DetectorDescription/DDCMS/interface/DDNamespace.h - master - githubmirror/cmssw

DetectorDescription/DDCMS/interface/DDNamespace.h
94 строки3 KB
Carl Vuosalo
DD4hep geometry DB payload creator
23 июл 2021, 23:37
23 июл 2021, 23:370ec07f2
#ifndef DETECTOR_DESCRIPTION_DD_NAMESPACE_H
#define DETECTOR_DESCRIPTION_DD_NAMESPACE_H

#include "XML/XML.h"
#include "DD4hep/Objects.h"
#include "DD4hep/Shapes.h"
#include "DD4hep/Volumes.h"
#include <unordered_map>
#include <vector>

namespace cms {

  namespace rotation_utils {
    std::string rotHash(const Double_t* rot);
    std::string rotHash(const dd4hep::Rotation3D& rot);
    double roundBinary(double value);
  }  // namespace rotation_utils

  class DDParsingContext;
  using DDVectorsMap = std::unordered_map<std::string, std::vector<double>>;

  class DDNamespace {
  public:
    DDNamespace(DDParsingContext*, xml_h);
    DDNamespace(DDParsingContext&, xml_h, bool);
    DDNamespace(DDParsingContext*);
    DDNamespace(DDParsingContext&);
    ~DDNamespace();

    DDNamespace() = delete;
    DDNamespace(const DDNamespace&) = delete;
    DDNamespace& operator=(const DDNamespace&) = delete;

    std::string prepend(const std::string&) const;
    std::string realName(const std::string&) const;
    static std::string objName(const std::string&);
    static std::string nsName(const std::string&);

    template <typename T>
    T attr(xml_elt_t element, const xml_tag_t& name) const {
      std::string val = realName(element.attr<std::string>(name));
      element.setAttr(name, val);
      return element.attr<T>(name);
    }

    template <typename T>
    T attr(xml_elt_t element, const xml_tag_t& name, T defaultValue) const {
      if (element.hasAttr(name)) {
        std::string val = realName(element.attr<std::string>(name));
        element.setAttr(name, val);
        return element.attr<T>(name);
      }
      return defaultValue;
    }

    void addConstant(const std::string& name, const std::string& value, const std::string& type) const;
    void addConstantNS(const std::string& name, const std::string& value, const std::string& type) const;

    dd4hep::Material material(const std::string& name) const;

    dd4hep::Solid solid(const std::string& name) const;
    dd4hep::Solid addSolid(const std::string& name, dd4hep::Solid solid) const;
    dd4hep::Solid addSolidNS(const std::string& name, dd4hep::Solid solid) const;

    dd4hep::Assembly assembly(const std::string& name, bool exception = true) const;
    dd4hep::Assembly addAssembly(dd4hep::Assembly asmb, bool addSolid = true) const;
    dd4hep::Assembly addAssemblySolid(dd4hep::Assembly assembly) const;

    dd4hep::Volume volume(const std::string& name, bool exc = true) const;
    dd4hep::Volume addVolume(dd4hep::Volume vol) const;
    dd4hep::Volume addVolumeNS(dd4hep::Volume vol) const;

    const dd4hep::Rotation3D& rotation(const std::string& name) const;
    void addRotation(const std::string& name, const dd4hep::Rotation3D& rot) const;

    DDParsingContext* const context() const { return m_context; }
    DDParsingContext* setContext() { return m_context; }

    std::string_view name() const { return m_name; }
    std::string noNamespace(const std::string&) const;

    std::vector<double> vecDbl(const std::string& name) const;
    std::vector<float> vecFloat(const std::string& name) const;

  private:
    DDParsingContext* m_context = nullptr;
    std::string m_name;
    bool m_pop = false;
  };
}  // namespace cms

#define NAMESPACE_SEP ':'

#endif