Geant4 Cross Reference |
1 // 1 // 2 // ******************************************* 2 // ******************************************************************** 3 // * License and Disclaimer 3 // * License and Disclaimer * 4 // * 4 // * * 5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. 9 // * include a list of copyright holders. * 10 // * 10 // * * 11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. * 17 // * 17 // * * 18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************* 24 // ******************************************************************** 25 // 25 // 26 // 26 // 27 /// \file field/field04/src/F04SteppingAction. 27 /// \file field/field04/src/F04SteppingAction.cc 28 /// \brief Implementation of the F04SteppingAc 28 /// \brief Implementation of the F04SteppingAction class 29 // 29 // 30 30 >> 31 #include "G4Track.hh" >> 32 31 #include "F04SteppingAction.hh" 33 #include "F04SteppingAction.hh" >> 34 #include "G4SteppingManager.hh" >> 35 #include "F04SteppingActionMessenger.hh" >> 36 >> 37 #include "G4ParticleTypes.hh" >> 38 #include "G4LogicalVolumeStore.hh" 32 39 33 #include "F04UserTrackInformation.hh" 40 #include "F04UserTrackInformation.hh" 34 41 35 #include "G4LogicalVolumeStore.hh" << 42 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 36 #include "G4ParticleTypes.hh" << 43 37 #include "G4SteppingManager.hh" << 44 F04SteppingAction::F04SteppingAction() 38 #include "G4Track.hh" << 45 : G4UserSteppingAction(), >> 46 fSteppingMessenger(0), >> 47 fTargetVolume(0), >> 48 fTestPlaneVolume(0) >> 49 { >> 50 fSteppingMessenger = new F04SteppingActionMessenger(); >> 51 } >> 52 >> 53 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 54 >> 55 F04SteppingAction::~F04SteppingAction() >> 56 { >> 57 delete fSteppingMessenger ; >> 58 } 39 59 40 //....oooOO0OOooo........oooOO0OOooo........oo 60 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 41 61 42 void F04SteppingAction::UserSteppingAction(con 62 void F04SteppingAction::UserSteppingAction(const G4Step* theStep) 43 { 63 { 44 G4Track* theTrack = theStep->GetTrack(); 64 G4Track* theTrack = theStep->GetTrack(); 45 << 65 46 // Get pointers to test volumes (only once) 66 // Get pointers to test volumes (only once) 47 if (!fTargetVolume) { << 67 if ( ! fTargetVolume ) { 48 fTargetVolume = G4LogicalVolumeStore::GetI << 68 fTargetVolume 49 fTestPlaneVolume = G4LogicalVolumeStore::G << 69 = G4LogicalVolumeStore::GetInstance()->GetVolume("Target"); 50 } << 70 fTestPlaneVolume 51 << 71 = G4LogicalVolumeStore::GetInstance()->GetVolume("TestPlane"); 52 if (theTrack->GetParentID() == 0) { << 72 } 53 // This is a primary track << 73 54 G4LogicalVolume* theVolume = << 74 if (theTrack->GetParentID()==0) { 55 theStep->GetPreStepPoint()->GetPhysicalV << 75 //This is a primary track >> 76 G4LogicalVolume* theVolume >> 77 = theStep->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume(); 56 if (theVolume != fTargetVolume) { 78 if (theVolume != fTargetVolume) { 57 theTrack->SetTrackStatus(fStopAndKill); << 79 theTrack->SetTrackStatus(fStopAndKill); 58 return; << 80 return; 59 } 81 } 60 } 82 } 61 83 62 // G4ParticleDefinition* particleType = the << 84 // G4ParticleDefinition* particleType = theTrack->GetDefinition(); 63 85 64 // check if it is entering the test volume 86 // check if it is entering the test volume 65 G4StepPoint* thePrePoint = theStep->GetPreSt 87 G4StepPoint* thePrePoint = theStep->GetPreStepPoint(); 66 G4VPhysicalVolume* thePrePV = thePrePoint->G 88 G4VPhysicalVolume* thePrePV = thePrePoint->GetPhysicalVolume(); 67 G4LogicalVolume* thePreLV = thePrePV->GetLog 89 G4LogicalVolume* thePreLV = thePrePV->GetLogicalVolume(); 68 90 69 G4LogicalVolume* thePostLV = nullptr; << 91 G4LogicalVolume* thePostLV = 0; 70 G4StepPoint* thePostPoint = theStep->GetPost 92 G4StepPoint* thePostPoint = theStep->GetPostStepPoint(); 71 93 72 if (thePostPoint) { 94 if (thePostPoint) { 73 G4VPhysicalVolume* thePostPV = thePostPoin << 95 G4VPhysicalVolume* thePostPV = thePostPoint->GetPhysicalVolume(); 74 if (thePostPV) thePostLV = thePostPV->GetL << 96 if (thePostPV) thePostLV = thePostPV->GetLogicalVolume(); 75 } 97 } 76 98 77 if (thePostLV == fTestPlaneVolume && thePreL << 99 if (thePostLV == fTestPlaneVolume && 78 // G4double x = theTrack->GetPosition( << 100 thePreLV != fTestPlaneVolume) { 79 // G4double y = theTrack->GetPosition( << 101 80 << 102 // G4double x = theTrack->GetPosition().x(); 81 G4ThreeVector theMomentumDirection = theTr << 103 // G4double y = theTrack->GetPosition().y(); 82 // G4double theTotalMomentum = theTrac << 104 83 // << 105 G4ThreeVector theMomentumDirection = theTrack-> 84 << 106 GetDynamicParticle()->GetMomentumDirection(); 85 // then kill the track << 107 // G4double theTotalMomentum = theTrack->GetDynamicParticle()-> 86 theTrack->SetTrackStatus(fStopAndKill); << 108 // GetTotalMomentum(); 87 return; << 109 >> 110 // then kill the track >> 111 theTrack->SetTrackStatus(fStopAndKill); >> 112 return; >> 113 88 } 114 } 89 115 90 // G4double z = theTrack->GetPosition().z() << 116 // G4double z = theTrack->GetPosition().z(); 91 117 92 auto trackInformation = (F04UserTrackInforma << 118 F04UserTrackInformation* trackInformation = >> 119 (F04UserTrackInformation*)theTrack->GetUserInformation(); 93 120 94 if (trackInformation->GetTrackStatusFlag() ! 121 if (trackInformation->GetTrackStatusFlag() != reverse) { 95 if (thePreLV != fTargetVolume) { << 122 if (thePreLV != fTargetVolume ) { 96 if (theTrack->GetMomentumDirection().z() << 123 if ( theTrack-> GetMomentumDirection().z()>0.0 && 97 && theTrack->GetVertexMomentumDirect << 124 theTrack->GetVertexMomentumDirection().z()<0.0 ) 98 { << 125 { 99 trackInformation->SetTrackStatusFlag(r << 126 trackInformation->SetTrackStatusFlag(reverse); 100 } << 127 } 101 } << 128 } 102 } 129 } 103 130 104 // check if it is alive 131 // check if it is alive 105 if (theTrack->GetTrackStatus() == fAlive) { << 132 if (theTrack->GetTrackStatus() == fAlive) { return; } 106 return; << 107 } << 108 133 109 if (thePostPoint->GetProcessDefinedStep() != << 134 if (thePostPoint->GetProcessDefinedStep() != 0) { 110 if (thePostPoint->GetProcessDefinedStep()- << 135 if (thePostPoint->GetProcessDefinedStep()->GetProcessName() != "Decay") >> 136 return; 111 } 137 } >> 138 112 } 139 } 113 140