#include <GoodSeedProducer.h>

Inheritance diagram for GoodSeedProducer:

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

	~GoodSeedProducer ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

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
	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

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

virtual void	endRun (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_

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_

std::vector< reco::PFCluster const * >	ps1Clus
	Vector of clusters of the PreShower. More...

std::vector< reco::PFCluster const * >	ps2Clus

float	pt

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

TMVA::Reader *	reader [9]
	READER FOR TMVA. More...

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

PFResolutionMap *	resMapEtaECAL_

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...

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::EDProducer
typedef EDProducer	ModuleType

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

Static Public Member Functions inherited from edm::EDProducer
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 40 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, 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");
 }

GoodSeedProducer::~GoodSeedProducer ( )

Definition at line 128 of file GoodSeedProducer.cc.

References j, pfTransformer_, reader, resMapEtaECAL_, resMapPhiECAL_, and useTmva_.

 {
   
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.) 
 
   delete pfTransformer_;
   delete resMapEtaECAL_;
   delete resMapPhiECAL_;
   if(useTmva_) {
     for (UInt_t j = 0; j < 9; ++j){
       delete reader[j];
     }
   }
 }

Member Function Documentation

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

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 497 of file GoodSeedProducer.cc.

References B_, chikfred, chiRatio, chired, conf_, dpt, eP, eta, edm::FileInPath::fullPath(), edm::EventSetup::get(), edm::ParameterSet::getParameter(), j, method_, nhit, PFTrackTransformer::OnlyProp(), 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_= new PFTrackTransformer(B_);
   pfTransformer_->OnlyProp();
   
   //Resolution maps
   FileInPath ecalEtaMap(conf_.getParameter<string>("EtaMap"));
   FileInPath ecalPhiMap(conf_.getParameter<string>("PhiMap"));
   resMapEtaECAL_ = new PFResolutionMap("ECAL_eta",ecalEtaMap.fullPath().c_str());
   resMapPhiECAL_ = 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] = 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::endRun	(	const edm::Run &	run,
		const edm::EventSetup &
	)

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 558 of file GoodSeedProducer.cc.

References pfTransformer_, resMapEtaECAL_, and resMapPhiECAL_.

                                                              {
   delete pfTransformer_;
   pfTransformer_ = nullptr;
   delete resMapEtaECAL_;
   resMapEtaECAL_ = nullptr;
   delete resMapPhiECAL_;
   resMapPhiECAL_ = nullptr;
 }

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

private

Definition at line 584 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 568 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

FIXME FIXME CLONE

Implements edm::EDProducer.

Definition at line 151 of file GoodSeedProducer.cc.

References funct::abs(), B_, DeDxDiscriminatorTools::charge(), chikfred, chiRatio, chired, Trajectory::chiSquared(), clusThreshold_, constexpr, disablePreId_, dpt, HWWFunctions::EE, eP, eta, edm::EventID::event(), f, edm::helper::Filler< Map >::fill(), fillPreIdRefValueMap(), PFResolutionMap::FindBin(), 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(), 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(), AlCaHLTBitMon_QueryRunRegistry::string, thr, tmp, 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;
       try {
         std::string theTrackerRecHitBuilderName("WithAngleAndTemplate");  // FIXME FIXME
         iSetup.get<TransientRecHitRecord>().get(theTrackerRecHitBuilderName,theTrackerRecHitBuilder);
       } catch(...) {
         std::string theTrackerRecHitBuilderName("hltESPTTRHBWithTrackAngle");  // FIXME FIXME
         iSetup.get<TransientRecHitRecord>().get(theTrackerRecHitBuilderName,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 168 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::chikfred

private

Definition at line 158 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::chiRatio

private

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::chired

private

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

double GoodSeedProducer::clusThreshold_

private

Cut on the energy of the clusters.

Definition at line 113 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

edm::ParameterSet GoodSeedProducer::conf_

private

Definition at line 135 of file GoodSeedProducer.h.

Referenced by beginRun().

bool GoodSeedProducer::disablePreId_

private

switch to disable the pre-id

Definition at line 123 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

float GoodSeedProducer::dpt

private

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

double GoodSeedProducer::EcalStripSumE_deltaEta_

private

Definition at line 106 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::EcalStripSumE_deltaPhiOverQ_maxValue_

private

Definition at line 108 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::EcalStripSumE_deltaPhiOverQ_minValue_

private

Definition at line 107 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::EcalStripSumE_minClusEnergy_

private

Definition at line 105 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

float GoodSeedProducer::eP

private

VARIABLES NEEDED FOR TMVA.

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::eta

private

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

std::unique_ptr<TrajectoryFitter> GoodSeedProducer::fitter_

private

Fitter.

Definition at line 85 of file GoodSeedProducer.h.

Referenced by produce().

std::string GoodSeedProducer::fitterName_

private

Definition at line 142 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::HcalIsolWindow_

private

Definition at line 104 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

TkClonerImpl GoodSeedProducer::hitCloner

private

Definition at line 91 of file GoodSeedProducer.h.

Referenced by produce().

double GoodSeedProducer::maxEp_

private

Definition at line 117 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::maxEta_

private

Definition at line 102 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::maxHoverP_

private

Definition at line 110 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::maxPt_

private

Definition at line 101 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::method_

private

TMVA method.

Definition at line 165 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

double GoodSeedProducer::Min_dr_

private

Definition at line 159 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::minEoverP_

private

Definition at line 109 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

double GoodSeedProducer::minEp_

private

Min and MAx allowed values forEoverP.

Definition at line 116 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

double GoodSeedProducer::minPt_

private

Minimum transverse momentum and maximum pseudorapidity.

Definition at line 100 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

float GoodSeedProducer::nhit

private

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

int GoodSeedProducer::nHitsInSeed_

private

Number of hits in the seed;.

Definition at line 97 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer().

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

private

Definition at line 137 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

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

private

Definition at line 138 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

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

private

Definition at line 136 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

PFGeometry GoodSeedProducer::pfGeometry_

private

Definition at line 173 of file GoodSeedProducer.h.

PFTrackTransformer* GoodSeedProducer::pfTransformer_

private

PFTrackTransformer.

Definition at line 94 of file GoodSeedProducer.h.

Referenced by beginRun(), endRun(), and ~GoodSeedProducer().

std::string GoodSeedProducer::preidckf_

private

Name of the Seed(Ckf) Collection.

Definition at line 76 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::preidgsf_

private

Name of the Seed(Gsf) Collection.

Definition at line 79 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::preidname_

private

Name of the preid Collection (FB)

Definition at line 82 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

bool GoodSeedProducer::produceCkfseed_

private

Produce the Seed for Ckf tracks?

Definition at line 120 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

bool GoodSeedProducer::producePreId_

private

Produce the pre-id debugging collection.

Definition at line 126 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

std::string GoodSeedProducer::propagatorName_

private

Definition at line 144 of file GoodSeedProducer.h.

std::vector<reco::PFCluster const *> GoodSeedProducer::ps1Clus

private

Vector of clusters of the PreShower.

Definition at line 72 of file GoodSeedProducer.h.

std::vector<reco::PFCluster const *> GoodSeedProducer::ps2Clus

private

Definition at line 73 of file GoodSeedProducer.h.

float GoodSeedProducer::pt

private

Definition at line 157 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

double GoodSeedProducer::PtThresholdSavePredId_

private

Threshold to save Pre Idinfo.

Definition at line 129 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

TMVA::Reader* GoodSeedProducer::reader[9]

private

READER FOR TMVA.

Definition at line 154 of file GoodSeedProducer.h.

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

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

private

Map used to create the TrackRef, PreIdRef value map.

Definition at line 171 of file GoodSeedProducer.h.

Referenced by fillPreIdRefValueMap(), and produce().

PFResolutionMap* GoodSeedProducer::resMapEtaECAL_

private

Definition at line 146 of file GoodSeedProducer.h.

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

PFResolutionMap* GoodSeedProducer::resMapPhiECAL_

private

Definition at line 147 of file GoodSeedProducer.h.

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

std::unique_ptr<TrajectorySmoother> GoodSeedProducer::smoother_

private

Smoother.

Definition at line 88 of file GoodSeedProducer.h.

Referenced by produce().

std::string GoodSeedProducer::smootherName_

private

Definition at line 143 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 132 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

reco::TrackBase::TrackQuality GoodSeedProducer::trackQuality_

private

Definition at line 151 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

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

private

Definition at line 140 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

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

private

Definition at line 139 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

float GoodSeedProducer::trk_ecalDeta

private

Definition at line 158 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

float GoodSeedProducer::trk_ecalDphi

private

Definition at line 158 of file GoodSeedProducer.h.

Referenced by beginRun(), and produce().

bool GoodSeedProducer::useQuality_

private

TRACK QUALITY.

Definition at line 150 of file GoodSeedProducer.h.

Referenced by GoodSeedProducer(), and produce().

bool GoodSeedProducer::useTmva_

private

USE OF TMVA.

Definition at line 162 of file GoodSeedProducer.h.

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

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