@@ -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