Skip to content

Commit 3663480

Browse files
gianniliverarogianniliveraro
committed
Minor changes in sigma0 tasks
1 parent 525b770 commit 3663480

File tree

3 files changed

+45
-28
lines changed

3 files changed

+45
-28
lines changed

‎PWGLF/DataModel/LFSigmaTables.h‎

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -481,14 +481,16 @@ namespace sigma0Gen
481481
{
482482
DECLARE_SOA_COLUMN(IsSigma0, isSigma0, bool); // true: sigma0, false: antisigma0
483483
DECLARE_SOA_COLUMN(ProducedByGenerator, producedByGenerator, bool);
484-
DECLARE_SOA_COLUMN(Sigma0MCPt, sigma0MCPt, float); // MC pT
484+
DECLARE_SOA_COLUMN(MCPt, mcpt, float); // MC pT
485+
DECLARE_SOA_COLUMN(MCY, mcy, float); // MC Y
485486

486487
} // namespace sigma0Gen
487488

488489
DECLARE_SOA_TABLE(Sigma0Gens, "AOD", "SIGMA0GENS",
489490
sigma0Gen::IsSigma0,
490491
sigma0Gen::ProducedByGenerator,
491-
sigma0Gen::Sigma0MCPt);
492+
sigma0Gen::MCPt,
493+
sigma0Gen::MCY);
492494

493495
DECLARE_SOA_TABLE(SigmaCollRef, "AOD", "SIGMACOLLREF", //! optional table to refer back to a collision
494496
o2::soa::Index<>, v0data::StraCollisionId);
@@ -859,12 +861,14 @@ DECLARE_SOA_TABLE(Pi0CollRef, "AOD", "PI0COLLREF", //! optional table to refer b
859861
namespace pi0Gen
860862
{
861863
DECLARE_SOA_COLUMN(ProducedByGenerator, producedByGenerator, bool);
862-
DECLARE_SOA_COLUMN(Pi0MCPt, pi0MCPt, float); // MC pT
864+
DECLARE_SOA_COLUMN(MCPt, mcpt, float); // MC pT
865+
DECLARE_SOA_COLUMN(MCY, mcy, float); // MC Y
863866
} // namespace pi0Gen
864867

865868
DECLARE_SOA_TABLE(Pi0Gens, "AOD", "PI0GENS",
866869
pi0Gen::ProducedByGenerator,
867-
pi0Gen::Pi0MCPt);
870+
pi0Gen::MCPt,
871+
pi0Gen::MCY);
868872

869873
DECLARE_SOA_TABLE(Pi0GenCollRef, "AOD", "PI0GENCOLLREF", //! optional table to refer back to a collision
870874
o2::soa::Index<>, v0data::StraMCCollisionId);

‎PWGLF/TableProducer/Strangeness/sigma0builder.cxx‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1259,13 +1259,13 @@ struct sigma0builder {
12591259
// Fill tables
12601260
// Pi0
12611261
if (fillPi0Tables && MCGenInfo.IsPi0) {
1262-
pi0Gens(MCGenInfo.IsProducedByGenerator, MCGenInfo.MCPt); // optional table to store generated pi0 candidates. Be careful, this is a large table!
1262+
pi0Gens(MCGenInfo.IsProducedByGenerator, MCGenInfo.MCPt, mcParticle.y()); // optional table to store generated pi0 candidates. Be careful, this is a large table!
12631263
pi0GenCollRefs(MCGenInfo.MCCollId); // link to stramccollision table
12641264
}
12651265

12661266
// Sigma0/ASigma0
12671267
if (fillSigma0Tables && (MCGenInfo.IsSigma0 || MCGenInfo.IsAntiSigma0)) {
1268-
sigma0Gens(MCGenInfo.IsSigma0, MCGenInfo.IsProducedByGenerator, MCGenInfo.MCPt);
1268+
sigma0Gens(MCGenInfo.IsSigma0, MCGenInfo.IsProducedByGenerator, MCGenInfo.MCPt, mcParticle.y());
12691269
sigma0GenCollRefs(MCGenInfo.MCCollId); // link to stramccollision table
12701270
}
12711271
}

‎PWGLF/Tasks/Strangeness/sigmaanalysis.cxx‎

Lines changed: 35 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ using namespace o2::framework;
5959
using namespace o2::framework::expressions;
6060

6161
using std::array;
62-
using MCSigma0s = soa::Join<aod::Sigma0Cores, aod::Sigma0PhotonExtras, aod::Sigma0LambdaExtras, aod::Sigma0MCCores, aod::SigmaCollRef, aod::SigmaMCLabels>;
62+
using MCSigma0s = soa::Join<aod::Sigma0Cores, aod::Sigma0PhotonExtras, aod::Sigma0LambdaExtras, aod::Sigma0MCCores, aod::SigmaCollRef>;
6363
using Sigma0s = soa::Join<aod::Sigma0Cores, aod::Sigma0PhotonExtras, aod::Sigma0LambdaExtras, aod::SigmaCollRef>;
6464

6565
static const std::vector<std::string> PhotonSels = {"NoSel", "V0Type", "DCADauToPV",
@@ -124,7 +124,13 @@ struct sigmaanalysis {
124124
} eventSelections;
125125

126126
// Generated Sigma0s
127-
Configurable<bool> mc_keepOnlyFromGenerator{"mc_keepOnlyFromGenerator", true, "if true, consider only particles from generator to calculate efficiency."};
127+
struct : ConfigurableGroup {
128+
std::string prefix = "genSelections"; // JSON group name
129+
Configurable<bool> mc_keepOnlyFromGenerator{"mc_keepOnlyFromGenerator", true, "if true, consider only particles from generator to calculate efficiency."};
130+
Configurable<float> mc_rapidityMin{"mc_rapidityMin", -0.5, "Min generated particle rapidity"};
131+
Configurable<float> mc_rapidityMax{"mc_rapidityMax", 0.5, "Max generated particle rapidity"};
132+
} genSelections;
133+
128134

129135
// QA
130136
Configurable<bool> fdoSigma0QA{"doSigma0QA", false, "if true, perform Sigma0 QA analysis. Only works with MC."};
@@ -545,7 +551,8 @@ struct sigmaanalysis {
545551
histos.add("Gen/h2dGenSigma0VsMultMC", "h2dGenSigma0VsMultMC", kTH2D, {axisNch, axisPt});
546552
histos.add("Gen/h2dGenAntiSigma0VsMultMC", "h2dGenAntiSigma0VsMultMC", kTH2D, {axisNch, axisPt});
547553

548-
} else { // Pi0 specific
554+
}
555+
if (doprocessPi0GeneratedRun3) { // Pi0 specific
549556
histos.add("Gen/h2dGenPi0VsMultMC_RecoedEvt", "h2dGenPi0VsMultMC_RecoedEvt", kTH2D, {axisNch, axisPt});
550557
histos.add("Gen/h2dGenPi0", "h2dGenPi0", kTH2D, {axisCentrality, axisPt});
551558
histos.add("Gen/h2dGenPi0VsMultMC", "h2dGenPi0VsMultMC", kTH2D, {axisNch, axisPt});
@@ -813,8 +820,12 @@ struct sigmaanalysis {
813820
continue;
814821
}
815822

823+
// Rapidity selection
824+
if ((genParticle.mcy() < genSelections.mc_rapidityMin) || (genParticle.mcy() > genSelections.mc_rapidityMax))
825+
continue;
826+
816827
// Selection on the source (generator/transport)
817-
if (!genParticle.producedByGenerator() && mc_keepOnlyFromGenerator)
828+
if (!genParticle.producedByGenerator() && genSelections.mc_keepOnlyFromGenerator)
818829
continue;
819830

820831
// Select corresponding mc collision && Basic event selection
@@ -834,33 +845,35 @@ struct sigmaanalysis {
834845

835846
//______________________________________________________________________________
836847
// Generated Sigma0 processing
837-
if constexpr (requires { genParticle.sigma0MCPt(); }) {
848+
if constexpr (requires { genParticle.isSigma0(); }) {
849+
if (doprocessGeneratedRun3) {
838850

839-
float ptmc = genParticle.sigma0MCPt();
851+
float ptmc = genParticle.mcpt();
840852

841-
if (listBestCollisionIdx[mcCollision.globalIndex()] > -1) {
842-
auto collision = collisions.iteratorAt(listBestCollisionIdx[mcCollision.globalIndex()]);
843-
centrality = doPPAnalysis ? collision.centFT0M() : collision.centFT0C();
853+
if (listBestCollisionIdx[mcCollision.globalIndex()] > -1) {
854+
auto collision = collisions.iteratorAt(listBestCollisionIdx[mcCollision.globalIndex()]);
855+
centrality = doPPAnalysis ? collision.centFT0M() : collision.centFT0C();
844856

845-
if (genParticle.isSigma0())
846-
histos.fill(HIST("Gen/h2dGenSigma0VsMultMC_RecoedEvt"), mcCollision.multMCNParticlesEta05(), ptmc);
857+
if (genParticle.isSigma0())
858+
histos.fill(HIST("Gen/h2dGenSigma0VsMultMC_RecoedEvt"), mcCollision.multMCNParticlesEta05(), ptmc);
847859

848-
else
849-
histos.fill(HIST("Gen/h2dGenAntiSigma0VsMultMC_RecoedEvt"), mcCollision.multMCNParticlesEta05(), ptmc);
850-
}
860+
else
861+
histos.fill(HIST("Gen/h2dGenAntiSigma0VsMultMC_RecoedEvt"), mcCollision.multMCNParticlesEta05(), ptmc);
862+
}
851863

852-
if (genParticle.isSigma0()) {
853-
histos.fill(HIST("Gen/h2dGenSigma0"), centrality, ptmc);
854-
histos.fill(HIST("Gen/h2dGenSigma0VsMultMC"), mcCollision.multMCNParticlesEta05(), ptmc);
855-
} else {
856-
histos.fill(HIST("Gen/h2dGenAntiSigma0"), centrality, ptmc);
857-
histos.fill(HIST("Gen/h2dGenAntiSigma0VsMultMC"), mcCollision.multMCNParticlesEta05(), ptmc);
864+
if (genParticle.isSigma0()) {
865+
histos.fill(HIST("Gen/h2dGenSigma0"), centrality, ptmc);
866+
histos.fill(HIST("Gen/h2dGenSigma0VsMultMC"), mcCollision.multMCNParticlesEta05(), ptmc);
867+
} else {
868+
histos.fill(HIST("Gen/h2dGenAntiSigma0"), centrality, ptmc);
869+
histos.fill(HIST("Gen/h2dGenAntiSigma0VsMultMC"), mcCollision.multMCNParticlesEta05(), ptmc);
870+
}
858871
}
859872
}
860873
//______________________________________________________________________________
861874
// Generated Pi0 processing
862-
if constexpr (requires { genParticle.pi0MCPt(); }) {
863-
float ptmc = genParticle.pi0MCPt();
875+
if (doprocessPi0GeneratedRun3) {
876+
float ptmc = genParticle.mcpt();
864877

865878
if (listBestCollisionIdx[mcCollision.globalIndex()] > -1) {
866879
auto collision = collisions.iteratorAt(listBestCollisionIdx[mcCollision.globalIndex()]);

0 commit comments

Comments
 (0)