source: trunk/source/geometry/magneticfield/test/NTST/src/NTSTGunMessenger.cc @ 1202

Last change on this file since 1202 was 1199, checked in by garnier, 15 years ago

nvx fichiers dans CVS
File size: 9.1 KB
Line 
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// $Id: NTSTGunMessenger.cc,v 1.4 2006/06/29 18:26:18 gunter Exp $
27// GEANT4 tag $Name: HEAD $
28//
29
30#include "NTSTGunMessenger.hh"
31#include "NTSTGunGenerator.hh"
32#include "G4Geantino.hh"
33#include "G4ThreeVector.hh"
34#include "G4ParticleTable.hh"
35#include "G4UIdirectory.hh"
36#include "G4UIcmdWithoutParameter.hh"
37#include "G4UIcmdWithAString.hh"
38#include "G4UIcmdWithADouble.hh"
39#include "G4UIcmdWithADoubleAndUnit.hh"
40#include "G4UIcmdWith3Vector.hh"
41#include "G4UIcmdWith3VectorAndUnit.hh"
42#include "G4UIcmdWithAnInteger.hh"
43#include "G4ios.hh"
44
45NTSTGunMessenger::NTSTGunMessenger(NTSTGunGenerator * NTSTGun)
46  : fNTSTGun(NTSTGun), gunDirectory(0), listCmd(0), particleCmd(0),
47    plowCmd(0), phighCmd(0), t0Cmd(0), polCmd(0),  numberCmd(0),
48    meanVertexCmd(0), rmsVertexCmd(0), 
49    coslowCmd(0), coshighCmd(0), philowCmd(0), phihighCmd(0)
50{
51  particleTable = G4ParticleTable::GetParticleTable();
52
53  // directory
54  gunDirectory = new G4UIdirectory("/gen/gun/");
55  gunDirectory->SetGuidance("NTST Particle Gun control commands.");
56
57  // list command
58  listCmd = new G4UIcmdWithoutParameter("/gen/gun/List",this);
59  listCmd->SetGuidance("List available particles.");
60  listCmd->SetGuidance(" Invokes G4ParticleTable.");
61
62  // particle command
63  particleCmd = new G4UIcmdWithAString("/gen/gun/particle",this);
64  particleCmd->SetGuidance("Set particle to be generated.");
65  particleCmd->SetGuidance(" (geantino is default)");
66  particleCmd->SetParameterName("particleName",true);
67  particleCmd->SetDefaultValue("geantino");
68  G4String candidateList; 
69  G4int nPtcl = particleTable->entries();
70  for(G4int i=0;i<nPtcl;i++)
71  {
72    candidateList += particleTable->GetParticleName(i);
73    candidateList += " ";
74  }
75  particleCmd->SetCandidates(candidateList);
76
77  // meanVertex command
78
79  meanVertexCmd = new G4UIcmdWith3VectorAndUnit("/gen/gun/vertex", this);
80  meanVertexCmd->SetGuidance("Set vertex position.");
81  meanVertexCmd->SetParameterName("x","y","z", true, true);
82  meanVertexCmd->SetDefaultUnit("mm");
83  meanVertexCmd->SetDefaultValue(G4ThreeVector(0,0,0));
84
85  // rmsVertex command
86
87  rmsVertexCmd = new G4UIcmdWith3VectorAndUnit("/gen/gun/spotsize", this);
88  rmsVertexCmd->SetGuidance("Set vertex size.");
89  rmsVertexCmd->SetParameterName("sx","sy","sz", true, true);
90  rmsVertexCmd->SetDefaultUnit("mm");
91  rmsVertexCmd->SetDefaultValue(G4ThreeVector(0,0,0));
92
93  // plow command
94  plowCmd = new G4UIcmdWithADoubleAndUnit("/gen/gun/plow",this);
95  plowCmd->SetGuidance("Set low momentum limit.");
96  plowCmd->SetParameterName("plow",true,true);
97  plowCmd->SetDefaultValue(1.0);
98  plowCmd->SetDefaultUnit("GeV");
99
100  // phigh command
101  phighCmd = new G4UIcmdWithADoubleAndUnit("/gen/gun/phigh",this);
102  phighCmd->SetGuidance("Set high momentum limit.");
103  phighCmd->SetParameterName("phigh",true,true);
104  phighCmd->SetDefaultValue(1.0);
105  phighCmd->SetDefaultUnit("GeV");
106
107  // coslow command
108  coslowCmd=new G4UIcmdWithADouble("/gen/gun/coslow",this);
109  coslowCmd->SetGuidance("Set lower cosine limit.");
110  coslowCmd->SetDefaultValue(-1.0);
111  coslowCmd->SetParameterName("coslow",true,true);
112
113  // coshigh command
114  coshighCmd=new G4UIcmdWithADouble("/gen/gun/coshigh",this);
115  coshighCmd->SetGuidance("Set upper cosine limit.");
116  coshighCmd->SetDefaultValue(1.0);
117  coshighCmd->SetParameterName("coshigh",true,true);
118
119  // philow command
120  philowCmd = new G4UIcmdWithADoubleAndUnit("/gen/gun/philow",this);
121  philowCmd->SetGuidance("Set low phi limit.");
122  philowCmd->SetParameterName("philow",true,true);
123  philowCmd->SetDefaultValue(0.0);
124  philowCmd->SetDefaultUnit("degree");
125
126  // phihigh command
127  phihighCmd = new G4UIcmdWithADoubleAndUnit("/gen/gun/phihigh",this);
128  phihighCmd->SetGuidance("Set high phi limit.");
129  phihighCmd->SetParameterName("phihigh",true,true);
130  phihighCmd->SetDefaultValue(360.0);
131  phihighCmd->SetDefaultUnit("degree");
132
133  // t0 command
134  t0Cmd = new G4UIcmdWithADoubleAndUnit("/gen/gun/t0",this);
135  t0Cmd->SetGuidance("Set initial t0 of the particle.");
136  t0Cmd->SetParameterName("t0",true,true);
137  t0Cmd->SetDefaultValue(0.0);
138  t0Cmd->SetDefaultUnit("ns");
139 
140  // polarization command
141  polCmd = new G4UIcmdWith3Vector("/gen/gun/polarization",this);
142  polCmd->SetGuidance("Set polarization.");
143  polCmd->SetParameterName("Px","Py","Pz",true,true); 
144  polCmd->SetRange("Px>=-1.&&Px<=1.&&Py>=-1.&&Py<=1.&&Pz>=-1.&&Pz<=1.");
145  polCmd->SetDefaultValue(G4ThreeVector(0,0,0));
146
147  // number command
148  numberCmd = new G4UIcmdWithAnInteger("/gen/gun/number",this);
149  numberCmd->SetGuidance("Set number of particles to be generated.");
150  numberCmd->SetParameterName("N",true,true);
151  numberCmd->SetRange("N>0");
152 
153  // set initial value to NTSTGunGenerator
154  fNTSTGun->SetParticleDefinition( G4Geantino::Geantino() );
155}
156
157NTSTGunMessenger::~NTSTGunMessenger()
158{
159  delete listCmd;
160  delete particleCmd;
161  delete meanVertexCmd;
162  delete rmsVertexCmd;
163  delete plowCmd;
164  delete phighCmd;
165  delete coslowCmd;
166  delete coshighCmd;
167  delete philowCmd;
168  delete phihighCmd;
169  delete t0Cmd;
170  delete polCmd;
171  delete numberCmd;
172  delete gunDirectory;
173}
174
175void NTSTGunMessenger::SetNewValue(G4UIcommand * command,G4String newValues)
176{
177  if( command==listCmd )
178  { particleTable->DumpTable(); }
179  else if( command==particleCmd )
180  {
181    G4ParticleDefinition* pd = particleTable->FindParticle(newValues);
182    if(pd != NULL)
183    { fNTSTGun->SetParticleDefinition( pd ); }
184  }
185  else if( command==meanVertexCmd )
186  { fNTSTGun->SetMeanVertex(meanVertexCmd->GetNew3VectorValue(newValues)); }
187  else if( command==rmsVertexCmd )
188  { fNTSTGun->SetRmsVertex(rmsVertexCmd->GetNew3VectorValue(newValues)); }
189  else if( command==plowCmd )
190  { fNTSTGun->SetPlow(plowCmd->GetNewDoubleValue(newValues)); }
191  else if( command==phighCmd )
192  { fNTSTGun->SetPhigh(phighCmd->GetNewDoubleValue(newValues)); }
193  else if( command==coslowCmd )
194  { fNTSTGun->SetCoslow(coslowCmd->GetNewDoubleValue(newValues)); }
195  else if( command==coshighCmd )
196  { fNTSTGun->SetCoshigh(coshighCmd->GetNewDoubleValue(newValues)); }
197  else if( command==philowCmd )
198  { fNTSTGun->SetPhilow(philowCmd->GetNewDoubleValue(newValues)); }
199  else if( command==phihighCmd )
200  { fNTSTGun->SetPhihigh(phihighCmd->GetNewDoubleValue(newValues)); }
201  else if( command==t0Cmd )
202  { fNTSTGun->SetT0(t0Cmd->GetNewDoubleValue(newValues)); }
203  else if( command==polCmd )
204  { fNTSTGun->SetPolarization(polCmd->GetNew3VectorValue(newValues)); }
205  else if( command==numberCmd )
206  { fNTSTGun->SetNumberOfParticles(numberCmd->GetNewIntValue(newValues)); }
207}
208
209G4String NTSTGunMessenger::GetCurrentValue(G4UIcommand * command)
210{
211  G4String cv;
212 
213  if( command==particleCmd )
214    { cv = fNTSTGun->GetParticleDefinition()->GetParticleName(); }
215  else if( command==plowCmd )
216    { cv = plowCmd->ConvertToString(fNTSTGun->GetPlow(),"GeV"); }
217  else if( command==plowCmd )
218    { cv = phighCmd->ConvertToString(fNTSTGun->GetPhigh(),"GeV"); }
219  else if( command==coslowCmd )
220    { cv = coslowCmd->ConvertToString(fNTSTGun->GetCoslow());}
221  else if( command==coshighCmd )
222    { cv = coshighCmd->ConvertToString(fNTSTGun->GetCoshigh()); }
223  else if( command==philowCmd )
224    { cv = philowCmd->ConvertToString(fNTSTGun->GetPhilow(),"degree"); }
225  else if( command==plowCmd )
226    { cv = phihighCmd->ConvertToString(fNTSTGun->GetPhihigh(),"degree"); }
227  else if( command==t0Cmd )
228    { cv = t0Cmd->ConvertToString(fNTSTGun->GetT0(),"ns"); }
229  else if( command==polCmd )
230    { cv = polCmd->ConvertToString(fNTSTGun->GetPolarization()); }
231  else if( command==numberCmd )
232    { cv = numberCmd->ConvertToString(fNTSTGun->GetNumberOfParticles()); }
233 
234  return cv;
235}
236
237
238
239
Note: See TracBrowser for help on using the repository browser.