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 runAndEvent/RE01/src/RE01EventAction 26 /// \file runAndEvent/RE01/src/RE01EventAction.cc 27 /// \brief Implementation of the RE01EventActi 27 /// \brief Implementation of the RE01EventAction class 28 // 28 // 29 // 29 // 30 30 31 #include "RE01EventAction.hh" 31 #include "RE01EventAction.hh" 32 << 33 #include "RE01CalorimeterHit.hh" << 34 #include "RE01TrackerHit.hh" 32 #include "RE01TrackerHit.hh" 35 #include "RE01Trajectory.hh" << 33 #include "RE01CalorimeterHit.hh" 36 34 37 #include "G4Event.hh" 35 #include "G4Event.hh" 38 #include "G4EventManager.hh" 36 #include "G4EventManager.hh" 39 #include "G4HCofThisEvent.hh" 37 #include "G4HCofThisEvent.hh" 40 #include "G4PrimaryParticle.hh" << 41 #include "G4PrimaryVertex.hh" << 42 #include "G4SDManager.hh" << 43 #include "G4SystemOfUnits.hh" << 44 #include "G4TrajectoryContainer.hh" << 45 #include "G4UImanager.hh" << 46 #include "G4VHitsCollection.hh" 38 #include "G4VHitsCollection.hh" >> 39 #include "G4TrajectoryContainer.hh" 47 #include "G4VVisManager.hh" 40 #include "G4VVisManager.hh" >> 41 #include "G4SDManager.hh" >> 42 #include "G4UImanager.hh" 48 #include "G4ios.hh" 43 #include "G4ios.hh" >> 44 #include "RE01Trajectory.hh" >> 45 #include "G4PrimaryVertex.hh" >> 46 #include "G4PrimaryParticle.hh" >> 47 #include "G4SystemOfUnits.hh" 49 48 50 //....oooOO0OOooo........oooOO0OOooo........oo 49 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 51 RE01EventAction::RE01EventAction() : G4UserEve << 50 RE01EventAction::RE01EventAction() 52 { << 51 :G4UserEventAction(), 53 ; << 52 fTrackerCollID(-1),fCalorimeterCollID(-1) 54 } << 53 {;} 55 54 56 //....oooOO0OOooo........oooOO0OOooo........oo 55 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 57 RE01EventAction::~RE01EventAction() 56 RE01EventAction::~RE01EventAction() 58 { << 57 {;} 59 ; << 60 } << 61 58 62 //....oooOO0OOooo........oooOO0OOooo........oo 59 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 63 void RE01EventAction::BeginOfEventAction(const 60 void RE01EventAction::BeginOfEventAction(const G4Event*) 64 { 61 { 65 G4SDManager* SDman = G4SDManager::GetSDMpoin << 62 G4SDManager * SDman = G4SDManager::GetSDMpointer(); 66 if (fTrackerCollID < 0 || fCalorimeterCollID << 63 if(fTrackerCollID<0||fCalorimeterCollID<0) >> 64 { 67 G4String colNam; 65 G4String colNam; 68 fTrackerCollID = SDman->GetCollectionID(co << 66 fTrackerCollID = SDman->GetCollectionID(colNam="trackerCollection"); 69 fCalorimeterCollID = SDman->GetCollectionI << 67 fCalorimeterCollID = SDman->GetCollectionID(colNam="calCollection"); 70 } 68 } 71 } 69 } 72 70 73 //....oooOO0OOooo........oooOO0OOooo........oo 71 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 74 void RE01EventAction::EndOfEventAction(const G 72 void RE01EventAction::EndOfEventAction(const G4Event* evt) 75 { 73 { 76 G4cout << ">>> Summary of Event " << evt->Ge 74 G4cout << ">>> Summary of Event " << evt->GetEventID() << G4endl; 77 if (evt->GetNumberOfPrimaryVertex() == 0) { << 75 if(evt->GetNumberOfPrimaryVertex()==0) >> 76 { 78 G4cout << "Event is empty." << G4endl; 77 G4cout << "Event is empty." << G4endl; 79 return; 78 return; 80 } 79 } >> 80 >> 81 if(fTrackerCollID<0||fCalorimeterCollID<0) return; 81 82 82 if (fTrackerCollID < 0 || fCalorimeterCollID << 83 G4HCofThisEvent * HCE = evt->GetHCofThisEvent(); 83 << 84 G4HCofThisEvent* HCE = evt->GetHCofThisEvent << 85 RE01TrackerHitsCollection* THC = 0; 84 RE01TrackerHitsCollection* THC = 0; 86 RE01CalorimeterHitsCollection* CHC = 0; 85 RE01CalorimeterHitsCollection* CHC = 0; 87 if (HCE) { << 86 if(HCE) >> 87 { 88 THC = (RE01TrackerHitsCollection*)(HCE->Ge 88 THC = (RE01TrackerHitsCollection*)(HCE->GetHC(fTrackerCollID)); 89 CHC = (RE01CalorimeterHitsCollection*)(HCE 89 CHC = (RE01CalorimeterHitsCollection*)(HCE->GetHC(fCalorimeterCollID)); 90 } 90 } 91 91 92 if (THC) { << 92 if(THC) >> 93 { 93 int n_hit = THC->entries(); 94 int n_hit = THC->entries(); 94 G4cout << G4endl; 95 G4cout << G4endl; 95 G4cout << "Tracker hits " << 96 G4cout << "Tracker hits " << 96 << "------------------------------- << 97 "--------------------------------------------------------------" 97 G4cout << n_hit << " hits are stored in RE << 98 << G4endl; 98 if (fpEventManager->GetVerboseLevel() > 0) << 99 G4cout << n_hit << " hits are stored in RE01TrackerHitsCollection." >> 100 << G4endl; >> 101 if(fpEventManager->GetVerboseLevel()>0) >> 102 { 99 G4cout << "List of hits in tracker" << G 103 G4cout << "List of hits in tracker" << G4endl; 100 for (int i = 0; i < n_hit; i++) { << 104 for(int i=0;i<n_hit;i++) 101 (*THC)[i]->Print(); << 105 { (*THC)[i]->Print(); } 102 } << 103 } 106 } 104 } 107 } 105 if (CHC) { << 108 if(CHC) >> 109 { 106 int n_hit = CHC->entries(); 110 int n_hit = CHC->entries(); 107 G4cout << G4endl; 111 G4cout << G4endl; 108 G4cout << "Calorimeter hits " << 112 G4cout << "Calorimeter hits "<< 109 << "------------------------------- << 113 "--------------------------------------------------------------" 110 G4cout << n_hit << " hits are stored in RE << 114 << G4endl; >> 115 G4cout << n_hit << " hits are stored in RE01CalorimeterHitsCollection." >> 116 << G4endl; 111 G4double totE = 0; 117 G4double totE = 0; 112 for (int i = 0; i < n_hit; i++) { << 118 for(int i=0;i<n_hit;i++) 113 totE += (*CHC)[i]->GetEdep(); << 119 { totE += (*CHC)[i]->GetEdep(); } 114 } << 120 G4cout << " Total energy deposition in calorimeter : " 115 G4cout << " Total energy deposition in << 121 << totE / GeV << " (GeV)" << G4endl; 116 } 122 } 117 123 118 // get number of stored trajectories 124 // get number of stored trajectories 119 G4TrajectoryContainer* trajectoryContainer = 125 G4TrajectoryContainer* trajectoryContainer = evt->GetTrajectoryContainer(); 120 G4int n_trajectories = 0; 126 G4int n_trajectories = 0; 121 if (trajectoryContainer) n_trajectories = tr 127 if (trajectoryContainer) n_trajectories = trajectoryContainer->entries(); 122 // extract the trajectories and print them o 128 // extract the trajectories and print them out 123 G4cout << G4endl; 129 G4cout << G4endl; 124 G4cout << "Trajectories in tracker " << 130 G4cout << "Trajectories in tracker "<< 125 << "--------------------------------- << 131 "--------------------------------------------------------------" 126 if (fpEventManager->GetVerboseLevel() > 0) { << 132 << G4endl; 127 for (G4int i = 0; i < n_trajectories; i++) << 133 if(fpEventManager->GetVerboseLevel()>0) 128 RE01Trajectory* trj = (RE01Trajectory*)( << 134 { >> 135 for(G4int i=0; i<n_trajectories; i++) >> 136 { >> 137 RE01Trajectory* trj = >> 138 (RE01Trajectory*)((*(evt->GetTrajectoryContainer()))[i]); 129 trj->ShowTrajectory(); 139 trj->ShowTrajectory(); 130 } 140 } 131 } 141 } 132 << 142 133 G4cout << G4endl; 143 G4cout << G4endl; 134 G4cout << "Primary particles " << 144 G4cout << "Primary particles "<< 135 << "--------------------------------- << 145 "--------------------------------------------------------------" >> 146 << G4endl; 136 G4int n_vertex = evt->GetNumberOfPrimaryVert 147 G4int n_vertex = evt->GetNumberOfPrimaryVertex(); 137 for (G4int iv = 0; iv < n_vertex; iv++) { << 148 for(G4int iv=0;iv<n_vertex;iv++) >> 149 { 138 G4PrimaryVertex* pv = evt->GetPrimaryVerte 150 G4PrimaryVertex* pv = evt->GetPrimaryVertex(iv); 139 G4cout << G4endl; 151 G4cout << G4endl; 140 G4cout << "Primary vertex " << G4ThreeVect << 152 G4cout << "Primary vertex " 141 << " at t = " << (pv->GetT0()) / << 153 << G4ThreeVector(pv->GetX0(),pv->GetY0(),pv->GetZ0()) 142 if (fpEventManager->GetVerboseLevel() > 0) << 154 << " at t = " << (pv->GetT0())/ns << " [ns]" << G4endl; >> 155 if(fpEventManager->GetVerboseLevel()>0) >> 156 { 143 G4PrimaryParticle* pp = pv->GetPrimary() 157 G4PrimaryParticle* pp = pv->GetPrimary(); 144 while (pp) { << 158 while(pp) 145 PrintPrimary(pp, 0); << 159 { >> 160 PrintPrimary(pp,0); 146 pp = pp->GetNext(); 161 pp = pp->GetNext(); 147 } 162 } 148 } 163 } 149 } 164 } 150 } 165 } 151 166 152 //....oooOO0OOooo........oooOO0OOooo........oo 167 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 153 void RE01EventAction::PrintPrimary(G4PrimaryPa << 168 void RE01EventAction::PrintPrimary(G4PrimaryParticle* pp,G4int ind) 154 { 169 { 155 for (G4int ii = 0; ii <= ind; ii++) { << 170 for(G4int ii=0;ii<=ind;ii++) 156 G4cout << " "; << 171 { G4cout << " "; } 157 } << 158 G4cout << "==PDGcode " << pp->GetPDGcode() < 172 G4cout << "==PDGcode " << pp->GetPDGcode() << " "; 159 if (pp->GetG4code() != 0) { << 173 if(pp->GetG4code()!=0) 160 G4cout << "(" << pp->GetG4code()->GetParti << 174 { G4cout << "(" << pp->GetG4code()->GetParticleName() << ")"; } 161 } << 175 else 162 else { << 176 { G4cout << "is not defined in G4"; } 163 G4cout << "is not defined in G4"; << 177 G4cout << " " << pp->GetMomentum()/GeV << " [GeV] "; 164 } << 178 if(pp->GetTrackID()<0) 165 G4cout << " " << pp->GetMomentum() / GeV << << 179 { G4cout << G4endl; } 166 if (pp->GetTrackID() < 0) { << 180 else 167 G4cout << G4endl; << 181 { G4cout << ">>> G4Track ID " << pp->GetTrackID() << G4endl; } 168 } << 169 else { << 170 G4cout << ">>> G4Track ID " << pp->GetTrac << 171 } << 172 182 173 G4PrimaryParticle* daughter = pp->GetDaughte 183 G4PrimaryParticle* daughter = pp->GetDaughter(); 174 while (daughter) { << 184 while(daughter) 175 PrintPrimary(daughter, ind + 1); << 185 { >> 186 PrintPrimary(daughter,ind+1); 176 daughter = daughter->GetNext(); 187 daughter = daughter->GetNext(); 177 } 188 } 178 } 189 } >> 190 >> 191 179 192