UtilAlgos/interface/SingleElementCollectionRefSelector.h

0001 #ifndef RecoAlgos_SingleElementCollectionRefSelector_h
0002 #define RecoAlgos_SingleElementCollectionRefSelector_h
0003 /** \class SingleElementCollectionRefSelector
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: SingleElementCollectionRefSelector.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/ParameterSetDescription.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 #include "DataFormats/Common/interface/View.h"
0022
0023 namespace reco {
0024   namespace modules {
0025     template <typename S>
0026     struct SingleElementCollectionRefSelectorEventSetupInit;
0027   }
0028 }  // namespace reco
0029 namespace edm {
0030   class Event;
0031   class EventSetup;
0032 }  // namespace edm
0033
0034 template <typename InputType,
0035           typename Selector,
0036           typename OutputCollection = typename ::helper::SelectedOutputCollectionTrait<edm::View<InputType> >::type,
0037           typename StoreContainer = typename ::helper::StoreContainerTrait<OutputCollection>::type,
0038           typename RefAdder = typename ::helper::SelectionAdderTrait<edm::View<InputType>, StoreContainer>::type>
0039 struct SingleElementCollectionRefSelector {
0040   typedef edm::View<InputType> InputCollection;
0041   typedef InputCollection collection;
0042   typedef StoreContainer container;
0043   typedef Selector selector;
0044   typedef typename container::const_iterator const_iterator;
0045   SingleElementCollectionRefSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC)
0046       : select_(reco::modules::make<Selector>(cfg, iC)) {}
0047   const_iterator begin() const { return selected_.begin(); }
0048   const_iterator end() const { return selected_.end(); }
0049   void select(const edm::Handle<InputCollection>& c, const edm::Event&, const edm::EventSetup&) {
0050     selected_.clear();
0051     for (size_t idx = 0; idx < c->size(); ++idx) {
0052       if (select_(c->refAt(idx)))
0053         addRef_(selected_, c, idx);
0054     }
0055   }
0056
0057   static void fillPSetDescription(edm::ParameterSetDescription& desc) {
0058     // Use ParameterAdapter to fill the descriptions
0059     reco::modules::ParameterAdapter<Selector>::fillPSetDescription(desc);
0060   }
0061
0062 private:
0063   container selected_;
0064   selector select_;
0065   RefAdder addRef_;
0066   friend struct reco::modules::SingleElementCollectionRefSelectorEventSetupInit<SingleElementCollectionRefSelector>;
0067 };
0068
0069 #include "CommonTools/UtilAlgos/interface/EventSetupInitTrait.h"
0070
0071 namespace reco {
0072   namespace modules {
0073     template <typename S>
0074     struct SingleElementCollectionRefSelectorEventSetupInit {
0075       explicit SingleElementCollectionRefSelectorEventSetupInit(edm::ConsumesCollector iC) : esi_(iC) {}
0076       void init(S& s, const edm::Event& ev, const edm::EventSetup& es) { esi_.init(s.select_, ev, es); }
0077       typedef typename EventSetupInit<typename S::selector>::type ESI;
0078       ESI esi_;
0079     };
0080
0081     template <typename I, typename S, typename O, typename C, typename R>
0082     struct EventSetupInit<SingleElementCollectionRefSelector<I, S, O, C, R> > {
0083       typedef SingleElementCollectionRefSelectorEventSetupInit<SingleElementCollectionRefSelector<I, S, O, C, R> > type;
0084     };
0085   }  // namespace modules
0086 }  // namespace reco
0087
0088 #endif