CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
MatcherUsingTracks.cc
Go to the documentation of this file.
1 //
2 //
3 
15 
18 
20 
22 
24 
25 // template-related workaround for bug in OwnVector+Ptr
26 namespace edm {
27  using std::advance;
28 }
29 
30 namespace pat {
31 
33  public:
34  explicit MatcherUsingTracks(const edm::ParameterSet &iConfig);
35  ~MatcherUsingTracks() override {}
36 
37  void produce(edm::Event &iEvent, const edm::EventSetup &iSetup) override;
38 
39  private:
43 
46 
50 
52  template <typename T>
55  const std::vector<T> &values,
56  const std::string &label) const;
57  };
58 
59 } // namespace pat
60 
62  : srcToken_(consumes<edm::View<reco::Candidate> >(iConfig.getParameter<edm::InputTag>("src"))),
63  matchedToken_(consumes<edm::View<reco::Candidate> >(iConfig.getParameter<edm::InputTag>("matched"))),
64  algo_(iConfig, consumesCollector()),
65  dontFailOnMissingInput_(iConfig.existsAs<bool>("dontFailOnMissingInput")
66  ? iConfig.getParameter<bool>("dontFailOnMissingInput")
67  : false),
68  writeExtraPATOutput_(
69  iConfig.existsAs<bool>("writeExtraPATOutput") ? iConfig.getParameter<bool>("writeExtraPATOutput") : false) {
70  // this is the basic output (edm::Association is not generic)
71  produces<edm::ValueMap<reco::CandidatePtr> >();
73  // this is the crazy stuff to get the same with UserData
74  produces<edm::OwnVector<pat::UserData> >();
75  produces<edm::ValueMap<edm::Ptr<pat::UserData> > >();
76  }
77  // this is the extra stuff
78  if (algo_.hasMetrics()) {
79  produces<edm::ValueMap<float> >("deltaR");
80  produces<edm::ValueMap<float> >("deltaEta");
81  produces<edm::ValueMap<float> >("deltaPhi");
82  produces<edm::ValueMap<float> >("deltaLocalPos");
83  produces<edm::ValueMap<float> >("deltaPtRel");
84  if (algo_.hasChi2()) {
85  produces<edm::ValueMap<float> >("chi2");
86  }
87  } else {
88  produces<edm::ValueMap<int> >("matched");
89  }
90 }
91 
93  using namespace edm;
94  using namespace std;
95 
96  algo_.init(iSetup);
97 
99 
100  iEvent.getByToken(srcToken_, src);
101  iEvent.getByToken(matchedToken_, matched);
102 
103  // declare loop variables and some intermediate stuff
105  int isrc, nsrc = src->size();
106 
107  // working and output variables
108  vector<int> match(nsrc, -1);
109  vector<float> deltaRs(nsrc, 999);
110  vector<float> deltaEtas(nsrc, 999);
111  vector<float> deltaPhis(nsrc, 999);
112  vector<float> deltaPtRel(nsrc, 999);
113  vector<float> deltaLocalPos(nsrc, 999);
114  vector<float> chi2(nsrc, 999999);
115 
116  // don't try matching if the input collection is missing and the module is configured to fail silently
117  if (!(matched.failedToGet() && dontFailOnMissingInput_)) {
118  // loop on the source collection, and request for the match
119  for (itsrc = src->begin(), edsrc = src->end(), isrc = 0; itsrc != edsrc; ++itsrc, ++isrc) {
120  match[isrc] = algo_.match(*itsrc,
121  *matched,
122  deltaRs[isrc],
123  deltaEtas[isrc],
124  deltaPhis[isrc],
125  deltaLocalPos[isrc],
126  deltaPtRel[isrc],
127  chi2[isrc]);
128  }
129  }
130 
131  std::vector<reco::CandidatePtr> ptrs(nsrc);
132  for (isrc = 0; isrc < nsrc; ++isrc) {
133  if (match[isrc] != -1) {
134  ptrs[isrc] = matched->ptrAt(match[isrc]);
135  }
136  }
137  storeValueMap<reco::CandidatePtr>(iEvent, src, ptrs, "");
138 
139  if (writeExtraPATOutput_) {
140  auto outUDVect = std::make_unique<edm::OwnVector<pat::UserData> >();
141  std::vector<int> idxUD(nsrc, -1);
142  for (isrc = 0; isrc < nsrc; ++isrc) {
143  if (match[isrc] != -1) {
144  outUDVect->push_back(pat::UserData::make(ptrs[isrc]));
145  idxUD[isrc] = outUDVect->size() - 1;
146  }
147  }
149  std::vector<edm::Ptr<pat::UserData> > ptrUD(nsrc);
150  for (isrc = 0; isrc < nsrc; ++isrc) {
151  if (idxUD[isrc] != -1)
152  ptrUD[isrc] = edm::Ptr<pat::UserData>(doneUDVect, idxUD[isrc]);
153  }
154  storeValueMap<edm::Ptr<pat::UserData> >(iEvent, src, ptrUD, "");
155  }
156 
157  if (algo_.hasMetrics()) {
158  storeValueMap<float>(iEvent, src, deltaRs, "deltaR");
159  storeValueMap<float>(iEvent, src, deltaEtas, "deltaEta");
160  storeValueMap<float>(iEvent, src, deltaPhis, "deltaPhi");
161  storeValueMap<float>(iEvent, src, deltaLocalPos, "deltaLocalPos");
162  storeValueMap<float>(iEvent, src, deltaPtRel, "deltaPtRel");
163  if (algo_.hasChi2()) {
164  storeValueMap<float>(iEvent, src, chi2, "chi2");
165  }
166  } else {
167  std::vector<int> ismatched(nsrc, 0);
168  for (isrc = 0; isrc < nsrc; ++isrc) {
169  ismatched[isrc] = (match[isrc] != -1);
170  }
171  storeValueMap<int>(iEvent, src, ismatched, "matched");
172  }
173 }
174 
175 template <typename T>
178  const std::vector<T> &values,
179  const std::string &label) const {
180  using namespace edm;
181  using namespace std;
182  unique_ptr<ValueMap<T> > valMap(new ValueMap<T>());
183  typename edm::ValueMap<T>::Filler filler(*valMap);
184  filler.insert(handle, values.begin(), values.end());
185  filler.fill();
186  iEvent.put(std::move(valMap), label);
187 }
188 
190 using namespace pat;
edm::EDGetTokenT< edm::View< reco::Candidate > > matchedToken_
Matcher of reconstructed objects to other reconstructed objects using the tracks inside them...
edm::EDGetTokenT< edm::View< reco::Candidate > > srcToken_
Labels for input collections.
static std::unique_ptr< UserData > make(const T &value, bool transientOnly=false)
Definition: HeavyIon.h:7
MatcherUsingTracksAlgorithm algo_
The real workhorse.
char const * label
int iEvent
Definition: GenABIO.cc:224
Matcher of reconstructed objects to other reconstructed objects using the tracks inside them...
void storeValueMap(edm::Event &iEvent, const edm::Handle< edm::View< reco::Candidate > > &handle, const std::vector< T > &values, const std::string &label) const
Store extra information in a ValueMap.
void produce(edm::Event &iEvent, const edm::EventSetup &iSetup) override
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:16
bool dontFailOnMissingInput_
Some extra configurables.
fixed size matrix
HLT enums.
boost::indirect_iterator< typename seq_t::const_iterator > const_iterator
Definition: View.h:88
MatcherUsingTracks(const edm::ParameterSet &iConfig)
def move(src, dest)
Definition: eostools.py:511