Skip to content

Please consider the following formatting changes to #6565 #308

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ddobrigk merged 1 commit into from
Jun 18, 2024
Merged

Please consider the following formatting changes to #6565 #308

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
62 changes: 30 additions & 32 deletions PWGLF/TableProducer/Strangeness/cascademcbuilder.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ struct cascademcbuilder {
{

// to be used if using the asymmetric mode, kept empty otherwise
std::vector<mcCascinfo> mcCascinfos; // V0MCCore information
std::vector<mcCascinfo> mcCascinfos; // V0MCCore information
std::vector<bool> mcParticleIsReco(mcParticles.size(), false); // mc Particle not recoed by V0s

for (auto& casc : cascTable) {
Expand All @@ -111,9 +111,9 @@ struct cascademcbuilder {
thisInfo.bachP[0] = -999.0f, thisInfo.bachP[1] = -999.0f, thisInfo.bachP[2] = -999.0f;
thisInfo.momentum[0] = -999.0f, thisInfo.momentum[1] = -999.0f, thisInfo.momentum[2] = -999.0f;
thisInfo.label = -1, thisInfo.motherLabel = -1;
thisInfo.mcParticlePositive=-1;
thisInfo.mcParticleNegative=-1;
thisInfo.mcParticleBachelor=-1;
thisInfo.mcParticlePositive = -1;
thisInfo.mcParticleNegative = -1;
thisInfo.mcParticleBachelor = -1;

// Acquire all three daughter tracks, please
auto lBachTrack = casc.template bachelor_as<aod::McTrackLabels>();
Expand Down Expand Up @@ -162,7 +162,7 @@ struct cascademcbuilder {
if (lV0Mother == lBachMother) {
thisInfo.label = lV0Mother.globalIndex();

if(lV0Mother.has_mcCollision()){
if (lV0Mother.has_mcCollision()) {
thisInfo.mcCollision = lV0Mother.mcCollisionId(); // save this reference, please
}

Expand Down Expand Up @@ -194,19 +194,19 @@ struct cascademcbuilder {
thisInfo.label, thisInfo.motherLabel);

// Mark mcParticle as recoed (no searching necessary afterwards)
if(thisInfo.motherLabel>-1){
mcParticleIsReco[thisInfo.motherLabel] = true;
if (thisInfo.motherLabel > -1) {
mcParticleIsReco[thisInfo.motherLabel] = true;
}

if (populateCascMCCoresSymmetric) {
cascmccores(
thisInfo.pdgCode, thisInfo.pdgCodeMother, thisInfo.pdgCodeV0, thisInfo.isPhysicalPrimary,
thisInfo.pdgCodePositive, thisInfo.pdgCodeNegative, thisInfo.pdgCodeBachelor,
thisInfo.xyz[0], thisInfo.xyz[1], thisInfo.xyz[2],
thisInfo.xyz[0], thisInfo.xyz[1], thisInfo.xyz[2],
thisInfo.lxyz[0], thisInfo.lxyz[1], thisInfo.lxyz[2],
thisInfo.posP[0], thisInfo.posP[1], thisInfo.posP[2],
thisInfo.negP[0], thisInfo.negP[1], thisInfo.negP[2],
thisInfo.bachP[0], thisInfo.bachP[1], thisInfo.bachP[2],
thisInfo.posP[0], thisInfo.posP[1], thisInfo.posP[2],
thisInfo.negP[0], thisInfo.negP[1], thisInfo.negP[2],
thisInfo.bachP[0], thisInfo.bachP[1], thisInfo.bachP[2],
thisInfo.momentum[0], thisInfo.momentum[1], thisInfo.momentum[2]);
cascmccollrefs(thisInfo.mcCollision);
}
Expand All @@ -219,10 +219,9 @@ struct cascademcbuilder {
if (
thisInfo.mcParticlePositive == mcCascinfos[ii].mcParticlePositive && mcCascinfos[ii].mcParticlePositive > 0 &&
thisInfo.mcParticleNegative == mcCascinfos[ii].mcParticleNegative && mcCascinfos[ii].mcParticleNegative > 0 &&
thisInfo.mcParticleBachelor == mcCascinfos[ii].mcParticleBachelor && mcCascinfos[ii].mcParticleBachelor > 0
) {
thisInfo.mcParticleBachelor == mcCascinfos[ii].mcParticleBachelor && mcCascinfos[ii].mcParticleBachelor > 0) {
thisCascMCCoreIndex = ii;
break; // this exists already in list
break; // this exists already in list
}
}
if (thisCascMCCoreIndex < 0) {
Expand All @@ -236,8 +235,8 @@ struct cascademcbuilder {

// now populate V0MCCores if in asymmetric mode
if (populateCascMCCoresAsymmetric) {
// first step: add any un-recoed v0mmcores that were requested
for (auto & mcParticle : mcParticles) {
// first step: add any un-recoed v0mmcores that were requested
for (auto& mcParticle : mcParticles) {
thisInfo.pdgCode = -1, thisInfo.pdgCodeMother = -1;
thisInfo.pdgCodePositive = -1, thisInfo.pdgCodeNegative = -1;
thisInfo.pdgCodeBachelor = -1, thisInfo.pdgCodeV0 = -1;
Expand All @@ -249,32 +248,31 @@ struct cascademcbuilder {
thisInfo.bachP[0] = 0.0f, thisInfo.bachP[1] = 0.0f, thisInfo.bachP[2] = 0.0f;
thisInfo.momentum[0] = 0.0f, thisInfo.momentum[1] = 0.0f, thisInfo.momentum[2] = 0.0f;
thisInfo.label = -1, thisInfo.motherLabel = -1;
thisInfo.mcParticlePositive=-1;
thisInfo.mcParticleNegative=-1;
thisInfo.mcParticleBachelor=-1;
thisInfo.mcParticlePositive = -1;
thisInfo.mcParticleNegative = -1;
thisInfo.mcParticleBachelor = -1;

if(mcParticleIsReco[mcParticle.globalIndex()] == true)
if (mcParticleIsReco[mcParticle.globalIndex()] == true)
continue; // skip if already created in list
if(TMath::Abs(mcParticle.y()) > rapidityWindow)

if (TMath::Abs(mcParticle.y()) > rapidityWindow)
continue; // skip outside midrapidity

if(
(addGeneratedXiMinus && mcParticle.pdgCode() == 3312) ||
if (
(addGeneratedXiMinus && mcParticle.pdgCode() == 3312) ||
(addGeneratedXiPlus && mcParticle.pdgCode() == -3312) ||
(addGeneratedOmegaMinus && mcParticle.pdgCode() == 3334) ||
(addGeneratedOmegaPlus && mcParticle.pdgCode() == -3334)
){
(addGeneratedOmegaPlus && mcParticle.pdgCode() == -3334)) {
thisInfo.pdgCode = mcParticle.pdgCode();
thisInfo.isPhysicalPrimary = mcParticle.isPhysicalPrimary();

if(mcParticle.has_mcCollision()){
if (mcParticle.has_mcCollision()) {
thisInfo.mcCollision = mcParticle.mcCollisionId(); // save this reference, please
}
thisInfo.momentum[0] = mcParticle.px();
thisInfo.momentum[1] = mcParticle.py();
thisInfo.momentum[2] = mcParticle.pz();

// if I got here, it means this MC particle was not recoed and is of interest. Add it please
mcCascinfos.push_back(thisInfo);
}
Expand All @@ -284,11 +282,11 @@ struct cascademcbuilder {
cascmccores( // a lot of the info below will be compressed in case of not-recoed MC (good!)
thisInfo.pdgCode, thisInfo.pdgCodeMother, thisInfo.pdgCodeV0, thisInfo.isPhysicalPrimary,
thisInfo.pdgCodePositive, thisInfo.pdgCodeNegative, thisInfo.pdgCodeBachelor,
thisInfo.xyz[0], thisInfo.xyz[1], thisInfo.xyz[2],
thisInfo.xyz[0], thisInfo.xyz[1], thisInfo.xyz[2],
thisInfo.lxyz[0], thisInfo.lxyz[1], thisInfo.lxyz[2],
thisInfo.posP[0], thisInfo.posP[1], thisInfo.posP[2],
thisInfo.negP[0], thisInfo.negP[1], thisInfo.negP[2],
thisInfo.bachP[0], thisInfo.bachP[1], thisInfo.bachP[2],
thisInfo.posP[0], thisInfo.posP[1], thisInfo.posP[2],
thisInfo.negP[0], thisInfo.negP[1], thisInfo.negP[2],
thisInfo.bachP[0], thisInfo.bachP[1], thisInfo.bachP[2],
thisInfo.momentum[0], thisInfo.momentum[1], thisInfo.momentum[2]);
cascmccollrefs(thisInfo.mcCollision);
}
Expand Down
31 changes: 15 additions & 16 deletions PWGLF/TableProducer/Strangeness/lambdakzeromcbuilder.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ struct lambdakzeromcbuilder {
Produces<aod::McV0Labels> v0labels; // MC labels for V0s
Produces<aod::V0MCCores> v0mccores; // optionally aggregate information from MC side for posterior analysis (derived data)
Produces<aod::V0CoreMCLabels> v0CoreMCLabels; // interlink V0Cores -> V0MCCores in asymmetric mode
Produces<aod::V0MCCollRefs> v0mccollref; // references collisions from V0MCCores
Produces<aod::V0MCCollRefs> v0mccollref; // references collisions from V0MCCores

Configurable<bool> populateV0MCCoresSymmetric{"populateV0MCCoresSymmetric", false, "populate V0MCCores table for derived data analysis, keep V0MCCores joinable with V0Cores"};
Configurable<bool> populateV0MCCoresAsymmetric{"populateV0MCCoresAsymmetric", false, "populate V0MCCores table for derived data analysis, create V0Cores -> V0MCCores interlink. Saves only labeled V0s."};
Expand Down Expand Up @@ -147,7 +147,7 @@ struct lambdakzeromcbuilder {
if (lNegMother.globalIndex() == lPosMother.globalIndex()) {
thisInfo.label = lNegMother.globalIndex();

if(lNegMother.has_mcCollision()){
if (lNegMother.has_mcCollision()) {
thisInfo.mcCollision = lNegMother.mcCollisionId(); // save this reference, please
}

Expand All @@ -173,8 +173,8 @@ struct lambdakzeromcbuilder {
thisInfo.label, thisInfo.motherLabel);

// Mark mcParticle as recoed (no searching necessary afterwards)
if(thisInfo.motherLabel>-1){
mcParticleIsReco[thisInfo.motherLabel] = true;
if (thisInfo.motherLabel > -1) {
mcParticleIsReco[thisInfo.motherLabel] = true;
}

// ---] Symmetric populate [---
Expand Down Expand Up @@ -222,8 +222,8 @@ struct lambdakzeromcbuilder {

// now populate V0MCCores if in asymmetric mode
if (populateV0MCCoresAsymmetric) {
// first step: add any un-recoed v0mmcores that were requested
for (auto & mcParticle : mcParticles) {
// first step: add any un-recoed v0mmcores that were requested
for (auto& mcParticle : mcParticles) {
thisInfo.packedMcParticleIndices = 0;
thisInfo.label = -1;
thisInfo.motherLabel = -1;
Expand All @@ -236,31 +236,30 @@ struct lambdakzeromcbuilder {
thisInfo.posP[0] = thisInfo.posP[1] = thisInfo.posP[2] = 0.0f;
thisInfo.negP[0] = thisInfo.negP[1] = thisInfo.negP[2] = 0.0f;

if(mcParticleIsReco[mcParticle.globalIndex()] == true)
if (mcParticleIsReco[mcParticle.globalIndex()] == true)
continue; // skip if already created in list
if(TMath::Abs(mcParticle.y()) > rapidityWindow)

if (TMath::Abs(mcParticle.y()) > rapidityWindow)
continue; // skip outside midrapidity

if(
(addGeneratedK0Short && mcParticle.pdgCode() == 310) ||
if (
(addGeneratedK0Short && mcParticle.pdgCode() == 310) ||
(addGeneratedLambda && mcParticle.pdgCode() == 3122) ||
(addGeneratedAntiLambda && mcParticle.pdgCode() == -3122) ||
(addGeneratedGamma && mcParticle.pdgCode() == 22)
){
(addGeneratedGamma && mcParticle.pdgCode() == 22)) {
thisInfo.pdgCode = mcParticle.pdgCode();
thisInfo.isPhysicalPrimary = mcParticle.isPhysicalPrimary();

if(mcParticle.has_mcCollision()){
if (mcParticle.has_mcCollision()) {
thisInfo.mcCollision = mcParticle.mcCollisionId(); // save this reference, please
}

// guarantee compressibility: keep momentum entirely in the positive prong
// guarantee compressibility: keep momentum entirely in the positive prong
// WARNING: THIS IS AS ARBITRARY AS IT GETS but should be ok
thisInfo.posP[0] = mcParticle.px();
thisInfo.posP[1] = mcParticle.py();
thisInfo.posP[2] = mcParticle.pz();

// if I got here, it means this MC particle was not recoed and is of interest. Add it please
mcV0infos.push_back(thisInfo);
}
Expand Down