Skip to content

Commit ccb326e

Browse files
authored
[PWGHF] Add normalized decay length to Omegac0 DCAFitter ML inputs (#18576)
1 parent f67e4a5 commit ccb326e

3 files changed

Lines changed: 10 additions & 3 deletions

File tree

‎PWGHF/Core/HfMlResponseOmegacToOmegaPiQa.h‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,7 @@ enum class InputFeaturesOmegacToOmegaPi : uint8_t {
7070
impactParCascZ,
7171
cosPACharmBaryon,
7272
cosPACasc,
73+
normDecayLenCharmBar,
7374

7475
// KFParticle variables
7576
cosPaOmegacToPv,
@@ -136,6 +137,7 @@ class HfMlResponseOmegacToOmegaPi : public HfMlResponse<TypeOutputScore>
136137
CHECK_AND_FILL_VEC_OMEGAC0(impactParCascZ);
137138
CHECK_AND_FILL_VEC_OMEGAC0(cosPACharmBaryon);
138139
CHECK_AND_FILL_VEC_OMEGAC0(cosPACasc);
140+
CHECK_AND_FILL_VEC_OMEGAC0(normDecayLenCharmBar);
139141
}
140142
}
141143

@@ -192,7 +194,8 @@ class HfMlResponseOmegacToOmegaPi : public HfMlResponse<TypeOutputScore>
192194
FILL_MAP_OMEGAC0(impactParCascXY),
193195
FILL_MAP_OMEGAC0(impactParCascZ),
194196
FILL_MAP_OMEGAC0(cosPACharmBaryon),
195-
FILL_MAP_OMEGAC0(cosPACasc)};
197+
FILL_MAP_OMEGAC0(cosPACasc),
198+
FILL_MAP_OMEGAC0(normDecayLenCharmBar)};
196199

197200
MlResponse<TypeOutputScore>::mAvailableInputFeatures.insert(mapDcaFeatures.begin(), mapDcaFeatures.end());
198201
}

‎PWGHF/DataModel/CandidateReconstructionTables.h‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1222,6 +1222,9 @@ DECLARE_SOA_DYNAMIC_COLUMN(PtPiFromCharmBaryon, ptPiFromCharmBaryon,
12221222
DECLARE_SOA_DYNAMIC_COLUMN(PtKaFromCasc, ptKaFromCasc,
12231223
[](float px, float py) -> float { return RecoDecay::pt(px, py); });
12241224

1225+
DECLARE_SOA_DYNAMIC_COLUMN(NormDecayLenCharmBar, normDecayLenCharmBar,
1226+
[](float decLen, float errDecLen) -> float { return decLen / errDecLen; });
1227+
12251228
} // end of namespace hf_cand_xic0_omegac0
12261229

12271230
// declare dedicated Omegac and Xic to Xi Pi candidate table
@@ -1288,7 +1291,8 @@ DECLARE_SOA_TABLE(HfCandToOmegaPi, "AOD", "HFCANDTOOMEGAPI",
12881291
hf_cand_xic0_omegac0::DcaXYToPvV0Dau0, hf_cand_xic0_omegac0::DcaXYToPvV0Dau1, hf_cand_xic0_omegac0::DcaXYToPvCascDau,
12891292
hf_cand_xic0_omegac0::DcaZToPvV0Dau0, hf_cand_xic0_omegac0::DcaZToPvV0Dau1, hf_cand_xic0_omegac0::DcaZToPvCascDau,
12901293
hf_cand_xic0_omegac0::DcaCascDau, hf_cand_xic0_omegac0::DcaV0Dau, hf_cand_xic0_omegac0::DcaCharmBaryonDau,
1291-
hf_cand_xic0_omegac0::DecLenCharmBaryon, hf_cand_xic0_omegac0::DecLenCascade, hf_cand_xic0_omegac0::DecLenV0, hf_cand_xic0_omegac0::ErrorDecayLengthCharmBaryon, hf_cand_xic0_omegac0::ErrorDecayLengthXYCharmBaryon, hf_track_index::HFflag);
1294+
hf_cand_xic0_omegac0::DecLenCharmBaryon, hf_cand_xic0_omegac0::DecLenCascade, hf_cand_xic0_omegac0::DecLenV0, hf_cand_xic0_omegac0::ErrorDecayLengthCharmBaryon, hf_cand_xic0_omegac0::ErrorDecayLengthXYCharmBaryon, hf_track_index::HFflag,
1295+
hf_cand_xic0_omegac0::NormDecayLenCharmBar<hf_cand_xic0_omegac0::DecLenCharmBaryon, hf_cand_xic0_omegac0::ErrorDecayLengthCharmBaryon>);
12921296

12931297
DECLARE_SOA_TABLE(HfCandToOmegaK, "AOD", "HFCANDTOOMEGAK",
12941298
o2::soa::Index<>,

‎PWGHF/TableProducer/treeCreatorOmegac0ToOmegaPiQa.cxx‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,7 @@ struct HfTreeCreatorToOmegaPiQa {
377377
candidate.errorDecayLengthCharmBaryon(),
378378
candidate.impactParCascXY() / candidate.errImpactParCascXY(),
379379
candidate.impactParBachFromCharmBaryonXY() / candidate.errImpactParBachFromCharmBaryonXY(),
380-
candidate.decLenCharmBaryon() / candidate.errorDecayLengthCharmBaryon(),
380+
candidate.normDecayLenCharmBar(),
381381
candidate.template bachelorFromCharmBaryon_as<MyTrackTable>().isGlobalTrackWoDCA(),
382382
candidate.template bachelorFromCharmBaryon_as<MyTrackTable>().itsNCls(),
383383
candidate.template bachelorFromCharmBaryon_as<MyTrackTable>().tpcNClsCrossedRows(),

0 commit comments

Comments
 (0)