#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 47 of file TrackAssociator.h.

Constructor & Destructor Documentation

HTrackAssociator::HTrackAssociator ( )

Definition at line 53 of file TrackAssociator.cc.

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

HTrackAssociator::~HTrackAssociator ( )

Definition at line 62 of file TrackAssociator.cc.

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

 {
    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, HTrackAssociator::HAssociatorParameters::idRCalo, HTrackAssociator::HAssociatorParameters::idREcal, HTrackAssociator::HAssociatorParameters::idRHcal, info(), init, HTrackAssociator::HAssociatorParameters::useCalo, HTrackAssociator::HAssociatorParameters::useEcal, and HTrackAssociator::HAssociatorParameters::useHcal.

 {
    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 HTrackDetMatchInfo::coneEcalRecHits, HTrackDetMatchInfo::crossedEcalRecHits, PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, HTrackAssociator::HAssociatorParameters::dREcal, info(), Parameters::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 HTrackDetMatchInfo::coneTowers, HTrackDetMatchInfo::crossedTowers, PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, HTrackAssociator::HAssociatorParameters::dRHcal, info(), Parameters::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	idR,
		const double	dR
	)

private

Definition at line 311 of file TrackAssociator.cc.

References HTrackDetMatchInfo::boxTowers, HiRecoJets_cff::caloTowers, HTrackDetMatchInfo::coneTowers, gather_cfg::cout, HTrackDetMatchInfo::crossedTowers, edm::Event::getByLabel(), HTrackDetMatchInfo::isGoodCalo, HTrackDetMatchInfo::isGoodEcal, HTrackDetMatchInfo::isGoodHcal, edm::HandleBase::isValid(), LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), HTrackDetMatchInfo::regionTowers, HTrackDetMatchInfo::trkGlobPosAtEcal, 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(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	idR,
		const double	dR
	)

private

Definition at line 226 of file TrackAssociator.cc.

References HTrackDetMatchInfo::coneEcalRecHits, HTrackDetMatchInfo::crossedEcalRecHits, egHLT::errCodes::EBRecHits, egHLT::errCodes::EERecHits, edm::Event::getByLabel(), HTrackDetMatchInfo::isGoodEcal, LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), and HTrackDetMatchInfo::trkGlobPosAtEcal.

 {
    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	idR,
		const double	dR
	)

private

Definition at line 448 of file TrackAssociator.cc.

References HTrackDetMatchInfo::boxHcalRecHits, HTrackDetMatchInfo::coneHcalRecHits, HTrackDetMatchInfo::crossedHcalRecHits, edm::Event::getByLabel(), egHLT::errCodes::HBHERecHits, HTrackDetMatchInfo::isGoodHcal, edm::HandleBase::isValid(), LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), HTrackDetMatchInfo::regionHcalRecHits, and HTrackDetMatchInfo::trkGlobPosAtHcal.

                                                        {
   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	idR,
		const double	dR
	)

private

Definition at line 536 of file TrackAssociator.cc.

References HiRecoJets_cff::caloTowers, HTrackDetMatchInfo::coneTowers, gather_cfg::cout, HTrackDetMatchInfo::crossedTowers, edm::Event::getByLabel(), HTrackDetMatchInfo::isGoodCalo, edm::HandleBase::isValid(), LogTrace, HTimerStack::pop_and_push(), HTimerStack::push(), 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 PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, HTrackAssociator::HAssociatorParameters::dREcal, HTrackDetMatchInfo::ecalConeEnergyFromRecHits(), HTrackDetMatchInfo::ecalEnergyFromRecHits(), info(), Parameters::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 reco::Track &	track
	)

get FreeTrajectoryState from different track representations

Definition at line 640 of file TrackAssociator.cc.

References ecalTB2006H4_GenSimDigiReco_cfg::bField, reco::TrackBase::charge(), alignCSCRings::e, edm::EventSetup::get(), reco::TrackBase::momentum(), point, 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.
    AlgebraicSymMatrix66 covT=  AlgebraicMatrixID(); covT *= 1e-6; // initialize to sigma=1e-3
    CartesianTrajectoryError tCov(covT);
    
    return FreeTrajectoryState(tPars, tCov);
 }

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

Definition at line 620 of file TrackAssociator.cc.

References ecalTB2006H4_GenSimDigiReco_cfg::bField, DeDxDiscriminatorTools::charge(), alignCSCRings::e, edm::EventSetup::get(), CoreSimTrack::momentum(), point, 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);
    
    AlgebraicSymMatrix66 covT=  AlgebraicMatrixID(); 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 PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, HTrackAssociator::HAssociatorParameters::dRHcal, HTrackDetMatchInfo::hcalConeEnergyFromRecHits(), HTrackDetMatchInfo::hcalEnergyFromRecHits(), info(), Parameters::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 GlobalPoint & point )

inlineprivate

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

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

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

inlineprivate

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::XYZVector HTrackAssociator::getVector ( const GlobalVector & vec )

inlineprivate

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

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

inlineprivate

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

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

private

Definition at line 116 of file TrackAssociator.cc.

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

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

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

 {
    useDefaultPropagator_ = true;
 }

Member Data Documentation

HCaloDetIdAssociator HTrackAssociator::caloDetIdAssociator_

private

Definition at line 201 of file TrackAssociator.h.

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

private

CaloTowers.

Definition at line 212 of file TrackAssociator.h.

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

private

Definition at line 197 of file TrackAssociator.h.

int HTrackAssociator::debug_

private

Definition at line 196 of file TrackAssociator.h.

Propagator* HTrackAssociator::defProp_

private

Definition at line 194 of file TrackAssociator.h.

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

private

Labels of the detector EDProducts (empty by default) ECAL

Definition at line 207 of file TrackAssociator.h.

HEcalDetIdAssociator HTrackAssociator::ecalDetIdAssociator_

private

Definition at line 199 of file TrackAssociator.h.

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

private

Definition at line 208 of file TrackAssociator.h.

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

private

HCAL.

Definition at line 210 of file TrackAssociator.h.

HHcalDetIdAssociator HTrackAssociator::hcalDetIdAssociator_

private

Definition at line 200 of file TrackAssociator.h.

Propagator* HTrackAssociator::ivProp_

private

Definition at line 193 of file TrackAssociator.h.

edm::ESHandle<CaloGeometry> HTrackAssociator::theCaloGeometry_

private

Definition at line 203 of file TrackAssociator.h.

bool HTrackAssociator::useDefaultPropagator_

private

Definition at line 195 of file TrackAssociator.h.

Classes

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation