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/HepMC/HepMCEx01/src/E << 27 /// \brief Implementation of the ExN04EventAct << 28 // << 29 // << 30 26 31 #include "ExN04EventAction.hh" 27 #include "ExN04EventAction.hh" 32 28 >> 29 #include "ExN04TrackerHit.hh" 33 #include "ExN04CalorimeterHit.hh" 30 #include "ExN04CalorimeterHit.hh" 34 #include "ExN04MuonHit.hh" 31 #include "ExN04MuonHit.hh" 35 #include "ExN04TrackerHit.hh" << 36 32 37 #include "G4Event.hh" 33 #include "G4Event.hh" 38 #include "G4EventManager.hh" 34 #include "G4EventManager.hh" 39 #include "G4HCofThisEvent.hh" 35 #include "G4HCofThisEvent.hh" >> 36 #include "G4VHitsCollection.hh" >> 37 #include "G4TrajectoryContainer.hh" >> 38 #include "G4Trajectory.hh" >> 39 #include "G4VVisManager.hh" 40 #include "G4SDManager.hh" 40 #include "G4SDManager.hh" 41 #include "G4SystemOfUnits.hh" << 42 #include "G4UImanager.hh" 41 #include "G4UImanager.hh" 43 #include "G4VHitsCollection.hh" << 44 #include "G4ios.hh" 42 #include "G4ios.hh" 45 43 46 //....oooOO0OOooo........oooOO0OOooo........oo << 47 ExN04EventAction::ExN04EventAction() 44 ExN04EventAction::ExN04EventAction() 48 : G4UserEventAction(), ftrackerCollID(-1), f << 45 { 49 {} << 46 trackerCollID = -1; >> 47 calorimeterCollID = -1; >> 48 muonCollID = -1; >> 49 } 50 50 51 //....oooOO0OOooo........oooOO0OOooo........oo << 51 ExN04EventAction::~ExN04EventAction() 52 ExN04EventAction::~ExN04EventAction() {} << 52 {;} 53 53 54 //....oooOO0OOooo........oooOO0OOooo........oo << 55 void ExN04EventAction::BeginOfEventAction(cons 54 void ExN04EventAction::BeginOfEventAction(const G4Event*) 56 { 55 { 57 G4SDManager* SDman = G4SDManager::GetSDMpoin << 56 G4SDManager * SDman = G4SDManager::GetSDMpointer(); 58 if (ftrackerCollID < 0 || fcalorimeterCollID << 57 if(trackerCollID<0||calorimeterCollID<0||muonCollID<0) >> 58 { 59 G4String colNam; 59 G4String colNam; 60 ftrackerCollID = SDman->GetCollectionID(co << 60 trackerCollID = SDman->GetCollectionID(colNam="trackerCollection"); 61 fcalorimeterCollID = SDman->GetCollectionI << 61 calorimeterCollID = SDman->GetCollectionID(colNam="calCollection"); 62 fmuonCollID = SDman->GetCollectionID(colNa << 62 muonCollID = SDman->GetCollectionID(colNam="muonCollection"); 63 } 63 } 64 } 64 } 65 65 66 //....oooOO0OOooo........oooOO0OOooo........oo << 67 void ExN04EventAction::EndOfEventAction(const 66 void ExN04EventAction::EndOfEventAction(const G4Event* evt) 68 { 67 { 69 G4cout << ">>> Event " << evt->GetEventID() 68 G4cout << ">>> Event " << evt->GetEventID() << G4endl; >> 69 >> 70 if(trackerCollID<0||calorimeterCollID<0||muonCollID<0) return; 70 71 71 if (ftrackerCollID < 0 || fcalorimeterCollID << 72 G4HCofThisEvent * HCE = evt->GetHCofThisEvent(); 72 << 73 ExN04TrackerHitsCollection* THC = 0; 73 G4HCofThisEvent* HCE = evt->GetHCofThisEvent << 74 ExN04CalorimeterHitsCollection* CHC = 0; 74 ExN04TrackerHitsCollection* THC = NULL; << 75 ExN04MuonHitsCollection* MHC = 0; 75 ExN04CalorimeterHitsCollection* CHC = NULL; << 76 if(HCE) 76 ExN04MuonHitsCollection* MHC = NULL; << 77 { 77 << 78 THC = (ExN04TrackerHitsCollection*)(HCE->GetHC(trackerCollID)); 78 if (HCE) { << 79 CHC = (ExN04CalorimeterHitsCollection*)(HCE->GetHC(calorimeterCollID)); 79 THC = (ExN04TrackerHitsCollection*)(HCE->G << 80 MHC = (ExN04MuonHitsCollection*)(HCE->GetHC(muonCollID)); 80 CHC = (ExN04CalorimeterHitsCollection*)(HC << 81 MHC = (ExN04MuonHitsCollection*)(HCE->GetH << 82 } 81 } 83 82 84 if (THC) { << 83 if(THC) 85 G4int n_hit = THC->entries(); << 84 { 86 G4cout << " " << n_hit << " hits are s << 85 int n_hit = THC->entries(); >> 86 G4cout << " " << n_hit >> 87 << " hits are stored in ExN04TrackerHitsCollection." << G4endl; 87 } 88 } 88 << 89 if(CHC) 89 if (CHC) { << 90 { 90 G4int n_hit = CHC->entries(); << 91 int n_hit = CHC->entries(); 91 G4cout << " " << n_hit << " hits are s << 92 G4cout << " " << n_hit >> 93 << " hits are stored in ExN04CalorimeterHitsCollection." << G4endl; 92 G4double totE = 0; 94 G4double totE = 0; 93 for (int i = 0; i < n_hit; i++) { << 95 for(int i=0;i<n_hit;i++) 94 totE += (*CHC)[i]->GetEdep(); << 96 { totE += (*CHC)[i]->GetEdep(); } 95 } << 97 G4cout << " Total energy deposition in calorimeter : " 96 G4cout << " Total energy deposition in << 98 << totE / GeV << " (GeV)" << G4endl; 97 } 99 } 98 << 100 if(MHC) 99 if (MHC) { << 101 { 100 G4int n_hit = MHC->entries(); << 102 int n_hit = MHC->entries(); 101 G4cout << " " << n_hit << " hits are s << 103 G4cout << " " << n_hit >> 104 << " hits are stored in ExN04MuonHitsCollection." << G4endl; 102 } 105 } 103 } 106 } >> 107 104 108