1 | // this : |
---|
2 | #include <G4Lab/DigitsCollectionAccessor.h> |
---|
3 | |
---|
4 | // Inventor : |
---|
5 | #include <Inventor/nodes/SoSeparator.h> |
---|
6 | #include <Inventor/nodes/SoTransform.h> |
---|
7 | #include <Inventor/nodes/SoLightModel.h> |
---|
8 | #include <Inventor/nodes/SoDrawStyle.h> |
---|
9 | #include <Inventor/nodes/SoCoordinate3.h> |
---|
10 | |
---|
11 | // HEPVis : |
---|
12 | #include <HEPVis/misc/SoStyleCache.h> |
---|
13 | #include <HEPVis/nodes/SoPolyhedron.h> |
---|
14 | #include <HEPVis/nodes/SoHighlightMaterial.h> |
---|
15 | |
---|
16 | #ifdef WIN32 |
---|
17 | #undef pascal // Clash between windef.h and Geant4/SystemOfUnits.hh |
---|
18 | #endif |
---|
19 | |
---|
20 | // Geant4 : |
---|
21 | #include <G4RunManager.hh> |
---|
22 | #include <G4Event.hh> |
---|
23 | #include <G4DigiManager.hh> |
---|
24 | #include <G4DCofThisEvent.hh> |
---|
25 | #include <G4LogicalVolume.hh> |
---|
26 | #include <G4Transform3D.hh> |
---|
27 | #include <G4AttDef.hh> |
---|
28 | #include <G4AttValue.hh> |
---|
29 | #include <G4Polyhedron.hh> |
---|
30 | #include <G4Colour.hh> |
---|
31 | #include <G4VDigi.hh> |
---|
32 | #include <G4VDigiCollection.hh> |
---|
33 | |
---|
34 | // Lib : |
---|
35 | #include <Slash/Core/ISession.h> |
---|
36 | #include <Slash/Data/IIterator.h> |
---|
37 | #include <Lib/Out.h> |
---|
38 | #include <Lib/Value.h> |
---|
39 | #include <Lib/smanip.h> |
---|
40 | #include <Lib/sout.h> |
---|
41 | |
---|
42 | // G4Lab : |
---|
43 | #include <G4Lab/Transform3D.h> |
---|
44 | #include <G4Lab/Polyhedron.h> |
---|
45 | |
---|
46 | ////////////////////////////////////////////////////////////////////////////// |
---|
47 | G4Lab::DigitsCollectionBaseAccessor::DigitsCollectionBaseAccessor( |
---|
48 | Slash::Core::ISession& aSession |
---|
49 | ,const std::string& aDC |
---|
50 | ) |
---|
51 | :OnX::InventorAccessor(aSession) |
---|
52 | ,fType("DC") |
---|
53 | ,fAttDefs(0) |
---|
54 | ,fDC(aDC) |
---|
55 | ////////////////////////////////////////////////////////////////////////////// |
---|
56 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
57 | { |
---|
58 | if(fDC!="") fType = fDC; |
---|
59 | } |
---|
60 | ////////////////////////////////////////////////////////////////////////////// |
---|
61 | G4Lab::DigitsCollectionBaseAccessor::~DigitsCollectionBaseAccessor( |
---|
62 | ) |
---|
63 | ////////////////////////////////////////////////////////////////////////////// |
---|
64 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
65 | { |
---|
66 | } |
---|
67 | ////////////////////////////////////////////////////////////////////////////// |
---|
68 | std::string G4Lab::DigitsCollectionBaseAccessor::name( |
---|
69 | ) const |
---|
70 | ////////////////////////////////////////////////////////////////////////////// |
---|
71 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
72 | { |
---|
73 | return fType; |
---|
74 | } |
---|
75 | ////////////////////////////////////////////////////////////////////////////// |
---|
76 | const std::string& G4Lab::DigitsCollectionBaseAccessor::DCName( |
---|
77 | ) const |
---|
78 | ////////////////////////////////////////////////////////////////////////////// |
---|
79 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
80 | { |
---|
81 | return fDC; |
---|
82 | } |
---|
83 | ////////////////////////////////////////////////////////////////////////////// |
---|
84 | Slash::Core::IValue* G4Lab::DigitsCollectionBaseAccessor::findValue( |
---|
85 | Slash::Data::IAccessor::Data aData |
---|
86 | ,const std::string& aName |
---|
87 | ,void* |
---|
88 | ) |
---|
89 | ////////////////////////////////////////////////////////////////////////////// |
---|
90 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
91 | { |
---|
92 | if(!fAttDefs) { |
---|
93 | Lib::Out out(printer()); |
---|
94 | out << "G4Lab::DigitsCollectionBaseAccessor::value :" |
---|
95 | << " problem to get AttDefs fot digi collection " << Lib::sout(fDC) |
---|
96 | << "." |
---|
97 | << Lib::endl; |
---|
98 | return new Lib::Value(); |
---|
99 | } |
---|
100 | |
---|
101 | G4VDigi* obj = (G4VDigi*)aData; |
---|
102 | std::vector<G4AttValue>* vec = obj->CreateAttValues(); |
---|
103 | if(!vec) { |
---|
104 | Lib::Out out(printer()); |
---|
105 | out << "G4Lab::DigitsCollectionBaseAccessor::value :" |
---|
106 | << " problem to get AttValues fot digi collection " << Lib::sout(fDC) |
---|
107 | << "." |
---|
108 | << Lib::endl; |
---|
109 | return new Lib::Value(); |
---|
110 | } |
---|
111 | |
---|
112 | unsigned int number = vec->size(); |
---|
113 | for(unsigned int index=0;index<number;index++) { |
---|
114 | const G4AttValue& val = (*vec)[index]; |
---|
115 | if(aName==val.GetName()) { |
---|
116 | const std::string& stype = (*fAttDefs)[val.GetName()].GetValueType(); |
---|
117 | if(stype=="G4int") { |
---|
118 | int v; |
---|
119 | if(!Lib::smanip::toint(val.GetValue(),v)) { |
---|
120 | Lib::Out out(printer()); |
---|
121 | out << "G4Lab::DigitsCollectionBaseAccessor::value :" |
---|
122 | << " " << Lib::sout(val.GetValue()) << " not a G4int." |
---|
123 | << Lib::endl; |
---|
124 | delete vec; |
---|
125 | return new Lib::Value(); |
---|
126 | } |
---|
127 | delete vec; |
---|
128 | return new Lib::Value(v); |
---|
129 | } else if((stype=="G4double")||(stype=="G4float")) { |
---|
130 | double v; |
---|
131 | if(!Lib::smanip::todouble(val.GetValue(),v)) { |
---|
132 | Lib::Out out(printer()); |
---|
133 | out << "G4Lab::DigitsCollectionBaseAccessor::value :" |
---|
134 | << " " << Lib::sout(val.GetValue()) << " not a double." |
---|
135 | << Lib::endl; |
---|
136 | delete vec; |
---|
137 | return new Lib::Value(); |
---|
138 | } |
---|
139 | delete vec; |
---|
140 | return new Lib::Value(v); |
---|
141 | } else { |
---|
142 | delete vec; |
---|
143 | return new Lib::Value(val.GetValue()); |
---|
144 | } |
---|
145 | } |
---|
146 | } |
---|
147 | Lib::Out out(printer()); |
---|
148 | out << "G4Lab::DigitsCollectionBaseAccessor::value :" |
---|
149 | << " AttValue not found for property " << Lib::sout(aName) << "." |
---|
150 | << Lib::endl; |
---|
151 | delete vec; |
---|
152 | return new Lib::Value(); |
---|
153 | } |
---|
154 | ////////////////////////////////////////////////////////////////////////////// |
---|
155 | void G4Lab::DigitsCollectionBaseAccessor::beginVisualize( |
---|
156 | ) |
---|
157 | ////////////////////////////////////////////////////////////////////////////// |
---|
158 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
159 | { |
---|
160 | OnX::InventorAccessor::beginVisualize(); |
---|
161 | |
---|
162 | std::string style = "DC("+fDC+")"; |
---|
163 | if(!isStyle(style)) style = "DC"; //Default. |
---|
164 | fillSoGC(style); |
---|
165 | } |
---|
166 | ////////////////////////////////////////////////////////////////////////////// |
---|
167 | void G4Lab::DigitsCollectionBaseAccessor::visualize( |
---|
168 | Slash::Data::IAccessor::Data aData |
---|
169 | ,void* |
---|
170 | ) |
---|
171 | ////////////////////////////////////////////////////////////////////////////// |
---|
172 | // The digi must have the "LV" and "TSF" AttDef. |
---|
173 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
174 | { |
---|
175 | G4VDigi* obj = (G4VDigi*)aData; |
---|
176 | G4LogicalVolume* lv = digiLogicalVolume(*obj); |
---|
177 | if(!lv) return; |
---|
178 | G4Transform3D* tsf = digiTransform3D(*obj); |
---|
179 | if(!tsf) return; |
---|
180 | G4VSolid* solid = lv->GetSolid(); |
---|
181 | if(!solid) return; |
---|
182 | //G4Material* material = lv->GetMaterial(); |
---|
183 | |
---|
184 | int old_nstep = HepPolyhedron::GetNumberOfRotationSteps(); |
---|
185 | HepPolyhedron::SetNumberOfRotationSteps(fSoGC.getRotationSteps()); |
---|
186 | G4Polyhedron* g4Polyhedron = solid->CreatePolyhedron(); |
---|
187 | HepPolyhedron::SetNumberOfRotationSteps(old_nstep); |
---|
188 | if(!g4Polyhedron) return; |
---|
189 | |
---|
190 | G4Lab::Polyhedron* polyhedron = new G4Lab::Polyhedron(*g4Polyhedron); |
---|
191 | delete g4Polyhedron; |
---|
192 | if(!polyhedron) return; |
---|
193 | |
---|
194 | SoSeparator* separator = new SoSeparator; |
---|
195 | separator->setName("sceneGraph"); |
---|
196 | |
---|
197 | G4Colour color; |
---|
198 | if(digiColor(*obj,color)) { |
---|
199 | SbColor sbColor((float)color.GetRed(), |
---|
200 | (float)color.GetGreen(), |
---|
201 | (float)color.GetBlue()); |
---|
202 | float transp = 1.0F - (float)color.GetAlpha(); |
---|
203 | SoStyleCache* styleCache = fSoGC.getStyleCache(); |
---|
204 | separator->addChild( |
---|
205 | styleCache->getHighlightMaterial |
---|
206 | (sbColor,fSoGC.getHighlightColor(),transp)); |
---|
207 | } else { |
---|
208 | separator->addChild(fSoGC.getHighlightMaterial()); |
---|
209 | } |
---|
210 | |
---|
211 | separator->addChild(fSoGC.getDrawStyle()); |
---|
212 | separator->addChild(fSoGC.getLightModel()); |
---|
213 | |
---|
214 | SoTransform* transform = new SoTransform; |
---|
215 | G4Lab::Transform3D* t = new G4Lab::Transform3D(*tsf); |
---|
216 | SbMatrix* matrix = t->getMatrix(); |
---|
217 | transform->setMatrix(*matrix); |
---|
218 | delete matrix; |
---|
219 | delete t; |
---|
220 | separator->addChild(transform); |
---|
221 | |
---|
222 | // Build name (for picking) : |
---|
223 | std::string s; |
---|
224 | Lib::smanip::printf(s,128,"%s/0x%lx",fDC.c_str(),(unsigned long)obj); |
---|
225 | SbName name(s.c_str()); |
---|
226 | //FIXME : can't cache due to the below setName for picking. |
---|
227 | //FIXME SoPolyhedron* soPolyhedron = fSoGC.getPolyhedron(*polyhedron); |
---|
228 | SoPolyhedron* soPolyhedron = new SoPolyhedron(*polyhedron); |
---|
229 | if(fSoGC.getModeling()==SbModeling_wire_frame) { |
---|
230 | soPolyhedron->solid.setValue(FALSE); |
---|
231 | //FIXME : handle reduceWireFrame. |
---|
232 | soPolyhedron->reducedWireFrame.setValue(TRUE); |
---|
233 | } else { |
---|
234 | soPolyhedron->solid.setValue(TRUE); |
---|
235 | } |
---|
236 | delete polyhedron; |
---|
237 | soPolyhedron->setName(name); |
---|
238 | |
---|
239 | separator->addChild(soPolyhedron); |
---|
240 | |
---|
241 | fSoRegion->doIt(SbAddNode(separator,"dynamicScene")); |
---|
242 | } |
---|
243 | ////////////////////////////////////////////////////////////////////////////// |
---|
244 | void G4Lab::DigitsCollectionBaseAccessor::setProperties( |
---|
245 | G4VDigi& aDigi |
---|
246 | ) |
---|
247 | ////////////////////////////////////////////////////////////////////////////// |
---|
248 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
249 | { |
---|
250 | fAttDefs = (std::map<G4String,G4AttDef>*)aDigi.GetAttDefs(); |
---|
251 | if(!fAttDefs) return; |
---|
252 | std::map<G4String,G4AttDef>::const_iterator it; |
---|
253 | for(it=fAttDefs->begin();it!=fAttDefs->end();++it) { |
---|
254 | const std::string& sname = (*it).second.GetName(); |
---|
255 | if((sname!="LV") && |
---|
256 | (sname!="TSF") && |
---|
257 | (sname!="Color")) { |
---|
258 | const std::string& stype = (*it).second.GetValueType(); |
---|
259 | if(stype=="G4int") { |
---|
260 | addProperty((*it).first,Lib::Property::INTEGER); |
---|
261 | } else if((stype=="G4double")||(stype=="G4float")) { |
---|
262 | addProperty((*it).first,Lib::Property::DOUBLE); |
---|
263 | } else { |
---|
264 | addProperty((*it).first,Lib::Property::STRING); |
---|
265 | } |
---|
266 | } |
---|
267 | } |
---|
268 | } |
---|
269 | ////////////////////////////////////////////////////////////////////////////// |
---|
270 | ////////////////////////////////////////////////////////////////////////////// |
---|
271 | ////////////////////////////////////////////////////////////////////////////// |
---|
272 | G4VDigiCollection* G4Lab::DigitsCollectionBaseAccessor::getCollection( |
---|
273 | Slash::Core::IWriter& aPrinter |
---|
274 | ,const std::string& aName |
---|
275 | ) |
---|
276 | ////////////////////////////////////////////////////////////////////////////// |
---|
277 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
278 | { |
---|
279 | G4RunManager* runManager = G4RunManager::GetRunManager(); |
---|
280 | if(!runManager) { |
---|
281 | Lib::Out out(aPrinter); |
---|
282 | out << "No G4RunManager." << Lib::endl; |
---|
283 | return 0; |
---|
284 | } |
---|
285 | const G4Event* event = runManager->GetCurrentEvent(); |
---|
286 | if(!event) { |
---|
287 | Lib::Out out(aPrinter); |
---|
288 | out << "No event." << Lib::endl; |
---|
289 | return 0; |
---|
290 | } |
---|
291 | G4DigiManager* digiManager = G4DigiManager::GetDMpointer(); |
---|
292 | if(!digiManager) { |
---|
293 | Lib::Out out(aPrinter); |
---|
294 | out << "No G4DigiManager." << Lib::endl; |
---|
295 | return 0; |
---|
296 | } |
---|
297 | |
---|
298 | // Get Digi collection of this event |
---|
299 | G4DCofThisEvent* dce = event->GetDCofThisEvent(); |
---|
300 | if(!dce) { |
---|
301 | Lib::Out out(aPrinter); |
---|
302 | out << "No G4DCofThisEvent." << Lib::endl; |
---|
303 | return 0; |
---|
304 | } |
---|
305 | G4int cid = digiManager->GetDigiCollectionID(aName); |
---|
306 | int nc = dce->GetNumberOfCollections(); |
---|
307 | if((cid<0)||(cid>=nc)) { |
---|
308 | Lib::Out out(aPrinter); |
---|
309 | out << "Collection id not found for " << Lib::sout(aName) << Lib::endl; |
---|
310 | return 0; |
---|
311 | } |
---|
312 | |
---|
313 | G4VDigiCollection* dc = dce->GetDC(cid); |
---|
314 | if(!dce) { |
---|
315 | Lib::Out out(aPrinter); |
---|
316 | out << "No G4VDigiCollection for " << Lib::sout(aName) << Lib::endl; |
---|
317 | return 0; |
---|
318 | } |
---|
319 | return dc; |
---|
320 | } |
---|
321 | ////////////////////////////////////////////////////////////////////////////// |
---|
322 | G4LogicalVolume* G4Lab::DigitsCollectionBaseAccessor::digiLogicalVolume( |
---|
323 | const G4VDigi& aDigi |
---|
324 | ) |
---|
325 | ////////////////////////////////////////////////////////////////////////////// |
---|
326 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
327 | { |
---|
328 | std::vector<G4AttValue>* vec = aDigi.CreateAttValues(); |
---|
329 | if(!vec) return 0; |
---|
330 | std::string value; |
---|
331 | unsigned int number = vec->size(); |
---|
332 | for(unsigned int index=0;index<number;index++) { |
---|
333 | const G4AttValue& val = (*vec)[index]; |
---|
334 | if(val.GetName()=="LV") { |
---|
335 | std::string s = val.GetValue(); |
---|
336 | void* p; |
---|
337 | if(!Lib::smanip::topointer(s,p)) return 0; |
---|
338 | return (G4LogicalVolume*)p; //Beurk |
---|
339 | } |
---|
340 | } |
---|
341 | return 0; |
---|
342 | } |
---|
343 | ////////////////////////////////////////////////////////////////////////////// |
---|
344 | G4Transform3D* G4Lab::DigitsCollectionBaseAccessor::digiTransform3D( |
---|
345 | const G4VDigi& aDigi |
---|
346 | ) |
---|
347 | ////////////////////////////////////////////////////////////////////////////// |
---|
348 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
349 | { |
---|
350 | std::vector<G4AttValue>* vec = aDigi.CreateAttValues(); |
---|
351 | if(!vec) return 0; |
---|
352 | std::string value; |
---|
353 | unsigned int number = vec->size(); |
---|
354 | for(unsigned int index=0;index<number;index++) { |
---|
355 | const G4AttValue& val = (*vec)[index]; |
---|
356 | if(val.GetName()=="TSF") { |
---|
357 | std::string s = val.GetValue(); |
---|
358 | void* p; |
---|
359 | if(!Lib::smanip::topointer(s,p)) return 0; |
---|
360 | return (G4Transform3D*)p; //Beurk |
---|
361 | } |
---|
362 | } |
---|
363 | return 0; |
---|
364 | } |
---|
365 | ////////////////////////////////////////////////////////////////////////////// |
---|
366 | bool G4Lab::DigitsCollectionBaseAccessor::digiColor( |
---|
367 | const G4VDigi& aDigi |
---|
368 | ,G4Colour& aColor |
---|
369 | ) |
---|
370 | ////////////////////////////////////////////////////////////////////////////// |
---|
371 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
372 | { |
---|
373 | std::vector<G4AttValue>* vec = aDigi.CreateAttValues(); |
---|
374 | if(vec) { |
---|
375 | std::string value; |
---|
376 | unsigned int number = vec->size(); |
---|
377 | for(unsigned int index=0;index<number;index++) { |
---|
378 | const G4AttValue& val = (*vec)[index]; |
---|
379 | if(val.GetName()=="Color") { |
---|
380 | std::string s = val.GetValue(); |
---|
381 | double r,g,b,a; |
---|
382 | if(!Lib::smanip::torgba(s,r,g,b,a)) break; |
---|
383 | aColor = G4Colour(r,g,b,a); |
---|
384 | return true; |
---|
385 | } |
---|
386 | } |
---|
387 | } |
---|
388 | aColor = G4Colour(); |
---|
389 | return false; |
---|
390 | } |
---|