#include <TrackAssociator.h>

Classes
class	HAssociatorParameters

Public Member Functions
void	addDataLabels (const std::string className, const std::string moduleLabel, const std::string productInstanceLabel="")
	specify names of EDProducts to use for different input data types More...

HTrackDetMatchInfo	associate (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const HAssociatorParameters &)

std::vector< EcalRecHit >	associateEcal (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)

std::vector< CaloTower >	associateHcal (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)

double	getEcalEnergy (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)

FreeTrajectoryState	getFreeTrajectoryState (const edm::EventSetup &, const reco::Track &)
	get FreeTrajectoryState from different track representations More...

FreeTrajectoryState	getFreeTrajectoryState (const edm::EventSetup &, const SimTrack &, const SimVertex &)

double	getHcalEnergy (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)

	HTrackAssociator ()

void	setPropagator (Propagator *)
	use a user configured propagator More...

void	useDefaultPropagator ()
	use the default propagator More...

	~HTrackAssociator ()

Private Member Functions
void	fillCaloTowers (const edm::Event &, const edm::EventSetup &, HTrackDetMatchInfo &, const FreeTrajectoryState &, const int, const double)

void	fillEcal (const edm::Event &, const edm::EventSetup &, HTrackDetMatchInfo &, const FreeTrajectoryState &, const int, const double)

void	fillHcal (const edm::Event &, const edm::EventSetup &, HTrackDetMatchInfo &, const FreeTrajectoryState &, const int, const double)

void	fillHcalTowers (const edm::Event &, const edm::EventSetup &, HTrackDetMatchInfo &, const FreeTrajectoryState &, const int, const double)

math::XYZPoint	getPoint (const GlobalPoint &point)

math::XYZPoint	getPoint (const LocalPoint &point)

math::XYZVector	getVector (const GlobalVector &vec)

math::XYZVector	getVector (const LocalVector &vec)

void	init (const edm::EventSetup &)

Private Attributes
HCaloDetIdAssociator	caloDetIdAssociator_

std::vector< std::string >	CaloTowerCollectionLabels
	CaloTowers. More...

std::vector< std::vector< std::set< uint32_t > > > *	caloTowerMap_

int	debug_

Propagator *	defProp_

std::vector< std::string >	EBRecHitCollectionLabels

HEcalDetIdAssociator	ecalDetIdAssociator_

std::vector< std::string >	EERecHitCollectionLabels

std::vector< std::string >	HBHERecHitCollectionLabels
	HCAL. More...

HHcalDetIdAssociator	hcalDetIdAssociator_

Propagator *	ivProp_

edm::ESHandle< CaloGeometry >	theCaloGeometry_

bool	useDefaultPropagator_

Detailed Description

Definition at line 46 of file TrackAssociator.h.

Constructor & Destructor Documentation

◆ HTrackAssociator()

HTrackAssociator::HTrackAssociator ( )

Definition at line 44 of file TrackAssociator.cc.

                                    {
   ivProp_ = nullptr;
   defProp_ = nullptr;
   debug_ = 0;
   caloTowerMap_ = nullptr;
   useDefaultPropagator_ = false;
 }

◆ ~HTrackAssociator()

HTrackAssociator::~HTrackAssociator ( )

Definition at line 52 of file TrackAssociator.cc.

                                     {
   if (defProp_)
     delete defProp_;
 }

Member Function Documentation

◆ addDataLabels()

void HTrackAssociator::addDataLabels	(	const std::string	className,
		const std::string	moduleLabel,
		const std::string	productInstanceLabel = `""`
	)

specify names of EDProducts to use for different input data types

Definition at line 58 of file TrackAssociator.cc.

                                                                            {
   if (className == "EBRecHitCollection") {
     EBRecHitCollectionLabels.clear();
     EBRecHitCollectionLabels.push_back(moduleLabel);
     EBRecHitCollectionLabels.push_back(productInstanceLabel);
   }
   if (className == "EERecHitCollection") {
     EERecHitCollectionLabels.clear();
     EERecHitCollectionLabels.push_back(moduleLabel);
     EERecHitCollectionLabels.push_back(productInstanceLabel);
   }
   if (className == "HBHERecHitCollection") {
     HBHERecHitCollectionLabels.clear();
     HBHERecHitCollectionLabels.push_back(moduleLabel);
     HBHERecHitCollectionLabels.push_back(productInstanceLabel);
   }
   if (className == "CaloTowerCollection") {
     CaloTowerCollectionLabels.clear();
     CaloTowerCollectionLabels.push_back(moduleLabel);
     CaloTowerCollectionLabels.push_back(productInstanceLabel);
   }
 }

References className(), and HerwigMaxPtPartonFilter_cfi::moduleLabel.

◆ associate()

HTrackDetMatchInfo HTrackAssociator::associate	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const FreeTrajectoryState &	trackOrigin,
		const HAssociatorParameters &	parameters
	)

propagate a track across the whole detector and find associated objects. Association is done in two modes 1) an object is associated to a track only if crossed by track 2) an object is associated to a track if it is withing an eta-phi cone of some radius with respect to a track. (the cone origin is at (0,0,0))

Definition at line 116 of file TrackAssociator.cc.

                                                                                         {
   HTrackDetMatchInfo info;
   using namespace edm;
  
   init(iSetup);
  
   const FreeTrajectoryState& currentPosition(trackOrigin);
  
   if (parameters.useEcal)
     fillEcal(iEvent, iSetup, info, currentPosition, parameters.idREcal, parameters.dREcal);
   if (parameters.useHcal)
     fillHcal(iEvent, iSetup, info, currentPosition, parameters.idRHcal, parameters.dRHcal);
   if (parameters.useCalo)
     fillCaloTowers(iEvent, iSetup, info, currentPosition, parameters.idRCalo, parameters.dRCalo);
  
   return info;
 }

References HLT_2018_cff::fillCaloTowers, iEvent, info(), and init.

Referenced by SequenceTypes._ModuleSequenceType::remove(), and SequenceTypes._ModuleSequenceType::replace().

◆ associateEcal()

std::vector< EcalRecHit > HTrackAssociator::associateEcal	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const FreeTrajectoryState &	trackOrigin,
		const double	dR = `-1`
	)

associate ECAL only and return RecHits negative dR means only crossed elements

Definition at line 138 of file TrackAssociator.cc.

                                                                          {
   HAssociatorParameters parameters;
   parameters.useHcal = false;
   parameters.dREcal = dR;
   HTrackDetMatchInfo info(associate(iEvent, iSetup, trackOrigin, parameters));
   if (dR > 0)
     return info.coneEcalRecHits;
   else
     return info.crossedEcalRecHits;
 }

References HGC3DClusterGenMatchSelector_cfi::dR, iEvent, and info().

◆ associateHcal()

std::vector< CaloTower > HTrackAssociator::associateHcal	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const FreeTrajectoryState &	trackOrigin,
		const double	dR = `-1`
	)

associate ECAL only and return RecHits negative dR means only crossed elements

Definition at line 168 of file TrackAssociator.cc.

                                                                         {
   HAssociatorParameters parameters;
   parameters.useEcal = false;
   parameters.dRHcal = dR;
   HTrackDetMatchInfo info(associate(iEvent, iSetup, trackOrigin, parameters));
   if (dR > 0)
     return info.coneTowers;
   else
     return info.crossedTowers;
 }

References HGC3DClusterGenMatchSelector_cfi::dR, iEvent, and info().

◆ fillCaloTowers()

void HTrackAssociator::fillCaloTowers	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		HTrackDetMatchInfo &	info,
		const FreeTrajectoryState &	trajectoryPoint,
		const int	idR,
		const double	dR
	)

private

Definition at line 275 of file TrackAssociator.cc.

                                                        {
   // ECAL hits are not used for the CaloTower identification
   caloDetIdAssociator_.setGeometry(&*theCaloGeometry_);
  
   // HCAL points (HB+HE)
   std::vector<GlobalPoint> hcalPoints;
   hcalPoints.push_back(GlobalPoint(135., 0, 310.));
   hcalPoints.push_back(GlobalPoint(150., 0, 340.));
   hcalPoints.push_back(GlobalPoint(170., 0, 370.));
   hcalPoints.push_back(GlobalPoint(190., 0, 400.));
   hcalPoints.push_back(GlobalPoint(240., 0, 500.));
   hcalPoints.push_back(GlobalPoint(280., 0, 550.));
  
   std::vector<GlobalPoint> hcalTrajectory = caloDetIdAssociator_.getTrajectory(trajectoryPoint, hcalPoints);
   //   if(hcalTrajectory.empty()) throw cms::Exception("FatalError") << "Failed to propagate the track to HCAL\n";
  
   if (hcalTrajectory.empty()) {
     LogTrace("HTrackAssociator::fillEcal") << "Failed to propagate a track to ECAL; moving on\n";
     info.isGoodCalo = false;
     info.isGoodEcal = false;
     std::cout << " HTrackAssociator::fillCaloTowers::Failed to propagate a track to ECAL " << std::endl;
     return;
   }
  
   info.isGoodCalo = true;
   info.isGoodEcal = true;
   info.trkGlobPosAtEcal = getPoint(hcalTrajectory[0]);
  
   if (hcalTrajectory.size() < 4) {
     LogTrace("HTrackAssociator::fillEcal") << "Failed to propagate a track to HCAL; moving on\n";
     info.isGoodHcal = false;
   }
  
   info.isGoodHcal = true;
  
   info.trkGlobPosAtHcal = getPoint(hcalTrajectory[4]);
  
   // find crossed CaloTowers
   edm::Handle<CaloTowerCollection> caloTowers;
  
   if (CaloTowerCollectionLabels.empty())
     throw cms::Exception("FatalError") << "Module lable is not set for CaloTowers.\n";
   else
     iEvent.getByLabel(CaloTowerCollectionLabels[0], CaloTowerCollectionLabels[1], caloTowers);
   if (!caloTowers.isValid())
     throw cms::Exception("FatalError") << "Unable to find CaloTowers in event!\n";
  
   // first get DetIds in a predefined NxN region
   //   std::set<DetId> caloTowerIdsInBigRegion = caloDetIdAssociator_.getDetIdsCloseToAPoint(hcalTrajectory[0],idR+1);
   std::set<DetId> caloTowerIdsInRegion = caloDetIdAssociator_.getDetIdsCloseToAPoint(hcalTrajectory[0], idR);
  
   std::set<DetId> caloTowerIdsInACone;
   std::set<DetId> crossedCaloTowerIds;
   std::set<DetId> caloTowerIdsInBox;
   caloTowerIdsInACone = caloDetIdAssociator_.getDetIdsInACone(caloTowerIdsInRegion, hcalTrajectory, dR);
   // get DetId of the most energetic tower in that region
   crossedCaloTowerIds = caloDetIdAssociator_.getMaxEDetId(caloTowerIdsInRegion, caloTowers);
   // get DetIds of the towers surrounding the most energetic one
   caloTowerIdsInBox = caloDetIdAssociator_.getDetIdsInACone(crossedCaloTowerIds, hcalTrajectory, -1.);
  
   //
   //  Debug prints
   //
   //   std::cout <<" Debug printout in CaloTowers "<<std::endl;
   //   std::cout <<" with position at outer layer:r,z,phi "<<trajectoryPoint.position().eta()<<
   //               " "<<trajectoryPoint.position().phi()<<
   //           " "<<trajectoryPoint.position().perp()<<
   //               " "<<trajectoryPoint.position().z()<<
   //           " "<<trajectoryPoint.charge()<<std::endl;
   //   std::cout <<" Trajectory point at ECAL surface:eta:phi:radius:z "<<(hcalTrajectory[0]).eta()<<
   //               " "<<(hcalTrajectory[0]).phi()<<
   //               " "<<(hcalTrajectory[0]).perp()<<
   //           " "<<(hcalTrajectory[0]).z()<<
   //           " momentum "<<trajectoryPoint.momentum().perp()<<std::endl;
   //
   //   std::cout<<" Number of towers in the region "<<caloTowerIdsInRegion.size()<<" idR= "<<idR<<std::endl;
  
   // add CaloTowers
   for (std::set<DetId>::const_iterator itr = crossedCaloTowerIds.begin(); itr != crossedCaloTowerIds.end(); itr++) {
     DetId id(*itr);
     CaloTowerCollection::const_iterator tower = (*caloTowers).find(id);
     if (tower != (*caloTowers).end())
       info.crossedTowers.push_back(*tower);
     else
       LogTrace("HTrackAssociator::fillEcal") << "CaloTower is not found for DetId: " << id.rawId() << "\n";
   }
  
   for (std::set<DetId>::const_iterator itr = caloTowerIdsInACone.begin(); itr != caloTowerIdsInACone.end(); itr++) {
     DetId id(*itr);
     CaloTowerCollection::const_iterator tower = (*caloTowers).find(id);
     if (tower != (*caloTowers).end()) {
       info.coneTowers.push_back(*tower);
     } else
       LogTrace("HTrackAssociator::fillEcal") << "CaloTower is not found for DetId: " << id.rawId() << "\n";
   }
  
   for (std::set<DetId>::const_iterator itr = caloTowerIdsInBox.begin(); itr != caloTowerIdsInBox.end(); itr++) {
     DetId id(*itr);
     CaloTowerCollection::const_iterator tower = (*caloTowers).find(id);
     if (tower != (*caloTowers).end()) {
       info.boxTowers.push_back(*tower);
     } else
       LogTrace("HTrackAssociator::fillEcal") << "CaloTower is not found for DetId: " << id.rawId() << "\n";
   }
  
   for (std::set<DetId>::const_iterator itr = caloTowerIdsInRegion.begin(); itr != caloTowerIdsInRegion.end(); itr++) {
     DetId id(*itr);
     CaloTowerCollection::const_iterator tower = (*caloTowers).find(id);
     if (tower != (*caloTowers).end()) {
       info.regionTowers.push_back(*tower);
     } else
       LogTrace("HTrackAssociator::fillEcal") << "CaloTower is not found for DetId: " << id.rawId() << "\n";
   }
 }

References eleHcalExtractorBlocks_cff::caloTowers, gather_cfg::cout, HGC3DClusterGenMatchSelector_cfi::dR, triggerObjects_cff::id, iEvent, info(), LogTrace, and hgcalTowerProducer_cfi::tower.

◆ fillEcal()

void HTrackAssociator::fillEcal	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		HTrackDetMatchInfo &	info,
		const FreeTrajectoryState &	trajectoryPoint,
		const int	idR,
		const double	dR
	)

private

Definition at line 198 of file TrackAssociator.cc.

                                                  {
   ecalDetIdAssociator_.setGeometry(&*theCaloGeometry_);
  
   // ECAL points (EB+EE)
   std::vector<GlobalPoint> ecalPoints;
   ecalPoints.push_back(GlobalPoint(135., 0, 310.));
   ecalPoints.push_back(GlobalPoint(150., 0, 340.));
   ecalPoints.push_back(GlobalPoint(170., 0, 370.));
  
   std::vector<GlobalPoint> ecalTrajectory = ecalDetIdAssociator_.getTrajectory(trajectoryPoint, ecalPoints);
   //   if(ecalTrajectory.empty()) throw cms::Exception("FatalError") << "Failed to propagate a track to ECAL\n";
  
   if (ecalTrajectory.empty()) {
     LogTrace("HTrackAssociator::fillEcal") << "Failed to propagate a track to ECAL; moving on\n";
     info.isGoodEcal = false;
     return;
   }
  
   info.isGoodEcal = true;
  
   info.trkGlobPosAtEcal = getPoint(ecalTrajectory[0]);
  
   // Find ECAL crystals
   edm::Handle<EBRecHitCollection> EBRecHits;
   edm::Handle<EERecHitCollection> EERecHits;
   //   if (EBRecHitCollectionLabels.empty())
   //     throw cms::Exception("FatalError") << "Module lable is not set for EBRecHitCollection.\n";
   //   else
   iEvent.getByLabel(EBRecHitCollectionLabels[0], EBRecHitCollectionLabels[1], EBRecHits);
   //   if (!EBRecHits.isValid()) throw cms::Exception("FatalError") << "Unable to find EBRecHitCollection in event!\n";
   if (EERecHitCollectionLabels[1] != EBRecHitCollectionLabels[1])
     iEvent.getByLabel(EERecHitCollectionLabels[0], EERecHitCollectionLabels[1], EERecHits);
   //   if (!EERecHits.isValid()) throw cms::Exception("FatalError") << "Unable to find EERecHitCollection in event!\n";
  
   std::set<DetId> ecalIdsInRegion = ecalDetIdAssociator_.getDetIdsCloseToAPoint(ecalTrajectory[0], idR);
   std::set<DetId> ecalIdsInACone = ecalDetIdAssociator_.getDetIdsInACone(ecalIdsInRegion, ecalTrajectory, dR);
   std::set<DetId> crossedEcalIds = ecalDetIdAssociator_.getCrossedDetIds(ecalIdsInRegion, ecalTrajectory);
  
   // add EcalRecHits
   for (std::set<DetId>::const_iterator itr = crossedEcalIds.begin(); itr != crossedEcalIds.end(); itr++) {
     std::vector<EcalRecHit>::const_iterator hit = (*EBRecHits).find(*itr);
     if (hit != (*EBRecHits).end())
       info.crossedEcalRecHits.push_back(*hit);
     else
       LogTrace("HTrackAssociator::fillEcal") << "EcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
   for (std::set<DetId>::const_iterator itr = ecalIdsInACone.begin(); itr != ecalIdsInACone.end(); itr++) {
     std::vector<EcalRecHit>::const_iterator hit = (*EBRecHits).find(*itr);
     if (hit != (*EBRecHits).end()) {
       info.coneEcalRecHits.push_back(*hit);
     } else
       LogTrace("HTrackAssociator::fillEcal") << "EcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
   if (EERecHitCollectionLabels[1] == EBRecHitCollectionLabels[1])
     return;
   for (std::set<DetId>::const_iterator itr = crossedEcalIds.begin(); itr != crossedEcalIds.end(); itr++) {
     std::vector<EcalRecHit>::const_iterator hit = (*EERecHits).find(*itr);
     if (hit != (*EERecHits).end())
       info.crossedEcalRecHits.push_back(*hit);
     else
       LogTrace("HTrackAssociator::fillEcal") << "EcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
   for (std::set<DetId>::const_iterator itr = ecalIdsInACone.begin(); itr != ecalIdsInACone.end(); itr++) {
     std::vector<EcalRecHit>::const_iterator hit = (*EERecHits).find(*itr);
     if (hit != (*EERecHits).end()) {
       info.coneEcalRecHits.push_back(*hit);
     } else
       LogTrace("HTrackAssociator::fillEcal") << "EcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
 }

References HGC3DClusterGenMatchSelector_cfi::dR, filterRecHits_cfi::EBRecHits, filterRecHits_cfi::EERecHits, iEvent, info(), and LogTrace.

◆ fillHcal()

void HTrackAssociator::fillHcal	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		HTrackDetMatchInfo &	info,
		const FreeTrajectoryState &	trajectoryPoint,
		const int	idR,
		const double	dR
	)

private

Definition at line 396 of file TrackAssociator.cc.

                                                  {
   hcalDetIdAssociator_.setGeometry(&*theCaloGeometry_);
  
   // HCAL points (HB+HE)
   std::vector<GlobalPoint> hcalPoints;
   hcalPoints.push_back(GlobalPoint(190., 0, 400.));
   hcalPoints.push_back(GlobalPoint(240., 0, 500.));
   hcalPoints.push_back(GlobalPoint(280., 0, 550.));
  
   std::vector<GlobalPoint> hcalTrajectory = hcalDetIdAssociator_.getTrajectory(trajectoryPoint, hcalPoints);
   //   if(hcalTrajectory.empty()) throw cms::Exception("FatalError") << "Failed to propagate the track to HCAL\n";
  
   if (hcalTrajectory.empty()) {
     LogTrace("HTrackAssociator::fillHcal") << "Failed to propagate a track to HCAL; moving on\n";
     info.isGoodHcal = false;
     //    std::cout<<" HTrackAssociator::fillHcal::Failed to propagate a track to HCAL "<<std::endl;
     return;
   }
  
   info.isGoodHcal = true;
  
   info.trkGlobPosAtHcal = getPoint(hcalTrajectory[0]);
  
   edm::Handle<HBHERecHitCollection> HBHERecHits;
   //  if (HBHERecHitCollectionLabels.empty())
   //    throw cms::Exception("FatalError") << "Module label is not set for HBHERecHitCollection.\n";
   //  else
   iEvent.getByLabel(HBHERecHitCollectionLabels[0], HBHERecHitCollectionLabels[1], HBHERecHits);
   if (!HBHERecHits.isValid())
     throw cms::Exception("FatalError") << "Unable to find HBHERecHitCollection in event!\n";
  
   // first get DetIds in a predefined NxN region
   //  std::set<DetId> hcalIdsInBigRegion = hcalDetIdAssociator_.getDetIdsCloseToAPoint(hcalTrajectory[0],idR+1);
   std::set<DetId> hcalIdsInRegion = hcalDetIdAssociator_.getDetIdsCloseToAPoint(hcalTrajectory[0], idR);
  
   std::set<DetId> hcalIdsInACone;
   std::set<DetId> crossedHcalIds;
   std::set<DetId> hcalIdsInBox;
   hcalIdsInACone = hcalDetIdAssociator_.getDetIdsInACone(hcalIdsInRegion, hcalTrajectory, dR);
   // get DetId of the most energetic tower in that region
   crossedHcalIds = hcalDetIdAssociator_.getMaxEDetId(hcalIdsInRegion, HBHERecHits);
   // get DetIds of the towers surrounding the most energetic one
   hcalIdsInBox = hcalDetIdAssociator_.getDetIdsInACone(crossedHcalIds, hcalTrajectory, -1.);
  
   // add HcalRecHits
   for (std::set<DetId>::const_iterator itr = crossedHcalIds.begin(); itr != crossedHcalIds.end(); itr++) {
     std::vector<HBHERecHit>::const_iterator hit = (*HBHERecHits).find(*itr);
     if (hit != (*HBHERecHits).end())
       info.crossedHcalRecHits.push_back(*hit);
     else
       LogTrace("HTrackAssociator::fillHcal") << "HcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
   for (std::set<DetId>::const_iterator itr = hcalIdsInACone.begin(); itr != hcalIdsInACone.end(); itr++) {
     std::vector<HBHERecHit>::const_iterator hit = (*HBHERecHits).find(*itr);
     if (hit != (*HBHERecHits).end())
       info.coneHcalRecHits.push_back(*hit);
     else
       LogTrace("HTrackAssociator::fillHcal") << "HcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
   for (std::set<DetId>::const_iterator itr = hcalIdsInBox.begin(); itr != hcalIdsInBox.end(); itr++) {
     std::vector<HBHERecHit>::const_iterator hit = (*HBHERecHits).find(*itr);
     if (hit != (*HBHERecHits).end())
       info.boxHcalRecHits.push_back(*hit);
     else
       LogTrace("HTrackAssociator::fillHcal") << "HcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
   for (std::set<DetId>::const_iterator itr = hcalIdsInRegion.begin(); itr != hcalIdsInRegion.end(); itr++) {
     std::vector<HBHERecHit>::const_iterator hit = (*HBHERecHits).find(*itr);
     if (hit != (*HBHERecHits).end())
       info.regionHcalRecHits.push_back(*hit);
     else
       LogTrace("HTrackAssociator::fillHcal") << "HcalRecHit is not found for DetId: " << itr->rawId() << "\n";
   }
 }

References HGC3DClusterGenMatchSelector_cfi::dR, egHLT::errCodes::HBHERecHits, iEvent, info(), and LogTrace.

◆ fillHcalTowers()

void HTrackAssociator::fillHcalTowers	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		HTrackDetMatchInfo &	info,
		const FreeTrajectoryState &	trajectoryPoint,
		const int	idR,
		const double	dR
	)

private

Definition at line 477 of file TrackAssociator.cc.

                                                        {
   // ECAL hits are not used for the CaloTower identification
   caloDetIdAssociator_.setGeometry(&*theCaloGeometry_);
  
   // HCAL points (HB+HE)
   std::vector<GlobalPoint> hcalPoints;
   hcalPoints.push_back(GlobalPoint(190., 0, 400.));
   hcalPoints.push_back(GlobalPoint(240., 0, 500.));
   hcalPoints.push_back(GlobalPoint(280., 0, 550.));
  
   std::vector<GlobalPoint> hcalTrajectory = caloDetIdAssociator_.getTrajectory(trajectoryPoint, hcalPoints);
   //   if(hcalTrajectory.empty()) throw cms::Exception("FatalError") << "Failed to propagate the track to HCAL\n";
  
   if (hcalTrajectory.empty()) {
     LogTrace("HTrackAssociator::fillEcal") << "Failed to propagate a track to HCAL; moving on\n";
     info.isGoodCalo = false;
     std::cout << " HTrackAssociator::fillCaloTowers::Failed to propagate a track to HCAL " << std::endl;
     return;
   }
  
   info.isGoodCalo = true;
  
   info.trkGlobPosAtHcal = getPoint(hcalTrajectory[0]);
  
   // find crossed CaloTowers
   edm::Handle<CaloTowerCollection> caloTowers;
  
   if (CaloTowerCollectionLabels.empty())
     throw cms::Exception("FatalError") << "Module lable is not set for CaloTowers.\n";
   else
     iEvent.getByLabel(CaloTowerCollectionLabels[0], CaloTowerCollectionLabels[1], caloTowers);
   if (!caloTowers.isValid())
     throw cms::Exception("FatalError") << "Unable to find CaloTowers in event!\n";
  
   // first get DetIds in a predefined NxN region
   std::set<DetId> caloTowerIdsInBigRegion = caloDetIdAssociator_.getDetIdsCloseToAPoint(hcalTrajectory[0], idR + 1);
   std::set<DetId> caloTowerIdsInRegion = caloDetIdAssociator_.getDetIdsCloseToAPoint(hcalTrajectory[0], idR);
  
   std::set<DetId> caloTowerIdsInACone;
   std::set<DetId> crossedCaloTowerIds;
   std::set<DetId> caloTowerIdsInBox;
   caloTowerIdsInACone = caloDetIdAssociator_.getDetIdsInACone(caloTowerIdsInBigRegion, hcalTrajectory, dR);
   // get DetId of the most energetic tower in that region
   crossedCaloTowerIds = caloDetIdAssociator_.getMaxEDetId(caloTowerIdsInRegion, caloTowers);
   // get DetIds of the towers surrounding the most energetic one
   caloTowerIdsInBox = caloDetIdAssociator_.getDetIdsInACone(crossedCaloTowerIds, hcalTrajectory, -1.);
  
   // add CaloTowers
   for (std::set<DetId>::const_iterator itr = crossedCaloTowerIds.begin(); itr != crossedCaloTowerIds.end(); itr++) {
     DetId id(*itr);
     CaloTowerCollection::const_iterator tower = (*caloTowers).find(id);
     if (tower != (*caloTowers).end())
       info.crossedTowers.push_back(*tower);
     else
       LogTrace("HTrackAssociator::fillEcal") << "CaloTower is not found for DetId: " << id.rawId() << "\n";
   }
  
   for (std::set<DetId>::const_iterator itr = caloTowerIdsInACone.begin(); itr != caloTowerIdsInACone.end(); itr++) {
     DetId id(*itr);
     CaloTowerCollection::const_iterator tower = (*caloTowers).find(id);
     if (tower != (*caloTowers).end())
       info.coneTowers.push_back(*tower);
     else
       LogTrace("HTrackAssociator::fillEcal") << "CaloTower is not found for DetId: " << id.rawId() << "\n";
   }
 }

References eleHcalExtractorBlocks_cff::caloTowers, gather_cfg::cout, HGC3DClusterGenMatchSelector_cfi::dR, triggerObjects_cff::id, iEvent, info(), LogTrace, and hgcalTowerProducer_cfi::tower.

◆ getEcalEnergy()

double HTrackAssociator::getEcalEnergy	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const FreeTrajectoryState &	trackOrigin,
		const double	dR = `-1`
	)

associate ECAL only and return energy negative dR means only crossed elements

Definition at line 153 of file TrackAssociator.cc.

                                                         {
   HAssociatorParameters parameters;
   parameters.useHcal = false;
   parameters.dREcal = dR;
   HTrackDetMatchInfo info = associate(iEvent, iSetup, trackOrigin, parameters);
   if (dR > 0)
     return info.ecalConeEnergyFromRecHits();
   else
     return info.ecalEnergyFromRecHits();
 }

References HGC3DClusterGenMatchSelector_cfi::dR, iEvent, and info().

◆ getFreeTrajectoryState() [1/2]

FreeTrajectoryState HTrackAssociator::getFreeTrajectoryState	(	const edm::EventSetup &	iSetup,
		const reco::Track &	track
	)

get FreeTrajectoryState from different track representations

Definition at line 570 of file TrackAssociator.cc.

                                                                                                                 {
   edm::ESHandle<MagneticField> bField;
   iSetup.get<IdealMagneticFieldRecord>().get(bField);
  
   GlobalVector vector(track.momentum().x(), track.momentum().y(), track.momentum().z());
  
   GlobalPoint point(track.vertex().x(), track.vertex().y(), track.vertex().z());
  
   GlobalTrajectoryParameters tPars(point, vector, track.charge(), &*bField);
  
   // FIX THIS !!!
   // need to convert from perigee to global or helix (curvilinear) frame
   // for now just an arbitrary matrix.
   AlgebraicSymMatrix66 covT = AlgebraicMatrixID();
   covT *= 1e-6;  // initialize to sigma=1e-3
   CartesianTrajectoryError tCov(covT);
  
   return FreeTrajectoryState(tPars, tCov);
 }

References Calorimetry_cff::bField, MillePedeFileConverter_cfg::e, reco::get(), edm::EventSetup::get(), point, and HLT_2018_cff::track.

◆ getFreeTrajectoryState() [2/2]

FreeTrajectoryState HTrackAssociator::getFreeTrajectoryState	(	const edm::EventSetup &	iSetup,
		const SimTrack &	track,
		const SimVertex &	vertex
	)

Definition at line 550 of file TrackAssociator.cc.

                                                                                       {
   edm::ESHandle<MagneticField> bField;
   iSetup.get<IdealMagneticFieldRecord>().get(bField);
  
   GlobalVector vector(track.momentum().x(), track.momentum().y(), track.momentum().z());
   // convert mm to cm
   GlobalPoint point(vertex.position().x() * .1, vertex.position().y() * .1, vertex.position().z() * .1);
   int charge = track.type() > 0 ? -1 : 1;
   GlobalTrajectoryParameters tPars(point, vector, charge, &*bField);
  
   AlgebraicSymMatrix66 covT = AlgebraicMatrixID();
   covT *= 1e-6;  // initialize to sigma=1e-3
   CartesianTrajectoryError tCov(covT);
  
   return FreeTrajectoryState(tPars, tCov);
 }

References Calorimetry_cff::bField, ALCARECOTkAlJpsiMuMu_cff::charge, MillePedeFileConverter_cfg::e, reco::get(), edm::EventSetup::get(), point, HLT_2018_cff::track, and bphysicsOniaDQM_cfi::vertex.

◆ getHcalEnergy()

double HTrackAssociator::getHcalEnergy	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup,
		const FreeTrajectoryState &	trackOrigin,
		const double	dR = `-1`
	)

associate ECAL only and return energy negative dR means only crossed elements

Definition at line 183 of file TrackAssociator.cc.

                                                         {
   HAssociatorParameters parameters;
   parameters.useEcal = false;
   parameters.dRHcal = dR;
   HTrackDetMatchInfo info(associate(iEvent, iSetup, trackOrigin, parameters));
   if (dR > 0)
     return info.hcalConeEnergyFromRecHits();
   else
     return info.hcalEnergyFromRecHits();
 }

References HGC3DClusterGenMatchSelector_cfi::dR, iEvent, and info().

◆ getPoint() [1/2]

math::XYZPoint HTrackAssociator::getPoint ( const GlobalPoint & point )

inlineprivate

Definition at line 162 of file TrackAssociator.h.

162 { return math::XYZPoint(point.x(), point.y(), point.z()); }

References point.

◆ getPoint() [2/2]

math::XYZPoint HTrackAssociator::getPoint ( const LocalPoint & point )

inlineprivate

Definition at line 164 of file TrackAssociator.h.

164 { return math::XYZPoint(point.x(), point.y(), point.z()); }

References point.

◆ getVector() [1/2]

math::XYZVector HTrackAssociator::getVector ( const GlobalVector & vec )

inlineprivate

Definition at line 166 of file TrackAssociator.h.

166 { return math::XYZVector(vec.x(), vec.y(), vec.z()); }

References PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

◆ getVector() [2/2]

math::XYZVector HTrackAssociator::getVector ( const LocalVector & vec )

inlineprivate

Definition at line 168 of file TrackAssociator.h.

168 { return math::XYZVector(vec.x(), vec.y(), vec.z()); }

References PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

◆ init()

void HTrackAssociator::init ( const edm::EventSetup & iSetup )

private

Definition at line 95 of file TrackAssociator.cc.

                                                        {
   // access the calorimeter geometry
   iSetup.get<CaloGeometryRecord>().get(theCaloGeometry_);
   if (!theCaloGeometry_.isValid())
     throw cms::Exception("FatalError") << "Unable to find IdealGeometryRecord in event!\n";
  
   if (useDefaultPropagator_ && !defProp_) {
     // setup propagator
     edm::ESHandle<MagneticField> bField;
     iSetup.get<IdealMagneticFieldRecord>().get(bField);
  
     SteppingHelixPropagator* prop = new SteppingHelixPropagator(&*bField, anyDirection);
     prop->setMaterialMode(false);
     prop->applyRadX0Correction(true);
     // prop->setDebug(true); // tmp
     defProp_ = prop;
     setPropagator(defProp_);
   }
 }

References anyDirection, SteppingHelixPropagator::applyRadX0Correction(), Calorimetry_cff::bField, reco::get(), edm::EventSetup::get(), SteppingHelixPropagator::setMaterialMode(), and SteppingHelixPropagator_cfi::SteppingHelixPropagator.

◆ setPropagator()

void HTrackAssociator::setPropagator ( Propagator * ptr )

use a user configured propagator

Definition at line 84 of file TrackAssociator.cc.

                                                     {
   ivProp_ = ptr;
   caloDetIdAssociator_.setPropagator(ivProp_);
   ecalDetIdAssociator_.setPropagator(ivProp_);
   hcalDetIdAssociator_.setPropagator(ivProp_);
 }

◆ useDefaultPropagator()

void HTrackAssociator::useDefaultPropagator ( )

use the default propagator

Definition at line 92 of file TrackAssociator.cc.

92 { useDefaultPropagator_ = true; }

Member Data Documentation

◆ caloDetIdAssociator_

HCaloDetIdAssociator HTrackAssociator::caloDetIdAssociator_

private

Definition at line 178 of file TrackAssociator.h.

◆ CaloTowerCollectionLabels

std::vector<std::string> HTrackAssociator::CaloTowerCollectionLabels

private

CaloTowers.

Definition at line 189 of file TrackAssociator.h.

◆ caloTowerMap_

std::vector<std::vector<std::set<uint32_t> > >* HTrackAssociator::caloTowerMap_

private

Definition at line 174 of file TrackAssociator.h.

◆ debug_

int HTrackAssociator::debug_

private

Definition at line 173 of file TrackAssociator.h.

◆ defProp_

Propagator* HTrackAssociator::defProp_

private

Definition at line 171 of file TrackAssociator.h.

◆ EBRecHitCollectionLabels

std::vector<std::string> HTrackAssociator::EBRecHitCollectionLabels

private

Labels of the detector EDProducts (empty by default) ECAL

Definition at line 184 of file TrackAssociator.h.

◆ ecalDetIdAssociator_

HEcalDetIdAssociator HTrackAssociator::ecalDetIdAssociator_

private

Definition at line 176 of file TrackAssociator.h.

◆ EERecHitCollectionLabels

std::vector<std::string> HTrackAssociator::EERecHitCollectionLabels

private

Definition at line 185 of file TrackAssociator.h.

◆ HBHERecHitCollectionLabels

std::vector<std::string> HTrackAssociator::HBHERecHitCollectionLabels

private

HCAL.

Definition at line 187 of file TrackAssociator.h.

◆ hcalDetIdAssociator_

HHcalDetIdAssociator HTrackAssociator::hcalDetIdAssociator_

private

Definition at line 177 of file TrackAssociator.h.

◆ ivProp_

Propagator* HTrackAssociator::ivProp_

private

Definition at line 170 of file TrackAssociator.h.

◆ theCaloGeometry_

edm::ESHandle<CaloGeometry> HTrackAssociator::theCaloGeometry_

private

Definition at line 180 of file TrackAssociator.h.

◆ useDefaultPropagator_

bool HTrackAssociator::useDefaultPropagator_

private

Definition at line 172 of file TrackAssociator.h.

Classes

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ HTrackAssociator()

◆ ~HTrackAssociator()

Member Function Documentation

◆ addDataLabels()

◆ associate()

◆ associateEcal()

◆ associateHcal()

◆ fillCaloTowers()

◆ fillEcal()

◆ fillHcal()

◆ fillHcalTowers()

◆ getEcalEnergy()

◆ getFreeTrajectoryState() [1/2]

◆ getFreeTrajectoryState() [2/2]

◆ getHcalEnergy()

◆ getPoint() [1/2]

◆ getPoint() [2/2]

◆ getVector() [1/2]

◆ getVector() [2/2]

◆ init()

◆ setPropagator()

◆ useDefaultPropagator()

Member Data Documentation

◆ caloDetIdAssociator_

◆ CaloTowerCollectionLabels

◆ caloTowerMap_

◆ debug_

◆ defProp_

◆ EBRecHitCollectionLabels

◆ ecalDetIdAssociator_

◆ EERecHitCollectionLabels

◆ HBHERecHitCollectionLabels

◆ hcalDetIdAssociator_

◆ ivProp_

◆ theCaloGeometry_

◆ useDefaultPropagator_