#include <PFConversionProducer.h>

Inheritance diagram for PFConversionProducer:

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

	~PFConversionProducer ()
	Destructor. More...

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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
virtual void	beginRun (const edm::Run &, const edm::EventSetup &) override

virtual void	endRun (const edm::Run &, const edm::EventSetup &) override

virtual 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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 14 of file PFConversionProducer.h.

Constructor & Destructor Documentation

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

explicit

Constructor.

Definition at line 22 of file PFConversionProducer.cc.

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

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

PFConversionProducer::~PFConversionProducer ( )

Destructor.

Definition at line 33 of file PFConversionProducer.cc.

References pfTransformer_.

 {
   delete pfTransformer_;
 }

Member Function Documentation

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

overrideprivatevirtual

Definition at line 181 of file PFConversionProducer.cc.

References edm::EventSetup::get(), MagneticField::inTesla(), seedCreatorFromRegionConsecutiveHitsEDProducer_cff::magneticField, PFTrackTransformer::OnlyProp(), and pfTransformer_.

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

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

overrideprivatevirtual

Definition at line 192 of file PFConversionProducer.cc.

References pfTransformer_.

                                            {
   delete pfTransformer_;
   pfTransformer_=nullptr;
 }

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

overrideprivatevirtual

Produce the PFRecTrack collection.

Definition at line 39 of file PFConversionProducer.cc.

References PFTrackTransformer::addPoints(), reco::Conversion::arbitratedMergedEcalGeneral, TransientTrackBuilder::build(), HiEvtPlane_cfi::chi2, ChiSquaredProbability(), MillePedeFileConverter_cfg::e, reco::PFTrajectoryPoint::ECALEntrance, cropTnPTrees::frac, edm::EventSetup::get(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), reco::Conversion::highPurity, training_settings::idx, edm::HandleBase::isValid(), reco::PFTrajectoryPoint::isValid(), TrackingRecHit::isValid(), edm::Ref< C, T, F >::key(), reco::PFRecTrack::KF, eostools::move(), ndof, AlCaHLTBitMon_ParallelJobs::p, pfConversionContainer_, pfTransformer_, edm::Event::put(), MetAnalyzer::pv(), btvTracks_cfi::quality, edm::second(), TrackingRecHit::sharesInput(), IPTools::signedTransverseImpactParameter(), TrackingRecHit::some, and vtx_h.

 {
   
   //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(trackingRecHit_iterator iHit1=trackRef1->recHitsBegin(); iHit1!=trackRef1->recHitsEnd(); iHit1++) 
             {
               const TrackingRecHit *h_1=(*iHit1);
               if(h_1->isValid()){         
             for(trackingRecHit_iterator iHit2=trackRef2->recHitsBegin(); iHit2!=trackRef2->recHitsEnd(); iHit2++)
               {
                 const TrackingRecHit *h_2=(*iHit2);
                 if(h_2->isValid() && h_1->sharesInput(h_2, TrackingRecHit::some))shared++;//count number of shared hits
               }
               }
             }         
           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));    
 }