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