PFNuclearProducer.cc

Go to the documentation of this file.

#include "DataFormats/ParticleFlowReco/interface/PFNuclearInteraction.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "RecoParticleFlow/PFTracking/interface/PFTrackTransformer.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
 
class PFNuclearProducer : public edm::stream::EDProducer<> {
public:
  explicit PFNuclearProducer(const edm::ParameterSet&);
 
  ~PFNuclearProducer() override;
 
private:
  void beginRun(const edm::Run&, const edm::EventSetup&) override;
  void endRun(const edm::Run&, const edm::EventSetup&) override;
 
  void produce(edm::Event&, const edm::EventSetup&) override;
 
  PFTrackTransformer* pfTransformer_;
  double likelihoodCut_;
  std::vector<edm::EDGetTokenT<reco::NuclearInteractionCollection> > nuclearContainers_;
 
  const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magneticFieldToken_;
};
 
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(PFNuclearProducer);
 
using namespace std;
using namespace edm;
PFNuclearProducer::PFNuclearProducer(const ParameterSet& iConfig)
    : pfTransformer_(nullptr), magneticFieldToken_(esConsumes<edm::Transition::BeginRun>()) {
  produces<reco::PFRecTrackCollection>();
  produces<reco::PFNuclearInteractionCollection>();
 
  std::vector<edm::InputTag> tags = iConfig.getParameter<vector<InputTag> >("nuclearColList");
 
  for (unsigned int i = 0; i < tags.size(); ++i)
    nuclearContainers_.push_back(consumes<reco::NuclearInteractionCollection>(tags[i]));
 
  likelihoodCut_ = iConfig.getParameter<double>("likelihoodCut");
}
 
PFNuclearProducer::~PFNuclearProducer() { delete pfTransformer_; }
 
void PFNuclearProducer::produce(Event& iEvent, const EventSetup& iSetup) {
  typedef reco::NuclearInteraction::trackRef_iterator trackRef_iterator;
 
  //create the empty collections
  auto pfNuclearColl = std::make_unique<reco::PFNuclearInteractionCollection>();
  auto pfNuclearRecTrackColl = std::make_unique<reco::PFRecTrackCollection>();
 
  reco::PFRecTrackRefProd pfTrackRefProd = iEvent.getRefBeforePut<reco::PFRecTrackCollection>();
  int hid = 0;
 
  // loop on the nuclear interaction collections
  for (unsigned int istr = 0; istr < nuclearContainers_.size(); istr++) {
    Handle<reco::NuclearInteractionCollection> nuclCollH;
    iEvent.getByToken(nuclearContainers_[istr], nuclCollH);
    const reco::NuclearInteractionCollection& nuclColl = *(nuclCollH.product());
 
    // loop on all NuclearInteraction
    for (unsigned int icoll = 0; icoll < nuclColl.size(); icoll++) {
      if (nuclColl[icoll].likelihood() < likelihoodCut_)
        continue;
 
      reco::PFRecTrackRefVector pfRecTkcoll;
 
      // convert the secondary tracks
      for (trackRef_iterator it = nuclColl[icoll].secondaryTracks_begin(); it != nuclColl[icoll].secondaryTracks_end();
           it++) {
        reco::PFRecTrack pftrack(
            (*it)->charge(), reco::PFRecTrack::KF, it->key(), (reco::TrackRef)((*it).castTo<reco::TrackRef>()));
        Trajectory FakeTraj;
        bool valid = pfTransformer_->addPoints(pftrack, **it, FakeTraj);
        if (valid) {
          pfRecTkcoll.push_back(reco::PFRecTrackRef(pfTrackRefProd, hid++));
          pfNuclearRecTrackColl->push_back(pftrack);
        }
      }
      reco::NuclearInteractionRef niRef(nuclCollH, icoll);
      pfNuclearColl->push_back(reco::PFNuclearInteraction(niRef, pfRecTkcoll));
    }
  }
  iEvent.put(std::move(pfNuclearRecTrackColl));
  iEvent.put(std::move(pfNuclearColl));
}
 
// ------------ method called once each job just before starting event loop  ------------
void PFNuclearProducer::beginRun(const edm::Run& run, const EventSetup& iSetup) {
  auto const& magneticField = &iSetup.getData(magneticFieldToken_);
  pfTransformer_ = new PFTrackTransformer(math::XYZVector(magneticField->inTesla(GlobalPoint(0, 0, 0))));
  pfTransformer_->OnlyProp();
}
 
// ------------ method called once each job just after ending the event loop  ------------
void PFNuclearProducer::endRun(const edm::Run& run, const EventSetup& iSetup) {
  delete pfTransformer_;
  pfTransformer_ = nullptr;
}