Skip to content
Closed
Changes from all commits
Commits
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
31 changes: 17 additions & 14 deletions PWGDQ/TableProducer/tableMakerMC_withAssoc.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -543,24 +543,26 @@ struct TableMakerMC {
if (mcflags == 0) {
continue;
}

auto mcCollision = mctrack.template mcCollision_as<MyEventsMcWithMults>();

bool isjpsi = false;
if(mctrack.pdgCode() == 443) {
if (mctrack.pdgCode() == 443) {
isjpsi = true;
}

//add rapidity and pT cuts for mcparticles
if(isjpsi && (fConfigJpsiMinPtMC.value > 0 && std::abs(mctrack.pt()) < fConfigJpsiMinPtMC.value)) continue;
if(isjpsi && (fConfigJpsiMaxPtMC.value > 0 && std::abs(mctrack.pt()) > fConfigJpsiMaxPtMC.value)) continue;
if(isjpsi && (fConfigTrackMaxRapidityMC.value > 0 && std::abs(mctrack.y()) > fConfigTrackMaxRapidityMC.value)) continue;
// add rapidity and pT cuts for mcparticles
if (isjpsi && (fConfigJpsiMinPtMC.value > 0 && std::abs(mctrack.pt()) < fConfigJpsiMinPtMC.value))
continue;
if (isjpsi && (fConfigJpsiMaxPtMC.value > 0 && std::abs(mctrack.pt()) > fConfigJpsiMaxPtMC.value))
continue;
if (isjpsi && (fConfigTrackMaxRapidityMC.value > 0 && std::abs(mctrack.y()) > fConfigTrackMaxRapidityMC.value))
continue;

//Save jpsi events that satisfied the cut conditions
if(fConfigApplyselctionMC && isjpsi && ((std::abs(mctrack.pt()) > fConfigJpsiMinPtMC.value) && (std::abs(mctrack.pt()) < fConfigJpsiMaxPtMC.value) && (std::abs(mctrack.y()) < fConfigTrackMaxRapidityMC.value)))
{
// Save jpsi events that satisfied the cut conditions
if (fConfigApplyselctionMC && isjpsi && ((std::abs(mctrack.pt()) > fConfigJpsiMinPtMC.value) && (std::abs(mctrack.pt()) < fConfigJpsiMaxPtMC.value) && (std::abs(mctrack.y()) < fConfigTrackMaxRapidityMC.value))) {
labelsToReserve.insert(mcCollision.globalIndex());
}
}

// If this MC track was not already added to the map, add it now
if (fLabelsMap.find(mctrack.globalIndex()) == fLabelsMap.end()) {
Expand Down Expand Up @@ -785,11 +787,12 @@ struct TableMakerMC {
// However, in data analysis one should loop over associations, so this one should not be used.
// In the case of Run2-like analysis, there will be no associations, so this ID will be the one originally assigned in the AO2Ds (updated for the skims)
uint32_t reducedEventIdx = fCollIndexMap[track.collisionId()];

// apply the event selection for reconstrcted tracks
if (fConfigApplyselctionMC && track.has_mcParticle()) {
auto mcCollision = collision.template mcCollision_as<MyEventsMcWithMults>();
if ((labelsToReserve.count(mcCollision.globalIndex()) == 0)) continue;
auto mcCollision = collision.template mcCollision_as<MyEventsMcWithMults>();
if ((labelsToReserve.count(mcCollision.globalIndex()) == 0))
continue;
}

// NOTE: trackBarrelInfo stores the index of the collision as in AO2D (for use in some cases where the analysis on skims is done
Expand Down