Geometry/CaloGeometry/src/FlatTrd.cc - master - githubmirror/cmssw

Geometry/CaloGeometry/src/FlatTrd.cc
260 строк10 KB
Sunanda
Code check
01 май 2020, 15:43
01 май 2020, 15:43e6949d1
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "Geometry/CaloGeometry/interface/FlatTrd.h"
#include "Geometry/CaloGeometry/interface/TruncatedPyramid.h"
#include <algorithm>
#include <iostream>

//#define EDM_ML_DEBUG

typedef FlatTrd::CCGFloat CCGFloat;
typedef FlatTrd::Pt3D Pt3D;
typedef FlatTrd::Pt3DVec Pt3DVec;
typedef FlatTrd::Tr3D Tr3D;

typedef HepGeom::Vector3D<CCGFloat> FVec3D;
typedef HepGeom::Plane3D<CCGFloat> Plane3D;

typedef HepGeom::Vector3D<double> DVec3D;
typedef HepGeom::Plane3D<double> DPlane3D;
typedef HepGeom::Point3D<double> DPt3D;

static const float tolmin = 1.e-12;
//----------------------------------------------------------------------

FlatTrd::FlatTrd()
    : CaloCellGeometry(), m_axis(0., 0., 0.), m_corOne(0., 0., 0.), m_local(0., 0., 0.), m_global(0., 0., 0.) {}

FlatTrd::FlatTrd(const FlatTrd& tr) : CaloCellGeometry(tr) { *this = tr; }

FlatTrd& FlatTrd::operator=(const FlatTrd& tr) {
  CaloCellGeometry::operator=(tr);
  if (this != &tr) {
    m_axis = tr.m_axis;
    m_corOne = tr.m_corOne;
    m_local = tr.m_local;
    m_global = tr.m_global;
    m_tr = tr.m_tr;
  }
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("CaloGeometry") << "FlatTrd(Copy): Local " << m_local << " Global " << m_global << " eta "
                                   << etaPos() << " phi " << phiPos() << " Translation " << m_tr.getTranslation()
                                   << " and rotation " << m_tr.getRotation();
#endif
  return *this;
}

FlatTrd::FlatTrd(
    CornersMgr* cMgr, const GlobalPoint& fCtr, const GlobalPoint& bCtr, const GlobalPoint& cor1, const CCGFloat* parV)
    : CaloCellGeometry(fCtr, cMgr, parV),
      m_axis((bCtr - fCtr).unit()),
      m_corOne(cor1.x(), cor1.y(), cor1.z()),
      m_local(0., 0., 0.) {
  getTransform(m_tr, nullptr);
  Pt3D glb = m_tr * m_local;
  m_global = GlobalPoint(glb.x(), glb.y(), glb.z());
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("CaloGeometry") << "FlatTrd: Local " << m_local << " Global " << glb << " eta " << etaPos()
                                   << " phi " << phiPos() << " Translation " << m_tr.getTranslation()
                                   << " and rotation " << m_tr.getRotation();
#endif
}

FlatTrd::FlatTrd(const CornersVec& corn, const CCGFloat* par)
    : CaloCellGeometry(corn, par), m_corOne(corn[0].x(), corn[0].y(), corn[0].z()), m_local(0., 0., 0.) {
  getTransform(m_tr, nullptr);
  m_axis = makeAxis();
  Pt3D glb = m_tr * m_local;
  m_global = GlobalPoint(glb.x(), glb.y(), glb.z());
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("CaloGeometry") << "FlatTrd: Local " << m_local << " Global " << glb << " eta " << etaPos()
                                   << " phi " << phiPos() << " Translation " << m_tr.getTranslation()
                                   << " and rotation " << m_tr.getRotation();
#endif
}

FlatTrd::FlatTrd(const FlatTrd& tr, const Pt3D& local) : CaloCellGeometry(tr), m_local(local) {
  *this = tr;
  Pt3D glb = m_tr * m_local;
  m_global = GlobalPoint(glb.x(), glb.y(), glb.z());
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("CaloGeometry") << "FlatTrd: Local " << m_local << " Global " << glb << " eta " << etaPos()
                                   << " phi " << phiPos() << " Translation " << m_tr.getTranslation()
                                   << " and rotation " << m_tr.getRotation();
#endif
}

FlatTrd::~FlatTrd() {}

GlobalPoint FlatTrd::getPosition(const Pt3D& local) const {
  Pt3D glb = m_tr * local;
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("CaloGeometry") << "FlatTrd::Local " << local.x() << ":" << local.y() << ":" << local.z()
                                   << " Global " << glb.x() << ":" << glb.y() << ":" << glb.z() << " TR " << m_tr.xx()
                                   << ":" << m_tr.xy() << ":" << m_tr.xz() << ":" << m_tr.yx() << ":" << m_tr.yy()
                                   << ":" << m_tr.yz() << ":" << m_tr.zx() << ":" << m_tr.zy() << ":" << m_tr.zz()
                                   << ":" << m_tr.dx() << ":" << m_tr.dy() << ":" << m_tr.dz();
#endif
  return GlobalPoint(glb.x(), glb.y(), glb.z());
}

float FlatTrd::etaSpan() const {
  assert(param() != nullptr);
  float eta1 = -std::log(
      std::tan(0.5 * std::atan((m_global.perp() + param()[k_dY1]) / std::max(tolmin, std::abs(m_global.z())))));
  float eta2 = -std::log(
      std::tan(0.5 * std::atan((m_global.perp() - param()[k_dY1]) / std::max(tolmin, std::abs(m_global.z())))));
  float dEta = std::abs(eta1 - eta2);
  return dEta;
}

float FlatTrd::phiSpan() const {
  assert(param() != nullptr);
  float dPhi = 2.0 * std::atan(0.5 * (param()[k_dX1] + param()[k_dX2]) / std::max(tolmin, m_global.perp()));
  return dPhi;
}

Pt3D FlatTrd::getLocal(const GlobalPoint& global) const {
  Pt3D local = m_tr.inverse() * Pt3D(global.x(), global.y(), global.z());
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("CaloGeometry") << "FlatTrd::Global " << global.x() << ":" << global.y() << ":" << global.z()
                                   << " Local " << local.x() << ":" << local.y() << ":" << local.z();
#endif
  return local;
}

CCGFloat FlatTrd::getThetaAxis() const { return m_axis.theta(); }

CCGFloat FlatTrd::getPhiAxis() const { return m_axis.phi(); }

const GlobalVector& FlatTrd::axis() const { return m_axis; }

void FlatTrd::vocalCorners(Pt3DVec& vec, const CCGFloat* pv, Pt3D& ref) const { localCorners(vec, pv, ref); }

void FlatTrd::createCorners(const std::vector<CCGFloat>& pv, const Tr3D& tr, std::vector<GlobalPoint>& co) {
  assert(11 <= pv.size());
  assert(ncorner_ == co.size());

  Pt3DVec ko(ncorner_, Pt3D(0, 0, 0));

  Pt3D tmp;
  Pt3DVec to(ncorner_, Pt3D(0, 0, 0));
  localCorners(to, &pv.front(), tmp);

  for (unsigned int i(0); i != ncorner_; ++i) {
    ko[i] = tr * to[i];  // apply transformation
    const Pt3D& p(ko[i]);
    co[i] = GlobalPoint(p.x(), p.y(), p.z());
#ifdef EDM_ML_DEBUG
    edm::LogVerbatim("CaloGeometry") << "Corner[" << i << "] = " << co[i];
#endif
  }
}

void FlatTrd::localCorners(Pt3DVec& lc, const CCGFloat* pv, Pt3D& ref) {
  assert(nullptr != pv);
  assert(ncorner_ == lc.size());

  const CCGFloat dz(pv[FlatTrd::k_dZ]);
  const CCGFloat h(pv[FlatTrd::k_dY1]);
  const CCGFloat bl(pv[FlatTrd::k_dX1]);
  const CCGFloat tl(pv[FlatTrd::k_dX2]);
  const CCGFloat a1(pv[FlatTrd::k_Alp1]);

  const CCGFloat ta1(tan(a1));

  lc[0] = Pt3D(-h * ta1 - bl, -h, -dz);  // (-,-,-)
  lc[1] = Pt3D(+h * ta1 - tl, +h, -dz);  // (-,+,-)
  lc[2] = Pt3D(+h * ta1 + tl, +h, -dz);  // (+,+,-)
  lc[3] = Pt3D(-h * ta1 + bl, -h, -dz);  // (+,-,-)
  lc[4] = Pt3D(-h * ta1 - bl, -h, dz);   // (-,-,+)
  lc[5] = Pt3D(+h * ta1 - tl, +h, dz);   // (-,+,+)
  lc[6] = Pt3D(+h * ta1 + tl, +h, dz);   // (+,+,+)
  lc[7] = Pt3D(-h * ta1 + bl, -h, dz);   // (+,-,+)

  ref = 0.25 * (lc[0] + lc[1] + lc[2] + lc[3]);
#ifdef EDM_ML_DEBUG
  edm::LogVerbatim("CaloGeometry") << "Ref " << ref << " Local Corners " << lc[0] << "|" << lc[1] << "|" << lc[2] << "|"
                                   << lc[3] << "|" << lc[4] << "|" << lc[5] << "|" << lc[6] << "|" << lc[7];
#endif
}

void FlatTrd::getTransform(Tr3D& tr, Pt3DVec* lptr) const {
  const GlobalPoint& p(CaloCellGeometry::getPosition());
  const Pt3D gFront(p.x(), p.y(), p.z());
  const DPt3D dgFront(p.x(), p.y(), p.z());

  Pt3D lFront;
  assert(nullptr != param());
  std::vector<Pt3D> lc(ncorner_, Pt3D(0, 0, 0));
  localCorners(lc, param(), lFront);

  // figure out if reflction volume or not

  Pt3D lBack(0.25 * (lc[4] + lc[5] + lc[6] + lc[7]));

  const DPt3D dlFront(lFront.x(), lFront.y(), lFront.z());
  const DPt3D dlBack(lBack.x(), lBack.y(), lBack.z());
  const DPt3D dlOne(lc[0].x(), lc[0].y(), lc[0].z());

  const FVec3D dgAxis(axis().x(), axis().y(), axis().z());

  const DPt3D dmOne(m_corOne.x(), m_corOne.y(), m_corOne.z());

  const DPt3D dgBack(dgFront + (dlBack - dlFront).mag() * dgAxis);
  DPt3D dgOne(dgFront + (dlOne - dlFront).mag() * (dmOne - dgFront).unit());

  const double dlangle((dlBack - dlFront).angle(dlOne - dlFront));
  const double dgangle((dgBack - dgFront).angle(dgOne - dgFront));
  const double dangle(dlangle - dgangle);

  if (1.e-6 < fabs(dangle)) {  //guard against precision problems
    const DPlane3D dgPl(dgFront, dgOne, dgBack);
    const DPt3D dp2(dgFront + dgPl.normal().unit());

    DPt3D dgOld(dgOne);

    dgOne = (dgFront + HepGeom::Rotate3D(-dangle, dgFront, dp2) * DVec3D(dgOld - dgFront));
  }

  tr = Tr3D(dlFront, dlBack, dlOne, dgFront, dgBack, dgOne);

  if (nullptr != lptr)
    (*lptr) = lc;
}

void FlatTrd::initCorners(CaloCellGeometry::CornersVec& co) {
  if (co.uninitialized()) {
    CornersVec& corners(co);
    Pt3DVec lc;
    Tr3D tr;
    getTransform(tr, &lc);

    for (unsigned int i(0); i != ncorner_; ++i) {
      const Pt3D corn(tr * lc[i]);
      corners[i] = GlobalPoint(corn.x(), corn.y(), corn.z());
    }
  }
}

GlobalVector FlatTrd::makeAxis() { return GlobalVector(backCtr() - getPosition()).unit(); }

GlobalPoint FlatTrd::backCtr() const {
  float dz = (getCorners()[ncornerBy2_].z() > getCorners()[0].z()) ? param()[FlatTrd::k_dZ] : -param()[FlatTrd::k_dZ];
  Pt3D local_b(m_local.x(), m_local.y(), m_local.z() + dz);
  Pt3D global_b = m_tr * local_b;
  GlobalPoint global(global_b.x(), global_b.y(), global_b.z());
  return global;
}

//----------------------------------------------------------------------
//----------------------------------------------------------------------

std::ostream& operator<<(std::ostream& s, const FlatTrd& cell) {
  s << "Center: " << cell.getPosition() << " eta " << cell.etaPos() << " phi " << cell.phiPos() << std::endl;
  s << "Axis: " << cell.getThetaAxis() << " " << cell.getPhiAxis() << std::endl;
  const CaloCellGeometry::CornersVec& corners(cell.getCorners());
  for (unsigned int i = 0; i != corners.size(); ++i) {
    s << "Corner: " << corners[i] << std::endl;
  }
  return s;
}