PFTauViewRefMerger.cc

Go to the documentation of this file.

/* \class PFTauViewRefMerger
 *
 * Produces a RefVector of PFTaus from a different views of PFTau.
 *
 * Note that the collections must all come from the same original collection!
 *
 * Author: Evan K. Friis
 *
 */

#include "DataFormats/TauReco/interface/PFTauFwd.h"
#include "RecoTauTag/RecoTau/interface/RecoTauCommonUtilities.h"

#include <FWCore/ParameterSet/interface/ConfigurationDescriptions.h>
#include <FWCore/ParameterSet/interface/ParameterSetDescription.h>

#include "FWCore/Framework/interface/MakerMacros.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"

class PFTauViewRefMerger : public edm::EDProducer {
  public:
    explicit PFTauViewRefMerger(const edm::ParameterSet& cfg) :
        src_(cfg.getParameter<std::vector<edm::InputTag> >("src")) {
          produces<reco::PFTauRefVector>();
        }
    static void fillDescriptions(edm::ConfigurationDescriptions & descriptions);
  private:
    void produce(edm::Event & evt, const edm::EventSetup &) override {
      auto out = std::make_unique<reco::PFTauRefVector>();
      for(auto const& inputSrc : src_) {
        edm::Handle<reco::CandidateView> src;
        evt.getByLabel(inputSrc, src);
        reco::PFTauRefVector inputRefs =
            reco::tau::castView<reco::PFTauRefVector>(src);
        // Merge all the collections
        for(auto const& tau : inputRefs) {
          out->push_back(tau);
        }
      }
      evt.put(std::move(out));
    }
    std::vector<edm::InputTag> src_;
};

void
PFTauViewRefMerger::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  // pfTauViewRefMerger
  edm::ParameterSetDescription desc;

  desc.add<std::vector<edm::InputTag>>("src");

  descriptions.add("pfTauViewRefMerger", desc);
}


DEFINE_FWK_MODULE(PFTauViewRefMerger);