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 void end();
65
67 virtual void finaliseCCDB(framework::ConcreteDataMatcher& matcher, void* obj);
68
69 // Custom
70 void setTraitsFromProvider(VertexerTraits*, TrackerTraits7*, TimeFrame7*);
72 {
74 LOGP(fatal, "ITS Tracking mode Unset is meant to be a default. Specify the mode");
75 }
76 mMode = mode;
77 }
78
79 auto getTracker() const { return mTracker.get(); }
80 auto getVertexer() const { return mVertexer.get(); }
81
83
84 protected:
85 virtual void loadROF(gsl::span<itsmft::ROFRecord>& trackROFspan,
86 gsl::span<const itsmft::CompClusterExt> clusters,
87 gsl::span<const unsigned char>::iterator& pattIt,
90
91 private:
92 bool mIsMC = false;
93 bool mRunVertexer = true;
94 bool mCosmicsProcessing = false;
95 int mUseTriggers = 0;
97 bool mOverrideBeamEstimation = false;
98 const o2::itsmft::TopologyDictionary* mDict = nullptr;
99 std::unique_ptr<Tracker> mTracker = nullptr;
100 std::unique_ptr<Vertexer> mVertexer = nullptr;
101 const o2::dataformats::MeanVertexObject* mMeanVertex;
102 std::shared_ptr<BoundedMemoryResource> mMemoryPool;
103 std::shared_ptr<tbb::task_arena> mTaskArena;
104};
105
106} // namespace o2::its
107#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 setClusterDictionary(const o2::itsmft::TopologyDictionary *d)
void setTrackingMode(TrackingMode mode=TrackingMode::Unset)
virtual void updateTimeDependentParams(framework::ProcessingContext &pc)
void getConfiguration(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