| 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 // 26 //
27 /// \file optical/wls/src/WLSEventAction.cc <<
28 /// \brief Implementation of the WLSEventActio <<
29 // <<
30 // 27 //
31 28
32 #include "WLSEventAction.hh" 29 #include "WLSEventAction.hh"
33 30
>> 31 #include "WLSRunAction.hh"
>> 32
34 #include "WLSEventActionMessenger.hh" 33 #include "WLSEventActionMessenger.hh"
>> 34
35 #include "WLSPhotonDetHit.hh" 35 #include "WLSPhotonDetHit.hh"
36 #include "WLSRun.hh" <<
37 #include "WLSRunAction.hh" <<
38 36
39 #include "G4AnalysisManager.hh" <<
40 #include "G4Event.hh" 37 #include "G4Event.hh"
41 #include "G4EventManager.hh" 38 #include "G4EventManager.hh"
42 #include "G4RunManager.hh" << 39 #include "WLSTrajectory.hh"
43 #include "G4SDManager.hh" <<
44 #include "G4TrajectoryContainer.hh" 40 #include "G4TrajectoryContainer.hh"
45 #include "G4VVisManager.hh" 41 #include "G4VVisManager.hh"
>> 42 #include "G4SDManager.hh"
>> 43
>> 44 //#include "G4ThreeVector.hh"
>> 45
>> 46 //#include "G4UnitsTable.hh"
>> 47
46 #include "Randomize.hh" 48 #include "Randomize.hh"
47 49
48 // Purpose: Accumulates statistics regarding h << 50 // Purpose: Invoke visualization at the end
>> 51 // Also can accumulate statistics regarding hits
49 // in the PhotonDet detector 52 // in the PhotonDet detector
50 53
51 //....oooOO0OOooo........oooOO0OOooo........oo << 54 WLSEventAction::WLSEventAction(WLSRunAction* RA)
52 << 55 : runaction(RA), verboselevel(0),
53 WLSEventAction::WLSEventAction() << 56 printModulo(100), drawFlag("all")
54 { 57 {
55 fEventMessenger = new WLSEventActionMessenge << 58 eventMessenger = new WLSEventActionMessenger(this);
56 } <<
57 59
58 //....oooOO0OOooo........oooOO0OOooo........oo << 60 forcedrawphotons = false;
>> 61 forcenophotons = false;
>> 62
>> 63 }
59 64
60 WLSEventAction::~WLSEventAction() 65 WLSEventAction::~WLSEventAction()
61 { 66 {
62 delete fEventMessenger; << 67 delete eventMessenger;
63 } 68 }
64 69
65 //....oooOO0OOooo........oooOO0OOooo........oo << 70 void WLSEventAction::BeginOfEventAction(const G4Event* evt)
66 <<
67 void WLSEventAction::BeginOfEventAction(const <<
68 { 71 {
69 fNTIR = 0; << 72 G4int evtNb = evt->GetEventID();
70 fNExiting = 0; << 73 if (evtNb%printModulo == 0)
71 fEscapedEnd = 0; << 74 G4cout << "\n---> Begin of Event: " << evtNb << G4endl;
72 fEscapedMid = 0; <<
73 fBounce = 0; <<
74 fWLSBounce = 0; <<
75 fClad1Bounce = 0; <<
76 fClad2Bounce = 0; <<
77 fReflected = 0; <<
78 fEscaped = 0; <<
79 fMirror = 0; <<
80 } <<
81 75
82 //....oooOO0OOooo........oooOO0OOooo........oo << 76 if(verboselevel>0)
>> 77 G4cout << "<<< Event " << evtNb << " started." << G4endl;
>> 78 }
83 79
84 void WLSEventAction::EndOfEventAction(const G4 80 void WLSEventAction::EndOfEventAction(const G4Event* evt)
85 { 81 {
>> 82 G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
>> 83
>> 84 // Visualization of Trajectory
>> 85 if(pVVisManager)
>> 86 {
>> 87 G4TrajectoryContainer* trajectoryContainer = evt->GetTrajectoryContainer();
>> 88
>> 89 G4int n_trajectories = 0;
>> 90 if (trajectoryContainer) n_trajectories = trajectoryContainer->entries();
>> 91 G4cout << "n_trajectories: " << n_trajectories << G4endl;
>> 92 if (drawFlag == "all") G4cout << "draw all trajectories" << G4endl;
>> 93 if (drawFlag == "charged") G4cout << "draw only charged" << G4endl;
>> 94
>> 95 for(G4int i=0; i<n_trajectories; i++)
>> 96 { WLSTrajectory* trj =
>> 97 (WLSTrajectory *)((*(evt->GetTrajectoryContainer()))[i]);
>> 98 if (drawFlag == "all") {
>> 99 G4cout << "Now calling DrawTrajectory" << G4endl;
>> 100 G4cout << "Particle Name: " << trj->GetParticleName() << G4endl;
>> 101 trj->DrawTrajectory(50);
>> 102 }
>> 103 else if ((drawFlag == "charged")&&(trj->GetCharge() != 0.))
>> 104 trj->DrawTrajectory(50);
>> 105 else if (trj->GetParticleName()=="opticalphoton")
>> 106 {
>> 107 G4cout << "We should be drawing an opticalphoton" << G4endl;
>> 108 trj->SetForceDrawTrajectory(forcedrawphotons);
>> 109 trj->SetForceNoDrawTrajectory(forcenophotons);
>> 110 trj->DrawTrajectory(50);
>> 111 }
>> 112 }
>> 113 }
>> 114
>> 115 if (verboselevel>0)
>> 116 G4cout << "<<< Event " << evt->GetEventID() << " ended." << G4endl;
>> 117
>> 118 // Save the Random Engine Status at the of event
>> 119 if (runaction->GetRndmFreq() == 2)
>> 120 {
>> 121 CLHEP::HepRandom::saveEngineStatus("endOfEvent.rndm");
>> 122 G4int evtNb = evt->GetEventID();
>> 123 if (evtNb%printModulo == 0)
>> 124 {
>> 125 G4cout << "\n---> End of Event: " << evtNb << G4endl;
>> 126 CLHEP::HepRandom::showEngineStatus();
>> 127 }
>> 128 }
>> 129
86 // Get Hits from the detector if any 130 // Get Hits from the detector if any
87 G4SDManager* SDman = G4SDManager::GetSDMpoin << 131 G4SDManager * SDman = G4SDManager::GetSDMpointer();
88 G4String colName = "PhotonDetHitCollection"; 132 G4String colName = "PhotonDetHitCollection";
89 fMPPCCollID = SDman->GetCollectionID(colName << 133 mppcCollID = SDman->GetCollectionID(colName);
90 134
91 G4HCofThisEvent* HCE = evt->GetHCofThisEvent 135 G4HCofThisEvent* HCE = evt->GetHCofThisEvent();
92 WLSPhotonDetHitsCollection* mppcHC = nullptr << 136 WLSPhotonDetHitsCollection* mppcHC = 0;
93 137
94 // Get the hit collections 138 // Get the hit collections
95 if (HCE) { << 139 if (HCE)
96 if (fMPPCCollID >= 0) { << 140 {
97 mppcHC = (WLSPhotonDetHitsCollection*)(H << 141 if (mppcCollID>=0) mppcHC =
98 } << 142 (WLSPhotonDetHitsCollection*)(HCE->GetHC(mppcCollID));
99 } 143 }
100 144
101 // Get hit information about photons that re 145 // Get hit information about photons that reached the detector in this event
102 G4int n_hit = 0; << 146 if (mppcHC)
103 if (mppcHC) { << 147 {
104 n_hit = mppcHC->entries(); << 148 // G4int n_hit = mppcHC->entries();
105 } 149 }
106 <<
107 auto analysisManager = G4AnalysisManager::In <<
108 analysisManager->FillH1(2, mppcHC->entries() <<
109 for (size_t i = 0; i < mppcHC->entries(); ++ <<
110 auto pdHit = (*mppcHC)[i]; <<
111 analysisManager->FillH1(0, pdHit->GetEnerg <<
112 analysisManager->FillH1(1, pdHit->GetArriv <<
113 } <<
114 <<
115 if (fVerboseLevel > 1) { <<
116 G4cout << "------------------------------- <<
117 << " In this event, number of:" << <<
118 << " Exiting: " << fNExiting <<
119 << G4endl << " Escaped End: " << <<
120 << " Bounced: " << fBounce < <<
121 << G4endl << " Clad1 Bounce: " << <<
122 << " Clad2 Bounce: " << fClad2Bou <<
123 << G4endl << " Escaped: " << <<
124 << G4endl << " Detector hit: " << <<
125 } <<
126 <<
127 auto run = static_cast<WLSRun*>(G4RunManager <<
128 <<
129 run->AddTIR(fNTIR); <<
130 run->AddExiting(fNExiting); <<
131 run->AddEscapedEnd(fEscapedEnd); <<
132 run->AddEscapedMid(fEscapedMid); <<
133 run->AddBounce(fBounce); <<
134 run->AddClad1Bounce(fClad1Bounce); <<
135 run->AddClad2Bounce(fClad2Bounce); <<
136 run->AddReflected(fReflected); <<
137 run->AddEscaped(fEscaped); <<
138 run->AddMirror(fMirror); <<
139 run->AddDetectorHits(n_hit); <<
140 } 150 }
141 151
142 //....oooOO0OOooo........oooOO0OOooo........oo <<
143 <<
144 G4int WLSEventAction::GetEventNo() 152 G4int WLSEventAction::GetEventNo()
145 { 153 {
146 return fpEventManager->GetConstCurrentEvent( << 154 G4int evno = fpEventManager->GetConstCurrentEvent()->GetEventID();
>> 155 return evno;
147 } 156 }
148 157
149 //....oooOO0OOooo........oooOO0OOooo........oo <<
150 <<
151 void WLSEventAction::SetEventVerbose(G4int lev 158 void WLSEventAction::SetEventVerbose(G4int level)
152 { 159 {
153 fVerboseLevel = level; << 160 verboselevel = level;
154 } 161 }
155 162