LArRecoMaterialTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// ***************************************************************************
//
// 3.8.2005 Creation of the tool by claire.bourdarios@cern.ch
//
// ***************************************************************************
 
#ifndef LARDETDESCR_LARRECOMATERIALTOOL_H
#define LARDETDESCR_LARRECOMATERIALTOOL_H
 
#include "CaloDetDescr/ICaloRecoMaterialTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
class CaloCell_ID;
class CaloDM_ID;
 
#include "GeoModelKernel/GeoVPhysVol.h"
#include "GeoModelKernel/GeoLogVol.h"
 
#include <string>
 
class LArRecoMaterialTool : public extends<AthAlgTool, ICaloRecoMaterialTool>
{
public:
  
  // Constructor:
  LArRecoMaterialTool(const std::string& type, 
              const std::string& name,
              const IInterface* parent);
 
  ~LArRecoMaterialTool() = default;
  virtual StatusCode initialize() override;
 
  virtual
  bool get_material (CaloSubdetNames::ALIGNVOL alvol, 
             double& mass, double& volume,
             double& x0, double& dEdx,
             double& aveA, double& aveZ
             ) const  override;
 
  virtual
  void get_default_material (double& density,
                 double& x0, double& dEdx,
                 double& aveA, double& aveZ
                 ) const override;
 
  void print() const override;
 
private:
 
  bool  m_parseGeoModelForMaterial = false;
 
  void ScanTree(); 
 
  bool ScanCRYO(CaloSubdetNames::ALIGNVOL  alvol,
        double& mass, double& x0, double& dEdx,
        double& aveA, double& aveZ); 
 
  bool ScanPS( CaloSubdetNames::ALIGNVOL alvol,
           double& mass, double& x0, double& dEdx,
           double& aveA, double& aveZ); 
 
  bool ScanEMB( CaloSubdetNames::ALIGNVOL alvol,
        double& mass, double& x0, double& dEdx,
        double& aveA, double& aveZ); 
 
  bool ScanEMEC( CaloSubdetNames::ALIGNVOL alvol,
        double& mass, double& x0, double& dEdx,
        double& aveA, double& aveZ); 
 
  bool ScanHEC( CaloSubdetNames::ALIGNVOL alvol,
        double& mass, double& x0, double& dEdx,
        double& aveA, double& aveZ); 
 
  bool ScanFCAL( CaloSubdetNames::ALIGNVOL alvol,
        double& mass, double& x0, double& dEdx,
        double& aveA, double& aveZ); 
 
  void blendGeoVolume(GeoPVConstLink& geoVolLink,
              std::vector<double>& volume,
              std::vector<double>& mass,
              std::vector<double>& x0,
              std::vector<double>& dEdX,
              std::vector<double>& aveA,
              std::vector<double>& aveZ) const;
                          
  void addMaterialFraction(const GeoLogVol& geoVol,
               std::vector<double>& volume, 
               std::vector<double>& mass,
               std::vector<double>& x0,
               std::vector<double>& dEdX,
               std::vector<double>& aveA,
               std::vector<double>& aveZ) const;
 
  void averageFraction (std::vector<double>& volumeFractions, 
            std::vector<double>& massFractions,
            std::vector<double>& x0Fractions,
            std::vector<double>& dEdXFractions,
            std::vector<double>& aveAFractions,
            std::vector<double>& aveZFractions,
            double& mass, double& volume, 
            double& x0, double& dEdx,
            double& aveA, double& aveZ, double& rho );
 
  // mapping between ALIGNVOL and strings (need to retrieve StoredPV-s)
  std::string map_av(CaloSubdetNames::ALIGNVOL alvol);
 
 
  const CaloCell_ID*        m_calo_id{nullptr};
  const CaloDM_ID*          m_dm_id{nullptr};
 
  // Store results per ALIGNVOL :
  std::vector<CaloSubdetNames::ALIGNVOL>  m_alignvol_number;
  std::vector<double> m_alignvol_mass;
  std::vector<double> m_alignvol_x0;
  std::vector<double> m_alignvol_dEdX;
  std::vector<double> m_alignvol_aveA;
  std::vector<double> m_alignvol_aveZ;
 
  // intermediate vectors, cleared and filled within several methods :     
  std::vector<double>   m_massFractions;
  std::vector<double>   m_volumeFractions;
  std::vector<double>   m_x0Fractions;
  std::vector<double>   m_dEdXFractions;
  std::vector<double>   m_averageAFractions;
  std::vector<double>   m_averageZFractions;
 
  std::vector<double>   m_child_massFractions;
  std::vector<double>   m_child_volumeFractions;
  std::vector<double>   m_child_x0Fractions;
  std::vector<double>   m_child_dEdXFractions;
  std::vector<double>   m_child_averageAFractions;
  std::vector<double>   m_child_averageZFractions;
  std::vector<double>   m_child_Rho;
  std::vector<double>   m_child_Volume;
  std::vector<double>   m_child_Mass;
 
};
 
#endif // LARDETDESCR_LARMATERIALTOOL_H