|
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
|
#ifndef CommonTools_UtilAlgos_ObjectCountEventSelector_h
#define CommonTools_UtilAlgos_ObjectCountEventSelector_h
/** \class ObjectCountEventSelector
*
* Selects an event if a collection has at least N entries
*
* \author Luca Lista, INFN
*
* \version $Revision: 1.2 $
*
* $Id: ObjectCountEventSelector.h,v 1.2 2010/02/20 20:55:24 wmtan Exp $
*
*/
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "CommonTools/UtilAlgos/interface/ParameterAdapter.h"
#include "CommonTools/UtilAlgos/interface/CollectionFilterTrait.h"
#include "CommonTools/UtilAlgos/interface/EventSelectorBase.h"
template <typename C,
typename S = AnySelector,
typename N = MinNumberSelector,
typename CS = typename helper::CollectionFilterTrait<C, S, N>::type>
class ObjectCountEventSelector : public EventSelectorBase {
public:
/// constructor
explicit ObjectCountEventSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
: srcToken_(iC.consumes<C>(cfg.template getParameter<edm::InputTag>("src"))),
select_(reco::modules::make<S>(cfg, iC)),
sizeSelect_(reco::modules::make<N>(cfg, iC)) {}
static void fillPSetDescription(edm::ParameterSetDescription& desc) {
desc.add<edm::InputTag>("src", edm::InputTag());
reco::modules::fillPSetDescription<S>(desc);
reco::modules::fillPSetDescription<N>(desc);
}
bool operator()(edm::Event& evt, const edm::EventSetup&) const override {
edm::Handle<C> source;
evt.getByToken(srcToken_, source);
return CS::filter(*source, select_, sizeSelect_);
}
private:
/// source collection label
edm::EDGetTokenT<C> srcToken_;
/// object filter
S select_;
/// minimum number of entries in a collection
N sizeSelect_;
};
#endif
|