@@ -43,7 +43,7 @@ void EveWorkflowHelper::selectTracks(const CalibObjectsConst* calib,
4343 this ->mRecoCont .createTracksVariadic (creator);
4444}
4545
46- void EveWorkflowHelper::draw (std::string jsonPath, int numberOfFiles, int numberOfTracks,
46+ void EveWorkflowHelper::draw (const std::string& jsonPath, int numberOfFiles, int numberOfTracks,
4747 o2::dataformats::GlobalTrackID::mask_t trkMask,
4848 o2::dataformats::GlobalTrackID::mask_t clMask, float workflowVersion)
4949{
@@ -196,108 +196,60 @@ void EveWorkflowHelper::drawITSTPC(GID gid, float trackTime)
196196 // LOG(INFO) << "EveWorkflowHelper::drawITSTPC " << gid;
197197 const auto & track = mRecoCont .getTPCITSTrack (gid);
198198 addTrackToEvent (track, gid, trackTime, 0 .);
199- const auto detRefs = mRecoCont .getSingleDetectorRefs (gid);
200- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITS].isIndexSet ()) {
201- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITS], trackTime);
202- }
203- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB].isIndexSet ()) {
204- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB], trackTime);
205- }
206- if (detRefs[o2::dataformats::GlobalTrackID::Source::TPC].isIndexSet ()) {
207- drawTPCClusters (detRefs[o2::dataformats::GlobalTrackID::Source::TPC], trackTime);
208- }
199+ drawITSClusters (track.getRefITS (), trackTime);
200+ drawTPCClusters (track.getRefTPC (), trackTime);
209201}
210202
211203void EveWorkflowHelper::drawITSTPCTOF (GID gid, float trackTime)
212204{
213205 const auto & track = mRecoCont .getITSTPCTOFTrack (gid);
214206 addTrackToEvent (track, gid, trackTime, 0 .);
215- const auto detRefs = mRecoCont .getSingleDetectorRefs (gid);
216- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITS].isIndexSet ()) {
217- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITS], trackTime);
218- }
219- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB].isIndexSet ()) {
220- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB], trackTime);
221- }
222- if (detRefs[o2::dataformats::GlobalTrackID::Source::TPC].isIndexSet ()) {
223- drawTPCClusters (detRefs[o2::dataformats::GlobalTrackID::Source::TPC], trackTime);
224- }
207+ drawITSClusters (track.getRefITS (), trackTime);
208+ drawTPCClusters (track.getRefTPC (), trackTime);
225209 drawTOFClusters (gid, trackTime);
226210}
227211
228212void EveWorkflowHelper::drawTPCTRD (GID gid, float trackTime)
229213{
230214 // LOG(INFO) << "EveWorkflowHelper::drawTPCTRD " << gid;
231-
232- const auto detRefs = mRecoCont .getSingleDetectorRefs (gid);
233- if (detRefs[o2::dataformats::GlobalTrackID::Source::TPC].isIndexSet ()) {
234- const auto & tpcTrack = mRecoCont .getTPCTrack (detRefs[o2::dataformats::GlobalTrackID::Source::TPC]);
235- addTrackToEvent (tpcTrack, gid, trackTime, 0 );
236- drawTPCClusters (detRefs[o2::dataformats::GlobalTrackID::Source::TPC], trackTime);
237- }
238215 const auto & tpcTrdTrack = mRecoCont .getTPCTRDTrack <o2::trd::TrackTRD>(gid);
216+ drawTPCClusters (tpcTrdTrack.getRefGlobalTrackId (), trackTime);
239217 drawTRDClusters (tpcTrdTrack, trackTime);
240218}
241219
242220void EveWorkflowHelper::drawITSTPCTRD (GID gid, float trackTime)
243221{
244222 // LOG(INFO) << "EveWorkflowHelper::drawITSTPCTRD " << gid;
245- const auto & itsTpcTrdTrack = mRecoCont .getTPCTRDTrack <o2::trd::TrackTRD>(gid);
246- addTrackToEvent (itsTpcTrdTrack, gid, trackTime, 0 );
247- const auto detRefs = mRecoCont .getSingleDetectorRefs (gid);
248- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITS].isIndexSet ()) {
249- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITS], trackTime);
250- }
251- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB].isIndexSet ()) {
252- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB], trackTime);
253- }
254- if (detRefs[o2::dataformats::GlobalTrackID::Source::TPC].isIndexSet ()) {
255- drawTPCClusters (detRefs[o2::dataformats::GlobalTrackID::Source::TPC], trackTime);
256- }
223+ const auto & itsTpcTrdTrack = mRecoCont .getITSTPCTRDTrack <o2::trd::TrackTRD>(gid);
224+ drawITSTPC (itsTpcTrdTrack.getRefGlobalTrackId (), trackTime);
257225 drawTRDClusters (itsTpcTrdTrack, trackTime);
258226}
259227
260228void EveWorkflowHelper::drawITSTPCTRDTOF (GID gid, float trackTime)
261229{
262230 // LOG(INFO) << "EveWorkflowHelper::drawITSTPCTRDTOF " << gid;
263- const auto & itsTpcTrdTofTrack = mRecoCont .getTPCTRDTrack <o2::trd::TrackTRD>(gid);
264- addTrackToEvent (itsTpcTrdTofTrack, gid, trackTime, 0 );
265- const auto detRefs = mRecoCont .getSingleDetectorRefs (gid);
266- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITS].isIndexSet ()) {
267- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITS], trackTime);
268- }
269- if (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB].isIndexSet ()) {
270- drawITSClusters (detRefs[o2::dataformats::GlobalTrackID::Source::ITSAB], trackTime);
271- }
272- if (detRefs[o2::dataformats::GlobalTrackID::Source::TPC].isIndexSet ()) {
273- drawTPCClusters (detRefs[o2::dataformats::GlobalTrackID::Source::TPC], trackTime);
274- }
275- drawTRDClusters (itsTpcTrdTofTrack, trackTime);
231+ const auto & match = mRecoCont .getITSTPCTRDTOFMatches ()[gid.getIndex ()];
232+ auto gidITSTPCTRD = match.getTrackRef ();
233+ drawITSTPCTRD (gidITSTPCTRD, trackTime);
276234 drawTOFClusters (gid, trackTime);
277235}
278236
279237void EveWorkflowHelper::drawTPCTRDTOF (GID gid, float trackTime)
280238{
281239 // LOG(INFO) << "EveWorkflowHelper::drawTPCTRDTOF " << gid;
282- const auto & tpcTrdTofTrack = mRecoCont .getTPCTRDTrack <o2::trd::TrackTRD>(gid);
283- addTrackToEvent (tpcTrdTofTrack, gid, trackTime, 0 );
284- const auto detRefs = mRecoCont .getSingleDetectorRefs (gid);
285- if (detRefs[o2::dataformats::GlobalTrackID::Source::TPC].isIndexSet ()) {
286- drawTPCClusters (detRefs[o2::dataformats::GlobalTrackID::Source::TPC], trackTime);
287- }
288- drawTRDClusters (tpcTrdTofTrack, trackTime);
240+ const auto & match = mRecoCont .getTPCTRDTOFMatches ()[gid.getIndex ()];
241+ auto gidTPCTRD = match.getTrackRef ();
242+ drawTPCTRD (gidTPCTRD, trackTime);
289243 drawTOFClusters (gid, trackTime);
290244}
291245
292246void EveWorkflowHelper::drawTPCTOF (GID gid, float trackTime)
293247{
294248 // LOG(INFO) << "EveWorkflowHelper::drawTPCTRDTOF " << gid;
295- const auto detRefs = mRecoCont .getSingleDetectorRefs (gid);
296- if (detRefs[o2::dataformats::GlobalTrackID::Source::TPC].isIndexSet ()) {
297- const auto & tpcTrack = mRecoCont .getTPCTrack (detRefs[o2::dataformats::GlobalTrackID::Source::TPC]);
298- addTrackToEvent (tpcTrack, gid, trackTime, 0 );
299- drawTPCClusters (detRefs[o2::dataformats::GlobalTrackID::Source::TPC], trackTime);
300- }
249+ const auto & trTPCTOF = mRecoCont .getTPCTOFTrack (gid);
250+ const auto & match = mRecoCont .getTPCTOFMatch (gid.getIndex ());
251+ addTrackToEvent (trTPCTOF, gid, trackTime, 0 );
252+ drawTPCClusters (match.getTrackRef (), trackTime);
301253 drawTOFClusters (gid, trackTime);
302254}
303255
0 commit comments