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};
40
41 public:
43 int trgType,
44 const bool overrBeamEst)
45 : mIsMC{isMC},
46 mUseTriggers{trgType},
47 mOverrideBeamEstimation{overrBeamEst} {}
48
51 {
52 if (v == nullptr) {
53 LOGP(error, "Mean Vertex Object is nullptr");
54 return;
55 } else {
56 LOGP(info, "Mean Vertex set with x: {} y: {}", v->getX(), v->getY());
57 }
58 mMeanVertex = v;
59 }
60 // Task callbacks
61 void initialise();
63 void printSummary() const;
64
66 virtual void finaliseCCDB(framework::ConcreteDataMatcher& matcher, void* obj);
67
68 // Custom
69 void setTraitsFromProvider(VertexerTraits*, TrackerTraits7*, TimeFrame7*);
71
72 auto getTracker() const { return mTracker.get(); }
73 auto getVertexer() const { return mVertexer.get(); }
74
76
77 protected:
78 virtual void loadROF(gsl::span<itsmft::ROFRecord>& trackROFspan,
79 gsl::span<const itsmft::CompClusterExt> clusters,
80 gsl::span<const unsigned char>::iterator& pattIt,
82
83 private:
84 bool mIsMC = false;
85 bool mRunVertexer = true;
86 bool mCosmicsProcessing = false;
87 int mUseTriggers = 0;
89 bool mOverrideBeamEstimation = false;
90 const o2::itsmft::TopologyDictionary* mDict = nullptr;
91 std::unique_ptr<Tracker> mTracker = nullptr;
92 std::unique_ptr<Vertexer> mVertexer = nullptr;
93 const o2::dataformats::MeanVertexObject* mMeanVertex;
94 std::shared_ptr<BoundedMemoryResource> mMemoryPool;
95 std::shared_ptr<tbb::task_arena> mTaskArena;
96};
97
98} // namespace o2::its
99#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 *, TrackerTraits7 *, TimeFrame7 *)
GLenum mode
Definition glcorearb.h:266
const GLdouble * v
Definition glcorearb.h:832
std::vector< Cluster > clusters