|
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
|
/** \class AlignmentMuonSelectorModule
*
* selects a subset of a muon collection and clones
* Track, TrackExtra parts and RecHits collection
* for SA, GB and Tracker Only options
*
* \author Javier Fernandez, IFCA
*
* \version $Revision: 1.3 $
*
* $Id: AlignmentMuonSelectorModule.cc,v 1.3 2008/02/04 19:32:26 flucke Exp $
*
*/
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "CommonTools/UtilAlgos/interface/ObjectSelector.h"
#include "Alignment/CommonAlignmentProducer/interface/AlignmentMuonSelector.h"
#include "DataFormats/MuonReco/interface/MuonFwd.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 recoMuons branch!
struct MuonConfigSelector {
typedef std::vector<const reco::Muon *> container;
typedef container::const_iterator const_iterator;
typedef reco::MuonCollection collection;
MuonConfigSelector(const edm::ParameterSet &cfg, edm::ConsumesCollector &&iC) : theSelector(cfg) {}
const_iterator begin() const { return selected_.begin(); }
const_iterator end() const { return selected_.end(); }
size_t size() const { return selected_.size(); }
void select(const edm::Handle<reco::MuonCollection> &c, const edm::Event &evt, const edm::EventSetup & /* dummy*/) {
all_.clear();
selected_.clear();
for (collection::const_iterator i = c.product()->begin(), iE = c.product()->end(); i != iE; ++i) {
all_.push_back(&*i);
}
selected_ = theSelector.select(all_, evt); // might add dummy
}
static void fillPSetDescription(edm::ParameterSetDescription &desc) {
AlignmentMuonSelector::fillPSetDescription(desc);
}
private:
container all_, selected_;
AlignmentMuonSelector theSelector;
};
typedef ObjectSelector<MuonConfigSelector> AlignmentMuonSelectorModule;
DEFINE_FWK_MODULE(AlignmentMuonSelectorModule);
|