Import Geant4 10.5.0 source tree

examples/extended/medical/DICOM/include/DicomNestedPhantomParameterisation.hh

@@ -23,7 +23,6 @@
 // * acceptance of all terms of the Geant4 Software license.          *
 // ********************************************************************
 //
-//$Id: DicomNestedPhantomParameterisation.hh 92820 2015-09-17 15:22:14Z gcosmo $
 //
 /// \file medical/DICOM/include/DicomNestedPhantomParameterisation.hh
 /// \brief Definition of the DicomNestedPhantomParameterisation class
@@ -37,10 +36,10 @@
 
 #include "G4Types.hh"
 #include "G4ThreeVector.hh"
-#include "G4VNestedParameterisation.hh" 
+#include "G4VNestedParameterisation.hh"
 
 class G4VPhysicalVolume;
-class G4VTouchable; 
+class G4VTouchable;
 class G4VSolid;
 class G4Material;
 class G4VisAttributes;
@@ -66,39 +65,48 @@ class G4Hype;
 class DicomNestedPhantomParameterisation : public G4VNestedParameterisation
 {
 public:
-  
+    typedef std::map<G4String, G4VisAttributes*> ColourMap_t;
+
+    static G4String defaultColorFile;
+
+public:
+
   DicomNestedPhantomParameterisation(const G4ThreeVector& voxelSize,
-                                     std::vector<G4Material*>& mat,
-                                     G4int fnZ_ = 0, G4int fnY_ = 0, G4int fnX_ = 0);
-  ~DicomNestedPhantomParameterisation(); 
-  
+                std::vector<G4Material*>& mat,
+                G4int fnZ_ = 0, G4int fnY_ = 0, G4int fnX_ = 0,
+                G4String colorFile = defaultColorFile);
+  ~DicomNestedPhantomParameterisation();
+
   G4Material* ComputeMaterial(G4VPhysicalVolume *currentVol,
-                              const G4int repNo, 
+                              const G4int repNo,
                               const G4VTouchable *parentTouch );
   // Must cope with parentTouch for navigator's SetupHierarchy
-  
+
   G4int       GetNumberOfMaterials() const;
   G4Material* GetMaterial(G4int idx) const;
-  // Needed to define materials for instances of Nested Parameterisation 
-  // Current convention: each call should return the materials 
+  // Needed to define materials for instances of Nested Parameterisation
+  // Current convention: each call should return the materials
   // of all instances with the same mother/ancestor volume
-  
-  unsigned int GetMaterialIndex( unsigned int nx, unsigned int ny, unsigned int nz) const;
+
+  unsigned int GetMaterialIndex( unsigned int nx, unsigned int ny,
+                                 unsigned int nz) const;
   unsigned int GetMaterialIndex( unsigned int copyNo) const;
   void SetMaterialIndices( size_t* matInd ) { fMaterialIndices = matInd; }
   void SetNoVoxel( unsigned int nx, unsigned int ny, unsigned int nz );
-  
+
   void ComputeTransformation(const G4int no,
                              G4VPhysicalVolume *currentPV) const;
-  
-  // Additional standard Parameterisation methods, 
+
+  // Additional standard Parameterisation methods,
   // which can be optionally defined, in case solid is used.
   void ComputeDimensions(G4Box &, const G4int,
                          const G4VPhysicalVolume *) const;
-  
-  
+
+  const ColourMap_t& GetColourMap() const { return fColours; }
+  ColourMap_t& GetColourMap() { return fColours; }
+
 private:  // Dummy declarations to get rid of warnings ...
-  
+
   void ComputeDimensions (G4Trd&, const G4int,
                           const G4VPhysicalVolume*) const {}
   void ComputeDimensions (G4Trap&, const G4int,
@@ -123,18 +131,20 @@ private:  // Dummy declarations to get rid of warnings ...
                           const G4VPhysicalVolume*) const {}
   void ComputeDimensions (G4Polyhedra&, const G4int,
                           const G4VPhysicalVolume*) const {}
-  
-  void ReadColourData();
-  
+
+  void ReadColourData(G4String);
+
   using G4VNestedParameterisation::ComputeMaterial;
-  
+
 private:
-  
+
   G4double fdX,fdY,fdZ;
   G4int fnX,fnY,fnZ;
   std::vector<G4Material*> fMaterials;
   size_t* fMaterialIndices; // Index in materials corresponding to each voxel
-  std::map<G4String,G4VisAttributes*> fColours;
+  ColourMap_t fColours;
+  std::map<G4int, G4VisAttributes*> mColours;
   std::vector<G4double> fpZ;
 };
+
 #endif