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
31namespace o2::its
32{
34{
35 static constexpr int NLayers{7};
38
39 public:
41 int trgType,
42 const bool overrBeamEst)
43 : mIsMC{isMC},
44 mUseTriggers{trgType},
45 mOverrideBeamEstimation{overrBeamEst} {}
46
49 {
50 if (v == nullptr) {
51 LOGP(error, "Mean Vertex Object is nullptr");
52 return;
53 } else {
54 LOGP(info, "Mean Vertex set with x: {} y: {}", v->getX(), v->getY());
55 }
56 mMeanVertex = v;
57 }
58 // Task callbacks
59 void initialise();
61 void printSummary() const;
62
64 virtual void finaliseCCDB(framework::ConcreteDataMatcher& matcher, void* obj);
65
66 // Custom
67 void setTraitsFromProvider(VertexerTraits*, TrackerTraits7*, TimeFrame7*);
69 {
71 LOGP(fatal, "ITS Tracking mode Unset is meant to be a default. Specify the mode");
72 }
73 mMode = mode;
74 }
75
76 auto getTracker() const { return mTracker.get(); }
77 auto getVertexer() const { return mVertexer.get(); }
78
80
81 protected:
82 virtual void loadROF(gsl::span<itsmft::ROFRecord>& trackROFspan,
83 gsl::span<const itsmft::CompClusterExt> clusters,
84 gsl::span<const unsigned char>::iterator& pattIt,
87
88 private:
89 bool mIsMC = false;
90 bool mRunVertexer = true;
91 bool mCosmicsProcessing = false;
92 int mUseTriggers = 0;
94 bool mOverrideBeamEstimation = false;
95 const o2::itsmft::TopologyDictionary* mDict = nullptr;
96 std::unique_ptr<Tracker> mTracker = nullptr;
97 std::unique_ptr<Vertexer> mVertexer = nullptr;
98 const o2::dataformats::MeanVertexObject* mMeanVertex;
99 std::shared_ptr<BoundedMemoryResource> mMemoryPool;
100};
101
102} // namespace o2::its
103#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