Skip to content

Commit aec4d2c

Browse files
authored
Fix for compatibility with O2 vertexer (#39)
1 parent 279cb08 commit aec4d2c

File tree

1 file changed

+9
-15
lines changed

1 file changed

+9
-15
lines changed

examples/aod/createO2tables.C

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -45,17 +45,7 @@ class TrackAlice3 : public o2::track::TrackParCov
4545
~TrackAlice3() = default;
4646
TrackAlice3(const TrackAlice3& src) = default;
4747
TrackAlice3(const o2::track::TrackParCov& src, const float t = 0, const float te = 1, const int label = 0) : o2::track::TrackParCov(src), mTimeMUS{t, te}, mLabel{label} {}
48-
4948
const timeEst& getTimeMUS() const { return mTimeMUS; }
50-
timeEst& getTimeMUS() { return mTimeMUS; }
51-
void setTimeMUS(const timeEst& t) { mTimeMUS = t; }
52-
void setTimeMUS(float t, float te)
53-
{
54-
mTimeMUS.setTimeStamp(t);
55-
mTimeMUS.setTimeStampError(te);
56-
}
57-
58-
void print() const;
5949
const int mLabel;
6050

6151
private:
@@ -177,6 +167,7 @@ void createO2tables(const char* inputFile = "delphes.root",
177167

178168
// loop over tracks
179169
std::vector<TrackAlice3> tracks_for_vertexing;
170+
std::vector<o2::InteractionRecord> bcData;
180171
std::vector<Track*> tof_tracks;
181172
for (Int_t itrack = 0; itrack < tracks->GetEntries(); ++itrack) {
182173

@@ -271,7 +262,7 @@ void createO2tables(const char* inputFile = "delphes.root",
271262
vertexer.setBunchFilling(bcfill);
272263
vertexer.init();
273264

274-
std::vector<o2::MCCompLabel> lblITS;
265+
std::vector<o2::MCCompLabel> lblTracks;
275266
std::vector<o2::vertexing::PVertex> vertices;
276267
std::vector<o2::vertexing::GIndex> vertexTrackIDs;
277268
std::vector<o2::vertexing::V2TRef> v2tRefs;
@@ -280,14 +271,17 @@ void createO2tables(const char* inputFile = "delphes.root",
280271
std::vector<o2::dataformats::GlobalTrackID> idxVec; // here we will the global IDs of all used tracks
281272
idxVec.reserve(tracks_for_vertexing.size());
282273
for (unsigned i = 0; i < tracks_for_vertexing.size(); i++) {
283-
lblITS.emplace_back(tracks_for_vertexing[i].mLabel, ientry + eventOffset, 1, false);
274+
lblTracks.emplace_back(tracks_for_vertexing[i].mLabel, ientry + eventOffset, 1, false);
284275
idxVec.emplace_back(i, o2::dataformats::GlobalTrackID::ITS);
285276
}
286-
std::vector<o2::ft0::RecPoints> ft0Data;
287277
vertexer.setStartIR({0, 0});
288278
const int n_vertices = vertexer.process(gsl::span<const TrackAlice3>{tracks_for_vertexing},
289-
idxVec, ft0Data, vertices, vertexTrackIDs, v2tRefs,
290-
gsl::span<const o2::MCCompLabel>{lblITS},
279+
idxVec,
280+
gsl::span<o2::InteractionRecord>{bcData},
281+
vertices,
282+
vertexTrackIDs,
283+
v2tRefs,
284+
gsl::span<const o2::MCCompLabel>{lblTracks},
291285
lblVtx);
292286
Printf("Found %i vertices with %zu tracks", n_vertices, tracks_for_vertexing.size());
293287
if (n_vertices == 0) {

0 commit comments

Comments
 (0)