HcalDDDGeometry.cc

Go to the documentation of this file.

#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"
#include "Geometry/HcalTowerAlgo/interface/HcalDDDGeometry.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <algorithm>

//#define DebugLog

HcalDDDGeometry::HcalDDDGeometry() : lastReqDet_(DetId::Detector(0)), 
                     lastReqSubdet_(0), etaMax_(0),
                     firstHFQuadRing_(40) {
  twopi = M_PI + M_PI;
  deg   = M_PI/180.;
}


HcalDDDGeometry::~HcalDDDGeometry() {}


std::vector<DetId> const & HcalDDDGeometry::getValidDetIds(DetId::Detector det,
                               int subdet) const {

  const std::vector<DetId>& baseIds(CaloSubdetectorGeometry::getValidDetIds());
  if (det    == DetId::Detector( 0 ) && subdet == 0) {
    return baseIds ;
  }
   
  if (lastReqDet_  != det || lastReqSubdet_ != subdet ) {
    lastReqDet_     = det    ;
    lastReqSubdet_  = subdet ;
    m_validIds.clear();
    m_validIds.reserve( baseIds.size() ) ;
  }

  if (m_validIds.empty() ) {
    for (unsigned int i = 0 ; i != baseIds.size() ; ++i ) {
      const DetId id ( baseIds[i] );
      if (id.det()      == det   && id.subdetId() == subdet ) { 
    m_validIds.push_back( id ) ;
      }
    }
    std::sort(m_validIds.begin(),m_validIds.end());
  }
  
#ifdef DebugLog
  LogDebug("HCalGeom") << "HcalDDDGeometry::getValidDetIds: "
               << m_validIds.size() << " valid IDs found for detector "
               << det << " Sub-detector " << subdet;
#endif
  return m_validIds;
}


DetId HcalDDDGeometry::getClosestCell(const GlobalPoint& r) const {

  // Now find the closest eta_bin, eta value of a bin i is average
  // of eta[i] and eta[i-1]
  double abseta = fabs(r.eta());
  double phi    = r.phi();
  if (phi < 0) phi += twopi;
  double radius = r.mag();
  double z      = fabs(r.z());
#ifdef DebugLog
  LogDebug("HCalGeom") << "HcalDDDGeometry::getClosestCell for eta "
               << r.eta() << " phi " << phi/deg << " z " << r.z()
               << " radius " << radius;
#endif

  HcalDetId bestId;
  if (abseta <= etaMax_) {
    for (unsigned int i=0; i<hcalCells_.size(); i++) {
      if (abseta >=hcalCells_[i].etaMin() && abseta <=hcalCells_[i].etaMax()) {
    HcalSubdetector bc = hcalCells_[i].detType();
    int etaring = hcalCells_[i].etaBin();
    int phibin  = 0;
    if (hcalCells_[i].unitPhi() == 4) {
      // rings 40 and 41 are offset wrt the other phi numbering
      //  1        1         1         2
      //  ------------------------------
      //  72       36        36        1
      phibin = static_cast<int>(((phi/deg)+hcalCells_[i].phiOffset()+
                     0.5*hcalCells_[i].phiBinWidth())/
                    hcalCells_[i].phiBinWidth());
      if (phibin == 0) phibin = hcalCells_[i].nPhiBins();
      phibin = phibin*4 - 1; 
    } else {
      phibin = static_cast<int>(((phi/deg)+hcalCells_[i].phiOffset())/
                    hcalCells_[i].phiBinWidth()) + 1;
      // convert to the convention of numbering 1,3,5, in 36 phi bins
      phibin = (phibin-1)*(hcalCells_[i].unitPhi()) + 1;
    }

    int dbin   = 1;
    int etabin = (r.z() > 0) ? etaring : -etaring;
    if (bc == HcalForward) {
      bestId   = HcalDetId(bc, etabin, phibin, dbin);
      break;
    } else {
      double rz = z;
      if (hcalCells_[i].depthType()) rz = radius;
      if (rz < hcalCells_[i].depthMax()) {
        dbin   = hcalCells_[i].depthSegment();
        bestId = HcalDetId(bc, etabin, phibin, dbin);
        break;
      }
    }
      }
    }
  }
#ifdef DebugLog
  LogDebug("HCalGeom") << "HcalDDDGeometry::getClosestCell " << bestId;
#endif
  return bestId;
}


int HcalDDDGeometry::insertCell(std::vector<HcalCellType> const & cells){

  hcalCells_.insert(hcalCells_.end(), cells.begin(), cells.end());
  int num = static_cast<int>(hcalCells_.size());
  for (unsigned int i=0; i<cells.size(); i++) {
    if (cells[i].etaMax() > etaMax_ ) etaMax_ = cells[i].etaMax();
  }
#ifdef DebugLog
  LogDebug("HCalGeom") << "HcalDDDGeometry::insertCell " << cells.size()
               << " cells inserted == Total " << num
               << " EtaMax = " << etaMax_;
#endif
  return num;
}