Adding processes for realigned muons in track-to-collision association

Author: mcoquet642

Common/DataModel/FwdTrackReAlignTables.h

@@ -72,6 +72,19 @@ using FwdTrackRealign = FwdTracksReAlign::iterator;
 using FwdTrkCovRealign = FwdTrksCovReAlign::iterator;
 using FullFwdTracksRealign = soa::Join<FwdTracksReAlign, FwdTrksCovReAlign>;
 using FullFwdTrackRealign = FullFwdTracksRealign::iterator;
+
+// ambiguity table for realigned muons
+namespace fwdtrackrealignambiguous
+{
+DECLARE_SOA_INDEX_COLUMN_FULL(FwdTrackRealign, fwdTrackRealign, int, FwdTracksReAlign, ""); //! FwdTracksReAlign index
+DECLARE_SOA_SLICE_INDEX_COLUMN(BC, bc);
+
+} // namespace fwdtrackrealignambiguous
+
+DECLARE_SOA_TABLE(AmbiguousFwdTrksReAlign, "AOD", "AMBIFWDREALIGN", //! Table for FwdTracksReAlign which are not associated with a collision
+                  o2::soa::Index<>, fwdtrackrealignambiguous::FwdTrackRealignId, fwdtrackrealignambiguous::BCIdSlice);
+
+using AmbiguousFwdTrkRealign = AmbiguousFwdTrksReAlign::iterator;
 } // namespace o2::aod
 
 #endif // COMMON_DATAMODEL_FWDTRACKREALIGNTABLES_H_

Common/TableProducer/fwdtrackToCollisionAssociator.cxx

@@ -16,6 +16,7 @@
 
 #include "Common/Core/CollisionAssociation.h"
 #include "Common/DataModel/CollisionAssociationTables.h"
+#include "Common/DataModel/FwdTrackReAlignTables.h"
 
 #include <Framework/AnalysisDataModel.h>
 #include <Framework/AnalysisHelpers.h>
@@ -44,6 +45,7 @@ struct FwdTrackToCollisionAssociation {
   CollisionAssociation<false> collisionAssociator;
 
   Preslice<FwdTracks> muonsPerCollisions = aod::fwdtrack::collisionId;
+  Preslice<FwdTracksReAlign> realignmuonsPerCollisions = aod::fwdtrack::collisionId;
   Preslice<MFTTracks> mftsPerCollisions = aod::fwdtrack::collisionId;
 
   void init(InitContext const&)
@@ -55,7 +57,7 @@ struct FwdTrackToCollisionAssociation {
       LOGP(fatal, "Exactly one process function between standard and time-based association should be enabled!");
     }
 
-    if (!(doprocessMFTAssocWithTime || doprocessMFTStandardAssoc || doprocessFwdAssocWithTime || doprocessFwdStandardAssoc)) {
+    if (!(doprocessMFTAssocWithTime || doprocessMFTStandardAssoc || doprocessFwdAssocWithTime || doprocessFwdStandardAssoc || doprocessFwdRealignAssocWithTime || doprocessFwdRealignStandardAssoc)) {
       LOGP(fatal, "At least one process function should be enabled!");
     }
 
@@ -85,6 +87,22 @@ struct FwdTrackToCollisionAssociation {
   }
   PROCESS_SWITCH(FwdTrackToCollisionAssociation, processFwdStandardAssoc, "Use standard fwdtrack-to-collision association", false);
 
+  void processFwdRealignAssocWithTime(Collisions const& collisions,
+                                      FwdTracksReAlign const& muons,
+                                      AmbiguousFwdTrksReAlign const& ambiTracksFwd,
+                                      BCs const& bcs)
+  {
+    collisionAssociator.runAssocWithTime(collisions, muons, muons, ambiTracksFwd, bcs, fwdassociation, fwdreverseIndices);
+  }
+  PROCESS_SWITCH(FwdTrackToCollisionAssociation, processFwdRealignAssocWithTime, "Use fwdrealigntrack-to-collision association based on time", false);
+
+  void processFwdRealignStandardAssoc(Collisions const& collisions,
+                                      FwdTracksReAlign const& muons)
+  {
+    collisionAssociator.runStandardAssoc(collisions, muons, realignmuonsPerCollisions, fwdassociation, fwdreverseIndices);
+  }
+  PROCESS_SWITCH(FwdTrackToCollisionAssociation, processFwdRealignStandardAssoc, "Use standard fwdrealigntrack-to-collision association", false);
+
   void processMFTAssocWithTime(Collisions const& collisions,
                                MFTTracks const& tracks,
                                AmbiguousMFTTracks const& ambiguousTracks,