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 | // SBTvoxelMessenger.cc |
---|
28 | // |
---|
29 | // Implementation of the messenger for the voxel test |
---|
30 | // |
---|
31 | |
---|
32 | #include "SBTvoxelMessenger.hh" |
---|
33 | #include "SBTvoxel.hh" |
---|
34 | #include "SBTVisManager.hh" |
---|
35 | #include "G4SolidQuery.hh" |
---|
36 | |
---|
37 | #include "G4VoxelLimits.hh" |
---|
38 | #include "G4AffineTransform.hh" |
---|
39 | |
---|
40 | #include "G4ios.hh" |
---|
41 | #include "G4UIdirectory.hh" |
---|
42 | #include "G4UIcommand.hh" |
---|
43 | #include "G4UIcmdWith3VectorAndUnit.hh" |
---|
44 | #include "G4UIcmdWithAnInteger.hh" |
---|
45 | #include "G4UIcmdWithAString.hh" |
---|
46 | #include "G4UIcmdWithoutParameter.hh" |
---|
47 | #include "G4UIcmdWith3Vector.hh" |
---|
48 | |
---|
49 | #include "G4UIcmdWithPargs.hh" |
---|
50 | #include "G4UIcmdPargDouble.hh" |
---|
51 | #include "G4UIcmdPargInteger.hh" |
---|
52 | |
---|
53 | #include <fstream> |
---|
54 | |
---|
55 | // |
---|
56 | // Constructor |
---|
57 | // |
---|
58 | SBTvoxelMessenger::SBTvoxelMessenger( const G4String prefix, |
---|
59 | const G4SolidQuery *theSolidQuery, SBTVisManager *theVisManager ) |
---|
60 | : tester(), errorFile("sbtvoxel.log") |
---|
61 | { |
---|
62 | // |
---|
63 | // Initialize |
---|
64 | // |
---|
65 | visManager = theVisManager; |
---|
66 | translate = new G4ThreeVector; |
---|
67 | rotate = new G4RotationMatrix; |
---|
68 | voxel = 0; |
---|
69 | point = 0; |
---|
70 | limits[0] = +1000.0; |
---|
71 | limits[1] = -1000.0; |
---|
72 | axis = kXAxis; |
---|
73 | |
---|
74 | // |
---|
75 | // Store solid query |
---|
76 | // |
---|
77 | solidQuery = theSolidQuery; |
---|
78 | |
---|
79 | // |
---|
80 | // Declare directory |
---|
81 | // |
---|
82 | voxelDirectory = new G4UIdirectory( prefix ); |
---|
83 | voxelDirectory->SetGuidance( "Controls for CSG batch voxel tester" ); |
---|
84 | |
---|
85 | // |
---|
86 | // Target command |
---|
87 | // |
---|
88 | G4String com = prefix+"target"; |
---|
89 | targetCmd = new G4UIcmdWith3VectorAndUnit( com, this ); |
---|
90 | targetCmd->SetGuidance( "Center of distribution of random points" ); |
---|
91 | targetCmd->SetParameterName( "X", "Y", "Z", true, true ); |
---|
92 | |
---|
93 | // |
---|
94 | // Widths command |
---|
95 | // |
---|
96 | com = prefix+"widths"; |
---|
97 | widthsCmd = new G4UIcmdWith3VectorAndUnit( com, this ); |
---|
98 | widthsCmd->SetGuidance( "Widths of distribution of random points" ); |
---|
99 | widthsCmd->SetParameterName( "Dx", "Dy", "Dz", true, true ); |
---|
100 | |
---|
101 | // |
---|
102 | // Max Voxels command |
---|
103 | // |
---|
104 | com = prefix+"maxVoxels"; |
---|
105 | maxVoxelsCmd = new G4UIcmdWithAnInteger( com, this ); |
---|
106 | maxVoxelsCmd->SetGuidance( "Maximum number of Voxels before test ends" ); |
---|
107 | |
---|
108 | // |
---|
109 | // Max Errors command |
---|
110 | // |
---|
111 | com = prefix+"maxErrors"; |
---|
112 | maxErrorsCmd = new G4UIcmdWithAnInteger( com, this ); |
---|
113 | maxErrorsCmd->SetGuidance( "Maximum number of errors before test ends" ); |
---|
114 | |
---|
115 | // |
---|
116 | // Error filename command |
---|
117 | // |
---|
118 | com = prefix+"errorFileName"; |
---|
119 | errorFileCmd = new G4UIcmdWithAString( com, this ); |
---|
120 | errorFileCmd->SetGuidance( "Filename in which to send error listings" ); |
---|
121 | |
---|
122 | // |
---|
123 | // Run command |
---|
124 | // |
---|
125 | com = prefix+"run"; |
---|
126 | runCmd = new G4UIcmdWithoutParameter( com, this ); |
---|
127 | runCmd->SetGuidance( "Execute voxel test" ); |
---|
128 | |
---|
129 | // |
---|
130 | // Picture directory |
---|
131 | // |
---|
132 | com = prefix+"picture/"; |
---|
133 | pictDirectory = new G4UIdirectory( com ); |
---|
134 | pictDirectory->SetGuidance( "Controls for drawing CSG voxel tests" ); |
---|
135 | |
---|
136 | // |
---|
137 | // picture/voxel command |
---|
138 | // |
---|
139 | pictVoxelArgs[0] = new G4UIcmdPargDouble( "xmin", 1.0, mm ); |
---|
140 | pictVoxelArgs[1] = new G4UIcmdPargDouble( "xmax", 1.0, mm ); |
---|
141 | pictVoxelArgs[2] = new G4UIcmdPargDouble( "ymin", 1.0, mm ); |
---|
142 | pictVoxelArgs[3] = new G4UIcmdPargDouble( "ymax", 1.0, mm ); |
---|
143 | pictVoxelArgs[4] = new G4UIcmdPargDouble( "zmin", 1.0, mm ); |
---|
144 | pictVoxelArgs[5] = new G4UIcmdPargDouble( "zmax", 1.0, mm ); |
---|
145 | com = prefix+"picture/voxel"; |
---|
146 | pictVoxelCmd = new G4UIcmdWithPargs( com, this, pictVoxelArgs, 6 ); |
---|
147 | pictVoxelCmd->SetGuidance( "Specifies the voxel to draw" ); |
---|
148 | |
---|
149 | // |
---|
150 | // picture/translate command |
---|
151 | // |
---|
152 | com = prefix+"picture/translate"; |
---|
153 | pictTranCmd = new G4UIcmdWith3Vector( com, this ); |
---|
154 | pictTranCmd->SetGuidance( "Specifies the voxel translation" ); |
---|
155 | |
---|
156 | // |
---|
157 | // picture/rotate command |
---|
158 | // |
---|
159 | pictRotArgs[0] = new G4UIcmdPargDouble( "xaxis", 1.0, 1 ); |
---|
160 | pictRotArgs[1] = new G4UIcmdPargDouble( "yaxis", 1.0, 1 ); |
---|
161 | pictRotArgs[2] = new G4UIcmdPargDouble( "zaxis", 1.0, 1 ); |
---|
162 | pictRotArgs[3] = new G4UIcmdPargDouble( "amount", 1.0, 1 ); |
---|
163 | com = prefix+"picture/rotate"; |
---|
164 | pictRotCmd = new G4UIcmdWithPargs( com, this, pictRotArgs, 4 ); |
---|
165 | pictRotCmd->SetGuidance( "Specifies rotation matrix" ); |
---|
166 | |
---|
167 | // |
---|
168 | // picture/point command |
---|
169 | // |
---|
170 | com = prefix+"picture/point"; |
---|
171 | pictPointCmd = new G4UIcmdWith3Vector( com, this ); |
---|
172 | pictPointCmd->SetGuidance( "Specifies one point to include in the picture" ); |
---|
173 | |
---|
174 | // |
---|
175 | // picture/limit command |
---|
176 | // |
---|
177 | pictLimitArgs[0] = new G4UIcmdPargInteger( "axis", 1 ); |
---|
178 | pictLimitArgs[1] = new G4UIcmdPargDouble( "min", 1.0, mm ); |
---|
179 | pictLimitArgs[2] = new G4UIcmdPargDouble( "max", 1.0, mm ); |
---|
180 | com = prefix+"picture/limit"; |
---|
181 | pictLimitCmd = new G4UIcmdWithPargs( com, this, pictLimitArgs, 3 ); |
---|
182 | pictLimitCmd->SetGuidance( "Specifies voxel limits" ); |
---|
183 | |
---|
184 | // |
---|
185 | // picture/draw command |
---|
186 | // |
---|
187 | com = prefix+"picture/draw"; |
---|
188 | pictDrawCmd = new G4UIcmdWithoutParameter( com, this ); |
---|
189 | pictDrawCmd->SetGuidance( "Draw the picture" ); |
---|
190 | |
---|
191 | // |
---|
192 | // picture/debug command |
---|
193 | // |
---|
194 | com = prefix+"picture/debug"; |
---|
195 | pictDebugCmd = new G4UIcmdWithoutParameter( com, this ); |
---|
196 | pictDebugCmd->SetGuidance( "Call CalculateExtent for the picture" ); |
---|
197 | } |
---|
198 | |
---|
199 | |
---|
200 | |
---|
201 | // |
---|
202 | // Destructor |
---|
203 | // |
---|
204 | SBTvoxelMessenger::~SBTvoxelMessenger() |
---|
205 | { |
---|
206 | if (voxel) delete voxel; |
---|
207 | if (point) delete point; |
---|
208 | |
---|
209 | // delete targetCmd; |
---|
210 | // delete widthsCmd; |
---|
211 | // delete maxVoxelsCmd; |
---|
212 | // delete maxVoxelsCmd; |
---|
213 | // delete errorFileCmd; |
---|
214 | delete voxelDirectory; |
---|
215 | |
---|
216 | delete pictDirectory; |
---|
217 | } |
---|
218 | |
---|
219 | |
---|
220 | // |
---|
221 | // InvokeTest |
---|
222 | // |
---|
223 | // Run the test |
---|
224 | // |
---|
225 | void SBTvoxelMessenger::InvokeTest() |
---|
226 | { |
---|
227 | // |
---|
228 | // Is there a Solid to test? |
---|
229 | // |
---|
230 | G4VSolid *testSolid = solidQuery->GetSolid(); |
---|
231 | if (testSolid == 0) { |
---|
232 | G4cerr << "Please initialize geometry before running test" << G4endl; |
---|
233 | G4cerr << "Test ABORTED" << G4endl; |
---|
234 | return; |
---|
235 | } |
---|
236 | |
---|
237 | // |
---|
238 | // Open output file |
---|
239 | // |
---|
240 | std::ofstream logFile( errorFile ); |
---|
241 | |
---|
242 | // |
---|
243 | // Run the test |
---|
244 | // |
---|
245 | tester.RunTest( testSolid, logFile ); |
---|
246 | } |
---|
247 | |
---|
248 | |
---|
249 | void SBTvoxelMessenger::Draw() |
---|
250 | { |
---|
251 | if (!visManager) |
---|
252 | G4cerr << "Visualization is not available in this executable" << G4endl; |
---|
253 | |
---|
254 | // |
---|
255 | // Is there a Solid to test? |
---|
256 | // |
---|
257 | G4VSolid *testSolid = solidQuery->GetSolid(); |
---|
258 | if (testSolid == 0) { |
---|
259 | G4cerr << "Please initialize geometry before running test" << G4endl; |
---|
260 | return; |
---|
261 | } |
---|
262 | |
---|
263 | // |
---|
264 | // Is there a voxel to test? |
---|
265 | // |
---|
266 | if (voxel == 0) { |
---|
267 | G4cerr << "Please specify a voxel to draw using /pict/voxel command" << G4endl; |
---|
268 | return; |
---|
269 | } |
---|
270 | |
---|
271 | // |
---|
272 | // Go do it |
---|
273 | // |
---|
274 | visManager->BuildFakeWorld(); |
---|
275 | |
---|
276 | G4AffineTransform transform( *rotate, *translate ); |
---|
277 | |
---|
278 | tester.Draw( testSolid, *voxel, transform, point, point ? 1 : 0, axis, limits, visManager ); |
---|
279 | } |
---|
280 | |
---|
281 | |
---|
282 | void SBTvoxelMessenger::Debug() |
---|
283 | { |
---|
284 | // |
---|
285 | // Is there a Solid to test? |
---|
286 | // |
---|
287 | G4VSolid *testSolid = solidQuery->GetSolid(); |
---|
288 | if (testSolid == 0) { |
---|
289 | G4cerr << "Please initialize geometry before running test" << G4endl; |
---|
290 | return; |
---|
291 | } |
---|
292 | |
---|
293 | // |
---|
294 | // Is there a voxel to test? |
---|
295 | // |
---|
296 | if (voxel == 0) { |
---|
297 | G4cerr << "Please specify a voxel to debug using /pict/voxel command" << G4endl; |
---|
298 | return; |
---|
299 | } |
---|
300 | |
---|
301 | // |
---|
302 | // Go do it |
---|
303 | // |
---|
304 | G4AffineTransform transform( *rotate, *translate ); |
---|
305 | |
---|
306 | tester.Debug( testSolid, axis, *voxel, transform, point ); |
---|
307 | } |
---|
308 | |
---|
309 | |
---|
310 | // |
---|
311 | // SetNewValue |
---|
312 | // |
---|
313 | // Call by the UI when user requests a change |
---|
314 | // |
---|
315 | void SBTvoxelMessenger::SetNewValue( G4UIcommand *command, G4String newValues ) |
---|
316 | { |
---|
317 | if (command == targetCmd) { |
---|
318 | tester.SetTarget( targetCmd->GetNew3VectorValue( newValues ) ); |
---|
319 | } |
---|
320 | else if (command == widthsCmd) { |
---|
321 | tester.SetWidths( widthsCmd->GetNew3VectorValue( newValues ) ); |
---|
322 | } |
---|
323 | else if (command == maxVoxelsCmd) { |
---|
324 | tester.SetMaxVoxels( maxVoxelsCmd->GetNewIntValue( newValues ) ); |
---|
325 | } |
---|
326 | else if (command == maxErrorsCmd) { |
---|
327 | tester.SetMaxErrors( maxErrorsCmd->GetNewIntValue( newValues ) ); |
---|
328 | } |
---|
329 | else if (command == errorFileCmd) { |
---|
330 | errorFile = newValues; |
---|
331 | } |
---|
332 | else if (command == runCmd) { |
---|
333 | InvokeTest(); |
---|
334 | } |
---|
335 | else if (command == pictVoxelCmd) { |
---|
336 | if (pictVoxelCmd->GetArguments( newValues )) { |
---|
337 | G4UIcmdPargDouble *minX = (G4UIcmdPargDouble *)pictVoxelArgs[0], |
---|
338 | *maxX = (G4UIcmdPargDouble *)pictVoxelArgs[1], |
---|
339 | *minY = (G4UIcmdPargDouble *)pictVoxelArgs[2], |
---|
340 | *maxY = (G4UIcmdPargDouble *)pictVoxelArgs[3], |
---|
341 | *minZ = (G4UIcmdPargDouble *)pictVoxelArgs[4], |
---|
342 | *maxZ = (G4UIcmdPargDouble *)pictVoxelArgs[5]; |
---|
343 | |
---|
344 | if (voxel) delete voxel; |
---|
345 | |
---|
346 | voxel = new G4VoxelLimits; |
---|
347 | |
---|
348 | if (minX->GetValue() < maxX->GetValue()) |
---|
349 | voxel->AddLimit( kXAxis, minX->GetValue(), maxX->GetValue() ); |
---|
350 | if (minY->GetValue() < maxY->GetValue()) |
---|
351 | voxel->AddLimit( kYAxis, minY->GetValue(), maxY->GetValue() ); |
---|
352 | if (minZ->GetValue() < maxZ->GetValue()) |
---|
353 | voxel->AddLimit( kZAxis, minZ->GetValue(), maxZ->GetValue() ); |
---|
354 | } |
---|
355 | else |
---|
356 | G4cerr << "Syntax error" << G4endl; |
---|
357 | } |
---|
358 | else if (command == pictTranCmd) { |
---|
359 | *translate = pictTranCmd->GetNew3VectorValue( newValues ); |
---|
360 | } |
---|
361 | else if (command == pictRotCmd) { |
---|
362 | if (pictRotCmd->GetArguments( newValues )) { |
---|
363 | G4UIcmdPargDouble *rotX = (G4UIcmdPargDouble *)pictRotArgs[0], |
---|
364 | *rotY = (G4UIcmdPargDouble *)pictRotArgs[1], |
---|
365 | *rotZ = (G4UIcmdPargDouble *)pictRotArgs[2], |
---|
366 | *amount = (G4UIcmdPargDouble *)pictRotArgs[3]; |
---|
367 | |
---|
368 | G4ThreeVector rotAxis( rotX->GetValue(), rotY->GetValue(), rotZ->GetValue() ); |
---|
369 | |
---|
370 | delete rotate; |
---|
371 | rotate = new G4RotationMatrix(); |
---|
372 | rotate->rotate( amount->GetValue(), &rotAxis ); |
---|
373 | } |
---|
374 | else |
---|
375 | G4cerr << "Syntax error" << G4endl; |
---|
376 | } |
---|
377 | else if (command == pictPointCmd) { |
---|
378 | if (point) delete point; |
---|
379 | point = new G4ThreeVector(); |
---|
380 | *point = pictPointCmd->GetNew3VectorValue( newValues ); |
---|
381 | } |
---|
382 | else if (command == pictLimitCmd) { |
---|
383 | if (pictLimitCmd->GetArguments( newValues )) { |
---|
384 | static const EAxis axes[3] = { kXAxis, kYAxis, kZAxis }; |
---|
385 | |
---|
386 | G4UIcmdPargInteger *axisArg = (G4UIcmdPargInteger *)pictLimitArgs[0]; |
---|
387 | G4UIcmdPargDouble *min = (G4UIcmdPargDouble *)pictLimitArgs[1], |
---|
388 | *max = (G4UIcmdPargDouble *)pictLimitArgs[2]; |
---|
389 | |
---|
390 | G4int axisIndex = axisArg->GetValue(); |
---|
391 | if (axisIndex < 0 || axisIndex > 2) { |
---|
392 | G4cerr << "Axis value must be from 0 to 2" << G4endl; |
---|
393 | } |
---|
394 | else { |
---|
395 | axis = axes[axisIndex]; |
---|
396 | limits[0] = min->GetValue(); |
---|
397 | limits[1] = max->GetValue(); |
---|
398 | } |
---|
399 | } |
---|
400 | else |
---|
401 | G4cerr << "Syntax error" << G4endl; |
---|
402 | } |
---|
403 | else if (command == pictDrawCmd) { |
---|
404 | Draw(); |
---|
405 | } |
---|
406 | else if (command == pictDebugCmd) { |
---|
407 | Debug(); |
---|
408 | } |
---|
409 | else { |
---|
410 | G4Exception( "Unrecognized command" ); |
---|
411 | } |
---|
412 | } |
---|
413 | |
---|
414 | |
---|
415 | |
---|
416 | // |
---|
417 | // GetCurrentValue |
---|
418 | // |
---|
419 | G4String SBTvoxelMessenger::GetCurrentValue( G4UIcommand *command ) |
---|
420 | { |
---|
421 | if (command == targetCmd) { |
---|
422 | return targetCmd->ConvertToString( tester.GetTarget(), "m" ); |
---|
423 | } |
---|
424 | else if (command == widthsCmd) { |
---|
425 | return widthsCmd->ConvertToString( tester.GetWidths(), "m" ); |
---|
426 | } |
---|
427 | else if (command == maxVoxelsCmd) { |
---|
428 | return maxVoxelsCmd->ConvertToString( tester.GetMaxVoxels() ); |
---|
429 | } |
---|
430 | else if (command == maxErrorsCmd) { |
---|
431 | return maxErrorsCmd->ConvertToString( tester.GetMaxErrors() ); |
---|
432 | } |
---|
433 | else if (command == errorFileCmd) { |
---|
434 | return errorFile; |
---|
435 | } |
---|
436 | else if (command == runCmd) { |
---|
437 | return ""; |
---|
438 | } |
---|
439 | else if (command == pictVoxelCmd) { |
---|
440 | return ""; |
---|
441 | } |
---|
442 | else if (command == pictTranCmd) { |
---|
443 | return ""; |
---|
444 | } |
---|
445 | else if (command == pictRotCmd) { |
---|
446 | return ""; |
---|
447 | } |
---|
448 | else if (command == pictPointCmd) { |
---|
449 | return ""; |
---|
450 | } |
---|
451 | else if (command == pictLimitCmd) { |
---|
452 | return ""; |
---|
453 | } |
---|
454 | else if (command == pictDebugCmd) { |
---|
455 | return ""; |
---|
456 | } |
---|
457 | else if (command == pictDrawCmd) { |
---|
458 | return ""; |
---|
459 | } |
---|
460 | |
---|
461 | G4Exception( "Unrecognized command" ); |
---|
462 | return "foo!"; |
---|
463 | } |
---|