RecoParticleFlow/PFTracking/plugins/PFNuclearProducer.cc - master - githubmirror/cmssw

RecoParticleFlow/PFTracking/plugins/PFNuclearProducer.cc
112 строк5 KB
Christopher Jones
Added edm::stream::Watch* to RecoParticleFlow modules
25 мар 2026, 23:40
25 мар 2026, 23:40ad2c509
#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"

#include <memory>

class PFNuclearProducer : public edm::stream::EDProducer<edm::stream::WatchRuns> {
public:
  ///Constructor
  explicit PFNuclearProducer(const edm::ParameterSet&);

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:
  void beginRun(const edm::Run&, const edm::EventSetup&) override;
  void endRun(const edm::Run&, const edm::EventSetup&) override;

  ///Produce the PFRecTrack collection
  void produce(edm::Event&, const edm::EventSetup&) override;

  ///PFTrackTransformer
  std::unique_ptr<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);

void PFNuclearProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  // cut on the likelihood of the nuclear interaction
  desc.add<double>("likelihoodCut", 0.1);
  desc.add<std::vector<edm::InputTag>>("nuclearColList", {edm::InputTag("firstnuclearInteractionMaker")});
  descriptions.add("pfNuclear", desc);
}

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");
}

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_ = std::make_unique<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) { pfTransformer_.reset(); }