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 // ------------------------------------------- 27 // -------------------------------------------------------------- 28 // GEANT 4 - ULTRA experiment 28 // GEANT 4 - ULTRA experiment example 29 // ------------------------------------------- 29 // -------------------------------------------------------------- 30 // 30 // 31 // Code developed by: 31 // Code developed by: 32 // B. Tome, M.C. Espirito-Santo, A. Trindade, << 32 // B. Tome, M.C. Espirito-Santo, A. Trindade, P. Rodrigues 33 // 33 // 34 // **************************************** 34 // **************************************************** 35 // * UltraEventAction.cc 35 // * UltraEventAction.cc 36 // **************************************** 36 // **************************************************** 37 // 37 // 38 // Ultra EventAction class. The UltraAnalys 38 // Ultra EventAction class. The UltraAnalysisManager class is used for histogram 39 // filling << 39 // filling if the G4ANALYSIS_USE environment variable is set. 40 // 40 // 41 #include "UltraEventAction.hh" 41 #include "UltraEventAction.hh" >> 42 #include "UltraRunAction.hh" 42 #include "UltraPrimaryGeneratorAction.hh" 43 #include "UltraPrimaryGeneratorAction.hh" 43 #include "UltraOpticalHit.hh" 44 #include "UltraOpticalHit.hh" 44 45 45 #include "G4RunManager.hh" << 46 #include "G4RunManager.hh" 46 #include "G4Event.hh" 47 #include "G4Event.hh" 47 #include "G4EventManager.hh" << 48 #include "G4EventManager.hh" 48 #include "G4SDManager.hh" 49 #include "G4SDManager.hh" 49 #include "G4HCofThisEvent.hh" 50 #include "G4HCofThisEvent.hh" 50 #include "G4VHitsCollection.hh" 51 #include "G4VHitsCollection.hh" 51 #include "G4GeneralParticleSource.hh" << 52 #include "G4GeneralParticleSource.hh" 52 #include "G4AnalysisManager.hh" << 53 #include "G4SystemOfUnits.hh" << 54 53 >> 54 #include "G4TrajectoryContainer.hh" >> 55 #include "G4Trajectory.hh" >> 56 #include "G4VVisManager.hh" >> 57 >> 58 #ifdef G4ANALYSIS_USE >> 59 #include "UltraAnalysisManager.hh" >> 60 #endif 55 //....oooOO0OOooo........oooOO0OOooo........oo 61 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 56 62 57 UltraEventAction::UltraEventAction() << 63 UltraEventAction::UltraEventAction(UltraRunAction* run) 58 :OpticalHitsCollID(-1) << 64 :UltraRun(run),OpticalHitsCollID(-1) 59 {;} 65 {;} 60 66 61 //....oooOO0OOooo........oooOO0OOooo........oo 67 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 62 68 63 UltraEventAction::~UltraEventAction(){;} 69 UltraEventAction::~UltraEventAction(){;} 64 70 65 //....oooOO0OOooo........oooOO0OOooo........oo 71 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 66 72 67 void UltraEventAction::BeginOfEventAction(cons 73 void UltraEventAction::BeginOfEventAction(const G4Event* evt) 68 { 74 { 69 auto printModulo = 100; << 75 G4int printModulo = 100; 70 76 71 evtNb = evt->GetEventID(); 77 evtNb = evt->GetEventID(); 72 78 73 auto SDman = G4SDManager::GetSDMpointer(); << 79 G4SDManager * SDman = G4SDManager::GetSDMpointer(); 74 80 75 81 76 if(OpticalHitsCollID==-1) { 82 if(OpticalHitsCollID==-1) { 77 OpticalHitsCollID = SDman->GetCollectionID 83 OpticalHitsCollID = SDman->GetCollectionID("OpticalHitsCollection"); 78 } 84 } 79 85 80 86 81 if (evtNb%printModulo == 0) 87 if (evtNb%printModulo == 0) 82 G4cout << "\n---> Begin of Event: " << evt 88 G4cout << "\n---> Begin of Event: " << evtNb << G4endl; 83 89 84 } 90 } 85 //....oooOO0OOooo........oooOO0OOooo........oo 91 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 86 92 87 void UltraEventAction::EndOfEventAction(const 93 void UltraEventAction::EndOfEventAction(const G4Event* evt) 88 { 94 { 89 95 90 auto HCE = evt->GetHCofThisEvent(); << 96 G4HCofThisEvent* HCE = evt->GetHCofThisEvent(); 91 UltraOpticalHitsCollection* OpticalHitsColl << 97 UltraOpticalHitsCollection* OpticalHitsColl = 0; >> 98 >> 99 if(HCE){ 92 100 93 // Fill histograms << 101 if(OpticalHitsCollID != -1) OpticalHitsColl = 94 auto man = G4AnalysisManager::Instance(); << 102 (UltraOpticalHitsCollection*)(HCE->GetHC(OpticalHitsCollID)); 95 103 96 if(HCE){ << 104 } 97 if(OpticalHitsCollID != -1) OpticalHitsCol << 105 G4int nOptHits = 0 ; 98 (UltraOpticalHitsCollection*)(HCE->GetHC << 99 } << 100 << 101 auto nOptHits = 0 ; << 102 106 103 if(OpticalHitsColl){ 107 if(OpticalHitsColl){ >> 108 104 nOptHits = OpticalHitsColl->entries(); 109 nOptHits = OpticalHitsColl->entries(); 105 110 106 #ifdef ULTRA_VERBOSE 111 #ifdef ULTRA_VERBOSE 107 if (nOptHits > 0){ 112 if (nOptHits > 0){ 108 G4cout << " Optical Hit # " << " " << "E << 113 G4cout << " Optical Hit # " << " " << "Energy (eV)" << " " << "x,y,z (cm)" << G4endl ; 109 } 114 } 110 #endif 115 #endif 111 116 112 for(auto iHit=0; iHit<nOptHits; iHit++){ << 117 for(G4int iHit=0; iHit<nOptHits; iHit++){ 113 auto HitEnergy = (*OpticalHitsColl)[iHit << 118 114 man->FillH1(1,HitEnergy/eV); << 119 G4double HitEnergy ; >> 120 HitEnergy = (*OpticalHitsColl)[iHit]->GetEnergy() ; >> 121 G4ThreeVector HitPos = (*OpticalHitsColl)[iHit]->GetPosition()/cm; >> 122 >> 123 #ifdef G4ANALYSIS_USE >> 124 UltraAnalysisManager* analysis = UltraAnalysisManager::getInstance(); >> 125 analysis->FillHistogram(1,HitEnergy/eV); >> 126 #endif >> 127 115 } 128 } 116 129 117 } << 130 } >> 131 >> 132 #ifdef G4ANALYSIS_USE >> 133 UltraAnalysisManager* analysis = UltraAnalysisManager::getInstance(); >> 134 analysis->FillHistogram(2,nOptHits); >> 135 #endif >> 136 >> 137 // Define visualization atributes: 118 138 119 man->FillH1(2,nOptHits); << 139 G4String drawFlag = "all"; >> 140 G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance(); >> 141 >> 142 if(pVVisManager) >> 143 { >> 144 G4TrajectoryContainer* trajectoryContainer = evt->GetTrajectoryContainer(); >> 145 G4int n_trajectories = 0; >> 146 if (trajectoryContainer) n_trajectories = trajectoryContainer->entries(); >> 147 for(G4int i=0; i<n_trajectories; i++) >> 148 { G4Trajectory* trj = (G4Trajectory *)((*(evt->GetTrajectoryContainer()))[i]); >> 149 if (drawFlag == "all") trj->DrawTrajectory(); >> 150 else if ((drawFlag == "charged")&&(trj->GetCharge() != 0.)) >> 151 trj->DrawTrajectory(); >> 152 } >> 153 } 120 154 121 } 155 } >> 156 >> 157 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.... 122 158