HTrackAssociator Class Reference

#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

(

)

Definition at line 44 of file TrackAssociator.cc.

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

HTrackAssociator::~HTrackAssociator

(

)

Definition at line 52 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 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);
  }
}

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.

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

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

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

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.

References HTrackDetMatchInfo::coneEcalRecHits, HTrackDetMatchInfo::crossedEcalRecHits, HGC3DClusterGenMatchSelector_cfi::dR, HTrackAssociator::HAssociatorParameters::dREcal, info(), 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 168 of file TrackAssociator.cc.

References HTrackDetMatchInfo::coneTowers, HTrackDetMatchInfo::crossedTowers, HGC3DClusterGenMatchSelector_cfi::dR, HTrackAssociator::HAssociatorParameters::dRHcal, info(), 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 275 of file TrackAssociator.cc.

References HTrackDetMatchInfo::boxTowers, eleHcalExtractorBlocks_cff::caloTowers, HTrackDetMatchInfo::coneTowers, gather_cfg::cout, HTrackDetMatchInfo::crossedTowers, edm::Event::getByLabel(), triggerObjects_cff::id, HTrackDetMatchInfo::isGoodCalo, HTrackDetMatchInfo::isGoodEcal, HTrackDetMatchInfo::isGoodHcal, edm::HandleBase::isValid(), LogTrace, HTrackDetMatchInfo::regionTowers, hgcalTowerProducer_cfi::tower, HTrackDetMatchInfo::trkGlobPosAtEcal, and HTrackDetMatchInfo::trkGlobPosAtHcal.

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

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.

References HTrackDetMatchInfo::coneEcalRecHits, HTrackDetMatchInfo::crossedEcalRecHits, filterRecHits_cfi::EBRecHits, filterRecHits_cfi::EERecHits, edm::Event::getByLabel(), HTrackDetMatchInfo::isGoodEcal, LogTrace, and HTrackDetMatchInfo::trkGlobPosAtEcal.

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

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.

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

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

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.

References eleHcalExtractorBlocks_cff::caloTowers, HTrackDetMatchInfo::coneTowers, gather_cfg::cout, HTrackDetMatchInfo::crossedTowers, edm::Event::getByLabel(), triggerObjects_cff::id, HTrackDetMatchInfo::isGoodCalo, edm::HandleBase::isValid(), LogTrace, hgcalTowerProducer_cfi::tower, and HTrackDetMatchInfo::trkGlobPosAtHcal.

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

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.

References HGC3DClusterGenMatchSelector_cfi::dR, HTrackAssociator::HAssociatorParameters::dREcal, HTrackDetMatchInfo::ecalConeEnergyFromRecHits(), HTrackDetMatchInfo::ecalEnergyFromRecHits(), info(), 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 570 of file TrackAssociator.cc.

References Calorimetry_cff::bField, reco::TrackBase::charge(), MillePedeFileConverter_cfg::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 550 of file TrackAssociator.cc.

References Calorimetry_cff::bField, ALCARECOTkAlJpsiMuMu_cff::charge, MillePedeFileConverter_cfg::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 183 of file TrackAssociator.cc.

References HGC3DClusterGenMatchSelector_cfi::dR, HTrackAssociator::HAssociatorParameters::dRHcal, HTrackDetMatchInfo::hcalConeEnergyFromRecHits(), HTrackDetMatchInfo::hcalEnergyFromRecHits(), info(), 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 162 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 164 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 166 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 168 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 95 of file TrackAssociator.cc.

References anyDirection, SteppingHelixPropagator::applyRadX0Correction(), Calorimetry_cff::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 84 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 92 of file TrackAssociator.cc.

{ useDefaultPropagator_ = true; }

Member Data Documentation

HCaloDetIdAssociator HTrackAssociator::caloDetIdAssociator_

private

Definition at line 178 of file TrackAssociator.h.

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

private

CaloTowers.

Definition at line 189 of file TrackAssociator.h.

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

private

Definition at line 174 of file TrackAssociator.h.

int HTrackAssociator::debug_

private

Definition at line 173 of file TrackAssociator.h.

Propagator* HTrackAssociator::defProp_

private

Definition at line 171 of file TrackAssociator.h.

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

private

Labels of the detector EDProducts (empty by default) ECAL

Definition at line 184 of file TrackAssociator.h.

HEcalDetIdAssociator HTrackAssociator::ecalDetIdAssociator_

private

Definition at line 176 of file TrackAssociator.h.

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

private

Definition at line 185 of file TrackAssociator.h.

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

private

HCAL.

Definition at line 187 of file TrackAssociator.h.

HHcalDetIdAssociator HTrackAssociator::hcalDetIdAssociator_

private

Definition at line 177 of file TrackAssociator.h.

Propagator* HTrackAssociator::ivProp_

private

Definition at line 170 of file TrackAssociator.h.

edm::ESHandle<CaloGeometry> HTrackAssociator::theCaloGeometry_

private

Definition at line 180 of file TrackAssociator.h.

bool HTrackAssociator::useDefaultPropagator_

private

Definition at line 172 of file TrackAssociator.h.