| 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 //
27 /// \file field/field04/src/F04SteppingAction.cc
28 /// \brief Implementation of the F04SteppingAction class
29 //
30
31 #include "F04SteppingAction.hh"
32
33 #include "F04UserTrackInformation.hh"
34
35 #include "G4LogicalVolumeStore.hh"
36 #include "G4ParticleTypes.hh"
37 #include "G4SteppingManager.hh"
38 #include "G4Track.hh"
39
40 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
41
42 void F04SteppingAction::UserSteppingAction(const G4Step* theStep)
43 {
44 G4Track* theTrack = theStep->GetTrack();
45
46 // Get pointers to test volumes (only once)
47 if (!fTargetVolume) {
48 fTargetVolume = G4LogicalVolumeStore::GetInstance()->GetVolume("Target");
49 fTestPlaneVolume = G4LogicalVolumeStore::GetInstance()->GetVolume("TestPlane");
50 }
51
52 if (theTrack->GetParentID() == 0) {
53 // This is a primary track
54 G4LogicalVolume* theVolume =
55 theStep->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume();
56 if (theVolume != fTargetVolume) {
57 theTrack->SetTrackStatus(fStopAndKill);
58 return;
59 }
60 }
61
62 // G4ParticleDefinition* particleType = theTrack->GetDefinition();
63
64 // check if it is entering the test volume
65 G4StepPoint* thePrePoint = theStep->GetPreStepPoint();
66 G4VPhysicalVolume* thePrePV = thePrePoint->GetPhysicalVolume();
67 G4LogicalVolume* thePreLV = thePrePV->GetLogicalVolume();
68
69 G4LogicalVolume* thePostLV = nullptr;
70 G4StepPoint* thePostPoint = theStep->GetPostStepPoint();
71
72 if (thePostPoint) {
73 G4VPhysicalVolume* thePostPV = thePostPoint->GetPhysicalVolume();
74 if (thePostPV) thePostLV = thePostPV->GetLogicalVolume();
75 }
76
77 if (thePostLV == fTestPlaneVolume && thePreLV != fTestPlaneVolume) {
78 // G4double x = theTrack->GetPosition().x();
79 // G4double y = theTrack->GetPosition().y();
80
81 G4ThreeVector theMomentumDirection = theTrack->GetDynamicParticle()->GetMomentumDirection();
82 // G4double theTotalMomentum = theTrack->GetDynamicParticle()->
83 // GetTotalMomentum();
84
85 // then kill the track
86 theTrack->SetTrackStatus(fStopAndKill);
87 return;
88 }
89
90 // G4double z = theTrack->GetPosition().z();
91
92 auto trackInformation = (F04UserTrackInformation*)theTrack->GetUserInformation();
93
94 if (trackInformation->GetTrackStatusFlag() != reverse) {
95 if (thePreLV != fTargetVolume) {
96 if (theTrack->GetMomentumDirection().z() > 0.0
97 && theTrack->GetVertexMomentumDirection().z() < 0.0)
98 {
99 trackInformation->SetTrackStatusFlag(reverse);
100 }
101 }
102 }
103
104 // check if it is alive
105 if (theTrack->GetTrackStatus() == fAlive) {
106 return;
107 }
108
109 if (thePostPoint->GetProcessDefinedStep() != nullptr) {
110 if (thePostPoint->GetProcessDefinedStep()->GetProcessName() != "Decay") return;
111 }
112 }
113