GeometryManager.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef GEOMETRYMANAGER_H
#define GEOMETRYMANAGER_H
 
#include <vector>
#include <set>
#include <string>
#include <memory>
#include <map>
 
class G4VPhysicalVolume;
class G4VisAttributes;
 
namespace Belle2 {
  class GearDir;
  class GeoConfiguration;
  class IntervalOfValidity;
 
  namespace geometry {
    class CreatorBase;
 
    enum GeometryTypes {
      FullGeometry,     
      TrackingGeometry, 
      DisplayGeometry   
    };
 
    class GeometryManager {
    public:
      static GeometryManager& getInstance();
 
      G4VPhysicalVolume* getTopVolume() { return m_topVolume; }
 
      void createGeometry(const GearDir& params, GeometryTypes type = FullGeometry);
 
      void createGeometry(const GeoConfiguration& params, GeometryTypes type = FullGeometry, bool useDB = true);
 
      GeoConfiguration createGeometryConfig(const GearDir& detectorDir, const IntervalOfValidity& iov);
 
      void setDetectorComponents(const std::vector<std::string>& components)
      {
        m_components.clear();
        m_components.insert(components.begin(), components.end());
      }
 
      void setExcludedComponents(const std::vector<std::string>& components)
      {
        m_excluded.clear();
        m_excluded.insert(components.begin(), components.end());
      }
 
      void setAdditionalComponents(const std::vector<std::string>& components)
      {
        m_additional.clear();
        m_additional.insert(components.begin(), components.end());
      }
 
      void setDensityScaling(const std::map<std::string, double>& components)
      {
        m_densityScaling = components;
      }
 
      void setAssignRegions(bool assignRegions)
      {
        m_assignRegions = assignRegions;
      }
 
      void clear();
 
      void createTGeoRepresentation();
 
      G4VisAttributes* newVisAttributes();
 
    private:
      GeometryManager(): m_topVolume(0) {};
      GeometryManager(const GeometryManager&) = delete;
      GeometryManager& operator=(const GeometryManager&) = delete;
      ~GeometryManager() {}
 
      G4VPhysicalVolume* m_topVolume;
      std::vector<CreatorBase*> m_creators;
      std::set<std::string> m_components;
      std::set<std::string> m_excluded;
      std::set<std::string> m_additional;
      std::map<std::string, double> m_densityScaling;
      bool m_assignRegions {false};
      std::vector<G4VisAttributes*> m_VisAttributes;
      friend struct std::default_delete<GeometryManager>;
    };
 
  }
 
} //Belle2 namespace
#endif //GEOMETRYMANAGER_H