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 /// \file TrackingAction.hh 27 /// \brief Implementation of the TrackingAction class. 28 29 #include "TrackingAction.hh" 30 31 #include "UserTrackInformation.hh" 32 33 #include "G4SystemOfUnits.hh" 34 #include "G4Track.hh" 35 #include "G4TrackingManager.hh" 36 #include "G4UnitsTable.hh" 37 38 TrackingAction::TrackingAction() : G4UserTrackingAction() {} 39 40 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 41 42 TrackingAction::~TrackingAction() {} 43 44 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 45 46 void TrackingAction::PreUserTrackingAction(const G4Track* aTrack) 47 { 48 // All primaries have a flag/label equal to 1 49 if (aTrack->GetParentID() == 0) { 50 UserTrackInformation* parentInformation = new UserTrackInformation(); 51 parentInformation->SetSplitTrackID(1); 52 G4Track* newTrack = (G4Track*)aTrack; 53 newTrack->SetUserInformation(parentInformation); 54 } 55 } 56 57 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 58 59 void TrackingAction::PostUserTrackingAction(const G4Track* aTrack) 60 { 61 UserTrackInformation* parentInformation = (UserTrackInformation*)(aTrack->GetUserInformation()); 62 // Pass the flag to all seconaries 63 G4TrackVector* secondaries = fpTrackingManager->GimmeSecondaries(); 64 if (secondaries) { 65 size_t nbSec = secondaries->size(); 66 for (size_t iSec = 0; iSec < nbSec; iSec++) { 67 if ((*secondaries)[iSec]->GetUserInformation() == 0) { 68 // Retrieve the flag 69 UserTrackInformation* secondaryInformation = new UserTrackInformation(); 70 (*secondaries)[iSec]->SetUserInformation(secondaryInformation); 71 72 G4int splitTrackID = parentInformation->GetSplitTrackID(); 73 // Pass the flag 74 secondaryInformation->SetSplitTrackID(splitTrackID); 75 } 76 } 77 } 78 } 79 80 void TrackingAction::Initialize() 81 { 82 return; 83 } 84