|
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
|
#include "Alignment/CommonAlignmentProducer/interface/AlignmentCSCBeamHaloSelector.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/MuonDetId/interface/MuonSubdetId.h"
#include "DataFormats/MuonDetId/interface/CSCDetId.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
// constructor ----------------------------------------------------------------
AlignmentCSCBeamHaloSelector::AlignmentCSCBeamHaloSelector(const edm::ParameterSet &iConfig, edm::ConsumesCollector &iC)
: m_minStations(iConfig.getParameter<unsigned int>("minStations")),
m_minHitsPerStation(iConfig.getParameter<unsigned int>("minHitsPerStation")) {
edm::LogInfo("AlignmentCSCBeamHaloSelector")
<< "Acceptable tracks must have at least " << m_minHitsPerStation << " hits in " << m_minStations
<< " different CSC stations." << std::endl;
}
void AlignmentCSCBeamHaloSelector::fillPSetDescription(edm::ParameterSetDescription &desc) {
desc.add<unsigned int>("minStations", 0);
desc.add<unsigned int>("minHitsPerStation", 1);
}
// destructor -----------------------------------------------------------------
AlignmentCSCBeamHaloSelector::~AlignmentCSCBeamHaloSelector() {}
// do selection ---------------------------------------------------------------
AlignmentCSCBeamHaloSelector::Tracks AlignmentCSCBeamHaloSelector::select(const Tracks &tracks,
const edm::Event &iEvent) const {
Tracks result;
for (auto const &track : tracks) {
std::map<int, unsigned int> station_map;
for (auto const &hit : track->recHits()) {
DetId id = hit->geographicalId();
if (id.det() == DetId::Muon && id.subdetId() == MuonSubdetId::CSC) {
CSCDetId cscid(id.rawId());
int station = (cscid.endcap() == 1 ? 1 : -1) * cscid.station();
std::map<int, unsigned int>::const_iterator station_iter = station_map.find(station);
if (station_iter == station_map.end()) {
station_map[station] = 0;
}
station_map[station]++;
} // end if it's a CSC hit
} // end loop over hits
unsigned int stations = 0;
for (std::map<int, unsigned int>::const_iterator station_iter = station_map.begin();
station_iter != station_map.end();
++station_iter) {
if (station_iter->second > m_minHitsPerStation)
stations++;
}
if (stations >= m_minStations) {
result.push_back(track);
}
} // end loop over tracks
return result;
}
|