#include <GoodSeedProducer.h>

Inheritance diagram for GoodSeedProducer:

Public Member Functions
	GoodSeedProducer (const edm::ParameterSet &)

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
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

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

	EDConsumerBase ()

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

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

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

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Types
typedef TrajectoryStateOnSurface	TSOS

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

void	fillPreIdRefValueMap (edm::Handle< reco::TrackCollection > tkhandle, const edm::OrphanHandle< reco::PreIdCollection > &, edm::ValueMap< reco::PreIdRef >::Filler &filler)

int	getBin (float, float)
	Find the bin in pt and eta. More...

virtual void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
math::XYZVector	B_
	B field. More...

float	chikfred

float	chiRatio

float	chired

double	clusThreshold_
	Cut on the energy of the clusters. More...

edm::ParameterSet	conf_

bool	disablePreId_
	switch to disable the pre-id More...

float	dpt

double	EcalStripSumE_deltaEta_

double	EcalStripSumE_deltaPhiOverQ_maxValue_

double	EcalStripSumE_deltaPhiOverQ_minValue_

double	EcalStripSumE_minClusEnergy_

float	eP
	VARIABLES NEEDED FOR TMVA. More...

float	eta

std::unique_ptr< TrajectoryFitter >	fitter_
	Fitter. More...

std::string	fitterName_

double	HcalIsolWindow_

TkClonerImpl	hitCloner

double	maxEp_

double	maxEta_

double	maxHoverP_

double	maxPt_

std::string	method_
	TMVA method. More...

double	Min_dr_

double	minEoverP_

double	minEp_
	Min and MAx allowed values forEoverP. More...

double	minPt_
	Minimum transverse momentum and maximum pseudorapidity. More...

float	nhit

int	nHitsInSeed_
	Number of hits in the seed;. More...

edm::EDGetTokenT < reco::PFClusterCollection >	pfCLusTagECLabel_

edm::EDGetTokenT < reco::PFClusterCollection >	pfCLusTagHCLabel_

edm::EDGetTokenT < reco::PFClusterCollection >	pfCLusTagPSLabel_

PFGeometry	pfGeometry_

std::unique_ptr < PFTrackTransformer >	pfTransformer_
	PFTrackTransformer. More...

std::string	preidckf_
	Name of the Seed(Ckf) Collection. More...

std::string	preidgsf_
	Name of the Seed(Gsf) Collection. More...

std::string	preidname_
	Name of the preid Collection (FB) More...

bool	produceCkfseed_
	Produce the Seed for Ckf tracks? More...

bool	producePreId_
	Produce the pre-id debugging collection. More...

std::string	propagatorName_

float	pt

double	PtThresholdSavePredId_
	Threshold to save Pre Idinfo. More...

std::array< std::unique_ptr < TMVA::Reader >, 9 >	reader {}
	READER FOR TMVA. More...

std::map< reco::TrackRef, unsigned >	refMap_
	Map used to create the TrackRef, PreIdRef value map. More...

std::unique_ptr< PFResolutionMap >	resMapEtaECAL_

std::unique_ptr< PFResolutionMap >	resMapPhiECAL_

std::unique_ptr < TrajectorySmoother >	smoother_
	Smoother. More...

std::string	smootherName_

float	thr [150]
	vector of thresholds for different bins of eta and pt More...

std::string	trackerRecHitBuilderName_

reco::TrackBase::TrackQuality	trackQuality_

std::vector< edm::EDGetTokenT < reco::TrackCollection > >	tracksContainers_

std::vector< edm::EDGetTokenT < std::vector< Trajectory > > >	trajContainers_

float	trk_ecalDeta

float	trk_ecalDphi

bool	useQuality_
	TRACK QUALITY. More...

bool	useTmva_
	USE OF TMVA. More...

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

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::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

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 52 of file GoodSeedProducer.h.

Member Typedef Documentation

typedef TrajectoryStateOnSurface GoodSeedProducer::TSOS

private

Definition at line 53 of file GoodSeedProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 41 of file GoodSeedProducer.cc.

References clusThreshold_, edm::EDConsumerBase::consumes(), disablePreId_, EcalStripSumE_deltaEta_, EcalStripSumE_deltaPhiOverQ_maxValue_, EcalStripSumE_deltaPhiOverQ_minValue_, EcalStripSumE_minClusEnergy_, fitterName_, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), HcalIsolWindow_, i, LogDebug, maxEp_, maxEta_, maxHoverP_, maxPt_, Min_dr_, minEoverP_, minEp_, minPt_, nHitsInSeed_, pfCLusTagECLabel_, pfCLusTagHCLabel_, pfCLusTagPSLabel_, preidckf_, preidgsf_, preidname_, produceCkfseed_, producePreId_, edm::ProductRegistryHelper::produces(), PtThresholdSavePredId_, smootherName_, AlCaHLTBitMon_QueryRunRegistry::string, o2o::tags, trackerRecHitBuilderName_, trackQuality_, tracksContainers_, trajContainers_, useQuality_, and useTmva_.

                                                              :
   pfTransformer_(nullptr),
   conf_(iConfig),
   resMapEtaECAL_(nullptr),
   resMapPhiECAL_(nullptr)
 {
   LogInfo("GoodSeedProducer")<<"Electron PreIdentification started  ";
   
   //now do what ever initialization is needed
   std::vector<edm::InputTag> tags =   iConfig.getParameter< vector < InputTag > >("TkColList");
   for(unsigned int i=0;i<tags.size();++i) {
     trajContainers_.push_back(consumes<vector<Trajectory> >(tags[i]));
     tracksContainers_.push_back(consumes<reco::TrackCollection>(tags[i]));
   }
   
   minPt_=iConfig.getParameter<double>("MinPt");
   maxPt_=iConfig.getParameter<double>("MaxPt");
   maxEta_=iConfig.getParameter<double>("MaxEta");
 
   HcalIsolWindow_                       =iConfig.getParameter<double>("HcalWindow");
   EcalStripSumE_minClusEnergy_ = iConfig.getParameter<double>("EcalStripSumE_minClusEnergy");
   EcalStripSumE_deltaEta_ = iConfig.getParameter<double>("EcalStripSumE_deltaEta");
   EcalStripSumE_deltaPhiOverQ_minValue_ = iConfig.getParameter<double>("EcalStripSumE_deltaPhiOverQ_minValue");
   EcalStripSumE_deltaPhiOverQ_maxValue_ = iConfig.getParameter<double>("EcalStripSumE_deltaPhiOverQ_maxValue");
   minEoverP_= iConfig.getParameter<double>("EOverPLead_minValue");
   maxHoverP_= iConfig.getParameter<double>("HOverPLead_maxValue");
  
   pfCLusTagECLabel_=consumes<reco::PFClusterCollection>(iConfig.getParameter<InputTag>("PFEcalClusterLabel"));
 
   pfCLusTagHCLabel_=consumes<reco::PFClusterCollection>(iConfig.getParameter<InputTag>("PFHcalClusterLabel"));  
 
   pfCLusTagPSLabel_=consumes<reco::PFClusterCollection>(iConfig.getParameter<InputTag>("PFPSClusterLabel"));
   
   preidgsf_ = iConfig.getParameter<string>("PreGsfLabel");
   preidckf_ = iConfig.getParameter<string>("PreCkfLabel");
   preidname_= iConfig.getParameter<string>("PreIdLabel");
   
   
   fitterName_ = iConfig.getParameter<string>("Fitter");
   smootherName_ = iConfig.getParameter<string>("Smoother");
   
   
   nHitsInSeed_=iConfig.getParameter<int>("NHitsInSeed");
 
   clusThreshold_=iConfig.getParameter<double>("ClusterThreshold");
   
   minEp_=iConfig.getParameter<double>("MinEOverP");
   maxEp_=iConfig.getParameter<double>("MaxEOverP");
 
   //collection to produce
   produceCkfseed_ = iConfig.getUntrackedParameter<bool>("ProduceCkfSeed",false);
 
   // to disable the electron part (for HI collisions for examples) 
   disablePreId_ = iConfig.getUntrackedParameter<bool>("DisablePreId",false);  
 
   producePreId_ = iConfig.getUntrackedParameter<bool>("ProducePreId",true);  
   //  if no electron, cannot produce the preid
   if(disablePreId_) 
     producePreId_=false;
   PtThresholdSavePredId_ = iConfig.getUntrackedParameter<double>("PtThresholdSavePreId",1.);  
 
   LogDebug("GoodSeedProducer")<<"Seeds for GSF will be produced ";
 
   // no disablePreId_ switch here. The collection will be empty if it is true
   produces<ElectronSeedCollection>(preidgsf_);
 
   if(produceCkfseed_){
     LogDebug("GoodSeedProducer")<<"Seeds for CKF will be produced ";
     produces<TrajectorySeedCollection>(preidckf_);
   }
 
   if(producePreId_){
     LogDebug("GoodSeedProducer")<<"PreId debugging information will be produced ";
 
     produces<PreIdCollection>(preidname_);
     if(tracksContainers_.size()==1) // do not make a value map if more than one input track collection
       produces<edm::ValueMap<reco::PreIdRef> >(preidname_);
   } 
   
   useQuality_   = iConfig.getParameter<bool>("UseQuality");
   trackQuality_=TrackBase::qualityByName(iConfig.getParameter<std::string>("TrackQuality"));
 
   useTmva_= iConfig.getUntrackedParameter<bool>("UseTMVA",false);
 
   Min_dr_ = iConfig.getParameter<double>("Min_dr");
 
   trackerRecHitBuilderName_ = iConfig.getParameter<std::string>("TTRHBuilder");
 
 }

Member Function Documentation

void GoodSeedProducer::beginRun	(	const edm::Run &	run,
		const edm::EventSetup &	es
	)

overrideprivatevirtual

Reimplemented from edm::stream::EDProducerBase.

Definition at line 477 of file GoodSeedProducer.cc.

References B_, chikfred, chiRatio, chired, conf_, dpt, eP, eta, edm::FileInPath::fullPath(), edm::EventSetup::get(), edm::ParameterSet::getParameter(), j, HLT_25ns14e33_v1_cff::magneticField, method_, nhit, pfTransformer_, pt, reader, resMapEtaECAL_, resMapPhiECAL_, thr, trk_ecalDeta, trk_ecalDphi, and useTmva_.

 {
   //Magnetic Field
   ESHandle<MagneticField> magneticField;
   es.get<IdealMagneticFieldRecord>().get(magneticField);
   B_=magneticField->inTesla(GlobalPoint(0,0,0));
   
   pfTransformer_.reset( new PFTrackTransformer(B_) );
   pfTransformer_->OnlyProp();
   
   //Resolution maps
   FileInPath ecalEtaMap(conf_.getParameter<string>("EtaMap"));
   FileInPath ecalPhiMap(conf_.getParameter<string>("PhiMap"));
   resMapEtaECAL_.reset( new PFResolutionMap("ECAL_eta",ecalEtaMap.fullPath().c_str()) );
   resMapPhiECAL_.reset( new PFResolutionMap("ECAL_phi",ecalPhiMap.fullPath().c_str()) );
 
   if(useTmva_){
     method_ = conf_.getParameter<string>("TMVAMethod");
     FileInPath Weigths1(conf_.getParameter<string>("Weights1"));
     FileInPath Weigths2(conf_.getParameter<string>("Weights2"));
     FileInPath Weigths3(conf_.getParameter<string>("Weights3"));
     FileInPath Weigths4(conf_.getParameter<string>("Weights4"));
     FileInPath Weigths5(conf_.getParameter<string>("Weights5"));
     FileInPath Weigths6(conf_.getParameter<string>("Weights6"));
     FileInPath Weigths7(conf_.getParameter<string>("Weights7"));
     FileInPath Weigths8(conf_.getParameter<string>("Weights8"));
     FileInPath Weigths9(conf_.getParameter<string>("Weights9"));
 
     for(UInt_t j = 0; j < 9; ++j){
       reader[j].reset( new TMVA::Reader("!Color:Silent"));
       
       reader[j]->AddVariable("NHits", &nhit);
       reader[j]->AddVariable("NormChi", &chikfred);
       reader[j]->AddVariable("dPtGSF", &dpt);
       reader[j]->AddVariable("EoP", &eP);
       reader[j]->AddVariable("ChiRatio", &chiRatio);
       reader[j]->AddVariable("RedChi", &chired);
       reader[j]->AddVariable("EcalDEta", &trk_ecalDeta);
       reader[j]->AddVariable("EcalDPhi", &trk_ecalDphi);
       reader[j]->AddVariable("pt", &pt);
       reader[j]->AddVariable("eta", &eta);
       
       if(j==0) reader[j]->BookMVA(method_, Weigths1.fullPath().c_str());
       if(j==1) reader[j]->BookMVA(method_, Weigths2.fullPath().c_str());
       if(j==2) reader[j]->BookMVA(method_, Weigths3.fullPath().c_str());
       if(j==3) reader[j]->BookMVA(method_, Weigths4.fullPath().c_str());
       if(j==4) reader[j]->BookMVA(method_, Weigths5.fullPath().c_str());
       if(j==5) reader[j]->BookMVA(method_, Weigths6.fullPath().c_str());
       if(j==6) reader[j]->BookMVA(method_, Weigths7.fullPath().c_str());
       if(j==7) reader[j]->BookMVA(method_, Weigths8.fullPath().c_str());
       if(j==8) reader[j]->BookMVA(method_, Weigths9.fullPath().c_str());
     }    
   }
   //read threshold
   FileInPath parFile(conf_.getParameter<string>("ThresholdFile"));
   ifstream ifs(parFile.fullPath().c_str());
   for (int iy=0;iy<81;++iy) ifs >> thr[iy];
 }

void GoodSeedProducer::fillPreIdRefValueMap	(	edm::Handle< reco::TrackCollection >	tkhandle,
		const edm::OrphanHandle< reco::PreIdCollection > &	preidhandle,
		edm::ValueMap< reco::PreIdRef >::Filler &	filler
	)

private

Definition at line 554 of file GoodSeedProducer.cc.

References edm::helper::Filler< Map >::insert(), refMap_, and makeHLTPrescaleTable::values.

Referenced by produce().

 {
   std::vector<reco::PreIdRef> values;
 
   unsigned ntracks=tracks->size();
   for(unsigned itrack=0;itrack<ntracks;++itrack)
    {
      reco::TrackRef theTrackRef(tracks,itrack);
      std::map<reco::TrackRef,unsigned>::const_iterator itcheck=refMap_.find(theTrackRef);
      if(itcheck==refMap_.end()) 
        {
      // the track has been early discarded
      values.push_back(reco::PreIdRef());
        }
      else
        {
      edm::Ref<reco::PreIdCollection> preIdRef(preidhandle,itcheck->second);
      values.push_back(preIdRef);
      //  std::cout << " Checking Refs " << (theTrackRef==preIdRef->trackRef()) << std::endl;
        }
    }
   filler.insert(tracks,values.begin(),values.end());
 }

int GoodSeedProducer::getBin	(	float	eta,
		float	pt
	)

private

Find the bin in pt and eta.

Definition at line 538 of file GoodSeedProducer.cc.

References LogDebug.

Referenced by produce().

                                            {
   int ie=0;
   int ip=0;
   if (fabs(eta)<0.8) ie=0;
   else{ if (fabs(eta)<1.479) ie=1;
     else ie=2;
   }
   if (pt<6) ip=0;
   else {  if (pt<12) ip=1;     
     else ip=2;
   }
   int iep= ie*3+ip;
   LogDebug("GoodSeedProducer")<<"Track pt ="<<pt<<" eta="<<eta<<" bin="<<iep;
   return iep;
 }

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

overrideprivatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 138 of file GoodSeedProducer.cc.

References funct::abs(), B_, chikfred, chiRatio, chired, Trajectory::chiSquared(), clusThreshold_, constexpr, disablePreId_, dpt, HWWFunctions::EE, eP, eta, edm::EventID::event(), f, edm::helper::Filler< Map >::fill(), fillPreIdRefValueMap(), Trajectory::firstMeasurement(), fitter_, fitterName_, edm::EventSetup::get(), getBin(), edm::Event::getByToken(), hitCloner, i, edm::EventBase::id(), Trajectory::isValid(), Trajectory::lastMeasurement(), LogDebug, maxEp_, maxEta_, maxPt_, method_, Min_dr_, minEp_, minPt_, nhit, PV3DBase< T, PVType, FrameType >::perp2(), pfCLusTagECLabel_, pfCLusTagHCLabel_, pfCLusTagPSLabel_, Pi, funct::pow(), preidckf_, preidgsf_, preidname_, produceCkfseed_, producePreId_, edm::Handle< T >::product(), edm::ESHandle< class >::product(), reco::PreId::pt(), pt, PtThresholdSavePredId_, edm::Event::put(), HLT_25ns14e33_v1_cff::quality, reader, refMap_, resMapEtaECAL_, resMapPhiECAL_, edm::EventID::run(), RawParticle::setCharge(), reco::ElectronSeed::setCtfTrack(), reco::PreId::setECALMatching(), reco::PreId::setECALMatchingProperties(), reco::PreId::setFinalDecision(), reco::PreId::setMVA(), reco::PreId::setTrack(), reco::PreId::setTrackFiltering(), reco::PreId::setTrackProperties(), smoother_, smootherName_, mathSSE::sqrt(), thr, tmp, trackerRecHitBuilderName_, trackQuality_, tracksContainers_, trajContainers_, trk_ecalDeta, trk_ecalDphi, TwoPi, csvLumiCalc::unit, useQuality_, useTmva_, x, PV3DBase< T, PVType, FrameType >::x(), detailsBasic3DVector::y, PV3DBase< T, PVType, FrameType >::y(), detailsBasic3DVector::z, and PV3DBase< T, PVType, FrameType >::z().

 {
 
   LogDebug("GoodSeedProducer")<<"START event: "<<iEvent.id().event()
                   <<" in run "<<iEvent.id().run();
   //Create empty output collections
   auto_ptr<ElectronSeedCollection> output_preid(new ElectronSeedCollection);
   auto_ptr<TrajectorySeedCollection> output_nopre(new TrajectorySeedCollection);
   auto_ptr<PreIdCollection> output_preidinfo(new PreIdCollection);
   auto_ptr<edm::ValueMap<reco::PreIdRef> > preIdMap_p(new edm::ValueMap<reco::PreIdRef>);
   edm::ValueMap<reco::PreIdRef>::Filler mapFiller(*preIdMap_p);
 
   //Tracking Tools
   if(!disablePreId_)
     {
       edm::ESHandle<TrajectoryFitter> aFitter;
       edm::ESHandle<TrajectorySmoother> aSmoother;
       iSetup.get<TrajectoryFitter::Record>().get(fitterName_, aFitter);
       iSetup.get<TrajectoryFitter::Record>().get(smootherName_, aSmoother);
       smoother_.reset(aSmoother->clone());
       fitter_ = aFitter->clone();
       edm::ESHandle<TransientTrackingRecHitBuilder> theTrackerRecHitBuilder;
       iSetup.get<TransientRecHitRecord>().get(trackerRecHitBuilderName_,theTrackerRecHitBuilder);
       hitCloner = static_cast<TkTransientTrackingRecHitBuilder const *>(theTrackerRecHitBuilder.product())->cloner();
       fitter_->setHitCloner(&hitCloner);
       smoother_->setHitCloner(&hitCloner);
     }
 
   // clear temporary maps
   refMap_.clear();
 
   //Handle input collections
   //ECAL clusters         
   Handle<PFClusterCollection> theECPfClustCollection;
   iEvent.getByToken(pfCLusTagECLabel_,theECPfClustCollection);
   
 
  vector<PFCluster const *> basClus;
   for ( auto const & klus : *theECPfClustCollection.product() ) {
     if(klus.correctedEnergy()>clusThreshold_) basClus.push_back(&klus);
   }
 
   //HCAL clusters
   Handle<PFClusterCollection> theHCPfClustCollection;
   iEvent.getByToken(pfCLusTagHCLabel_,theHCPfClustCollection);
   
   //PS clusters
   Handle<PFClusterCollection> thePSPfClustCollection;
   iEvent.getByToken(pfCLusTagPSLabel_,thePSPfClustCollection);
 
   //Vector of track collections
   for (unsigned int istr=0; istr<tracksContainers_.size();++istr){
     
     //Track collection
     Handle<TrackCollection> tkRefCollection;
     iEvent.getByToken(tracksContainers_[istr], tkRefCollection);
     const TrackCollection&  Tk=*(tkRefCollection.product());
     
     //Trajectory collection
     Handle<vector<Trajectory> > tjCollection;
     iEvent.getByToken(trajContainers_[istr], tjCollection);
     auto const & Tj=*(tjCollection.product());
     
     LogDebug("GoodSeedProducer")<<"Number of tracks in collection "
                                 <<tracksContainers_[istr] <<" to be analyzed "
                                 <<Tj.size();
 
     //loop over the track collection
     for(unsigned int i=0;i<Tk.size();++i){      
       if (useQuality_ &&
       (!(Tk[i].quality(trackQuality_)))) continue;
       
       reco::PreId myPreId;
       bool GoodPreId=false;
 
       TrackRef trackRef(tkRefCollection, i);
       // TrajectorySeed Seed=Tj[i].seed();
       math::XYZVectorF tkmom(Tk[i].momentum());
       auto tketa= tkmom.eta();
       auto tkpt = std::sqrt(tkmom.perp2());
       auto const & Seed=(*trackRef->seedRef());
       if(!disablePreId_)
     {
       int ipteta=getBin(Tk[i].eta(),Tk[i].pt());
       int ibin=ipteta*9;
       
       float oPTOB=1.f/Tj[i].lastMeasurement().updatedState().globalMomentum().mag();
       //  float chikfred=Tk[i].normalizedChi2();
       float nchi=Tk[i].normalizedChi2();
 
       int nhitpi=Tj[i].foundHits();
       float EP=0;
       
       // set track info
       myPreId.setTrack(trackRef);
       //CLUSTERS - TRACK matching
       
       auto pfmass=  0.0005;
       auto pfoutenergy=sqrt((pfmass*pfmass)+Tk[i].outerMomentum().Mag2());
 
       XYZTLorentzVector mom =XYZTLorentzVector(Tk[i].outerMomentum().x(),
                            Tk[i].outerMomentum().y(),
                            Tk[i].outerMomentum().z(),
                            pfoutenergy);
       XYZTLorentzVector pos =   XYZTLorentzVector(Tk[i].outerPosition().x(),
                               Tk[i].outerPosition().y(),
                               Tk[i].outerPosition().z(),
                               0.);
 
       BaseParticlePropagator theOutParticle( RawParticle(mom,pos),
                     0,0,B_.z());
       theOutParticle.setCharge(Tk[i].charge());
       
       theOutParticle.propagateToEcalEntrance(false);
       
 
       
       float toteta=1000.f;
       float totphi=1000.f;
       float dr=1000.f;
       float EE=0.f;
       float feta=0.f;
       GlobalPoint ElecTrkEcalPos(0,0,0);
 
       PFClusterRef clusterRef;
       math::XYZPoint meanShowerSaved;
       if(theOutParticle.getSuccess()!=0){
          ElecTrkEcalPos=GlobalPoint(theOutParticle.vertex().x(),
                             theOutParticle.vertex().y(),
                     theOutParticle.vertex().z()
                                        );
 
             constexpr float psLim = 2.50746495928f; // std::sinh(1.65f);
             bool isBelowPS= (ElecTrkEcalPos.z()*ElecTrkEcalPos.z()) > (psLim*psLim)*ElecTrkEcalPos.perp2();
         // bool isBelowPS=(std::abs(ElecTrkEcalPos.eta())>1.65f);   
     
         unsigned clusCounter=0;
         float max_ee = 0;
         for(auto aClus : basClus) {
 
           float tmp_ep=float(aClus->correctedEnergy())*oPTOB;
               if ((tmp_ep<minEp_)|(tmp_ep>maxEp_)) { ++clusCounter; continue;}
         
           double ecalShowerDepth
         = PFCluster::getDepthCorrection(aClus->correctedEnergy(),
                         isBelowPS,
                         false);
           auto mom = theOutParticle.momentum().Vect();
           auto meanShower = ElecTrkEcalPos +
         GlobalVector(mom.x(),mom.y(),mom.z()).unit()*ecalShowerDepth;   
       
           float etarec=meanShower.eta();
           float phirec=meanShower.phi();
          
 
           float tmp_phi=std::abs(aClus->positionREP().phi()-phirec);
           if (tmp_phi>float(TMath::Pi())) tmp_phi-= float(TMath::TwoPi());
           
           float tmp_dr=std::sqrt(std::pow(tmp_phi,2.f)+
                 std::pow(aClus->positionREP().eta()-etarec,2.f));
       
           if (tmp_dr<dr){
         dr=tmp_dr;
         if(dr < Min_dr_){ // find the most closest and energetic ECAL cluster
           if(aClus->correctedEnergy() > max_ee){
 
             toteta=aClus->positionREP().eta()-etarec;
             totphi=tmp_phi;
             EP=tmp_ep;
             EE=aClus->correctedEnergy();
             feta= aClus->positionREP().eta();
             clusterRef = PFClusterRef(theECPfClustCollection,clusCounter);
             meanShowerSaved = meanShower;
             
           }
         }
           }
               ++clusCounter;
         }
       }
       float trk_ecalDeta_ = fabs(toteta);
       float trk_ecalDphi_ = fabs(totphi);
 
       //Resolution maps
       auto ecaletares 
         = resMapEtaECAL_->GetBinContent(resMapEtaECAL_->FindBin(feta,EE));
       auto ecalphires 
         = resMapPhiECAL_->GetBinContent(resMapPhiECAL_->FindBin(feta,EE)); 
       
       //geomatrical compatibility
       float chieta=(toteta!=1000.f)? toteta/ecaletares : toteta;
       float chiphi=(totphi!=1000.f)? totphi/ecalphires : totphi;
       float chichi= sqrt(chieta*chieta + chiphi*chiphi);
       
       //Matching criteria
       float eta_cut = thr[ibin+0];
       float phi_cut = thr[ibin+1];
       float ep_cutmin=thr[ibin+2];
       bool GoodMatching= ((trk_ecalDeta_<eta_cut) && (trk_ecalDphi_<phi_cut) && (EP>ep_cutmin) && (nhitpi>10));
 
       bool EcalMatching=GoodMatching;
       
       if (tkpt>maxPt_) GoodMatching=true;
       if (tkpt<minPt_) GoodMatching=false;
 
 
   
       math::XYZPoint myPoint(ElecTrkEcalPos.x(),ElecTrkEcalPos.y(),ElecTrkEcalPos.z());
       myPreId.setECALMatchingProperties(clusterRef,myPoint,meanShowerSaved,std::abs(toteta),std::abs(totphi),chieta,
                         chiphi,chichi,EP);
       myPreId.setECALMatching(EcalMatching);
 
 
       bool GoodRange= ((std::abs(tketa)<maxEta_) & 
                (tkpt>minPt_));
       //KF FILTERING FOR UNMATCHED EVENTS
       int hit1max=int(thr[ibin+3]);
       float chiredmin=thr[ibin+4];
       bool GoodKFFiltering =
         ((nchi>chiredmin) | (nhitpi<hit1max));
       
 
       myPreId.setTrackFiltering(GoodKFFiltering);
 
       bool GoodTkId= false;
       
       if((!GoodMatching) &&(GoodKFFiltering) &&(GoodRange)){
         chired=1000;
         chiRatio=1000;
         dpt=0;
         nhit=nhitpi;
         chikfred = nchi;
         trk_ecalDeta = trk_ecalDeta_;
         trk_ecalDphi = trk_ecalDphi_;
       
         Trajectory::ConstRecHitContainer tmp;
         Trajectory::ConstRecHitContainer && hits=Tj[i].recHits();
         for (int ih=hits.size()-1; ih>=0; ih--)  tmp.push_back(hits[ih]);
         Trajectory  && FitTjs= fitter_->fitOne(Seed,tmp,Tj[i].lastMeasurement().updatedState());
     
           if(FitTjs.isValid()){
         Trajectory && SmooTjs= smoother_->trajectory(FitTjs);
           if(SmooTjs.isValid()){
         
             //Track refitted with electron hypothesis
         
             float pt_out=SmooTjs.firstMeasurement().
               updatedState().globalMomentum().perp();
             float pt_in=SmooTjs.lastMeasurement().
               updatedState().globalMomentum().perp();
             dpt=(pt_in>0) ? fabs(pt_out-pt_in)/pt_in : 0.;
             // the following is simply the number of degrees of freedom
             chiRatio=SmooTjs.chiSquared()/Tj[i].chiSquared();
             chired=chiRatio*chikfred;
 
           }
         }
          
     
         //TMVA Analysis
         if(useTmva_){
     
           eta=tketa;
           pt=tkpt;
           eP=EP;
       
           float Ytmva=reader[ipteta]->EvaluateMVA( method_ );
           
           float BDTcut=thr[ibin+5]; 
           if ( Ytmva>BDTcut) GoodTkId=true;
           myPreId.setMVA(GoodTkId,Ytmva);
           myPreId.setTrackProperties(chired,chiRatio,dpt);
         }else{ 
               
           float chiratiocut=thr[ibin+6]; 
           float gschicut=thr[ibin+7]; 
           float gsptmin=thr[ibin+8];
 
           GoodTkId=((dpt>gsptmin)&(chired<gschicut)&(chiRatio<chiratiocut));      
        
         }
       }
     
       GoodPreId= GoodTkId | GoodMatching; 
 
       myPreId.setFinalDecision(GoodPreId);
       
       if(GoodPreId)
         LogDebug("GoodSeedProducer")<<"Track (pt= "<<Tk[i].pt()<<
           "GeV/c, eta= "<<Tk[i].eta() <<
           ") preidentified for agreement between  track and ECAL cluster";
       if(GoodPreId &&(!GoodMatching))
         LogDebug("GoodSeedProducer")<<"Track (pt= "<<Tk[i].pt()<<
           "GeV/c, eta= "<<Tk[i].eta() <<
           ") preidentified only for track properties";
     
     } // end of !disablePreId_
       
       if (GoodPreId){
     //NEW SEED with n hits  
     ElectronSeed NewSeed(Seed);
     NewSeed.setCtfTrack(trackRef);
     output_preid->push_back(NewSeed);
       }else{
     if (produceCkfseed_){
       output_nopre->push_back(Seed);
     }
       }
       if(producePreId_ && myPreId.pt()>PtThresholdSavePredId_)
     {
       // save the index of the PreId object as to be able to create a Ref later
       refMap_[trackRef] = output_preidinfo->size();
       output_preidinfo->push_back(myPreId);   
     }
     } //end loop on track collection
   } //end loop on the vector of track collections
   
   // no disablePreId_ switch, it is simpler to have an empty collection rather than no collection
   iEvent.put(output_preid,preidgsf_);
   if (produceCkfseed_)
     iEvent.put(output_nopre,preidckf_);
   if(producePreId_)
     {
       const edm::OrphanHandle<reco::PreIdCollection> preIdRefProd = iEvent.put(output_preidinfo,preidname_);
       // now make the Value Map, but only if one input collection
       if(tracksContainers_.size()==1)
     {
       Handle<TrackCollection> tkRefCollection ;
       iEvent.getByToken(tracksContainers_[0],tkRefCollection);
       fillPreIdRefValueMap(tkRefCollection,preIdRefProd,mapFiller);
       mapFiller.fill();
       iEvent.put(preIdMap_p,preidname_);
     }
     }
   // clear temporary maps
   refMap_.clear();
 }

Member Data Documentation

math::XYZVector GoodSeedProducer::B_

private

B field.

Definition at line 163 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::chikfred

private

Definition at line 153 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::chiRatio

private

Definition at line 152 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::chired

private

Definition at line 152 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

double GoodSeedProducer::clusThreshold_

private

Cut on the energy of the clusters.

Definition at line 107 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

edm::ParameterSet GoodSeedProducer::conf_

private

Definition at line 129 of file GoodSeedProducer.h.

Referenced by beginRun().

bool GoodSeedProducer::disablePreId_

private

switch to disable the pre-id

Definition at line 117 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

float GoodSeedProducer::dpt

private

Definition at line 152 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

double GoodSeedProducer::EcalStripSumE_deltaEta_

private

Definition at line 100 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::EcalStripSumE_deltaPhiOverQ_maxValue_

private

Definition at line 102 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::EcalStripSumE_deltaPhiOverQ_minValue_

private

Definition at line 101 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::EcalStripSumE_minClusEnergy_

private

Definition at line 99 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

float GoodSeedProducer::eP

private

VARIABLES NEEDED FOR TMVA.

Definition at line 152 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::eta

private

Definition at line 152 of file GoodSeedProducer.h.

Referenced by Particle.Particle::__str__(), Muon.Muon::absEffAreaIso(), beginRun(), Jet.Jet::jetID(), produce(), and Jet.Jet::puJetId().

std::unique_ptr<TrajectoryFitter> GoodSeedProducer::fitter_

private

Fitter.

Definition at line 79 of file GoodSeedProducer.h.

Referenced by produce().

std::string GoodSeedProducer::fitterName_

private

Definition at line 136 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::HcalIsolWindow_

private

Definition at line 98 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

TkClonerImpl GoodSeedProducer::hitCloner

private

Definition at line 85 of file GoodSeedProducer.h.

Referenced by produce().

double GoodSeedProducer::maxEp_

private

Definition at line 111 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::maxEta_

private

Definition at line 96 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::maxHoverP_

private

Definition at line 104 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::maxPt_

private

Definition at line 95 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::method_

private

TMVA method.

Definition at line 160 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

double GoodSeedProducer::Min_dr_

private

Definition at line 154 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::minEoverP_

private

Definition at line 103 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::minEp_

private

Min and MAx allowed values forEoverP.

Definition at line 110 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::minPt_

private

Minimum transverse momentum and maximum pseudorapidity.

Definition at line 94 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

float GoodSeedProducer::nhit

private

Definition at line 152 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

int GoodSeedProducer::nHitsInSeed_

private

Number of hits in the seed;.

Definition at line 91 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

edm::EDGetTokenT<reco::PFClusterCollection> GoodSeedProducer::pfCLusTagECLabel_

private

Definition at line 131 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

edm::EDGetTokenT<reco::PFClusterCollection> GoodSeedProducer::pfCLusTagHCLabel_

private

Definition at line 132 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

edm::EDGetTokenT<reco::PFClusterCollection> GoodSeedProducer::pfCLusTagPSLabel_

private

Definition at line 130 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

PFGeometry GoodSeedProducer::pfGeometry_

private

Definition at line 168 of file GoodSeedProducer.h.

std::unique_ptr<PFTrackTransformer> GoodSeedProducer::pfTransformer_

private

PFTrackTransformer.

Definition at line 88 of file GoodSeedProducer.h.

Referenced by beginRun().

std::string GoodSeedProducer::preidckf_

private

Name of the Seed(Ckf) Collection.

Definition at line 70 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::preidgsf_

private

Name of the Seed(Gsf) Collection.

Definition at line 73 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::preidname_

private

Name of the preid Collection (FB)

Definition at line 76 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

bool GoodSeedProducer::produceCkfseed_

private

Produce the Seed for Ckf tracks?

Definition at line 114 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

bool GoodSeedProducer::producePreId_

private

Produce the pre-id debugging collection.

Definition at line 120 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::propagatorName_

private

Definition at line 138 of file GoodSeedProducer.h.

float GoodSeedProducer::pt

private

Definition at line 152 of file GoodSeedProducer.h.

Referenced by Particle.Particle::__str__(), ZMuMuRochCorAnalyzer.DiMuon::__str__(), DiObject.DiMuon::__str__(), beginRun(), Photon.Photon::calScaledIsoValue(), Electron.Electron::mvaIDLoose(), Electron.Electron::mvaIDTight(), produce(), Lepton.Lepton::relEffAreaIso(), and Lepton.Lepton::relIso().

double GoodSeedProducer::PtThresholdSavePredId_

private

Threshold to save Pre Idinfo.

Definition at line 123 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::array<std::unique_ptr<TMVA::Reader>,9> GoodSeedProducer::reader {}

private

READER FOR TMVA.

Definition at line 149 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

std::map<reco::TrackRef,unsigned> GoodSeedProducer::refMap_

private

Map used to create the TrackRef, PreIdRef value map.

Definition at line 166 of file GoodSeedProducer.h.

Referenced by fillPreIdRefValueMap(), and produce().

std::unique_ptr<PFResolutionMap> GoodSeedProducer::resMapEtaECAL_

private

Definition at line 141 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

std::unique_ptr<PFResolutionMap> GoodSeedProducer::resMapPhiECAL_

private

Definition at line 142 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

std::unique_ptr<TrajectorySmoother> GoodSeedProducer::smoother_

private

Smoother.

Definition at line 82 of file GoodSeedProducer.h.

Referenced by produce().

std::string GoodSeedProducer::smootherName_

private

Definition at line 137 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

float GoodSeedProducer::thr[150]

private

vector of thresholds for different bins of eta and pt

Definition at line 126 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

std::string GoodSeedProducer::trackerRecHitBuilderName_

private

Definition at line 139 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

reco::TrackBase::TrackQuality GoodSeedProducer::trackQuality_

private

Definition at line 146 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::vector<edm::EDGetTokenT<reco::TrackCollection > > GoodSeedProducer::tracksContainers_

private

Definition at line 134 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::vector<edm::EDGetTokenT<std::vector<Trajectory> > > GoodSeedProducer::trajContainers_

private

Definition at line 133 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

float GoodSeedProducer::trk_ecalDeta

private

Definition at line 153 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::trk_ecalDphi

private

Definition at line 153 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

bool GoodSeedProducer::useQuality_

private

TRACK QUALITY.

Definition at line 145 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

bool GoodSeedProducer::useTmva_

private

USE OF TMVA.

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), GoodSeedProducer(), and produce().

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation