[Common] add dipole shift correction to muon QA task (#16440)

Author: aferrero2707

Common/Tasks/qaMuon.cxx

@@ -223,6 +223,7 @@ struct muonQa {
     Configurable<double> fChamberResolutionX{"cfgChamberResolutionX", 0.4, "Chamber resolution along X configuration for refit"}; // 0.4cm pp, 0.2cm PbPb
     Configurable<double> fChamberResolutionY{"cfgChamberResolutionY", 0.4, "Chamber resolution along Y configuration for refit"}; // 0.4cm pp, 0.2cm PbPb
     Configurable<double> fSigmaCutImprove{"cfgSigmaCutImprove", 6., "Sigma cut for track improvement"};
+    Configurable<double> fDipoleZcorr{"cfgDipoleZcorr", 0.0f, "Correction to the dipole z position"};
   } configRealign;
 
   ///    Variables to event mixing criteria
@@ -1651,6 +1652,11 @@ struct muonQa {
       fgValues.errorClusters.emplace_back(eCls);
       fgValues.DEIDs.emplace_back(cluster.deId());
 
+      // subtract the dipole shift correction from the cluster z position
+      if (configRealign.fDipoleZcorr != 0) {
+        clusterMCH->z -= configRealign.fDipoleZcorr;
+      }
+
       // Add transformed cluster into temporary variable
       convertedTrack.createParamAtCluster(*clusterMCH);
     }
@@ -1663,6 +1669,12 @@ struct muonQa {
         LOGF(fatal, "Muon track %d has no associated clusters.", muon.globalIndex());
       }
 
+      // add back the dipole shift correction to the track z
+      if (configRealign.fDipoleZcorr != 0) {
+        auto& trackParam = *(convertedTrack.begin());
+        trackParam.setZ(trackParam.getZ() + configRealign.fDipoleZcorr);
+      }
+
       for (auto it = convertedTrack.begin(); it != convertedTrack.end(); it++) {
         std::vector<float> pos = {static_cast<float>(it->getNonBendingCoor()), static_cast<float>(it->getBendingCoor()), static_cast<float>(it->getZ())};
         fgValues.posClusters.emplace_back(pos);