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 SAXSSteppingAction.cc 27 /// \brief Definition of the SAXSSteppingAction class 28 // 29 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 30 31 #ifdef G4MULTITHREADED 32 # include "G4MTRunManager.hh" 33 #else 34 # include "G4RunManager.hh" 35 #endif 36 37 #include "SAXSDetectorConstruction.hh" 38 #include "SAXSEventAction.hh" 39 #include "SAXSSteppingAction.hh" 40 41 #include "G4AnalysisManager.hh" 42 #include "G4Event.hh" 43 #include "G4EventManager.hh" 44 #include "G4HCofThisEvent.hh" 45 #include "G4OpticalPhoton.hh" 46 #include "G4SDManager.hh" 47 #include "G4Step.hh" 48 #include "G4SystemOfUnits.hh" 49 #include "G4Track.hh" 50 #include "G4Trajectory.hh" 51 #include "G4TrajectoryContainer.hh" 52 #include "G4UImanager.hh" 53 #include "G4VHitsCollection.hh" 54 #include "G4VVisManager.hh" 55 #include "G4ios.hh" 56 57 #include <cmath> 58 59 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 60 61 SAXSSteppingAction::SAXSSteppingAction(SAXSEventAction* eventAction) 62 : G4UserSteppingAction(), fEventAction(eventAction), fPhantom(0), fEventNumber(-1), fNSe(0) 63 {} 64 65 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 66 67 SAXSSteppingAction::~SAXSSteppingAction() {} 68 69 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 70 71 void SAXSSteppingAction::UserSteppingAction(const G4Step* step) 72 { 73 // DetectorConstruction instance 74 const SAXSDetectorConstruction* detectorConstruction = 75 static_cast<const SAXSDetectorConstruction*>( 76 G4RunManager::GetRunManager()->GetUserDetectorConstruction()); 77 78 // get Phantom volume 79 fPhantom = detectorConstruction->GetPhantom(); 80 81 // get pre and post step points 82 G4StepPoint* preStepPoint = step->GetPreStepPoint(); 83 G4StepPoint* postStepPoint = step->GetPostStepPoint(); 84 85 // get the volume of the current step 86 G4LogicalVolume* volume = preStepPoint->GetTouchableHandle()->GetVolume()->GetLogicalVolume(); 87 88 // get track and particle name 89 G4Track* track = step->GetTrack(); 90 G4int ID = track->GetTrackID(); 91 G4String partName = track->GetDefinition()->GetParticleName(); 92 93 // update the primary particle (event) weight 94 G4double PrimaryWeight = 1.; 95 if (partName == "gamma" && ID == 1) { 96 PrimaryWeight = track->GetWeight(); 97 fEventAction->UpdateEventWeight(PrimaryWeight); 98 } 99 100 // if the particle is not a primary photon stepping in the phantom, exit! 101 if (volume != fPhantom || partName != "gamma" || ID != 1) return; 102 103 // if it is a new event, reset variable values to 0 104 G4int eventNumber = G4RunManager::GetRunManager()->GetCurrentEvent()->GetEventID(); 105 if (eventNumber != fEventNumber) { 106 fEventNumber = eventNumber; 107 fNSe = 0; 108 } 109 110 // identify the process that occurred 111 G4String Process; 112 if (postStepPoint->GetProcessDefinedStep() != NULL) { 113 Process = postStepPoint->GetProcessDefinedStep()->GetProcessName(); 114 } 115 else { 116 Process = "UserLimit"; 117 } 118 // G4cout << "Process: " << Process << G4endl; 119 G4int ProcIndex = -1; 120 if (Process == "Transportation" || Process == "CoupledTransportation" || Process == "UserLimit") { 121 ProcIndex = 0; 122 } 123 if (Process == "Rayl" || Process == "biasWrapper(Rayl)") { 124 ProcIndex = 1; 125 fNSe++; 126 fEventAction->AddNRi(); 127 } 128 if (Process == "compt" || Process == "biasWrapper(compt)") { 129 ProcIndex = 2; 130 fNSe++; 131 fEventAction->AddNCi(); 132 } 133 if (Process == "phot" || Process == "biasWrapper(phot)") ProcIndex = 3; 134 if (Process == "conv" || Process == "biasWrapper(conv)") ProcIndex = 4; 135 if (Process == "photonNuclear" || Process == "biasWrapper(photonNuclear)") { 136 ProcIndex = 5; 137 } 138 if (Process == "PowderDiffraction" || Process == "biasWrapper(PowderDiffraction)") { 139 ProcIndex = 6; 140 fNSe++; 141 fEventAction->AddNDi(); 142 } 143 // G4cout << "ProcIndex: " << ProcIndex << G4endl; 144 145 // calculate the scattering angle 146 G4ThreeVector mom1 = preStepPoint->GetMomentumDirection(); 147 G4ThreeVector mom2 = postStepPoint->GetMomentumDirection(); 148 G4double theta = mom1.angle(mom2); 149 150 // fill the Scattering ntuple 151 G4AnalysisManager* analysisManager = G4AnalysisManager::Instance(); 152 if (theta > 1e-6) { // record only if the angle is not 0 153 analysisManager->FillNtupleIColumn(1, 0, ProcIndex); 154 analysisManager->FillNtupleDColumn(1, 1, preStepPoint->GetKineticEnergy() / CLHEP::keV); 155 analysisManager->FillNtupleDColumn(1, 2, theta / CLHEP::deg); 156 analysisManager->FillNtupleDColumn(1, 3, track->GetWeight()); 157 analysisManager->AddNtupleRow(1); 158 } 159 } 160 161 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 162