Skip to content
Merged
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
7 changes: 5 additions & 2 deletions PWGJE/TableProducer/berkeleyTreeProducer.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ namespace o2::aod
{
namespace berkeleytree
{
DECLARE_SOA_COLUMN(RunNumber, runNumber, int);
DECLARE_SOA_COLUMN(VtxZ, vtxZ, float);
DECLARE_SOA_COLUMN(Weight, weight, float);
DECLARE_SOA_COLUMN(PtHat, ptHat, float);
Expand All @@ -63,6 +64,7 @@ DECLARE_SOA_COLUMN(PdgId, pdgId, std::vector<int>);
} // namespace berkeleytree

DECLARE_SOA_TABLE(BerkeleyTree, "AOD", "BERKELEYTREE",
berkeleytree::RunNumber,
berkeleytree::VtxZ,
berkeleytree::Weight,
berkeleytree::PtHat,
Expand Down Expand Up @@ -133,7 +135,7 @@ struct BerkeleyTreeProducer {
}

using JetParticlesWithOriginal = soa::Join<aod::JetParticles, aod::JMcParticlePIs>;
void processMCJJ(aod::JetCollisionsMCD::iterator const& collision, aod::JetTracksMCD const& tracks, JetParticlesWithOriginal const& mcParticles, aod::JetMcCollisions const&)
void processMCJJ(aod::JetCollisionsMCD::iterator const& collision, aod::JetTracksMCD const& tracks, JetParticlesWithOriginal const& mcParticles, aod::JetMcCollisions const&, aod::JBCs const&)
{
// do not do any RCT selections, will be done on analysis level
if (!jetderiveddatautilities::selectCollision(collision, eventSelectionBits, skipMBGapEvents, false, "", false, false))
Expand All @@ -143,6 +145,7 @@ struct BerkeleyTreeProducer {

float weight = collision.has_mcCollision() ? collision.mcCollision().weight() : 1.f;
float pthat = collision.has_mcCollision() ? collision.mcCollision().ptHard() : 1.f;
int runNumber = collision.bc_as<aod::JBCs>().runNumber();

std::vector<float> detPt, detEta, detPhi;
std::vector<uint8_t> detTrackSel;
Expand Down Expand Up @@ -196,7 +199,7 @@ struct BerkeleyTreeProducer {
}
}

tree(collision.posZ(), weight, pthat, collision.multFT0C(), collision.eventSel(), collision.trackOccupancyInTimeRange(), collision.rct_raw(), detPt, detEta, detPhi, detTrackSel, detMcId, genPt, genEta, genPhi, genE, genCharge, genMcId, pdgId);
tree(runNumber, collision.posZ(), weight, pthat, collision.multFT0C(), collision.eventSel(), collision.trackOccupancyInTimeRange(), collision.rct_raw(), detPt, detEta, detPhi, detTrackSel, detMcId, genPt, genEta, genPhi, genE, genCharge, genMcId, pdgId);
}

PROCESS_SWITCH(BerkeleyTreeProducer, processMCJJ, "MC processing for JJ simulations", false);
Expand Down
Loading