Cleanup

Author: gianniliveraro

PWGLF/TableProducer/Strangeness/sigma0builder.cxx

@@ -407,15 +407,17 @@ struct sigma0builder {
         histos.fill(HIST("Gen/hPrimaryPi0s"), 0);
         if (isPrimary) histos.fill(HIST("Gen/hPrimaryPi0s"), 1);
 
-        pi0Gens(ptmc); // optional table to store generated pi0 candidates. Be careful, this is a large table!
-        pi0GenCollRefs(mccollisionid); // link to stramccollision table
+        if (fillPi0Tables){
+          pi0Gens(ptmc); // optional table to store generated pi0 candidates. Be careful, this is a large table!
+          pi0GenCollRefs(mccollisionid); // link to stramccollision table
+        }        
       }  
 
       // Sigma0-specific
       if (isSigma0 || isAntiSigma0){     
 
         // Checking decay mode
-        auto daughtersIDs = mcParticle.daughtersIds();
+        //auto daughtersIDs = mcParticle.daughtersIds();
         auto const& daughters = mcParticle.template daughters_as<aod::McParticles>();
 
         if (daughters.size() == 2) {

PWGLF/Tasks/Strangeness/sigmaanalysis.cxx

@@ -486,30 +486,31 @@ struct sigmaanalysis {
       histos.add("Gen/hCentralityVsPVzMC", "hCentralityVsPVzMC", kTH2F, {{101, 0.0f, 101.0f}, {100, -20.0f, +20.0f}});
 
       // Sigma0 specific
-      histos.add("Gen/h2dGenSigma0", "h2dGenSigma0", kTH2D, {axisCentrality, axisPt});
-      histos.add("Gen/h2dGenAntiSigma0", "h2dGenAntiSigma0", kTH2D, {axisCentrality, axisPt});      
-      histos.add("Gen/h2dGenSigma0VsMultMC_RecoedEvt", "h2dGenSigma0VsMultMC_RecoedEvt", kTH2D, {axisNch, axisPt});
-      histos.add("Gen/h2dGenAntiSigma0VsMultMC_RecoedEvt", "h2dGenAntiSigma0VsMultMC_RecoedEvt", kTH2D, {axisNch, axisPt});
-      histos.add("Gen/h2dGenSigma0VsMultMC", "h2dGenSigma0VsMultMC", kTH2D, {axisNch, axisPt});
-      histos.add("Gen/h2dGenAntiSigma0VsMultMC", "h2dGenAntiSigma0VsMultMC", kTH2D, {axisNch, axisPt});      
-
-      auto hGenSigma0Type = histos.add<TH1>("Gen/hGenSigma0Type", "hGenSigma0Type", kTH1D, {{4, -0.5f, 3.5f}});
-      hGenSigma0Type->GetXaxis()->SetBinLabel(1, "All Sigma0s");
-      hGenSigma0Type->GetXaxis()->SetBinLabel(2, "Sterile");
-      hGenSigma0Type->GetXaxis()->SetBinLabel(3, "Lambda+Gamma");
-      hGenSigma0Type->GetXaxis()->SetBinLabel(4, "Others");
-
-      auto h2DGenSigma0Type = histos.add<TH2>("Gen/h2DGenSigma0Type", "h2DGenSigma0Type", kTH2D, {{4, -0.5f, 3.5f}, axisGeneratorIds});
-      h2DGenSigma0Type->GetXaxis()->SetBinLabel(1, "All Sigma0s");
-      h2DGenSigma0Type->GetXaxis()->SetBinLabel(2, "Sterile");
-      h2DGenSigma0Type->GetXaxis()->SetBinLabel(3, "Lambda+Gamma");
-      h2DGenSigma0Type->GetXaxis()->SetBinLabel(4, "Others");
-
-      // Pi0 specific
-      histos.add("Gen/h2dGenPi0VsMultMC_RecoedEvt", "h2dGenPi0VsMultMC_RecoedEvt", kTH2D, {axisNch, axisPt});
-      histos.add("Gen/h2dGenPi0", "h2dGenPi0", kTH2D, {axisCentrality, axisPt});
-      histos.add("Gen/h2dGenPi0VsMultMC", "h2dGenPi0VsMultMC", kTH2D, {axisNch, axisPt});
-      
+      if (doprocessGeneratedRun3){
+        histos.add("Gen/h2dGenSigma0", "h2dGenSigma0", kTH2D, {axisCentrality, axisPt});
+        histos.add("Gen/h2dGenAntiSigma0", "h2dGenAntiSigma0", kTH2D, {axisCentrality, axisPt});      
+        histos.add("Gen/h2dGenSigma0VsMultMC_RecoedEvt", "h2dGenSigma0VsMultMC_RecoedEvt", kTH2D, {axisNch, axisPt});
+        histos.add("Gen/h2dGenAntiSigma0VsMultMC_RecoedEvt", "h2dGenAntiSigma0VsMultMC_RecoedEvt", kTH2D, {axisNch, axisPt});
+        histos.add("Gen/h2dGenSigma0VsMultMC", "h2dGenSigma0VsMultMC", kTH2D, {axisNch, axisPt});
+        histos.add("Gen/h2dGenAntiSigma0VsMultMC", "h2dGenAntiSigma0VsMultMC", kTH2D, {axisNch, axisPt});      
+
+        auto hGenSigma0Type = histos.add<TH1>("Gen/hGenSigma0Type", "hGenSigma0Type", kTH1D, {{4, -0.5f, 3.5f}});
+        hGenSigma0Type->GetXaxis()->SetBinLabel(1, "All Sigma0s");
+        hGenSigma0Type->GetXaxis()->SetBinLabel(2, "Sterile");
+        hGenSigma0Type->GetXaxis()->SetBinLabel(3, "Lambda+Gamma");
+        hGenSigma0Type->GetXaxis()->SetBinLabel(4, "Others");
+
+        auto h2DGenSigma0Type = histos.add<TH2>("Gen/h2DGenSigma0Type", "h2DGenSigma0Type", kTH2D, {{4, -0.5f, 3.5f}, axisGeneratorIds});
+        h2DGenSigma0Type->GetXaxis()->SetBinLabel(1, "All Sigma0s");
+        h2DGenSigma0Type->GetXaxis()->SetBinLabel(2, "Sterile");
+        h2DGenSigma0Type->GetXaxis()->SetBinLabel(3, "Lambda+Gamma");
+        h2DGenSigma0Type->GetXaxis()->SetBinLabel(4, "Others");
+      }
+      else{ // Pi0 specific        
+        histos.add("Gen/h2dGenPi0VsMultMC_RecoedEvt", "h2dGenPi0VsMultMC_RecoedEvt", kTH2D, {axisNch, axisPt});
+        histos.add("Gen/h2dGenPi0", "h2dGenPi0", kTH2D, {axisCentrality, axisPt});
+        histos.add("Gen/h2dGenPi0VsMultMC", "h2dGenPi0VsMultMC", kTH2D, {axisNch, axisPt});        
+      }      
     }
   }