Skip to content

Commit bdee7b3

Browse files
committed
adding fixes for derived data
1 parent 29c68ad commit bdee7b3

4 files changed

Lines changed: 134 additions & 46 deletions

File tree

PWGJE/Core/JetV0Utilities.h

Lines changed: 27 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ bool isV0Particle(T const& particles, U const& particle, bool v0ChargedDecaysOnl
142142
if (v0ChargedDecaysOnly) {
143143
return RecoDecay::isMatchedMCGen(particles, particle, +kK0Short, std::array{+kPiPlus, -kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, +kLambda0, std::array{+kProton, -kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, -kLambda0, std::array{-kProton, +kPiPlus}, true);
144144
} else {
145-
return RecoDecay::isMatchedMCGen(particles, particle, +kK0Short, std::array{+kPiPlus, -kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, +kK0Short, std::array{+kPi0, +kPi0}, true) || RecoDecay::isMatchedMCGen(particles, particle, +kLambda0, std::array{+kProton, -kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, +kLambda0, std::array{+kNeutron, +kPi0}, true) || RecoDecay::isMatchedMCGen(particles, particle, -kLambda0, std::array{-kProton, +kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, -kLambda0, std::array{+kNeutron, +kPi0}, true);
145+
return RecoDecay::isMatchedMCGen(particles, particle, +kK0Short, std::array{+kPiPlus, -kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, +kK0Short, std::array{+kPi0, +kPi0}, true) || RecoDecay::isMatchedMCGen(particles, particle, +kLambda0, std::array{+kProton, -kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, +kLambda0, std::array{+kNeutron, +kPi0}, true) || RecoDecay::isMatchedMCGen(particles, particle, -kLambda0, std::array{-kProton, +kPiPlus}, true) || RecoDecay::isMatchedMCGen(particles, particle, -kLambda0, std::array{-kNeutron, +kPi0}, true);
146146
}
147147
}
148148

@@ -183,6 +183,32 @@ uint8_t setV0ParticleDecayBit(T const& particles, U const& particle)
183183
return bit;
184184
}
185185

186+
// need to update this
187+
template <typename T, typename U, typename V>
188+
void fillV0CandidateTable(T const& candidate, U& V0CoresTable, V& V0SignalFlagsTable)
189+
{
190+
191+
V0CoresTable(
192+
candidate.x(),
193+
candidate.y(),
194+
candidate.z(),
195+
candidate.pxpos(),
196+
candidate.pypos(),
197+
candidate.pzpos(),
198+
candidate.pxneg(),
199+
candidate.pyneg(),
200+
candidate.pzneg(),
201+
candidate.dcaV0daughters(),
202+
candidate.dcapostopv(),
203+
candidate.dcanegtopv(),
204+
candidate.v0cosPA(),
205+
candidate.dcav0topv(),
206+
candidate.v0Type());
207+
208+
V0SignalFlagsTable(
209+
candidate.signalFlag());
210+
}
211+
186212
}; // namespace jetv0utilities
187213

188214
#endif // PWGJE_CORE_JETV0UTILITIES_H_

PWGJE/DataModel/JetReducedDataV0.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,10 +53,10 @@ DECLARE_SOA_TABLE_STAGED(JV0CollisionIds, "JV0COLLID",
5353
DECLARE_SOA_TABLE_STAGED(JV0McCollisionIds, "JV0MCCOLLID",
5454
jv0indices::JMcCollisionId);
5555

56-
DECLARE_SOA_TABLE(JV0Ids, "AOD", "JV0ID",
57-
jv0indices::JCollisionId,
58-
jv0indices::PosTrackId,
59-
jv0indices::NegTrackId);
56+
DECLARE_SOA_TABLE_STAGED(JV0Ids, "JV0ID",
57+
jv0indices::JCollisionId,
58+
jv0indices::PosTrackId,
59+
jv0indices::NegTrackId);
6060

6161
namespace jv0mc
6262
{

0 commit comments

Comments
 (0)