|
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
|
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "CommonTools/UtilAlgos/interface/ObjectSelector.h"
//the selectores used to select the tracks
#include "Alignment/CommonAlignmentProducer/interface/AlignmentTracksFromVertexCompositeCandidateSelector.h"
// the following include is necessary to clone all track branches
// including recoTrackExtras and TrackingRecHitsOwned.
// if you remove it the code will compile, but the cloned
// tracks have only the recoTracks branch!
#include "CommonTools/RecoAlgos/interface/TrackSelector.h"
struct TrackFromVertexCompositeCandidateConfigSelector {
typedef std::vector<const reco::Track *> container;
typedef container::const_iterator const_iterator;
typedef reco::TrackCollection collection;
TrackFromVertexCompositeCandidateConfigSelector(const edm::ParameterSet &cfg, edm::ConsumesCollector &&iC)
: theBaseSelector(cfg, iC) {}
const_iterator begin() const { return theSelectedTracks.begin(); }
const_iterator end() const { return theSelectedTracks.end(); }
size_t size() const { return theSelectedTracks.size(); }
void select(const edm::Handle<reco::TrackCollection> &c, const edm::Event &evt, const edm::EventSetup &setup) {
theSelectedTracks = theBaseSelector.select(c, evt, setup);
}
private:
container theSelectedTracks;
AlignmentTrackFromVertexCompositeCandidateSelector theBaseSelector;
};
class AlignmentTrackFromVertexCompositeCandidateSelectorModule
: public ObjectSelector<TrackFromVertexCompositeCandidateConfigSelector> {
public:
AlignmentTrackFromVertexCompositeCandidateSelectorModule(const edm::ParameterSet &ps)
: ObjectSelector<TrackFromVertexCompositeCandidateConfigSelector>(ps) {}
static void fillDescriptions(edm::ConfigurationDescriptions &descriptions) {
edm::ParameterSetDescription desc;
desc.setComment("Alignment Tracks Selector from VertexCompositeCandidates");
desc.add<edm::InputTag>("src", edm::InputTag("generalTracks"));
desc.add<bool>("filter", false);
desc.add<edm::InputTag>("vertexCompositeCandidates", edm::InputTag("generalV0Candidates:Kshort"));
descriptions.addWithDefaultLabel(desc);
}
};
DEFINE_FWK_MODULE(AlignmentTrackFromVertexCompositeCandidateSelectorModule);
|