Refactor includes and update process functions

Author: miedema-11

PWGUD/Tasks/flowMcUpc.cxx

@@ -14,31 +14,27 @@
 /// \since  Apr/2/2026
 /// \brief  flow efficiency analysis on UPC MC
 
+#include <CCDB/BasicCCDBManager.h>
 #include "PWGUD/Core/SGSelector.h"
 #include "PWGUD/DataModel/UDTables.h"
-
-#include "Common/Core/RecoDecay.h"
-#include "Common/Core/TrackSelection.h"
-#include "Common/Core/TrackSelectionDefaults.h"
-#include "Common/Core/trackUtilities.h"
-#include "Common/DataModel/TrackSelectionTables.h"
-
-#include "Framework/ASoAHelpers.h"
+#include <vector>
+#include <string>
 #include "Framework/AnalysisDataModel.h"
 #include "Framework/AnalysisTask.h"
 #include "Framework/HistogramRegistry.h"
-#include "Framework/RunningWorkflowInfo.h"
 #include "Framework/runDataProcessing.h"
+#include "Framework/ASoAHelpers.h"
+#include "Framework/RunningWorkflowInfo.h"
+#include "Common/DataModel/TrackSelectionTables.h"
+#include "Common/Core/RecoDecay.h"
+#include "Common/Core/TrackSelection.h"
+#include "Common/Core/TrackSelectionDefaults.h"
+#include "Common/Core/trackUtilities.h"
 #include "ReconstructionDataFormats/Track.h"
-#include <CCDB/BasicCCDBManager.h>
-
-#include <TF1.h>
-#include <TPDGCode.h>
 #include <TProfile.h>
 #include <TRandom3.h>
-
-#include <string>
-#include <vector>
+#include <TPDGCode.h>
+#include <TF1.h>
 
 using namespace o2;
 using namespace o2::framework;
@@ -49,8 +45,6 @@ using namespace o2::framework::expressions;
 struct FlowMcUpc {
   HistogramRegistry histos{"Histos", {}, OutputObjHandlingPolicy::AnalysisObject};
 
-  // using BCs = soa::Join<aod::BCsWithTimestamps, aod::BcSels, aod::Run3MatchedToBCSparse>;
-
   Configurable<float> minB{"minB", 0.0f, "min impact parameter"};
   Configurable<float> maxB{"maxB", 20.0f, "max impact parameter"};
   O2_DEFINE_CONFIGURABLE(cfgCutVertex, float, 10.0f, "Accepted z-vertex range")
@@ -60,6 +54,7 @@ struct FlowMcUpc {
   O2_DEFINE_CONFIGURABLE(cfgCutDCAxy, float, 0.2f, "DCAxy cut for tracks")
   O2_DEFINE_CONFIGURABLE(cfgDcaxy, bool, true, "choose dcaxy")
 
+
   ConfigurableAxis axisB{"axisB", {100, 0.0f, 20.0f}, ""};
   ConfigurableAxis axisPt{"axisPt", {VARIABLE_WIDTH, 0.0f, 0.1f, 0.2f, 0.3f, 0.4f, 0.5f, 0.6f, 0.7f, 0.8f, 0.9f, 1.0f, 1.1f, 1.2f, 1.3f, 1.4f, 1.5f, 1.6f, 1.7f, 1.8f, 1.9f, 2.0f, 2.2f, 2.4f, 2.6f, 2.8f, 3.0f, 3.2f, 3.4f, 3.6f, 3.8f, 4.0f, 4.4f, 4.8f, 5.2f, 5.6f, 6.0f, 6.5f, 7.0f, 7.5f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f}, "pt axis"};
   // Connect to ccdb
@@ -70,6 +65,7 @@ struct FlowMcUpc {
 
   using McParticles = soa::Join<aod::UDMcParticles, aod::UDMcTrackLabels>;
 
+
   void init(InitContext&)
   {
     ccdb->setURL(ccdbUrl.value);
@@ -89,14 +85,15 @@ struct FlowMcUpc {
     histos.add<TH3>("hEtaPtVtxzMCGen", "Monte Carlo Truth; #eta; p_{T} (GeV/c); V_{z} (cm);", {HistType::kTH3D, {axisEta, axisPt, axisVertex}});
     histos.add<TH1>("hPtReco", "Monte Carlo Reco Global; pT (GeV/c);", {HistType::kTH1D, {axisPt}});
     histos.add<TH3>("hEtaPtVtxzMCReco", "Monte Carlo Global; #eta; p_{T} (GeV/c); V_{z} (cm);", {HistType::kTH3D, {axisEta, axisPt, axisVertex}});
-  }
 
-  template <typename TCollision>
-  bool eventSelected(TCollision collision)
-  {
-    return true;
   }
 
+  // template <typename TCollision>
+  // bool eventSelected(TCollision collision)
+  // {
+  //   return true;
+  // }
+
   template <typename TTrack>
   bool trackSelected(TTrack track)
   {
@@ -112,11 +109,11 @@ struct FlowMcUpc {
     return true;
   }
 
-  void processMCTrue(aod::UDMcCollisions::iterator const& mcCollision, McParticles const& mcParticles)
+  void processMCTrue(aod::UDMcCollisions::iterator const& mcCollision, McParticles const& mcParticles, aod::BCs const& bcs)
   {
-    // if (bcs.size() == 0) {
-    //   return;
-    // }
+    if (bcs.size() == 0) {
+      return;
+    }
     histos.fill(HIST("mcEventCounter"), 0.5);
     float imp = mcCollision.impactParameter();
     float vtxz = mcCollision.posZ();
@@ -139,22 +136,23 @@ struct FlowMcUpc {
           continue;
         // if (std::fabs(mcParticle.eta()) > cfgCutEta) // main acceptance
         //   continue;
-
+        
         histos.fill(HIST("hPtMCGen"), pt);
         histos.fill(HIST("hEtaPtVtxzMCGen"), eta, pt, vtxz);
       }
     }
+
   }
   PROCESS_SWITCH(FlowMcUpc, processMCTrue, "process pure simulation information", true);
 
   using MCRecoTracks = soa::Join<aod::UDTracks, aod::UDTracksPID, aod::UDTracksExtra, aod::UDTracksFlags, aod::UDTracksDCA, aod::UDMcTrackLabels>;
-  using MCRecoCollisions = soa::Join<aod::UDCollisions, aod::UDCollisionsSels, aod::UDMcCollsLabels>;
+  using MCRecoCollisions = soa::Join<aod::UDCollisions, aod::SGCollisions, aod::UDCollisionSelExtras, aod::UDCollisionsSels, aod::UDZdcsReduced, aod::UDMcCollsLabels>;
 
   void processReco(MCRecoCollisions::iterator const& collision, MCRecoTracks const& tracks)
   {
     histos.fill(HIST("RecoProcessEventCounter"), 0.5);
-    if (!eventSelected(collision))
-      return;
+    // if (!eventSelected(collision))
+    //   return;
     histos.fill(HIST("RecoProcessEventCounter"), 1.5);
     if (!collision.has_udMcCollision())
       return;
@@ -167,9 +165,9 @@ struct FlowMcUpc {
 
     for (const auto& track : tracks) {
       auto momentum = std::array<double, 3>{track.px(), track.py(), track.pz()};
-      double pt = RecoDecay::pt(momentum);
-      double phi = RecoDecay::phi(momentum);
-      double eta = RecoDecay::eta(momentum);
+        double pt = RecoDecay::pt(momentum);
+        double phi = RecoDecay::phi(momentum);
+        double eta = RecoDecay::eta(momentum);
       if (!trackSelected(track) || (!track.has_udMcParticle()))
         continue;
       auto mcParticle = track.udMcParticle();
@@ -180,7 +178,7 @@ struct FlowMcUpc {
       //   continue;
       if (!mcParticle.isPhysicalPrimary())
         continue;
-
+      
       histos.fill(HIST("hPtReco"), pt);
       histos.fill(HIST("hEtaPtVtxzMCReco"), eta, pt, vtxz);
     }