HGCalParameters.cc

Go to the documentation of this file.

#include "Geometry/HGCalCommonData/interface/HGCalParameters.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "Geometry/HGCalCommonData/interface/HGCalWaferIndex.h"

//#define EDM_ML_DEBUG

HGCalParameters::HGCalParameters(const std::string& nam) : name_(nam) {
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("HGCalGeom") << "Construct HGCalParameters for " << name_;
#endif
}

HGCalParameters::~HGCalParameters() {}

void HGCalParameters::fillModule(const HGCalParameters::hgtrap& mytr,
                                 bool reco) {
  if (reco) {
    moduleLayR_.emplace_back(mytr.lay);
    moduleBlR_.emplace_back(mytr.bl);
    moduleTlR_.emplace_back(mytr.tl);
    moduleHR_.emplace_back(mytr.h);
    moduleDzR_.emplace_back(mytr.dz);
    moduleAlphaR_.emplace_back(mytr.alpha);
    moduleCellR_.emplace_back(mytr.cellSize);
  } else {
    moduleLayS_.emplace_back(mytr.lay);
    moduleBlS_.emplace_back(mytr.bl);
    moduleTlS_.emplace_back(mytr.tl);
    moduleHS_.emplace_back(mytr.h);
    moduleDzS_.emplace_back(mytr.dz);
    moduleAlphaS_.emplace_back(mytr.alpha);
    moduleCellS_.emplace_back(mytr.cellSize);
  }
}

HGCalParameters::hgtrap HGCalParameters::getModule(unsigned int k,
                                                   bool reco) const {
  HGCalParameters::hgtrap mytr;
  if (reco) {
    if (k < moduleLayR_.size()) {
      mytr.lay = moduleLayR_[k];
      mytr.bl = moduleBlR_[k];
      mytr.tl = moduleTlR_[k];
      mytr.h = moduleHR_[k];
      mytr.dz = moduleDzR_[k];
      mytr.alpha = moduleAlphaR_[k];
      mytr.cellSize = moduleCellR_[k];
    } else {
      mytr.lay = -1;
      mytr.bl = mytr.tl = mytr.h = mytr.dz = mytr.alpha = mytr.cellSize = 0;
    }
  } else {
    if (k < moduleLayS_.size()) {
      mytr.lay = moduleLayS_[k];
      mytr.bl = moduleBlS_[k];
      mytr.tl = moduleTlS_[k];
      mytr.h = moduleHS_[k];
      mytr.dz = moduleDzS_[k];
      mytr.alpha = moduleAlphaS_[k];
      mytr.cellSize = moduleCellS_[k];
    } else {
      mytr.lay = -1;
      mytr.bl = mytr.tl = mytr.h = mytr.dz = mytr.alpha = mytr.cellSize = 0;
    }
  }
  return mytr;
}

void HGCalParameters::fillTrForm(const HGCalParameters::hgtrform& mytr) {
  int zp = (mytr.zp == 1) ? 1 : 0;
  uint32_t indx = ((zp & kMaskZside) << kShiftZside);
  indx |= ((mytr.lay & kMaskLayer) << kShiftLayer);
  indx |= ((mytr.sec & kMaskSector) << kShiftSector);
  indx |= ((mytr.subsec & kMaskSubSec) << kShiftSubSec);
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("HGCalGeom")
      << "ZP " << zp << ":" << kMaskZside << ":" << kShiftZside
      << ((zp & kMaskZside) << kShiftZside) << " Lay " << mytr.lay << ":"
      << kMaskLayer << ":" << kShiftLayer << ":"
      << ((mytr.lay & kMaskLayer) << kShiftLayer) << " Sector " << mytr.sec
      << ":" << kMaskSector << ":" << kShiftSector << ":"
      << ((mytr.sec & kMaskSector) << kShiftSector) << " SubSec " << mytr.subsec
      << ":" << kMaskSubSec << ":" << kShiftSubSec << ":"
      << ((mytr.subsec & kMaskSubSec) << kShiftSubSec) << " Index " << std::hex
      << indx << std::dec;
#endif
  trformIndex_.emplace_back(indx);
  trformTranX_.emplace_back(mytr.h3v.x());
  trformTranY_.emplace_back(mytr.h3v.y());
  trformTranZ_.emplace_back(mytr.h3v.z());
  trformRotXX_.emplace_back(mytr.hr.xx());
  trformRotYX_.emplace_back(mytr.hr.yx());
  trformRotZX_.emplace_back(mytr.hr.zx());
  trformRotXY_.emplace_back(mytr.hr.xy());
  trformRotYY_.emplace_back(mytr.hr.yy());
  trformRotZY_.emplace_back(mytr.hr.zy());
  trformRotXZ_.emplace_back(mytr.hr.xz());
  trformRotYZ_.emplace_back(mytr.hr.yz());
  trformRotZZ_.emplace_back(mytr.hr.zz());
#ifdef EDM_ML_DEBUG
  unsigned int k = trformIndex_.size() - 1;
  edm::LogVerbatim("HGCalGeom")
      << "HGCalParameters[" << k << "] Index " << std::hex << trformIndex_[k]
      << std::dec << " (" << mytr.zp << ", " << mytr.lay << ", " << mytr.sec
      << ", " << mytr.subsec << ") Translation (" << trformTranX_[k] << ", "
      << trformTranY_[k] << ", " << trformTranZ_[k] << ") Rotation ("
      << trformRotXX_[k] << ", " << trformRotYX_[k] << ", " << trformRotZX_[k]
      << ", " << trformRotXY_[k] << ", " << trformRotYY_[k] << ", "
      << trformRotZY_[k] << ", " << trformRotXZ_[k] << ", " << trformRotYZ_[k]
      << ", " << trformRotZZ_[k];
#endif
}

HGCalParameters::hgtrform HGCalParameters::getTrForm(unsigned int k) const {
  HGCalParameters::hgtrform mytr;
  if (k < trformIndex_.size()) {
    const auto& id = getID(k);
    mytr.zp = id[0];
    mytr.lay = id[1];
    mytr.sec = id[2];
    mytr.subsec = id[3];
    mytr.h3v =
        CLHEP::Hep3Vector(trformTranX_[k], trformTranY_[k], trformTranZ_[k]);
    const CLHEP::HepRep3x3 rotation(
        trformRotXX_[k], trformRotXY_[k], trformRotXZ_[k], trformRotYX_[k],
        trformRotYY_[k], trformRotYZ_[k], trformRotZX_[k], trformRotZY_[k],
        trformRotZZ_[k]);
    mytr.hr = CLHEP::HepRotation(rotation);
  } else {
    mytr.zp = mytr.lay = mytr.sec = mytr.subsec = 0;
  }
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("HGCalGeom")
      << "HGCalParameters[" << k << "] Index " << std::hex << trformIndex_[k]
      << std::dec << " (" << mytr.zp << ", " << mytr.lay << ", " << mytr.sec
      << ", " << mytr.subsec << ") Translation (" << mytr.h3v.x() << ", "
      << mytr.h3v.y() << ", " << mytr.h3v.z() << ") Rotation (" << mytr.hr.xx()
      << ", " << mytr.hr.yx() << ", " << mytr.hr.zx() << ", " << mytr.hr.xy()
      << ", " << mytr.hr.yy() << ", " << mytr.hr.zy() << ", " << mytr.hr.xz()
      << ", " << mytr.hr.yz() << ", " << mytr.hr.zz();
#endif
  return mytr;
}

void HGCalParameters::addTrForm(const CLHEP::Hep3Vector& h3v) {
  unsigned int k = trformTranX_.size();
  if (k > 0) {
    trformTranX_[k - 1] += h3v.x();
    trformTranY_[k - 1] += h3v.y();
    trformTranZ_[k - 1] += h3v.z();
  }
}

void HGCalParameters::scaleTrForm(double scale) {
  unsigned int k = trformTranX_.size();
  if (k > 0) {
    trformTranX_[k - 1] *= scale;
    trformTranY_[k - 1] *= scale;
    trformTranZ_[k - 1] *= scale;
  }
}

std::array<int, 4> HGCalParameters::getID(unsigned int k) const {
  int zp = ((trformIndex_[k] >> kShiftZside) & kMaskZside);
  if (zp != 1) zp = -1;
  int lay = ((trformIndex_[k] >> kShiftLayer) & kMaskLayer);
  int sec = ((trformIndex_[k] >> kShiftSector) & kMaskSector);
  int subsec = ((trformIndex_[k] >> kShiftSubSec) & kMaskSubSec);
  return std::array<int, 4>{{zp, lay, sec, subsec}};
}

#include "FWCore/Utilities/interface/typelookup.h"

TYPELOOKUP_DATA_REG(HGCalParameters);