| 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 // G4RK547FEq1 implementation 26 // G4RK547FEq1 implementation
27 // 27 //
28 // The Butcher table of the Higham & Hall 5(4) 28 // The Butcher table of the Higham & Hall 5(4)7 method is:
29 // 29 //
30 // 0 | 30 // 0 |
31 // 2/9 | 2/9 31 // 2/9 | 2/9
32 // 1/3 | 1/12 1/4 32 // 1/3 | 1/12 1/4
33 // 1/2 | 1/8 0 3/8 33 // 1/2 | 1/8 0 3/8
34 // 3/5 | 91/500 -27/100 78/125 34 // 3/5 | 91/500 -27/100 78/125 8/125
35 // 1 | -11/20 27/20 12/5 35 // 1 | -11/20 27/20 12/5 -36/5 5
36 // 1 | 1/12 0 27/32 36 // 1 | 1/12 0 27/32 -4/3 125/96 5/48
37 //-------------------------------------------- 37 //----------------------------------------------------------------------------
38 // 1/12 0 27/32 38 // 1/12 0 27/32 -4/3 125/96 5/48 0
39 // 2/15 0 27/80 39 // 2/15 0 27/80 -2/15 25/48 1/24 1/10
40 // 40 //
41 // Author: Dmitry Sorokin, Google Summer of Co 41 // Author: Dmitry Sorokin, Google Summer of Code 2017
42 // Supervision: John Apostolakis, CERN 42 // Supervision: John Apostolakis, CERN
43 // ------------------------------------------- 43 // --------------------------------------------------------------------
44 44
45 #include "G4RK547FEq1.hh" 45 #include "G4RK547FEq1.hh"
46 #include "G4LineSection.hh" 46 #include "G4LineSection.hh"
47 #include "G4FieldUtils.hh" 47 #include "G4FieldUtils.hh"
48 48
49 using namespace field_utils; 49 using namespace field_utils;
50 50
51 G4RK547FEq1::G4RK547FEq1(G4EquationOfMotion* E 51 G4RK547FEq1::G4RK547FEq1(G4EquationOfMotion* EqRhs, G4int integrationVariables)
52 : G4MagIntegratorStepper(EqRhs, integrationV 52 : G4MagIntegratorStepper(EqRhs, integrationVariables)
53 { 53 {
54 } 54 }
55 55
56 void G4RK547FEq1::makeStep( const G4double yIn 56 void G4RK547FEq1::makeStep( const G4double yInput[],
57 const G4double dyd 57 const G4double dydx[],
58 const G4double hst 58 const G4double hstep,
59 G4double yOu 59 G4double yOutput[],
60 G4double* dy 60 G4double* dydxOutput,
61 G4double* yE 61 G4double* yError ) const
62 { 62 {
63 G4double yTemp[G4FieldTrack::ncompSVEC]; 63 G4double yTemp[G4FieldTrack::ncompSVEC];
64 for (G4int i=GetNumberOfVariables(); i<Get 64 for (G4int i=GetNumberOfVariables(); i<GetNumberOfStateVariables(); ++i)
65 { 65 {
66 yOutput[i] = yTemp[i] = yInput[i]; 66 yOutput[i] = yTemp[i] = yInput[i];
67 } 67 }
68 68
69 G4double ak2[G4FieldTrack::ncompSVEC], 69 G4double ak2[G4FieldTrack::ncompSVEC],
70 ak3[G4FieldTrack::ncompSVEC], 70 ak3[G4FieldTrack::ncompSVEC],
71 ak4[G4FieldTrack::ncompSVEC], 71 ak4[G4FieldTrack::ncompSVEC],
72 ak5[G4FieldTrack::ncompSVEC], 72 ak5[G4FieldTrack::ncompSVEC],
73 ak6[G4FieldTrack::ncompSVEC]; 73 ak6[G4FieldTrack::ncompSVEC];
74 74
75 const G4double b21 = 2./9., 75 const G4double b21 = 2./9.,
76 b31 = 1./12., b32 = 1./4., 76 b31 = 1./12., b32 = 1./4.,
77 b41 = 1./8., b42 = 0., b43 77 b41 = 1./8., b42 = 0., b43 = 3./8.,
78 b51 = 91./500., b52 = -27./ 78 b51 = 91./500., b52 = -27./100.,
79 b53 = 78./125., b54 = 8./ 79 b53 = 78./125., b54 = 8./125.,
80 b61 = -11./20., b62 = 27./2 80 b61 = -11./20., b62 = 27./20., b63 = 12./5.,
81 b64 = -36./5., b65 = 5., 81 b64 = -36./5., b65 = 5.,
82 b71 = 1./12., b72 = 0., 82 b71 = 1./12., b72 = 0., b73 = 27./32.,
83 b74 = -4./3., b75 = 125./ 83 b74 = -4./3., b75 = 125./96., b76 = 5./48.;
84 84
85 const G4double dc1 = b71 - 2./15., 85 const G4double dc1 = b71 - 2./15.,
86 dc2 = b72 - 0., 86 dc2 = b72 - 0.,
87 dc3 = b73 - 27./80., 87 dc3 = b73 - 27./80.,
88 dc4 = b74 + 2./15., 88 dc4 = b74 + 2./15.,
89 dc5 = b75 - 25./48., 89 dc5 = b75 - 25./48.,
90 dc6 = b76 - 1./24., 90 dc6 = b76 - 1./24.,
91 dc7 = 0. - 1./10.; 91 dc7 = 0. - 1./10.;
92 92
93 // RightHandSide(yInput, dydx); 93 // RightHandSide(yInput, dydx);
94 for(G4int i = 0; i < GetNumberOfVariables( 94 for(G4int i = 0; i < GetNumberOfVariables(); ++i)
95 { << 95 yTemp[i] = yInput[i] + hstep * b21 * dydx[i];
96 yTemp[i] = yInput[i] + hstep * b21 * dyd <<
97 } <<
98 96
99 RightHandSide(yTemp, ak2); 97 RightHandSide(yTemp, ak2);
100 for(G4int i = 0; i < GetNumberOfVariables( 98 for(G4int i = 0; i < GetNumberOfVariables(); ++i)
101 { << 99 yTemp[i] = yInput[i] + hstep * (b31 * dydx[i] + b32 * ak2[i]);
102 yTemp[i] = yInput[i] + hstep * (b31 * dy <<
103 } <<
104 100
105 RightHandSide(yTemp, ak3); 101 RightHandSide(yTemp, ak3);
106 for(G4int i = 0;i < GetNumberOfVariables() 102 for(G4int i = 0;i < GetNumberOfVariables(); ++i)
107 { << 103 yTemp[i] = yInput[i] + hstep * (b41 * dydx[i] + b42 * ak2[i] +
108 yTemp[i] = yInput[i] + hstep * (b41 * dy << 104 b43 * ak3[i]);
109 b43 * ak <<
110 } <<
111 105
112 RightHandSide(yTemp, ak4); 106 RightHandSide(yTemp, ak4);
113 for(G4int i = 0; i < GetNumberOfVariables( 107 for(G4int i = 0; i < GetNumberOfVariables(); ++i)
114 { << 108 yTemp[i] = yInput[i] + hstep * (b51 * dydx[i] + b52 * ak2[i] +
115 yTemp[i] = yInput[i] + hstep * (b51 * dy << 109 b53 * ak3[i] + b54 * ak4[i]);
116 b53 * ak <<
117 } <<
118 110
119 RightHandSide(yTemp, ak5); 111 RightHandSide(yTemp, ak5);
120 for(G4int i = 0; i < GetNumberOfVariables( 112 for(G4int i = 0; i < GetNumberOfVariables(); ++i)
121 { << 113 yTemp[i] = yInput[i] + hstep * (b61 * dydx[i] + b62 * ak2[i] +
122 yTemp[i] = yInput[i] + hstep * (b61 * dy << 114 b63 * ak3[i] + b64 * ak4[i] +
123 b63 * ak << 115 b65 * ak5[i]);
124 b65 * ak <<
125 } <<
126 116
127 RightHandSide(yTemp, ak6); 117 RightHandSide(yTemp, ak6);
128 for(G4int i = 0; i < GetNumberOfVariables( 118 for(G4int i = 0; i < GetNumberOfVariables(); ++i)
129 { << 119 yOutput[i] = yInput[i] + hstep * (b71 * dydx[i] + b72 * ak2[i] +
130 yOutput[i] = yInput[i] + hstep * (b71 * << 120 b73 * ak3[i] + b74 * ak4[i] +
131 b73 * << 121 b75 * ak5[i] + b76 * ak6[i]);
132 b75 * << 122 if (dydxOutput && yError)
133 } <<
134 if ((dydxOutput != nullptr) && (yError != <<
135 { 123 {
136 RightHandSide(yOutput, dydxOutput); 124 RightHandSide(yOutput, dydxOutput);
137 for(G4int i = 0; i < GetNumberOfVariab 125 for(G4int i = 0; i < GetNumberOfVariables(); ++i)
138 { << 126 yError[i] = hstep * (dc1 * dydx[i] + dc2 * ak2[i] + dc3 * ak3[i] +
139 yError[i] = hstep * (dc1 * dydx[i] + << 127 dc4 * ak4[i] + dc5 * ak5[i] + dc6 * ak6[i] +
140 dc4 * ak4[i] + << 128 dc7 * dydxOutput[i]);
141 dc7 * dydxOutpu <<
142 } <<
143 } 129 }
144 } 130 }
145 131
146 void G4RK547FEq1::Stepper( const G4double yInp 132 void G4RK547FEq1::Stepper( const G4double yInput[],
147 const G4double dydx 133 const G4double dydx[],
148 G4double hste 134 G4double hstep,
149 G4double yOut 135 G4double yOutput[],
150 G4double yErr 136 G4double yError[] )
151 { 137 {
152 copy(fyIn, yInput); 138 copy(fyIn, yInput);
153 copy(fdydx, dydx); 139 copy(fdydx, dydx);
154 fhstep = hstep; 140 fhstep = hstep;
155 141
156 makeStep(fyIn, fdydx, fhstep, fyOut, fdydx 142 makeStep(fyIn, fdydx, fhstep, fyOut, fdydxOut, yError);
157 143
158 copy(yOutput, fyOut); 144 copy(yOutput, fyOut);
159 } 145 }
160 146
161 void G4RK547FEq1::Stepper( const G4double yInp 147 void G4RK547FEq1::Stepper( const G4double yInput[],
162 const G4double dydx 148 const G4double dydx[],
163 G4double hste 149 G4double hstep,
164 G4double yOut 150 G4double yOutput[],
165 G4double yErr 151 G4double yError[],
166 G4double dydx 152 G4double dydxOutput[] )
167 { 153 {
168 copy(fyIn, yInput); 154 copy(fyIn, yInput);
169 copy(fdydx, dydx); 155 copy(fdydx, dydx);
170 fhstep = hstep; 156 fhstep = hstep;
171 157
172 makeStep(fyIn,fdydx, fhstep, fyOut, fdydxO 158 makeStep(fyIn,fdydx, fhstep, fyOut, fdydxOut, yError);
173 159
174 copy(yOutput, fyOut); 160 copy(yOutput, fyOut);
175 copy(dydxOutput, fdydxOut); 161 copy(dydxOutput, fdydxOut);
176 } 162 }
177 163
178 G4double G4RK547FEq1::DistChord() const 164 G4double G4RK547FEq1::DistChord() const
179 { 165 {
180 G4double yMid[G4FieldTrack::ncompSVEC]; 166 G4double yMid[G4FieldTrack::ncompSVEC];
181 makeStep(fyIn, fdydx, fhstep / 2., yMid); 167 makeStep(fyIn, fdydx, fhstep / 2., yMid);
182 168
183 const G4ThreeVector begin = makeVector(fyI 169 const G4ThreeVector begin = makeVector(fyIn, Value3D::Position);
184 const G4ThreeVector mid = makeVector(yMid, 170 const G4ThreeVector mid = makeVector(yMid, Value3D::Position);
185 const G4ThreeVector end = makeVector(fyOut 171 const G4ThreeVector end = makeVector(fyOut, Value3D::Position);
186 172
187 return G4LineSection::Distline(mid, begin, 173 return G4LineSection::Distline(mid, begin, end);
188 } 174 }
189 175