[1316] | 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 | // FredMessenger.cc |
---|
| 28 | // |
---|
| 29 | // Implementation of fred's (primary) options |
---|
| 30 | // |
---|
| 31 | |
---|
| 32 | #include "FredMessenger.hh" |
---|
| 33 | #include "FredTest3Messenger.hh" |
---|
| 34 | #include "FredVoxelTestMessenger.hh" |
---|
| 35 | |
---|
| 36 | #include "G4UIdirectory.hh" |
---|
| 37 | #include "G4UIcmdWithAString.hh" |
---|
| 38 | #include "G4UIcmdWithADouble.hh" |
---|
| 39 | #include "G4UIcmdWithoutParameter.hh" |
---|
| 40 | #include "G4UIcmdWithAnInteger.hh" |
---|
| 41 | #include "G4ios.hh" |
---|
| 42 | |
---|
| 43 | // |
---|
| 44 | // Constructor |
---|
| 45 | // |
---|
| 46 | FredMessenger::FredMessenger( ) |
---|
| 47 | { |
---|
| 48 | testVolume = 0; |
---|
| 49 | |
---|
| 50 | volumeNames[NATALIA] = "NATALIA"; // Yeah, yeah. Really cheezy, I know |
---|
| 51 | volumeNames[VOXEL] = "VOXEL"; |
---|
| 52 | |
---|
| 53 | volumeNames[BOX] = "BOX"; |
---|
| 54 | volumeNames[CONE] = "CONE"; |
---|
| 55 | volumeNames[CONE2] = "CONE2"; |
---|
| 56 | volumeNames[ORB] = "ORB"; |
---|
| 57 | volumeNames[PARA] = "PARA"; |
---|
| 58 | volumeNames[SPHERE] = "SPHERE"; |
---|
| 59 | volumeNames[TORUS1] = "TORUS1"; |
---|
| 60 | volumeNames[TORUS2] = "TORUS2"; |
---|
| 61 | volumeNames[TRAP] = "TRAP"; |
---|
| 62 | volumeNames[TRD] = "TRD"; |
---|
| 63 | volumeNames[TUBS] = "TUBS"; |
---|
| 64 | |
---|
| 65 | volumeNames[ELLIPS] = "ELLIPS"; |
---|
| 66 | volumeNames[ELCONE] = "ELCONE"; |
---|
| 67 | volumeNames[ELTUBE] = "ELTUBE"; |
---|
| 68 | volumeNames[EXTRU1] = "EXTRU1"; |
---|
| 69 | volumeNames[EXTRU2] = "EXTRU2"; |
---|
| 70 | volumeNames[EXTRU3] = "EXTRU3"; |
---|
| 71 | volumeNames[EXTRU4] = "EXTRU4"; |
---|
| 72 | volumeNames[HYPE] = "HYPE"; |
---|
| 73 | volumeNames[PCON] = "PCON"; |
---|
| 74 | volumeNames[PCON2] = "PCON2"; |
---|
| 75 | volumeNames[PCON3] = "PCON3"; |
---|
| 76 | volumeNames[PCON4] = "PCON4"; |
---|
| 77 | volumeNames[PGON] = "PGON"; |
---|
| 78 | volumeNames[PGON2] = "PGON2"; |
---|
| 79 | volumeNames[PGON3] = "PGON3"; |
---|
| 80 | volumeNames[TESSEL1] = "TESSEL1"; |
---|
| 81 | volumeNames[TESSEL2] = "TESSEL2"; |
---|
| 82 | volumeNames[TESSEL3] = "TESSEL3"; |
---|
| 83 | volumeNames[TESSEL4] = "TESSEL4"; |
---|
| 84 | volumeNames[TET] = "TET"; |
---|
| 85 | volumeNames[TWBOX] = "TWBOX"; |
---|
| 86 | volumeNames[TWTRAP1] = "TWTRAP1"; |
---|
| 87 | volumeNames[TWTRAP2] = "TWTRAP2"; |
---|
| 88 | volumeNames[TWTRD] = "TWTRD"; |
---|
| 89 | volumeNames[TWTUBS] = "TWTUBS"; |
---|
| 90 | |
---|
| 91 | volumeNames[BOOL1] = "BOOL1"; |
---|
| 92 | |
---|
| 93 | gunNames[SPRAY] = "SPRAY"; |
---|
| 94 | gunNames[GRID] = "GRID"; |
---|
| 95 | gunNames[G4] = "G4"; |
---|
| 96 | |
---|
| 97 | drawNames[NORMAL] = "NORMAL"; |
---|
| 98 | drawNames[SHADOW] = "SHADOW"; |
---|
| 99 | |
---|
| 100 | // |
---|
| 101 | // Defaults |
---|
| 102 | // |
---|
| 103 | testVolumeType = BOX; |
---|
| 104 | gunType = SPRAY; |
---|
| 105 | drawType = NORMAL; |
---|
| 106 | |
---|
| 107 | startPhi = 0; |
---|
| 108 | deltaPhi = 360; |
---|
| 109 | |
---|
| 110 | numSide = 8; |
---|
| 111 | |
---|
| 112 | // |
---|
| 113 | // Declare directory |
---|
| 114 | // |
---|
| 115 | fredDirectory = new G4UIdirectory( "/fred/" ); |
---|
| 116 | fredDirectory->SetGuidance( "Fred's options" ); |
---|
| 117 | |
---|
| 118 | // |
---|
| 119 | // Volume command |
---|
| 120 | // |
---|
| 121 | volumeTypeNameCmd = new G4UIcmdWithAString( "/fred/volume", this ); |
---|
| 122 | volumeTypeNameCmd->SetGuidance( "Test Volume Type" ); |
---|
| 123 | volumeTypeNameCmd->SetParameterName( "VolumeType", true ); |
---|
| 124 | |
---|
| 125 | G4String choices = volumeNames[NATALIA] + " " |
---|
| 126 | + volumeNames[VOXEL] + " " |
---|
| 127 | + volumeNames[BOX] + " " |
---|
| 128 | + volumeNames[CONE] + " " |
---|
| 129 | + volumeNames[CONE2] + " " |
---|
| 130 | + volumeNames[ORB] + " " |
---|
| 131 | + volumeNames[PARA] + " " |
---|
| 132 | + volumeNames[SPHERE] + " " |
---|
| 133 | + volumeNames[TORUS1] + " " |
---|
| 134 | + volumeNames[TORUS2] + " " |
---|
| 135 | + volumeNames[TRAP] + " " |
---|
| 136 | + volumeNames[TRD] + " " |
---|
| 137 | + volumeNames[TUBS] + " " |
---|
| 138 | + volumeNames[ELLIPS] + " " |
---|
| 139 | + volumeNames[ELCONE] + " " |
---|
| 140 | + volumeNames[ELTUBE] + " " |
---|
| 141 | + volumeNames[EXTRU1] + " " |
---|
| 142 | + volumeNames[EXTRU2] + " " |
---|
| 143 | + volumeNames[EXTRU3] + " " |
---|
| 144 | + volumeNames[EXTRU4] + " " |
---|
| 145 | + volumeNames[HYPE] + " " |
---|
| 146 | + volumeNames[PCON] + " " |
---|
| 147 | + volumeNames[PCON2] + " " |
---|
| 148 | + volumeNames[PCON3] + " " |
---|
| 149 | + volumeNames[PCON4] + " " |
---|
| 150 | + volumeNames[PGON] + " " |
---|
| 151 | + volumeNames[PGON2] + " " |
---|
| 152 | + volumeNames[PGON3] + " " |
---|
| 153 | + volumeNames[TESSEL1] + " " |
---|
| 154 | + volumeNames[TESSEL2] + " " |
---|
| 155 | + volumeNames[TESSEL3] + " " |
---|
| 156 | + volumeNames[TESSEL4] + " " |
---|
| 157 | + volumeNames[TET] + " " |
---|
| 158 | + volumeNames[TWBOX] + " " |
---|
| 159 | + volumeNames[TWTRAP1] + " " |
---|
| 160 | + volumeNames[TWTRAP2] + " " |
---|
| 161 | + volumeNames[TWTRD] + " " |
---|
| 162 | + volumeNames[TWTUBS] + " " |
---|
| 163 | + volumeNames[BOOL1]; |
---|
| 164 | volumeTypeNameCmd->SetCandidates( choices ); |
---|
| 165 | |
---|
| 166 | volumeTypeNameCmd->AvailableForStates( G4State_PreInit, G4State_Idle ); |
---|
| 167 | |
---|
| 168 | // |
---|
| 169 | // gun command |
---|
| 170 | // |
---|
| 171 | gunTypeNameCmd = new G4UIcmdWithAString( "/fred/gun", this ); |
---|
| 172 | gunTypeNameCmd->SetGuidance( "Type of particle gun to use" ); |
---|
| 173 | gunTypeNameCmd->SetParameterName( "GunType", true ); |
---|
| 174 | |
---|
| 175 | G4String choices2 = gunNames[SPRAY] + " " |
---|
| 176 | + gunNames[GRID] + " " |
---|
| 177 | + gunNames[G4]; |
---|
| 178 | gunTypeNameCmd->SetCandidates( choices2 ); |
---|
| 179 | |
---|
| 180 | // |
---|
| 181 | // draw command |
---|
| 182 | // |
---|
| 183 | drawTypeNameCmd = new G4UIcmdWithAString( "/fred/draw", this ); |
---|
| 184 | drawTypeNameCmd->SetGuidance( "Type of drawing to make" ); |
---|
| 185 | drawTypeNameCmd->SetParameterName( "DrawType", true ); |
---|
| 186 | |
---|
| 187 | G4String choices3 = drawNames[NORMAL] + " " |
---|
| 188 | + drawNames[SHADOW]; |
---|
| 189 | drawTypeNameCmd->SetCandidates( choices3 ); |
---|
| 190 | |
---|
| 191 | // |
---|
| 192 | // Start Phi Command |
---|
| 193 | // |
---|
| 194 | startPhiCmd = new G4UIcmdWithADouble( "/fred/startPhi", this ); |
---|
| 195 | startPhiCmd->SetGuidance( "Starting phi value (degrees) for test volume" ); |
---|
| 196 | startPhiCmd->SetParameterName( "StartPhi", true ); |
---|
| 197 | startPhiCmd->AvailableForStates( G4State_PreInit, G4State_Idle ); |
---|
| 198 | |
---|
| 199 | // |
---|
| 200 | // Delta Phi Command |
---|
| 201 | // |
---|
| 202 | deltaPhiCmd = new G4UIcmdWithADouble( "/fred/deltaPhi", this ); |
---|
| 203 | deltaPhiCmd->SetGuidance( "Delta phi value (degrees) for test volume" ); |
---|
| 204 | deltaPhiCmd->SetParameterName( "DeltaPhi", true ); |
---|
| 205 | deltaPhiCmd->AvailableForStates( G4State_PreInit, G4State_Idle ); |
---|
| 206 | |
---|
| 207 | // |
---|
| 208 | // Number sides command |
---|
| 209 | // |
---|
| 210 | numSideCmd = new G4UIcmdWithAnInteger( "/fred/numSide", this ); |
---|
| 211 | numSideCmd->SetGuidance( "Number phi segments for test volume" ); |
---|
| 212 | numSideCmd->SetParameterName( "NumberSide", true ); |
---|
| 213 | numSideCmd->AvailableForStates( G4State_PreInit, G4State_Idle ); |
---|
| 214 | |
---|
| 215 | // |
---|
| 216 | // Pause command |
---|
| 217 | // |
---|
| 218 | pauseCmd = new G4UIcmdWithoutParameter( "/fred/pause", this ); |
---|
| 219 | pauseCmd->SetGuidance( "Prompts for return" ); |
---|
| 220 | pauseCmd->AvailableForStates( G4State_PreInit, G4State_Idle ); |
---|
| 221 | |
---|
| 222 | // |
---|
| 223 | // Declare test3 messenger |
---|
| 224 | // |
---|
| 225 | test3Messenger = new FredTest3Messenger(); |
---|
| 226 | |
---|
| 227 | // |
---|
| 228 | // Declare voxel test messenger |
---|
| 229 | // |
---|
| 230 | voxelTestMessenger = new FredVoxelTestMessenger(); |
---|
| 231 | } |
---|
| 232 | |
---|
| 233 | // |
---|
| 234 | // Destructor |
---|
| 235 | // |
---|
| 236 | FredMessenger::~FredMessenger() |
---|
| 237 | { |
---|
| 238 | delete test3Messenger; |
---|
| 239 | delete voxelTestMessenger; |
---|
| 240 | delete volumeTypeNameCmd; |
---|
| 241 | delete fredDirectory; |
---|
| 242 | } |
---|
| 243 | |
---|
| 244 | |
---|
| 245 | // |
---|
| 246 | // SetTestVolume |
---|
| 247 | // |
---|
| 248 | // Set the target volume for test 3. We need to funnel this down |
---|
| 249 | // to our test 3 messenger |
---|
| 250 | // |
---|
| 251 | void FredMessenger::SetTestVolume( const G4VSolid *theTestVolume ) |
---|
| 252 | { |
---|
| 253 | testVolume = theTestVolume; |
---|
| 254 | test3Messenger->SetTestVolume( testVolume ); |
---|
| 255 | voxelTestMessenger->SetTestVolume( testVolume ); |
---|
| 256 | } |
---|
| 257 | |
---|
| 258 | |
---|
| 259 | // |
---|
| 260 | // SelectedVolume |
---|
| 261 | // |
---|
| 262 | // Return selected test volume type |
---|
| 263 | // |
---|
| 264 | VolumeType FredMessenger::SelectedVolume() |
---|
| 265 | { |
---|
| 266 | return(testVolumeType); |
---|
| 267 | } |
---|
| 268 | |
---|
| 269 | // |
---|
| 270 | // SelectedGun |
---|
| 271 | // |
---|
| 272 | // Return selected gun |
---|
| 273 | // |
---|
| 274 | GunType FredMessenger::SelectedGun() |
---|
| 275 | { |
---|
| 276 | return(gunType); |
---|
| 277 | } |
---|
| 278 | |
---|
| 279 | // |
---|
| 280 | // SelectedDrawing |
---|
| 281 | // |
---|
| 282 | // Return selected drawing type |
---|
| 283 | // |
---|
| 284 | DrawType FredMessenger::SelectedDrawing() |
---|
| 285 | { |
---|
| 286 | return( drawType ); |
---|
| 287 | } |
---|
| 288 | |
---|
| 289 | |
---|
| 290 | // |
---|
| 291 | // PauseInput |
---|
| 292 | // |
---|
| 293 | // This is to make up for a deficiency in the basic user-inteface: |
---|
| 294 | // Wait for input from the user |
---|
| 295 | // |
---|
| 296 | void FredMessenger::PauseInput() |
---|
| 297 | { |
---|
| 298 | G4cout << "Press <return> to continue: " << std::flush; |
---|
| 299 | char c; |
---|
| 300 | G4cin.get(c); |
---|
| 301 | } |
---|
| 302 | |
---|
| 303 | |
---|
| 304 | // |
---|
| 305 | // SetNewValue |
---|
| 306 | // |
---|
| 307 | void FredMessenger::SetNewValue( G4UIcommand *command, G4String newValues ) |
---|
| 308 | { |
---|
| 309 | if (command == volumeTypeNameCmd) { |
---|
| 310 | for( G4int vol = 0; vol < FRED_VOLUMETYPE_NUM; vol++ ) { |
---|
| 311 | if (volumeNames[vol] == newValues) { |
---|
| 312 | testVolumeType = (VolumeType)vol; |
---|
| 313 | break; |
---|
| 314 | } |
---|
| 315 | } |
---|
| 316 | } |
---|
| 317 | else if (command == gunTypeNameCmd) { |
---|
| 318 | for( G4int gun = 0; gun < FRED_GUNTYPE_NUM; gun++ ) { |
---|
| 319 | if (gunNames[gun] == newValues) { |
---|
| 320 | gunType = (GunType)gun; |
---|
| 321 | break; |
---|
| 322 | } |
---|
| 323 | } |
---|
| 324 | } |
---|
| 325 | else if (command == drawTypeNameCmd) { |
---|
| 326 | for( G4int draw = 0; draw < FRED_DRAWTYPE_NUM; draw++ ) { |
---|
| 327 | if (drawNames[draw] == newValues) { |
---|
| 328 | drawType = (DrawType)draw; |
---|
| 329 | break; |
---|
| 330 | } |
---|
| 331 | } |
---|
| 332 | } |
---|
| 333 | else if (command == startPhiCmd) { |
---|
| 334 | startPhi = startPhiCmd->GetNewDoubleValue( newValues ); |
---|
| 335 | } |
---|
| 336 | else if (command == deltaPhiCmd) { |
---|
| 337 | deltaPhi = deltaPhiCmd->GetNewDoubleValue( newValues ); |
---|
| 338 | } |
---|
| 339 | else if (command == numSideCmd) { |
---|
| 340 | numSide = numSideCmd->GetNewIntValue( newValues ); |
---|
| 341 | } |
---|
| 342 | else if (command == pauseCmd) { |
---|
| 343 | PauseInput(); |
---|
| 344 | } |
---|
| 345 | } |
---|
| 346 | |
---|
| 347 | // |
---|
| 348 | // GetCurrentValue |
---|
| 349 | // |
---|
| 350 | G4String FredMessenger::GetCurrentValue( G4UIcommand *command ) |
---|
| 351 | { |
---|
| 352 | if (command == volumeTypeNameCmd) { |
---|
| 353 | return volumeNames[testVolumeType]; |
---|
| 354 | } |
---|
| 355 | else if (command == gunTypeNameCmd) { |
---|
| 356 | return gunNames[gunType]; |
---|
| 357 | } |
---|
| 358 | else if (command == drawTypeNameCmd) { |
---|
| 359 | return drawNames[drawType]; |
---|
| 360 | } |
---|
| 361 | else if (command == startPhiCmd) { |
---|
| 362 | return startPhiCmd->ConvertToString( startPhi ); |
---|
| 363 | } |
---|
| 364 | else if (command == deltaPhiCmd) { |
---|
| 365 | return deltaPhiCmd->ConvertToString( deltaPhi ); |
---|
| 366 | } |
---|
| 367 | else if (command == numSideCmd) { |
---|
| 368 | return numSideCmd->ConvertToString( numSide ); |
---|
| 369 | } |
---|
| 370 | return "baloney"; |
---|
| 371 | } |
---|