CMS 3D CMS Logo
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
CandCombinerBase.h
Go to the documentation of this file.
1 #ifndef CommonTools_CandUtils_CandCombinerBase_h
2 #define CommonTools_CandUtils_CandCombinerBase_h
3 
8 #include "DataFormats/Candidate/interface/Candidate.h"
9 #include "DataFormats/Candidate/interface/OverlapChecker.h"
10 #include <vector>
11 #include <string>
12 
13 template <typename OutputCollection, typename CandPtr>
14 class CandCombinerBase {
15 public:
16  typedef std::vector<std::string> vstring;
18  explicit CandCombinerBase(const std::string = "");
20  CandCombinerBase(int, int, const std::string = "");
22  CandCombinerBase(int, int, int, const std::string = "");
24  CandCombinerBase(int, int, int, int, const std::string = "");
26  CandCombinerBase(bool checkCharge, bool checkOverlap, const std::vector<int> &, const std::string = "");
28  virtual ~CandCombinerBase();
30  std::unique_ptr<OutputCollection> combine(const std::vector<edm::Handle<reco::CandidateView> > &,
31  const vstring & = vstring()) const;
33  std::unique_ptr<OutputCollection> combine(const edm::Handle<reco::CandidateView> &,
34  const vstring & = vstring()) const;
36  std::unique_ptr<OutputCollection> combine(const edm::Handle<reco::CandidateView> &,
37  const edm::Handle<reco::CandidateView> &,
38  const vstring & = vstring()) const;
40  std::unique_ptr<OutputCollection> combine(const edm::Handle<reco::CandidateView> &,
41  const edm::Handle<reco::CandidateView> &,
42  const edm::Handle<reco::CandidateView> &,
43  const vstring & = vstring()) const;
45  std::unique_ptr<OutputCollection> combine(const edm::Handle<reco::CandidateView> &,
46  const edm::Handle<reco::CandidateView> &,
47  const edm::Handle<reco::CandidateView> &,
48  const edm::Handle<reco::CandidateView> &,
49  const vstring & = vstring()) const;
50 
51 private:
53  bool preselect(const reco::Candidate &, const reco::Candidate &) const;
55  void combine(typename OutputCollection::value_type &,
56  const CandPtr &,
57  const CandPtr &,
58  const std::string = "",
59  const std::string = "") const;
61  typedef std::vector<
62  std::pair<std::pair<CandPtr, size_t>, std::vector<edm::Handle<reco::CandidateView> >::const_iterator> >
63  CandStack;
64  typedef std::vector<int> ChargeStack;
66  void combine(size_t collectionIndex,
67  CandStack &,
68  ChargeStack &,
69  std::vector<edm::Handle<reco::CandidateView> >::const_iterator begin,
70  std::vector<edm::Handle<reco::CandidateView> >::const_iterator end,
71  OutputCollection *comps,
72  const vstring &name = vstring()) const;
74  virtual bool select(const reco::Candidate &) const = 0;
76  virtual bool selectPair(const reco::Candidate &c1, const reco::Candidate &c2) const = 0;
78  virtual void setup(typename OutputCollection::value_type &) const = 0;
80  virtual void addDaughter(typename OutputCollection::value_type &cmp,
81  const CandPtr &c,
82  const std::string = "") const = 0;
84  bool checkCharge_;
86  bool checkOverlap_;
88  std::vector<int> dauCharge_;
90  OverlapChecker overlap_;
92  std::string name_;
93 };
94 
95 template <typename OutputCollection, typename CandPtr>
96 CandCombinerBase<OutputCollection, CandPtr>::CandCombinerBase(const std::string name)
97  : checkCharge_(false), checkOverlap_(true), dauCharge_(), overlap_(), name_(name) {}
98 
99 template <typename OutputCollection, typename CandPtr>
100 CandCombinerBase<OutputCollection, CandPtr>::CandCombinerBase(int q1, int q2, const std::string name)
101  : checkCharge_(true), checkOverlap_(true), dauCharge_(2), overlap_(), name_(name) {
102  dauCharge_[0] = q1;
103  dauCharge_[1] = q2;
104 }
105 
106 template <typename OutputCollection, typename CandPtr>
107 CandCombinerBase<OutputCollection, CandPtr>::CandCombinerBase(int q1, int q2, int q3, const std::string name)
108  : checkCharge_(true), checkOverlap_(true), dauCharge_(3), overlap_(), name_(name) {
109  dauCharge_[0] = q1;
110  dauCharge_[1] = q2;
111  dauCharge_[2] = q3;
112 }
113 
114 template <typename OutputCollection, typename CandPtr>
115 CandCombinerBase<OutputCollection, CandPtr>::CandCombinerBase(int q1, int q2, int q3, int q4, const std::string name)
116  : checkCharge_(true), checkOverlap_(true), dauCharge_(4), overlap_(), name_(name) {
117  dauCharge_[0] = q1;
118  dauCharge_[1] = q2;
119  dauCharge_[2] = q3;
120  dauCharge_[3] = q4;
121 }
122 
123 template <typename OutputCollection, typename CandPtr>
124 CandCombinerBase<OutputCollection, CandPtr>::CandCombinerBase(bool checkCharge,
125  bool checkOverlap,
126  const std::vector<int> &dauCharge,
127  const std::string name)
128  : checkCharge_(checkCharge), checkOverlap_(checkOverlap), dauCharge_(dauCharge), overlap_(), name_(name) {}
129 
130 template <typename OutputCollection, typename CandPtr>
131 CandCombinerBase<OutputCollection, CandPtr>::~CandCombinerBase() {}
132 
133 template <typename OutputCollection, typename CandPtr>
134 bool CandCombinerBase<OutputCollection, CandPtr>::preselect(const reco::Candidate &c1,
135  const reco::Candidate &c2) const {
136  if (checkCharge_) {
137  int dq1 = dauCharge_[0], dq2 = dauCharge_[1], q1 = c1.charge(), q2 = c2.charge();
138  bool matchCharge = (q1 == dq1 && q2 == dq2) || (q1 == -dq1 && q2 == -dq2);
139  if (!matchCharge)
140  return false;
141  }
142  if (checkOverlap_ && overlap_(c1, c2))
143  return false;
144  return selectPair(c1, c2);
145 }
146 
147 template <typename OutputCollection, typename CandPtr>
148 void CandCombinerBase<OutputCollection, CandPtr>::combine(typename OutputCollection::value_type &cmp,
149  const CandPtr &c1,
150  const CandPtr &c2,
151  const std::string name1,
152  const std::string name2) const {
153  addDaughter(cmp, c1, name1);
154  addDaughter(cmp, c2, name2);
155  setup(cmp);
156 }
157 
158 template <typename OutputCollection, typename CandPtr>
159 std::unique_ptr<OutputCollection> CandCombinerBase<OutputCollection, CandPtr>::combine(
160  const std::vector<edm::Handle<reco::CandidateView> > &src, const vstring &names) const {
161  size_t srcSize = src.size();
162  if (checkCharge_ && dauCharge_.size() != srcSize)
163  throw edm::Exception(edm::errors::Configuration)
164  << "CandCombiner: trying to combine " << srcSize << " collections"
165  << " but configured to check against " << dauCharge_.size() << " charges.";
166  std::unique_ptr<OutputCollection> comps(new OutputCollection);
167  size_t namesSize = names.size();
168  if (srcSize == 2) {
169  std::string name1 = "", name2 = "";
170  if (namesSize > 0) {
171  if (namesSize != 2)
172  throw edm::Exception(edm::errors::Configuration)
173  << "CandCombiner: should specify exactly two "
174  << " names in configuration (" << namesSize << " specified).\n";
175  name1 = names[0];
176  name2 = names[1];
177  }
178  edm::Handle<reco::CandidateView> src1 = src[0], src2 = src[1];
179  if (src1.id() == src2.id()) {
180  const reco::CandidateView &cands = *src1;
181  const size_t n = cands.size();
182  for (size_t i1 = 0; i1 < n; ++i1) {
183  const reco::Candidate &c1 = cands[i1];
184  CandPtr cr1(src1, i1);
185  for (size_t i2 = i1 + 1; i2 < n; ++i2) {
186  const reco::Candidate &c2 = cands[i2];
187  if (preselect(c1, c2)) {
188  CandPtr cr2(src2, i2);
189  typename OutputCollection::value_type c;
190  combine(c, cr1, cr2, name1, name2);
191  if (select(c))
192  comps->push_back(c);
193  }
194  }
195  }
196  } else {
197  const reco::CandidateView &cands1 = *src1, &cands2 = *src2;
198  const size_t n1 = cands1.size(), n2 = cands2.size();
199  for (size_t i1 = 0; i1 < n1; ++i1) {
200  const reco::Candidate &c1 = cands1[i1];
201  CandPtr cr1(src1, i1);
202  for (size_t i2 = 0; i2 < n2; ++i2) {
203  const reco::Candidate &c2 = cands2[i2];
204  if (preselect(c1, c2)) {
205  CandPtr cr2(src2, i2);
206  typename OutputCollection::value_type c;
207  combine(c, cr1, cr2, name1, name2);
208  if (select(c))
209  comps->push_back(c);
210  }
211  }
212  }
213  }
214  } else {
215  CandStack stack;
216  ChargeStack qStack;
217  combine(0, stack, qStack, src.begin(), src.end(), comps.get(), names);
218  }
219 
220  return comps;
221 }
222 
223 template <typename OutputCollection, typename CandPtr>
224 std::unique_ptr<OutputCollection> CandCombinerBase<OutputCollection, CandPtr>::combine(
225  const edm::Handle<reco::CandidateView> &src, const vstring &names) const {
226  if (checkCharge_ && dauCharge_.size() != 2)
227  throw edm::Exception(edm::errors::Configuration)
228  << "CandCombiner: trying to combine 2 collections"
229  << " but configured to check against " << dauCharge_.size() << " charges.";
230 
231  std::unique_ptr<OutputCollection> comps(new OutputCollection);
232  size_t namesSize = names.size();
233  std::string name1, name2;
234  if (namesSize > 0) {
235  if (namesSize != 2)
236  throw edm::Exception(edm::errors::Configuration) << "CandCombiner: should specify exactly two "
237  << " names in configuration (" << namesSize << " specified).\n";
238  name1 = names[0];
239  name2 = names[1];
240  }
241  const reco::CandidateView &cands = *src;
242  const size_t n = cands.size();
243  for (size_t i1 = 0; i1 < n; ++i1) {
244  const reco::Candidate &c1 = cands[i1];
245  CandPtr cr1(src, i1);
246  for (size_t i2 = i1 + 1; i2 < n; ++i2) {
247  const reco::Candidate &c2 = cands[i2];
248  if (preselect(c1, c2)) {
249  CandPtr cr2(src, i2);
250  typename OutputCollection::value_type c;
251  combine(c, cr1, cr2, name1, name2);
252  if (select(c))
253  comps->push_back(c);
254  }
255  }
256  }
257 
258  return comps;
259 }
260 
261 template <typename OutputCollection, typename CandPtr>
262 std::unique_ptr<OutputCollection> CandCombinerBase<OutputCollection, CandPtr>::combine(
263  const edm::Handle<reco::CandidateView> &src1,
264  const edm::Handle<reco::CandidateView> &src2,
265  const vstring &names) const {
266  std::vector<edm::Handle<reco::CandidateView> > src;
267  src.push_back(src1);
268  src.push_back(src2);
269  return combine(src, names);
270 }
271 
272 template <typename OutputCollection, typename CandPtr>
273 std::unique_ptr<OutputCollection> CandCombinerBase<OutputCollection, CandPtr>::combine(
274  const edm::Handle<reco::CandidateView> &src1,
275  const edm::Handle<reco::CandidateView> &src2,
276  const edm::Handle<reco::CandidateView> &src3,
277  const vstring &names) const {
278  std::vector<edm::Handle<reco::CandidateView> > src;
279  src.push_back(src1);
280  src.push_back(src2);
281  src.push_back(src3);
282  return combine(src, names);
283 }
284 
285 template <typename OutputCollection, typename CandPtr>
286 std::unique_ptr<OutputCollection> CandCombinerBase<OutputCollection, CandPtr>::combine(
287  const edm::Handle<reco::CandidateView> &src1,
288  const edm::Handle<reco::CandidateView> &src2,
289  const edm::Handle<reco::CandidateView> &src3,
290  const edm::Handle<reco::CandidateView> &src4,
291  const vstring &names) const {
292  std::vector<edm::Handle<reco::CandidateView> > src;
293  src.push_back(src1);
294  src.push_back(src2);
295  src.push_back(src3);
296  src.push_back(src4);
297  return combine(src, names);
298 }
299 
300 template <typename OutputCollection, typename CandPtr>
301 void CandCombinerBase<OutputCollection, CandPtr>::combine(
302  size_t collectionIndex,
303  CandStack &stack,
304  ChargeStack &qStack,
305  std::vector<edm::Handle<reco::CandidateView> >::const_iterator collBegin,
306  std::vector<edm::Handle<reco::CandidateView> >::const_iterator collEnd,
307  OutputCollection *comps,
308  const vstring &names) const {
309  if (collBegin == collEnd) {
310  static const int undetermined = 0, sameDecay = 1, conjDecay = -1, wrongDecay = 2;
311  int decayType = undetermined;
312  if (checkCharge_) {
313  assert(qStack.size() == stack.size());
314  for (size_t i = 0; i < qStack.size(); ++i) {
315  int q = qStack[i], dq = dauCharge_[i];
316  if (decayType == undetermined) {
317  if (q != 0 && dq != 0) {
318  if (q == dq)
319  decayType = sameDecay;
320  else if (q == -dq)
321  decayType = conjDecay;
322  else
323  decayType = wrongDecay;
324  }
325  } else if ((decayType == sameDecay && q != dq) || (decayType == conjDecay && q != -dq)) {
326  decayType = wrongDecay;
327  }
328  if (decayType == wrongDecay)
329  break;
330  }
331  }
332  if (decayType != wrongDecay) {
333  typename OutputCollection::value_type c;
334  size_t nameIndex = 0;
335  for (typename CandStack::const_iterator i = stack.begin(); i != stack.end(); ++i, ++nameIndex) {
336  if (!names.empty())
337  addDaughter(c, i->first.first, names[nameIndex]);
338  else
339  addDaughter(c, i->first.first);
340  }
341  setup(c);
342  if (select(c))
343  comps->push_back(c);
344  }
345  } else {
346  const edm::Handle<reco::CandidateView> &srcRef = *collBegin;
347  const reco::CandidateView &src = *srcRef;
348  size_t candBegin = 0, candEnd = src.size();
349  for (typename CandStack::const_iterator i = stack.begin(); i != stack.end(); ++i)
350  if (srcRef.id() == i->second->id())
351  candBegin = i->first.second + 1;
352  for (size_t candIndex = candBegin; candIndex != candEnd; ++candIndex) {
353  CandPtr candRef(srcRef, candIndex);
354  bool noOverlap = true;
355  const reco::Candidate &cand = *candRef;
356  for (typename CandStack::const_iterator i = stack.begin(); i != stack.end(); ++i)
357  if (checkOverlap_ && overlap_(cand, *(i->first.first))) {
358  noOverlap = false;
359  break;
360  }
361  if (noOverlap) {
362  stack.push_back(std::make_pair(std::make_pair(candRef, candIndex), collBegin));
363  if (checkCharge_)
364  qStack.push_back(cand.charge());
365  combine(collectionIndex + 1, stack, qStack, collBegin + 1, collEnd, comps, names);
366  stack.pop_back();
367  qStack.pop_back();
368  }
369  }
370  }
371 }
372 
373 #endif
readCalibrations2004_v2_cff.src1
tuple src1
Definition: readCalibrations2004_v2_cff.py:3
c
const edm::EventSetup & c
Definition: SiStripLAProfileBooker.cc:66
mps_fire.i
i
Definition: mps_fire.py:428
CandCombinerBase::select
virtual bool select(const reco::Candidate &) const =0
select a candidate
CandCombinerBase::selectPair
virtual bool selectPair(const reco::Candidate &c1, const reco::Candidate &c2) const =0
select a candidate pair
g4TestGeometry_cfi.checkOverlap
def checkOverlap
Definition: g4TestGeometry_cfi.py:3
edm::HandleBase::id
ProductID id() const
Definition: HandleBase.cc:29
CandCombinerBase::checkOverlap_
bool checkOverlap_
flag to specify the checking of overlaps
Definition: CandCombinerBase.h:86
HLT_FULL_cff.cands
tuple cands
Definition: HLT_FULL_cff.py:15391
CandCombinerBase::checkCharge_
bool checkCharge_
flag to specify the checking of electric charge
Definition: CandCombinerBase.h:84
CandCombinerBase::preselect
bool preselect(const reco::Candidate &, const reco::Candidate &) const
verify that the two candidate don&#39;t overlap and check charge
Definition: CandCombinerBase.h:134
CandCombinerBase::overlap_
OverlapChecker overlap_
utility to check candidate daughters overlap
Definition: CandCombinerBase.h:90
edm::View::size
size_type size() const
CandCombinerBase::CandCombinerBase
CandCombinerBase(const std::string="")
default construct
Definition: CandCombinerBase.h:96
readConfiguration2006_v0_cff.src2
tuple src2
Definition: readConfiguration2006_v0_cff.py:20
cms::cuda::assert
assert(be >=bs)
names
const std::string names[nVars_]
Definition: PhotonIDValueMapProducer.cc:124
AlCaHLTBitMon_QueryRunRegistry.string
string string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:256
cscdqm::h::namesSize
static const unsigned int namesSize
Definition: CSCDQM_HistoDef.h:831
funct::combine
static type combine(const A &_1, const B &_2)
Definition: Factorize.h:176
svgfig.stack
stack
Definition: svgfig.py:559
CandCombinerBase::~CandCombinerBase
virtual ~CandCombinerBase()
destructor
Definition: CandCombinerBase.h:131
PixelPluginsPhase0_cfi.select
tuple select
Definition: PixelPluginsPhase0_cfi.py:16
CandCombinerBase::addDaughter
virtual void addDaughter(typename OutputCollection::value_type &cmp, const CandPtr &c, const std::string="") const =0
add candidate daughter
alcazmumu_cfi.src
tuple src
Definition: alcazmumu_cfi.py:30
reco::Candidate::charge
virtual int charge() const =0
electric charge
for
for(Iditer=Id.begin();Iditer!=Id.end();Iditer++)
Definition: SiStripLAProfileBooker.cc:79
HLT_FULL_cff.checkCharge
tuple checkCharge
Definition: HLT_FULL_cff.py:26644
CandCombinerBase::combine
std::unique_ptr< OutputCollection > combine(const std::vector< edm::Handle< reco::CandidateView > > &, const vstring &=vstring()) const
return all selected candidate pairs
Definition: CandCombinerBase.h:159
CandCombinerBase::dauCharge_
std::vector< int > dauCharge_
electric charges of the daughters
Definition: CandCombinerBase.h:88
CandCombinerBase::CandStack
std::vector< std::pair< std::pair< CandPtr, size_t >, std::vector< edm::Handle< reco::CandidateView > >::const_iterator > > CandStack
temporary candidate stack
Definition: CandCombinerBase.h:63
CandCombinerBase::vstring
std::vector< std::string > vstring
Definition: CandCombinerBase.h:16
dataset.end
string end
Definition: dataset.py:937
edm::ProductID::id
ProductIndex id() const
Definition: ProductID.h:35
CandCombinerBase::name_
std::string name_
composite name (if applies)
Definition: CandCombinerBase.h:92
CandCombinerBase::setup
virtual void setup(typename OutputCollection::value_type &) const =0
set kinematics to reconstructed composite
CandCombinerBase::ChargeStack
std::vector< int > ChargeStack
Definition: CandCombinerBase.h:64
Generated for CMSSW Reference Manual by doxygen 1.8.5