Add files via upload

Author: YazhenLin

PWGDQ/Tasks/dqEnergyCorrelator_direct.cxx

@@ -116,6 +116,8 @@ struct AnalysisEnergyCorrelator {
     Configurable<float> fConfigJpsiPtMin{"cfgJpsiPtMin", 0.0, "J/psi pt minimum"};
     Configurable<float> fConfigJpsiPtMax{"cfgJpsiPtMax", 100.0, "J/psi pt maximum"};
     Configurable<float> fConfigJpsiRapMax{"cfgJpsiRapMax", 0.9, "J/psi rapidity maximum"};
+    Configurable<float> fConfigJpsiTauxyMin{"cfgJpsiTauxyMin", -10000, "J/psi tauxy minimum"};
+    Configurable<float> fConfigJpsiTauxyMax{"cfgJpsiTauxyMax", 10000.0, "J/psi tauxy maximum"};
     Configurable<std::string> fConfigAddSEPHistogram{"cfgAddSEPHistogram", "", "Comma separated list of histograms"};
     Configurable<std::string> recSignals{"cfgMCRecSignals", "eeFromJpsi", "Comma separated list of MC signals (reconstructed)"};
     Configurable<std::string> recSignalsJSON{"cfgMCRecSignalsJSON", "", "Comma separated list of MC signals (reconstructed) via JSON"};
@@ -627,8 +629,8 @@ struct AnalysisEnergyCorrelator {
   }
 
   // Template function to run same event pairing (barrel-barrel, muon-muon, barrel-muon)
-  template <uint32_t TTrackFillMap, int TPairType, typename TTrack1, typename TTrack2>
-  void runSameEventPairing(TTrack1 const& t1, TTrack2 const& t2, int iEleCut, aod::McParticles const& /*mcParticles*/)
+  template <uint32_t TTrackFillMap, int TPairType, typename TTrack1, typename TTrack2, typename TEvent>
+  void runSameEventPairing(TTrack1 const& t1, TTrack2 const& t2, int iEleCut, TEvent const& event, aod::McParticles const& /*mcParticles*/)
   {
     std::map<int, std::vector<TString>> histNames = fTrackHistNames;
     std::map<int, std::vector<TString>> histNamesMC = fBarrelHistNamesMCmatched;
@@ -647,6 +649,7 @@ struct AnalysisEnergyCorrelator {
     } // end loop over MC signals
 
     VarManager::FillPair<TPairType, TTrackFillMap>(t1, t2);
+    VarManager::FillPairVertexing<TPairType, gkEventFillMapWithMults, TTrackFillMap>(event, t1, t2, false, VarManager::fgValues);
 
     if (sign1 * sign2 < 0) {                                                       // +- pairs
       fHistMan->FillHistClass(histNames[iEleCut][0].Data(), VarManager::fgValues); // reconstructed, unmatched
@@ -802,18 +805,20 @@ struct AnalysisEnergyCorrelator {
           // Fill pair histograms for all cuts that both tracks pass
           for (size_t iCut = 0; iCut < fTrackCuts.size(); iCut++) {
             if (twoTrackFilter & (static_cast<uint32_t>(1) << iCut)) {
-              runSameEventPairing<gkTrackFillMapWithCov, VarManager::kDecayToEE>(t1, t2, iCut, mcParticles);
+              runSameEventPairing<gkTrackFillMapWithCov, VarManager::kDecayToEE>(t1, t2, iCut, event, mcParticles);
             }
           }
 
           float mass = VarManager::fgValues[VarManager::kMass];
           float pt = VarManager::fgValues[VarManager::kPt];
           float rap = VarManager::fgValues[VarManager::kRap];
+          float tauxy = VarManager::fgValues[VarManager::kVertexingTauxyProjectedPoleJPsiMass];
 
           // Apply J/psi cuts
           if (mass < fConfigPairOptions.fConfigJpsiMassMin.value || mass > fConfigPairOptions.fConfigJpsiMassMax.value ||
               pt < fConfigPairOptions.fConfigJpsiPtMin.value || pt > fConfigPairOptions.fConfigJpsiPtMax.value ||
-              std::abs(rap) > fConfigPairOptions.fConfigJpsiRapMax.value) {
+              std::abs(rap) > fConfigPairOptions.fConfigJpsiRapMax.value ||
+              tauxy < fConfigPairOptions.fConfigJpsiTauxyMin.value || tauxy > fConfigPairOptions.fConfigJpsiTauxyMax.value) {
             continue;
           }
 
@@ -928,13 +933,17 @@ struct AnalysisEnergyCorrelator {
           }
           // Fill pair variables for cut
           VarManager::FillPair<VarManager::kDecayToEE, gkTrackFillMapWithCov>(t1, t2, VarManager::fgValues);
+          VarManager::FillPairVertexing<VarManager::kDecayToEE, gkEventFillMapWithMults, gkTrackFillMapWithCov>(event1, t1, t2, false, VarManager::fgValues);
           float mass = VarManager::fgValues[VarManager::kMass];
           float pt = VarManager::fgValues[VarManager::kPt];
           float rap = VarManager::fgValues[VarManager::kRap];
+          float tauxy = VarManager::fgValues[VarManager::kVertexingTauxyProjectedPoleJPsiMass];
+
           // Apply J/psi cuts
           if (mass < fConfigPairOptions.fConfigJpsiMassMin.value || mass > fConfigPairOptions.fConfigJpsiMassMax.value ||
               pt < fConfigPairOptions.fConfigJpsiPtMin.value || pt > fConfigPairOptions.fConfigJpsiPtMax.value ||
-              std::abs(rap) > fConfigPairOptions.fConfigJpsiRapMax.value) {
+              std::abs(rap) > fConfigPairOptions.fConfigJpsiRapMax.value ||
+              tauxy < fConfigPairOptions.fConfigJpsiTauxyMin.value || tauxy > fConfigPairOptions.fConfigJpsiTauxyMax.value) {
             continue;
           }
           if (t1.sign() * t2.sign() >= 0) {

PWGDQ/Tasks/tableReader_withAssoc.h

@@ -4133,7 +4133,7 @@ struct AnalysisDileptonTrack {
   o2::framework::Configurable<bool> fConfigApplyMassEC{"cfgApplyMassEC", false, "Apply fit mass for sideband for the energy correlator study"};
   o2::framework::Configurable<std::vector<float>> fConfigFitmassEC{"cfgTFitmassEC", std::vector<float>{-0.541438, 2.8, 3.2}, "parameter from the fit fuction and fit range"};
   o2::framework::Configurable<std::vector<float>> fConfigTransRange{"cfgTransRange", std::vector<float>{0.333333, 0.666667}, "Transverse region for the energy correlstor analysis"};
-
+  o2::framework::Configurable<int> fConfigbootstrapRN{"cfgBootstrapRN", -1, "Number of bootstrap samples for the energy correlator study"};
   o2::framework::Configurable<bool> fConfigApplyEfficiency{"cfgApplyEfficiency", false, "If true, apply efficiency correction for the energy correlator study"};
   o2::framework::Configurable<bool> fConfigApplyEfficiencyME{"cfgApplyEfficiencyME", false, "If true, apply efficiency correction for the energy correlator study"};
   o2::framework::Configurable<std::string> fConfigAccCCDBPath{"AccCCDBPath", "Users/y/yalin/pptest/test2", "Path of the efficiency corrections"};
@@ -4176,6 +4176,9 @@ struct AnalysisDileptonTrack {
   TH2F* hEfficiency_dilepton = nullptr;
   TH2F* hEfficiency_hadron = nullptr;
   TH1F* hMasswindow = nullptr;
+  TRandom3 fRndm{0};
+  using BootstrapEventKey = std::pair<uint64_t, uint64_t>;
+  std::map<BootstrapEventKey, int> fBootstrapSampleMap;
 
   void init(o2::framework::InitContext& context)
   {
@@ -4459,6 +4462,24 @@ struct AnalysisDileptonTrack {
     return hEff->Interpolate(safeX, safeY);
   }
 
+  template <typename TEvent>
+  int getBootstrapSample(TEvent const& event)
+  {
+    const BootstrapEventKey key{
+      static_cast<uint64_t>(event.timestamp()),
+      static_cast<uint64_t>(event.globalIndex())};
+
+    if (auto it = fBootstrapSampleMap.find(key);
+        it != fBootstrapSampleMap.end()) {
+      return it->second;
+    }
+
+    const int sample = static_cast<int>(fRndm.Rndm() * fConfigbootstrapRN.value);
+    fBootstrapSampleMap.emplace(key, sample);
+
+    return sample;
+  }
+
   // Template function to run pair - hadron combinations
   template <int TCandidateType, uint32_t TEventFillMap, uint32_t TTrackFillMap, typename TEvent, typename TTracks, typename TTrackAssocs, typename TDileptons>
   void runDileptonHadron(TEvent const& event, TTrackAssocs const& assocs, TTracks const& tracks, TDileptons const& dileptons)
@@ -4468,6 +4489,10 @@ struct AnalysisDileptonTrack {
     VarManager::FillEvent<TEventFillMap>(event, fValuesHadron);
     VarManager::FillEvent<TEventFillMap>(event, fValuesDilepton);
 
+    if (fConfigbootstrapRN.value > 0) {
+      const int bootstrapSample = getBootstrapSample(event);
+      fValuesHadron[VarManager::kECBootstrapSample] = bootstrapSample;
+    }
     for (auto const& dilepton : dileptons) {
       // get full track info of tracks based on the index
       auto lepton1 = tracks.rawIteratorAt(dilepton.index0Id());
@@ -4740,6 +4765,10 @@ struct AnalysisDileptonTrack {
       auto evAssocs = assocs.sliceBy(trackAssocsPerCollision, event2.globalIndex());
       evAssocs.bindExternalIndices(&events);
 
+      if (fConfigbootstrapRN.value > 0) {
+        const int bootstrapSample = getBootstrapSample(event1);
+        dqtablereader_helpers::varValues()[VarManager::kECBootstrapSample] = bootstrapSample;
+      }
       // loop over associations
       for (auto const& assoc : evAssocs) {