File indexing completed on 2023-10-25 09:36:14
0001 #ifndef RecoAlgos_SingleElementCollectionSelector_h
0002 #define RecoAlgos_SingleElementCollectionSelector_h
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015 #include "FWCore/Framework/interface/ConsumesCollector.h"
0016 #include "CommonTools/UtilAlgos/interface/SelectionAdderTrait.h"
0017 #include "CommonTools/UtilAlgos/interface/StoreContainerTrait.h"
0018 #include "CommonTools/UtilAlgos/interface/ParameterAdapter.h"
0019 #include "CommonTools/UtilAlgos/interface/SelectedOutputCollectionTrait.h"
0020
0021 namespace reco {
0022 namespace modules {
0023 template <typename S>
0024 struct SingleElementCollectionSelectorEventSetupInit;
0025 }
0026 }
0027 namespace edm {
0028 class Event;
0029 class EventSetup;
0030 }
0031
0032 template <typename InputCollection,
0033 typename Selector,
0034 typename OutputCollection = typename ::helper::SelectedOutputCollectionTrait<InputCollection>::type,
0035 typename StoreContainer = typename ::helper::StoreContainerTrait<OutputCollection>::type,
0036 typename RefAdder = typename ::helper::SelectionAdderTrait<InputCollection, StoreContainer>::type>
0037 struct SingleElementCollectionSelector {
0038 typedef InputCollection collection;
0039 typedef StoreContainer container;
0040 typedef Selector selector;
0041 typedef typename container::const_iterator const_iterator;
0042 SingleElementCollectionSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
0043 : select_(reco::modules::make<Selector>(cfg, iC)) {}
0044 const_iterator begin() const { return selected_.begin(); }
0045 const_iterator end() const { return selected_.end(); }
0046 void select(const edm::Handle<InputCollection>& c, const edm::Event&, const edm::EventSetup&) {
0047 selected_.clear();
0048 for (size_t idx = 0; idx < c->size(); ++idx) {
0049 if (select_((*c)[idx]))
0050 addRef_(selected_, c, idx);
0051 }
0052 }
0053
0054 private:
0055 container selected_;
0056 selector select_;
0057 RefAdder addRef_;
0058 friend struct reco::modules::SingleElementCollectionSelectorEventSetupInit<SingleElementCollectionSelector>;
0059 };
0060
0061 #include "CommonTools/UtilAlgos/interface/EventSetupInitTrait.h"
0062
0063 namespace reco {
0064 namespace modules {
0065 template <typename S>
0066 struct SingleElementCollectionSelectorEventSetupInit {
0067 explicit SingleElementCollectionSelectorEventSetupInit(edm::ConsumesCollector iC) : esi_(iC) {}
0068 typedef typename EventSetupInit<typename S::selector>::type ESI;
0069 void init(S& s, const edm::Event& ev, const edm::EventSetup& es) { esi_.init(s.select_, ev, es); }
0070 ESI esi_;
0071 };
0072
0073 template <typename I, typename S, typename O, typename C, typename R>
0074 struct EventSetupInit<SingleElementCollectionSelector<I, S, O, C, R> > {
0075 typedef SingleElementCollectionSelectorEventSetupInit<SingleElementCollectionSelector<I, S, O, C, R> > type;
0076 };
0077 }
0078 }
0079
0080 #endif