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

Geometry/CaloGeometry/src/CaloCellGeometry.cc
157 строк5 KB
Sunanda
Code check
27 апр 2020, 18:59
27 апр 2020, 18:59a31914b
#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h"
#include <CLHEP/Geometry/Plane3D.h>

typedef CaloCellGeometry::CCGFloat CCGFloat;
typedef CaloCellGeometry::Pt3D Pt3D;
typedef CaloCellGeometry::Pt3DVec Pt3DVec;
typedef CaloCellGeometry::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;
typedef std::vector<DPt3D> DPt3DVec;

const float CaloCellGeometry::k_ScaleFromDDDtoGeant(0.1);

CaloCellGeometry::CaloCellGeometry()
    : m_refPoint(0., 0., 0.), m_corners(), m_parms((CCGFloat*)nullptr), m_dEta(0), m_dPhi(0) {}

CaloCellGeometry::~CaloCellGeometry() {}

CaloCellGeometry::CaloCellGeometry(CornersVec::const_reference gp, CornersMgr* mgr, const CCGFloat* par)
    : m_refPoint(gp), m_corners(mgr), m_parms(par), m_rep(gp.perp(), gp.eta(), gp.barePhi()), m_dEta(0), m_dPhi(0) {}

CaloCellGeometry::CaloCellGeometry(const CornersVec& cv, const CCGFloat* par)
    : m_refPoint(0.25 * (cv[0].x() + cv[1].x() + cv[2].x() + cv[3].x()),
                 0.25 * (cv[0].y() + cv[1].y() + cv[2].y() + cv[3].y()),
                 0.25 * (cv[0].z() + cv[1].z() + cv[2].z() + cv[3].z())),
      m_corners(cv),
      m_parms(par),
      m_rep(m_refPoint.perp(), m_refPoint.eta(), m_refPoint.barePhi()),
      m_dEta(0),
      m_dPhi(0) {}

std::ostream& operator<<(std::ostream& s, const CaloCellGeometry& cell) {
  s << ", Center: " << cell.getPosition() << std::endl;

  if (cell.emptyCorners()) {
    s << "Corners vector is empty." << std::endl;
  } else {
    const CaloCellGeometry::CornersVec& corners(cell.getCorners());
    for (unsigned int i(0); i != corners.size(); ++i) {
      s << "Corner: " << corners[i] << std::endl;
    }
  }
  return s;
}

void CaloCellGeometry::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());
  Pt3DVec lc(8, Pt3D(0, 0, 0));
  vocalCorners(lc, param(), lFront);

  DPt3D dlFront(lFront.x(), lFront.y(), lFront.z());

  const Pt3D lBack(0.25 * (lc[4] + lc[5] + lc[6] + lc[7]));
  const DPt3D dlBack(lBack.x(), lBack.y(), lBack.z());

  const Pt3D dlOne(lc[0].x(), lc[0].y(), lc[0].z());

  const CornersVec& cor(getCorners());
  DPt3DVec dkor(8, DPt3D(0, 0, 0));
  for (unsigned int i(0); i != 8; ++i) {
    dkor[i] = DPt3D(cor[i].x(), cor[i].y(), cor[i].z());
  }

  DPt3D dgBack(0.25 * (dkor[4] + dkor[5] + dkor[6] + dkor[7]));

  const DVec3D dgAxis((dgBack - dgFront).unit());

  dgBack = (dgFront + (dlBack - dlFront).mag() * dgAxis);
  const DPt3D dgOneT(dgFront + (dlOne - dlFront).mag() * (dkor[0] - dgFront).unit());

  const double dlangle((dlBack - dlFront).angle(dlOne - dlFront));
  const double dgangle((dgBack - dgFront).angle(dgOneT - dgFront));
  const double ddangle(dlangle - dgangle);

  const DPlane3D dgPl(dgFront, dgOneT, dgBack);
  const DPt3D dp2(dgFront + dgPl.normal().unit());

  const DPt3D dgOne(dgFront + HepGeom::Rotate3D(-ddangle, dgFront, dp2) * DVec3D(dgOneT - dgFront));

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

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

const float* CaloCellGeometry::checkParmPtr(const std::vector<float>& vv, CaloCellGeometry::ParVecVec& pvv) {
  const float* pP(nullptr);

  for (unsigned int ii(0); ii != pvv.size(); ++ii) {
    const ParVec& v(pvv[ii]);
    assert(v.size() == vv.size());

    bool same(true);
    for (unsigned int j(0); j != vv.size(); ++j) {
      same = same && (fabs(vv[j] - v[j]) < 1.e-6);
      if (!same)
        break;
    }
    if (same) {
      pP = &(*v.begin());
      break;
    }
  }
  return pP;
}

const float* CaloCellGeometry::getParmPtr(const std::vector<float>& vv,
                                          CaloCellGeometry::ParMgr* mgr,
                                          CaloCellGeometry::ParVecVec& pvv) {
  const float* pP(checkParmPtr(vv, pvv));

  if (nullptr == pP) {
    pvv.emplace_back(ParVec(mgr));
    ParVec& back(pvv.back());
    for (unsigned int i(0); i != vv.size(); ++i) {
      back[i] = vv[i];
    }
    pP = &(*back.begin());
  }
  return pP;
}

bool CaloCellGeometry::inside(const GlobalPoint& point) const {
  bool ans(false);
  const Pt3D p(point.x(), point.y(), point.z());
  const CornersVec& cog(getCorners());
  Pt3D co[8];
  for (unsigned int i(0); i != 8; ++i) {
    co[i] = Pt3D(cog[i].x(), cog[i].y(), cog[i].z());
  }

  const Plane3D AA(co[0], co[1], co[2]);  // z<0
  const Plane3D BB(co[6], co[5], co[4]);  // z>0

  if (AA.distance(p) * BB.distance(p) >= 0) {
    const Plane3D CC(co[0], co[4], co[5]);  // x<0
    const Plane3D DD(co[2], co[6], co[7]);  // x>0
    if (CC.distance(p) * DD.distance(p) >= 0) {
      const Plane3D EE(co[3], co[7], co[4]);  // y<0
      const Plane3D FF(co[1], co[5], co[6]);  // y>0
      if (EE.distance(p) * FF.distance(p) >= 0) {
        ans = true;
      }
    }
  }
  return ans;
}