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 /// \file eventgenerator/particleGun/src/Track << 27 /// \brief Implementation of the TrackingActio << 28 // 26 // 29 // << 27 // $Id: TrackingAction.cc,v 1.3 2010-07-16 07:37:48 maire Exp $ 30 // << 28 // GEANT4 tag $Name: not supported by cvs2svn $ 31 //....oooOO0OOooo........oooOO0OOooo........oo << 29 // 32 //....oooOO0OOooo........oooOO0OOooo........oo 30 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 31 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 33 32 34 #include "TrackingAction.hh" 33 #include "TrackingAction.hh" 35 << 36 #include "HistoManager.hh" << 37 #include "PrimaryGeneratorAction.hh" 34 #include "PrimaryGeneratorAction.hh" 38 #include "PrimaryGeneratorAction2.hh" 35 #include "PrimaryGeneratorAction2.hh" 39 #include "PrimaryGeneratorAction3.hh" 36 #include "PrimaryGeneratorAction3.hh" 40 #include "PrimaryGeneratorAction4.hh" 37 #include "PrimaryGeneratorAction4.hh" 41 #include "RunAction.hh" 38 #include "RunAction.hh" >> 39 #include "HistoManager.hh" 42 40 43 #include "G4PhysicalConstants.hh" << 44 #include "G4Track.hh" 41 #include "G4Track.hh" 45 42 46 //....oooOO0OOooo........oooOO0OOooo........oo 43 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 47 44 48 TrackingAction::TrackingAction(PrimaryGenerato << 45 TrackingAction::TrackingAction(PrimaryGeneratorAction* prim,HistoManager* histo) 49 { << 46 :primary(prim),histoManager(histo) 50 // parameters for generator action #3 << 47 { 51 fNewUz = fPrimary->GetAction3()->GetNewUz(); << 48 // parameters for generator action #3 >> 49 newUz = primary->GetAction3()->GetNewUz(); >> 50 >> 51 // parameters for generator action #4 >> 52 deltaR3 = >> 53 (primary->GetAction4()->GetRmax3() - primary->GetAction4()->GetRmin3())/3.; >> 54 cosAlphaMin = primary->GetAction4()->GetCosAlphaMin(); >> 55 cosAlphaMax = primary->GetAction4()->GetCosAlphaMax(); 52 } 56 } 53 57 54 //....oooOO0OOooo........oooOO0OOooo........oo 58 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 55 59 56 void TrackingAction::PreUserTrackingAction(con 60 void TrackingAction::PreUserTrackingAction(const G4Track* track) 57 { 61 { 58 G4int selectedGeneratorAction = fPrimary->Ge << 62 G4int selectedGeneratorAction = primary->GetSelectedAction(); 59 G4AnalysisManager* analysis = G4AnalysisMana << 63 G4int id = 0; 60 G4int id = 0; << 64 61 << 65 if(selectedGeneratorAction==2) 62 if (selectedGeneratorAction == 2) { << 66 { 63 // energy spectrum << 67 //energy spectrum 64 // << 68 // 65 id = 1; << 69 id = 1; 66 analysis->FillH1(id, track->GetKineticEner << 70 histoManager->FillHisto(id, track->GetKineticEnergy()); 67 } << 71 } 68 << 72 else if(selectedGeneratorAction==3) 69 else if (selectedGeneratorAction == 0) { << 73 { 70 // particle direction: cos(alpha) << 74 //momentum direction : angular distr dN/dOmega = f(alpha) 71 // << 75 // 72 id = 5; << 76 id = 2; 73 G4ThreeVector um = track->GetMomentumDirec << 77 G4ThreeVector um = track->GetMomentumDirection(); 74 G4double cosalpha = um.z(); << 78 G4double alpha = std::acos(newUz*um); 75 analysis->FillH1(id, cosalpha); << 79 G4double dalfa = histoManager->GetBinWidth(id); 76 << 80 G4double dOmega = twopi*std::sin(alpha)*dalfa; 77 // particle direction: psi << 81 if (dOmega > 0.) histoManager->FillHisto(id, alpha, 1./dOmega); 78 // << 82 79 id = 6; << 83 //momentum direction : angular distr dN/dOmega = f(psi) 80 G4double psi = std::atan2(um.y(), um.x()); << 84 // 81 if (psi < 0.) psi += twopi; << 85 id = 3; 82 analysis->FillH1(id, psi); << 86 // complete local frame 83 } << 87 G4ThreeVector u1(1.,0.,0.); u1.rotateUz(newUz); 84 else if (selectedGeneratorAction == 3) { << 88 G4ThreeVector u2(0.,1.,0.); u2.rotateUz(newUz); 85 // particle direction in local frame: cos( << 89 // 86 // << 90 G4double psi = std::atan2(u2*um, u1*um); 87 id = 5; << 91 if (psi < 0.) psi += twopi; 88 G4ThreeVector um = track->GetMomentumDirec << 92 G4double dpsi = histoManager->GetBinWidth(id); 89 G4double cosalpha = fNewUz * um; << 93 G4double alphaMax = primary->GetAction3()->GetAlphaMax(); 90 analysis->FillH1(id, cosalpha); << 94 dOmega = (1. - std::cos(alphaMax))*dpsi; 91 << 95 if (dOmega > 0.) histoManager->FillHisto(id, psi, 1./dOmega); 92 // particle direction in local frame: psi << 96 } 93 // << 97 else if(selectedGeneratorAction==4) 94 id = 6; << 98 { 95 // complete local frame << 99 G4ThreeVector vertex = track->GetVertexPosition(); 96 G4ThreeVector u1(1., 0., 0.); << 100 G4double r = vertex.mag(); 97 u1.rotateUz(fNewUz); << 101 if (r <= 0.0) return; 98 G4ThreeVector u2(0., 1., 0.); << 102 //local frame : new uz = ur 99 u2.rotateUz(fNewUz); << 103 G4ThreeVector ur = vertex/r; 100 // << 104 101 G4double psi = std::atan2(u2 * um, u1 * um << 105 //vertex position: radial distribution : dN/dv = f(r) 102 if (psi < 0.) psi += twopi; << 106 // 103 analysis->FillH1(id, psi); << 107 id = 4; 104 } << 108 G4double dr = histoManager->GetBinWidth(id); 105 << 109 G4double dv = 2*twopi*r*r*dr; 106 else if (selectedGeneratorAction == 4) { << 110 if (dv > 0.) histoManager->FillHisto(id, r, 1./dv); 107 G4ThreeVector vertex = track->GetVertexPos << 111 108 G4double r = vertex.mag(); << 112 //vertex position: angular distribution : dN/dv = f(theta) 109 if (r <= 0.0) return; << 113 // 110 // local frame : new uz = ur << 114 id = 5; 111 G4ThreeVector ur = vertex / r; << 115 G4double theta = std::acos(ur.z()); 112 << 116 G4double dteta = histoManager->GetBinWidth(id); 113 // vertex position. radial distribution dN << 117 dv = deltaR3*twopi*std::sin(theta)*dteta; 114 // << 118 if (dv > 0.) histoManager->FillHisto(id, theta, 1./dv); 115 id = 2; << 119 116 G4double dr = analysis->GetH1Width(id); << 120 //vertex position: angular distribution : dN/dv = f(phi) 117 G4double dv = 2 * twopi * r * r * dr; << 121 // 118 if (dv > 0.) analysis->FillH1(id, r, 1. / << 122 id = 6; 119 << 123 G4double phi = std::atan2(ur.y(), ur.x()); 120 // vertex position: cos(theta) << 124 if (phi < 0.) phi += twopi; 121 // << 125 G4double dphi = histoManager->GetBinWidth(id); 122 id = 3; << 126 dv = 2*deltaR3*dphi; 123 G4double costheta = ur.z(); << 127 if (dv > 0.) histoManager->FillHisto(id, phi, 1./dv); 124 analysis->FillH1(id, costheta); << 128 125 << 129 //momentum direction : angular distr dN/dOmega = f(alpha) 126 // vertex position: phi << 130 // 127 // << 131 id = 7; 128 id = 4; << 132 G4ThreeVector um = track->GetMomentumDirection(); 129 G4double phi = std::atan2(ur.y(), ur.x()); << 133 G4double alpha = std::acos(ur*um); 130 if (phi < 0.) phi += twopi; << 134 G4double dalfa = histoManager->GetBinWidth(id); 131 analysis->FillH1(id, phi); << 135 G4double dOmega = twopi*std::sin(alpha)*dalfa; 132 << 136 if (dOmega > 0.) histoManager->FillHisto(id, alpha, 1./dOmega); 133 // particle direction in local frame: cos( << 137 134 // << 138 //momentum direction : angular distr dN/dOmega = f(psi) 135 id = 5; << 139 // 136 G4ThreeVector um = track->GetMomentumDirec << 140 id = 8; 137 G4double cosalpha = ur * um; << 141 // complete local frame 138 analysis->FillH1(id, cosalpha); << 142 G4ThreeVector u1(1.,0.,0.); u1.rotateUz(ur); 139 << 143 G4ThreeVector u2(0.,1.,0.); u2.rotateUz(ur); 140 // particle direction in local frame: psi << 144 // 141 // << 145 G4double psi = std::atan2(u2*um, u1*um); 142 id = 6; << 146 if (psi < 0.) psi += twopi; 143 // complete local frame << 147 G4double dpsi = histoManager->GetBinWidth(id); 144 G4ThreeVector u1(1., 0., 0.); << 148 dOmega = (cosAlphaMin - cosAlphaMax)*dpsi; 145 u1.rotateUz(ur); << 149 if (dOmega > 0.) histoManager->FillHisto(id, psi, 1./dOmega); 146 G4ThreeVector u2(0., 1., 0.); << 150 } 147 u2.rotateUz(ur); << 148 // << 149 G4double psi = std::atan2(u2 * um, u1 * um << 150 if (psi < 0.) psi += twopi; << 151 analysis->FillH1(id, psi); << 152 } << 153 } 151 } 154 152 155 //....oooOO0OOooo........oooOO0OOooo........oo 153 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 156 154 157 void TrackingAction::PostUserTrackingAction(co << 155 void TrackingAction::PostUserTrackingAction(const G4Track*) >> 156 { } 158 157 159 //....oooOO0OOooo........oooOO0OOooo........oo 158 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 159 160 160