| 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 // Hadrontherapy advanced example for Geant4 << 26 // $Id: HadrontherapyEventAction.cc;
27 // See more at: https://twiki.cern.ch/twiki/bi << 27 // Last modified: G.A.P.Cirrone March 2008;
28 << 28 //
29 #include "G4SystemOfUnits.hh" << 29 // See more at: http://geant4infn.wikispaces.com/HadrontherapyExample
>> 30 //
>> 31 // ----------------------------------------------------------------------------
>> 32 // GEANT 4 - Hadrontherapy example
>> 33 // ----------------------------------------------------------------------------
>> 34 // Code developed by:
>> 35 //
>> 36 // G.A.P. Cirrone(a)*, G. Candiano, F. Di Rosa(a), S. Guatelli(b), G. Russo(a)
>> 37 //
>> 38 // (a) Laboratori Nazionali del Sud
>> 39 // of the National Institute for Nuclear Physics, Catania, Italy
>> 40 // (b) National Institute for Nuclear Physics Section of Genova, genova, Italy
>> 41 //
>> 42 // * cirrone@lns.infn.it
>> 43 // --------------------------------------------------------------
30 #include "G4Event.hh" 44 #include "G4Event.hh"
31 #include "G4EventManager.hh" 45 #include "G4EventManager.hh"
32 #include "G4HCofThisEvent.hh" 46 #include "G4HCofThisEvent.hh"
33 #include "G4VHitsCollection.hh" 47 #include "G4VHitsCollection.hh"
>> 48 #include "G4TrajectoryContainer.hh"
>> 49 #include "G4Trajectory.hh"
>> 50 #include "G4VVisManager.hh"
34 #include "G4SDManager.hh" 51 #include "G4SDManager.hh"
35 #include "G4VVisManager.hh" 52 #include "G4VVisManager.hh"
36 <<
37 #include "HadrontherapyEventAction.hh" 53 #include "HadrontherapyEventAction.hh"
38 #include "HadrontherapyDetectorHit.hh" 54 #include "HadrontherapyDetectorHit.hh"
39 #include "HadrontherapyDetectorSD.hh" 55 #include "HadrontherapyDetectorSD.hh"
40 #include "HadrontherapyDetectorConstruction.hh 56 #include "HadrontherapyDetectorConstruction.hh"
41 #include "HadrontherapyMatrix.hh" 57 #include "HadrontherapyMatrix.hh"
42 #include "HadrontherapyEventActionMessenger.hh <<
43 58
44 ////////////////////////////////////////////// << 59 HadrontherapyEventAction::HadrontherapyEventAction(HadrontherapyMatrix* matrixPointer) :
45 HadrontherapyEventAction::HadrontherapyEventAc << 60 drawFlag("all" ),printModulo(10000)
46 drawFlag("all" ),printModulo(10), pointerEvent << 61 {
47 { << 62 hitsCollectionID = -1;
48 hitsCollectionID = -1; << 63 matrix = matrixPointer;
49 pointerEventMessenger = new HadrontherapyE <<
50 } 64 }
51 65
52 ////////////////////////////////////////////// <<
53 HadrontherapyEventAction::~HadrontherapyEventA 66 HadrontherapyEventAction::~HadrontherapyEventAction()
54 { 67 {
55 delete pointerEventMessenger; <<
56 } 68 }
57 69
58 ////////////////////////////////////////////// << 70 void HadrontherapyEventAction::BeginOfEventAction(const G4Event* evt)
59 void HadrontherapyEventAction::BeginOfEventAct <<
60 { 71 {
61 G4SDManager* pSDManager = G4SDManager::Get << 72
62 if(hitsCollectionID == -1) << 73 G4int evtNb = evt->GetEventID();
63 hitsCollectionID = pSDManager -> GetCo << 74
>> 75 //printing survey
>> 76 if (evtNb%printModulo == 0)
>> 77 G4cout << "\n---> Begin of Event: " << evtNb << G4endl;
>> 78
>> 79 G4SDManager* pSDManager = G4SDManager::GetSDMpointer();
>> 80 if(hitsCollectionID == -1)
>> 81 hitsCollectionID = pSDManager -> GetCollectionID("HadrontherapyDetectorHitsCollection");
64 } 82 }
65 83
66 ////////////////////////////////////////////// <<
67 void HadrontherapyEventAction::EndOfEventActio 84 void HadrontherapyEventAction::EndOfEventAction(const G4Event* evt)
68 { << 85 {
69 if(hitsCollectionID < 0) << 86 if(hitsCollectionID < 0)
70 return; << 87 return;
71 G4HCofThisEvent* HCE = evt -> GetHCofThisE << 88
72 << 89 G4HCofThisEvent* HCE = evt -> GetHCofThisEvent();
73 // Clear voxels hit list << 90 HadrontherapyDetectorHitsCollection* CHC = NULL;
74 HadrontherapyMatrix* matrix = Hadrontherap << 91
75 if (matrix) matrix -> ClearHitTrack(); << 92 if(HCE)
76 << 93 CHC = (HadrontherapyDetectorHitsCollection*)(HCE -> GetHC(hitsCollectionID));
77 if(HCE) << 94
>> 95 if(CHC)
78 { 96 {
79 HadrontherapyDetectorHitsCollection* C << 97 if(matrix)
80 if(CHC) << 98 {
>> 99 // Fill the matrix with the information: voxel and associated energy deposit
>> 100 // in the detector at the end of the event
>> 101
>> 102 G4int HitCount = CHC -> entries();
>> 103 for (G4int h=0; h<HitCount; h++)
>> 104 {
>> 105 G4int i = ((*CHC)[h]) -> GetXID();
>> 106 G4int j = ((*CHC)[h]) -> GetYID();
>> 107 G4int k = ((*CHC)[h]) -> GetZID();
>> 108 G4double energyDeposit = ((*CHC)[h]) -> GetEdep();
>> 109 matrix -> Fill(i, j, k, energyDeposit/MeV);
>> 110 }
>> 111 }
>> 112 }
>> 113
>> 114 // Extract the trajectories and draw them in the visualisation
>> 115
>> 116 if (G4VVisManager::GetConcreteInstance())
>> 117 {
>> 118 G4TrajectoryContainer * trajectoryContainer = evt -> GetTrajectoryContainer();
>> 119 G4int n_trajectories = 0;
>> 120 if (trajectoryContainer) n_trajectories = trajectoryContainer -> entries();
>> 121
>> 122 for (G4int i = 0; i < n_trajectories; i++)
81 { 123 {
82 if(matrix) << 124 G4Trajectory* trj = (G4Trajectory*)
83 { << 125 ((*(evt -> GetTrajectoryContainer()))[i]);
84 // Fill the matrix with the in << 126 if(drawFlag == "all") trj -> DrawTrajectory(50);
85 // in the detector at the end << 127 else if((drawFlag == "charged")&&(trj -> GetCharge() != 0.))
86 << 128 trj -> DrawTrajectory(50);
87 size_t HitCount = CHC -> entri << 129 else if ((drawFlag == "neutral")&&(trj -> GetCharge() == 0.))
88 for (size_t h=0; h<HitCount; h << 130 trj -> DrawTrajectory(50);
89 { << 131 }
90 G4int i = ((*CHC)[h]) -> G <<
91 G4int j = ((*CHC)[h]) -> G <<
92 G4int k = ((*CHC)[h]) -> G <<
93 G4double energyDeposit = ( <<
94 matrix -> Fill(i, j, k, en <<
95 } <<
96 } <<
97 } <<
98 } 132 }
99 } 133 }
>> 134
100 135