[PWGCF] Fill registry only when events are selected (#8940)

Co-authored-by: ALICE Action Bot <alibuild@cern.ch>

Author: cnkoster

PWGCF/Flow/TableProducer/zdcQVectors.cxx

@@ -176,6 +176,7 @@ struct ZdcQVectors {
     for (int step = 0; step < 6; step++) {
       registry.add<TH2>(Form("step%i/QA/hSPplaneA", step), "hSPplaneA", kTH2D, {{100, -4, 4}, axisCent10});
       registry.add<TH2>(Form("step%i/QA/hSPplaneC", step), "hSPplaneC", kTH2D, {{100, -4, 4}, axisCent10});
+      registry.add<TH2>(Form("step%i/QA/hSPplaneFull", step), "hSPplaneFull", kTH2D, {{100, -4, 4}, axisCent10});
       for (const auto& side : sides) {
         hQxvsQy[step] = registry.add<TH2>(Form("step%i/hZN%s_Qx_vs_Qy", step, side), Form("hZN%s_Qx_vs_Qy", side), kTH2F, {axisQ, axisQ});
       }
@@ -328,6 +329,14 @@ struct ZdcQVectors {
     registry.fill(HIST("step0/QA/hQXC_vs_vz"), v[2], q[0][0][2]);
     registry.fill(HIST("step0/QA/hQYC_vs_vz"), v[2], q[0][0][3]);
 
+    // add psi!!
+    double psiA = 1.0 * std::atan2(q[0][0][2], q[0][0][0]);
+    registry.fill(HIST("step0/QA/hSPplaneA"), psiA, centrality, 1);
+    double psiC = 1.0 * std::atan2(q[0][0][3], q[0][0][1]);
+    registry.fill(HIST("step0/QA/hSPplaneC"), psiC, centrality, 1);
+    double psiFull = 1.0 * std::atan2(q[0][0][2] + q[0][0][3], q[0][0][0] + q[0][0][1]);
+    registry.fill(HIST("step0/QA/hSPplaneFull"), psiFull, centrality, 1);
+
     static constexpr std::string_view SubDir[] = {"step1/", "step2/", "step3/", "step4/", "step5/"};
     static_for<0, 4>([&](auto Ind) {
       constexpr int Index = Ind.value;
@@ -361,11 +370,12 @@ struct ZdcQVectors {
       registry.fill(HIST(SubDir[Index]) + HIST("QA/hQXC_vs_vz"), v[2], q[iteration][indexRt][2]);
       registry.fill(HIST(SubDir[Index]) + HIST("QA/hQYC_vs_vz"), v[2], q[iteration][indexRt][3]);
 
-      // add psi!!
-      double psiA = 1.0 * std::atan2(q[iteration][indexRt][2], q[iteration][indexRt][0]);
+      psiA = 1.0 * std::atan2(q[iteration][indexRt][2], q[iteration][indexRt][0]);
       registry.fill(HIST(SubDir[Index]) + HIST("QA/hSPplaneA"), psiA, centrality, 1);
-      double psiC = 1.0 * std::atan2(q[iteration][indexRt][3], q[iteration][indexRt][1]);
+      psiC = 1.0 * std::atan2(q[iteration][indexRt][3], q[iteration][indexRt][1]);
       registry.fill(HIST(SubDir[Index]) + HIST("QA/hSPplaneC"), psiC, centrality, 1);
+      psiFull = 1.0 * std::atan2(q[iteration][indexRt][2] + q[iteration][indexRt][3], q[iteration][indexRt][0] + q[iteration][indexRt][1]);
+      registry.fill(HIST(SubDir[Index]) + HIST("QA/hSPplaneFull"), psiFull, centrality, 1);
     });
   }
 
@@ -456,7 +466,7 @@ struct ZdcQVectors {
     } else if (hist->InheritsFrom("TProfile")) {
       TProfile* h = reinterpret_cast<TProfile*>(hist);
       TString name = h->GetName();
-      int bin{};
+      int bin;
       if (name.Contains("mean_vx"))
         bin = h->GetXaxis()->FindBin(v[0]);
       if (name.Contains("mean_vy"))
@@ -471,20 +481,17 @@ struct ZdcQVectors {
     } else if (hist->InheritsFrom("THnSparse")) {
       std::vector<int> sparsePars;
       THnSparseD* h = reinterpret_cast<THnSparseD*>(hist);
-      if (step == 0 && iteration > 0) {
-        // Axis(0) is runnuber, but we don't need this
-        sparsePars.push_back(h->GetAxis(1)->FindBin(centrality));
-        sparsePars.push_back(h->GetAxis(2)->FindBin(v[0]));
-        sparsePars.push_back(h->GetAxis(3)->FindBin(v[1]));
-        sparsePars.push_back(h->GetAxis(4)->FindBin(v[2]));
-      }
+      sparsePars.push_back(h->GetAxis(0)->FindBin(centrality));
+      sparsePars.push_back(h->GetAxis(1)->FindBin(v[0]));
+      sparsePars.push_back(h->GetAxis(2)->FindBin(v[1]));
+      sparsePars.push_back(h->GetAxis(3)->FindBin(v[2]));
 
-      for (std::size_t i = 0; i < sparsePars.size() + 1; i++) {
-        h->GetAxis(i + 1)->SetRange(sparsePars[i], sparsePars[i]);
+      for (std::size_t i = 0; i < sparsePars.size(); i++) {
+        h->GetAxis(i)->SetRange(sparsePars[i], sparsePars[i]);
       }
-      calibConstant = h->Projection(5)->GetMean();
+      calibConstant = h->Projection(4)->GetMean();
 
-      if (h->Projection(sparsePars.size())->GetEntries() < cfgMinEntriesSparseBin) {
+      if (h->Projection(4)->GetEntries() < cfgMinEntriesSparseBin) {
         LOGF(debug, "1 entry in sparse bin! Not used... (increase binsize)");
         calibConstant = 0;
         isSelected = false;
@@ -694,7 +701,8 @@ struct ZdcQVectors {
     if (cal.atIteration == 0) {
       if (counter < 1)
         LOGF(warning, "Calibation files missing!!! Output created with q-vectors right after energy gain eq. !!");
-      fillAllRegistries(0, 0);
+      if (isSelected)
+        fillAllRegistries(0, 0);
       spTableZDC(runnumber, centrality, v[0], v[1], v[2], q[0][0][0], q[0][0][1], q[0][0][2], q[0][0][3], isSelected, 0, 0);
       counter++;
       return;
@@ -722,8 +730,10 @@ struct ZdcQVectors {
 
       if (counter < 1)
         LOGF(info, "Output created with q-vectors at iteration %i and step %i!!!!", cal.atIteration, cal.atStep + 1);
-      fillAllRegistries(cal.atIteration, cal.atStep + 1);
-      registry.fill(HIST("QA/centrality_after"), centrality);
+      if (isSelected) {
+        fillAllRegistries(cal.atIteration, cal.atStep + 1);
+        registry.fill(HIST("QA/centrality_after"), centrality);
+      }
       spTableZDC(runnumber, centrality, v[0], v[1], v[2], q[cal.atIteration][cal.atStep][0], q[cal.atIteration][cal.atStep][1], q[cal.atIteration][cal.atStep][2], q[cal.atIteration][cal.atStep][3], isSelected, cal.atIteration, cal.atStep);
       counter++;
       return;