Added a couple of routines to handle the link between regions and volumes

[u/mrichter/AliRoot.git] / Flugg / FGeometryInit.hh

diff --git a/Flugg/FGeometryInit.hh b/Flugg/FGeometryInit.hh
index 6abeed01d78a8d74b0ce6f3914c6aa2bcf3dafa1..bbd97591fcd706745bc9658a9bd38a399ea5408c 100644 (file)
--- a/Flugg/FGeometryInit.hh
+++ b/Flugg/FGeometryInit.hh
@@ -30,55 +30,88 @@
 #include "G4TransportationManager.hh"
 
 
 #include "G4TransportationManager.hh"
 
 

+#include <map>
+

 class FluggNavigator;
 class FluggNavigator;

+class FlukaMaterial;
+class FlukaCompound;
+
+class FGeometryInit : public G4TransportationManager {
+
+public:
+  ~FGeometryInit();    //destructor
+  static FGeometryInit *GetInstance();
+  inline FluggNavigator *getNavigatorForTracking();
+  inline G4FieldManager * getFieldManager();
+  inline void setDetConstruction(G4VUserDetectorConstruction* detector);
+  inline void setDetector();
+  inline void setMotherVolume();
+  void createFlukaMatFile();
+  void closeGeometry();
+  
+  void PrintHistories();
+  void InitHistories();
+  void DeleteHistories();
+  void UpdateHistories(const G4NavigationHistory *, G4int);
+  inline G4TouchableHistory * GetTouchableHistory();
+  inline G4TouchableHistory * GetOldNavHist();
+  inline G4TouchableHistory * GetTempNavHist();
+  
+  void InitHistArray();
+  inline void DelHistArray();
+  inline G4int * GetHistArray();
+  
+  void InitJrLtGeantArray();
+  inline G4int * GetJrLtGeantArray();
+  inline G4int GetLttcFlagGeant();
+  void SetLttcFlagGeant(G4int);
+  void PrintJrLtGeant();
+
+  //Map access methods
+  G4int GetRegionFromName(const char* volName) const;
+

 
 

-class FGeometryInit : public G4TransportationManager
-{
-   public:
-       ~FGeometryInit();       //destructor
-       static FGeometryInit *GetInstance();
-       inline FluggNavigator *getNavigatorForTracking();
-        inline G4FieldManager * getFieldManager();
-       inline void setDetConstruction(G4VUserDetectorConstruction* detector);
-       inline void setDetector();
-       inline void setMotherVolume();
-       void createFlukaMatFile();
-        void closeGeometry();
-
-        void PrintHistories();
-       void InitHistories();
-        void DeleteHistories();
-        void UpdateHistories(const G4NavigationHistory *, G4int);
-       inline G4TouchableHistory * GetTouchableHistory();
-       inline G4TouchableHistory * GetOldNavHist();
-       inline G4TouchableHistory * GetTempNavHist();
-
-       void InitHistArray();
-       inline void DelHistArray();
-       inline G4int * GetHistArray();
-
-       void InitJrLtGeantArray();
-       inline G4int * GetJrLtGeantArray();
-        inline G4int GetLttcFlagGeant();
-        void SetLttcFlagGeant(G4int);
-        void PrintJrLtGeant(); 
-
-   private:    
-       FGeometryInit();        //costructor
-       G4VUserDetectorConstruction * fDetector;
-       G4FieldManager * fFieldManager; 
-        G4TransportationManager * fTransportationManager;
-       static FGeometryInit *flagInstance;
-        G4VPhysicalVolume * myTopNode;
-       G4GeometryManager * ptrGeoMan;
-       G4int * ptrArray;
-        G4TouchableHistory * ptrTouchHist;
-       G4TouchableHistory * ptrOldNavHist;
-       G4TouchableHistory * ptrTempNavHist;
-        G4int * ptrJrLtGeant;
-        G4int flagLttcGeant;
+protected:
+  void BuildRegionsMap();
+  void PrintRegionsMap(G4std::ostream& os);
+  void BuildMaterialTables();
+  FlukaMaterial* BuildFlukaMaterialFromElement(const G4Element* element, 
+                                              G4double matDensity);
+  FlukaMaterial* BuildFlukaMaterialFromIsotope(const G4Isotope* isotope,
+                                              G4double matDensity);
+  FlukaCompound* BuildFlukaCompoundFromMaterial(const G4Material* material); 
+  FlukaCompound* BuildFlukaCompoundFromElement(const G4Element* element,
+                                              G4double matDensity);
+  void PrintMaterialTables(G4std::ostream& os);
+  void PrintAssignmat(G4std::ostream& os);
+  void PrintMagneticField(G4std::ostream& os);
+
+private:    
+  FGeometryInit();     //costructor
+
+private:
+  G4VUserDetectorConstruction * fDetector;
+  G4FieldManager * fFieldManager; 
+  G4TransportationManager * fTransportationManager;
+  static FGeometryInit *flagInstance;
+  G4VPhysicalVolume * myTopNode;
+  G4GeometryManager * ptrGeoMan;
+  G4int * ptrArray;
+  G4TouchableHistory * ptrTouchHist;
+  G4TouchableHistory * ptrOldNavHist;
+  G4TouchableHistory * ptrTempNavHist;
+  G4int * ptrJrLtGeant;
+  G4int flagLttcGeant;
+
+  G4std::map<G4VPhysicalVolume*, int, G4std::less<G4VPhysicalVolume*> > fRegionVolumeMap;
+  G4std::map<G4Material*, FlukaMaterial*, G4std::less<G4Material*> > G4FlukaMaterialMap;
+  G4std::map<G4Material*, FlukaCompound*, G4std::less<G4Material*> > G4FlukaCompoundMap;
+  //G4int NOfMaterials;

 };
 
 };
 

+typedef  G4std::map<G4VPhysicalVolume*, int, G4std::less<G4VPhysicalVolume*> >::const_iterator RegionIterator;
+typedef  G4std::vector<G4Material*>::const_iterator MatTableIterator;
+

 
 //Include the file with the inline methods
 #include "FGeometryInit.icc"
 
 //Include the file with the inline methods
 #include "FGeometryInit.icc"