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