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