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 electromagnetic/TestEm5/src/Tracking 26 /// \file electromagnetic/TestEm5/src/TrackingAction.cc 27 /// \brief Implementation of the TrackingActio 27 /// \brief Implementation of the TrackingAction class 28 // 28 // 29 // 29 // >> 30 // $Id$ 30 // 31 // 31 //....oooOO0OOooo........oooOO0OOooo........oo 32 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 32 //....oooOO0OOooo........oooOO0OOooo........oo 33 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 33 34 34 #include "TrackingAction.hh" 35 #include "TrackingAction.hh" 35 36 36 #include "DetectorConstruction.hh" 37 #include "DetectorConstruction.hh" >> 38 #include "RunAction.hh" 37 #include "EventAction.hh" 39 #include "EventAction.hh" 38 #include "HistoManager.hh" 40 #include "HistoManager.hh" 39 #include "Run.hh" << 40 41 >> 42 #include "G4Track.hh" 41 #include "G4PhysicalConstants.hh" 43 #include "G4PhysicalConstants.hh" 42 #include "G4RunManager.hh" << 43 #include "G4SystemOfUnits.hh" 44 #include "G4SystemOfUnits.hh" 44 #include "G4Track.hh" << 45 45 46 //....oooOO0OOooo........oooOO0OOooo........oo 46 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 47 47 48 TrackingAction::TrackingAction(DetectorConstru << 48 TrackingAction::TrackingAction(DetectorConstruction* DET,RunAction* RA, 49 : fDetector(det), fEventAction(event) << 49 EventAction* EA) 50 {} << 50 :fDetector(DET), fRunAction(RA), fEventAction(EA) 51 << 51 { } >> 52 52 //....oooOO0OOooo........oooOO0OOooo........oo 53 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 53 54 54 void TrackingAction::PreUserTrackingAction(con << 55 void TrackingAction::PreUserTrackingAction(const G4Track* aTrack ) 55 { 56 { 56 // few initialisations 57 // few initialisations 57 // 58 // 58 if (aTrack->GetTrackID() == 1) { 59 if (aTrack->GetTrackID() == 1) { 59 fXstartAbs = fDetector->GetxstartAbs(); 60 fXstartAbs = fDetector->GetxstartAbs(); 60 fXendAbs = fDetector->GetxendAbs(); << 61 fXendAbs = fDetector->GetxendAbs(); 61 fPrimaryCharge = aTrack->GetDefinition()-> 62 fPrimaryCharge = aTrack->GetDefinition()->GetPDGCharge(); 62 fDirX = aTrack->GetMomentumDirection().x() << 63 } 63 } 64 } 64 } 65 65 66 //....oooOO0OOooo........oooOO0OOooo........oo 66 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 67 67 68 void TrackingAction::PostUserTrackingAction(co 68 void TrackingAction::PostUserTrackingAction(const G4Track* aTrack) 69 { 69 { 70 G4AnalysisManager* analysisManager = G4Analy 70 G4AnalysisManager* analysisManager = G4AnalysisManager::Instance(); 71 71 72 Run* run = static_cast<Run*>(G4RunManager::G << 73 << 74 G4ThreeVector position = aTrack->GetPosition 72 G4ThreeVector position = aTrack->GetPosition(); 75 G4ThreeVector vertex = aTrack->GetVertexPosi << 73 G4ThreeVector vertex = aTrack->GetVertexPosition(); 76 G4double charge = aTrack->GetDefinition()->G << 74 G4double charge = aTrack->GetDefinition()->GetPDGCharge(); 77 G4double energy = aTrack->GetKineticEnergy() << 75 78 << 76 G4bool transmit = ((position.x() >= fXendAbs) && (vertex.x() < fXendAbs)); 79 G4bool transmit = << 77 G4bool reflect = (position.x() <= fXstartAbs); 80 (((fDirX >= 0.0 && position.x() >= fXendAb << 81 && energy > 0.0); << 82 G4bool reflect = << 83 (((fDirX >= 0.0 && position.x() <= fXstart << 84 && energy > 0.0); << 85 G4bool notabsor = (transmit || reflect); 78 G4bool notabsor = (transmit || reflect); 86 79 87 // transmitted + reflected particles counter << 80 //transmitted + reflected particles counter 88 // 81 // 89 G4int flag = 0; 82 G4int flag = 0; 90 if (charge == fPrimaryCharge) flag = 1; << 83 if (charge == fPrimaryCharge) flag = 1; 91 if (aTrack->GetTrackID() == 1) flag = 2; 84 if (aTrack->GetTrackID() == 1) flag = 2; 92 if (transmit) fEventAction->SetTransmitFlag( 85 if (transmit) fEventAction->SetTransmitFlag(flag); 93 if (reflect) fEventAction->SetReflectFlag(fl << 86 if (reflect) fEventAction->SetReflectFlag(flag); 94 << 87 95 // 88 // 96 // histograms << 89 //histograms 97 // 90 // 98 G4bool charged = (charge != 0.); << 91 G4bool charged = (charge != 0.); 99 G4bool neutral = !charged; 92 G4bool neutral = !charged; 100 93 101 // energy spectrum at exit << 94 //energy spectrum at exit 102 // zero energy charged particle are not take << 103 // 95 // 104 G4int id = 0; << 96 G4int id = 0; 105 if (transmit && charged) << 97 if (transmit && charged) id = 10; 106 id = 10; << 98 else if (transmit && neutral) id = 20; 107 else if (transmit && neutral) << 99 else if (reflect && charged) id = 30; 108 id = 20; << 100 else if (reflect && neutral) id = 40; 109 else if (reflect && charged) << 101 110 id = 30; << 102 if (id>0) 111 else if (reflect && neutral) << 103 analysisManager->FillH1(id, aTrack->GetKineticEnergy()); 112 id = 40; << 104 113 << 105 //energy leakage 114 if (id > 0) { << 115 analysisManager->FillH1(id, energy); << 116 } << 117 << 118 // energy leakage << 119 // 106 // 120 if (notabsor) { 107 if (notabsor) { 121 G4int trackID = aTrack->GetTrackID(); 108 G4int trackID = aTrack->GetTrackID(); 122 G4int index = 0; << 109 G4int index = 0; if (trackID > 1) index = 1; //primary=0, secondaries=1 123 if (trackID > 1) index = 1; // primary=0, << 124 G4double eleak = aTrack->GetKineticEnergy( 110 G4double eleak = aTrack->GetKineticEnergy(); 125 if ((aTrack->GetDefinition() == G4Positron 111 if ((aTrack->GetDefinition() == G4Positron::Positron()) && (trackID > 1)) 126 eleak += 2 * electron_mass_c2; << 112 eleak += 2*electron_mass_c2; 127 run->AddEnergyLeak(eleak, index); << 113 fRunAction->AddEnergyLeak(eleak,index); 128 } 114 } 129 115 130 // space angle distribution at exit : dN/dOm << 116 //space angle distribution at exit : dN/dOmega 131 // 117 // 132 G4ThreeVector direction = aTrack->GetMomentu << 118 G4ThreeVector direction = aTrack->GetMomentumDirection(); 133 id = 0; << 119 id = 0; 134 if (transmit && charged) << 120 if (transmit && charged) id = 12; 135 id = 12; << 121 else if (transmit && neutral) id = 22; 136 else if (transmit && neutral) << 122 else if (reflect && charged) id = 32; 137 id = 22; << 123 else if (reflect && neutral) id = 42; 138 else if (reflect && charged) << 139 id = 32; << 140 else if (reflect && neutral) << 141 id = 42; << 142 124 143 if (id > 0 && std::abs(direction.x()) < 1.0) << 125 if (id>0) { 144 G4double theta = std::acos(direction.x()); << 126 G4double theta = std::acos(direction.x()); 145 if (theta > 0.0) { 127 if (theta > 0.0) { 146 G4double dteta = analysisManager->GetH1W << 128 G4double dteta = analysisManager->GetH1Width(id); 147 G4double unit = analysisManager->GetH1Un << 129 G4double unit = analysisManager->GetH1Unit(id); 148 G4double weight = unit / (twopi * std::s << 130 G4double weight = (unit*unit)/(twopi*std::sin(theta)*dteta); 149 /* 131 /* 150 G4cout << "theta, dteta, unit, weight: " << 132 G4cout << "theta, dteta, unit, weight: " 151 << theta << " " << 133 << theta << " " 152 << dteta << " " << 134 << dteta << " " 153 << unit << " " << 135 << unit << " " 154 << weight << G4endl; << 136 << weight << G4endl; 155 */ 137 */ 156 analysisManager->FillH1(id, theta, weigh << 138 analysisManager->FillH1(id,theta,weight); 157 } 139 } 158 } 140 } 159 << 141 160 // energy fluence at exit : dE(MeV)/dOmega << 142 //energy fluence at exit : dE(MeV)/dOmega 161 // 143 // 162 id = 0; << 144 id = 0; 163 if (transmit && charged) << 145 if (transmit && charged) id = 11; 164 id = 11; << 146 else if (transmit && neutral) id = 21; 165 else if (reflect && charged) << 147 else if (reflect && charged) id = 31; 166 id = 31; << 148 else if (reflect && neutral) id = 41; 167 else if (transmit && neutral) << 168 id = 21; << 169 else if (reflect && neutral) << 170 id = 41; << 171 149 172 if (id > 0 && std::abs(direction.x()) < 1.0) << 150 if (id>0) { 173 G4double theta = std::acos(direction.x()); << 151 G4double theta = std::acos(direction.x()); 174 if (theta > 0.0) { 152 if (theta > 0.0) { 175 G4double dteta = analysisManager->GetH1W << 153 G4double dteta = analysisManager->GetH1Width(id); 176 G4double unit = analysisManager->GetH1Un << 154 G4double unit = analysisManager->GetH1Unit(id); 177 G4double weight = unit / (twopi * std::s << 155 G4double weight = (unit*unit)/(twopi*std::sin(theta)*dteta); 178 weight *= (aTrack->GetKineticEnergy() / << 156 weight *= (aTrack->GetKineticEnergy()/MeV); 179 analysisManager->FillH1(id, theta, weigh << 157 analysisManager->FillH1(id,theta,weight); 180 } << 158 } 181 } << 159 } >> 160 >> 161 //projected angles distribution at exit >> 162 // >> 163 id = 0; >> 164 if (transmit && charged) id = 13; >> 165 else if (transmit && neutral) id = 23; >> 166 else if (reflect && charged) id = 33; >> 167 else if (reflect && neutral) id = 43; 182 168 183 // projected angles distribution at exit << 169 if (id>0) { 184 // << 185 id = 0; << 186 if (transmit && charged) << 187 id = 13; << 188 else if (transmit && neutral) << 189 id = 23; << 190 else if (reflect && charged) << 191 id = 33; << 192 else if (reflect && neutral) << 193 id = 43; << 194 << 195 if (id > 0) { << 196 if (direction.x() != 0.0) { 170 if (direction.x() != 0.0) { 197 G4double tet = std::atan(direction.y() / << 171 G4double tet = std::atan(direction.y()/std::fabs(direction.x())); 198 analysisManager->FillH1(id, tet); << 172 analysisManager->FillH1(id,tet); 199 if (transmit && (flag == 2)) run->AddMsc << 173 if (transmit && (flag == 2)) fRunAction->AddMscProjTheta(tet); 200 << 174 201 tet = std::atan(direction.z() / std::fab << 175 tet = std::atan(direction.z()/std::fabs(direction.x())); 202 analysisManager->FillH1(id, tet); << 176 analysisManager->FillH1(id,tet); 203 if (transmit && (flag == 2)) run->AddMsc << 177 if (transmit && (flag == 2)) fRunAction->AddMscProjTheta(tet); 204 } 178 } 205 } 179 } 206 180 207 // projected position and radius at exit << 181 //projected position and radius at exit 208 // 182 // 209 if (transmit && energy > 0.0) { << 183 id = 0; >> 184 if (transmit && charged) id = 14; >> 185 >> 186 if (id>0) { 210 G4double y = position.y(), z = position.z( 187 G4double y = position.y(), z = position.z(); 211 G4double r = std::sqrt(y * y + z * z); << 188 G4double r = std::sqrt(y*y + z*z); 212 analysisManager->FillH1(14, y); << 189 analysisManager->FillH1(id, y); 213 analysisManager->FillH1(14, z); << 190 analysisManager->FillH1(id, z); 214 analysisManager->FillH1(15, r); << 191 analysisManager->FillH1(id+1, r); 215 } 192 } 216 << 193 217 // x-vertex of charged secondaries << 194 //x-vertex of charged secondaries 218 // 195 // 219 if ((aTrack->GetParentID() == 1) && charged) 196 if ((aTrack->GetParentID() == 1) && charged) { 220 G4double xVertex = (aTrack->GetVertexPosit 197 G4double xVertex = (aTrack->GetVertexPosition()).x(); 221 analysisManager->FillH1(6, xVertex); 198 analysisManager->FillH1(6, xVertex); 222 if (notabsor) analysisManager->FillH1(7, x << 199 if (notabsor) analysisManager->FillH1(7, xVertex); 223 } 200 } 224 } 201 } 225 202 226 //....oooOO0OOooo........oooOO0OOooo........oo 203 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 204 227 205