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 // $Id: F04SteppingAction.cc 73014 2013-08-15 08:54:47Z gcosmo $ 26 // 27 // 27 /// \file field/field04/src/F04SteppingAction. 28 /// \file field/field04/src/F04SteppingAction.cc 28 /// \brief Implementation of the F04SteppingAc 29 /// \brief Implementation of the F04SteppingAction class 29 // 30 // 30 31 >> 32 #include "G4Track.hh" >> 33 31 #include "F04SteppingAction.hh" 34 #include "F04SteppingAction.hh" >> 35 #include "G4SteppingManager.hh" >> 36 #include "F04SteppingActionMessenger.hh" >> 37 >> 38 #include "G4ParticleTypes.hh" >> 39 #include "G4LogicalVolumeStore.hh" 32 40 33 #include "F04UserTrackInformation.hh" 41 #include "F04UserTrackInformation.hh" 34 42 35 #include "G4LogicalVolumeStore.hh" << 43 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 36 #include "G4ParticleTypes.hh" << 44 37 #include "G4SteppingManager.hh" << 45 F04SteppingAction::F04SteppingAction() 38 #include "G4Track.hh" << 46 : G4UserSteppingAction(), >> 47 fSteppingMessenger(0), >> 48 fTargetVolume(0), >> 49 fTestPlaneVolume(0) >> 50 { >> 51 fSteppingMessenger = new F04SteppingActionMessenger(); >> 52 } >> 53 >> 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 55 >> 56 F04SteppingAction::~F04SteppingAction() >> 57 { >> 58 delete fSteppingMessenger ; >> 59 } 39 60 40 //....oooOO0OOooo........oooOO0OOooo........oo 61 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 41 62 42 void F04SteppingAction::UserSteppingAction(con 63 void F04SteppingAction::UserSteppingAction(const G4Step* theStep) 43 { 64 { 44 G4Track* theTrack = theStep->GetTrack(); 65 G4Track* theTrack = theStep->GetTrack(); 45 << 66 46 // Get pointers to test volumes (only once) 67 // Get pointers to test volumes (only once) 47 if (!fTargetVolume) { << 68 if ( ! fTargetVolume ) { 48 fTargetVolume = G4LogicalVolumeStore::GetI << 69 fTargetVolume 49 fTestPlaneVolume = G4LogicalVolumeStore::G << 70 = G4LogicalVolumeStore::GetInstance()->GetVolume("Target"); 50 } << 71 fTestPlaneVolume 51 << 72 = G4LogicalVolumeStore::GetInstance()->GetVolume("TestPlane"); 52 if (theTrack->GetParentID() == 0) { << 73 } 53 // This is a primary track << 74 54 G4LogicalVolume* theVolume = << 75 if (theTrack->GetParentID()==0) { 55 theStep->GetPreStepPoint()->GetPhysicalV << 76 //This is a primary track >> 77 G4LogicalVolume* theVolume >> 78 = theStep->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume(); 56 if (theVolume != fTargetVolume) { 79 if (theVolume != fTargetVolume) { 57 theTrack->SetTrackStatus(fStopAndKill); << 80 theTrack->SetTrackStatus(fStopAndKill); 58 return; << 81 return; 59 } 82 } 60 } 83 } 61 84 62 // G4ParticleDefinition* particleType = the << 85 // G4ParticleDefinition* particleType = theTrack->GetDefinition(); 63 86 64 // check if it is entering the test volume 87 // check if it is entering the test volume 65 G4StepPoint* thePrePoint = theStep->GetPreSt 88 G4StepPoint* thePrePoint = theStep->GetPreStepPoint(); 66 G4VPhysicalVolume* thePrePV = thePrePoint->G 89 G4VPhysicalVolume* thePrePV = thePrePoint->GetPhysicalVolume(); 67 G4LogicalVolume* thePreLV = thePrePV->GetLog 90 G4LogicalVolume* thePreLV = thePrePV->GetLogicalVolume(); 68 91 69 G4LogicalVolume* thePostLV = nullptr; << 92 G4LogicalVolume* thePostLV = 0; 70 G4StepPoint* thePostPoint = theStep->GetPost 93 G4StepPoint* thePostPoint = theStep->GetPostStepPoint(); 71 94 72 if (thePostPoint) { 95 if (thePostPoint) { 73 G4VPhysicalVolume* thePostPV = thePostPoin << 96 G4VPhysicalVolume* thePostPV = thePostPoint->GetPhysicalVolume(); 74 if (thePostPV) thePostLV = thePostPV->GetL << 97 if (thePostPV) thePostLV = thePostPV->GetLogicalVolume(); 75 } 98 } 76 99 77 if (thePostLV == fTestPlaneVolume && thePreL << 100 if (thePostLV == fTestPlaneVolume && 78 // G4double x = theTrack->GetPosition( << 101 thePreLV != fTestPlaneVolume) { 79 // G4double y = theTrack->GetPosition( << 102 80 << 103 // G4double x = theTrack->GetPosition().x(); 81 G4ThreeVector theMomentumDirection = theTr << 104 // G4double y = theTrack->GetPosition().y(); 82 // G4double theTotalMomentum = theTrac << 105 83 // << 106 G4ThreeVector theMomentumDirection = theTrack-> 84 << 107 GetDynamicParticle()->GetMomentumDirection(); 85 // then kill the track << 108 // G4double theTotalMomentum = theTrack->GetDynamicParticle()-> 86 theTrack->SetTrackStatus(fStopAndKill); << 109 // GetTotalMomentum(); 87 return; << 110 >> 111 // then kill the track >> 112 theTrack->SetTrackStatus(fStopAndKill); >> 113 return; >> 114 88 } 115 } 89 116 90 // G4double z = theTrack->GetPosition().z() << 117 // G4double z = theTrack->GetPosition().z(); 91 118 92 auto trackInformation = (F04UserTrackInforma << 119 F04UserTrackInformation* trackInformation = >> 120 (F04UserTrackInformation*)theTrack->GetUserInformation(); 93 121 94 if (trackInformation->GetTrackStatusFlag() ! 122 if (trackInformation->GetTrackStatusFlag() != reverse) { 95 if (thePreLV != fTargetVolume) { << 123 if (thePreLV != fTargetVolume ) { 96 if (theTrack->GetMomentumDirection().z() << 124 if ( theTrack-> GetMomentumDirection().z()>0.0 && 97 && theTrack->GetVertexMomentumDirect << 125 theTrack->GetVertexMomentumDirection().z()<0.0 ) 98 { << 126 { 99 trackInformation->SetTrackStatusFlag(r << 127 trackInformation->SetTrackStatusFlag(reverse); 100 } << 128 } 101 } << 129 } 102 } 130 } 103 131 104 // check if it is alive 132 // check if it is alive 105 if (theTrack->GetTrackStatus() == fAlive) { << 133 if (theTrack->GetTrackStatus() == fAlive) { return; } 106 return; << 107 } << 108 134 109 if (thePostPoint->GetProcessDefinedStep() != << 135 if (thePostPoint->GetProcessDefinedStep() != 0) { 110 if (thePostPoint->GetProcessDefinedStep()- << 136 if (thePostPoint->GetProcessDefinedStep()->GetProcessName() != "Decay") >> 137 return; 111 } 138 } >> 139 112 } 140 } 113 141