PFRecHitCaloNavigator.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFClusterProducer_PFRecHitCaloNavigator_h
#define RecoParticleFlow_PFClusterProducer_PFRecHitCaloNavigator_h


#include "RecoParticleFlow/PFClusterProducer/interface/PFRecHitNavigatorBase.h"
#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"

#include "RecoCaloTools/Navigation/interface/CaloNavigator.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"
#include "DataFormats/EcalDetId/interface/ESDetId.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"

#include "Geometry/CaloTopology/interface/EcalEndcapTopology.h"
#include "Geometry/CaloTopology/interface/EcalBarrelTopology.h"
#include "Geometry/CaloTopology/interface/EcalPreshowerTopology.h"
#include "Geometry/CaloTopology/interface/HcalTopology.h"

#include "Geometry/CaloTopology/interface/CaloTowerTopology.h"
#include "DataFormats/CaloTowers/interface/CaloTowerDetId.h"


template <typename DET,typename TOPO,bool ownsTopo=true>
class PFRecHitCaloNavigator : public PFRecHitNavigatorBase {
 public:

 ~PFRecHitCaloNavigator() override { if(!ownsTopo) { topology_.release(); } }

  void associateNeighbours(reco::PFRecHit& hit,std::unique_ptr<reco::PFRecHitCollection>& hits,edm::RefProd<reco::PFRecHitCollection>& refProd) override {
      DetId detid( hit.detId() );
      
      CaloNavigator<DET> navigator(detid, topology_.get());
      
      DetId N(0);
      DetId E(0);
      DetId S(0);
      DetId W(0);
      DetId NW(0);
      DetId NE(0);
      DetId SW(0);
      DetId SE(0);


      N=navigator.north();  
      associateNeighbour(N,hit,hits,refProd,0,1,0);


      if (N !=DetId(0)) {
    NE=navigator.east();
      }
      else 
    {
      navigator.home();
      E=navigator.east();
      NE=navigator.north();
    }
      associateNeighbour(NE,hit,hits,refProd,1,1,0);
      navigator.home();

      S = navigator.south();
      associateNeighbour(S,hit,hits,refProd,0,-1,0);
      
      if (S !=DetId(0)) {
    SW = navigator.west();
      } else {
    navigator.home();
    W=navigator.west();
    SW=navigator.south();
      }
      associateNeighbour(SW,hit,hits,refProd,-1,-1,0);
      navigator.home();

      E = navigator.east();
      associateNeighbour(E,hit,hits,refProd,1,0,0);
      
      if (E !=DetId(0)) {
    SE = navigator.south();
      } else {
    navigator.home();
    S=navigator.south();
    SE=navigator.east();
      }
      associateNeighbour(SE,hit,hits,refProd,1,-1,0);
      navigator.home();


      W = navigator.west();
      associateNeighbour(W,hit,hits,refProd,-1,0,0);

      if (W !=DetId(0)) {
    NW = navigator.north();
      } else {
    navigator.home();
    N=navigator.north();
    NW=navigator.west();
      }
      associateNeighbour(NW,hit,hits,refProd,-1,1,0);
  }



 protected:
  std::unique_ptr<const TOPO> topology_;


};

#endif