PATCompositeCandidateProducer.cc

Go to the documentation of this file.

//
//

#include "PhysicsTools/PatAlgos/plugins/PATCompositeCandidateProducer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/Common/interface/View.h"
#include "CommonTools/CandUtils/interface/AddFourMomenta.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "CommonTools/Utils/interface/StringObjectFunction.h"
#include "FWCore/Utilities/interface/Exception.h"
#include <memory>

#include <iostream>

using namespace pat;
using namespace std;
using namespace edm;

PATCompositeCandidateProducer::PATCompositeCandidateProducer(const ParameterSet& iConfig)
    : srcToken_(consumes<edm::View<reco::CompositeCandidate> >(iConfig.getParameter<InputTag>("src"))),
      useUserData_(iConfig.exists("userData")),
      userDataHelper_(iConfig.getParameter<edm::ParameterSet>("userData"), consumesCollector()),
      addEfficiencies_(iConfig.getParameter<bool>("addEfficiencies")),
      addResolutions_(iConfig.getParameter<bool>("addResolutions")) {
  // Efficiency configurables
  if (addEfficiencies_) {
    efficiencyLoader_ =
        pat::helper::EfficiencyLoader(iConfig.getParameter<edm::ParameterSet>("efficiencies"), consumesCollector());
  }

  // Resolution configurables
  if (addResolutions_) {
    resolutionLoader_ = pat::helper::KinResolutionsLoader(iConfig.getParameter<edm::ParameterSet>("resolutions"));
  }

  // produces vector of particles
  produces<vector<pat::CompositeCandidate> >();
}

PATCompositeCandidateProducer::~PATCompositeCandidateProducer() {}

void PATCompositeCandidateProducer::produce(Event& iEvent, const EventSetup& iSetup) {
  // Get the vector of CompositeCandidate's from the event
  Handle<View<reco::CompositeCandidate> > cands;
  iEvent.getByToken(srcToken_, cands);

  if (efficiencyLoader_.enabled())
    efficiencyLoader_.newEvent(iEvent);
  if (resolutionLoader_.enabled())
    resolutionLoader_.newEvent(iEvent, iSetup);

  auto myCompositeCandidates = std::make_unique<vector<pat::CompositeCandidate> >();

  if (cands.isValid()) {
    View<reco::CompositeCandidate>::const_iterator ibegin = cands->begin(), iend = cands->end(), i = ibegin;
    for (; i != iend; ++i) {
      pat::CompositeCandidate cand(*i);

      if (useUserData_) {
        userDataHelper_.add(cand, iEvent, iSetup);
      }

      if (efficiencyLoader_.enabled())
        efficiencyLoader_.setEfficiencies(cand, cands->refAt(i - cands->begin()));
      if (resolutionLoader_.enabled())
        resolutionLoader_.setResolutions(cand);

      myCompositeCandidates->push_back(std::move(cand));
    }

  }  // end if the two handles are valid

  iEvent.put(std::move(myCompositeCandidates));
}

#include "FWCore/Framework/interface/MakerMacros.h"

DEFINE_FWK_MODULE(PATCompositeCandidateProducer);