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