CaloCellManager.cc

Go to the documentation of this file.

/* class CaloCellManager
 *
 * Simple eta-phi cell structure manager, mimic calorimetric tower structure
 *
 *
 */
 
#include "Validation/EventGenerator/interface/CaloCellManager.h"
 
#include "CLHEP/Units/defs.h"
#include "CLHEP/Units/SystemOfUnits.h"
 
#include <iomanip>
 
CaloCellManager::CaloCellManager(const unsigned int theVerbosity) : verbosity(theVerbosity) {
  //initialize constants
 
  init();
 
  // build the calo cells identifiers
 
  builder();
}
 
CaloCellManager::~CaloCellManager() {
  for (unsigned int i = 0; i < theCellCollection.size(); i++) {
    delete theCellCollection[i];
  }
}
 
void CaloCellManager::init() {
  etaLim.reserve(nBarrelEta + nEndcapEta + nForwardEta + 1);
  phiLimBar.reserve(nBarrelEta + 1);
  phiLimEnd.reserve(nEndcapEta + 1);
  phiLimFor.reserve(nForwardEta + 1);
 
  // Barrel ranges
 
  double firstEta = 0.;
  double etaBarrelBin = 0.087;
  for (unsigned int ietabin = 0; ietabin <= nBarrelEta; ietabin++) {
    etaLim.push_back(firstEta + ietabin * etaBarrelBin);
  }
 
  double firstPhi = -180.;
  double phiBarrelBin = (double)360 / nBarrelPhi;
  for (unsigned int iphibin = 0; iphibin <= nBarrelPhi; iphibin++) {
    phiLimBar.push_back((firstPhi + iphibin * phiBarrelBin) * CLHEP::degree);
  }
 
  // Endcap ranges (compromise wrt real CMS)
 
  firstEta = etaBarrelBin * nBarrelEta;
  double etaEndcapBin = 0.131;
  for (unsigned int ietabin = 1; ietabin <= nEndcapEta; ietabin++) {
    etaLim.push_back(firstEta + ietabin * etaEndcapBin);
  }
  double phiEndcapBin = (double)360 / nEndcapPhi;
  for (unsigned int iphibin = 0; iphibin <= nEndcapPhi; iphibin++) {
    phiLimEnd.push_back((firstPhi + iphibin * phiEndcapBin) * CLHEP::degree);
  }
 
  // Forward ranges (compromise wrt real CMS)
 
  etaLim.push_back(3.139);
  etaLim.push_back(3.314);
  etaLim.push_back(3.489);
  etaLim.push_back(3.664);
  etaLim.push_back(3.839);
  etaLim.push_back(4.013);
  etaLim.push_back(4.191);
  etaLim.push_back(4.363);
  etaLim.push_back(4.538);
  etaLim.push_back(4.716);
  etaLim.push_back(4.889);
  etaLim.push_back(5.191);
 
  double phiForwardBin = (double)360 / nForwardPhi;
  for (unsigned int iphibin = 0; iphibin <= nForwardPhi; iphibin++) {
    phiLimFor.push_back((firstPhi + iphibin * phiForwardBin) * CLHEP::degree);
  }
 
  if (verbosity > 0) {
    std::cout << "Number of eta ranges = " << nBarrelEta + nEndcapEta + nForwardEta << std::endl;
    for (unsigned int i = 0; i < etaLim.size(); i++) {
      std::cout << "Eta range limit # " << i << " = " << etaLim[i] << std::endl;
    }
    for (unsigned int i = 0; i < phiLimBar.size(); i++) {
      std::cout << "Phi barrel range limit # " << i << " = " << phiLimBar[i] << std::endl;
    }
    for (unsigned int i = 0; i < phiLimEnd.size(); i++) {
      std::cout << "Phi endcap range limit # " << i << " = " << phiLimEnd[i] << std::endl;
    }
    for (unsigned int i = 0; i < phiLimFor.size(); i++) {
      std::cout << "Phi forward range limit # " << i << " = " << phiLimFor[i] << std::endl;
    }
  }
}
 
void CaloCellManager::builder() {
  theCellCollection.reserve(nCaloCell);
 
  // Barrel
 
  CaloCellId::System theSys = CaloCellId::Barrel;
 
  for (unsigned int iphi = 0; iphi < nBarrelPhi; iphi++) {
    for (unsigned int ieta = 0; ieta < nBarrelEta; ieta++) {
      CaloCellId* thisCell =
          new CaloCellId(etaLim[ieta], etaLim[ieta + 1], phiLimBar[iphi], phiLimBar[iphi + 1], theSys);
      theCellCollection.push_back(thisCell);
    }
    for (unsigned int ieta = 0; ieta < nBarrelEta; ieta++) {
      CaloCellId* thisCell =
          new CaloCellId(-1. * etaLim[ieta + 1], -1. * etaLim[ieta], phiLimBar[iphi], phiLimBar[iphi + 1], theSys);
      theCellCollection.push_back(thisCell);
    }
  }
 
  // Endcap
 
  theSys = CaloCellId::Endcap;
 
  for (unsigned int iphi = 0; iphi < nEndcapPhi; iphi++) {
    for (unsigned int ieta = nBarrelEta; ieta < nBarrelEta + nEndcapEta; ieta++) {
      CaloCellId* thisCell =
          new CaloCellId(etaLim[ieta], etaLim[ieta + 1], phiLimEnd[iphi], phiLimEnd[iphi + 1], theSys);
      theCellCollection.push_back(thisCell);
    }
    for (unsigned int ieta = nBarrelEta; ieta < nBarrelEta + nEndcapEta; ieta++) {
      CaloCellId* thisCell =
          new CaloCellId(-1. * etaLim[ieta + 1], -1. * etaLim[ieta], phiLimEnd[iphi], phiLimEnd[iphi + 1], theSys);
      theCellCollection.push_back(thisCell);
    }
  }
 
  // Forward
 
  theSys = CaloCellId::Forward;
 
  for (unsigned int iphi = 0; iphi < nForwardPhi; iphi++) {
    for (unsigned int ieta = nBarrelEta + nEndcapEta; ieta < nBarrelEta + nEndcapEta + nForwardEta; ieta++) {
      CaloCellId* thisCell =
          new CaloCellId(etaLim[ieta], etaLim[ieta + 1], phiLimFor[iphi], phiLimFor[iphi + 1], theSys);
      theCellCollection.push_back(thisCell);
    }
    for (unsigned int ieta = nBarrelEta + nEndcapEta; ieta < nBarrelEta + nEndcapEta + nForwardEta; ieta++) {
      CaloCellId* thisCell =
          new CaloCellId(-1. * etaLim[ieta + 1], -1. * etaLim[ieta], phiLimFor[iphi], phiLimFor[iphi + 1], theSys);
      theCellCollection.push_back(thisCell);
    }
  }
 
  if (verbosity > 0) {
    std::cout << "Number of cells = " << nCaloCell << std::endl;
    for (unsigned int i = 0; i < theCellCollection.size(); i++) {
      std::cout << "Cell # " << std::setfill(' ') << std::setw(4) << i << " = " << *(theCellCollection[i]) << std::endl;
    }
  }
}
 
unsigned int CaloCellManager::getCellIndexFromAngle(double eta, double phi) {
  unsigned int theIndex = 1000000;
  for (unsigned int i = 0; i < theCellCollection.size(); i++) {
    if (theCellCollection[i]->isInCell(eta, phi)) {
      theIndex = i;
      continue;
    }
  }
  return theIndex;
}
 
CaloCellId* CaloCellManager::getCellFromIndex(unsigned int id) {
  if (id < theCellCollection.size()) {
    return theCellCollection[id];
  }
  return nullptr;
}
 
std::vector<double> CaloCellManager::getEtaRanges() {
  std::vector<double> theEtaRanges(etaLim);
  return theEtaRanges;
}