CandDecaySelector.h

Go to the documentation of this file.

#ifndef CandAlgos_CandDecaySelector_h
#define CandAlgos_CandDecaySelector_h
/* \class helper::CandDecayStoreManager
 *
 * \author: Luca Lista, INFN
 *
 */
#include "DataFormats/Candidate/interface/CompositeRefCandidate.h"
#include "CommonTools/UtilAlgos/interface/ObjectSelector.h"

namespace helper {
  class CandDecayStoreManager {
  public:
    typedef reco::CandidateCollection collection;
    CandDecayStoreManager(const edm::Handle<reco::CandidateCollection>&) : selCands_(new reco::CandidateCollection) {}
    template <typename I>
    void cloneAndStore(const I& begin, const I& end, edm::Event& evt) {
      using namespace reco;
      CandidateRefProd cands = evt.getRefBeforePut<CandidateCollection>();
      for (I i = begin; i != end; ++i)
        add(cands, **i);
    }
    edm::OrphanHandle<reco::CandidateCollection> put(edm::Event& evt) { return evt.put(std::move(selCands_)); }
    size_t size() const { return selCands_->size(); }

  private:
    reco::CandidateRef add(reco::CandidateRefProd cands, const reco::Candidate& c) {
      using namespace reco;
      using namespace std;
      auto cmp = std::make_unique<CompositeRefCandidate>(c);
      CompositeRefCandidate* p = cmp.get();
      CandidateRef ref(cands, selCands_->size());
      selCands_->push_back(std::move(cmp));
      size_t n = c.numberOfDaughters();
      for (size_t i = 0; i < n; ++i)
        p->addDaughter(add(cands, *c.daughter(i)));
      return ref;
    }
    std::unique_ptr<reco::CandidateCollection> selCands_;
  };

  template <typename EdmFilter>
  struct StoreManagerTrait<reco::CandidateCollection, EdmFilter> {
    typedef CandDecayStoreManager type;
    typedef ObjectSelectorBase<reco::CandidateCollection, EdmFilter> base;
  };

}  // namespace helper

#endif