BinnedMaterial.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrkGeometry/BinnedMaterial.h"
 
Trk::BinnedMaterial::BinnedMaterial(
    const Trk::Material* mat, const Trk::BinUtility& bu,
    const std::vector<size_t>& index,
    const std::vector<Trk::IdentifiedMaterial>& detailedMat)
    : Trk::Material(*mat),
      m_matVec(detailedMat),
      m_matBins(std::make_unique<Trk::CompactBinnedArray1D<const Trk::IdentifiedMaterial> >(ptrs(), index, bu))
     {}
 
Trk::BinnedMaterial::BinnedMaterial(
    const Trk::Material* mat, const Trk::BinUtility& bu,
    const std::vector<Trk::BinUtility>& bVec,
    const std::vector<std::vector<size_t> >& index,
    const std::vector<Trk::IdentifiedMaterial>& detailedMat)
    : Trk::Material(*mat),
      m_matVec(detailedMat),
      m_matBins(std::make_unique<Trk::CompactBinnedArray2D<const Trk::IdentifiedMaterial> >(ptrs(), index, bu, bVec))
     {}
 
//copy
Trk::BinnedMaterial::BinnedMaterial(const BinnedMaterial& amc)
    : Material(amc), m_matVec(amc.m_matVec) {
  if (amc.m_matBins) {
    if (!m_matVec.empty()) {
      m_matBins = binsPtr_t(amc.m_matBins->clone(ptrs()));
    } else {
      m_matBins = binsPtr_t(amc.m_matBins->clone());
    }
  }
}
 
Trk::BinnedMaterial& Trk::BinnedMaterial::operator=(const BinnedMaterial& amc) {
  if (this != &amc) {
    Material::operator=(amc);
    m_matVec = amc.m_matVec;
    m_matBins.reset();
    if (amc.m_matBins) {
      if (!m_matVec.empty()) {
        m_matBins = binsPtr_t(amc.m_matBins->clone(ptrs()));
      } else {
        m_matBins = binsPtr_t(amc.m_matBins->clone());
      }
    }
  }
  return (*this);
}
 
const Trk::IdentifiedMaterial* Trk::BinnedMaterial::material(
    const Amg::Vector3D& position) const {
  const Trk::IdentifiedMaterial* mat =
      m_matBins ? m_matBins->object(position) : nullptr;
  return mat;
}
 
const Trk::IdentifiedMaterial* Trk::BinnedMaterial::materialNext(
    const Amg::Vector3D& position, const Amg::Vector3D& direction,
    bool layOnly) const {
  const Trk::IdentifiedMaterial* mat =
      m_matBins ? m_matBins->nextObject(position, direction, layOnly) : nullptr;
  return mat;
}
 
//utility to return pointers to the elements of m_matVec
std::vector<const Trk::IdentifiedMaterial*> Trk::BinnedMaterial::ptrs() const {
  std::vector<const Trk::IdentifiedMaterial*> p;
  p.reserve(m_matVec.size());
  for (const Trk::IdentifiedMaterial& m : m_matVec) {
    p.push_back(&m);
  }
  return p;
}