modify process switch

Author: miedema-11

PWGUD/Tasks/flowCumulantsUpc.cxx

@@ -119,6 +119,9 @@ struct FlowCumulantsUpc {
   Configurable<std::vector<std::string>> cfgUserDefineGFWName{"cfgUserDefineGFWName", std::vector<std::string>{"Ch02Gap22", "Ch12Gap22"}, "User defined GFW Name"};
   Configurable<std::vector<int>> cfgRunRemoveList{"cfgRunRemoveList", std::vector<int>{-1}, "excluded run numbers"};
 
+  Configurable<int> cfgMaxEvents{"max-events", -1, "maximum number of events to process, -1 for all events"};
+  int processedEvents{0};
+
   ConfigurableAxis axisPtHist{"axisPtHist", {100, 0., 10.}, "pt axis for histograms"};
   ConfigurableAxis axisPt{"axisPt", {VARIABLE_WIDTH, 0.2, 0.25, 0.3, 0.35, 0.4, 0.45, 0.5, 0.55, 0.6, 0.65, 0.7, 0.75, 0.8, 0.85, 0.9, 0.95, 1, 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2, 2.2, 2.4, 2.6, 2.8, 3, 3.5, 4, 5, 6, 8, 10}, "pt axis for histograms"};
   ConfigurableAxis axisIndependent{"axisIndependent", {VARIABLE_WIDTH, 0, 5, 10, 20, 30, 40, 50, 60, 70, 80, 90}, "X axis for histograms"};
@@ -127,12 +130,12 @@ struct FlowCumulantsUpc {
   ConfigurableAxis axisDCAxy{"axisDCAxy", {200, -1, 1}, "DCA_{xy} (cm)"};
 
   // Added UPC Cuts
-  // SGSelector sgSelector;
-  // Configurable<float> cfgCutFV0{"cfgCutFV0", 50., "FV0A threshold"};
-  // Configurable<float> cfgCutFT0A{"cfgCutFT0A", 150., "FT0A threshold"};
-  // Configurable<float> cfgCutFT0C{"cfgCutFT0C", 50., "FT0C threshold"};
-  // Configurable<float> cfgCutZDC{"cfgCutZDC", 10., "ZDC threshold"};
-  // Configurable<float> cfgGapSideSelection{"cfgGapSideSelection", 2, "gap selection"};
+  SGSelector sgSelector;
+  Configurable<float> cfgCutFV0{"cfgCutFV0", 50., "FV0A threshold"};
+  Configurable<float> cfgCutFT0A{"cfgCutFT0A", 150., "FT0A threshold"};
+  Configurable<float> cfgCutFT0C{"cfgCutFT0C", 50., "FT0C threshold"};
+  Configurable<float> cfgCutZDC{"cfgCutZDC", 10., "ZDC threshold"};
+  Configurable<float> cfgGapSideSelection{"cfgGapSideSelection", 2, "gap selection"};
 
   // Corrections
   TH1D* mEfficiency = nullptr;
@@ -432,7 +435,6 @@ struct FlowCumulantsUpc {
     fGFWMC->AddRegion("olN", -0.8, -0.4, 1 + fPtAxis->GetNbins(), 4);
     fGFWMC->AddRegion("olN10", -0.8, -0.5, 1 + fPtAxis->GetNbins(), 4);
     fGFWMC->AddRegion("olfull", -0.8, 0.8, 1 + fPtAxis->GetNbins(), 4);
-    fGFWMC->CreateRegions();
 
     corrconfigs.push_back(fGFW->GetCorrelatorConfig("full {2 -2}", "ChFull22", kFALSE));
     corrconfigs.push_back(fGFW->GetCorrelatorConfig("full {3 -3}", "ChFull32", kFALSE));
@@ -526,9 +528,11 @@ struct FlowCumulantsUpc {
         }
         LOGF(info, "%d: %s %s", i, userDefineGFWCorr.at(i).c_str(), userDefineGFWName.at(i).c_str());
         corrconfigs.push_back(fGFW->GetCorrelatorConfig(userDefineGFWCorr.at(i).c_str(), userDefineGFWName.at(i).c_str(), kFALSE));
+        corrconfigsmc.push_back(fGFWMC->GetCorrelatorConfig(userDefineGFWCorr.at(i).c_str(), userDefineGFWName.at(i).c_str(), kFALSE));
       }
     }
     fGFW->CreateRegions();
+    fGFWMC->CreateRegions();
 
     if (cfgUseAdditionalEventCut) {
       fMultPVCutLow = new TF1("fMultPVCutLow", "[0]+[1]*x+[2]*x*x+[3]*x*x*x+[4]*x*x*x*x - 3.5*([5]+[6]*x+[7]*x*x+[8]*x*x*x+[9]*x*x*x*x)", 0, 100);
@@ -644,6 +648,8 @@ struct FlowCumulantsUpc {
     return;
   }
 
+  // ... 其余代码保持不变 ...
+
   void loadCorrections(uint64_t timestamp, int runNumber)
   {
     if (correctionsLoaded) {
@@ -950,7 +956,7 @@ struct FlowCumulantsUpc {
       registry.fill(HIST("eventCounterMC"), 0.5);
 
       // registry.fill(HIST("hEventCount"), 1.5);
-      float cent = 100;
+      float cent = 50;
       float vtxz = 0;
 
       vtxz = mcCollision.posZ();
@@ -990,6 +996,7 @@ struct FlowCumulantsUpc {
           std::cout << "mcParticle.isPhysicalPrimary() = " << mcParticle.isPhysicalPrimary() << std::endl;
           continue;
         }
+
         std::array<double, 3> momentum = {mcParticle.px(), mcParticle.py(), mcParticle.pz()};
         double energy = std::sqrt(momentum[0] * momentum[0] + momentum[1] * momentum[1] + momentum[2] * momentum[2] + massPion * massPion);
         ROOT::Math::LorentzVector<ROOT::Math::PxPyPzE4D<double>> protoMC(momentum[0], momentum[1], momentum[2], energy);
@@ -1036,6 +1043,7 @@ struct FlowCumulantsUpc {
         if (withinPtPOI && withinPtRef) {
           fGFWMC->Fill(eta, fPtAxis->FindBin(pt) - 1, phi, wacc * weff, 4);
         }
+        std::cout << "successfully filled" << std::endl;
       }
       registry.fill(HIST("hTrackCorrection2dMC"), mcParticles.size(), nTracksCorrected);
       // Filling Flow Container