Geant4 Cross Reference |
1 // 2 // ******************************************************************** 3 // * License and Disclaimer * 4 // * * 5 // * The Geant4 software is copyright of the Copyright Holders of * 6 // * the Geant4 Collaboration. It is provided under the terms and * 7 // * conditions of the Geant4 Software License, included in the file * 8 // * LICENSE and available at http://cern.ch/geant4/license . These * 9 // * include a list of copyright holders. * 10 // * * 11 // * Neither the authors of this software system, nor their employing * 12 // * institutes,nor the agencies providing financial support for this * 13 // * work make any representation or warranty, express or implied, * 14 // * regarding this software system or assume any liability for its * 15 // * use. Please see the license in the file LICENSE and URL above * 16 // * for the full disclaimer and the limitation of liability. * 17 // * * 18 // * This code implementation is the result of the scientific and * 19 // * technical work of the GEANT4 collaboration. * 20 // * By using, copying, modifying or distributing the software (or * 21 // * any work based on the software) you agree to acknowledge its * 22 // * use in resulting scientific publications, and indicate your * 23 // * acceptance of all terms of the Geant4 Software license. * 24 // ******************************************************************** 25 // 26 /// \file eventgenerator/HepMC/HepMCEx01/src/ExN04EventAction.cc 27 /// \brief Implementation of the ExN04EventAction class 28 // 29 // 30 31 #include "ExN04EventAction.hh" 32 33 #include "ExN04CalorimeterHit.hh" 34 #include "ExN04MuonHit.hh" 35 #include "ExN04TrackerHit.hh" 36 37 #include "G4Event.hh" 38 #include "G4EventManager.hh" 39 #include "G4HCofThisEvent.hh" 40 #include "G4SDManager.hh" 41 #include "G4SystemOfUnits.hh" 42 #include "G4UImanager.hh" 43 #include "G4VHitsCollection.hh" 44 #include "G4ios.hh" 45 46 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 47 ExN04EventAction::ExN04EventAction() 48 : G4UserEventAction(), ftrackerCollID(-1), fcalorimeterCollID(-1), fmuonCollID(-1) 49 {} 50 51 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 52 ExN04EventAction::~ExN04EventAction() {} 53 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 55 void ExN04EventAction::BeginOfEventAction(const G4Event*) 56 { 57 G4SDManager* SDman = G4SDManager::GetSDMpointer(); 58 if (ftrackerCollID < 0 || fcalorimeterCollID < 0 || fmuonCollID < 0) { 59 G4String colNam; 60 ftrackerCollID = SDman->GetCollectionID(colNam = "trackerCollection"); 61 fcalorimeterCollID = SDman->GetCollectionID(colNam = "calCollection"); 62 fmuonCollID = SDman->GetCollectionID(colNam = "muonCollection"); 63 } 64 } 65 66 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 67 void ExN04EventAction::EndOfEventAction(const G4Event* evt) 68 { 69 G4cout << ">>> Event " << evt->GetEventID() << G4endl; 70 71 if (ftrackerCollID < 0 || fcalorimeterCollID < 0 || fmuonCollID < 0) return; 72 73 G4HCofThisEvent* HCE = evt->GetHCofThisEvent(); 74 ExN04TrackerHitsCollection* THC = NULL; 75 ExN04CalorimeterHitsCollection* CHC = NULL; 76 ExN04MuonHitsCollection* MHC = NULL; 77 78 if (HCE) { 79 THC = (ExN04TrackerHitsCollection*)(HCE->GetHC(ftrackerCollID)); 80 CHC = (ExN04CalorimeterHitsCollection*)(HCE->GetHC(fcalorimeterCollID)); 81 MHC = (ExN04MuonHitsCollection*)(HCE->GetHC(fmuonCollID)); 82 } 83 84 if (THC) { 85 G4int n_hit = THC->entries(); 86 G4cout << " " << n_hit << " hits are stored in ExN04TrackerHitsCollection." << G4endl; 87 } 88 89 if (CHC) { 90 G4int n_hit = CHC->entries(); 91 G4cout << " " << n_hit << " hits are stored in ExN04CalorimeterHitsCollection." << G4endl; 92 G4double totE = 0; 93 for (int i = 0; i < n_hit; i++) { 94 totE += (*CHC)[i]->GetEdep(); 95 } 96 G4cout << " Total energy deposition in calorimeter : " << totE / GeV << " (GeV)" << G4endl; 97 } 98 99 if (MHC) { 100 G4int n_hit = MHC->entries(); 101 G4cout << " " << n_hit << " hits are stored in ExN04MuonHitsCollection." << G4endl; 102 } 103 } 104