Materials.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2019 Belle II Collaboration                          *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
#include <framework/core/MRUCache.h>
 
#include <optional>
#include <set>
#include <string>
#include <vector>
 
class G4Material;
class G4Element;
class G4MaterialPropertiesTable;
class G4OpticalSurface;
class G4NistElementBuilder;
class G4NistMaterialBuilder;
 
namespace Belle2 {
  namespace gearbox { class Interface; }
 
  class GeoMaterial;
  class GeoOpticalSurface;
  class GeoMaterialProperty;
 
  namespace geometry {
    class Materials {
    public:
      enum { CacheSize = 100 };
 
      static Materials& getInstance();
 
      ~Materials();
 
      static G4Material* get(const std::string& name) { return Materials::getInstance().getMaterial(name); }
 
      G4Material* getMaterial(const std::string& name, bool showErrors = true);
 
      G4Element* getElement(const std::string& name);
 
      G4Material* createMaterial(const gearbox::Interface& parameters);
 
      G4Material* createMaterial(const GeoMaterial& parameters);
 
      G4OpticalSurface* createOpticalSurface(const gearbox::Interface& parameters);
 
      G4OpticalSurface* createOpticalSurface(const GeoOpticalSurface& surface);
 
      GeoMaterial createMaterialConfig(const gearbox::Interface& parameters);
 
      GeoOpticalSurface createOpticalSurfaceConfig(const gearbox::Interface& parameters);
 
      void clear();
 
      void setDensityScale(double scale)
      {
        if (scale == 1.0) return resetDensityScale();
        m_densityScale = scale;
      }
      void resetDensityScale()
      {
        m_densityScale.reset();
      }
      std::set<std::string> getDensityScaleIgnoredMaterials() const { return m_ignoreScaling; }
      void setDensityScaleIgnoredMaterials(const std::set<std::string>& ignored) { m_ignoreScaling = ignored; }
    protected:
      Materials(): m_materialCache(CacheSize) {};
      Materials(const Materials&) = delete;
      void operator=(const Materials&) = delete;
      G4Material* findMaterial(const std::string& name);
 
      mutable MRUCache<std::string, G4Material*> m_materialCache;
 
      std::vector<G4MaterialPropertiesTable*> m_PropTables;
 
      G4NistElementBuilder* m_nistElementBuilder{nullptr};
      G4NistMaterialBuilder* m_nistMaterialBuilder{nullptr};
 
      G4MaterialPropertiesTable* createProperties(const std::vector<GeoMaterialProperty>& props);
 
      void initBuilders();
 
      std::optional<double> m_densityScale;
      std::set<std::string> m_ignoreScaling{"G4_AIR", "Air", "Vacuum"};
    };
  } //geometry namespace
} //Belle2 namespace