|
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
70
71
72
73
74
75
76
77
|
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "CommonTools/UtilAlgos/interface/ObjectSelector.h"
//the selectores used to select the tracks
#include "Alignment/CommonAlignmentProducer/interface/AlignmentTrackSelector.h"
#include "Alignment/CommonAlignmentProducer/interface/AlignmentGlobalTrackSelector.h"
#include "Alignment/CommonAlignmentProducer/interface/AlignmentTwoBodyDecayTrackSelector.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
// the following include is necessary to clone all track branches
// including recoTrackExtras and TrackingRecHitsOwned (in future also "owned clusters"?).
// if you remove it the code will compile, but the cloned
// tracks have only the recoTracks branch!
#include "CommonTools/RecoAlgos/interface/TrackSelector.h"
struct TrackConfigSelector {
typedef std::vector<const reco::Track*> container;
typedef container::const_iterator const_iterator;
typedef reco::TrackCollection collection;
TrackConfigSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
: theBaseSelector(cfg, iC),
theGlobalSelector(cfg.getParameter<edm::ParameterSet>("GlobalSelector"), iC),
theTwoBodyDecaySelector(cfg.getParameter<edm::ParameterSet>("TwoBodyDecaySelector"), iC) {
//TODO Wrap the BaseSelector into its own PSet
theBaseSwitch = theBaseSelector.useThisFilter();
theGlobalSwitch = theGlobalSelector.useThisFilter();
theTwoBodyDecaySwitch = theTwoBodyDecaySelector.useThisFilter();
}
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& eSetup) {
theSelectedTracks.clear();
for (reco::TrackCollection::const_iterator i = c.product()->begin(); i != c.product()->end(); ++i) {
theSelectedTracks.push_back(&*i);
}
// might add EvetSetup to the select(...) method of the Selectors
if (theBaseSwitch)
theSelectedTracks = theBaseSelector.select(theSelectedTracks, evt, eSetup);
if (theGlobalSwitch)
theSelectedTracks = theGlobalSelector.select(theSelectedTracks, evt, eSetup);
if (theTwoBodyDecaySwitch)
theSelectedTracks = theTwoBodyDecaySelector.select(theSelectedTracks, evt, eSetup);
}
static void fillPSetDescription(edm::ParameterSetDescription& desc) {
AlignmentTrackSelector::fillPSetDescription(desc);
edm::ParameterSetDescription globalSelectorDesc;
AlignmentGlobalTrackSelector::fillPSetDescription(globalSelectorDesc);
desc.add<edm::ParameterSetDescription>("GlobalSelector", globalSelectorDesc);
edm::ParameterSetDescription twoBodySelectorDesc;
AlignmentTwoBodyDecayTrackSelector::fillPSetDescription(twoBodySelectorDesc);
desc.add<edm::ParameterSetDescription>("TwoBodyDecaySelector", twoBodySelectorDesc);
}
private:
container theSelectedTracks;
bool theBaseSwitch, theGlobalSwitch, theTwoBodyDecaySwitch;
AlignmentTrackSelector theBaseSelector;
AlignmentGlobalTrackSelector theGlobalSelector;
AlignmentTwoBodyDecayTrackSelector theTwoBodyDecaySelector;
};
typedef ObjectSelector<TrackConfigSelector> AlignmentTrackSelectorModule;
DEFINE_FWK_MODULE(AlignmentTrackSelectorModule);
|