| Geant4 Cross Reference |
1 // 1 //
2 // ******************************************* 2 // ********************************************************************
3 // * License and Disclaimer << 3 // * DISCLAIMER *
4 // * 4 // * *
5 // * The Geant4 software is copyright of th << 5 // * The following disclaimer summarizes all the specific disclaimers *
6 // * the Geant4 Collaboration. It is provided << 6 // * of contributors to this software. The specific disclaimers,which *
7 // * conditions of the Geant4 Software License << 7 // * govern, are listed with their locations in: *
8 // * LICENSE and available at http://cern.ch/ << 8 // * http://cern.ch/geant4/license *
9 // * include a list of copyright holders. <<
10 // * 9 // * *
11 // * Neither the authors of this software syst 10 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing fin 11 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warran 12 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assum 13 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file << 14 // * use. *
16 // * for the full disclaimer and the limitatio <<
17 // * 15 // * *
18 // * This code implementation is the result << 16 // * This code implementation is the intellectual property of the *
19 // * technical work of the GEANT4 collaboratio << 17 // * GEANT4 collaboration. *
20 // * By using, copying, modifying or distri << 18 // * By copying, distributing or modifying the Program (or any work *
21 // * any work based on the software) you ag << 19 // * based on the Program) you indicate your acceptance of this *
22 // * use in resulting scientific publicati << 20 // * statement, and all its terms. *
23 // * acceptance of all terms of the Geant4 Sof <<
24 // ******************************************* 21 // ********************************************************************
25 // 22 //
26 /// \file eventgenerator/HepMC/HepMCEx01/src/E <<
27 /// \brief Implementation of the ExN04EventAct <<
28 // <<
29 // <<
30 23
31 #include "ExN04EventAction.hh" 24 #include "ExN04EventAction.hh"
32 25
>> 26 #include "ExN04TrackerHit.hh"
33 #include "ExN04CalorimeterHit.hh" 27 #include "ExN04CalorimeterHit.hh"
34 #include "ExN04MuonHit.hh" 28 #include "ExN04MuonHit.hh"
35 #include "ExN04TrackerHit.hh" <<
36 29
37 #include "G4Event.hh" 30 #include "G4Event.hh"
38 #include "G4EventManager.hh" 31 #include "G4EventManager.hh"
39 #include "G4HCofThisEvent.hh" 32 #include "G4HCofThisEvent.hh"
>> 33 #include "G4VHitsCollection.hh"
>> 34 #include "G4TrajectoryContainer.hh"
>> 35 #include "G4Trajectory.hh"
>> 36 #include "G4VVisManager.hh"
40 #include "G4SDManager.hh" 37 #include "G4SDManager.hh"
41 #include "G4SystemOfUnits.hh" <<
42 #include "G4UImanager.hh" 38 #include "G4UImanager.hh"
43 #include "G4VHitsCollection.hh" <<
44 #include "G4ios.hh" 39 #include "G4ios.hh"
45 40
46 //....oooOO0OOooo........oooOO0OOooo........oo <<
47 ExN04EventAction::ExN04EventAction() 41 ExN04EventAction::ExN04EventAction()
48 : G4UserEventAction(), ftrackerCollID(-1), f << 42 {
49 {} << 43 trackerCollID = -1;
>> 44 calorimeterCollID = -1;
>> 45 muonCollID = -1;
>> 46 }
50 47
51 //....oooOO0OOooo........oooOO0OOooo........oo << 48 ExN04EventAction::~ExN04EventAction()
52 ExN04EventAction::~ExN04EventAction() {} << 49 {;}
53 50
54 //....oooOO0OOooo........oooOO0OOooo........oo <<
55 void ExN04EventAction::BeginOfEventAction(cons 51 void ExN04EventAction::BeginOfEventAction(const G4Event*)
56 { 52 {
57 G4SDManager* SDman = G4SDManager::GetSDMpoin << 53 G4SDManager * SDman = G4SDManager::GetSDMpointer();
58 if (ftrackerCollID < 0 || fcalorimeterCollID << 54 if(trackerCollID<0||calorimeterCollID<0||muonCollID<0)
>> 55 {
59 G4String colNam; 56 G4String colNam;
60 ftrackerCollID = SDman->GetCollectionID(co << 57 trackerCollID = SDman->GetCollectionID(colNam="trackerCollection");
61 fcalorimeterCollID = SDman->GetCollectionI << 58 calorimeterCollID = SDman->GetCollectionID(colNam="calCollection");
62 fmuonCollID = SDman->GetCollectionID(colNa << 59 muonCollID = SDman->GetCollectionID(colNam="muonCollection");
63 } 60 }
64 } 61 }
65 62
66 //....oooOO0OOooo........oooOO0OOooo........oo <<
67 void ExN04EventAction::EndOfEventAction(const 63 void ExN04EventAction::EndOfEventAction(const G4Event* evt)
68 { 64 {
69 G4cout << ">>> Event " << evt->GetEventID() 65 G4cout << ">>> Event " << evt->GetEventID() << G4endl;
70 << 66
71 if (ftrackerCollID < 0 || fcalorimeterCollID << 67 // get number of stored trajectories
72 << 68 //
73 G4HCofThisEvent* HCE = evt->GetHCofThisEvent << 69 G4TrajectoryContainer* trajectoryContainer = evt->GetTrajectoryContainer();
74 ExN04TrackerHitsCollection* THC = NULL; << 70 G4int n_trajectories = 0;
75 ExN04CalorimeterHitsCollection* CHC = NULL; << 71 if (trajectoryContainer) n_trajectories = trajectoryContainer->entries();
76 ExN04MuonHitsCollection* MHC = NULL; << 72
77 << 73 // extract the trajectories and draw them
78 if (HCE) { << 74 //
79 THC = (ExN04TrackerHitsCollection*)(HCE->G << 75 if (G4VVisManager::GetConcreteInstance())
80 CHC = (ExN04CalorimeterHitsCollection*)(HC << 76 {
81 MHC = (ExN04MuonHitsCollection*)(HCE->GetH << 77 for (G4int i=0; i<n_trajectories; i++)
>> 78 { G4Trajectory* trj = (G4Trajectory*)
>> 79 ((*(evt->GetTrajectoryContainer()))[i]);
>> 80 trj->DrawTrajectory(50);
>> 81 }
>> 82 }
>> 83
>> 84
>> 85 if(trackerCollID<0||calorimeterCollID<0||muonCollID<0) return;
>> 86
>> 87 G4HCofThisEvent * HCE = evt->GetHCofThisEvent();
>> 88 ExN04TrackerHitsCollection* THC = 0;
>> 89 ExN04CalorimeterHitsCollection* CHC = 0;
>> 90 ExN04MuonHitsCollection* MHC = 0;
>> 91 if(HCE)
>> 92 {
>> 93 THC = (ExN04TrackerHitsCollection*)(HCE->GetHC(trackerCollID));
>> 94 CHC = (ExN04CalorimeterHitsCollection*)(HCE->GetHC(calorimeterCollID));
>> 95 MHC = (ExN04MuonHitsCollection*)(HCE->GetHC(muonCollID));
82 } 96 }
83 97
84 if (THC) { << 98 if(THC)
85 G4int n_hit = THC->entries(); << 99 {
86 G4cout << " " << n_hit << " hits are s << 100 int n_hit = THC->entries();
>> 101 G4cout << " " << n_hit
>> 102 << " hits are stored in ExN04TrackerHitsCollection." << G4endl;
87 } 103 }
88 << 104 if(CHC)
89 if (CHC) { << 105 {
90 G4int n_hit = CHC->entries(); << 106 int n_hit = CHC->entries();
91 G4cout << " " << n_hit << " hits are s << 107 G4cout << " " << n_hit
>> 108 << " hits are stored in ExN04CalorimeterHitsCollection." << G4endl;
92 G4double totE = 0; 109 G4double totE = 0;
93 for (int i = 0; i < n_hit; i++) { << 110 for(int i=0;i<n_hit;i++)
94 totE += (*CHC)[i]->GetEdep(); << 111 { totE += (*CHC)[i]->GetEdep(); }
95 } << 112 G4cout << " Total energy deposition in calorimeter : "
96 G4cout << " Total energy deposition in << 113 << totE / GeV << " (GeV)" << G4endl;
97 } 114 }
98 << 115 if(MHC)
99 if (MHC) { << 116 {
100 G4int n_hit = MHC->entries(); << 117 int n_hit = MHC->entries();
101 G4cout << " " << n_hit << " hits are s << 118 G4cout << " " << n_hit
>> 119 << " hits are stored in ExN04MuonHitsCollection." << G4endl;
102 } 120 }
103 } 121 }
>> 122
104 123