JNUA Creator added

Author: nmallick19

PWGCF/JCorran/Tasks/jflucWeightsLoader.cxx

@@ -30,12 +30,14 @@
 #include "Framework/RunningWorkflowInfo.h"
 #include "Framework/runDataProcessing.h"
 #include "ReconstructionDataFormats/V0.h"
+#include "CommonConstants/MathConstants.h"
 
 #include <TFile.h>
 #include <THn.h>
 
 #include <experimental/type_traits>
 #include <string>
+#include <vector>
 
 using namespace o2;
 using namespace o2::framework;
@@ -243,7 +245,7 @@ struct JflucWeightsLoader {
   {
     loadWeights(output, collision, tracks);
   }
-  PROCESS_SWITCH(JflucWeightsLoader, processLoadWeightsCF, "Load weights histograms for CF derived data table", true);
+  PROCESS_SWITCH(JflucWeightsLoader, processLoadWeightsCF, "Load weights histograms for CF derived data table", false);
 
   Produces<aod::J2ProngWeights> output2p;
   void processLoadWeightsCF2Prong(aod::CFCollision const& collision, aod::CF2ProngTracks const& tracks2p)
@@ -253,7 +255,66 @@ struct JflucWeightsLoader {
   PROCESS_SWITCH(JflucWeightsLoader, processLoadWeightsCF2Prong, "Load weights histograms for CF derived 2-prong tracks data table", false);
 };
 
+struct JNUACreator {
+
+  O2_DEFINE_CONFIGURABLE(cfgPtMin, float, 0.2f, "Minimum pT used for track selection.");
+  O2_DEFINE_CONFIGURABLE(cfgPtMax, float, 5.0f, "Maximum pT used for track selection.");
+  O2_DEFINE_CONFIGURABLE(cfgEtaMax, float, 1.0f, "Maximum eta used for track selection.");
+  O2_DEFINE_CONFIGURABLE(cfgMinMultiplicity, int, 5, "Minimum number of particles required for the event to have.");
+  O2_DEFINE_CONFIGURABLE(cfgTrackBitMask, uint16_t, 0, "Track selection bitmask to use as defined in the filterCorrelations.cxx task");
+  
+  ConfigurableAxis cfgAxisMultiplicity{"axisMultiplicity", {VARIABLE_WIDTH, 0, 2.0, 5, 10, 20, 30, 40, 50, 60, 70, 80, 100.1}, "multiplicity / centrality axis for histograms"};
+  ConfigurableAxis cfgAxisPhi{"axisPhi", {50, 0.0, o2::constants::math::TwoPI}, "phi axis for histograms"};
+  ConfigurableAxis cfgAxisEta{"axisEta", {40, -2.0, 2.0}, "eta axis for histograms"};
+  ConfigurableAxis cfgAxisZVertex{"axisZVertex", {20, -10.0, 10.0}, "zvertex axis for histograms"};
+
+
+  //we read cffilter collisions and tracks -- zvtx, ptMin, eta, multiplicity, trackSelection cuts and bits
+  //come from the PWGCF/TableProducer/filterCorrelations.cxx task
+
+  HistogramRegistry qaHistRegistry{"qaHistRegistry", {}, OutputObjHandlingPolicy::AnalysisObject, true, true};
+
+  void init(InitContext const&)
+  {
+    qaHistRegistry.add("trackType", "trackType;trackType;counts", HistType::kTH1F, {{65, -0.5, 64.5}});
+    // NUA histogram structure matches loader: THnF with axes (mult, partType, phi, eta, z) - same order as GetBin(nuaCoords) in loader
+    const AxisSpec axisMult{cfgAxisMultiplicity, "multiplicity/centrality"};
+    const AxisSpec axisType{2, -0.5, 1.5, "type"}; // 0 = all charged hadrons (matches loader partType)
+    const AxisSpec axisPhi{cfgAxisPhi, "#varphi"};
+    const AxisSpec axisEta{cfgAxisEta, "#eta"};
+    const AxisSpec axisZVertex{cfgAxisZVertex, "z_{vtx} [cm]"};
+   
+    qaHistRegistry.add("NUACreation/h_phietaz", "(NUA) mult, type, phi, eta, z", HistType::kTHnF, {axisMult, axisType, axisPhi, axisEta, axisZVertex});
+  }
+
+  Filter derivedTracks = (nabs(aod::cftrack::eta) < cfgEtaMax) && (aod::cftrack::pt > cfgPtMin) && (aod::cftrack::pt < cfgPtMax) && ncheckbit(aod::track::trackType, as<uint8_t>(cfgTrackBitMask));
+
+  void processNUACreationCFDerived(aod::CFCollision const& collision, soa::Filtered<aod::CFTracks> const& tracks) {
+
+    if (tracks.size() < cfgMinMultiplicity) {
+      return; //reject if not enough tracks
+    }
+
+    const auto multiplicity = collision.multiplicity(); //this comes from the filterCorrelations.cxx task
+    if (multiplicity < 0. || multiplicity > 100.1) {
+      return; // match upper edge of cfgAxisMultiplicity (100.1)
+    }
+
+    const float posZ = collision.posZ();
+    // Fill NUA histogram with same coordinate order as loader: (mult, partType, phi, eta, z)
+    // partType = 0 for all charged hadrons (matches loader where partType is always 0 for now)
+    for (auto& track : tracks) {
+      qaHistRegistry.fill(HIST("NUACreation/h_phietaz"), multiplicity, 0.0f, track.phi(), track.eta(), posZ);
+      qaHistRegistry.fill(HIST("trackType"), track.trackType());
+    }
+  }
+
+  PROCESS_SWITCH(JNUACreator, processNUACreationCFDerived, "Create NUA histograms from CF derived data", false);
+};
+
 WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
 {
-  return WorkflowSpec{adaptAnalysisTask<JflucWeightsLoader>(cfgc)};
+  return WorkflowSpec{
+    adaptAnalysisTask<JflucWeightsLoader>(cfgc),
+    adaptAnalysisTask<JNUACreator>(cfgc)};
 }