Project
Loading...
Searching...
No Matches
TrackingInterface.h
Go to the documentation of this file.
1// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
2// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
3// All rights not expressly granted are reserved.
4//
5// This software is distributed under the terms of the GNU General Public
6// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
7//
8// In applying this license CERN does not waive the privileges and immunities
9// granted to it by virtue of its status as an Intergovernmental Organization
10// or submit itself to any jurisdiction.
11
12#ifndef O2_ITS_TRACKINGINTERFACE
13#define O2_ITS_TRACKINGINTERFACE
14
16
18#include "ITStracking/Tracker.h"
26
27#include "GPUDataTypes.h"
28#include "GPUO2Interface.h"
29#include "GPUChainITS.h"
30
31#include <oneapi/tbb/task_arena.h>
32
33namespace o2::its
34{
36{
37 static constexpr int NLayers{7};
41
42 public:
44 int trgType,
45 const bool overrBeamEst)
46 : mIsMC{isMC},
47 mUseTriggers{trgType},
48 mOverrideBeamEstimation{overrBeamEst} {}
49
52 {
53 if (v == nullptr) {
54 LOGP(error, "Mean Vertex Object is nullptr");
55 return;
56 } else {
57 LOGP(info, "Mean Vertex set with x: {} y: {}", v->getX(), v->getY());
58 }
59 mMeanVertex = v;
60 }
61 // Task callbacks
62 void initialise();
64 void printSummary() const;
65
67 virtual void finaliseCCDB(framework::ConcreteDataMatcher& matcher, void* obj);
68
69 // Custom
70 void setTraitsFromProvider(VertexerTraits*, TrackerTraitsN*, TimeFrameN*);
72
73 auto getTracker() const { return mTracker.get(); }
74 auto getVertexer() const { return mVertexer.get(); }
75
77
78 protected:
79 virtual void loadROF(gsl::span<itsmft::ROFRecord>& trackROFspan,
80 gsl::span<const itsmft::CompClusterExt> clusters,
81 gsl::span<const unsigned char>::iterator& pattIt,
83
84 private:
85 bool mIsMC = false;
86 bool mRunVertexer = true;
87 bool mCosmicsProcessing = false;
88 int mUseTriggers = 0;
90 bool mOverrideBeamEstimation = false;
91 const o2::itsmft::TopologyDictionary* mDict = nullptr;
92 std::unique_ptr<TrackerN> mTracker = nullptr;
93 std::unique_ptr<Vertexer> mVertexer = nullptr;
94 const o2::dataformats::MeanVertexObject* mMeanVertex;
95 std::shared_ptr<BoundedMemoryResource> mMemoryPool;
96 std::shared_ptr<tbb::task_arena> mTaskArena;
97};
98
99} // namespace o2::its
100#endif // O2_ITS_TRACKINGINTERFACE
Definition of the ClusterTopology class.
Header of the General Run Parameters object.
Class to compute the primary vertex in ITS from tracklets.
A container to hold and manage MC truth information/labels.
void run(framework::ProcessingContext &pc)
ITSTrackingInterface(bool isMC, int trgType, const bool overrBeamEst)
void setMeanVertex(const o2::dataformats::MeanVertexObject *v)
virtual void loadROF(gsl::span< itsmft::ROFRecord > &trackROFspan, gsl::span< const itsmft::CompClusterExt > clusters, gsl::span< const unsigned char >::iterator &pattIt, const dataformats::MCTruthContainer< MCCompLabel > *mcLabels)
void setTrackingMode(TrackingMode::Type mode=TrackingMode::Unset)
void setClusterDictionary(const o2::itsmft::TopologyDictionary *d)
virtual void updateTimeDependentParams(framework::ProcessingContext &pc)
virtual void finaliseCCDB(framework::ConcreteDataMatcher &matcher, void *obj)
void setTraitsFromProvider(VertexerTraits *, TrackerTraitsN *, TimeFrameN *)
GLenum mode
Definition glcorearb.h:266
const GLdouble * v
Definition glcorearb.h:832
std::vector< Cluster > clusters