GeoHandler.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 DD4HEP_GEOHANDLER_H
#define DD4HEP_GEOHANDLER_H
 
#include <DD4hep/Detector.h>
 
#include <set>
#include <map>
#include <vector>
 
// Forward declarations
class TGeoMatrix;
class TGeoVolume;
class TGeoMedium;
class TGeoShape;
class TGeoNode;
 
namespace dd4hep {
 
  // Forward declarations
  class  Detector;
  class  NamedObject;
  class  DetElement;
  class  SensitiveDetector;
  class  VisAttrObject;
  class  Volume;
  class  PlacedVolume;
 
  namespace detail {
 
 
    class GeoHandlerTypes {
    public:
 
      class GeometryInfo {
      public:
        std::vector<TGeoShape*>   solids;
        std::set<TGeoShape*>   solid_set;
        std::set<Volume>       volumeSet;
        std::vector<Volume>    volumes;
        std::set<VisAttr>      vis;
        std::set<Ref_t>        fields;
        std::set<Material>     materials;
        std::set<TGeoMedium*>  media;
        std::set<TGeoElement*> elements;
        std::vector<std::pair<std::string, TGeoMatrix*> > trafos;
      };
    };
 
 
    class GeoHandler: public GeoHandlerTypes {
 
    protected:
      bool  m_propagateRegions { false };
 
      std::map<int, std::vector<const TGeoNode*> >*    m_data      { nullptr };
      std::map<int, std::set<const TGeoNode*> >* m_set_data { nullptr };
 
      std::map<const TGeoNode*, std::vector<TGeoNode*> >* m_daughters { nullptr };
      GeoHandler& i_collect(const TGeoNode* parent,
                            const TGeoNode* node,
                            int level, Region rg, LimitSet ls);
 
    private:
      GeoHandler(const GeoHandler&) {
      }
      GeoHandler& operator=(const GeoHandler&) {
        return *this;
      }
 
    public:
      GeoHandler();
      GeoHandler(std::map<int, std::vector<const TGeoNode*> >* ptr,
         std::map<int, std::set<const TGeoNode*> >* ptr_set,
                 std::map<const TGeoNode*, std::vector<TGeoNode*> >* daus = nullptr);
      virtual ~GeoHandler();
      bool setPropagateRegions(bool value);
      GeoHandler& collect(DetElement top);
      GeoHandler& collect(DetElement top, GeometryInfo& info);
      std::map<int, std::vector<const TGeoNode*> >* release();
    };
 
 
    class GeoScan {
    protected:
      std::map<int, std::vector<const TGeoNode*> >* m_data;
    public:
      GeoScan(DetElement e);
      GeoScan(DetElement e, bool propagateRegions);
      virtual ~GeoScan();
      virtual GeoScan& operator()();
    };
  }    // End namespace detail
}      // End namespace dd4hep
 
#endif // DD4HEP_GEOHANDLER_H