PFV0Producer.cc

Go to the documentation of this file.

#include <memory>
#include "RecoParticleFlow/PFTracking/plugins/PFV0Producer.h"
#include "DataFormats/ParticleFlowReco/interface/PFV0.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrackFwd.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "RecoParticleFlow/PFTracking/interface/PFTrackTransformer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

using namespace std;
using namespace edm;
using namespace reco;
PFV0Producer::PFV0Producer(const ParameterSet& iConfig):
  pfTransformer_(0)
{

  produces<reco::PFV0Collection>();
  produces<reco::PFRecTrackCollection>();

  std::vector<edm::InputTag> tags =     iConfig.getParameter< vector < InputTag > >("V0List");

  for (unsigned int i=0;i<tags.size();++i)
    V0list_.push_back(consumes<reco::VertexCompositeCandidateCollection>(tags[i])); 
}

PFV0Producer::~PFV0Producer()
{
  delete pfTransformer_;
}

void
PFV0Producer::produce(Event& iEvent, const EventSetup& iSetup)
{
  LogDebug("PFV0Producer")<<"START event: "<<iEvent.id().event()
              <<" in run "<<iEvent.id().run();
  //create the empty collections 
  auto pfV0Coll = std::make_unique<PFV0Collection>();

  auto pfV0RecTrackColl = std::make_unique<reco::PFRecTrackCollection>();


  reco::PFRecTrackRefProd pfTrackRefProd = iEvent.getRefBeforePut<reco::PFRecTrackCollection>();
  int idx = 0;

  for (unsigned int il=0; il<V0list_.size(); il++){
    Handle<VertexCompositeCandidateCollection> V0coll;
    iEvent.getByToken(V0list_[il],V0coll);
    LogDebug("PFV0Producer")<< "V0list_[" << il <<"] contains "<<V0coll->size()<<" V0 candidates ";
    for (unsigned int iv=0;iv<V0coll->size();iv++){
      VertexCompositeCandidateRef V0(V0coll, iv);
      vector<TrackRef> Tracks;
      vector<PFRecTrackRef> PFTracks;
      for( unsigned int ndx = 0; ndx < V0->numberOfDaughters(); ndx++ ) {
    
    Tracks.push_back( (dynamic_cast<const RecoChargedCandidate*>(V0->daughter(ndx)))->track() );
    TrackRef trackRef = (dynamic_cast<const RecoChargedCandidate*>(V0->daughter(ndx)))->track();

    reco::PFRecTrack pfRecTrack( trackRef->charge(), 
                   reco::PFRecTrack::KF, 
                   trackRef.key(), 
                   trackRef );


    Trajectory FakeTraj;
    bool valid = pfTransformer_->addPoints( pfRecTrack, *trackRef, FakeTraj);
    if(valid) {
      PFTracks.push_back(reco::PFRecTrackRef( pfTrackRefProd, idx++));
      pfV0RecTrackColl->push_back(pfRecTrack);
      
    }
      }
      if ((PFTracks.size()==2)&&(Tracks.size()==2)){
    pfV0Coll->push_back(PFV0(V0,PFTracks,Tracks));
      }
    
      

    }
  }
  
  
  iEvent.put(std::move(pfV0Coll));
  iEvent.put(std::move(pfV0RecTrackColl));
}

// ------------ method called once each job just before starting event loop  ------------
void 
PFV0Producer::beginRun(const edm::Run& run,
                       const EventSetup& iSetup)
{
  ESHandle<MagneticField> magneticField;
  iSetup.get<IdealMagneticFieldRecord>().get(magneticField);
  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 
PFV0Producer::endRun(const edm::Run& run,
             const EventSetup& iSetup) {
  delete pfTransformer_;
  pfTransformer_=nullptr;
}