Skip to content

Commit 6c9e913

Browse files
committed
Add missing if condition for TrackQA table in D0 data creator for calibration studies
1 parent 644fb83 commit 6c9e913

File tree

1 file changed

+37
-35
lines changed

1 file changed

+37
-35
lines changed

DPG/Tasks/AOTTrack/derivedDataCreatorD0Calibration.cxx

Lines changed: 37 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ struct DerivedDataCreatorD0Calibration {
169169
}
170170

171171
// main function
172-
template <bool withTrackQa, typename TTrackQa>
172+
template<bool withTrackQa, typename TTrackQa>
173173
void runDataCreation(CollisionsWEvSel const& collisions,
174174
aod::TrackAssoc const& trackIndices,
175175
TracksWCovExtraPid const&,
@@ -421,7 +421,7 @@ struct DerivedDataCreatorD0Calibration {
421421
// apply BDT models
422422
if (cfgMl.apply) {
423423
std::vector<float> featuresCandD0 = {dcaPos.getY(), dcaNeg.getY(), chi2PCA, cosPaD0, cosPaXYD0, decLenXYD0, decLenD0, dcaPos.getY() * dcaNeg.getY(), aod::pid_tpc_tof_utils::combineNSigma<false>(trackPos.tpcNSigmaPi(), trackPos.tofNSigmaPi()), aod::pid_tpc_tof_utils::combineNSigma<false>(trackNeg.tpcNSigmaKa(), trackNeg.tofNSigmaKa()), trackPos.tpcNSigmaPi(), trackPos.tpcNSigmaKa(), aod::pid_tpc_tof_utils::combineNSigma<false>(trackPos.tpcNSigmaKa(), trackPos.tofNSigmaKa()), trackNeg.tpcNSigmaPi(), trackNeg.tpcNSigmaKa(), aod::pid_tpc_tof_utils::combineNSigma<false>(trackNeg.tpcNSigmaPi(), trackNeg.tofNSigmaPi())};
424-
if (!mlResponse.isSelectedMl(featuresCandD0, ptD0, bdtScoresD0)) {
424+
if(!mlResponse.isSelectedMl(featuresCandD0, ptD0, bdtScoresD0)) {
425425
massHypo -= D0MassHypo::D0;
426426
}
427427
}
@@ -436,7 +436,7 @@ struct DerivedDataCreatorD0Calibration {
436436
// apply BDT models
437437
if (cfgMl.apply) {
438438
std::vector<float> featuresCandD0bar = {dcaPos.getY(), dcaNeg.getY(), chi2PCA, cosPaD0, cosPaXYD0, decLenXYD0, decLenD0, dcaPos.getY() * dcaNeg.getY(), aod::pid_tpc_tof_utils::combineNSigma<false>(trackNeg.tpcNSigmaPi(), trackNeg.tofNSigmaPi()), aod::pid_tpc_tof_utils::combineNSigma<false>(trackPos.tpcNSigmaKa(), trackPos.tofNSigmaKa()), trackNeg.tpcNSigmaPi(), trackNeg.tpcNSigmaKa(), aod::pid_tpc_tof_utils::combineNSigma<false>(trackNeg.tpcNSigmaKa(), trackNeg.tofNSigmaKa()), trackPos.tpcNSigmaPi(), trackPos.tpcNSigmaKa(), aod::pid_tpc_tof_utils::combineNSigma<false>(trackPos.tpcNSigmaPi(), trackPos.tofNSigmaPi())};
439-
if (!mlResponse.isSelectedMl(featuresCandD0bar, ptD0, bdtScoresD0bar)) {
439+
if(!mlResponse.isSelectedMl(featuresCandD0bar, ptD0, bdtScoresD0bar)) {
440440
massHypo -= D0MassHypo::D0Bar;
441441
}
442442
}
@@ -686,33 +686,35 @@ struct DerivedDataCreatorD0Calibration {
686686
int8_t deltaRefGloParamQ2Pt{0};
687687
int8_t deltaTOFdX{0};
688688
int8_t deltaTOFdZ{0};
689-
if (trackNeg.has_trackQA()) {
690-
auto trackQA = trackNeg.trackQA_as<aod::TracksQAVersion>(); // obtain track QA
691-
tpcTime0 = trackQA.tpcTime0();
692-
tpcdEdxNorm = trackQA.tpcdEdxNorm();
693-
tpcDcaR = trackQA.tpcdcaR();
694-
tpcDcaZ = trackQA.tpcdcaZ();
695-
tpcClusterByteMask = trackQA.tpcClusterByteMask();
696-
tpcdEdxMax0R = trackQA.tpcdEdxMax0R();
697-
tpcdEdxMax1R = trackQA.tpcdEdxMax1R();
698-
tpcdEdxMax2R = trackQA.tpcdEdxMax2R();
699-
tpcdEdxMax3R = trackQA.tpcdEdxMax3R();
700-
tpcdEdxTot0R = trackQA.tpcdEdxTot0R();
701-
tpcdEdxTot1R = trackQA.tpcdEdxTot1R();
702-
tpcdEdxTot2R = trackQA.tpcdEdxTot2R();
703-
tpcdEdxTot3R = trackQA.tpcdEdxTot3R();
704-
deltaRefContParamY = trackQA.deltaRefContParamY();
705-
deltaRefITSParamZ = trackQA.deltaRefITSParamZ();
706-
deltaRefContParamSnp = trackQA.deltaRefContParamSnp();
707-
deltaRefContParamTgl = trackQA.deltaRefContParamTgl();
708-
deltaRefContParamQ2Pt = trackQA.deltaRefContParamQ2Pt();
709-
deltaRefGloParamY = trackQA.deltaRefGloParamY();
710-
deltaRefGloParamZ = trackQA.deltaRefGloParamZ();
711-
deltaRefGloParamSnp = trackQA.deltaRefGloParamSnp();
712-
deltaRefGloParamTgl = trackQA.deltaRefGloParamTgl();
713-
deltaRefGloParamQ2Pt = trackQA.deltaRefGloParamQ2Pt();
714-
deltaTOFdX = trackQA.deltaTOFdX();
715-
deltaTOFdZ = trackQA.deltaTOFdZ();
689+
if constexpr (withTrackQa) {
690+
if (trackNeg.has_trackQA()) {
691+
auto trackQA = trackNeg.template trackQA_as<TTrackQa>(); // obtain track QA
692+
tpcTime0 = trackQA.tpcTime0();
693+
tpcdEdxNorm = trackQA.tpcdEdxNorm();
694+
tpcDcaR = trackQA.tpcdcaR();
695+
tpcDcaZ = trackQA.tpcdcaZ();
696+
tpcClusterByteMask = trackQA.tpcClusterByteMask();
697+
tpcdEdxMax0R = trackQA.tpcdEdxMax0R();
698+
tpcdEdxMax1R = trackQA.tpcdEdxMax1R();
699+
tpcdEdxMax2R = trackQA.tpcdEdxMax2R();
700+
tpcdEdxMax3R = trackQA.tpcdEdxMax3R();
701+
tpcdEdxTot0R = trackQA.tpcdEdxTot0R();
702+
tpcdEdxTot1R = trackQA.tpcdEdxTot1R();
703+
tpcdEdxTot2R = trackQA.tpcdEdxTot2R();
704+
tpcdEdxTot3R = trackQA.tpcdEdxTot3R();
705+
deltaRefContParamY = trackQA.deltaRefContParamY();
706+
deltaRefITSParamZ = trackQA.deltaRefITSParamZ();
707+
deltaRefContParamSnp = trackQA.deltaRefContParamSnp();
708+
deltaRefContParamTgl = trackQA.deltaRefContParamTgl();
709+
deltaRefContParamQ2Pt = trackQA.deltaRefContParamQ2Pt();
710+
deltaRefGloParamY = trackQA.deltaRefGloParamY();
711+
deltaRefGloParamZ = trackQA.deltaRefGloParamZ();
712+
deltaRefGloParamSnp = trackQA.deltaRefGloParamSnp();
713+
deltaRefGloParamTgl = trackQA.deltaRefGloParamTgl();
714+
deltaRefGloParamQ2Pt = trackQA.deltaRefGloParamQ2Pt();
715+
deltaTOFdX = trackQA.deltaTOFdX();
716+
deltaTOFdZ = trackQA.deltaTOFdZ();
717+
}
716718
}
717719

718720
trackTable(selectedCollisions[collision.globalIndex()], // stored at PV
@@ -832,11 +834,11 @@ struct DerivedDataCreatorD0Calibration {
832834

833835
// process functions
834836
void processWithTrackQa(CollisionsWEvSel const& collisions,
835-
aod::TrackAssoc const& trackIndices,
836-
TracksWCovExtraPid const& tracks,
837-
aod::BCsWithTimestamps const& bcs,
838-
TrackMeanOccs const& occ,
839-
aod::TracksQAVersion const& trackQa)
837+
aod::TrackAssoc const& trackIndices,
838+
TracksWCovExtraPid const& tracks,
839+
aod::BCsWithTimestamps const& bcs,
840+
TrackMeanOccs const& occ,
841+
aod::TracksQAVersion const& trackQa)
840842
{
841843
runDataCreation<true>(collisions, trackIndices, tracks, bcs, occ, trackQa);
842844
}

0 commit comments

Comments
 (0)