TG4RootDetectorConstruction.h

Go to the documentation of this file.

// @(#)root/g4root:$Id$
// Author: Andrei Gheata   07/08/06
 
/*************************************************************************
 * Copyright (C) 1995-2000, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
 
#ifndef ROOT_TG4RootDetectorConstruction
#define ROOT_TG4RootDetectorConstruction
 
#include "G4RotationMatrix.hh"
#include "G4VUserDetectorConstruction.hh"
 
#include "TGeoManager.h"
#include "TGeoNode.h"
 
#include <unordered_map>
 
class TObjArray;
class TGeoManager;
class TGeoMaterial;
class TGeoShape;
class TGeoVolume;
class TGeoMatrix;
 
class G4FieldManager;
class G4VSolid;
class G4LogicalVolume;
class G4VPhysicalVolume;
class G4Material;
class G4Element;
class TVirtualUserPostDetConstruction;
 
 
class TG4RootDetectorConstruction : public G4VUserDetectorConstruction
{
 
 private:
  typedef std::unordered_map<const TGeoMaterial*, G4Material*> G4MaterialMap_t;
  typedef G4MaterialMap_t::const_iterator G4MaterialIt_t;
  typedef G4MaterialMap_t::value_type G4MaterialVal_t;
  G4MaterialMap_t fG4MaterialMap; 
 
  typedef std::unordered_map<const TGeoVolume*, G4LogicalVolume*> G4VolumeMap_t;
  typedef G4VolumeMap_t::const_iterator G4VolumeIt_t;
  typedef G4VolumeMap_t::value_type G4VolumeVal_t;
  G4VolumeMap_t fG4VolumeMap; 
 
  typedef std::unordered_map<const G4LogicalVolume*, TGeoVolume*> VolumeMap_t;
  typedef VolumeMap_t::const_iterator VolumeIt_t;
  typedef VolumeMap_t::value_type VolumeVal_t;
  VolumeMap_t fVolumeMap; 
 
  typedef std::unordered_map<const TGeoNode*, G4VPhysicalVolume*>
    G4PVolumeMap_t;
  typedef G4PVolumeMap_t::const_iterator G4PVolumeIt_t;
  typedef G4PVolumeMap_t::value_type G4PVolumeVal_t;
  G4PVolumeMap_t fG4PVolumeMap; 
 
  typedef std::unordered_map<const G4VPhysicalVolume*, TGeoNode*> PVolumeMap_t;
  typedef PVolumeMap_t::const_iterator PVolumeIt_t;
  typedef PVolumeMap_t::value_type PVolumeVal_t;
  PVolumeMap_t fPVolumeMap; 
 
 protected:
  Bool_t fIsConstructed;                    
  TGeoManager* fGeometry;                   
  G4VPhysicalVolume* fTopPV;                
  TVirtualUserPostDetConstruction* fSDInit; 
  // Geometry creators
  void CreateG4LogicalVolumes();
  void CreateG4Materials();
  void CreateG4Elements();
  void CreateG4PhysicalVolumes();
  // Converters TGeo->G4 for basic types
  G4VSolid* CreateG4Solid(TGeoShape* shape);
  G4LogicalVolume* CreateG4LogicalVolume(TGeoVolume* vol);
  G4VPhysicalVolume* CreateG4PhysicalVolume(TGeoNode* node);
  G4Material* CreateG4Material(const TGeoMaterial* mat);
  G4RotationMatrix* CreateG4Rotation(const TGeoMatrix* matrix);
  G4Element* CreateG4Element(TGeoElement* elem);
 
 public:
  TG4RootDetectorConstruction();
  TG4RootDetectorConstruction(TGeoManager* geom);
  virtual ~TG4RootDetectorConstruction();
 
  virtual G4VPhysicalVolume* Construct();
  virtual void ConstructSDandField();
 
  // Getters
  TGeoManager* GetGeometryManager() const { return fGeometry; }
  G4VPhysicalVolume* GetTopPV() const { return fTopPV; }
  G4Material* GetG4Material(const TGeoMaterial* mat) const;
  G4LogicalVolume* GetG4Volume(const TGeoVolume* vol) const;
  TGeoVolume* GetVolume(const G4LogicalVolume* g4vol) const;
  G4VPhysicalVolume* GetG4VPhysicalVolume(const TGeoNode* node) const;
  TGeoNode* GetNode(const G4VPhysicalVolume* g4vol) const;
  TVirtualUserPostDetConstruction* GetSDInit() const { return fSDInit; }
  Bool_t IsConstructed() const { return fIsConstructed; }
 
  void Initialize(TVirtualUserPostDetConstruction* sdinit = 0);
 
  //   ClassDef(TG4RootDetectorConstruction,0)  // Class creating a G4 gometry
  //   based on ROOT geometry
};
 
 
class TVirtualUserPostDetConstruction
{
 public:
  TVirtualUserPostDetConstruction() {}          
  virtual ~TVirtualUserPostDetConstruction() {} 
 
  virtual void Initialize(TG4RootDetectorConstruction* dc) = 0;
  virtual void InitializeSDandField() {}
};
#endif