#include <PFConversionProducer.h>

Inheritance diagram for PFConversionProducer:

Public Member Functions
	PFConversionProducer (const edm::ParameterSet &)
	Constructor. More...

	~PFConversionProducer () override
	Destructor. More...

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

Private Member Functions
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
	Produce the PFRecTrack collection. More...

Private Attributes
edm::EDGetTokenT< reco::ConversionCollection >	pfConversionContainer_

PFTrackTransformer *	pfTransformer_
	PFTrackTransformer. More...

edm::EDGetTokenT< reco::VertexCollection >	vtx_h

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 14 of file PFConversionProducer.h.

Constructor & Destructor Documentation

◆ PFConversionProducer()

PFConversionProducer::PFConversionProducer ( const edm::ParameterSet & iConfig )

explicit

Constructor.

Definition at line 19 of file PFConversionProducer.cc.

                                                                       : pfTransformer_(nullptr) {
   produces<reco::PFRecTrackCollection>();
   produces<reco::PFConversionCollection>();
  
   pfConversionContainer_ = consumes<reco::ConversionCollection>(iConfig.getParameter<InputTag>("conversionCollection"));
  
   vtx_h = consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("PrimaryVertexLabel"));
 }

References edm::ParameterSet::getParameter(), pfConversionContainer_, and vtx_h.

◆ ~PFConversionProducer()

PFConversionProducer::~PFConversionProducer ( )

override

Destructor.

Definition at line 28 of file PFConversionProducer.cc.

28 { delete pfTransformer_; }

References pfTransformer_.

Member Function Documentation

◆ beginRun()

void PFConversionProducer::beginRun	(	const edm::Run &	run,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 166 of file PFConversionProducer.cc.

                                                                                {
   ESHandle<MagneticField> magneticField;
   iSetup.get<IdealMagneticFieldRecord>().get(magneticField);
   pfTransformer_ = new PFTrackTransformer(math::XYZVector(magneticField->inTesla(GlobalPoint(0, 0, 0))));
   pfTransformer_->OnlyProp();
 }

References edm::EventSetup::get(), get, HLT_2018_cff::magneticField, PFTrackTransformer::OnlyProp(), and pfTransformer_.

◆ endRun()

void PFConversionProducer::endRun	(	const edm::Run &	run,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 174 of file PFConversionProducer.cc.

                                                                              {
   delete pfTransformer_;
   pfTransformer_ = nullptr;
 }

References pfTransformer_.

◆ produce()

void PFConversionProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Produce the PFRecTrack collection.

Definition at line 30 of file PFConversionProducer.cc.

                                                                           {
   //create the empty collections
   auto pfConversionColl = std::make_unique<reco::PFConversionCollection>();
   auto pfRecTrackColl = std::make_unique<reco::PFRecTrackCollection>();
  
   edm::ESHandle<TransientTrackBuilder> builder;
   iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", builder);
   TransientTrackBuilder thebuilder = *(builder.product());
   reco::PFRecTrackRefProd pfTrackRefProd = iEvent.getRefBeforePut<reco::PFRecTrackCollection>();
   Handle<reco::ConversionCollection> convCollH;
   iEvent.getByToken(pfConversionContainer_, convCollH);
  
   const reco::ConversionCollection& convColl = *(convCollH.product());
  
   Handle<reco::VertexCollection> vertex;
   iEvent.getByToken(vtx_h, vertex);
   //Find PV for IP calculation, if there is no PV in collection than use dummy
   reco::Vertex dummy;
   const reco::Vertex* pv = &dummy;
   if (vertex.isValid()) {
     pv = &*vertex->begin();
   } else {  // create a dummy PV
     reco::Vertex::Error e;
     e(0, 0) = 0.0015 * 0.0015;
     e(1, 1) = 0.0015 * 0.0015;
     e(2, 2) = 15. * 15.;
     reco::Vertex::Point p(0, 0, 0);
     dummy = reco::Vertex(p, e, 0, 0, 0);
   }
  
   int idx = 0;                                    //index of track in PFRecTrack collection
   multimap<unsigned int, unsigned int> trackmap;  //Map of Collections and tracks
   std::vector<unsigned int> conv_coll(0);
  
   // CLEAN CONVERSION COLLECTION FOR DUPLICATES
   for (unsigned int icoll1 = 0; icoll1 < convColl.size(); icoll1++) {
     if ((!convColl[icoll1].quality(reco::Conversion::arbitratedMergedEcalGeneral)) ||
         (!convColl[icoll1].quality(reco::Conversion::highPurity)))
       continue;
  
     bool greater_prob = false;
     std::vector<edm::RefToBase<reco::Track> > tracksRefColl1 = convColl[icoll1].tracks();
     for (unsigned it1 = 0; it1 < tracksRefColl1.size(); it1++) {
       reco::TrackRef trackRef1 = (tracksRefColl1[it1]).castTo<reco::TrackRef>();
  
       for (unsigned int icoll2 = 0; icoll2 < convColl.size(); icoll2++) {
         if (icoll1 == icoll2)
           continue;
         if ((!convColl[icoll2].quality(reco::Conversion::arbitratedMergedEcalGeneral)) ||
             (!convColl[icoll2].quality(reco::Conversion::highPurity)))
           continue;
         std::vector<edm::RefToBase<reco::Track> > tracksRefColl2 = convColl[icoll2].tracks();
         for (unsigned it2 = 0; it2 < tracksRefColl2.size(); it2++) {
           reco::TrackRef trackRef2 = (tracksRefColl2[it2]).castTo<reco::TrackRef>();
           double like1 = -999;
           double like2 = -999;
           //number of shared hits
           int shared = 0;
           for (auto const& hit1 : trackRef1->recHits())
             if (hit1->isValid()) {
               //count number of shared hits
               for (auto const& hit2 : trackRef2->recHits()) {
                 if (hit2->isValid() && hit1->sharesInput(hit2, TrackingRecHit::some))
                   shared++;
               }
             }
           float frac = 0;
           //number of valid hits in tracks that are duplicates
           float size1 = trackRef1->found();
           float size2 = trackRef2->found();
           //divide number of shared hits by the total number of hits for the track with less hits
           if (size1 > size2)
             frac = (double)shared / size2;
           else
             frac = (double)shared / size1;
           if (frac > 0.9) {
             like1 = ChiSquaredProbability(convColl[icoll1].conversionVertex().chi2(),
                                           convColl[icoll1].conversionVertex().ndof());
             like2 = ChiSquaredProbability(convColl[icoll2].conversionVertex().chi2(),
                                           convColl[icoll2].conversionVertex().ndof());
           }
           if (like2 > like1) {
             greater_prob = true;
             break;
           }
         }  //end loop over tracks in collection 2
  
         if (greater_prob)
           break;  //if a duplicate track is found in a collection with greater Chi^2 probability for Vertex fit then break out of comparison loop
       }           //end loop over collection 2 checking
       if (greater_prob)
         break;  //if a duplicate track is found in a collection with greater Chi^2 probability for Vertex fit then one does not need to check the other track the collection will not be stored
     }           //end loop over tracks in collection 1
     if (!greater_prob)
       conv_coll.push_back(icoll1);
   }  //end loop over collection 1
  
   //Finally fill empty collections
   for (unsigned iColl = 0; iColl < conv_coll.size(); iColl++) {
     unsigned int collindex = conv_coll[iColl];
     //std::cout<<"Filling this collection"<<collindex<<endl;
     std::vector<reco::PFRecTrackRef> pfRecTkcoll;
  
     std::vector<edm::RefToBase<reco::Track> > tracksRefColl = convColl[collindex].tracks();
     // convert the secondary tracks
     for (unsigned it = 0; it < tracksRefColl.size(); it++) {
       reco::TrackRef trackRef = (tracksRefColl[it]).castTo<reco::TrackRef>();
       reco::PFRecTrack pfRecTrack(trackRef->charge(), reco::PFRecTrack::KF, trackRef.key(), trackRef);
       //std::cout<<"Track Pt "<<trackRef->pt()<<std::endl;
       Trajectory FakeTraj;
       bool valid = pfTransformer_->addPoints(pfRecTrack, *trackRef, FakeTraj);
       if (valid) {
         double stip = -999;
         const reco::PFTrajectoryPoint& atECAL = pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance);
         //if extrapolation to ECAL is valid then calculate STIP
         if (atECAL.isValid()) {
           GlobalVector direction(pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().x(),
                                  pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().y(),
                                  pfRecTrack.extrapolatedPoint(reco::PFTrajectoryPoint::ECALEntrance).position().z());
           stip = IPTools::signedTransverseImpactParameter(thebuilder.build(*trackRef), direction, *pv)
                      .second.significance();
         }
         pfRecTrack.setSTIP(stip);
         pfRecTkcoll.push_back(reco::PFRecTrackRef(pfTrackRefProd, idx++));
         pfRecTrackColl->push_back(pfRecTrack);
       }
     }  //end loop over tracks
     //store reference to the Conversion collection
     reco::ConversionRef niRef(convCollH, collindex);
     pfConversionColl->push_back(reco::PFConversion(niRef, pfRecTkcoll));
   }  //end loop over collections
   iEvent.put(std::move(pfRecTrackColl));
   iEvent.put(std::move(pfConversionColl));
 }