| 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 | }
|
|---|