| 1 |
#ifndef ForwardAnalysis_Utilities_OneToManySelector_h
|
| 2 |
#define ForwardAnalysis_Utilities_OneToManySelector_h
|
| 3 |
|
| 4 |
#include "FWCore/Framework/interface/Event.h"
|
| 5 |
#include "FWCore/ParameterSet/interface/ParameterSet.h"
|
| 6 |
#include "FWCore/Utilities/interface/InputTag.h"
|
| 7 |
#include "CommonTools/UtilAlgos/interface/AnyPairSelector.h"
|
| 8 |
#include "CommonTools/UtilAlgos/interface/AnySelector.h"
|
| 9 |
|
| 10 |
namespace forwardAnalysis {
|
| 11 |
|
| 12 |
template <class Object, class Coll,
|
| 13 |
class PairSelector=AnyPairSelector, class S1=AnySelector, class S2=AnySelector>
|
| 14 |
class OneToManySelector {
|
| 15 |
public:
|
| 16 |
OneToManySelector(edm::ParameterSet const& pset):
|
| 17 |
src_( pset.template getParameter<edm::InputTag>("src") ),
|
| 18 |
selector_( reco::modules::make<PairSelector>(pset) ),
|
| 19 |
s1_( reco::modules::make<S1>(pset) ),
|
| 20 |
s2_( reco::modules::make<S2>(pset) ) {}
|
| 21 |
|
| 22 |
~OneToManySelector() {}
|
| 23 |
|
| 24 |
bool operator() (Object const& obj, edm::Event const& event){
|
| 25 |
edm::Handle<Coll> source;
|
| 26 |
event.getByLabel(src_,source);
|
| 27 |
typename Coll::const_iterator cand = source->begin(), source_end = source->end();
|
| 28 |
bool result = true;
|
| 29 |
if( !s1_(obj) ) { result=false; return result; } // Use only objects passing selector S1
|
| 30 |
for(; cand != source_end; ++cand){
|
| 31 |
if( !s2_(*cand) ) continue; // Use only objects passing selector S2
|
| 32 |
if( !selector_(obj,*cand) ) { result = false; break; }
|
| 33 |
}
|
| 34 |
return result;
|
| 35 |
}
|
| 36 |
|
| 37 |
private:
|
| 38 |
edm::InputTag src_;
|
| 39 |
PairSelector selector_;
|
| 40 |
S1 s1_;
|
| 41 |
S2 s2_;
|
| 42 |
};
|
| 43 |
|
| 44 |
} // namespace
|
| 45 |
|
| 46 |
#endif
|
