OniaAddV0TracksProducer.cc

Go to the documentation of this file.

#include "HeavyFlavorAnalysis/Onia2MuMu/interface/OniaAddV0TracksProducer.h"
#include "DataFormats/PatCandidates/interface/CompositeCandidate.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include <vector>

OniaAddV0TracksProducer::OniaAddV0TracksProducer(const edm::ParameterSet& ps) {
  LambdaCollectionToken_ =
      consumes<reco::VertexCompositeCandidateCollection>(ps.getParameter<edm::InputTag>("LambdaTag"));
  KShortCollectionToken_ =
      consumes<reco::VertexCompositeCandidateCollection>(ps.getParameter<edm::InputTag>("KShortTag"));

  produces<pat::CompositeCandidateCollection>("Kshort");
  produces<pat::CompositeCandidateCollection>("Lambda");

  events_v0 = 0;
  total_v0 = 0;
  total_lambda = 0;
  total_kshort = 0;
  std::cout << "running OniaAddV0TracksProducer..." << std::endl;
}

void OniaAddV0TracksProducer::produce(edm::Event& event, const edm::EventSetup& esetup) {
  // Create unique_ptr for each collection to be stored in the Event
  std::unique_ptr<pat::CompositeCandidateCollection> Enhanced_kShortCandidates(new pat::CompositeCandidateCollection);
  std::unique_ptr<pat::CompositeCandidateCollection> Enhanced_lambdaCandidates(new pat::CompositeCandidateCollection);

  edm::Handle<reco::VertexCompositeCandidateCollection> lcandidates;
  event.getByToken(LambdaCollectionToken_, lcandidates);

  edm::Handle<reco::VertexCompositeCandidateCollection> kcandidates;
  event.getByToken(KShortCollectionToken_, kcandidates);

  int exits_l = 0;
  int exits_k = 0;
  for (reco::VertexCompositeCandidateCollection::const_iterator ik = kcandidates->begin(); ik != kcandidates->end();
       ++ik) {
    pat::CompositeCandidate* kc = new pat::CompositeCandidate();
    edm::RefToBase<reco::Track> ktrk0((*(dynamic_cast<const reco::RecoChargedCandidate*>(ik->daughter(0)))).track());
    edm::RefToBase<reco::Track> ktrk1((*(dynamic_cast<const reco::RecoChargedCandidate*>(ik->daughter(1)))).track());
    kc->addUserData<reco::Track>("track0", *ktrk0);
    kc->addUserData<reco::Track>("track1", *ktrk1);
    Enhanced_kShortCandidates->push_back(*kc);
    exits_k++;
  }

  for (reco::VertexCompositeCandidateCollection::const_iterator il = lcandidates->begin(); il != lcandidates->end();
       ++il) {
    pat::CompositeCandidate* lc = new pat::CompositeCandidate();
    edm::RefToBase<reco::Track> ltrk0((*(dynamic_cast<const reco::RecoChargedCandidate*>(il->daughter(0)))).track());
    edm::RefToBase<reco::Track> ltrk1((*(dynamic_cast<const reco::RecoChargedCandidate*>(il->daughter(1)))).track());
    lc->addUserData<reco::Track>("track0", *ltrk0);
    lc->addUserData<reco::Track>("track1", *ltrk1);
    Enhanced_lambdaCandidates->push_back(*lc);
    exits_l++;
  }

  // Write the collections to the Event

  total_v0 += exits_k;
  total_v0 += exits_l;
  total_kshort += exits_k;
  total_lambda += exits_l;
  if (exits_k || exits_l)
    events_v0++;

  event.put(std::move(Enhanced_kShortCandidates), "Kshort");
  event.put(std::move(Enhanced_lambdaCandidates), "Lambda");
}

void OniaAddV0TracksProducer::endJob() {
  std::cout << "############################" << std::endl;
  std::cout << "OniaAddV0Tracks producer report " << std::endl;
  std::cout << "############################" << std::endl;
  std::cout << "Total events with v0 :      " << events_v0 << std::endl;
  std::cout << "Total v0 :                  " << total_v0 << std::endl;
  std::cout << "Total number of lambda :    " << total_lambda << std::endl;
  std::cout << "Total number of kshort :    " << total_kshort << std::endl;
  std::cout << "############################" << std::endl;
}
//define this as a plug-in
DEFINE_FWK_MODULE(OniaAddV0TracksProducer);