| 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 28
29 #include "G4RTPrimaryGeneratorAction.hh" 29 #include "G4RTPrimaryGeneratorAction.hh"
30 #include "G4ParticleDefinition.hh" 30 #include "G4ParticleDefinition.hh"
31 #include "G4ParticleTable.hh" 31 #include "G4ParticleTable.hh"
>> 32 #include "G4GeometryManager.hh"
32 #include "G4TransportationManager.hh" 33 #include "G4TransportationManager.hh"
>> 34 #include "G4VPhysicalVolume.hh"
33 #include "G4Event.hh" 35 #include "G4Event.hh"
34 #include "G4PrimaryVertex.hh" 36 #include "G4PrimaryVertex.hh"
35 #include "G4PrimaryParticle.hh" 37 #include "G4PrimaryParticle.hh"
36 38
37 #include "G4TheMTRayTracer.hh" 39 #include "G4TheMTRayTracer.hh"
38 40
39 G4RTPrimaryGeneratorAction::G4RTPrimaryGenerat 41 G4RTPrimaryGeneratorAction::G4RTPrimaryGeneratorAction()
40 { 42 {
41 G4ThreeVector zero; 43 G4ThreeVector zero;
42 particle_definition = 0; 44 particle_definition = 0;
43 particle_energy = 1.0*CLHEP::GeV; 45 particle_energy = 1.0*CLHEP::GeV;
44 particle_time = 0.0; 46 particle_time = 0.0;
45 particle_polarization = zero; 47 particle_polarization = zero;
46 48
47 pWorld = 0; 49 pWorld = 0;
48 whereisit = kInside; 50 whereisit = kInside;
49 51
50 nRow = 0; 52 nRow = 0;
51 nColumn = 0; 53 nColumn = 0;
52 54
53 eyePosition = zero; 55 eyePosition = zero;
54 eyeDirection = zero; 56 eyeDirection = zero;
55 up = G4ThreeVector(0,1,0); 57 up = G4ThreeVector(0,1,0);
56 headAngle = 0.0; 58 headAngle = 0.0;
57 viewSpan = 0.0; 59 viewSpan = 0.0;
58 stepAngle = 0.0; 60 stepAngle = 0.0;
59 viewSpanX = 0.0; 61 viewSpanX = 0.0;
60 viewSpanY = 0.0; 62 viewSpanY = 0.0;
61 63
62 distortionOn = false; 64 distortionOn = false;
63 } 65 }
64 66
65 G4RTPrimaryGeneratorAction::~G4RTPrimaryGenera 67 G4RTPrimaryGeneratorAction::~G4RTPrimaryGeneratorAction()
66 {;} 68 {;}
67 69
68 void G4RTPrimaryGeneratorAction::GeneratePrima 70 void G4RTPrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent)
69 { 71 {
70 // Note: We don't use G4ParticleGun here, as 72 // Note: We don't use G4ParticleGun here, as instantiating a G4ParticleGun
71 // object causes creation of UI commands an 73 // object causes creation of UI commands and corresponding UI messenger
72 // that interfare with normal G4ParticleGun 74 // that interfare with normal G4ParticleGun UI commands.
73 75
74 // evId = iRow * nColumn + iColumn 76 // evId = iRow * nColumn + iColumn
75 G4int evId = anEvent->GetEventID(); 77 G4int evId = anEvent->GetEventID();
76 G4int iRow = evId / nColumn; 78 G4int iRow = evId / nColumn;
77 G4int iColumn = evId % nColumn; 79 G4int iColumn = evId % nColumn;
78 G4double angleX = -(viewSpanX/2. - G4double( 80 G4double angleX = -(viewSpanX/2. - G4double(iColumn)*stepAngle);
79 G4double angleY = viewSpanY/2. - G4double(iR 81 G4double angleY = viewSpanY/2. - G4double(iRow)*stepAngle;
80 G4ThreeVector rayDirection; 82 G4ThreeVector rayDirection;
81 if(distortionOn) 83 if(distortionOn)
82 { rayDirection = G4ThreeVector(-std::tan(ang 84 { rayDirection = G4ThreeVector(-std::tan(angleX)/std::cos(angleY),std::tan(angleY)/std::cos(angleX),1.0); }
83 else 85 else
84 { rayDirection = G4ThreeVector(-std::tan(ang 86 { rayDirection = G4ThreeVector(-std::tan(angleX),std::tan(angleY),1.0); }
85 G4double cp = std::cos(eyeDirection.phi()); 87 G4double cp = std::cos(eyeDirection.phi());
86 G4double sp = std::sqrt(1.-cp*cp); 88 G4double sp = std::sqrt(1.-cp*cp);
87 G4double ct = std::cos(eyeDirection.theta()) 89 G4double ct = std::cos(eyeDirection.theta());
88 G4double st = std::sqrt(1.-ct*ct); 90 G4double st = std::sqrt(1.-ct*ct);
89 G4double gam = std::atan2(ct*cp*up.x()+ct*sp 91 G4double gam = std::atan2(ct*cp*up.x()+ct*sp*up.y()-st*up.z(), -sp*up.x()+cp*up.y());
90 rayDirection.rotateZ(-gam); 92 rayDirection.rotateZ(-gam);
91 rayDirection.rotateZ(headAngle); 93 rayDirection.rotateZ(headAngle);
92 rayDirection.rotateUz(eyeDirection); 94 rayDirection.rotateUz(eyeDirection);
93 95
94 G4ThreeVector rayPosition(eyePosition); 96 G4ThreeVector rayPosition(eyePosition);
95 if (whereisit != kInside) { 97 if (whereisit != kInside) {
96 // Eye position is outside the world, so m 98 // Eye position is outside the world, so move it inside.
97 G4double outsideDistance = pWorld->GetLogi 99 G4double outsideDistance = pWorld->GetLogicalVolume()->GetSolid()->
98 DistanceToIn(rayPosition,rayDirection); 100 DistanceToIn(rayPosition,rayDirection);
99 if(outsideDistance != kInfinity) 101 if(outsideDistance != kInfinity)
100 { rayPosition = rayPosition + (outsideDist 102 { rayPosition = rayPosition + (outsideDistance+0.001)*rayDirection; }
101 else 103 else
102 { 104 {
103 // Ray does not intercept world at all. 105 // Ray does not intercept world at all.
104 // Return without primary particle. 106 // Return without primary particle.
105 return; 107 return;
106 } 108 }
107 } 109 }
108 110
109 // create a new vertex 111 // create a new vertex
110 G4PrimaryVertex* vertex = new G4PrimaryVerte 112 G4PrimaryVertex* vertex = new G4PrimaryVertex(rayPosition,particle_time);
111 113
112 // create new primaries and set them to the 114 // create new primaries and set them to the vertex
113 G4double mass = particle_definition->GetPDGM 115 G4double mass = particle_definition->GetPDGMass();
114 G4PrimaryParticle* particle = new G4PrimaryP 116 G4PrimaryParticle* particle = new G4PrimaryParticle(particle_definition);
115 particle->SetKineticEnergy( particle_energy 117 particle->SetKineticEnergy( particle_energy );
116 particle->SetMass( mass ); 118 particle->SetMass( mass );
117 particle->SetMomentumDirection( rayDirection 119 particle->SetMomentumDirection( rayDirection.unit() );
118 particle->SetPolarization(particle_polarizat 120 particle->SetPolarization(particle_polarization.x(),
119 particle_polarizat 121 particle_polarization.y(),
120 particle_polarizat 122 particle_polarization.z());
121 vertex->SetPrimary( particle ); 123 vertex->SetPrimary( particle );
122 124
123 anEvent->AddPrimaryVertex( vertex ); 125 anEvent->AddPrimaryVertex( vertex );
124 } 126 }
125 127
126 void G4RTPrimaryGeneratorAction::SetUp() 128 void G4RTPrimaryGeneratorAction::SetUp()
127 { 129 {
128 G4ParticleTable* particleTable = G4ParticleT 130 G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable();
129 particle_definition = particleTable->FindPar 131 particle_definition = particleTable->FindParticle("geantino");
130 if(!particle_definition) 132 if(!particle_definition)
131 { 133 {
132 G4String msg; 134 G4String msg;
133 msg = " G4RayTracer uses geantino to trac 135 msg = " G4RayTracer uses geantino to trace the ray, but your physics list does not\n";
134 msg += "define G4Geantino. Please add G4Ge 136 msg += "define G4Geantino. Please add G4Geantino in your physics list.";
135 G4Exception("G4RTPrimaryGeneratorAction::S 137 G4Exception("G4RTPrimaryGeneratorAction::SetUp","VisRayTracer00101",FatalException,msg);
136 } 138 }
137 139
138 G4TheMTRayTracer* rt = G4TheMTRayTracer::the 140 G4TheMTRayTracer* rt = G4TheMTRayTracer::theInstance;
139 nRow = rt->nRow; 141 nRow = rt->nRow;
140 nColumn = rt->nColumn; 142 nColumn = rt->nColumn;
141 eyePosition = rt->eyePosition; 143 eyePosition = rt->eyePosition;
142 eyeDirection = rt->eyeDirection; 144 eyeDirection = rt->eyeDirection;
143 viewSpan = rt->viewSpan; 145 viewSpan = rt->viewSpan;
144 stepAngle = viewSpan/100.; 146 stepAngle = viewSpan/100.;
145 viewSpanX = stepAngle*nColumn; 147 viewSpanX = stepAngle*nColumn;
146 viewSpanY = stepAngle*nRow; 148 viewSpanY = stepAngle*nRow;
147 distortionOn = rt->distortionOn; 149 distortionOn = rt->distortionOn;
148 150
149 pWorld = G4TransportationManager::GetTranspo 151 pWorld = G4TransportationManager::GetTransportationManager()->
150 GetNavigatorForTracking()->GetWorldVol 152 GetNavigatorForTracking()->GetWorldVolume();
151 whereisit = pWorld->GetLogicalVolume()->GetS 153 whereisit = pWorld->GetLogicalVolume()->GetSolid()->Inside(eyePosition);
152 } 154 }
153 155
154 156