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:
#if 0
      typedef std::set<const TGeoVolume*> ConstVolumeSet;
      typedef std::map<SensitiveDetector, ConstVolumeSet> SensitiveVolumes;
      typedef std::map<Region,   ConstVolumeSet>          RegionVolumes;
      typedef std::map<LimitSet, ConstVolumeSet>          LimitVolumes;
      typedef std::map<int, std::set<const TGeoNode*> >   Data;
      typedef std::set<SensitiveDetector>                 SensitiveDetectorSet;
      typedef std::set<Region>                            RegionSet;
      typedef std::set<LimitSet>                          LimitSetSet;
      typedef std::set<TNamed*>                           ObjectSet;
#endif
 
      class GeometryInfo {
      public:
        std::set<TGeoShape*>   solids;
        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;
      std::map<int, std::set<const TGeoNode*> >* m_data;
 
      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::set<const TGeoNode*> >* ptr);
      virtual ~GeoHandler();
      bool setPropagateRegions(bool value);
      GeoHandler& collect(DetElement top);
      GeoHandler& collect(DetElement top, GeometryInfo& info);
      std::map<int, std::set<const TGeoNode*> >* release();
    };
 
 
    class GeoScan {
    protected:
      std::map<int, std::set<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