CompressedLayerMaterial.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/

// CompressedLayerMaterial.h, (c) ATLAS Detector software
 
#ifndef TRKGEOMETRY_COMPRESSEDLAYERMATERIAL_H
#define TRKGEOMETRY_COMPRESSEDLAYERMATERIAL_H
 
// Trk
#include "TrkDetDescrUtils/BinUtility.h"
#include "TrkGeometry/LayerMaterialProperties.h"
#include "TrkGeometry/MaterialProperties.h"
// Amg
#include "GeoPrimitives/GeoPrimitives.h"
// Gaudi
#include "GaudiKernel/MsgStream.h"
 
namespace Trk {

 
class CompressedLayerMaterial final : public LayerMaterialProperties {
 public:
  CompressedLayerMaterial();

  CompressedLayerMaterial(BinUtility& binutility);

  CompressedLayerMaterial(
      const Trk::BinUtility& binutility,
      const MaterialPropertiesVector& fullProperties,
      const std::vector<unsigned short int>& materialIndices,
      double splitFactor = 0.);
 
  CompressedLayerMaterial(
      std::unique_ptr<Trk::BinUtility> binutility,
      const MaterialPropertiesVector&& fullProperties,
      const std::vector<unsigned short int>& materialIndices,
      double splitFactor = 0.);

  CompressedLayerMaterial(const CompressedLayerMaterial& mprop);

  virtual ~CompressedLayerMaterial() override;

  virtual CompressedLayerMaterial* clone() const override final;

  CompressedLayerMaterial& operator=(const CompressedLayerMaterial& lmp);

  virtual CompressedLayerMaterial& operator*=(double scale) override final;

  virtual const BinUtility* binUtility() const override final;

  virtual void updateBinning(BinUtility* bu) override final;

  const MaterialPropertiesVector& fullMaterial() const;

  const std::vector<unsigned short int>& materialBins() const;

  virtual const MaterialProperties* fullMaterial(
      const Amg::Vector3D& gp) const override;

  virtual const MaterialProperties* material(size_t bin0,
                                             size_t bin1) const override final;

  virtual MsgStream& dump(MsgStream& sl) const override final;
  virtual std::ostream& dump(std::ostream& sl) const override final;
 
 private:
  BinUtility* m_binUtility;  

  MaterialPropertiesVector m_fullMaterial;
  std::vector<unsigned short int> m_materialBins;

  void clearMaterial();

  void fillMaterial(const MaterialPropertiesVector& matVector);
};
 
inline const BinUtility* CompressedLayerMaterial::binUtility() const {
  return m_binUtility;
}
 
inline const MaterialPropertiesVector& CompressedLayerMaterial::fullMaterial()
    const {
  return m_fullMaterial;
}
 
inline const std::vector<unsigned short int>&
CompressedLayerMaterial::materialBins() const {
  return m_materialBins;
}
 
inline const MaterialProperties* CompressedLayerMaterial::material(
    size_t bin0, size_t bin1) const {
  int accessBin = bin1 * (m_binUtility->max(0) + 1) + bin0;
  // safety check
  if (bin0 <= (unsigned int)m_binUtility->max(0) &&
      bin1 <= (unsigned int)m_binUtility->max(1) && !m_fullMaterial.empty() &&
      accessBin < int(m_materialBins.size()))
    return m_fullMaterial[m_materialBins[accessBin]];
  return nullptr;
}
 
inline void CompressedLayerMaterial::updateBinning(BinUtility* bu) {
  if (bu) {
    delete m_binUtility;
    m_binUtility = bu;
  }
}
 
}  // namespace Trk
 
#endif