Skip to content

Commit 0c4b214

Browse files
aalkinaalkin
committed
PWGHF: remove DCA from HFSelTrack table and use TracksExtended instead
1 parent 15648ff commit 0c4b214

9 files changed

Lines changed: 48 additions & 43 deletions

‎EventFiltering/PWGHF/HFFilter.cxx‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -278,10 +278,10 @@ struct HfFilter { // Main struct for HF triggers
278278
return false;
279279
}
280280

281-
if (std::abs(track.dcaPrim0()) < cutsSingleTrackBeauty[candType].get(pTBinTrack, "min_dcaxytoprimary")) {
281+
if (std::abs(track.dcaXY()) < cutsSingleTrackBeauty[candType].get(pTBinTrack, "min_dcaxytoprimary")) {
282282
return false; //minimum DCAxy
283283
}
284-
if (std::abs(track.dcaPrim0()) > cutsSingleTrackBeauty[candType].get(pTBinTrack, "max_dcaxytoprimary")) {
284+
if (std::abs(track.dcaXY()) > cutsSingleTrackBeauty[candType].get(pTBinTrack, "max_dcaxytoprimary")) {
285285
return false; //maximum DCAxy
286286
}
287287
return true;
@@ -339,8 +339,8 @@ struct HfFilter { // Main struct for HF triggers
339339

340340
using HfTrackIndexProng2withColl = soa::Join<aod::HfTrackIndexProng2, aod::Colls2Prong>;
341341
using HfTrackIndexProng3withColl = soa::Join<aod::HfTrackIndexProng3, aod::Colls3Prong>;
342-
using BigTracksWithProtonPID = soa::Join<aod::BigTracks, aod::pidTPCFullPr, aod::pidTOFFullPr>;
343-
using BigTracksMCPID = soa::Join<aod::BigTracksPID, aod::BigTracksMC>;
342+
using BigTracksWithProtonPID = soa::Join<aod::BigTracksExtended, aod::pidTPCFullPr, aod::pidTOFFullPr>;
343+
using BigTracksMCPID = soa::Join<aod::BigTracksPIDExtended, aod::BigTracksMC>;
344344

345345
void process(aod::Collision const& collision,
346346
HfTrackIndexProng2withColl const& cand2Prongs,
@@ -517,8 +517,8 @@ struct HfFilter { // Main struct for HF triggers
517517

518518
double pseudoRndm = trackPos.pt() * 1000. - (long)(trackPos.pt() * 1000);
519519
if ((fillSignal && indexRec > -1) || (fillBackground && indexRec < 0 && pseudoRndm < donwSampleBkgFactor)) {
520-
train2P(trackPos.pt(), trackPos.dcaPrim0(), trackPos.dcaPrim1(), trackPos.tpcNSigmaPi(), trackPos.tpcNSigmaKa(), trackPos.tofNSigmaPi(), trackPos.tofNSigmaKa(),
521-
trackNeg.pt(), trackNeg.dcaPrim0(), trackNeg.dcaPrim1(), trackNeg.tpcNSigmaPi(), trackNeg.tpcNSigmaKa(), trackNeg.tofNSigmaPi(), trackNeg.tofNSigmaKa(),
520+
train2P(trackPos.pt(), trackPos.dcaXY(), trackPos.dcaZ(), trackPos.tpcNSigmaPi(), trackPos.tpcNSigmaKa(), trackPos.tofNSigmaPi(), trackPos.tofNSigmaKa(),
521+
trackNeg.pt(), trackNeg.dcaXY(), trackNeg.dcaZ(), trackNeg.tpcNSigmaPi(), trackNeg.tpcNSigmaKa(), trackNeg.tofNSigmaPi(), trackNeg.tofNSigmaKa(),
522522
flag);
523523
}
524524
} // end loop over 2-prong candidates
@@ -576,9 +576,9 @@ struct HfFilter { // Main struct for HF triggers
576576

577577
double pseudoRndm = trackFirst.pt() * 1000. - (long)(trackFirst.pt() * 1000);
578578
if ((fillSignal && indexRec > -1) || (fillBackground && indexRec < 0 && pseudoRndm < donwSampleBkgFactor)) {
579-
train3P(trackFirst.pt(), trackFirst.dcaPrim0(), trackFirst.dcaPrim1(), trackFirst.tpcNSigmaPi(), trackFirst.tpcNSigmaKa(), trackFirst.tpcNSigmaPr(), trackFirst.tofNSigmaPi(), trackFirst.tofNSigmaKa(), trackFirst.tofNSigmaPr(),
580-
trackSecond.pt(), trackSecond.dcaPrim0(), trackSecond.dcaPrim1(), trackSecond.tpcNSigmaPi(), trackSecond.tpcNSigmaKa(), trackSecond.tpcNSigmaPr(), trackSecond.tofNSigmaPi(), trackSecond.tofNSigmaKa(), trackSecond.tofNSigmaPr(),
581-
trackThird.pt(), trackThird.dcaPrim0(), trackThird.dcaPrim1(), trackThird.tpcNSigmaPi(), trackThird.tpcNSigmaKa(), trackThird.tpcNSigmaPr(), trackThird.tofNSigmaPi(), trackThird.tofNSigmaKa(), trackThird.tofNSigmaPr(),
579+
train3P(trackFirst.pt(), trackFirst.dcaXY(), trackFirst.dcaZ(), trackFirst.tpcNSigmaPi(), trackFirst.tpcNSigmaKa(), trackFirst.tpcNSigmaPr(), trackFirst.tofNSigmaPi(), trackFirst.tofNSigmaKa(), trackFirst.tofNSigmaPr(),
580+
trackSecond.pt(), trackSecond.dcaXY(), trackSecond.dcaZ(), trackSecond.tpcNSigmaPi(), trackSecond.tpcNSigmaKa(), trackSecond.tpcNSigmaPr(), trackSecond.tofNSigmaPi(), trackSecond.tofNSigmaKa(), trackSecond.tofNSigmaPr(),
581+
trackThird.pt(), trackThird.dcaXY(), trackThird.dcaZ(), trackThird.tpcNSigmaPi(), trackThird.tpcNSigmaKa(), trackThird.tpcNSigmaPr(), trackThird.tofNSigmaPi(), trackThird.tofNSigmaKa(), trackThird.tofNSigmaPr(),
582582
flag, channel, cand3Prong.hfflag());
583583
}
584584
} // end loop over 3-prong candidates

‎PWGHF/DataModel/HFSecondaryVertex.h‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
#include "PWGHF/Core/HFSelectorCuts.h"
2525
#include "Common/Core/PID/PIDResponse.h"
2626
#include "Common/DataModel/StrangenessTables.h"
27+
#include "Common/DataModel/TrackSelectionTables.h"
2728

2829
using namespace o2::analysis;
2930

@@ -40,26 +41,24 @@ DECLARE_SOA_TABLE(HFSelCollision, "AOD", "HFSELCOLLISION", //!
4041
namespace hf_seltrack
4142
{
4243
DECLARE_SOA_COLUMN(IsSelProng, isSelProng, int); //!
43-
DECLARE_SOA_COLUMN(DCAPrim0, dcaPrim0, float); //!
44-
DECLARE_SOA_COLUMN(DCAPrim1, dcaPrim1, float); //!
4544
DECLARE_SOA_COLUMN(PxProng, pxProng, float); //!
4645
DECLARE_SOA_COLUMN(PyProng, pyProng, float); //!
4746
DECLARE_SOA_COLUMN(PzProng, pzProng, float); //!
4847
} // namespace hf_seltrack
4948

5049
DECLARE_SOA_TABLE(HFSelTrack, "AOD", "HFSELTRACK", //!
5150
hf_seltrack::IsSelProng,
52-
hf_seltrack::DCAPrim0,
53-
hf_seltrack::DCAPrim1,
5451
hf_seltrack::PxProng,
5552
hf_seltrack::PyProng,
5653
hf_seltrack::PzProng);
5754

5855
using BigTracks = soa::Join<Tracks, TracksCov, TracksExtra, HFSelTrack>;
56+
using BigTracksExtended = soa::Join<BigTracks, aod::TracksExtended>;
5957
using BigTracksMC = soa::Join<BigTracks, McTrackLabels>;
6058
using BigTracksPID = soa::Join<BigTracks,
6159
aod::pidTPCFullEl, aod::pidTPCFullMu, aod::pidTPCFullPi, aod::pidTPCFullKa, aod::pidTPCFullPr,
6260
aod::pidTOFFullEl, aod::pidTOFFullMu, aod::pidTOFFullPi, aod::pidTOFFullKa, aod::pidTOFFullPr>;
61+
using BigTracksPIDExtended = soa::Join<BigTracksPID, aod::TracksExtended>;
6362

6463
// FIXME: this is a workaround until we get the index columns to work with joins.
6564

‎PWGHF/TableProducer/HFD0CandidateSelector.cxx‎

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717

1818
#include "Framework/runDataProcessing.h"
1919
#include "Framework/AnalysisTask.h"
20+
#include "Common/DataModel/TrackSelectionTables.h"
2021
#include "PWGHF/DataModel/HFSecondaryVertex.h"
2122
#include "PWGHF/DataModel/HFCandidateSelectionTables.h"
2223
#include "Common/Core/TrackSelectorPID.h"
@@ -150,7 +151,7 @@ struct HFD0CandidateSelector {
150151
}
151152

152153
// cut on daughter DCA - need to add secondary vertex constraint here
153-
if (std::abs(trackPion.dcaPrim0()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaPrim0()) > cuts->get(pTBin, "d0K")) {
154+
if (std::abs(trackPion.dcaXY()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaXY()) > cuts->get(pTBin, "d0K")) {
154155
return false;
155156
}
156157

@@ -168,7 +169,7 @@ struct HFD0CandidateSelector {
168169
return true;
169170
}
170171

171-
void process(aod::HfCandProng2 const& candidates, aod::BigTracksPID const&)
172+
void process(aod::HfCandProng2 const& candidates, aod::BigTracksPIDExtended const&)
172173
{
173174
TrackSelectorPID selectorPion(kPiPlus);
174175
selectorPion.setRangePtTPC(d_pidTPCMinpT, d_pidTPCMaxpT);
@@ -198,8 +199,8 @@ struct HFD0CandidateSelector {
198199
}
199200
statusHFFlag = 1;
200201

201-
auto trackPos = candidate.index0_as<aod::BigTracksPID>(); // positive daughter
202-
auto trackNeg = candidate.index1_as<aod::BigTracksPID>(); // negative daughter
202+
auto trackPos = candidate.index0_as<aod::BigTracksPIDExtended>(); // positive daughter
203+
auto trackNeg = candidate.index1_as<aod::BigTracksPIDExtended>(); // negative daughter
203204

204205
/*
205206
if (!daughterSelection(trackPos) || !daughterSelection(trackNeg)) {

‎PWGHF/TableProducer/HFD0CandidateSelectorALICE3Barrel.cxx‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ struct HFD0CandidateSelectorALICE3Barrel {
171171
}
172172

173173
// cut on daughter DCA - need to add secondary vertex constraint here
174-
if (std::abs(trackPion.dcaPrim0()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaPrim0()) > cuts->get(pTBin, "d0K")) {
174+
if (std::abs(trackPion.dcaXY()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaXY()) > cuts->get(pTBin, "d0K")) {
175175
return false;
176176
}
177177

@@ -189,7 +189,7 @@ struct HFD0CandidateSelectorALICE3Barrel {
189189
return true;
190190
}
191191

192-
using Trks = soa::Join<aod::BigTracksPID, aod::RICHTracksIndex, aod::McTrackLabels>;
192+
using Trks = soa::Join<aod::BigTracksPIDExtended, aod::RICHTracksIndex, aod::McTrackLabels>;
193193
void process(aod::HfCandProng2 const& candidates, Trks const& barreltracks, const aod::McParticles& mcParticles, const aod::RICHs&, const aod::FRICHs&)
194194
{
195195

‎PWGHF/TableProducer/HFD0CandidateSelectorALICE3Forward.cxx‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ struct HFD0CandidateSelectorALICE3Forward {
171171
}
172172

173173
// cut on daughter DCA - need to add secondary vertex constraint here
174-
if (std::abs(trackPion.dcaPrim0()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaPrim0()) > cuts->get(pTBin, "d0K")) {
174+
if (std::abs(trackPion.dcaXY()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaXY()) > cuts->get(pTBin, "d0K")) {
175175
return false;
176176
}
177177

@@ -189,7 +189,7 @@ struct HFD0CandidateSelectorALICE3Forward {
189189
return true;
190190
}
191191

192-
using Trks = soa::Join<aod::BigTracksPID, aod::Tracks, aod::FRICHTracksIndex, aod::TracksExtra>;
192+
using Trks = soa::Join<aod::BigTracksPIDExtended, aod::Tracks, aod::FRICHTracksIndex, aod::TracksExtra>;
193193
void process(aod::HfCandProng2 const& candidates, Trks const& forwardtracks, const aod::McParticles& mcParticles, const aod::RICHs&, const aod::FRICHs&)
194194
{
195195

‎PWGHF/TableProducer/HFD0CandidateSelectorparametrizedPID.cxx‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ struct HFD0CandidateSelectorparametrizedPID {
171171
}
172172

173173
// cut on daughter DCA - need to add secondary vertex constraint here
174-
if (std::abs(trackPion.dcaPrim0()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaPrim0()) > cuts->get(pTBin, "d0K")) {
174+
if (std::abs(trackPion.dcaXY()) > cuts->get(pTBin, "d0pi") || std::abs(trackKaon.dcaXY()) > cuts->get(pTBin, "d0K")) {
175175
return false;
176176
}
177177

@@ -189,7 +189,7 @@ struct HFD0CandidateSelectorparametrizedPID {
189189
return true;
190190
}
191191

192-
using Trks = soa::Join<aod::BigTracksPID, aod::Tracks, aod::RICHTracksIndex, aod::McTrackLabels, aod::TracksExtra>;
192+
using Trks = soa::Join<aod::BigTracksPIDExtended, aod::Tracks, aod::RICHTracksIndex, aod::McTrackLabels, aod::TracksExtra>;
193193
void process(aod::HfCandProng2 const& candidates, Trks const& barreltracks, const aod::McParticles& mcParticles, const aod::RICHs&, const aod::FRICHs&)
194194
{
195195

‎PWGHF/TableProducer/HFJpsiCandidateSelector.cxx‎

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
/// \author Vít Kučera <vit.kucera@cern.ch>, CERN
1818

1919
#include "Framework/AnalysisTask.h"
20+
#include "Common/DataModel/TrackSelectionTables.h"
2021
#include "PWGHF/DataModel/HFSecondaryVertex.h"
2122
#include "PWGHF/DataModel/HFCandidateSelectionTables.h"
2223
#include "Common/Core/TrackSelectorPID.h"
@@ -120,12 +121,12 @@ struct HfJpsiCandidateSelector {
120121
}
121122

122123
// cut on daughter DCA - need to add secondary vertex constraint here
123-
if (std::abs(trackNeg.dcaPrim0()) > cuts->get(pTBin, "DCA_xy") || std::abs(trackPos.dcaPrim0()) > cuts->get(pTBin, "DCA_xy")) {
124+
if (std::abs(trackNeg.dcaXY()) > cuts->get(pTBin, "DCA_xy") || std::abs(trackPos.dcaXY()) > cuts->get(pTBin, "DCA_xy")) {
124125
return false;
125126
}
126127

127128
// cut on daughter DCA - need to add secondary vertex constraint here
128-
if (std::abs(trackNeg.dcaPrim1()) > cuts->get(pTBin, "DCA_z") || std::abs(trackPos.dcaPrim1()) > cuts->get(pTBin, "DCA_z")) {
129+
if (std::abs(trackNeg.dcaZ()) > cuts->get(pTBin, "DCA_z") || std::abs(trackPos.dcaZ()) > cuts->get(pTBin, "DCA_z")) {
129130
return false;
130131
}
131132

@@ -138,7 +139,9 @@ struct HfJpsiCandidateSelector {
138139

139140
using TracksPID = soa::Join<aod::BigTracksPID, aod::HfTrackIndexALICE3PID>;
140141

141-
void processAlice2(aod::HfCandProng2 const& candidates, aod::BigTracksPID const&)
142+
using ExtendedTracksPID = soa::Join<TracksPID, aod::TracksExtended>;
143+
144+
void processAlice2(aod::HfCandProng2 const& candidates, aod::BigTracksPIDExtended const&)
142145
{
143146
TrackSelectorPID selectorElectron(kElectron);
144147
selectorElectron.setRangePtTPC(d_pidTPCMinpT, d_pidTPCMaxpT);
@@ -159,8 +162,8 @@ struct HfJpsiCandidateSelector {
159162
continue;
160163
}
161164

162-
auto trackPos = candidate.index0_as<aod::BigTracksPID>(); // positive daughter
163-
auto trackNeg = candidate.index1_as<aod::BigTracksPID>(); // negative daughter
165+
auto trackPos = candidate.index0_as<aod::BigTracksPIDExtended>(); // positive daughter
166+
auto trackNeg = candidate.index1_as<aod::BigTracksPIDExtended>(); // negative daughter
164167

165168
int selectedEETopol = 1;
166169
int selectedEETpc = 1;
@@ -217,7 +220,7 @@ struct HfJpsiCandidateSelector {
217220

218221
PROCESS_SWITCH(HfJpsiCandidateSelector, processAlice2, "Use ALICE 2 detector setup", true);
219222

220-
void processAlice3(aod::HfCandProng2 const& candidates, TracksPID const&, aod::RICHs const&, aod::MIDs const&)
223+
void processAlice3(aod::HfCandProng2 const& candidates, ExtendedTracksPID const&, aod::RICHs const&, aod::MIDs const&)
221224
{
222225
TrackSelectorPID selectorElectron(kElectron);
223226
selectorElectron.setRangePtTPC(d_pidTPCMinpT, d_pidTPCMaxpT);
@@ -240,8 +243,8 @@ struct HfJpsiCandidateSelector {
240243
continue;
241244
}
242245

243-
auto trackPos = candidate.index0_as<TracksPID>(); // positive daughter
244-
auto trackNeg = candidate.index1_as<TracksPID>(); // negative daughter
246+
auto trackPos = candidate.index0_as<ExtendedTracksPID>(); // positive daughter
247+
auto trackNeg = candidate.index1_as<ExtendedTracksPID>(); // negative daughter
245248

246249
int selectedEETopol = 1;
247250
int selectedEETpc = 1;

‎PWGHF/TableProducer/HFTrackIndexSkimsCreator.cxx‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -503,7 +503,7 @@ struct HfTagSelTracks {
503503
}
504504

505505
// fill table row
506-
rowSelectedTrack(statusProng, dca[0], dca[1], track.px(), track.py(), track.pz());
506+
rowSelectedTrack(statusProng, track.px(), track.py(), track.pz());
507507
}
508508
}
509509
};
@@ -691,7 +691,7 @@ struct HfTrackIndexSkimsCreator {
691691

692692
// imp. par. product cut
693693
if (debug || TESTBIT(isSelected, iDecay2P)) {
694-
auto impParProduct = hfTrack0.dcaPrim0() * hfTrack1.dcaPrim0();
694+
auto impParProduct = hfTrack0.dcaXY() * hfTrack1.dcaXY();
695695
if (impParProduct > cut2Prong[iDecay2P].get(pTBin, d0d0Index[iDecay2P])) {
696696
CLRBIT(isSelected, iDecay2P);
697697
if (debug) {
@@ -885,7 +885,7 @@ struct HfTrackIndexSkimsCreator {
885885
Filter filterSelectTracks = aod::hf_seltrack::isSelProng > 0;
886886

887887
using SelectedCollisions = soa::Filtered<soa::Join<aod::Collisions, aod::HFSelCollision>>;
888-
using SelectedTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TracksCov, aod::TracksExtra, aod::HFSelTrack>>;
888+
using SelectedTracks = soa::Filtered<soa::Join<aod::Tracks, aod::TracksCov, aod::TracksExtra, aod::TracksExtended, aod::HFSelTrack>>;
889889

890890
// FIXME
891891
//Partition<SelectedTracks> tracksPos = aod::track::signed1Pt > 0.f;

‎PWGHF/Tasks/HFSelOptimisation.cxx‎

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
#include "Framework/runDataProcessing.h"
1818
#include "Framework/AnalysisTask.h"
1919
#include "Framework/HistogramRegistry.h"
20+
#include "Common/DataModel/TrackSelectionTables.h"
2021
#include "PWGHF/DataModel/HFSecondaryVertex.h"
2122
#include "PWGHF/DataModel/HFCandidateSelectionTables.h"
2223

@@ -171,12 +172,12 @@ struct HfSelOptimisation {
171172
/// \param candType is the candidate channel
172173
/// \param candOrig is candidate type (Prompt, NonPrompt, Bkg)
173174
/// \param candidate is a candidate
174-
/// \param tracks is the array of doughter tracks
175+
/// \param tracks is the array of daughter tracks
175176
template <std::size_t candType, std::size_t candOrig, typename T1, typename T2>
176177
void testSelections2Prong(const T1& candidate, const T2& tracks)
177178
{
178179
auto pT = candidate.pt();
179-
std::array<double, 2> absDCA{std::abs(tracks[0].dcaPrim0()), std::abs(tracks[1].dcaPrim0())};
180+
std::array<double, 2> absDCA{std::abs(tracks[0].dcaXY()), std::abs(tracks[1].dcaXY())};
180181
std::sort(absDCA.begin(), absDCA.end());
181182

182183
std::array<double, 2> ptTrack{tracks[0].pt(), tracks[1].pt()};
@@ -224,7 +225,7 @@ struct HfSelOptimisation {
224225
void testSelections3Prong(const T1& candidate, const T2& tracks)
225226
{
226227
auto pT = candidate.pt();
227-
std::array<double, 3> absDCA{std::abs(tracks[0].dcaPrim0()), std::abs(tracks[1].dcaPrim0()), std::abs(tracks[2].dcaPrim0())};
228+
std::array<double, 3> absDCA{std::abs(tracks[0].dcaXY()), std::abs(tracks[1].dcaXY()), std::abs(tracks[2].dcaXY())};
228229
std::sort(absDCA.begin(), absDCA.end());
229230

230231
std::array<double, 3> ptTrack{tracks[0].pt(), tracks[1].pt(), tracks[2].pt()};
@@ -257,15 +258,16 @@ struct HfSelOptimisation {
257258
}
258259
}
259260

261+
using ExtendedTracks = soa::Join<aod::BigTracks, aod::TracksExtended>;
260262
void process(soa::Join<aod::HfCandProng2, aod::HfCandProng2MCRec> const& cand2Prongs,
261263
soa::Join<aod::HfCandProng3, aod::HfCandProng3MCRec> const& cand3Prongs,
262-
aod::BigTracks const&)
264+
ExtendedTracks const&)
263265
{
264266
// looping over 2-prong candidates
265267
for (const auto& cand2Prong : cand2Prongs) {
266268

267-
auto trackPos = cand2Prong.index0_as<aod::BigTracks>(); // positive daughter
268-
auto trackNeg = cand2Prong.index1_as<aod::BigTracks>(); // negative daughter
269+
auto trackPos = cand2Prong.index0_as<ExtendedTracks>(); // positive daughter
270+
auto trackNeg = cand2Prong.index1_as<ExtendedTracks>(); // negative daughter
269271
std::array tracks = {trackPos, trackNeg};
270272

271273
bool isPrompt = false, isNonPrompt = false, isBkg = false;
@@ -319,9 +321,9 @@ struct HfSelOptimisation {
319321
// looping over 3-prong candidates
320322
for (const auto& cand3Prong : cand3Prongs) {
321323

322-
auto trackFirst = cand3Prong.index0_as<aod::BigTracks>(); // first daughter
323-
auto trackSecond = cand3Prong.index1_as<aod::BigTracks>(); // second daughter
324-
auto trackThird = cand3Prong.index2_as<aod::BigTracks>(); // third daughter
324+
auto trackFirst = cand3Prong.index0_as<ExtendedTracks>(); // first daughter
325+
auto trackSecond = cand3Prong.index1_as<ExtendedTracks>(); // second daughter
326+
auto trackThird = cand3Prong.index2_as<ExtendedTracks>(); // third daughter
325327
std::array tracks = {trackFirst, trackSecond, trackThird};
326328

327329
bool isPrompt = false, isNonPrompt = false, isBkg = false;

0 commit comments

Comments
 (0)