LCDDConverter.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================
 
#ifndef DDCORE_SRC_PLUGINS_LCDDCONVERTER_H
#define DDCORE_SRC_PLUGINS_LCDDCONVERTER_H
 
// Framework include files
#include <DD4hep/Detector.h>
#include <DD4hep/GeoHandler.h>
#include <DD4hep/DetFactoryHelper.h>
 
// C/C++ include files
#include <set>
#include <map>
 
// Forward declarations
class TGeoVolume;
class TGeoElement;
class TGeoShape;
class TGeoMedium;
class TGeoNode;
class TGeoMatrix;
 
namespace dd4hep {
 
  namespace detail {
 
    // Forward declarations
    class SensitiveDetectorObject;
 
 
    class LCDDConverter: public GeoHandler {
    public:
      typedef xml::XmlElement XmlElement;
      typedef std::map<Atom,              XmlElement*> ElementMap;
      typedef std::map<Material,          XmlElement*> MaterialMap;
      typedef std::map<LimitSet,          XmlElement*> LimitMap;
      typedef std::map<PlacedVolume,      XmlElement*> PlacementMap;
      typedef std::map<Region,            XmlElement*> RegionMap;
      typedef std::map<SensitiveDetector, XmlElement*> SensDetMap;
      typedef std::map<Volume,            XmlElement*> VolumeMap;
      typedef std::map<IDDescriptor,      XmlElement*> IdSpecMap;
      typedef std::map<VisAttr,           XmlElement*> VisMap;
      typedef std::map<const TGeoShape*,  XmlElement*> SolidMap;
      typedef std::map<OverlayedField,    XmlElement*> FieldMap;
      typedef std::map<const TGeoMatrix*, XmlElement*> TrafoMap;
 
      class GeometryInfo: public GeoHandler::GeometryInfo {
      public:
        ElementMap   xmlElements;
        MaterialMap  xmlMaterials;
        SolidMap     xmlSolids;
        VolumeMap    xmlVolumes;
        PlacementMap xmlPlacements;
        RegionMap    xmlRegions;
        VisMap       xmlVis;
        LimitMap     xmlLimits;
        IdSpecMap    xmlIdSpecs;
        SensDetMap   xmlSensDets;
        TrafoMap     xmlPositions;
        TrafoMap     xmlRotations;
        FieldMap     xmlFields;
        std::set<SensitiveDetector> sensitives;
        std::set<Region> regions;
        std::set<LimitSet> limits;
        // These we need for redundancy and checking the data integrity
        typedef std::map<std::string, const TNamed*> CheckIter;
        struct _checks {
          std::map<std::string, const TNamed*> positions, rotations, volumes, solids, materials;
        };
        mutable _checks checks;
        void check(const std::string& name, const TNamed* n, std::map<std::string, const TNamed*>& array) const;
        void checkPosition(const std::string& name, const TNamed* n) const {
          check(name, n, checks.positions);
        }
        void checkRotation(const std::string& name, const TNamed* n) const {
          check(name, n, checks.rotations);
        }
        void checkVolume(const std::string& name, const TNamed* n) const {
          check(name, n, checks.volumes);
        }
        void checkShape(const std::string& name, const TNamed* n) const {
          check(name, n, checks.solids);
        }
        void checkMaterial(const std::string& name, Material n) const {
          check(name, n.ptr(), checks.materials);
        }
 
        xml_doc_t doc;
        xml_h identity_rot, identity_pos;
        xml_elt_t doc_root, doc_header, doc_idDict, doc_detectors, doc_limits, doc_regions, doc_display, doc_gdml, doc_fields,
          doc_define, doc_materials, doc_solids, doc_structure, doc_setup;
        GeometryInfo();
      };
      typedef std::set<std::string> NameSet;
 
      Detector& m_detDesc;
      mutable NameSet m_checkNames;
      GeometryInfo* m_dataPtr;
 
      GeometryInfo& data() const {
        return *m_dataPtr;
      }
 
      void checkVolumes(const std::string& name, Volume volume) const;
 
      LCDDConverter(Detector& description);
 
      virtual ~LCDDConverter();
 
      xml_doc_t createGDML(DetElement top);
 
      xml_doc_t createDetector(DetElement top);
 
      xml_doc_t createVis(DetElement top);
 
      virtual void handleHeader() const;
 
      virtual xml_h handleMaterial(const std::string& name, Material medium) const;
 
      virtual xml_h handleElement(const std::string& name, Atom element) const;
 
      virtual xml_h handleSolid(const std::string& name, const TGeoShape* volume) const;
 
      virtual xml_h handleVolume(const std::string& name, Volume volume) const;
      virtual xml_h handleVolumeVis(const std::string& name, const TGeoVolume* volume) const;
      virtual void collectVolume(const std::string& name, const TGeoVolume* volume) const;
 
      virtual xml_h handlePlacement(const std::string& name, PlacedVolume node) const;
 
 
      virtual xml_h handleRegion(const std::string& name, Region region) const;
 
      virtual xml_h handleVis(const std::string& name, VisAttr vis) const;
 
      virtual xml_h handleIdSpec(const std::string& name, IDDescriptor idspec) const;
 
      virtual xml_h handleLimitSet(const std::string& name, LimitSet limitset) const;
 
      virtual xml_h handleSensitive(const std::string& name, SensitiveDetector sens_det) const;
 
      virtual xml_h handleSegmentation(Segmentation seg) const;
 
      virtual xml_h handlePosition(const std::string& name, const TGeoMatrix* trafo) const;
 
      virtual xml_h handleRotation(const std::string& name, const TGeoMatrix* trafo) const;
 
      virtual xml_h handleField(const std::string& name, OverlayedField field) const;
 
      void handleProperties(Detector::Properties& prp) const;
    };
  }    // End namespace xml
}      // End namespace dd4hep
#endif // DDCORE_SRC_PLUGINS_LCDDCONVERTER_H