GoodSeedProducer Class Reference

Inheritance diagram for GoodSeedProducer:

Public Member Functions
	GoodSeedProducer (const edm::ParameterSet &, const goodseedhelpers::HeavyObjectCache *)
Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< goodseedhelpers::HeavyObjectCache > >
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	globalEndJob (goodseedhelpers::HeavyObjectCache const *)
static std::unique_ptr< goodseedhelpers::HeavyObjectCache >	initializeGlobalCache (const edm::ParameterSet &conf)

Private Types
typedef TrajectoryStateOnSurface	TSOS

Private Member Functions
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...
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_
std::string	fitterName_
const edm::ESGetToken< TrajectoryFitter, TrajectoryFitter::Record >	fitterToken_
double	HcalIsolWindow_
TkClonerImpl	hitCloner
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	magneticFieldToken_
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	magneticFieldTokenBeginRun_
double	maxEp_
double	maxEta_
double	maxHoverP_
double	maxPt_
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
	VARIABLES NEEDED FOR TMVA. More...
int	nHitsInSeed_
	Number of hits in the seed;. More...
edm::EDGetTokenT< reco::PFClusterCollection >	pfCLusTagECLabel_
edm::EDGetTokenT< reco::PFClusterCollection >	pfCLusTagHCLabel_
edm::EDGetTokenT< reco::PFClusterCollection >	pfCLusTagPSLabel_
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_
double	PtThresholdSavePredId_
	Threshold to save Pre Idinfo. 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::string	smootherName_
const edm::ESGetToken< TrajectorySmoother, TrajectoryFitter::Record >	smootherToken_
float	thr [150]
	vector of thresholds for different bins of eta and pt More...
std::string	trackerRecHitBuilderName_
const edm::ESGetToken< TransientTrackingRecHitBuilder, TransientRecHitRecord >	trackerRecHitBuilderToken_
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< edm::GlobalCache< goodseedhelpers::HeavyObjectCache > >
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 68 of file GoodSeedProducer.cc.

Member Typedef Documentation

TSOS

typedef TrajectoryStateOnSurface GoodSeedProducer::TSOS

private

Definition at line 69 of file GoodSeedProducer.cc.

Constructor & Destructor Documentation

GoodSeedProducer()

GoodSeedProducer::GoodSeedProducer ( const edm::ParameterSet &  iConfig, const goodseedhelpers::HeavyObjectCache *    )

explicit

Definition at line 190 of file GoodSeedProducer.cc.

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

    : pfTransformer_(nullptr),
      conf_(iConfig),
      resMapEtaECAL_(nullptr),
      resMapPhiECAL_(nullptr),
      fitterToken_(esConsumes(edm::ESInputTag("", iConfig.getParameter<string>("Fitter")))),
      smootherToken_(esConsumes(edm::ESInputTag("", iConfig.getParameter<string>("Smoother")))),
      trackerRecHitBuilderToken_(esConsumes(edm::ESInputTag("", iConfig.getParameter<std::string>("TTRHBuilder")))),
      magneticFieldToken_(esConsumes()),
      magneticFieldTokenBeginRun_(esConsumes<edm::Transition::BeginRun>()) {
  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

beginRun()

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

overrideprivate

Definition at line 616 of file GoodSeedProducer.cc.

References B_, conf_, edm::FileInPath::fullPath(), edm::EventSetup::getData(), edm::ParameterSet::getParameter(), HLT_2022v12_cff::magneticField, magneticFieldTokenBeginRun_, pfTransformer_, resMapEtaECAL_, resMapPhiECAL_, and thr.

                                                                       {
  //Magnetic Field
  auto const& magneticField = &es.getData(magneticFieldTokenBeginRun_);
  B_ = magneticField->inTesla(GlobalPoint(0, 0, 0));

  pfTransformer_ = std::make_unique<PFTrackTransformer>(B_);
  pfTransformer_->OnlyProp();

  //Resolution maps
  FileInPath ecalEtaMap(conf_.getParameter<string>("EtaMap"));
  FileInPath ecalPhiMap(conf_.getParameter<string>("PhiMap"));
  resMapEtaECAL_ = std::make_unique<PFResolutionMap>("ECAL_eta", ecalEtaMap.fullPath().c_str());
  resMapPhiECAL_ = std::make_unique<PFResolutionMap>("ECAL_phi", ecalPhiMap.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];
}

fillPreIdRefValueMap()

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

private

Definition at line 661 of file GoodSeedProducer.cc.

References trigObjTnPSource_cfi::filler, ntracks, refMap_, tracks, and contentValuesCheck::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());
}

getBin()

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

private

Find the bin in pt and eta.

Definition at line 637 of file GoodSeedProducer.cc.

References PVValHelper::eta, LogDebug, and DiDispStaMuonMonitor_cfi::pt.

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

globalEndJob()

static void GoodSeedProducer::globalEndJob ( goodseedhelpers::HeavyObjectCache const *  )

inlinestatic

Definition at line 78 of file GoodSeedProducer.cc.

{}

initializeGlobalCache()

static std::unique_ptr<goodseedhelpers::HeavyObjectCache> GoodSeedProducer::initializeGlobalCache ( const edm::ParameterSet &  conf )

inlinestatic

Definition at line 74 of file GoodSeedProducer.cc.

                                                                                                           {
    return std::make_unique<goodseedhelpers::HeavyObjectCache>(conf);
  }

produce()

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

overrideprivate

Definition at line 286 of file GoodSeedProducer.cc.

References funct::abs(), B_, ALCARECOTkAlJpsiMuMu_cff::charge, chikfred, chiRatio, chired, Trajectory::chiSquared(), clusThreshold_, runTheMatrix::const, disablePreId_, dpt, flavorHistoryFilter_cfi::dr, EE, PVValHelper::eta, Basic3DVector< T >::eta(), f, JetMETHLTOfflineSource_cfi::feta, fillPreIdRefValueMap(), Trajectory::firstMeasurement(), fitterToken_, dqmMemoryStats::float, getBin(), edm::EventSetup::getData(), TrajectoryStateOnSurface::globalMomentum(), runTauDisplay::gp, hitCloner, mps_fire::i, iEvent, createfilelist::int, Trajectory::isValid(), Trajectory::lastMeasurement(), LogDebug, mag2(), HLT_2022v12_cff::magneticField, magneticFieldToken_, maxEp_, maxEta_, maxPt_, Min_dr_, minEp_, minPt_, eostools::move(), nhit, PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::perp2(), pfCLusTagECLabel_, pfCLusTagHCLabel_, pfCLusTagPSLabel_, Pi, funct::pow(), preidckf_, preidgsf_, preidname_, produceCkfseed_, producePreId_, edm::Handle< T >::product(), BaseParticlePropagator::propagateToEcalEntrance(), DiDispStaMuonMonitor_cfi::pt, reco::PreId::pt(), PtThresholdSavePredId_, quality, FastTrackerRecHitMaskProducer_cfi::recHits, refMap_, resMapEtaECAL_, resMapPhiECAL_, reco::ElectronSeed::setCtfTrack(), reco::PreId::setECALMatching(), reco::PreId::setECALMatchingProperties(), reco::PreId::setFinalDecision(), reco::PreId::setMVA(), reco::PreId::setTrack(), reco::PreId::setTrackFiltering(), reco::PreId::setTrackProperties(), smootherToken_, mathSSE::sqrt(), thr, createJobs::tmp, trackerRecHitBuilderToken_, trackQuality_, tracksContainers_, trk_ecalDeta, trk_ecalDphi, TwoPi, unit(), TrajectoryMeasurement::updatedState(), useQuality_, useTmva_, x, PV3DBase< T, PVType, FrameType >::x(), y, PV3DBase< T, PVType, FrameType >::y(), z, and PV3DBase< T, PVType, FrameType >::z().

                                                                      {
  LogDebug("GoodSeedProducer") << "START event: " << iEvent.id().event() << " in run " << iEvent.id().run();
  //Create empty output collections
  auto output_preid = std::make_unique<ElectronSeedCollection>();
  auto output_nopre = std::make_unique<TrajectorySeedCollection>();
  auto output_preidinfo = std::make_unique<PreIdCollection>();
  auto preIdMap_p = std::make_unique<edm::ValueMap<reco::PreIdRef>>();
  edm::ValueMap<reco::PreIdRef>::Filler mapFiller(*preIdMap_p);

  std::unique_ptr<TrajectoryFitter> fitter;
  std::unique_ptr<TrajectorySmoother> smoother;

  //Tracking Tools
  if (!disablePreId_) {
    auto const& aFitter = &iSetup.getData(fitterToken_);
    auto const& aSmoother = &iSetup.getData(smootherToken_);

    smoother.reset(aSmoother->clone());
    fitter = aFitter->clone();
    auto const& theTrackerRecHitBuilder = &iSetup.getData(trackerRecHitBuilderToken_);
    hitCloner = static_cast<TkTransientTrackingRecHitBuilder const*>(theTrackerRecHitBuilder)->cloner();
    fitter->setHitCloner(&hitCloner);
    smoother->setHitCloner(&hitCloner);
  }

  // clear temporary maps
  refMap_.clear();

  //Magnetic Field
  auto const& magneticField = &iSetup.getData(magneticFieldToken_);

  //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());

    LogDebug("GoodSeedProducer") << "Number of tracks in collection "
                                 << "tracksContainers_[" << istr << "] to be analyzed " << Tk.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);
      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;

        // FIXME the original code was buggy should be outerMomentum...
        float oPTOB = 1.f / std::sqrt(Tk[i].innerMomentum().mag2());
        //  float chikfred=Tk[i].normalizedChi2();
        float nchi = Tk[i].normalizedChi2();

        int nhitpi = Tk[i].found();
        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, Tk[i].charge()), 0, 0, B_.z());

        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.particle().vertex().x(),
                                       theOutParticle.particle().vertex().y(),
                                       theOutParticle.particle().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.particle().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;
          for (auto const& hit : Tk[i].recHits())
            tmp.push_back(hit->cloneSH());
          auto const& theTrack = Tk[i];
          GlobalVector gv(theTrack.innerMomentum().x(), theTrack.innerMomentum().y(), theTrack.innerMomentum().z());
          GlobalPoint gp(theTrack.innerPosition().x(), theTrack.innerPosition().y(), theTrack.innerPosition().z());
          GlobalTrajectoryParameters gtps(gp, gv, theTrack.charge(), &*magneticField);
          TrajectoryStateOnSurface tsos(gtps, theTrack.innerStateCovariance(), *tmp[0]->surface());
          Trajectory&& FitTjs = fitter->fitOne(Seed, tmp, tsos);

          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() / Tk[i].chi2();
              chired = chiRatio * chikfred;
            }
          }

          //TMVA Analysis
          if (useTmva_) {
            float vars[10] = {nhit, chikfred, dpt, EP, chiRatio, chired, trk_ecalDeta, trk_ecalDphi, tkpt, tketa};

            float Ytmva = globalCache()->gbr[ipteta]->GetClassifier(vars);

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

#ifdef EDM_ML_DEBUG
        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";
#endif

      }  // 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(std::move(output_preid), preidgsf_);
  if (produceCkfseed_)
    iEvent.put(std::move(output_nopre), preidckf_);
  if (producePreId_) {
    const edm::OrphanHandle<reco::PreIdCollection> preIdRefProd = iEvent.put(std::move(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(std::move(preIdMap_p), preidname_);
    }
  }
  // clear temporary maps
  refMap_.clear();
}

Member Data Documentation

B_

math::XYZVector GoodSeedProducer::B_

private

B field.

Definition at line 180 of file GoodSeedProducer.cc.

Referenced by beginRun(), and produce().

chikfred

float GoodSeedProducer::chikfred

private

Definition at line 173 of file GoodSeedProducer.cc.

Referenced by produce().

chiRatio

float GoodSeedProducer::chiRatio

private

Definition at line 172 of file GoodSeedProducer.cc.

Referenced by produce().

chired

float GoodSeedProducer::chired

private

Definition at line 172 of file GoodSeedProducer.cc.

Referenced by produce().

clusThreshold_

double GoodSeedProducer::clusThreshold_

private

Cut on the energy of the clusters.

Definition at line 124 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

conf_

edm::ParameterSet GoodSeedProducer::conf_

private

Definition at line 146 of file GoodSeedProducer.cc.

Referenced by beginRun().

disablePreId_

bool GoodSeedProducer::disablePreId_

private

switch to disable the pre-id

Definition at line 134 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

dpt

float GoodSeedProducer::dpt

private

Definition at line 172 of file GoodSeedProducer.cc.

Referenced by produce().

EcalStripSumE_deltaEta_

double GoodSeedProducer::EcalStripSumE_deltaEta_

private

Definition at line 117 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

EcalStripSumE_deltaPhiOverQ_maxValue_

double GoodSeedProducer::EcalStripSumE_deltaPhiOverQ_maxValue_

private

Definition at line 119 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

EcalStripSumE_deltaPhiOverQ_minValue_

double GoodSeedProducer::EcalStripSumE_deltaPhiOverQ_minValue_

private

Definition at line 118 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

EcalStripSumE_minClusEnergy_

double GoodSeedProducer::EcalStripSumE_minClusEnergy_

private

Definition at line 116 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

fitterName_

std::string GoodSeedProducer::fitterName_

private

Definition at line 153 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

fitterToken_

const edm::ESGetToken<TrajectoryFitter, TrajectoryFitter::Record> GoodSeedProducer::fitterToken_

private

Definition at line 161 of file GoodSeedProducer.cc.

Referenced by produce().

HcalIsolWindow_

double GoodSeedProducer::HcalIsolWindow_

private

Definition at line 115 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

hitCloner

TkClonerImpl GoodSeedProducer::hitCloner

private

Definition at line 102 of file GoodSeedProducer.cc.

Referenced by produce().

magneticFieldToken_

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> GoodSeedProducer::magneticFieldToken_

private

Definition at line 164 of file GoodSeedProducer.cc.

Referenced by produce().

magneticFieldTokenBeginRun_

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> GoodSeedProducer::magneticFieldTokenBeginRun_

private

Definition at line 165 of file GoodSeedProducer.cc.

Referenced by beginRun().

maxEp_

double GoodSeedProducer::maxEp_

private

Definition at line 128 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

maxEta_

double GoodSeedProducer::maxEta_

private

Definition at line 113 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

maxHoverP_

double GoodSeedProducer::maxHoverP_

private

Definition at line 121 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

maxPt_

double GoodSeedProducer::maxPt_

private

Definition at line 112 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

Min_dr_

double GoodSeedProducer::Min_dr_

private

Definition at line 174 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

minEoverP_

double GoodSeedProducer::minEoverP_

private

Definition at line 120 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

minEp_

double GoodSeedProducer::minEp_

private

Min and MAx allowed values forEoverP.

Definition at line 127 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

minPt_

double GoodSeedProducer::minPt_

private

Minimum transverse momentum and maximum pseudorapidity.

Definition at line 111 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

nhit

float GoodSeedProducer::nhit

private

VARIABLES NEEDED FOR TMVA.

Definition at line 172 of file GoodSeedProducer.cc.

Referenced by produce().

nHitsInSeed_

int GoodSeedProducer::nHitsInSeed_

private

Number of hits in the seed;.

Definition at line 108 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

pfCLusTagECLabel_

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

private

Definition at line 148 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

pfCLusTagHCLabel_

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

private

Definition at line 149 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

pfCLusTagPSLabel_

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

private

Definition at line 147 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

pfTransformer_

std::unique_ptr<PFTrackTransformer> GoodSeedProducer::pfTransformer_

private

PFTrackTransformer.

Definition at line 105 of file GoodSeedProducer.cc.

Referenced by beginRun().

preidckf_

std::string GoodSeedProducer::preidckf_

private

Name of the Seed(Ckf) Collection.

Definition at line 93 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

preidgsf_

std::string GoodSeedProducer::preidgsf_

private

Name of the Seed(Gsf) Collection.

Definition at line 96 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

preidname_

std::string GoodSeedProducer::preidname_

private

Name of the preid Collection (FB)

Definition at line 99 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

produceCkfseed_

bool GoodSeedProducer::produceCkfseed_

private

Produce the Seed for Ckf tracks?

Definition at line 131 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

producePreId_

bool GoodSeedProducer::producePreId_

private

Produce the pre-id debugging collection.

Definition at line 137 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

propagatorName_

std::string GoodSeedProducer::propagatorName_

private

Definition at line 155 of file GoodSeedProducer.cc.

PtThresholdSavePredId_

double GoodSeedProducer::PtThresholdSavePredId_

private

Threshold to save Pre Idinfo.

Definition at line 140 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

refMap_

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

private

Map used to create the TrackRef, PreIdRef value map.

Definition at line 183 of file GoodSeedProducer.cc.

Referenced by fillPreIdRefValueMap(), and produce().

resMapEtaECAL_

std::unique_ptr<PFResolutionMap> GoodSeedProducer::resMapEtaECAL_

private

Definition at line 158 of file GoodSeedProducer.cc.

Referenced by beginRun(), and produce().

resMapPhiECAL_

std::unique_ptr<PFResolutionMap> GoodSeedProducer::resMapPhiECAL_

private

Definition at line 159 of file GoodSeedProducer.cc.

Referenced by beginRun(), and produce().

smootherName_

std::string GoodSeedProducer::smootherName_

private

Definition at line 154 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

smootherToken_

const edm::ESGetToken<TrajectorySmoother, TrajectoryFitter::Record> GoodSeedProducer::smootherToken_

private

Definition at line 162 of file GoodSeedProducer.cc.

Referenced by produce().

thr

float GoodSeedProducer::thr[150]

private

vector of thresholds for different bins of eta and pt

Definition at line 143 of file GoodSeedProducer.cc.

Referenced by beginRun(), and produce().

trackerRecHitBuilderName_

std::string GoodSeedProducer::trackerRecHitBuilderName_

private

Definition at line 156 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

trackerRecHitBuilderToken_

const edm::ESGetToken<TransientTrackingRecHitBuilder, TransientRecHitRecord> GoodSeedProducer::trackerRecHitBuilderToken_

private

Definition at line 163 of file GoodSeedProducer.cc.

Referenced by produce().

trackQuality_

reco::TrackBase::TrackQuality GoodSeedProducer::trackQuality_

private

Definition at line 169 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

tracksContainers_

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

private

Definition at line 151 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

trajContainers_

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

private

Definition at line 150 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer().

trk_ecalDeta

float GoodSeedProducer::trk_ecalDeta

private

Definition at line 173 of file GoodSeedProducer.cc.

Referenced by produce().

trk_ecalDphi

float GoodSeedProducer::trk_ecalDphi

private

Definition at line 173 of file GoodSeedProducer.cc.

Referenced by produce().

useQuality_

bool GoodSeedProducer::useQuality_

private

TRACK QUALITY.

Definition at line 168 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().

useTmva_

bool GoodSeedProducer::useTmva_

private

USE OF TMVA.

Definition at line 177 of file GoodSeedProducer.cc.

Referenced by GoodSeedProducer(), and produce().