Skip to content

Commit 43e3cbb

Browse files
committed
Adding option to fill resolution matrix for 2kstar
1 parent 2e977ec commit 43e3cbb

1 file changed

Lines changed: 12 additions & 3 deletions

File tree

PWGCF/FemtoUniverse/Core/FemtoUniverseContainer.h

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -226,12 +226,16 @@ class FemtoUniverseContainer
226226
/// \param part2 Particle two
227227
/// \param mult Multiplicity of the event
228228
template <bool isMC, typename T>
229-
void setPair(T const& part1, T const& part2, const int mult, bool use3dplots, float weight = 1.0f)
229+
void setPair(T const& part1, T const& part2, const int mult, bool use3dplots, float weight = 1.0f, bool isiden = false)
230230
{
231231
float femtoObs, femtoObsMC;
232232
// Calculate femto observable and the mT with reconstructed information
233233
if constexpr (FemtoObs == femto_universe_container::Observable::kstar) {
234-
femtoObs = FemtoUniverseMath::getkstar(part1, mMassOne, part2, mMassTwo);
234+
if (!isiden) {
235+
femtoObs = FemtoUniverseMath::getkstar(part1, mMassOne, part2, mMassTwo);
236+
} else {
237+
femtoObs = 2.0*FemtoUniverseMath::getkstar(part1, mMassOne, part2, mMassTwo);
238+
}
235239
}
236240
const float mT = FemtoUniverseMath::getmT(part1, mMassOne, part2, mMassTwo);
237241

@@ -242,7 +246,11 @@ class FemtoUniverseContainer
242246
if (part1.has_fdMCParticle() && part2.has_fdMCParticle()) {
243247
// calculate the femto observable and the mT with MC truth information
244248
if constexpr (FemtoObs == femto_universe_container::Observable::kstar) {
245-
femtoObsMC = FemtoUniverseMath::getkstar(part1.fdMCParticle(), mMassOne, part2.fdMCParticle(), mMassTwo);
249+
if (!isiden) {
250+
femtoObsMC = FemtoUniverseMath::getkstar(part1.fdMCParticle(), mMassOne, part2.fdMCParticle(), mMassTwo);
251+
} else {
252+
femtoObsMC = 2.0*FemtoUniverseMath::getkstar(part1.fdMCParticle(), mMassOne, part2.fdMCParticle(), mMassTwo);
253+
}
246254
}
247255
const float mTMC = FemtoUniverseMath::getmT(part1.fdMCParticle(), mMassOne, part2.fdMCParticle(), mMassTwo);
248256

@@ -260,6 +268,7 @@ class FemtoUniverseContainer
260268
}
261269
}
262270

271+
263272
protected:
264273
HistogramRegistry* mHistogramRegistry = nullptr; ///< For QA output
265274
static constexpr std::string_view FolderSuffix[2] = {"SameEvent", "MixedEvent"}; ///< Folder naming for the output according to EventType

0 commit comments

Comments
 (0)