| Geant4 Cross Reference |
1 // 1 //
2 // ******************************************* 2 // ********************************************************************
3 // * License and Disclaimer 3 // * License and Disclaimer *
4 // * 4 // * *
5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. 9 // * include a list of copyright holders. *
10 // * 10 // * *
11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. *
17 // * 17 // * *
18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. *
24 // ******************************************* 24 // ********************************************************************
25 // 25 //
26 // G4ScoringManager << 26 //
27 // ------------------------------------------- << 27 // $Id: G4ScoringManager.cc,v 1.30 2007/11/14 22:08:15 asaim Exp $
>> 28 // GEANT4 tag $Name: geant4-09-01 $
>> 29 //
28 30
29 #include "G4ScoringManager.hh" 31 #include "G4ScoringManager.hh"
30 #include "G4ScoringMessenger.hh" 32 #include "G4ScoringMessenger.hh"
31 #include "G4ScoreQuantityMessenger.hh" 33 #include "G4ScoreQuantityMessenger.hh"
32 #include "G4VScoringMesh.hh" 34 #include "G4VScoringMesh.hh"
33 #include "G4THitsMap.hh" 35 #include "G4THitsMap.hh"
34 #include "G4VScoreColorMap.hh" 36 #include "G4VScoreColorMap.hh"
35 #include "G4DefaultLinearColorMap.hh" 37 #include "G4DefaultLinearColorMap.hh"
36 #include "G4ScoreLogColorMap.hh" <<
37 38
38 G4ThreadLocal G4ScoringManager* G4ScoringManag << 39 G4ScoringManager* G4ScoringManager::fSManager = 0;
39 40
40 G4ThreadLocal G4int G4ScoringManager::replicaL << 41 G4int G4ScoringManager::replicaLevel = 2;
41 42
42 G4ScoringManager* G4ScoringManager::GetScoring 43 G4ScoringManager* G4ScoringManager::GetScoringManager()
43 { 44 {
44 if(fSManager == nullptr) << 45 if(!fSManager)
45 { 46 {
46 fSManager = new G4ScoringManager; 47 fSManager = new G4ScoringManager;
47 } 48 }
48 return fSManager; 49 return fSManager;
49 } 50 }
50 51
51 G4ScoringManager* G4ScoringManager::GetScoring 52 G4ScoringManager* G4ScoringManager::GetScoringManagerIfExist()
52 { << 53 { return fSManager; }
53 return fSManager; <<
54 } <<
55 54
56 G4ScoringManager::G4ScoringManager() 55 G4ScoringManager::G4ScoringManager()
57 : verboseLevel(0) << 56 :verboseLevel(0),fCurrentMesh(0)
58 , fCurrentMesh(nullptr) <<
59 { 57 {
60 fMessenger = new G4ScoringMessen << 58 fMessenger = new G4ScoringMessenger(this);
61 fQuantityMessenger = new G4ScoreQuantity << 59 fQuantityMessenger = new G4ScoreQuantityMessenger(this);
62 fColorMapDict = new ColorMapDict(); << 60 fColorMapDict = new ColorMapDict();
63 fDefaultLinearColorMap = new G4DefaultLinear 61 fDefaultLinearColorMap = new G4DefaultLinearColorMap("defaultLinearColorMap");
64 (*fColorMapDict)[fDefaultLinearColorMap->Get 62 (*fColorMapDict)[fDefaultLinearColorMap->GetName()] = fDefaultLinearColorMap;
65 G4VScoreColorMap* logColorMap = new G4ScoreL << 63 writer = new G4VScoreWriter();
66 (*fColorMapDict)[logColorMap->GetName()] = l <<
67 writer = n <<
68 } 64 }
69 65
70 G4ScoringManager::~G4ScoringManager() 66 G4ScoringManager::~G4ScoringManager()
71 { << 67 {
72 delete writer; <<
73 delete fDefaultLinearColorMap; <<
74 delete fColorMapDict; <<
75 delete fQuantityMessenger; <<
76 delete fMessenger; 68 delete fMessenger;
77 fSManager = nullptr; << 69 fSManager = 0;
>> 70 if(writer) delete writer;
78 } 71 }
79 72
80 void G4ScoringManager::SetReplicaLevel(G4int l << 73 void G4ScoringManager::SetReplicaLevel(G4int lvl)
81 G4int G4ScoringManager::GetReplicaLevel() { re << 74 { replicaLevel = lvl; }
>> 75 G4int G4ScoringManager::GetReplicaLevel()
>> 76 { return replicaLevel; }
82 77
83 void G4ScoringManager::Accumulate(G4VHitsColle 78 void G4ScoringManager::Accumulate(G4VHitsCollection* map)
84 { 79 {
85 auto sm = FindMesh(map); << 80 G4String wName = map->GetSDname();
86 if(sm == nullptr) << 81 G4VScoringMesh* sm = FindMesh(wName);
87 return; << 82 if(sm == NULL) return;
88 if(verboseLevel > 9) << 83 if(verboseLevel>9)
89 { << 84 { G4cout << "G4ScoringManager::Accumulate() for " << map->GetSDname() << " / " << map->GetName() << G4endl;
90 G4cout << "G4ScoringManager::Accumulate() << 85 G4cout << " is calling G4VScoringMesh::Accumulate() of " << sm->GetWorldName() << G4endl; }
91 << map->GetName() << G4endl; <<
92 G4cout << " is calling G4VScoringMesh::Ac <<
93 << sm->GetWorldName() << G4endl; <<
94 } <<
95 sm->Accumulate(static_cast<G4THitsMap<double 86 sm->Accumulate(static_cast<G4THitsMap<double>*>(map));
96 } 87 }
97 88
98 G4VScoringMesh* G4ScoringManager::FindMesh(G4V << 89 G4VScoringMesh* G4ScoringManager::FindMesh(G4String wName)
99 { 90 {
100 auto colID = map->GetColID(); << 91 G4VScoringMesh* sm = 0;
101 G4VScoringMesh* sm = nullptr; << 92 for(MeshVecItr itr = fMeshVec.begin(); itr != fMeshVec.end(); itr++) {
102 auto msh = fMeshMap.find(colID); << 93 G4String smName = (*itr)->GetWorldName();
103 if(msh == fMeshMap.end()) << 94 if(wName == smName) {
104 { << 95 sm = *itr;
105 const auto& wName = map->GetSDname(); << 96 break;
106 sm = FindMesh(wName); << 97 }
107 fMeshMap[colID] = sm; <<
108 } <<
109 else <<
110 { <<
111 sm = (*msh).second; <<
112 } 98 }
113 return sm; << 99 if(!sm && verboseLevel>9)
114 } << 100 { G4cout << "WARNING : G4ScoringManager::FindMesh() --- <" << wName << "> is not found. Null returned." << G4endl; }
115 101
116 G4VScoringMesh* G4ScoringManager::FindMesh(con << 102 return sm;
117 { <<
118 G4VScoringMesh* sm = nullptr; <<
119 for(auto msh : fMeshVec) <<
120 { <<
121 if(msh->GetWorldName() == wName) <<
122 return msh; <<
123 } <<
124 if((sm == nullptr) && verboseLevel > 9) <<
125 { <<
126 G4cout << "WARNING : G4ScoringManager::Fin <<
127 << "> is not found. Null returned." <<
128 } <<
129 return nullptr; <<
130 } 103 }
131 104
132 void G4ScoringManager::List() const 105 void G4ScoringManager::List() const
133 { 106 {
134 G4cout << "G4ScoringManager has " << GetNumb << 107 G4cout << "G4ScoringManager has " << GetNumberOfMesh() << " scoring meshes." << G4endl;
135 << G4endl; << 108 for(MeshVecConstItr itr = fMeshVec.begin(); itr != fMeshVec.end(); itr++) {
136 for(auto msh : fMeshVec) << 109 (*itr)->List();
137 msh->List(); << 110 }
138 } 111 }
139 112
140 void G4ScoringManager::Dump() const 113 void G4ScoringManager::Dump() const
141 { 114 {
142 for(auto msh : fMeshVec) << 115 for(MeshVecConstItr itr = fMeshVec.begin(); itr != fMeshVec.end(); itr++) {
143 msh->Dump(); << 116 (*itr)->Dump();
>> 117 }
144 } 118 }
145 119
146 void G4ScoringManager::DrawMesh(const G4String << 120 void G4ScoringManager::DrawMesh(G4String meshName,G4String psName,G4String colorMapName,G4int axflg)
147 const G4String <<
148 const G4String <<
149 { 121 {
150 G4VScoringMesh* mesh = FindMesh(meshName); 122 G4VScoringMesh* mesh = FindMesh(meshName);
151 if(mesh != nullptr) << 123 if(mesh)
152 { 124 {
153 G4VScoreColorMap* colorMap = GetScoreColor 125 G4VScoreColorMap* colorMap = GetScoreColorMap(colorMapName);
154 if(colorMap == nullptr) << 126 if(!colorMap)
155 { 127 {
156 G4cerr << "WARNING : Score color map <" << 128 G4cerr << "WARNING : Score color map <" << colorMapName << "> is not found. Default linear color map is used." << G4endl;
157 << "> is not found. Default linea <<
158 colorMap = fDefaultLinearColorMap; 129 colorMap = fDefaultLinearColorMap;
159 } 130 }
160 mesh->DrawMesh(psName, colorMap, axflg); << 131 mesh->DrawMesh(psName,colorMap,axflg);
161 } << 132 } else {
162 else << 133 G4cerr << "ERROR : G4ScoringManager::DrawMesh() --- <"
163 { << 134 << meshName << "> is not found. Nothing is done." << G4endl;
164 G4cerr << "ERROR : G4ScoringManager::DrawM <<
165 << "> is not found. Nothing is done <<
166 } 135 }
167 } 136 }
168 137
169 void G4ScoringManager::DrawMesh(const G4String << 138 void G4ScoringManager::DrawMesh(G4String meshName,G4String psName,G4int idxPlane,G4int iColumn,G4String colorMapName)
170 const G4String <<
171 G4int iColumn, <<
172 { 139 {
173 G4VScoringMesh* mesh = FindMesh(meshName); 140 G4VScoringMesh* mesh = FindMesh(meshName);
174 if(mesh != nullptr) << 141 if(mesh)
175 { 142 {
176 G4VScoreColorMap* colorMap = GetScoreColor 143 G4VScoreColorMap* colorMap = GetScoreColorMap(colorMapName);
177 if(colorMap == nullptr) << 144 if(!colorMap)
178 { 145 {
179 G4cerr << "WARNING : Score color map <" << 146 G4cerr << "WARNING : Score color map <" << colorMapName << "> is not found. Default linear color map is used." << G4endl;
180 << "> is not found. Default linea <<
181 colorMap = fDefaultLinearColorMap; 147 colorMap = fDefaultLinearColorMap;
182 } 148 }
183 mesh->DrawMesh(psName, idxPlane, iColumn, << 149 mesh->DrawMesh(psName,idxPlane,iColumn,colorMap);
184 } << 150 } else {
185 else << 151 G4cerr << "ERROR : G4ScoringManager::DrawMesh() --- <"
186 { << 152 << meshName << "> is not found. Nothing is done." << G4endl;
187 G4cerr << "ERROR : G4ScoringManager::DrawM <<
188 << "> is not found. Nothing is done <<
189 } 153 }
190 } 154 }
191 155
192 void G4ScoringManager::DumpQuantityToFile(cons << 156 void G4ScoringManager::DumpQuantityToFile(G4String meshName,G4String psName,G4String fileName, G4String option)
193 cons <<
194 cons <<
195 cons <<
196 { 157 {
197 G4VScoringMesh* mesh = FindMesh(meshName); 158 G4VScoringMesh* mesh = FindMesh(meshName);
198 if(mesh != nullptr) << 159 if(mesh) {
199 { <<
200 writer->SetScoringMesh(mesh); 160 writer->SetScoringMesh(mesh);
201 writer->DumpQuantityToFile(psName, fileNam 161 writer->DumpQuantityToFile(psName, fileName, option);
202 } << 162 } else {
203 else << 163 G4cerr << "ERROR : G4ScoringManager::DrawQuantityToFile() --- <"
204 { << 164 << meshName << "> is not found. Nothing is done." << G4endl;
205 G4cerr << "ERROR : G4ScoringManager::DrawQ <<
206 << "> is not found. Nothing is done <<
207 } 165 }
208 } 166 }
209 167
210 void G4ScoringManager::DumpAllQuantitiesToFile << 168 void G4ScoringManager::DumpAllQuantitiesToFile(G4String meshName,G4String fileName, G4String option)
211 <<
212 <<
213 { 169 {
214 G4VScoringMesh* mesh = FindMesh(meshName); 170 G4VScoringMesh* mesh = FindMesh(meshName);
215 if(mesh != nullptr) << 171 if(mesh) {
216 { <<
217 writer->SetScoringMesh(mesh); 172 writer->SetScoringMesh(mesh);
218 writer->DumpAllQuantitiesToFile(fileName, 173 writer->DumpAllQuantitiesToFile(fileName, option);
219 } << 174 } else {
220 else <<
221 { <<
222 G4cerr << "ERROR : G4ScoringManager::DrawA 175 G4cerr << "ERROR : G4ScoringManager::DrawAllQuantitiesToFile() --- <"
223 << meshName << "> is not found. Not << 176 << meshName << "> is not found. Nothing is done." << G4endl;
224 } 177 }
225 } 178 }
226 179
227 void G4ScoringManager::RegisterScoreColorMap(G 180 void G4ScoringManager::RegisterScoreColorMap(G4VScoreColorMap* colorMap)
228 { 181 {
229 if(fColorMapDict->find(colorMap->GetName()) << 182 if(fColorMapDict->find(colorMap->GetName()) != fColorMapDict->end())
230 { 183 {
231 G4cerr << "ERROR : G4ScoringManager::Regis 184 G4cerr << "ERROR : G4ScoringManager::RegisterScoreColorMap -- "
232 << colorMap->GetName() << 185 << colorMap->GetName() << " has already been registered. Method ignored." << G4endl;
233 << " has already been registered. M <<
234 } 186 }
235 else 187 else
236 { 188 {
237 (*fColorMapDict)[colorMap->GetName()] = co 189 (*fColorMapDict)[colorMap->GetName()] = colorMap;
238 } 190 }
239 } 191 }
240 192
241 G4VScoreColorMap* G4ScoringManager::GetScoreCo << 193 G4VScoreColorMap* G4ScoringManager::GetScoreColorMap(G4String mapName)
242 { 194 {
243 auto mItr = fColorMapDict->find(mapName); << 195 ColorMapDictItr mItr = fColorMapDict->find(mapName);
244 if(mItr == fColorMapDict->cend()) << 196 if(mItr == fColorMapDict->end()) { return 0; }
245 { <<
246 return nullptr; <<
247 } <<
248 return (mItr->second); 197 return (mItr->second);
249 } 198 }
250 199
251 void G4ScoringManager::ListScoreColorMaps() 200 void G4ScoringManager::ListScoreColorMaps()
252 { 201 {
253 G4cout << "Registered Score Color Maps " << 202 G4cout << "Registered Score Color Maps -------------------------------------------------------" << G4endl;
254 "--------------------------------- << 203 ColorMapDictItr mItr = fColorMapDict->begin();
255 << G4endl; << 204 for(;mItr!=fColorMapDict->end();mItr++)
256 auto mItr = fColorMapDict->cbegin(); << 205 { G4cout << " " << mItr->first; }
257 for(; mItr != fColorMapDict->cend(); ++mItr) <<
258 { <<
259 G4cout << " " << mItr->first; <<
260 } <<
261 G4cout << G4endl; 206 G4cout << G4endl;
262 } 207 }
263 208
264 void G4ScoringManager::Merge(const G4ScoringMa << 209
265 { <<
266 for(auto i = 0; i < (G4int)GetNumberOfMesh() <<
267 { <<
268 G4VScoringMesh* fMesh = GetMesh(i); <<
269 G4VScoringMesh* scMesh = mgr->GetMesh(i); <<
270 fMesh->Merge(scMesh); <<
271 } <<
272 } <<
273 210