| Geant4 Cross Reference |
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 // G4TrackingMessenger class implementation
27 //
28 // Contact:
29 // Questions and comments to this code should be sent to
30 // Katsuya Amako (e-mail: Katsuya.Amako@kek.jp)
31 // Makoto Asai (e-mail: asai@slac.stanford.edu)
32 // Takashi Sasaki (e-mail: Takashi.Sasaki@kek.jp)
33 // --------------------------------------------------------------------
34
35 #include "G4TrackingMessenger.hh"
36
37 #include "G4IdentityTrajectoryFilter.hh"
38 #include "G4PropagatorInField.hh"
39 #include "G4SteppingManager.hh"
40 #include "G4TrackStatus.hh"
41 #include "G4TrackingManager.hh"
42 #include "G4TransportationManager.hh"
43 #include "G4UIcmdWithAnInteger.hh"
44 #include "G4UIcmdWithoutParameter.hh"
45 #include "G4UIdirectory.hh"
46 #include "G4UImanager.hh"
47 #include "G4Threading.hh"
48 #include "G4ClonedTrajectory.hh"
49 #include "G4ClonedTrajectoryPoint.hh"
50 #include "G4ClonedRichTrajectory.hh"
51 #include "G4ClonedRichTrajectoryPoint.hh"
52 #include "G4ClonedSmoothTrajectory.hh"
53 #include "G4ClonedSmoothTrajectoryPoint.hh"
54 #include "G4ios.hh"
55 #include "globals.hh"
56
57 ///////////////////////////////////////////////////////////////////
58 G4TrackingMessenger::G4TrackingMessenger(G4TrackingManager* trMan)
59 ///////////////////////////////////////////////////////////////////
60 : trackingManager(trMan)
61 {
62 steppingManager = trackingManager->GetSteppingManager();
63
64 TrackingDirectory = new G4UIdirectory("/tracking/");
65 TrackingDirectory->SetGuidance("TrackingManager and SteppingManager control commands.");
66
67 AbortCmd = new G4UIcmdWithoutParameter("/tracking/abort", this);
68 AbortCmd->SetGuidance("Abort current G4Track processing.");
69
70 ResumeCmd = new G4UIcmdWithoutParameter("/tracking/resume", this);
71 ResumeCmd->SetGuidance("Resume current G4Track processing.");
72
73 StoreTrajectoryCmd = new G4UIcmdWithAnInteger("/tracking/storeTrajectory", this);
74 StoreTrajectoryCmd->SetGuidance("Store trajectories or not.");
75 StoreTrajectoryCmd->SetGuidance(" 0 : Don't Store trajectories.");
76 StoreTrajectoryCmd->SetGuidance(" !=0 : Store trajectories.");
77 StoreTrajectoryCmd->SetGuidance(" 1 : Choose G4Trajectory as default.");
78 StoreTrajectoryCmd->SetGuidance(" 2 : Choose G4SmoothTrajectory as default.");
79 StoreTrajectoryCmd->SetGuidance(" 3 : Choose G4RichTrajectory as default.");
80 StoreTrajectoryCmd->SetGuidance(" 4 : Choose G4RichTrajectory with auxiliary points as default.");
81 StoreTrajectoryCmd->SetParameterName("Store", true);
82 StoreTrajectoryCmd->SetDefaultValue(0);
83 StoreTrajectoryCmd->SetRange("Store >=0 && Store <= 4");
84
85 VerboseCmd = new G4UIcmdWithAnInteger("/tracking/verbose", this);
86 #ifdef G4VERBOSE
87 VerboseCmd->SetGuidance("Set Verbose level of tracking category.");
88 VerboseCmd->SetGuidance(" -1 : Silent.");
89 VerboseCmd->SetGuidance(" 0 : Silent.");
90 VerboseCmd->SetGuidance(" 1 : Minimum information of each Step.");
91 VerboseCmd->SetGuidance(" 2 : Addition to Level=1, info of secondary particles.");
92 VerboseCmd->SetGuidance(" 3 : Addition to Level=1, pre/postStepoint information");
93 VerboseCmd->SetGuidance(" after all AlongStep/PostStep process executions.");
94 VerboseCmd->SetGuidance(" 4 : Addition to Level=3, pre/postStepoint information");
95 VerboseCmd->SetGuidance(" at each AlongStepPostStep process execution.");
96 VerboseCmd->SetGuidance(" 5 : Addition to Level=4, proposed Step length information");
97 VerboseCmd->SetGuidance(" from each AlongStepPostStep process.");
98 VerboseCmd->SetParameterName("verbose_level", true);
99 VerboseCmd->SetDefaultValue(0);
100 VerboseCmd->SetRange("verbose_level >=-1 ");
101 #else
102 VerboseCmd->SetGuidance("You need to recompile the tracking category defining G4VERBOSE ");
103 #endif
104 }
105
106 ////////////////////////////////////////////
107 G4TrackingMessenger::~G4TrackingMessenger()
108 ////////////////////////////////////////////
109 {
110 delete TrackingDirectory;
111 delete AbortCmd;
112 delete ResumeCmd;
113 delete StoreTrajectoryCmd;
114 delete VerboseCmd;
115 delete auxiliaryPointsFilter;
116 }
117
118 ///////////////////////////////////////////////////////////////////////////////
119 void G4TrackingMessenger::SetNewValue(G4UIcommand* command, G4String newValues)
120 ///////////////////////////////////////////////////////////////////////////////
121 {
122 if (command == VerboseCmd) {
123 trackingManager->SetVerboseLevel(VerboseCmd->ConvertToInt(newValues));
124 }
125
126 if (command == AbortCmd) {
127 steppingManager->GetTrack()->SetTrackStatus(fStopAndKill);
128 G4UImanager::GetUIpointer()->ApplyCommand("/control/exit");
129 }
130
131 if (command == ResumeCmd) {
132 G4UImanager::GetUIpointer()->ApplyCommand("/control/exit");
133 }
134
135 if (command == StoreTrajectoryCmd) {
136 G4int trajType = StoreTrajectoryCmd->ConvertToInt(newValues);
137 if (trajType == 2 || trajType == 4) {
138
139 if (nullptr == auxiliaryPointsFilter) {
140 auxiliaryPointsFilter = new G4IdentityTrajectoryFilter;
141 }
142 G4TransportationManager::GetTransportationManager()
143 ->GetPropagatorInField()
144 ->SetTrajectoryFilter(auxiliaryPointsFilter);
145 }
146 else {
147 G4TransportationManager::GetTransportationManager()
148 ->GetPropagatorInField()
149 ->SetTrajectoryFilter(nullptr);
150 }
151 trackingManager->SetStoreTrajectory(trajType);
152
153 // Make sure cloning works for sub-event parallel mode
154 if(G4Threading::IsMasterThread() && trajType>0) {
155 static G4bool traj_1 = false, traj_2 = false, traj_3 = false;
156 G4VTrajectory* traj = nullptr;
157 G4VTrajectoryPoint* trajp = nullptr;
158 switch (trajType) {
159 case 1:
160 if(!traj_1) {
161 traj = new G4ClonedTrajectory();
162 trajp = new G4ClonedTrajectoryPoint();
163 traj_1 = true;
164 }
165 break;
166 case 2:
167 if(!traj_2) {
168 traj = new G4ClonedSmoothTrajectory();
169 trajp = new G4ClonedSmoothTrajectoryPoint();
170 traj_2 = true;
171 }
172 break;
173 case 3:
174 case 4:
175 if(!traj_3) {
176 traj = new G4ClonedRichTrajectory();
177 trajp = new G4ClonedRichTrajectoryPoint();
178 traj_3 = true;
179 }
180 break;
181 default:
182 break;
183 }
184 if(traj!=nullptr) delete traj;
185 if(trajp!=nullptr) delete trajp;
186 }
187 }
188
189 }
190
191 ////////////////////////////////////////////////////////////////////
192 G4String G4TrackingMessenger::GetCurrentValue(G4UIcommand* command)
193 ////////////////////////////////////////////////////////////////////
194 {
195 if (command == VerboseCmd) {
196 return VerboseCmd->ConvertToString(trackingManager->GetVerboseLevel());
197 }
198 if (command == StoreTrajectoryCmd) {
199 return StoreTrajectoryCmd->ConvertToString(trackingManager->GetStoreTrajectory());
200 }
201 return G4String(1, '\0');
202 }
203