UtilAlgos/interface/SingleElementCollectionSelector.h

0001 #ifndef RecoAlgos_SingleElementCollectionSelector_h
0002 #define RecoAlgos_SingleElementCollectionSelector_h
0003 /** \class SingleElementCollectionSelector
0004  *
0005  * selects a subset of a collection based
0006  * on single element selection done via functor
0007  *
0008  * \author Luca Lista, INFN
0009  *
0010  * \version $Revision: 1.1 $
0011  *
0012  * $Id: SingleElementCollectionSelector.h,v 1.1 2009/03/03 13:07:28 llista Exp $
0013  *
0014  */
0015 #include "FWCore/Framework/interface/ConsumesCollector.h"
0016 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0017 #include "CommonTools/UtilAlgos/interface/SelectionAdderTrait.h"
0018 #include "CommonTools/UtilAlgos/interface/StoreContainerTrait.h"
0019 #include "CommonTools/UtilAlgos/interface/ParameterAdapter.h"
0020 #include "CommonTools/UtilAlgos/interface/SelectedOutputCollectionTrait.h"
0021
0022 namespace reco {
0023   namespace modules {
0024     template <typename S>
0025     struct SingleElementCollectionSelectorEventSetupInit;
0026   }
0027 }  // namespace reco
0028 namespace edm {
0029   class Event;
0030   class EventSetup;
0031 }  // namespace edm
0032
0033 template <typename InputCollection,
0034           typename Selector,
0035           typename OutputCollection = typename ::helper::SelectedOutputCollectionTrait<InputCollection>::type,
0036           typename StoreContainer = typename ::helper::StoreContainerTrait<OutputCollection>::type,
0037           typename RefAdder = typename ::helper::SelectionAdderTrait<InputCollection, StoreContainer>::type>
0038 struct SingleElementCollectionSelector {
0039   typedef InputCollection collection;
0040   typedef StoreContainer container;
0041   typedef Selector selector;
0042   typedef typename container::const_iterator const_iterator;
0043   SingleElementCollectionSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
0044       : select_(reco::modules::make<Selector>(cfg, iC)) {}
0045   const_iterator begin() const { return selected_.begin(); }
0046   const_iterator end() const { return selected_.end(); }
0047   void select(const edm::Handle<InputCollection>& c, const edm::Event&, const edm::EventSetup&) {
0048     selected_.clear();
0049     for (size_t idx = 0; idx < c->size(); ++idx) {
0050       if (select_((*c)[idx]))
0051         addRef_(selected_, c, idx);
0052     }
0053   }
0054
0055   static void fillPSetDescription(edm::ParameterSetDescription& desc) {
0056     reco::modules::ParameterAdapter<Selector>::fillPSetDescription(desc);
0057   };
0058
0059 private:
0060   container selected_;
0061   selector select_;
0062   RefAdder addRef_;
0063   friend struct reco::modules::SingleElementCollectionSelectorEventSetupInit<SingleElementCollectionSelector>;
0064 };
0065
0066 #include "CommonTools/UtilAlgos/interface/EventSetupInitTrait.h"
0067
0068 namespace reco {
0069   namespace modules {
0070     template <typename S>
0071     struct SingleElementCollectionSelectorEventSetupInit {
0072       explicit SingleElementCollectionSelectorEventSetupInit(edm::ConsumesCollector iC) : esi_(iC) {}
0073       typedef typename EventSetupInit<typename S::selector>::type ESI;
0074       void init(S& s, const edm::Event& ev, const edm::EventSetup& es) { esi_.init(s.select_, ev, es); }
0075       ESI esi_;
0076     };
0077
0078     template <typename I, typename S, typename O, typename C, typename R>
0079     struct EventSetupInit<SingleElementCollectionSelector<I, S, O, C, R> > {
0080       typedef SingleElementCollectionSelectorEventSetupInit<SingleElementCollectionSelector<I, S, O, C, R> > type;
0081     };
0082   }  // namespace modules
0083 }  // namespace reco
0084
0085 #endif