@@ -22,6 +22,7 @@ DECLARE_SOA_COLUMN(ZVTX, zvtx, float);
2222DECLARE_SOA_COLUMN (WEIGHT , weight, float );
2323DECLARE_SOA_COLUMN (PTHAT , ptHat, float );
2424DECLARE_SOA_COLUMN (MULTIPLICITY , multiplicity, float );
25+ DECLARE_SOA_BITMAP_COLUMN (RCT , rct, 32 );
2526
2627DECLARE_SOA_COLUMN (DETPt, detPt, std::vector<float >);
2728DECLARE_SOA_COLUMN (DETEta, detEta, std::vector<float >);
@@ -41,6 +42,7 @@ DECLARE_SOA_TABLE(TREE, "AOD", "TREE",
4142 tree::WEIGHT ,
4243 tree::PTHAT ,
4344 tree::MULTIPLICITY ,
45+ tree::RCT ,
4446 tree::DETPt,
4547 tree::DETEta,
4648 tree::DETPhi,
@@ -61,6 +63,7 @@ struct EECTreeCreatorTask
6163
6264 Configurable<std::string> eventSelections{" eventSelections" , " sel8" , " " };
6365 Configurable<std::string> trackSelections{" trackSelections" , " globalTracks" , " " };
66+ Configurable<bool > skipMBGapEvents{" skipMBGapEvents" , true , " skip MB gap events" };
6467
6568 Produces<aod::TREE > tree;
6669
@@ -98,7 +101,8 @@ struct EECTreeCreatorTask
98101
99102 void processMC (aod::JetCollisionsMCD::iterator const & collision, aod::JetTracks const & tracks, aod::JMcParticles const & mcParticles,aod::JetMcCollisions const &)
100103 {
101- if (!jetderiveddatautilities::selectCollision (collision, eventSelectionBits)) return ;
104+ // do not do any RCT selections, will be done on analysis level
105+ if (!jetderiveddatautilities::selectCollision (collision, eventSelectionBits, skipMBGapEvents, false , " " , false , false )) return ;
102106 if (std::abs (collision.posZ ()) > vertexZCut) return ;
103107
104108 float w = collision.has_mcCollision () ? collision.mcCollision ().weight () : 1 .f ;
@@ -147,7 +151,7 @@ struct EECTreeCreatorTask
147151 }
148152 }
149153
150- tree ( collision.posZ (), w, pthard, collision.multFT0C (), detPt, detEta, detPhi, detCharge, truthPt, truthEta, truthPhi, truthE, truthCharge, pdg);}
154+ tree ( collision.posZ (), w, pthard, collision.multFT0C (), collision. rct_raw (), detPt, detEta, detPhi, detCharge, truthPt, truthEta, truthPhi, truthE, truthCharge, pdg);}
151155
152156 PROCESS_SWITCH (EECTreeCreatorTask, processMC, " MC processing" , true );
153157};
0 commit comments