---

HTrackAssociator Class Reference

#include <Calibration/Tools/interface/TrackAssociator.h>

List of all members.

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
HTrackDetMatchInfo	associate (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const HAssociatorParameters &)
	propagate a track across the whole detector and find associated objects.
std::vector< EcalRecHit >	associateEcal (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)
	associate ECAL only and return RecHits negative dR means only crossed elements
std::vector< CaloTower >	associateHcal (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)
	associate ECAL only and return RecHits negative dR means only crossed elements
double	getEcalEnergy (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)
	associate ECAL only and return energy negative dR means only crossed elements
FreeTrajectoryState	getFreeTrajectoryState (const edm::EventSetup &, const SimTrack &, const SimVertex &)
FreeTrajectoryState	getFreeTrajectoryState (const edm::EventSetup &, const reco::Track &)
	get FreeTrajectoryState from different track representations
double	getHcalEnergy (const edm::Event &, const edm::EventSetup &, const FreeTrajectoryState &, const double dR=-1)
	associate ECAL only and return energy negative dR means only crossed elements
	HTrackAssociator ()
void	setPropagator (Propagator *)
	use a user configured propagator
void	useDefaultPropagator ()
	use the default propagator
	~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 LocalPoint &point)
math::XYZPoint	getPoint (const GlobalPoint &point)
math::XYZVector	getVector (const LocalVector &vec)
math::XYZVector	getVector (const GlobalVector &vec)
void	init (const edm::EventSetup &)
Private Attributes
HCaloDetIdAssociator	caloDetIdAssociator_
std::vector< std::string >	CaloTowerCollectionLabels
	CaloTowers.
std::vector< std::vector < std::set< uint32_t > > > *	caloTowerMap_
int	debug_
Propagator *	defProp_
std::vector< std::string >	EBRecHitCollectionLabels
	Labels of the detector EDProducts (empty by default) ECAL.
HEcalDetIdAssociator	ecalDetIdAssociator_
std::vector< std::string >	EERecHitCollectionLabels
std::vector< std::string >	HBHERecHitCollectionLabels
	HCAL.
HHcalDetIdAssociator	hcalDetIdAssociator_
Propagator *	ivProp_
edm::ESHandle< CaloGeometry >	theCaloGeometry_
bool	useDefaultPropagator_
Classes
class	HAssociatorParameters

---

Detailed Description

Definition at line 47 of file TrackAssociator.h.

---

Constructor & Destructor Documentation

HTrackAssociator::HTrackAssociator

(

)

Definition at line 53 of file TrackAssociator.cc.

References caloTowerMap_, debug_, defProp_, ivProp_, and useDefaultPropagator_.

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

HTrackAssociator::~HTrackAssociator

(

)

Definition at line 62 of file TrackAssociator.cc.

References defProp_.

{
   if (defProp_) delete defProp_;
}

---

Member Function Documentation

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 68 of file TrackAssociator.cc.

References CaloTowerCollectionLabels, EBRecHitCollectionLabels, EERecHitCollectionLabels, and HBHERecHitCollectionLabels.

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

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 140 of file TrackAssociator.cc.

References HTrackAssociator::HAssociatorParameters::dRCalo, HTrackAssociator::HAssociatorParameters::dREcal, HTrackAssociator::HAssociatorParameters::dRHcal, fillCaloTowers(), fillEcal(), fillHcal(), HTrackAssociator::HAssociatorParameters::idRCalo, HTrackAssociator::HAssociatorParameters::idREcal, HTrackAssociator::HAssociatorParameters::idRHcal, info, init(), HTrackAssociator::HAssociatorParameters::useCalo, HTrackAssociator::HAssociatorParameters::useEcal, and HTrackAssociator::HAssociatorParameters::useHcal.

Referenced by associateEcal(), associateHcal(), getEcalEnergy(), and getHcalEnergy().

{
   HTrackDetMatchInfo info;
   using namespace edm;
   HTimerStack timers;
   
   init( iSetup );
   
   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;
}

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 161 of file TrackAssociator.cc.

References associate(), HTrackDetMatchInfo::coneEcalRecHits, HTrackDetMatchInfo::crossedEcalRecHits, HTrackAssociator::HAssociatorParameters::dREcal, info, python::trackProbabilityAnalysis_cff::parameters, and HTrackAssociator::HAssociatorParameters::useHcal.

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

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 193 of file TrackAssociator.cc.

References associate(), HTrackDetMatchInfo::coneTowers, HTrackDetMatchInfo::crossedTowers, HTrackAssociator::HAssociatorParameters::dRHcal, info, python::trackProbabilityAnalysis_cff::parameters, and HTrackAssociator::HAssociatorParameters::useEcal.

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

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

Definition at line 311 of file TrackAssociator.cc.

References HTrackDetMatchInfo::boxTowers, caloDetIdAssociator_, CaloTowerCollectionLabels, IterativeConePu5Jets_PbPb_cff::caloTowers, HTrackDetMatchInfo::coneTowers, GenMuonPlsPt100GeV_cfg::cout, HTrackDetMatchInfo::crossedTowers, lat::endl(), edm::Event::getByLabel(), HDetIdAssociator::getDetIdsCloseToAPoint(), HDetIdAssociator::getDetIdsInACone(), HDetIdAssociator::getMaxEDetId(), getPoint(), HDetIdAssociator::getTrajectory(), id, HTrackDetMatchInfo::isGoodCalo, HTrackDetMatchInfo::isGoodEcal, HTrackDetMatchInfo::isGoodHcal, edm::Handle< T >::isValid(), LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), HTrackDetMatchInfo::regionTowers, HCaloDetIdAssociator::setGeometry(), theCaloGeometry_, HTrackDetMatchInfo::trkGlobPosAtEcal, and HTrackDetMatchInfo::trkGlobPosAtHcal.

Referenced by associate().

{
   // ECAL hits are not used for the CaloTower identification
   HTimerStack timers;
   timers.push("HTrackAssociator::fillCaloTowers");

   caloDetIdAssociator_.setGeometry(&*theCaloGeometry_);
   
   // HCAL points (HB+HE)
   timers.push("HTrackAssociator::fillCaloTowers::propagation");
   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 = 0;
      info.isGoodEcal = 0;
      std::cout<<" HTrackAssociator::fillCaloTowers::Failed to propagate a track to ECAL "<<std::endl;
      return;
   }
   
   info.isGoodCalo = 1;
   info.isGoodEcal = 1;
   info.trkGlobPosAtEcal = getPoint(hcalTrajectory[0]);
   
   if(hcalTrajectory.size()<4) {
      LogTrace("HTrackAssociator::fillEcal") << "Failed to propagate a track to HCAL; moving on\n";
      info.isGoodHcal = 0;
   }
   
   info.isGoodHcal = 1;
   
   info.trkGlobPosAtHcal = getPoint(hcalTrajectory[4]);

   // find crossed CaloTowers
   timers.pop_and_push("HTrackAssociator::fillCaloTowers::access::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";
   
   timers.push("HTrackAssociator::fillCaloTowers::matching");

// 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
   timers.push("HTrackAssociator::fillCaloTowers::addCaloTowers");
   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";
     }

}

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

Definition at line 226 of file TrackAssociator.cc.

References HTrackDetMatchInfo::coneEcalRecHits, HTrackDetMatchInfo::crossedEcalRecHits, EBRecHitCollectionLabels, ecalDetIdAssociator_, EERecHitCollectionLabels, edm::Event::getByLabel(), HDetIdAssociator::getCrossedDetIds(), HDetIdAssociator::getDetIdsCloseToAPoint(), HDetIdAssociator::getDetIdsInACone(), getPoint(), HDetIdAssociator::getTrajectory(), HTrackDetMatchInfo::isGoodEcal, LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), HCaloDetIdAssociator::setGeometry(), theCaloGeometry_, and HTrackDetMatchInfo::trkGlobPosAtEcal.

Referenced by associate().

{
   HTimerStack timers;
   timers.push("HTrackAssociator::fillEcal");
   
   ecalDetIdAssociator_.setGeometry(&*theCaloGeometry_);
   
   timers.push("HTrackAssociator::fillEcal::propagation");
   // 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 = 0;
      return;
   }
   
   info.isGoodEcal = 1;

   info.trkGlobPosAtEcal = getPoint(ecalTrajectory[0]);

   // Find ECAL crystals
   timers.pop_and_push("HTrackAssociator::fillEcal::access::EcalBarrel");
   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";

   timers.pop_and_push("HTrackAssociator::fillEcal::matching");
   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
   timers.pop_and_push("HTrackAssociator::fillEcal::addEcalRecHits");
   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";
   }
}

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

Definition at line 448 of file TrackAssociator.cc.

References HTrackDetMatchInfo::boxHcalRecHits, HTrackDetMatchInfo::coneHcalRecHits, HTrackDetMatchInfo::crossedHcalRecHits, edm::Event::getByLabel(), HDetIdAssociator::getDetIdsCloseToAPoint(), HDetIdAssociator::getDetIdsInACone(), HDetIdAssociator::getMaxEDetId(), getPoint(), HDetIdAssociator::getTrajectory(), HBHERecHitCollectionLabels, hcalDetIdAssociator_, HTrackDetMatchInfo::isGoodHcal, edm::Handle< T >::isValid(), LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), HTrackDetMatchInfo::regionHcalRecHits, HCaloDetIdAssociator::setGeometry(), theCaloGeometry_, and HTrackDetMatchInfo::trkGlobPosAtHcal.

Referenced by associate().

                                                       {
  HTimerStack timers;
  timers.push("HTrackAssociator::fillHcal");

  hcalDetIdAssociator_.setGeometry(&*theCaloGeometry_);

// HCAL points (HB+HE)
  timers.push("HTrackAssociator::fillHcal::propagation");
  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 = 0;
//    std::cout<<" HTrackAssociator::fillHcal::Failed to propagate a track to HCAL "<<std::endl;
    return;
  }

  info.isGoodHcal = 1;

  info.trkGlobPosAtHcal = getPoint(hcalTrajectory[0]);
  timers.pop_and_push("HTrackAssociator::fillHcal::access::Hcal");

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

  timers.pop_and_push("HTrackAssociator::fillHcal::matching");

// 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
  timers.pop_and_push("HTrackAssociator::fillHcal::addHcalRecHits");
  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";
  }
}

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

Definition at line 536 of file TrackAssociator.cc.

References caloDetIdAssociator_, CaloTowerCollectionLabels, IterativeConePu5Jets_PbPb_cff::caloTowers, HTrackDetMatchInfo::coneTowers, GenMuonPlsPt100GeV_cfg::cout, HTrackDetMatchInfo::crossedTowers, lat::endl(), edm::Event::getByLabel(), HDetIdAssociator::getDetIdsCloseToAPoint(), HDetIdAssociator::getDetIdsInACone(), HDetIdAssociator::getMaxEDetId(), getPoint(), HDetIdAssociator::getTrajectory(), id, HTrackDetMatchInfo::isGoodCalo, edm::Handle< T >::isValid(), LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), HCaloDetIdAssociator::setGeometry(), theCaloGeometry_, and HTrackDetMatchInfo::trkGlobPosAtHcal.

{
   // ECAL hits are not used for the CaloTower identification
   HTimerStack timers;
   timers.push("HTrackAssociator::fillCaloTowers");

   caloDetIdAssociator_.setGeometry(&*theCaloGeometry_);
   
   // HCAL points (HB+HE)
   timers.push("HTrackAssociator::fillCaloTowers::propagation");
   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 = 0;
      std::cout<<" HTrackAssociator::fillCaloTowers::Failed to propagate a track to HCAL "<<std::endl;
      return;
   }
   
   info.isGoodCalo = 1;

   info.trkGlobPosAtHcal = getPoint(hcalTrajectory[0]);
   
   // find crossed CaloTowers
   timers.pop_and_push("HTrackAssociator::fillCaloTowers::access::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";
   
   timers.push("HTrackAssociator::fillCaloTowers::matching");

// 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
   timers.push("HTrackAssociator::fillCaloTowers::addCaloTowers");
   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";
     }
   
}

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 177 of file TrackAssociator.cc.

References associate(), HTrackAssociator::HAssociatorParameters::dREcal, HTrackDetMatchInfo::ecalConeEnergyFromRecHits(), HTrackDetMatchInfo::ecalEnergyFromRecHits(), info, python::trackProbabilityAnalysis_cff::parameters, and HTrackAssociator::HAssociatorParameters::useHcal.

{
   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();
}

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

Definition at line 620 of file TrackAssociator.cc.

References e, edm::EventSetup::get(), CoreSimTrack::momentum(), CoreSimVertex::position(), and CoreSimTrack::type().

{
   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);
   
   HepSymMatrix covT(6,1); covT *= 1e-6; // initialize to sigma=1e-3
   CartesianTrajectoryError tCov(covT);
   
   return FreeTrajectoryState(tPars, tCov);
}

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

get FreeTrajectoryState from different track representations

Definition at line 640 of file TrackAssociator.cc.

References reco::TrackBase::charge(), e, edm::EventSetup::get(), reco::TrackBase::momentum(), and reco::TrackBase::vertex().

{
   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.
   HepSymMatrix covT(6,1); covT *= 1e-6; // initialize to sigma=1e-3
   CartesianTrajectoryError tCov(covT);
   
   return FreeTrajectoryState(tPars, tCov);
}

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 210 of file TrackAssociator.cc.

References associate(), HTrackAssociator::HAssociatorParameters::dRHcal, HTrackDetMatchInfo::hcalConeEnergyFromRecHits(), HTrackDetMatchInfo::hcalEnergyFromRecHits(), info, python::trackProbabilityAnalysis_cff::parameters, and HTrackAssociator::HAssociatorParameters::useEcal.

{
   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();
}

math::XYZPoint HTrackAssociator::getPoint

(

const LocalPoint &

point

)

[inline, private]

Definition at line 178 of file TrackAssociator.h.

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

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

math::XYZPoint HTrackAssociator::getPoint

(

const GlobalPoint &

point

)

[inline, private]

Definition at line 173 of file TrackAssociator.h.

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

Referenced by fillCaloTowers(), fillEcal(), fillHcal(), and fillHcalTowers().

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

math::XYZVector HTrackAssociator::getVector

(

const LocalVector &

vec

)

[inline, private]

Definition at line 188 of file TrackAssociator.h.

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

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

math::XYZVector HTrackAssociator::getVector

(

const GlobalVector &

vec

)

[inline, private]

Definition at line 183 of file TrackAssociator.h.

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

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

void HTrackAssociator::init

(

const edm::EventSetup &

iSetup

)

[private]

Definition at line 116 of file TrackAssociator.cc.

References anyDirection, SteppingHelixPropagator::applyRadX0Correction(), defProp_, edm::EventSetup::get(), edm::ESHandle< T >::isValid(), SteppingHelixPropagator::setMaterialMode(), setPropagator(), SteppingHelixPropagator_cfi::SteppingHelixPropagator, theCaloGeometry_, and useDefaultPropagator_.

Referenced by associate().

{
   // 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_);
   }
   
        
}

void HTrackAssociator::setPropagator

(

Propagator *

ptr

)

use a user configured propagator

Definition at line 100 of file TrackAssociator.cc.

References caloDetIdAssociator_, ecalDetIdAssociator_, hcalDetIdAssociator_, ivProp_, and HDetIdAssociator::setPropagator().

Referenced by init().

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

void HTrackAssociator::useDefaultPropagator

(

)

use the default propagator

Definition at line 109 of file TrackAssociator.cc.

References useDefaultPropagator_.

{
   useDefaultPropagator_ = true;
}

---

Member Data Documentation

HCaloDetIdAssociator HTrackAssociator::caloDetIdAssociator_ [private]

Definition at line 201 of file TrackAssociator.h.

Referenced by fillCaloTowers(), fillHcalTowers(), and setPropagator().

std::vector<std::string> HTrackAssociator::CaloTowerCollectionLabels [private]

CaloTowers.

Definition at line 212 of file TrackAssociator.h.

Referenced by addDataLabels(), fillCaloTowers(), and fillHcalTowers().

std::vector<std::vector<std::set<uint32_t> > >* HTrackAssociator::caloTowerMap_ [private]

Definition at line 197 of file TrackAssociator.h.

Referenced by HTrackAssociator().

int HTrackAssociator::debug_ [private]

Definition at line 196 of file TrackAssociator.h.

Referenced by HTrackAssociator().

Propagator* HTrackAssociator::defProp_ [private]

Definition at line 194 of file TrackAssociator.h.

Referenced by HTrackAssociator(), init(), and ~HTrackAssociator().

std::vector<std::string> HTrackAssociator::EBRecHitCollectionLabels [private]

Labels of the detector EDProducts (empty by default) ECAL.

Definition at line 207 of file TrackAssociator.h.

Referenced by addDataLabels(), and fillEcal().

HEcalDetIdAssociator HTrackAssociator::ecalDetIdAssociator_ [private]

Definition at line 199 of file TrackAssociator.h.

Referenced by fillEcal(), and setPropagator().

std::vector<std::string> HTrackAssociator::EERecHitCollectionLabels [private]

Definition at line 208 of file TrackAssociator.h.

Referenced by addDataLabels(), and fillEcal().

std::vector<std::string> HTrackAssociator::HBHERecHitCollectionLabels [private]

HCAL.

Definition at line 210 of file TrackAssociator.h.

Referenced by addDataLabels(), and fillHcal().

HHcalDetIdAssociator HTrackAssociator::hcalDetIdAssociator_ [private]

Definition at line 200 of file TrackAssociator.h.

Referenced by fillHcal(), and setPropagator().

Propagator* HTrackAssociator::ivProp_ [private]

Definition at line 193 of file TrackAssociator.h.

Referenced by HTrackAssociator(), and setPropagator().

edm::ESHandle<CaloGeometry> HTrackAssociator::theCaloGeometry_ [private]

Definition at line 203 of file TrackAssociator.h.

Referenced by fillCaloTowers(), fillEcal(), fillHcal(), fillHcalTowers(), and init().

bool HTrackAssociator::useDefaultPropagator_ [private]

Definition at line 195 of file TrackAssociator.h.

Referenced by HTrackAssociator(), init(), and useDefaultPropagator().

---

The documentation for this class was generated from the following files:

• Calibration/Tools/interface/TrackAssociator.h
• Calibration/Tools/src/TrackAssociator.cc

---