| Geant4 Cross Reference |
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 //
27 // Author: Haegin Han
28 // Reference: ICRP Publication 145. Ann. ICRP 49(3), 2020.
29 // Geant4 Contributors: J. Allison and S. Guatelli
30 //
31 #ifndef TETModelImport_h
32 #define TETModelImport_h 1
33
34 #include <stdlib.h>
35 #include <fstream>
36 #include <sstream>
37 #include <iomanip>
38 #include <vector>
39 #include <map>
40
41 #include "G4UIExecutive.hh"
42 #include "G4SystemOfUnits.hh"
43 #include "G4PhysicalConstants.hh"
44 #include "G4ThreeVector.hh"
45 #include "G4String.hh"
46 #include "G4Tet.hh"
47 #include "G4NistManager.hh"
48 #include "G4Material.hh"
49 #include "G4Colour.hh"
50
51 // *********************************************************************
52 // This class imports the phantom data from *.ele, *.node, and
53 // *.material files.
54 // -- DataRead: Construct G4Tet by reading data from *.ele and *.node
55 // files
56 // -- MaterialRead: Construct G4Material by reading material data from
57 // *.material file
58 // -- ColourRead: Construct std::map that contains G4Colour data
59 // according to data in colour.dat file
60 // -- PrintMaterialInformation: Print a table that contains organ ID,
61 // number of tetrahedrons, volume, density,
62 // mass, and name for each organ
63 // *********************************************************************
64
65 class TETModelImport
66 {
67 public:
68 TETModelImport(G4bool isAF, G4UIExecutive* ui);
69 virtual ~TETModelImport() {};
70
71 // get methods
72 G4String GetPhantomName() { return fPhantomName; };
73 G4Material* GetMaterial(G4int idx) { return fMaterialMap[idx];}
74 G4int GetNumTetrahedron() { return fTetVector.size();}
75 G4int GetMaterialIndex(G4int idx){ return fMaterialVector[idx]; }
76 G4Tet* GetTetrahedron(G4int idx) { return fTetVector[idx]; }
77 G4double GetVolume(G4int idx) { return fVolumeMap[idx]; }
78 std::map<G4int, G4double> GetMassMap() { return fMassMap; }
79 std::map<G4int, G4Colour> GetColourMap() { return fColourMap; }
80 G4ThreeVector GetPhantomSize() { return fPhantomSize; }
81 G4ThreeVector GetPhantomBoxMin() { return fBoundingBox_Min; }
82 G4ThreeVector GetPhantomBoxMax() { return fBoundingBox_Max; }
83
84 private:
85 // private methods
86 void DataRead(G4String, G4String);
87 void MaterialRead(G4String);
88 void ColourRead();
89 void PrintMaterialInfomation();
90
91 G4String fPhantomDataPath;
92 G4String fPhantomName;
93
94 G4ThreeVector fBoundingBox_Min;
95 G4ThreeVector fBoundingBox_Max;
96 G4ThreeVector fPhantomSize;
97
98 std::vector<G4ThreeVector> fVertexVector;
99 std::vector<G4Tet*> fTetVector;
100 std::vector<G4int*> fEleVector;
101 std::vector<G4int> fMaterialVector;
102 std::map<G4int, G4int> fNumTetMap;
103 std::map<G4int, G4double> fVolumeMap;
104 std::map<G4int, G4double> fMassMap;
105 std::map<G4int, G4Colour> fColourMap;
106
107 std::map<G4int, std::vector<std::pair<G4int, G4double>>> fMaterialIndexMap;
108 std::vector<G4int> fMaterialIndex;
109 std::map<G4int, G4Material*> fMaterialMap;
110 std::map<G4int, G4double> fDensityMap;
111 std::map<G4int, G4String> fOrganNameMap;
112 };
113
114 #endif
115