]>
Commit | Line | Data |
---|---|---|
676fb573 | 1 | // $Id$ |
2 | // Category: geometry | |
3 | // | |
4 | // See the class description in the header file. | |
5 | ||
6 | #include "AliMoreModulesConstruction.h" | |
7 | #include "AliSingleModuleConstruction.h" | |
8 | #include "AliSDManager.h" | |
9 | #include "AliSensitiveDetector.h" | |
10 | #include "AliModule.h" | |
11 | #include "AliRun.h" | |
12 | #include "AliGlobals.h" | |
13 | ||
14 | #include "TG4GeometryManager.h" | |
15 | ||
16 | #include <G4SDManager.hh> | |
17 | #include <G4LogicalVolume.hh> | |
18 | #include <G4LogicalVolumeStore.hh> | |
19 | ||
20 | #include <TROOT.h> | |
21 | #include <TCint.h> | |
22 | ||
23 | AliMoreModulesConstruction::AliMoreModulesConstruction() { | |
24 | // | |
25 | fSDManager = AliSDManager::Instance(); | |
26 | } | |
27 | ||
28 | AliMoreModulesConstruction::AliMoreModulesConstruction( | |
29 | const AliMoreModulesConstruction& right) | |
30 | { | |
31 | // delete current module constructions | |
32 | fModuleConstructionVector.erase( | |
33 | fModuleConstructionVector.begin(), fModuleConstructionVector.end()); | |
34 | ||
35 | // create new module constructions | |
36 | G4int nofModules = right.fModuleConstructionVector.size(); | |
37 | for (G4int i=0; i<nofModules; i++) { | |
38 | G4String name = right.fModuleConstructionVector[i]->GetDetName(); | |
39 | G4int version = right.fModuleConstructionVector[i]->GetVersion(); | |
40 | AliModuleType type = right.fModuleConstructionVector[i]->GetType(); | |
41 | AddModule(name, version, type); | |
42 | } | |
43 | ||
44 | fSDManager = right.fSDManager; | |
45 | } | |
46 | ||
47 | ||
48 | AliMoreModulesConstruction::~AliMoreModulesConstruction() | |
49 | { | |
50 | // delete module constructions | |
51 | fModuleConstructionVector.erase( | |
52 | fModuleConstructionVector.begin(), fModuleConstructionVector.end()); | |
53 | } | |
54 | ||
55 | // operators | |
56 | ||
57 | AliMoreModulesConstruction& | |
58 | AliMoreModulesConstruction::operator=(const AliMoreModulesConstruction& right) | |
59 | { | |
60 | // check assignement to self | |
61 | if (this == &right) return *this; | |
62 | ||
63 | // delete current module constructions | |
64 | fModuleConstructionVector.erase( | |
65 | fModuleConstructionVector.begin(), fModuleConstructionVector.end()); | |
66 | ||
67 | // create new module constructions | |
68 | G4int nofModules = right.fModuleConstructionVector.size(); | |
69 | for (G4int i=0; i<nofModules; i++) { | |
70 | G4String name = right.fModuleConstructionVector[i]->GetDetName(); | |
71 | G4int version = right.fModuleConstructionVector[i]->GetVersion(); | |
72 | AliModuleType type = right.fModuleConstructionVector[i]->GetType(); | |
73 | AddModule(name, version, type); | |
74 | } | |
75 | ||
76 | fSDManager = right.fSDManager; | |
77 | ||
78 | return *this; | |
79 | } | |
80 | ||
81 | // private methods | |
82 | ||
83 | void AliMoreModulesConstruction::CreateSensitiveDetectors( | |
84 | G4bool allLVSensitive) | |
85 | { | |
86 | // Creates sensitive detectors. | |
87 | // --- | |
88 | ||
89 | if (allLVSensitive) | |
90 | CreateSensitiveDetectors1(); | |
91 | else | |
92 | CreateSensitiveDetectors2(); | |
93 | ||
94 | // set static number of logical volumes already processed | |
95 | G4LogicalVolumeStore* pLVStore = G4LogicalVolumeStore::GetInstance(); | |
96 | fSDManager->SetNofLVWithSD(pLVStore->entries()); | |
97 | } | |
98 | ||
99 | void AliMoreModulesConstruction::CreateSensitiveDetectors1() | |
100 | { | |
101 | // Creates sensitive detectors. | |
102 | // Sensitive detectors are set to all logical volumes | |
103 | // --- | |
104 | ||
105 | G4LogicalVolumeStore* pLVStore = G4LogicalVolumeStore::GetInstance(); | |
106 | G4int nofLV = pLVStore->entries(); | |
107 | ||
108 | G4int nofLVWithSD = fSDManager->GetNofLVWithSD(); | |
109 | for (G4int i=nofLVWithSD; i<nofLV; i++) { | |
110 | G4LogicalVolume* lv = (*pLVStore)[i]; | |
111 | AliModule* module = fSDManager->FindAliModule(lv); | |
112 | fSDManager->CreateSD(lv, module); | |
113 | } | |
114 | } | |
115 | ||
116 | void AliMoreModulesConstruction::CreateSensitiveDetectors2() | |
117 | { | |
118 | // Creates sensitive detectors. | |
119 | // Sensitive detectors are set only to logical volumes | |
120 | // in G3SensVolVector. | |
121 | // --- | |
122 | ||
123 | TG4GeometryManager* pGeometryManager = TG4GeometryManager::Instance(); | |
124 | ||
125 | G3SensVolVector pSVVector | |
126 | = pGeometryManager->GetG3SensVolVector(); | |
127 | ||
128 | G4int nofSV = pSVVector.entries(); | |
129 | if (nofSV>0) | |
130 | for (G4int isv=0; isv<nofSV; isv++) { | |
131 | G4LogicalVolume* lv = pSVVector[isv]; | |
132 | AliModule* module = fSDManager->FindAliModule(lv); | |
133 | fSDManager->CreateSD(lv, module); | |
134 | } | |
135 | } | |
136 | ||
137 | // public methods | |
138 | ||
139 | void AliMoreModulesConstruction::AddModule(G4String moduleName, G4int version, | |
140 | AliModuleType moduleType) | |
141 | { | |
142 | // Adds module specified by name, version and type. | |
143 | // --- | |
144 | ||
145 | // create module construction | |
146 | AliSingleModuleConstruction* moduleConstruction | |
147 | = new AliSingleModuleConstruction(moduleName, version, moduleType); | |
148 | ||
149 | // add module, module construction to vectors | |
150 | fModuleConstructionVector.push_back(moduleConstruction); | |
151 | } | |
152 | ||
153 | void AliMoreModulesConstruction::Construct() | |
154 | { | |
155 | // Constructs geometry. | |
156 | // G3 tables are process for all modules alltogether. | |
157 | // -- | |
158 | ||
159 | // number of modules | |
160 | G4int nofModules = fModuleConstructionVector.size(); | |
161 | ||
162 | if (nofModules == 0) { | |
163 | AliGlobals::Warning( | |
164 | "AliMoreModulesConstruction::Construct(): No modules are defined."); | |
165 | } | |
166 | else { | |
167 | // get geometry manager | |
168 | TG4GeometryManager* pGeometryManager = TG4GeometryManager::Instance(); | |
169 | ||
170 | G4bool allLVSensitive = false; | |
171 | G4int i; | |
172 | for (i=0; i<nofModules; i++) { | |
173 | ||
174 | fModuleConstructionVector[i]->Configure(); | |
175 | ||
176 | // register module name in the name map | |
177 | AliModule* module = fModuleConstructionVector[i]->GetAliModule(); | |
178 | pGeometryManager->SetMapSecond(module->GetName()); | |
179 | ||
180 | G4bool readGeometry = fModuleConstructionVector[i]->GetReadGeometry(); | |
181 | G4bool writeGeometry = fModuleConstructionVector[i]->GetWriteGeometry(); | |
182 | G4String dataFilePath = fModuleConstructionVector[i]->GetDataFilePath(); | |
183 | ||
184 | if (readGeometry) { | |
185 | // create G3 geometry from g3calls.dat | |
186 | pGeometryManager->SetWriteGeometry(false); | |
187 | pGeometryManager->ReadG3Geometry(dataFilePath); | |
188 | } | |
189 | else { | |
190 | // set geometry output stream for this module | |
191 | pGeometryManager->SetWriteGeometry(writeGeometry); | |
192 | if (writeGeometry) | |
193 | pGeometryManager->OpenOutFile(dataFilePath); | |
194 | ||
195 | // create geometry from AliRoot | |
196 | ||
197 | // construct materials | |
198 | module->CreateMaterials(); | |
199 | ||
200 | // construct G3 geometry | |
201 | module->CreateGeometry(); | |
202 | } | |
203 | ||
204 | // all logical volumes will be made sensitive if any | |
205 | // module requires this | |
206 | if (fModuleConstructionVector[i]->GetAllLVSensitive()) | |
207 | allLVSensitive = true; | |
208 | } | |
209 | ||
210 | // construct G4 geometry | |
211 | pGeometryManager->CreateG4Geometry(); | |
212 | ||
213 | // print name map | |
214 | // pGeometryManager->PrintNameMap(); | |
215 | ||
216 | // create sensitive detectors | |
217 | CreateSensitiveDetectors(allLVSensitive); | |
218 | // each single module construction can build sensitive detectors | |
219 | // for all modules as all modules are in memory | |
220 | // (volumes entities do not know to which module belong) | |
221 | ||
222 | for (i=0; i<nofModules; i++) { | |
223 | // set the detector frame (envelope) | |
224 | // (without warning output if enevelope is not defined) | |
225 | fModuleConstructionVector[i]->SetDetFrame(false); | |
226 | ||
227 | // build sensitive detectors table | |
228 | fModuleConstructionVector[i]->GetAliModule()->Init(); | |
229 | } | |
230 | ||
231 | // reset TG4GeometryManager | |
232 | pGeometryManager->ClearG3Tables(); | |
233 | ||
234 | // print current total number of logical volumes | |
235 | G4cout << "Current total number of sensitive volumes: " | |
236 | << pGeometryManager->NofVolumes() << endl; | |
237 | ||
238 | #ifdef ALICE_VISUALIZE | |
239 | // set visualization attributes | |
240 | for (i=0; i<nofModules; i++) { | |
241 | if (fModuleConstructionVector[i]->GetDetFrame()) { | |
242 | fModuleConstructionVector[i]->SetDetVisibility(true); | |
243 | fModuleConstructionVector[i]->SetDetColour("Yellow"); | |
244 | } | |
245 | } | |
246 | #endif | |
247 | } | |
248 | } |