1 | // %%%%%%%%%% |
---|
2 | // G4 headers |
---|
3 | // %%%%%%%%%% |
---|
4 | #include "G4Event.hh" |
---|
5 | |
---|
6 | // %%%%%%%%%%%%% |
---|
7 | // gemc headers |
---|
8 | // %%%%%%%%%%%%% |
---|
9 | #include "MEventAction.h" |
---|
10 | #include "MHit.h" |
---|
11 | |
---|
12 | #include <iostream> |
---|
13 | using namespace std; |
---|
14 | |
---|
15 | MEventAction::MEventAction(gemc_opts opts) {gemcOpt = opts;} |
---|
16 | MEventAction::~MEventAction(){;} |
---|
17 | |
---|
18 | void MEventAction::BeginOfEventAction(const G4Event* evt) |
---|
19 | { |
---|
20 | string hd_msg = gemcOpt.args["LOG_MSG"].args + " Event Action: >> "; |
---|
21 | int Modulo = (int) gemcOpt.args["PRINT_EVENT"].arg ; |
---|
22 | |
---|
23 | if(evtN%Modulo == 0 ) |
---|
24 | cout << hd_msg << " Begin of event " << evtN << endl; |
---|
25 | |
---|
26 | } |
---|
27 | |
---|
28 | void MEventAction::EndOfEventAction(const G4Event* evt) |
---|
29 | { |
---|
30 | string hd_msg = gemcOpt.args["LOG_MSG"].args + " Event Action: >> "; |
---|
31 | int Modulo = (int) gemcOpt.args["PRINT_EVENT"].arg ; |
---|
32 | double VERB = gemcOpt.args["OUT_VERBOSITY"].arg ; |
---|
33 | string catch_v = gemcOpt.args["CATCH"].args; |
---|
34 | |
---|
35 | |
---|
36 | if(evtN%Modulo == 0 ) |
---|
37 | cout << hd_msg << " End of event " << evtN << endl << endl; |
---|
38 | |
---|
39 | MHitCollection* MHC; |
---|
40 | MHit* aHit; |
---|
41 | int nhits; |
---|
42 | MPHBaseClass *ProcessHitRoutine = NULL; |
---|
43 | MOutputBaseClass *ProcessOutput = NULL; |
---|
44 | |
---|
45 | PH_output output; |
---|
46 | int bitcid; |
---|
47 | |
---|
48 | double Etot; |
---|
49 | |
---|
50 | // Making sure the output routine exists in the ProcessOutput map |
---|
51 | map<string, MOutput_Factory>::iterator ito = Out->find(MOut->outType); |
---|
52 | if(ito == Out->end()) |
---|
53 | { |
---|
54 | if(MOut->outType != "no") |
---|
55 | cout << hd_msg << " Warning: output type <" << MOut->outType << "> is not registered in the <MOutput_Factory>. " << endl |
---|
56 | << " This event will not be written out." << endl; |
---|
57 | return; |
---|
58 | } |
---|
59 | ProcessOutput = GetMOutputClass((*Out), MOut->outType); |
---|
60 | |
---|
61 | // Header Bank contains event number |
---|
62 | // Need to change this to read DB header bank |
---|
63 | ProcessOutput->SetOutpHeader(evtN, MOut); |
---|
64 | |
---|
65 | // Getting Generated Particles info |
---|
66 | vector<MGeneratedParticle> MPrimaries; |
---|
67 | for(int pv=0; pv< evt->GetNumberOfPrimaryVertex(); pv++) |
---|
68 | { |
---|
69 | MGeneratedParticle Mparticle; |
---|
70 | G4PrimaryVertex* MPV = evt->GetPrimaryVertex(pv); |
---|
71 | Mparticle.vertex = MPV->GetPosition(); |
---|
72 | for(int pp = 0; pp<MPV->GetNumberOfParticle(); pp++) |
---|
73 | { |
---|
74 | |
---|
75 | G4PrimaryParticle *PP = MPV->GetPrimary(pp); |
---|
76 | Mparticle.momentum = PP->GetMomentum(); |
---|
77 | Mparticle.PID = PP->GetPDGcode(); |
---|
78 | } |
---|
79 | MPrimaries.push_back(Mparticle) ; |
---|
80 | } |
---|
81 | ProcessOutput-> WriteGenerated(MOut, MPrimaries); |
---|
82 | |
---|
83 | for(map<string, MSensitiveDetector*>::iterator it = SeDe_Map.begin(); it!= SeDe_Map.end(); it++) |
---|
84 | { |
---|
85 | string hitType; |
---|
86 | ProcessOutput->SetBankHeader(it->second->SDID.id, it->first, MOut); |
---|
87 | |
---|
88 | MHC = it->second->GetMHitCollection(); |
---|
89 | nhits = MHC->GetSize(); |
---|
90 | // The same ProcessHit Routine must apply to all the hits in this HitCollection. |
---|
91 | // Instantiating the ProcessHitRoutine only once for the first hit. |
---|
92 | if(nhits) |
---|
93 | { |
---|
94 | aHit = (*MHC)[0]; |
---|
95 | string vname = aHit->GetDetector().name; |
---|
96 | hitType = it->second->GetDetectorHitType(vname); |
---|
97 | |
---|
98 | // Making sure the Routine exists in the MProcessHit_Map |
---|
99 | map<string, MPHB_Factory>::iterator itp = MProcessHit_Map->find(hitType); |
---|
100 | if(itp == MProcessHit_Map->end()) |
---|
101 | { |
---|
102 | cout << hd_msg << " Warning: hit Type <" << hitType << "> is not registered in the <MProcessHit_Map>. " << endl |
---|
103 | << " This hit collection will not be processed." << endl; |
---|
104 | return; |
---|
105 | } |
---|
106 | ProcessHitRoutine = GetMPHClass((*MProcessHit_Map), hitType); |
---|
107 | |
---|
108 | for(int h=0; h<nhits; h++) |
---|
109 | { |
---|
110 | aHit = (*MHC)[h]; |
---|
111 | output = ProcessHitRoutine->ProcessHit(aHit, gemcOpt); |
---|
112 | bitcid = it->second->get_bit_compressed_id(output.identity); |
---|
113 | ProcessOutput->ProcessOutput(bitcid, output, MOut, (*MBank_Map)[hitType]); |
---|
114 | |
---|
115 | string vname = aHit->GetId()[aHit->GetId().size()-1].name; |
---|
116 | if(VERB > 4 || vname.find(catch_v) != string::npos) |
---|
117 | { |
---|
118 | cout << hd_msg << " Hit " << h + 1 << " -- total number of steps this hit: " << aHit->GetPos().size() << endl; |
---|
119 | cout << aHit->GetId(); |
---|
120 | Etot = 0; |
---|
121 | for(int e=0; e<aHit->GetPos().size(); e++) Etot = Etot + aHit->GetEdep()[e]; |
---|
122 | cout << " Total energy deposited: " << Etot/MeV << " MeV" << endl; |
---|
123 | } |
---|
124 | } |
---|
125 | // cout << it->second->HCname << " " << evtN << " " << nhits << endl; |
---|
126 | } |
---|
127 | ProcessOutput->RecordAndClear(MOut, (*MBank_Map)[hitType]); |
---|
128 | |
---|
129 | } |
---|
130 | ProcessOutput->WriteEvent(MOut); |
---|
131 | |
---|
132 | delete ProcessOutput; |
---|
133 | delete ProcessHitRoutine; |
---|
134 | |
---|
135 | evtN++; |
---|
136 | } |
---|
137 | |
---|
138 | |
---|
139 | |
---|
140 | |
---|
141 | |
---|
142 | |
---|
143 | |
---|
144 | |
---|
145 | |
---|
146 | |
---|
147 | |
---|