Add TRD gain calibration

martenole · martenole · commit e44b773d54bc · 2023-07-06T17:26:40.000+02:00
diff --git a/Detectors/TRD/workflow/src/TrackBasedCalibSpec.cxx b/Detectors/TRD/workflow/src/TrackBasedCalibSpec.cxx
@@ -141,9 +141,9 @@ DataProcessorSpec getTRDTrackBasedCalibSpec(o2::dataformats::GlobalTrackID::mask
   if (GTrackID::includesSource(GTrackID::Source::TPC, src)) {
     LOGF(info, "Found TPC tracks as input, loading TPC-TRD");
     srcTrk |= GTrackID::getSourcesMask("TPC-TRD");
-    if (gain) {
-      srcTrk |= GTrackID::getSourcesMask("TPC");
-    }
+  }
+  if (gain) {
+    srcTrk |= GTrackID::getSourcesMask("TPC");
   }
   GTrackID::mask_t srcClu = GTrackID::getSourcesMask("TRD");         // we don't need all clusters, only TRD tracklets
   dataRequest->requestTracks(srcTrk, false);
diff --git a/prodtests/full-system-test/aggregator-workflow.sh b/prodtests/full-system-test/aggregator-workflow.sh
@@ -64,6 +64,7 @@ if [[ "${GEN_TOPO_VERBOSE:-}" == "1" ]]; then
   echo "CALIB_PHS_RUNBYRUNCALIB = $CALIB_PHS_RUNBYRUNCALIB" 1>&2
   echo "CALIB_PHS_L1PHASE = $CALIB_PHS_L1PHASE" 1>&2
   echo "CALIB_TRD_VDRIFTEXB = $CALIB_TRD_VDRIFTEXB" 1>&2
+  echo "CALIB_TRD_GAIN = $CALIB_TRD_GAIN" 1>&2
   echo "CALIB_TPC_TIMEGAIN = $CALIB_TPC_TIMEGAIN" 1>&2
   echo "CALIB_TPC_RESPADGAIN = $CALIB_TPC_RESPADGAIN" 1>&2
   echo "CALIB_TPC_SCDCALIB = $CALIB_TPC_SCDCALIB" 1>&2
@@ -211,8 +212,15 @@ if [[ $AGGREGATOR_TASKS == BARREL_TF ]] || [[ $AGGREGATOR_TASKS == ALL ]]; then
     add_W o2-tpc-vdrift-tgl-calibration-workflow ""
   fi
   # TRD
+  TRD_CALIB_CONFIG=
   if [[ $CALIB_TRD_VDRIFTEXB == 1 ]]; then
-    add_W o2-calibration-trd-workflow "--vDriftAndExB"
+    TRD_CALIB_CONFIG+=" --vDriftAndExB"
+  fi
+  if [[ $CALIB_TRD_GAIN == 1 ]]; then
+    TRD_CALIB_CONFIG+=" --gain"
+  fi
+  if [[ ! -z ${TRD_CALIB_CONFIG} ]]; then
+    add_W o2-calibration-trd-workflow "${TRD_CALIB_CONFIG}"
   fi
 fi
 
diff --git a/prodtests/full-system-test/dpl-workflow.sh b/prodtests/full-system-test/dpl-workflow.sh
@@ -158,6 +158,7 @@ fi
 
 
 workflow_has_parameter CALIB && [[ $CALIB_TRD_VDRIFTEXB == 1 ]] && TRD_CONFIG+=" --enable-vdexb-calib"
+workflow_has_parameter CALIB && [[ $CALIB_TRD_GAIN == 1 ]] && TRD_CONFIG+=" --enable-gain-calib"
 ! has_detector FT0 && TRD_CONFIG+=" --disable-ft0-pileup-tagging"
 
 SEND_ITSTPC_DTGL=
