#include <PFHCALDenseIdNavigator.h>

Inheritance diagram for PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >:

Public Member Functions
void	associateNeighbours (reco::PFRecHit &hit, std::unique_ptr< reco::PFRecHitCollection > &hits, edm::RefProd< reco::PFRecHitCollection > &refProd) override

unsigned int	getIdx (const unsigned int denseid) const

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

	PFHCALDenseIdNavigator (const edm::ParameterSet &iConfig, edm::ConsumesCollector &cc)

bool	validNeighbours (const unsigned int denseid) const

	~PFHCALDenseIdNavigator () override

Public Member Functions inherited from PFRecHitNavigatorBase
	PFRecHitNavigatorBase ()=default

	PFRecHitNavigatorBase (const edm::ParameterSet &iConfig, edm::ConsumesCollector &cc)

virtual	~PFRecHitNavigatorBase ()=default

Protected Attributes
unsigned int	denseIdHcalMax_

unsigned int	denseIdHcalMin_

std::vector< std::vector< DetId > >	neighboursHcal_

edm::ESWatcher< HcalRecNumberingRecord >	theRecNumberWatcher_

std::unique_ptr< const TOPO >	topology_

std::vector< int >	vhcalEnum_

Private Attributes
edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	geomToken_

edm::ESGetToken< HcalTopology, HcalRecNumberingRecord >	hcalToken_

Additional Inherited Members
Public Types inherited from PFRecHitNavigatorBase
typedef std::unordered_map< unsigned, unsigned >	DetIdToHitIdx

Protected Member Functions inherited from PFRecHitNavigatorBase
void	associateNeighbour (const DetId &id, reco::PFRecHit &hit, std::unique_ptr< reco::PFRecHitCollection > &hits, edm::RefProd< reco::PFRecHitCollection > &refProd, short eta, short phi, short depth)

Detailed Description

template<typename DET, typename TOPO, bool ownsTopo = true>
class PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >

Definition at line 26 of file PFHCALDenseIdNavigator.h.

Constructor & Destructor Documentation

◆ ~PFHCALDenseIdNavigator()

template<typename DET, typename TOPO, bool ownsTopo = true>

PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >::~PFHCALDenseIdNavigator ( )

inlineoverride

Definition at line 28 of file PFHCALDenseIdNavigator.h.

                                      {
     if (!ownsTopo) {
       topology_.release();
     }
   }

◆ PFHCALDenseIdNavigator()

template<typename DET, typename TOPO, bool ownsTopo = true>

PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >::PFHCALDenseIdNavigator	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &	cc
	)

inline

Definition at line 34 of file PFHCALDenseIdNavigator.h.

       : vhcalEnum_(iConfig.getParameter<std::vector<int>>("hcalEnums")),
         hcalToken_(cc.esConsumes<edm::Transition::BeginLuminosityBlock>()),
         geomToken_(cc.esConsumes<edm::Transition::BeginLuminosityBlock>()) {}

Member Function Documentation

◆ associateNeighbours()

template<typename DET, typename TOPO, bool ownsTopo = true>

void PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >::associateNeighbours	(	reco::PFRecHit &	hit,
		std::unique_ptr< reco::PFRecHitCollection > &	hits,
		edm::RefProd< reco::PFRecHitCollection > &	refProd
	)

inlineoverridevirtual

Implements PFRecHitNavigatorBase.

Definition at line 143 of file PFHCALDenseIdNavigator.h.

                                                                                  {
     DetId detid(hit.detId());
     unsigned denseid = topology_.get()->detId2denseId(detid);
  
     std::vector<DetId> neighbours(9, DetId(0));
  
     if (denseid < denseIdHcalMin_ || denseid > denseIdHcalMax_) {
       edm::LogWarning("PFRecHitHCALCachedNavigator") << " DenseId for this cell is out of the range." << std::endl;
     } else if (!validNeighbours(denseid)) {
       edm::LogWarning("PFRecHitHCALCachedNavigator")
           << " DenseId for this cell does not have the neighbour information." << std::endl;
     } else {
       unsigned index = getIdx(denseid);
       neighbours = neighboursHcal_.at(index);
     }
  
     associateNeighbour(neighbours.at(NORTH), hit, hits, refProd, 0, 1, 0);        // N
     associateNeighbour(neighbours.at(NORTHEAST), hit, hits, refProd, 1, 1, 0);    // NE
     associateNeighbour(neighbours.at(SOUTH), hit, hits, refProd, 0, -1, 0);       // S
     associateNeighbour(neighbours.at(SOUTHWEST), hit, hits, refProd, -1, -1, 0);  // SW
     associateNeighbour(neighbours.at(EAST), hit, hits, refProd, 1, 0, 0);         // E
     associateNeighbour(neighbours.at(SOUTHEAST), hit, hits, refProd, 1, -1, 0);   // SE
     associateNeighbour(neighbours.at(WEST), hit, hits, refProd, -1, 0, 0);        // W
     associateNeighbour(neighbours.at(NORTHWEST), hit, hits, refProd, -1, 1, 0);   // NW
   }

◆ getIdx()

template<typename DET, typename TOPO, bool ownsTopo = true>

unsigned int PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >::getIdx ( const unsigned int denseid ) const

inline

Definition at line 179 of file PFHCALDenseIdNavigator.h.

                                                         {
     unsigned index = denseid - denseIdHcalMin_;
     return index;
   }

Referenced by PFHCALDenseIdNavigator< HcalDetId, HcalTopology, false >::associateNeighbours(), PFHCALDenseIdNavigator< HcalDetId, HcalTopology, false >::init(), and PFHCALDenseIdNavigator< HcalDetId, HcalTopology, false >::validNeighbours().

◆ init()

template<typename DET, typename TOPO, bool ownsTopo = true>

void PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >::init ( const edm::EventSetup & iSetup )

inlineoverridevirtual

Implements PFRecHitNavigatorBase.

Definition at line 39 of file PFHCALDenseIdNavigator.h.

                                                   {
     bool check = theRecNumberWatcher_.check(iSetup);
     if (!check)
       return;
  
     edm::ESHandle<HcalTopology> hcalTopology = iSetup.getHandle(hcalToken_);
     topology_.release();
     topology_.reset(hcalTopology.product());
  
     // Fill a vector of valid denseid's
     edm::ESHandle<CaloGeometry> hGeom = iSetup.getHandle(geomToken_);
     const CaloGeometry& caloGeom = *hGeom;
  
     std::vector<DetId> vecHcal;
     std::vector<unsigned int> vDenseIdHcal;
     neighboursHcal_.clear();
     for (auto hcalSubdet : vhcalEnum_) {
       std::vector<DetId> vecDetIds(caloGeom.getValidDetIds(DetId::Hcal, hcalSubdet));
       vecHcal.insert(vecHcal.end(), vecDetIds.begin(), vecDetIds.end());
     }
     vDenseIdHcal.reserve(vecHcal.size());
     for (auto hDetId : vecHcal) {
       vDenseIdHcal.push_back(topology_.get()->detId2denseId(hDetId));
     }
     std::sort(vDenseIdHcal.begin(), vDenseIdHcal.end());
  
     // Fill a vector of cell neighbours
     denseIdHcalMax_ = *max_element(vDenseIdHcal.begin(), vDenseIdHcal.end());
     denseIdHcalMin_ = *min_element(vDenseIdHcal.begin(), vDenseIdHcal.end());
     neighboursHcal_.resize(denseIdHcalMax_ - denseIdHcalMin_ + 1);
  
     for (auto denseid : vDenseIdHcal) {
       DetId N(0);
       DetId E(0);
       DetId S(0);
       DetId W(0);
       DetId NW(0);
       DetId NE(0);
       DetId SW(0);
       DetId SE(0);
       std::vector<DetId> neighbours(9, DetId(0));
  
       // the centre
       unsigned denseid_c = denseid;
       DetId detid_c = topology_.get()->denseId2detId(denseid_c);
       CaloNavigator<DET> navigator(detid_c, topology_.get());
  
       // Using enum in Geometry/CaloTopology/interface/CaloDirection.h
       // Order: CENTER(NONE),SOUTH,SOUTHEAST,SOUTHWEST,EAST,WEST,NORTHEAST,NORTHWEST,NORTH
       neighbours.at(NONE) = detid_c;
  
       navigator.home();
       N = navigator.north();
       neighbours.at(NORTH) = N;
       if (N != DetId(0)) {
         NE = navigator.east();
       } else {
         navigator.home();
         E = navigator.east();
         NE = navigator.north();
       }
       neighbours.at(NORTHEAST) = NE;
  
       navigator.home();
       S = navigator.south();
       neighbours.at(SOUTH) = S;
       if (S != DetId(0)) {
         SW = navigator.west();
       } else {
         navigator.home();
         W = navigator.west();
         SW = navigator.south();
       }
       neighbours.at(SOUTHWEST) = SW;
  
       navigator.home();
       E = navigator.east();
       neighbours.at(EAST) = E;
       if (E != DetId(0)) {
         SE = navigator.south();
       } else {
         navigator.home();
         S = navigator.south();
         SE = navigator.east();
       }
       neighbours.at(SOUTHEAST) = SE;
  
       navigator.home();
       W = navigator.west();
       neighbours.at(WEST) = W;
       if (W != DetId(0)) {
         NW = navigator.north();
       } else {
         navigator.home();
         N = navigator.north();
         NW = navigator.west();
       }
       neighbours.at(NORTHWEST) = NW;
  
       unsigned index = getIdx(denseid_c);
       neighboursHcal_[index] = neighbours;
     }
   }

◆ validNeighbours()

template<typename DET, typename TOPO, bool ownsTopo = true>

bool PFHCALDenseIdNavigator< DET, TOPO, ownsTopo >::validNeighbours ( const unsigned int denseid ) const

inline

Definition at line 171 of file PFHCALDenseIdNavigator.h.

                                                          {
     bool ok = true;
     unsigned index = getIdx(denseid);
     if (neighboursHcal_.at(index).size() != 9)
       ok = false;  // the neighbour vector size should be 3x3
     return ok;
   }