|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
#ifndef FastMatchedTrackerRecHit_H
#define FastMatchedTrackerRecHit_H
#include "DataFormats/TrackerRecHit2D/interface/FastTrackerRecHit.h"
#include "DataFormats/TrackerRecHit2D/interface/FastSingleTrackerRecHit.h"
class FastMatchedTrackerRecHit : public FastTrackerRecHit {
public:
FastMatchedTrackerRecHit() : stereoHitFirst_(false) {}
~FastMatchedTrackerRecHit() override {}
FastMatchedTrackerRecHit(const LocalPoint& pos,
const LocalError& err,
const GeomDet& idet,
const FastSingleTrackerRecHit& rMono,
const FastSingleTrackerRecHit& rStereo,
bool stereoHitFirst = false)
: FastTrackerRecHit(pos, err, idet, fastTrackerRecHitType::siStripMatched2D),
stereoHitFirst_(stereoHitFirst),
componentMono_(rMono),
componentStereo_(rStereo) {}
FastMatchedTrackerRecHit* clone() const override {
FastMatchedTrackerRecHit* p = new FastMatchedTrackerRecHit(*this);
p->load();
return p;
}
size_t nIds() const override { return 2; }
int32_t id(size_t i = 0) const override { return i == 0 ? monoHit().id() : stereoHit().id(); }
int32_t eventId(size_t i = 0) const override { return i == 0 ? monoHit().eventId() : stereoHit().eventId(); }
size_t nSimTrackIds() const override {
return componentMono_.nSimTrackIds() + componentStereo_.nSimTrackIds();
} ///< see addSimTrackId(int32_t simTrackId)
int32_t simTrackId(size_t i) const override {
return i < componentMono_.nSimTrackIds() ? componentMono_.simTrackId(i)
: componentStereo_.simTrackId(i - componentMono_.nSimTrackIds());
}
int32_t simTrackEventId(size_t i) const override {
return i < componentMono_.nSimTrackIds() ? componentMono_.simTrackEventId(i)
: componentStereo_.simTrackEventId(i - componentMono_.nSimTrackIds());
}
const FastSingleTrackerRecHit& monoHit() const { return componentMono_; }
const FastSingleTrackerRecHit& stereoHit() const { return componentStereo_; }
const FastSingleTrackerRecHit& firstHit() const { return stereoHitFirst_ ? componentStereo_ : componentMono_; }
const FastSingleTrackerRecHit& secondHit() const { return stereoHitFirst_ ? componentMono_ : componentStereo_; }
void setStereoLayerFirst(bool stereoHitFirst = true) { stereoHitFirst_ = stereoHitFirst; }
void setEventId(int32_t eventId) override {
componentMono_.setEventId(eventId);
componentStereo_.setEventId(eventId);
}
void setRecHitCombinationIndex(int32_t recHitCombinationIndex) override {
FastTrackerRecHit::setRecHitCombinationIndex(recHitCombinationIndex);
componentMono_.setRecHitCombinationIndex(recHitCombinationIndex);
componentStereo_.setRecHitCombinationIndex(recHitCombinationIndex);
}
private:
bool stereoHitFirst_;
FastSingleTrackerRecHit componentMono_;
FastSingleTrackerRecHit componentStereo_;
};
#endif
|