GeoHandler.h

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation for LCD
//--------------------------------------------------------------------------
// 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/LCDD.h"
#include <set>
#include <map>
#include <vector>

// Forward declarations
class TGeoMatrix;
class TGeoVolume;
class TGeoMedium;
class TGeoShape;
class TGeoNode;

namespace DD4hep {

  // Forward declarations
  class  NamedObject;

  namespace Geometry {

    class  LCDD;
    class  Volume;
    class  PlacedVolume;
    class  DetElement;
    class  SensitiveDetector;
    class  VisAttrObject;


    class GeoHandlerTypes {
    public:
      typedef std::set<Volume> VolumeSet;
      typedef std::vector<Volume> VolumeVector;
      typedef std::set<const TGeoVolume*> ConstVolumeSet;
      typedef std::vector<std::pair<std::string, TGeoMatrix*> > TransformSet;
      typedef std::set<TGeoShape*> SolidSet;
      typedef std::set<Material> MaterialSet;
      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<VisAttr> VisRefs;
      typedef std::set<SensitiveDetector> SensitiveDetectorSet;
      typedef std::set<Region>            RegionSet;
      typedef std::set<LimitSet>          LimitSetSet;
      typedef std::set<Ref_t>             Fields;
      typedef std::set<TNamed*> ObjectSet;
      typedef LCDD::HandleMap             DefinitionSet;


      class GeometryInfo {
      public:
        SolidSet solids;
        VolumeSet volumeSet;
        VolumeVector volumes;
        TransformSet trafos;
        VisRefs vis;
        Fields fields;
        MaterialSet materials;
        std::set<TGeoMedium*>  media;
        std::set<TGeoElement*> elements;
      };
    };


    class GeoHandler: public GeoHandlerTypes {

    protected:
      bool  m_propagateRegions;
      Data* m_data;

      GeoHandler& i_collect(const TGeoNode* node, int level, Region rg, LimitSet ls);

    private:
      GeoHandler(const GeoHandler&) {
      }
      GeoHandler& operator=(const GeoHandler&) {
        return *this;
      }

    public:
      GeoHandler();
      GeoHandler(Data* ptr);
      virtual ~GeoHandler();
      bool setPropagateRegions(bool value);
      GeoHandler& collect(DetElement top);
      GeoHandler& collect(DetElement top, GeometryInfo& info);
      Data* release();
    };


    class GeoScan {
    protected:
      GeoHandler::Data* m_data;
    public:
      GeoScan(DetElement e);
      GeoScan(DetElement e, bool propagateRegions);
      virtual ~GeoScan();
      virtual GeoScan& operator()();
    };
  }    // End namespace Geometry
}      // End namespace DD4hep

#endif // DD4HEP_GEOHANDLER_H