#include <MultiHitGeneratorFromChi2.h>

Inheritance diagram for MultiHitGeneratorFromChi2:

Public Member Functions
void	hitSets (const TrackingRegion &region, OrderedMultiHits &trs, const edm::Event &ev, const edm::EventSetup &es, SeedingLayerSetsHits::SeedingLayerSet pairLayers, std::vector< SeedingLayerSetsHits::SeedingLayer > thirdLayers) override

void	hitSets (const TrackingRegion &region, OrderedMultiHits &trs, const HitDoublets &doublets, const std::vector< SeedingLayerSetsHits::SeedingLayer > &thirdLayers, LayerCacheType &layerCache, cacheHits &refittedHitStorage)

void	hitSets (const TrackingRegion &region, OrderedMultiHits &result, const HitDoublets &doublets, const RecHitsSortedInPhi *thirdHitMap, const std::vector< const DetLayer > &thirdLayerDetLayer, const int nThirdLayers, cacheHits &refittedHitStorage)

void	hitTriplets (const TrackingRegion &region, OrderedMultiHits &result, const HitDoublets &doublets, const RecHitsSortedInPhi *thirdHitMap, const std::vector< const DetLayer > &thirdLayerDetLayer, const int nThirdLayers) override

void	initES (const edm::EventSetup &es) override

	MultiHitGeneratorFromChi2 (const edm::ParameterSet &cfg, edm::ConsumesCollector &&iC)

	MultiHitGeneratorFromChi2 (const edm::ParameterSet &cfg, edm::ConsumesCollector &)

	~MultiHitGeneratorFromChi2 () override

Public Member Functions inherited from MultiHitGeneratorFromPairAndLayers
void	clear ()

void	init (std::unique_ptr< HitPairGeneratorFromLayerPair > &&pairGenerator, LayerCacheType *layerCache)

	MultiHitGeneratorFromPairAndLayers (const edm::ParameterSet &pset)

const HitPairGeneratorFromLayerPair &	pairGenerator () const

virtual	~MultiHitGeneratorFromPairAndLayers ()

Static Public Member Functions
static void	fillDescriptions (edm::ParameterSetDescription &desc)

static const char *	fillDescriptionsLabel ()

Static Public Member Functions inherited from MultiHitGeneratorFromPairAndLayers
static void	fillDescriptions (edm::ParameterSetDescription &desc)

Private Types
using	HitOwnPtr = mayown_ptr< BaseTrackerRecHit >

typedef CombinedMultiHitGenerator::LayerCacheType	LayerCacheType

Private Member Functions
void	refit2Hits (HitOwnPtr &hit0, HitOwnPtr &hit1, TrajectoryStateOnSurface &tsos0, TrajectoryStateOnSurface &tsos1, const TrackingRegion &region, float nomField, bool isDebug)

Private Attributes
const MagneticField *	bfield

TkTransientTrackingRecHitBuilder const *	builder

std::string	builderName_

std::vector< double >	chi2_cuts

bool	chi2VsPtCut

TkClonerImpl	cloner

edm::ESGetToken < ClusterShapeHitFilter, CkfComponentsRecord >	clusterShapeHitFilterESToken_

std::vector< int >	detIdsToDebug

float	dphi

float	extraHitRPhitolerance

float	extraHitRZtolerance

float	extraPhiKDBox

float	extraRKDBox

float	extraZKDBox

const ClusterShapeHitFilter *	filter

std::string	filterName_

double	fnSigmaRZ

edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	magneticFieldESToken_

double	maxChi2

std::string	mfName_

float	nomField

double	nSigmaPhi

double	nSigmaRZ

std::vector< double >	pt_interv

bool	refitHits

edm::ESGetToken < TransientTrackingRecHitBuilder, TransientRecHitRecord >	transientTrackingRecHitBuilderESToken_

UniformMagneticField	ufield = 0.

bool	useFixedPreFiltering

bool	useSimpleMF_

Additional Inherited Members
Public Types inherited from MultiHitGeneratorFromPairAndLayers
typedef LayerHitMapCache	LayerCacheType

Protected Types inherited from MultiHitGeneratorFromPairAndLayers
using	cacheHitPointer = std::unique_ptr< BaseTrackerRecHit >

using	cacheHits = std::vector< cacheHitPointer >

Protected Attributes inherited from MultiHitGeneratorFromPairAndLayers
cacheHits	cache

LayerCacheType *	theLayerCache

const unsigned int	theMaxElement

std::unique_ptr < HitPairGeneratorFromLayerPair >	thePairGenerator

Detailed Description

Definition at line 36 of file MultiHitGeneratorFromChi2.h.

Member Typedef Documentation

using MultiHitGeneratorFromChi2::HitOwnPtr = mayown_ptr<BaseTrackerRecHit>

private

Definition at line 81 of file MultiHitGeneratorFromChi2.h.

typedef CombinedMultiHitGenerator::LayerCacheType MultiHitGeneratorFromChi2::LayerCacheType

private

Definition at line 37 of file MultiHitGeneratorFromChi2.h.

Constructor & Destructor Documentation

MultiHitGeneratorFromChi2::MultiHitGeneratorFromChi2	(	const edm::ParameterSet &	cfg,
		edm::ConsumesCollector &&	iC
	)

inline

Definition at line 40 of file MultiHitGeneratorFromChi2.h.

41 : MultiHitGeneratorFromChi2(cfg, iC) {}

MultiHitGeneratorFromChi2::MultiHitGeneratorFromChi2

MultiHitGeneratorFromChi2(const edm::ParameterSet &cfg, edm::ConsumesCollector &&iC)

Definition: MultiHitGeneratorFromChi2.h:40

MultiHitGeneratorFromChi2::MultiHitGeneratorFromChi2	(	const edm::ParameterSet &	cfg,
		edm::ConsumesCollector &	iC
	)

Definition at line 49 of file MultiHitGeneratorFromChi2.cc.

References bfield, builderName_, chi2_cuts, chi2VsPtCut, clusterShapeHitFilterESToken_, detIdsToDebug, dphi, edm::ConsumesCollector::esConsumes(), edm::ParameterSet::exists(), filter, filterName_, edm::ParameterSet::getParameter(), magneticFieldESToken_, mfName_, nomField, pt_interv, refitHits, AlCaHLTBitMon_QueryRunRegistry::string, transientTrackingRecHitBuilderESToken_, useFixedPreFiltering, and useSimpleMF_.

     : MultiHitGeneratorFromPairAndLayers(cfg),
       useFixedPreFiltering(cfg.getParameter<bool>("useFixedPreFiltering")),
       extraHitRZtolerance(
           cfg.getParameter<double>("extraHitRZtolerance")),  //extra window in ThirdHitRZPrediction range
       extraHitRPhitolerance(cfg.getParameter<double>(
           "extraHitRPhitolerance")),  //extra window in ThirdHitPredictionFromCircle range (divide by R to get phi)
       extraZKDBox(
           cfg.getParameter<double>("extraZKDBox")),  //extra windown in Z when building the KDTree box (used in barrel)
       extraRKDBox(
           cfg.getParameter<double>("extraRKDBox")),  //extra windown in R when building the KDTree box (used in endcap)
       extraPhiKDBox(cfg.getParameter<double>("extraPhiKDBox")),  //extra windown in Phi when building the KDTree box
       fnSigmaRZ(cfg.getParameter<double>(
           "fnSigmaRZ")),  //this multiplies the max hit error on the layer when building the KDTree box
       chi2VsPtCut(cfg.getParameter<bool>("chi2VsPtCut")),
       maxChi2(cfg.getParameter<double>("maxChi2")),
       refitHits(cfg.getParameter<bool>("refitHits")),
       filterName_(cfg.getParameter<std::string>("ClusterShapeHitFilterName")),
       builderName_(cfg.existsAs<std::string>("TTRHBuilder") ? cfg.getParameter<std::string>("TTRHBuilder")
                                                             : std::string("WithTrackAngle")),
       useSimpleMF_(false),
       mfName_("") {
   if (useFixedPreFiltering)
     dphi = cfg.getParameter<double>("phiPreFiltering");
   if (chi2VsPtCut) {
     pt_interv = cfg.getParameter<std::vector<double> >("pt_interv");
     chi2_cuts = cfg.getParameter<std::vector<double> >("chi2_cuts");
   }
 #ifdef EDM_ML_DEBUG
   detIdsToDebug = cfg.getParameter<std::vector<int> >("detIdsToDebug");
   //if (detIdsToDebug.size()<3) //fixme
 #else
   detIdsToDebug.push_back(0);
   detIdsToDebug.push_back(0);
   detIdsToDebug.push_back(0);
 #endif
   // 2014/02/11 mia:
   // we should get rid of the boolean parameter useSimpleMF,
   // and use only a string magneticField [instead of SimpleMagneticField]
   // or better an edm::ESInputTag (at the moment HLT does not handle ESInputTag)
   if (cfg.exists("SimpleMagneticField")) {
     useSimpleMF_ = true;
     mfName_ = cfg.getParameter<std::string>("SimpleMagneticField");
   }
   filter = nullptr;
   bfield = nullptr;
   nomField = -1.;
 
   if (useSimpleMF_) {
     magneticFieldESToken_ = iC.esConsumes(edm::ESInputTag("", mfName_));
   } else {
     magneticFieldESToken_ = iC.esConsumes();
   }
   if (refitHits) {
     clusterShapeHitFilterESToken_ = iC.esConsumes(edm::ESInputTag("", filterName_));
     transientTrackingRecHitBuilderESToken_ = iC.esConsumes(edm::ESInputTag("", builderName_));
   }
 }

MultiHitGeneratorFromChi2::~MultiHitGeneratorFromChi2 ( )

override

Definition at line 108 of file MultiHitGeneratorFromChi2.cc.

108 {}

Member Function Documentation

void MultiHitGeneratorFromChi2::fillDescriptions ( edm::ParameterSetDescription & desc )

static

Definition at line 110 of file MultiHitGeneratorFromChi2.cc.

References edm::ParameterSetDescription::add(), MultiHitGeneratorFromPairAndLayers::fillDescriptions(), and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by MultiHitFromChi2EDProducer::fillDescriptions().

                                                                                  {
   MultiHitGeneratorFromPairAndLayers::fillDescriptions(desc);
 
   // fixed phi filtering
   desc.add<bool>("useFixedPreFiltering", false);
   desc.add<double>("phiPreFiltering", 0.3);
 
   // box properties
   desc.add<double>("extraHitRPhitolerance", 0);
   desc.add<double>("extraHitRZtolerance", 0);
   desc.add<double>("extraZKDBox", 0.2);
   desc.add<double>("extraRKDBox", 0.2);
   desc.add<double>("extraPhiKDBox", 0.005);
   desc.add<double>("fnSigmaRZ", 2.0);
 
   // refit&filter hits
   desc.add<bool>("refitHits", true);
   desc.add<std::string>("ClusterShapeHitFilterName", "ClusterShapeHitFilter");
   desc.add<std::string>("TTRHBuilder", "WithTrackAngle");
 
   // chi2 cuts
   desc.add<double>("maxChi2", 5.0);
   desc.add<bool>("chi2VsPtCut", true);
   desc.add<std::vector<double> >("pt_interv", std::vector<double>{{0.4, 0.7, 1.0, 2.0}});
   desc.add<std::vector<double> >("chi2_cuts", std::vector<double>{{3.0, 4.0, 5.0, 5.0}});
 
   // debugging
   desc.add<std::vector<int> >("detIdsToDebug", std::vector<int>{{0, 0, 0}});
 }

static const char* MultiHitGeneratorFromChi2::fillDescriptionsLabel ( )

inlinestatic

Definition at line 47 of file MultiHitGeneratorFromChi2.h.

Referenced by MultiHitFromChi2EDProducer::fillDescriptions().

47 { return "multiHitFromChi2"; }

void MultiHitGeneratorFromChi2::hitSets	(	const TrackingRegion &	region,
		OrderedMultiHits &	trs,
		const edm::Event &	ev,
		const edm::EventSetup &	es,
		SeedingLayerSetsHits::SeedingLayerSet	pairLayers,
		std::vector< SeedingLayerSetsHits::SeedingLayer >	thirdLayers
	)

overridevirtual

Implements MultiHitGeneratorFromPairAndLayers.

Definition at line 165 of file MultiHitGeneratorFromChi2.cc.

References cms::cuda::assert(), MultiHitGeneratorFromPairAndLayers::cache, HLT_FULL_cff::doublets, LogDebug, LogTrace, MultiHitGeneratorFromPairAndLayers::theLayerCache, and MultiHitGeneratorFromPairAndLayers::thePairGenerator.

Referenced by hitSets(), hitTriplets(), and MultiHitFromChi2EDProducer::produce().

                                                                                                  {
   LogDebug("MultiHitGeneratorFromChi2") << "pair: " << thePairGenerator->innerLayer(pairLayers).name() << "+"
                                         << thePairGenerator->outerLayer(pairLayers).name()
                                         << " 3rd lay size: " << thirdLayers.size();
 
   auto const& doublets = thePairGenerator->doublets(region, ev, es, pairLayers);
   LogTrace("MultiHitGeneratorFromChi2") << "";
   if (doublets.empty()) {
     //  LogDebug("MultiHitGeneratorFromChi2") << "empy pairs";
     return;
   }
 
   assert(theLayerCache);
   hitSets(region, result, doublets, thirdLayers, *theLayerCache, cache);
 }

void MultiHitGeneratorFromChi2::hitSets	(	const TrackingRegion &	region,
		OrderedMultiHits &	trs,
		const HitDoublets &	doublets,
		const std::vector< SeedingLayerSetsHits::SeedingLayer > &	thirdLayers,
		LayerCacheType &	layerCache,
		cacheHits &	refittedHitStorage
	)

Definition at line 186 of file MultiHitGeneratorFromChi2.cc.

References hitSets(), and findQualityFiles::size.

                                                                        {
   int size = thirdLayers.size();
   const RecHitsSortedInPhi* thirdHitMap[size];
   vector<const DetLayer*> thirdLayerDetLayer(size, nullptr);
   for (int il = 0; il < size; ++il) {
     thirdHitMap[il] = &layerCache(thirdLayers[il], region);
 
     thirdLayerDetLayer[il] = thirdLayers[il].detLayer();
   }
   hitSets(region, result, doublets, thirdHitMap, thirdLayerDetLayer, size, refittedHitStorage);
 }

void MultiHitGeneratorFromChi2::hitSets	(	const TrackingRegion &	region,
		OrderedMultiHits &	result,
		const HitDoublets &	doublets,
		const RecHitsSortedInPhi **	thirdHitMap,
		const std::vector< const DetLayer * > &	thirdLayerDetLayer,
		const int	nThirdLayers,
		cacheHits &	refittedHitStorage
	)

Definition at line 212 of file MultiHitGeneratorFromChi2.cc.

References RecHitsSortedInPhi::all(), angle(), cms::cuda::assert(), bfield, TrajectoryStateOnSurface::charge(), RZLine::chi2(), HLT_FULL_cff::chi2, chi2_cuts, chi2VsPtCut, cloner, runTheMatrix::const, PixelRecoUtilities::curvature(), ThirdHitPredictionFromCircle::curvature(), declareDynArray, detIdsToDebug, DeadROC_duringRun::dir, dphi, relativeConstraints::empty, mayown_ptr< T, N >::empty(), PV3DBase< T, PVType, FrameType >::eta(), extraHitRPhitolerance, extraHitRZtolerance, extraPhiKDBox, extraRKDBox, extraZKDBox, validate-o2o-wbm::f, filter, fnSigmaRZ, TrajectoryStateOnSurface::globalMomentum(), HitDoublets::gp(), runTauDisplay::gp, hh, HitDoublets::hit(), mps_fire::i, IfLogTrace, HitDoublets::inner, HitDoublets::innerLayer(), RecHitsSortedInPhi::isBarrel, ClusterShapeHitFilter::isCompatible(), edm::isNotFinite(), mayown_ptr< T, N >::isOwn(), GetRecoTauVFromDQM_MC_cff::kk, phase1PixelTopology::layer, geometryCSVtoXML::line, LogTrace, M_PI, SiStripPI::max, maxChi2, gpuPixelDoublets::maxz, gpuPixelDoublets::minz, SiStripMatchedRecHit2D::monoCluster(), SiStripMatchedRecHit2D::monoId(), eostools::move(), nomField, normalizedPhi(), ProjectedSiStripRecHit2D::originalHit(), TrackingRegion::originRBound(), HitDoublets::outer, HitDoublets::outerLayer(), PV3DBase< T, PVType, FrameType >::perp(), ThirdHitPredictionFromCircle::phi(), PV3DBase< T, PVType, FrameType >::phi(), GeometricSearchDet::position(), proxim(), DiDispStaMuonMonitor_cfi::pt, pt_interv, TrackingRegion::ptMin(), CosmicsPD_Skims::radius, refit2Hits(), refitHits, mayown_ptr< T, N >::release(), mayown_ptr< T, N >::reset(), rho, FastCircle::rho(), OrderedMultiHits::size(), HitDoublets::size(), SiStripMatchedRecHit2D::stereoCluster(), SiStripMatchedRecHit2D::stereoId(), MultiHitGeneratorFromPairAndLayers::theMaxElement, SiStripDetId::TIB, SiStripDetId::TID, Geom::twoPi(), ufield, useFixedPreFiltering, and PV3DBase< T, PVType, FrameType >::z().

                                                                        {
 #ifdef EDM_ML_DEBUG
   unsigned int debug_Id0 = detIdsToDebug[0];
   unsigned int debug_Id1 = detIdsToDebug[1];
   unsigned int debug_Id2 = detIdsToDebug[2];
 #endif
 
   std::array<bool, 3> bl;
   bl[0] = doublets.innerLayer().isBarrel;
   bl[1] = doublets.outerLayer().isBarrel;
 
   //gc: these are all the layers compatible with the layer pairs (as defined in the config file)
 
   //gc: initialize a KDTree per each 3rd layer
   std::vector<KDTreeNodeInfo<RecHitsSortedInPhi::HitIter, 2> > layerTree;  // re-used throughout
   std::vector<RecHitsSortedInPhi::HitIter> foundNodes;                     // re-used thoughout
   foundNodes.reserve(100);
   declareDynArray(KDTreeLinkerAlgo<RecHitsSortedInPhi::HitIter>, nThirdLayers, hitTree);
   declareDynArray(LayerRZPredictions, nThirdLayers, mapPred);
   float rzError[nThirdLayers];  //save maximum errors
 
   const float maxDelphi = region.ptMin() < 0.3f ? float(M_PI) / 4.f : float(M_PI) / 8.f;  // FIXME move to config??
   const float maxphi = M_PI + maxDelphi, minphi = -maxphi;  // increase to cater for any range
   const float safePhi = M_PI - maxDelphi;                   // sideband
 
   //gc: loop over each layer
   for (int il = 0; il < nThirdLayers; il++) {
     LogTrace("MultiHitGeneratorFromChi2")
         << "considering third layer: with hits: " << thirdHitMap[il]->all().second - thirdHitMap[il]->all().first;
     const DetLayer* layer = thirdLayerDetLayer[il];
     LayerRZPredictions& predRZ = mapPred[il];
     predRZ.line.initLayer(layer);
     predRZ.line.initTolerance(extraHitRZtolerance);
 
     //gc: now we take all hits in the layer and fill the KDTree
     auto const& layer3 = *thirdHitMap[il];  // Get iterators
     layerTree.clear();
     float minz = 999999.0f, maxz = -minz;  // Initialise to extreme values in case no hits
     float maxErr = 0.0f;
     if (!layer3.empty()) {
       minz = layer3.v[0];
       maxz = minz;  //In case there's only one hit on the layer
       for (auto i = 0U; i < layer3.size(); ++i) {
         auto hi = layer3.theHits.begin() + i;
         auto angle = layer3.phi(i);
         auto myz = layer3.v[i];
 #ifdef EDM_ML_DEBUG
         IfLogTrace(hi->hit()->rawId() == debug_Id2, "MultiHitGeneratorFromChi2")
             << "filling KDTree with hit in id=" << debug_Id2 << " with pos: " << hi->hit()->globalPosition()
             << " phi=" << hi->hit()->globalPosition().phi() << " z=" << hi->hit()->globalPosition().z()
             << " r=" << hi->hit()->globalPosition().perp();
 #endif
         //use (phi,r) for endcaps rather than (phi,z)
         if (myz < minz) {
           minz = myz;
         } else {
           if (myz > maxz) {
             maxz = myz;
           }
         }
         auto myerr = layer3.dv[i];
         if (myerr > maxErr) {
           maxErr = myerr;
         }
         layerTree.push_back(KDTreeNodeInfo<RecHitsSortedInPhi::HitIter, 2>(hi, angle, myz));  // save it
         // populate side-bands
         if (angle > safePhi)
           layerTree.push_back(
               KDTreeNodeInfo<RecHitsSortedInPhi::HitIter, 2>(hi, float(angle - Geom::twoPi()), float(myz)));
         else if (angle < -safePhi)
           layerTree.push_back(
               KDTreeNodeInfo<RecHitsSortedInPhi::HitIter, 2>(hi, float(angle + Geom::twoPi()), float(myz)));
       }
     }
     KDTreeBox phiZ(minphi, maxphi, minz - 0.01f, maxz + 0.01f);  // declare our bounds
     //add fudge factors in case only one hit and also for floating-point inaccuracy
     hitTree[il].build(layerTree, phiZ);  // make KDtree
     rzError[il] = maxErr;                //save error
   }
   //gc: now we have initialized the KDTrees and we are out of the layer loop
 
   //gc: this sets the minPt of the triplet
   auto curv = PixelRecoUtilities::curvature(1. / region.ptMin(), *bfield);
 
   LogTrace("MultiHitGeneratorFromChi2") << "doublet size=" << doublets.size() << std::endl;
 
   //fixme add pixels
   auto filterHit = [&](TrackingRecHit const* hit, GlobalVector const& dir) -> bool {
     auto hh = reinterpret_cast<BaseTrackerRecHit const*>(hit);
     if (  //hh->geographicalId().subdetId() > 2
         hh->geographicalId().subdetId() == SiStripDetId::TIB || hh->geographicalId().subdetId() == SiStripDetId::TID
         //|| hh->geographicalId().subdetId()==SiStripDetId::TOB
         //|| hh->geographicalId().subdetId()==SiStripDetId::TEC
     ) {
       // carefull " for matched and projected local of tsos != local for individual clusters...
       if (hh->isMatched()) {
         const SiStripMatchedRecHit2D* matchedHit = reinterpret_cast<const SiStripMatchedRecHit2D*>(hh);
         if (filter->isCompatible(DetId(matchedHit->monoId()), matchedHit->monoCluster(), dir) == 0 ||
             filter->isCompatible(DetId(matchedHit->stereoId()), matchedHit->stereoCluster(), dir) == 0)
           return false;
       } else if (hh->isProjected()) {
         const ProjectedSiStripRecHit2D* precHit = reinterpret_cast<const ProjectedSiStripRecHit2D*>(hh);
         if (filter->isCompatible(precHit->originalHit(), dir) == 0)
           return false;  //FIXME??
       } else if (hh->isSingle()) {
         const SiStripRecHit2D* recHit = reinterpret_cast<const SiStripRecHit2D*>(hh);
         if (filter->isCompatible(*recHit, dir) == 0)
           return false;
       }
     }
     return true;
   };
 
   //gc: now we loop over all pairs
   for (std::size_t ip = 0; ip != doublets.size(); ip++) {
     int foundTripletsFromPair = 0;
     bool usePair = false;
     cacheHitPointer bestH2;
     float minChi2 = std::numeric_limits<float>::max();
 
     SeedingHitSet::ConstRecHitPointer oriHit0 = doublets.hit(ip, HitDoublets::inner);
     SeedingHitSet::ConstRecHitPointer oriHit1 = doublets.hit(ip, HitDoublets::outer);
 
     HitOwnPtr hit0(*oriHit0);
     HitOwnPtr hit1(*oriHit1);
     GlobalPoint gp0 = doublets.gp(ip, HitDoublets::inner);
     GlobalPoint gp1 = doublets.gp(ip, HitDoublets::outer);
 
 #ifdef EDM_ML_DEBUG
     bool debugPair = oriHit0->rawId() == debug_Id0 && oriHit1->rawId() == debug_Id1;
 #endif
     IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << endl
                                                        << endl
                                                        << "found new pair with ids " << oriHit0->rawId() << " "
                                                        << oriHit1->rawId() << " with pos: " << gp0 << " " << gp1;
 
     if (refitHits) {
       TrajectoryStateOnSurface tsos0, tsos1;
       assert(!hit0.isOwn());
       assert(!hit1.isOwn());
 #ifdef EDM_ML_DEBUG
       refit2Hits(hit0, hit1, tsos0, tsos1, region, nomField, debugPair);
 #else
       refit2Hits(hit0, hit1, tsos0, tsos1, region, nomField, false);
 #endif
 
       bool passFilterHit0 = filterHit(hit0->hit(), tsos0.globalMomentum());
       IfLogTrace(debugPair && !passFilterHit0, "MultiHitGeneratorFromChi2") << "hit0 did not pass cluster shape filter";
       if (!passFilterHit0)
         continue;
       bool passFilterHit1 = filterHit(hit1->hit(), tsos1.globalMomentum());
       IfLogTrace(debugPair && !passFilterHit1, "MultiHitGeneratorFromChi2") << "hit1 did not pass cluster shape filter";
       if (!passFilterHit1)
         continue;
       // refit hits
       hit0.reset((SeedingHitSet::RecHitPointer)(cloner(*hit0, tsos0)));
       hit1.reset((SeedingHitSet::RecHitPointer)(cloner(*hit1, tsos1)));
 
 #ifdef EDM_ML_DEBUG
       IfLogTrace(debugPair, "MultiHitGeneratorFromChi2")
           << "charge=" << tsos0.charge() << std::endl
           << "state1 pt=" << tsos0.globalMomentum().perp() << " eta=" << tsos0.globalMomentum().eta()
           << " phi=" << tsos0.globalMomentum().phi() << std::endl
           << "state2 pt=" << tsos1.globalMomentum().perp() << " eta=" << tsos1.globalMomentum().eta()
           << " phi=" << tsos1.globalMomentum().phi() << std::endl
           << "positions after refitting: " << hit0->globalPosition() << " " << hit1->globalPosition();
 #endif
     } else {
       // not refit clone anyhow
       hit0.reset((BaseTrackerRecHit*)hit0->clone());
       hit1.reset((BaseTrackerRecHit*)hit1->clone());
     }
 
     assert(hit0.isOwn());
     assert(hit1.isOwn());
 
     //gc: create the RZ line for the pair
     SimpleLineRZ line(PixelRecoPointRZ(gp0.perp(), gp0.z()), PixelRecoPointRZ(gp1.perp(), gp1.z()));
     ThirdHitPredictionFromCircle predictionRPhi(gp0, gp1, extraHitRPhitolerance);
 
     auto toPos = std::signbit(gp1.z() - gp0.z());
 
     //gc: this is the curvature of the two hits assuming the region
     Range pairCurvature = predictionRPhi.curvature(region.originRBound());
     //gc: intersect not only returns a bool but may change pairCurvature to intersection with curv
     if (!intersect(pairCurvature, Range(-curv, curv))) {
       IfLogTrace(debugPair, "MultiHitGeneratorFromChi2")
           << "curvature cut: curv=" << curv << " gc=(" << pairCurvature.first << ", " << pairCurvature.second << ")";
       continue;
     }
 
     std::array<GlobalPoint, 3> gp;
     std::array<GlobalError, 3> ge;
     gp[0] = hit0->globalPosition();
     ge[0] = hit0->globalPositionError();
     gp[1] = hit1->globalPosition();
     ge[1] = hit1->globalPositionError();
 
     //gc: loop over all third layers compatible with the pair
     for (int il = 0; (il < nThirdLayers) & (!usePair); il++) {
       IfLogTrace(debugPair, "MultiHitGeneratorFromChi2")
           << "cosider layer:"
           << " for this pair. Location: " << thirdLayerDetLayer[il]->location();
 
       if (hitTree[il].empty()) {
         IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "empty hitTree";
         continue;  // Don't bother if no hits
       }
 
       cacheHitPointer bestL2;
       float chi2FromThisLayer = std::numeric_limits<float>::max();
 
       const DetLayer* layer = thirdLayerDetLayer[il];
       // bool barrelLayer = layer->location() == GeomDetEnumerators::barrel;
       auto const& layer3 = *thirdHitMap[il];
       bool barrelLayer = layer3.isBarrel;
       bl[2] = layer3.isBarrel;
 
       if ((!barrelLayer) & (toPos != std::signbit(layer->position().z())))
         continue;
 
       LayerRZPredictions& predRZ = mapPred[il];
       predRZ.line.initPropagator(&line);
 
       //gc: this takes the z at R-thick/2 and R+thick/2 according to
       //    the line from the two points and the adds the extra tolerance
       Range rzRange = predRZ.line();
 
       if (rzRange.first >= rzRange.second) {
         IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "rzRange empty";
         continue;
       }
       //gc: check that rzRange is compatible with detector bounds
       //    note that intersect may change rzRange to intersection with bounds
       if (!intersect(rzRange, predRZ.line.detSize())) {  // theDetSize = Range(-maxZ, maxZ);
         IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "rzRange and detector do not intersect";
         continue;
       }
       Range radius = barrelLayer ? predRZ.line.detRange() : rzRange;
 
       //gc: define the phi range of the hits
       Range phiRange;
       if (useFixedPreFiltering) {
         //gc: in this case it takes as range the phi of the outer
         //    hit +/- the phiPreFiltering value from cfg
         float phi0 = oriHit0->globalPosition().phi();
         phiRange = Range(phi0 - dphi, phi0 + dphi);
       } else {
         //gc: predictionRPhi uses the cosine rule to find the phi of the 3rd point at radius, assuming the pairCurvature range [-c,+c]
         if (pairCurvature.first < 0. && pairCurvature.second < 0.) {
           radius.swap();
         } else if (pairCurvature.first >= 0. && pairCurvature.second >= 0.) {
           ;
         } else {
           radius.first = radius.second;
         }
         auto phi12 = predictionRPhi.phi(pairCurvature.first, radius.first);
         auto phi22 = predictionRPhi.phi(pairCurvature.second, radius.second);
         phi12 = normalizedPhi(phi12);
         phi22 = proxim(phi22, phi12);
         phiRange = Range(phi12, phi22);
         phiRange.sort();
       }
 
       float prmin = phiRange.min(), prmax = phiRange.max();
 
       if (prmax - prmin > maxDelphi) {
         auto prm = phiRange.mean();
         prmin = prm - 0.5f * maxDelphi;
         prmax = prm + 0.5f * maxDelphi;
       }
 
       //gc: this is the place where hits in the compatible region are put in the foundNodes
       using Hit = RecHitsSortedInPhi::Hit;
       foundNodes.clear();  // Now recover hits in bounding box...
 
       IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "defining kd tree box";
 
       if (barrelLayer) {
         KDTreeBox phiZ(prmin - extraPhiKDBox,
                        prmax + extraPhiKDBox,
                        float(rzRange.min() - fnSigmaRZ * rzError[il] - extraZKDBox),
                        float(rzRange.max() + fnSigmaRZ * rzError[il] + extraZKDBox));
         hitTree[il].search(phiZ, foundNodes);
 
         IfLogTrace(debugPair, "MultiHitGeneratorFromChi2")
             << "kd tree box bounds, phi: " << prmin - extraPhiKDBox << "," << prmax + extraPhiKDBox
             << " z: " << rzRange.min() - fnSigmaRZ * rzError[il] - extraZKDBox << ","
             << rzRange.max() + fnSigmaRZ * rzError[il] + extraZKDBox << " rzRange: " << rzRange.min() << ","
             << rzRange.max();
 
       } else {
         KDTreeBox phiR(prmin - extraPhiKDBox,
                        prmax + extraPhiKDBox,
                        float(rzRange.min() - fnSigmaRZ * rzError[il] - extraRKDBox),
                        float(rzRange.max() + fnSigmaRZ * rzError[il] + extraRKDBox));
         hitTree[il].search(phiR, foundNodes);
 
         IfLogTrace(debugPair, "MultiHitGeneratorFromChi2")
             << "kd tree box bounds, phi: " << prmin - extraPhiKDBox << "," << prmax + extraPhiKDBox
             << " r: " << rzRange.min() - fnSigmaRZ * rzError[il] - extraRKDBox << ","
             << rzRange.max() + fnSigmaRZ * rzError[il] + extraRKDBox << " rzRange: " << rzRange.min() << ","
             << rzRange.max();
       }
 
       IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "kd tree box size: " << foundNodes.size();
 
       //gc: now we loop over the hits in the box for this layer
       for (std::vector<RecHitsSortedInPhi::HitIter>::iterator ih = foundNodes.begin();
            ih != foundNodes.end() && !usePair;
            ++ih) {
         IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << endl << "triplet candidate";
 
         const RecHitsSortedInPhi::HitIter KDdata = *ih;
 
         auto oriHit2 = KDdata->hit();
         auto kk = KDdata - layer3.theHits.begin();
         cacheHitPointer hit2;
         auto gp2 = layer3.gp(kk);
         if (refitHits) {  //fixme
 
           //fitting all 3 hits takes too much time... do it quickly only for 3rd hit
           GlobalVector initMomentum(gp2 - gp1);
           initMomentum /= initMomentum.perp();  //set pT=1
 
           //fixme add pixels
           bool passFilterHit2 = filterHit(oriHit2->hit(), initMomentum);
           if (!passFilterHit2)
             continue;
           TrajectoryStateOnSurface state(GlobalTrajectoryParameters(gp2, initMomentum, 1, &ufield),
                                          *oriHit2->surface());
           hit2.reset((SeedingHitSet::RecHitPointer)(cloner(*oriHit2, state)));
 
         } else {
           // not refit clone anyhow
           hit2.reset((BaseTrackerRecHit*)oriHit2->clone());
         }
 
         //gc: add the chi2 cut
         gp[2] = hit2->globalPosition();
         ge[2] = hit2->globalPositionError();
         RZLine rzLine(gp, ge, bl);
         float chi2 = rzLine.chi2();
 
 #ifdef EDM_ML_DEBUG
         bool debugTriplet = debugPair && hit2->rawId() == debug_Id2;
 #endif
         IfLogTrace(debugTriplet, "MultiHitGeneratorFromChi2")
             << endl
             << "triplet candidate in debug id" << std::endl
             << "hit in id=" << hit2->rawId() << " (from KDTree) with pos: " << KDdata->hit()->globalPosition()
             << " refitted: " << hit2->globalPosition() << " chi2: " << chi2;
         // should fix nan
         if ((chi2 > maxChi2) | edm::isNotFinite(chi2))
           continue;
 
         if (chi2VsPtCut) {
           FastCircle theCircle(gp[2], gp[1], gp[0]);
           float tesla0 = 0.1f * nomField;
           float rho = theCircle.rho();
           float cm2GeV = 0.01f * 0.3f * tesla0;
           float pt = cm2GeV * rho;
           IfLogTrace(debugTriplet, "MultiHitGeneratorFromChi2") << "triplet pT=" << pt;
           if (pt < region.ptMin())
             continue;
 
           if (chi2_cuts.size() == 4) {
             if ((pt > pt_interv[0] && pt <= pt_interv[1] && chi2 > chi2_cuts[0]) ||
                 (pt > pt_interv[1] && pt <= pt_interv[2] && chi2 > chi2_cuts[1]) ||
                 (pt > pt_interv[2] && pt <= pt_interv[3] && chi2 > chi2_cuts[2]) ||
                 (pt > pt_interv[3] && chi2 > chi2_cuts[3]))
               continue;
           }
 
           // apparently this takes too much time...
           //      if (chi2_cuts.size()>1) {
           //        int ncuts = chi2_cuts.size();
           //        if ( pt<=pt_interv[0] && chi2 > chi2_cuts[0] ) continue;
           //        bool pass = true;
           //        for (int icut=1; icut<ncuts-1; icut++){
           //          if ( pt>pt_interv[icut-1] && pt<=pt_interv[icut] && chi2 > chi2_cuts[icut] ) pass=false;
           //        }
           //        if (!pass) continue;
           //        if ( pt>pt_interv[ncuts-2] && chi2 > chi2_cuts[ncuts-1] ) continue;
           //        if (hit0->rawId()==debug_Id0 && hit1->rawId()==debug_Id1 && hit2->rawId()==debug_Id2) {
           //          LogTrace("MultiHitGeneratorFromChi2") << "triplet passed chi2 vs pt cut" << std::endl;
           //        }
           //      }
         }
 
         if (theMaxElement != 0 && result.size() >= theMaxElement) {
           result.clear();
           edm::LogError("TooManyTriplets") << " number of triples exceed maximum. no triplets produced.";
           return;
         }
         IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "triplet made";
         //result.push_back(SeedingHitSet(hit0, hit1, hit2));
         /* no refit so keep only hit2
     assert(tripletFromThisLayer.empty());
     assert(hit0.isOwn()); assert(hit1.isOwn());assert(hit2.isOwn());
     tripletFromThisLayer.emplace_back(std::move(hit0));
     tripletFromThisLayer.emplace_back(std::move(hit1));
     tripletFromThisLayer.emplace_back(std::move(hit2));
     assert(hit0.isEmpty()); assert(hit1.isEmpty());assert(hit2.isEmpty());
     */
         bestL2 = std::move(hit2);
         chi2FromThisLayer = chi2;
         foundTripletsFromPair++;
         if (foundTripletsFromPair >= 2) {
           usePair = true;
           IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "using pair";
           break;
         }
       }  //loop over hits in KDTree
 
       if (usePair)
         break;
       else {
         //if there is one triplet in more than one layer, try picking the one with best chi2
         if (chi2FromThisLayer < minChi2) {
           bestH2 = std::move(bestL2);
           minChi2 = chi2FromThisLayer;
         }
         /*
     else {
       if (!bestH2 && foundTripletsFromPair>0)
         LogTrace("MultiHitGeneratorFromChi2") << "what?? " <<  minChi2 << ' '  << chi2FromThisLayer;
     }
     */
       }
 
     }  //loop over layers
 
     if (foundTripletsFromPair == 0)
       continue;
 
     //push back only (max) once per pair
     IfLogTrace(debugPair, "MultiHitGeneratorFromChi2") << "Done seed #" << result.size();
     if (usePair)
       result.push_back(SeedingHitSet(oriHit0, oriHit1));
     else {
       assert(1 == foundTripletsFromPair);
       assert(bestH2);
       result.emplace_back(&*hit0, &*hit1, &*bestH2);
       assert(hit0.isOwn());
       assert(hit1.isOwn());
       refittedHitStorage.emplace_back(const_cast<BaseTrackerRecHit*>(hit0.release()));
       refittedHitStorage.emplace_back(const_cast<BaseTrackerRecHit*>(hit1.release()));
       refittedHitStorage.emplace_back(std::move(bestH2));
       assert(hit0.empty());
       assert(hit1.empty());
       assert(!bestH2);
     }
     // LogTrace("MultiHitGeneratorFromChi2") << (usePair ? "pair " : "triplet ") << minChi2 <<' ' << refittedHitStorage.size();
 
   }  //loop over pairs
   LogTrace("MultiHitGeneratorFromChi2") << "triplet size=" << result.size();
   // std::cout << "MultiHitGeneratorFromChi2 " << "triplet size=" << result.size() << std::endl;
 }

void MultiHitGeneratorFromChi2::hitTriplets	(	const TrackingRegion &	region,
		OrderedMultiHits &	result,
		const HitDoublets &	doublets,
		const RecHitsSortedInPhi **	thirdHitMap,
		const std::vector< const DetLayer * > &	thirdLayerDetLayer,
		const int	nThirdLayers
	)

overridevirtual

Implements MultiHitGeneratorFromPairAndLayers.

Definition at line 203 of file MultiHitGeneratorFromChi2.cc.

References MultiHitGeneratorFromPairAndLayers::cache, and hitSets().

                                                                     {
   hitSets(region, result, doublets, thirdHitMap, thirdLayerDetLayer, nThirdLayers, cache);
 }

void MultiHitGeneratorFromChi2::initES ( const edm::EventSetup & es )

overridevirtual

Implements MultiHitGeneratorFromPairAndLayers.

Definition at line 140 of file MultiHitGeneratorFromChi2.cc.

References bfield, builder, cloner, clusterShapeHitFilterESToken_, filter, edm::EventSetup::getData(), magneticFieldESToken_, nomField, MagneticField::nominalValue(), refitHits, UniformMagneticField::set(), transientTrackingRecHitBuilderESToken_, and ufield.

Referenced by MultiHitFromChi2EDProducer::produce().

                                                               {
   bfield = &es.getData(magneticFieldESToken_);
   nomField = bfield->nominalValue();
   ufield.set(nomField);  // more than enough (never actually used)
 
   if (refitHits) {
     filter = &es.getData(clusterShapeHitFilterESToken_);
     auto const& builderRef = es.getData(transientTrackingRecHitBuilderESToken_);
     builder = (TkTransientTrackingRecHitBuilder const*)(&builderRef);
     cloner = (*builder).cloner();
   }
 }

void MultiHitGeneratorFromChi2::refit2Hits	(	HitOwnPtr &	hit0,
		HitOwnPtr &	hit1,
		TrajectoryStateOnSurface &	tsos0,
		TrajectoryStateOnSurface &	tsos1,
		const TrackingRegion &	region,
		float	nomField,
		bool	isDebug
	)

private

Definition at line 678 of file MultiHitGeneratorFromChi2.cc.

References IfLogTrace, nomField, TrackingRegion::origin(), fireworks::p1, fireworks::p2, PV3DBase< T, PVType, FrameType >::perp(), DiDispStaMuonMonitor_cfi::pt, submitPVResolutionJobs::q, rho, FastCircle::rho(), edm::swap(), ufield, PV3DBase< T, PVType, FrameType >::x(), FastCircle::x0(), y, PV3DBase< T, PVType, FrameType >::y(), FastCircle::y0(), PV3DBase< T, PVType, FrameType >::z(), and gpuVertexFinder::zv.

Referenced by hitSets().

                                                          {
   //these need to be sorted in R
   const GlobalPoint& gp0 = region.origin();
   GlobalPoint gp1 = hit1->globalPosition();
   GlobalPoint gp2 = hit2->globalPosition();
 
   IfLogTrace(isDebug, "MultiHitGeneratorFromChi2")
       << "positions before refitting: " << hit1->globalPosition() << " " << hit2->globalPosition();
 
   FastCircle theCircle(gp2, gp1, gp0);
   GlobalPoint cc(theCircle.x0(), theCircle.y0(), 0);
   float tesla0 = 0.1f * nomField;
   float rho = theCircle.rho();
   float cm2GeV = 0.01f * 0.3f * tesla0;
   float pt = cm2GeV * rho;
 
   GlobalVector vec20 = gp2 - gp0;
   //if (isDebug) { cout << "vec20.eta=" << vec20.eta() << endl; }
 
   GlobalVector p0(gp0.y() - cc.y(), -gp0.x() + cc.x(), 0.);
   p0 = p0 * pt / p0.perp();
   GlobalVector p1(gp1.y() - cc.y(), -gp1.x() + cc.x(), 0.);
   p1 = p1 * pt / p1.perp();
   GlobalVector p2(gp2.y() - cc.y(), -gp2.x() + cc.x(), 0.);
   p2 = p2 * pt / p2.perp();
 
   //check sign according to scalar product
   if ((p0.x() * (gp1.x() - gp0.x()) + p0.y() * (gp1.y() - gp0.y())) < 0) {
     p0 *= -1.;
     p1 *= -1.;
     p2 *= -1.;
   }
 
   //now set z component
   auto zv = vec20.z() / vec20.perp();
   p0 = GlobalVector(p0.x(), p0.y(), p0.perp() * zv);
   p1 = GlobalVector(p1.x(), p1.y(), p1.perp() * zv);
   p2 = GlobalVector(p2.x(), p2.y(), p2.perp() * zv);
 
   //get charge from vectorial product
   TrackCharge q = 1;
   if ((gp1 - cc).x() * p1.y() - (gp1 - cc).y() * p1.x() > 0)
     q = -q;
 
   TrajectoryStateOnSurface(GlobalTrajectoryParameters(gp1, p1, q, &ufield), *hit1->surface()).swap(state1);
   // hit1.reset((SeedingHitSet::RecHitPointer)(cloner(*hit1,state1)));
 
   TrajectoryStateOnSurface(GlobalTrajectoryParameters(gp2, p2, q, &ufield), *hit2->surface()).swap(state2);
   // hit2.reset((SeedingHitSet::RecHitPointer)(cloner(*hit2,state2)));
 }