[PWGDQ] add an option to produce mft table in muonGlobalAlignment.cxx

PWGDQ/Tasks/muonGlobalAlignment.cxx

@@ -13,12 +13,14 @@
 /// \brief Task to compute and evaluate DCA quantities
 /// \author Nicolas Bizé <nicolas.bize@cern.ch>, SUBATECH
 //
+
 #include "PWGDQ/Core/VarManager.h"
-#include "PWGDQ/DataModel/ReducedInfoTables.h"
+// #include "PWGDQ/DataModel/ReducedInfoTables.h"
 
 #include "Common/CCDB/EventSelectionParams.h"
 #include "Common/CCDB/RCTSelectionFlags.h"
 #include "Common/DataModel/EventSelection.h"
+#include "Common/DataModel/TrackSelectionTables.h"
 
 #include <CCDB/BasicCCDBManager.h>
 #include <CCDB/CcdbApi.h>
@@ -88,9 +90,9 @@
 
 using namespace o2::aod::rctsel;
 
-using MyReducedMuons = soa::Join<aod::ReducedMuons, aod::ReducedMuonsExtra, aod::ReducedMuonsCov>;
-using MyReducedEvents = soa::Join<aod::ReducedEvents, aod::ReducedEventsExtended>;
-using MyReducedEventsVtxCov = soa::Join<aod::ReducedEvents, aod::ReducedEventsExtended, aod::ReducedEventsVtxCov>;
+// using MyReducedMuons = soa::Join<aod::ReducedMuons, aod::ReducedMuonsExtra, aod::ReducedMuonsCov>;
+// using MyReducedEvents = soa::Join<aod::ReducedEvents, aod::ReducedEventsExtended>;
+// using MyReducedEventsVtxCov = soa::Join<aod::ReducedEvents, aod::ReducedEventsExtended, aod::ReducedEventsVtxCov>;
 
 using MyCollisions = aod::Collisions;
 using MyBCs = soa::Join<aod::BCs, aod::Timestamps, aod::BcSels>;
@@ -114,7 +116,22 @@
 using o2::dataformats::GlobalFwdTrack;
 using o2::track::TrackParCovFwd;
 
+namespace o2::aod
+{
+DECLARE_SOA_TABLE(CompactMFTTracks, "AOD", "COMPACTMFT", //! standalone table for studying alignment
+                  collision::PosX, collision::PosY, collision::PosZ,
+                  fwdtrack::Signed1Pt, fwdtrack::Tgl, fwdtrack::Phi,
+                  fwdtrack::FwdDcaX, fwdtrack::FwdDcaY, o2::aod::fwdtrack::CXX, o2::aod::fwdtrack::CYY, o2::aod::fwdtrack::CXY,
+                  fwdtrack::NClusters, fwdtrack::Chi2,
+                  fwdtrack::X, fwdtrack::Y, fwdtrack::Z);
+using CompactMFTTrack = CompactMFTTracks;
+} // namespace o2::aod
+
 struct muonGlobalAlignment {
+
+  Produces<aod::CompactMFTTracks> mftTable;
+  Configurable<bool> cfgProduceMFTTable{"cfgProduceMFTTable", false, "flag to produce MFTsa table"};
+
   ////   Variables for selecting MCH and MFT tracks
   Configurable<float> fTrackChi2MchUp{"cfgTrackChi2MchUp", 5.f, ""};
   Configurable<float> fPtMchLow{"cfgPtMchLow", 0.7f, ""};
@@ -255,7 +272,7 @@
                       std::map<uint64_t, CollisionInfo>& collisionInfos)
   {
     // fill collision information for global muon tracks (MFT-MCH-MID matches)
     for (auto muonTrack : muonTracks) {
       if (!muonTrack.has_collision())
         continue;
 
@@ -303,8 +320,8 @@
       return (track1.chi2MatchMCHMFT() < track2.chi2MatchMCHMFT());
     };
 
     for (auto& [collisionIndex, collisionInfo] : collisionInfos) {
       for (auto& [mchIndex, globalTracksVector] : collisionInfo.globalMuonTracks) {
         std::sort(globalTracksVector.begin(), globalTracksVector.end(), compareChi2);
       }
     }
@@ -319,7 +336,7 @@
     InitCollisions(collisions, bcs, muonTracks, collisionInfos);
 
     // fill collision information for MFT standalone tracks
     for (auto mftTrack : mftTracks) {
       if (!mftTrack.has_collision())
         continue;
 
@@ -1077,8 +1094,8 @@
     track.setBendingCoor(y + yCorrection);
     track.setBendingSlope(ySlope + ySlopeCorrection);
     /*
     std::cout << std::format("[TOTO] MFT position:    pos={:0.3f},{:0.3f}", x, y) << std::endl;
     std::cout << std::format("[TOTO] MFT corrections: pos={:0.3f},{:0.3f}  slope={:0.12f},{:0.12f}  angle={:0.12f},{:0.12f}",
         xCorrection, yCorrection, xSlopeCorrection, ySlopeCorrection,
         std::atan2(xSlopeCorrection, 1), std::atan2(ySlopeCorrection, 1)) << std::endl;
     */
@@ -1116,8 +1133,8 @@
   template <typename T>
   T UpdateTrackMomentum(const T& track, const double p, int sign)
   {
     double px = p * sin(M_PI / 2 - atan(track.tgl())) * cos(track.phi());
     double py = p * sin(M_PI / 2 - atan(track.tgl())) * sin(track.phi());
     double pt = std::sqrt(std::pow(px, 2) + std::pow(py, 2));
 
     SMatrix5 tpars = {track.x(), track.y(), track.phi(), track.tgl(), sign / pt};
@@ -1137,8 +1154,8 @@
   template <typename T>
   T UpdateTrackMomentum(const T& track, const o2::mch::TrackParam& track4mom)
   {
     double px = track4mom.p() * sin(M_PI / 2 - atan(track.tgl())) * cos(track.phi());
     double py = track4mom.p() * sin(M_PI / 2 - atan(track.tgl())) * sin(track.phi());
     double pt = std::sqrt(std::pow(px, 2) + std::pow(py, 2));
     double sign = track4mom.getCharge();
 
@@ -1452,6 +1469,14 @@
               registry.get<THnSparse>(HIST("DCA/MFT/DCA_x"))->Fill(dcax, collision.posZ(), mftTrack.x(), mftTrack.y(), mftNclusters);
               registry.get<THnSparse>(HIST("DCA/MFT/DCA_y"))->Fill(dcay, collision.posZ(), mftTrack.x(), mftTrack.y(), mftNclusters);
 
+              if (cfgProduceMFTTable) {
+                mftTable(collision.posX(), collision.posY(), collision.posZ(),
+                         mftTrack.signed1Pt(), mftTrack.tgl(), mftTrack.phi(),
+                         dcax, dcay, mftTrackAtDCA.getSigma2X(), mftTrackAtDCA.getSigma2Y(), mftTrackAtDCA.getSigmaXY(),
+                         mftNclusters, mftTrack.chi2(),
+                         mftTrack.x(), mftTrack.y(), mftTrack.z());
+              }
+
               if (fEnableMftDcaExtraPlots) {
                 if (mftNclusters >= 6) {
                   for (int i = 0; i < nMftLayers; i++) {