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 // This example is provided by the Geant4-DNA collaboration 27 // Any report or published results obtained using the Geant4-DNA software 28 // shall cite the following Geant4-DNA collaboration publication: 29 // Med. Phys. 37 (2010) 4692-4708 30 // J. Comput. Phys. 274 (2014) 841-882 31 // The Geant4-DNA web site is available at http://geant4-dna.org 32 // 33 // 34 /// \file ITTrackingInteractivity.hh 35 /// \brief Implementation of the ITTrackingInteractivity class 36 37 #include "ITTrackingInteractivity.hh" 38 39 #include "G4Event.hh" 40 #include "G4EventManager.hh" 41 #include "G4IT.hh" 42 #include "G4RichTrajectory.hh" 43 #include "G4SmoothTrajectory.hh" 44 #include "G4TrackingInformation.hh" 45 #include "G4Trajectory.hh" 46 #include "G4UserSteppingAction.hh" 47 #include "G4UserTrackingAction.hh" 48 #include "G4VSteppingVerbose.hh" 49 #include "G4VTrajectory.hh" 50 #include "G4VisManager.hh" 51 52 class G4Trajectory_Lock 53 { 54 friend class ITTrackingInteractivity; 55 56 G4Trajectory_Lock() : fpTrajectory(0) { ; } 57 58 ~G4Trajectory_Lock() { ; } 59 60 G4VTrajectory* fpTrajectory; 61 }; 62 63 ITTrackingInteractivity::ITTrackingInteractivity() 64 { 65 fStoreTrajectory = 0; 66 fVerboseLevel = 0; 67 68 fpUserTrackingAction = 0; 69 fpUserSteppingAction = 0; 70 71 //////////////////////////// 72 // In case you want to use same tracking/stepping action 73 // for normal and IT stepping 74 /* 75 fpUserTrackingAction = 76 trackingManager->GetUserTrackingAction(); 77 fpUserSteppingAction = 78 G4EventManager::GetEventManager()->GetUserSteppingAction(); 79 */ 80 //////////////////////////// 81 } 82 83 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 84 85 ITTrackingInteractivity::~ITTrackingInteractivity() 86 { 87 G4EventManager* eventManager = G4EventManager::GetEventManager(); 88 89 if (eventManager) { 90 G4UserTrackingAction* std_trackAct = eventManager->GetUserTrackingAction(); 91 if (fpUserTrackingAction != std_trackAct && fpUserTrackingAction) delete fpUserTrackingAction; 92 93 G4UserSteppingAction* std_stepAct = eventManager->GetUserSteppingAction(); 94 if (fpUserSteppingAction != std_stepAct && fpUserSteppingAction) delete fpUserSteppingAction; 95 } 96 else { 97 if (fpUserSteppingAction) { 98 delete fpUserSteppingAction; 99 } 100 101 if (fpUserTrackingAction) { 102 delete fpUserTrackingAction; 103 } 104 } 105 } 106 107 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 108 109 void ITTrackingInteractivity::Initialize() 110 { 111 G4TrackingManager* trackingManager = G4EventManager::GetEventManager()->GetTrackingManager(); 112 fStoreTrajectory = trackingManager->GetStoreTrajectory(); 113 fVerboseLevel = trackingManager->GetVerboseLevel(); 114 } 115 116 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 117 118 void ITTrackingInteractivity::StartTracking(G4Track* track) 119 { 120 #ifdef G4VERBOSE 121 if (fVerboseLevel) { 122 TrackBanner(track, "G4ITTrackingManager::StartTracking : "); 123 } 124 #endif 125 126 if (fVerboseLevel > 0 && (G4VSteppingVerbose::GetSilent() != 1)) TrackBanner(track); 127 128 // Pre tracking user intervention process. 129 if (fpUserTrackingAction != 0) { 130 fpUserTrackingAction->PreUserTrackingAction(track); 131 } 132 // #ifdef G4_STORE_TRAJECTORY 133 G4TrackingInformation* trackingInfo = GetIT(track)->GetTrackingInfo(); 134 G4Trajectory_Lock* trajectory_lock = trackingInfo->GetTrajectory_Lock(); 135 136 // Construct a trajectory if it is requested 137 if (fStoreTrajectory && (!trajectory_lock)) { 138 trajectory_lock = new G4Trajectory_Lock(); 139 trackingInfo->SetTrajectory_Lock(trajectory_lock); 140 G4VTrajectory* trajectory = 0; 141 // default trajectory concrete class object 142 switch (fStoreTrajectory) { 143 default: 144 case 1: 145 trajectory = new G4Trajectory(track); 146 break; 147 case 2: 148 trajectory = new G4SmoothTrajectory(track); 149 break; 150 case 3: 151 trajectory = new G4RichTrajectory(track); 152 break; 153 } 154 trajectory_lock->fpTrajectory = trajectory; 155 } 156 // #endif 157 } 158 159 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 160 161 void ITTrackingInteractivity::AppendStep(G4Track* track, G4Step* step) 162 { 163 //////////////////////////// 164 // If you want to use sensitive detector 165 /* 166 G4VPhysicalVolume* currentVolume = 167 step->GetPreStepPoint()->GetPhysicalVolume(); 168 G4SteppingControl stepControlFlag = step->GetControlFlag(); 169 170 if( currentVolume != 0 && stepControlFlag != AvoidHitInvocation) 171 { 172 G4VSensitiveDetector* sensitive = step->GetPreStepPoint()-> 173 GetSensitiveDetector(); 174 if( sensitive != 0 ) 175 { 176 sensitive->Hit(fpStep); 177 } 178 } 179 */ 180 //////////////////////////// 181 182 if (fpUserSteppingAction) fpUserSteppingAction->UserSteppingAction(step); 183 184 //////////////////////////// 185 // If you want to use regional stepping action 186 /* 187 G4UserSteppingAction* regionalAction 188 = fpStep->GetPreStepPoint()->GetPhysicalVolume()-> 189 GetLogicalVolume()->GetRegion()-> 190 GetRegionalSteppingAction(); 191 if( regionalAction ) regionalAction->UserSteppingAction(fpStep); 192 */ 193 //////////////////////////// 194 195 if (fStoreTrajectory) { 196 G4TrackingInformation* trackingInfo = GetIT(track)->GetTrackingInfo(); 197 G4Trajectory_Lock* trajectory_lock = trackingInfo->GetTrajectory_Lock(); 198 trajectory_lock->fpTrajectory->AppendStep(step); 199 } 200 } 201 202 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 203 204 void ITTrackingInteractivity::EndTracking(G4Track* track) 205 { 206 #ifdef G4VERBOSE 207 if (fVerboseLevel) { 208 TrackBanner(track, "G4ITTrackingManager::EndTracking : "); 209 } 210 #endif 211 // Post tracking user intervention process. 212 if (fpUserTrackingAction != 0) { 213 fpUserTrackingAction->PostUserTrackingAction(track); 214 } 215 216 // #ifdef G4_STORE_TRAJECTORY 217 G4TrackingInformation* trackingInfo = GetIT(track)->GetTrackingInfo(); 218 G4Trajectory_Lock* trajectory_lock = trackingInfo->GetTrajectory_Lock(); 219 220 if (trajectory_lock) { 221 G4VTrajectory*& trajectory = trajectory_lock->fpTrajectory; 222 223 if (fStoreTrajectory && trajectory) { 224 #ifdef G4VERBOSE 225 if (fVerboseLevel > 10) trajectory->ShowTrajectory(); 226 #endif 227 G4TrackStatus istop = track->GetTrackStatus(); 228 229 if (trajectory && (istop != fStopButAlive) && (istop != fSuspend)) { 230 G4Event* currentEvent = G4EventManager::GetEventManager()->GetNonconstCurrentEvent(); 231 232 if (currentEvent) { 233 G4TrajectoryContainer* trajectoryContainer = currentEvent->GetTrajectoryContainer(); 234 235 if (!trajectoryContainer) { 236 trajectoryContainer = new G4TrajectoryContainer; 237 currentEvent->SetTrajectoryContainer(trajectoryContainer); 238 } 239 trajectoryContainer->insert(trajectory); 240 } 241 else { 242 fTrajectories.push_back(trajectory); 243 } 244 } 245 } 246 // Destruct the trajectory if it was created 247 else if ((!fStoreTrajectory) && trajectory) { 248 delete trajectory; 249 trajectory = 0; 250 } 251 delete trajectory_lock; 252 trackingInfo->SetTrajectory_Lock(0); 253 } 254 // #endif 255 } 256 257 void ITTrackingInteractivity::Finalize() 258 { 259 for (std::vector<G4VTrajectory*>::iterator it = fTrajectories.begin(); it != fTrajectories.end(); 260 it++) 261 { 262 G4VisManager::GetConcreteInstance()->Draw(**it); 263 } 264 } 265