Krita Source Code Documentation
Loading...
Searching...
No Matches
Functions
multi_bspline_eval_z.h File Reference

Go to the source code of this file.

Functions

void eval_multi_UBspline_1d_z (multi_UBspline_1d_z *spline, double x, complex_double *restrict vals)
 
void eval_multi_UBspline_1d_z_vg (multi_UBspline_1d_z *spline, double x, complex_double *restrict vals, complex_double *restrict grads)
 
void eval_multi_UBspline_1d_z_vgh (multi_UBspline_1d_z *spline, double x, complex_double *restrict vals, complex_double *restrict grads, complex_double *restrict hess)
 
void eval_multi_UBspline_1d_z_vgl (multi_UBspline_1d_z *spline, double x, complex_double *restrict vals, complex_double *restrict grads, complex_double *restrict lapl)
 
void eval_multi_UBspline_2d_z (multi_UBspline_2d_z *spline, double x, double y, complex_double *restrict vals)
 
void eval_multi_UBspline_2d_z_vg (multi_UBspline_2d_z *spline, double x, double y, complex_double *restrict vals, complex_double *restrict grads)
 
void eval_multi_UBspline_2d_z_vgh (multi_UBspline_2d_z *spline, double x, double y, complex_double *restrict vals, complex_double *restrict grads, complex_double *restrict hess)
 
void eval_multi_UBspline_2d_z_vgl (multi_UBspline_2d_z *spline, double x, double y, complex_double *restrict vals, complex_double *restrict grads, complex_double *restrict lapl)
 
void eval_multi_UBspline_3d_z (multi_UBspline_3d_z *spline, double x, double y, double z, complex_double *restrict vals)
 
void eval_multi_UBspline_3d_z_vg (multi_UBspline_3d_z *spline, double x, double y, double z, complex_double *restrict vals, complex_double *restrict grads)
 
void eval_multi_UBspline_3d_z_vgh (multi_UBspline_3d_z *spline, double x, double y, double z, complex_double *restrict vals, complex_double *restrict grads, complex_double *restrict hess)
 
void eval_multi_UBspline_3d_z_vgl (multi_UBspline_3d_z *spline, double x, double y, double z, complex_double *restrict vals, complex_double *restrict grads, complex_double *restrict lapl)
 

Function Documentation

◆ eval_multi_UBspline_1d_z()

void eval_multi_UBspline_1d_z ( multi_UBspline_1d_z * spline,
double x,
complex_double *restrict vals )

Definition at line 38 of file multi_bspline_eval_std_z.h.

41{
42 x -= spline->x_grid.start;
43 double ux = x*spline->x_grid.delta_inv;
44 double ipartx, tx;
45 tx = modf (ux, &ipartx); int ix = (int) ipartx;
46
47 double tpx[4], a[4];
48 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
49 complex_double* restrict coefs = spline->coefs;
50
51 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
52 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
53 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
54 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
55
56 intptr_t xs = spline->x_stride;
57
58 for (int n=0; n<spline->num_splines; n++)
59 vals[n] = 0.0;
60
61 for (int i=0; i<4; i++) {
62 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs);
63 for (int n=0; n<spline->num_splines; n++)
64 vals[n] += a[i] * coefs[n];
65 }
66}
complex_double
complex double complex_double
Definition bspline_base.h:34
restrict
#define restrict
Definition local_definitions.h:4
Ad
const double *restrict Ad
Definition bspline_data.cpp:181
resize_to_all_layers.x
x
Definition resize_to_all_layers.py:18
Ugrid::start
double start
Definition bspline_base.h:85
Ugrid::delta_inv
double delta_inv
Definition bspline_base.h:89
multi_UBspline_1d_z::coefs
complex_double *restrict coefs
Definition multi_bspline_structs.h:148
multi_UBspline_1d_z::x_stride
intptr_t x_stride
Definition multi_bspline_structs.h:149
multi_UBspline_1d_z::x_grid
Ugrid x_grid
Definition multi_bspline_structs.h:150
multi_UBspline_1d_z::num_splines
int num_splines
Definition multi_bspline_structs.h:152

References Ad, multi_UBspline_1d_z::coefs, Ugrid::delta_inv, multi_UBspline_1d_z::num_splines, restrict, Ugrid::start, multi_UBspline_1d_z::x_grid, and multi_UBspline_1d_z::x_stride.

◆ eval_multi_UBspline_1d_z_vg()

void eval_multi_UBspline_1d_z_vg ( multi_UBspline_1d_z * spline,
double x,
complex_double *restrict vals,
complex_double *restrict grads )

Definition at line 71 of file multi_bspline_eval_std_z.h.

75{
76 x -= spline->x_grid.start;
77 double ux = x*spline->x_grid.delta_inv;
78 double ipartx, tx;
79 tx = modf (ux, &ipartx); int ix = (int) ipartx;
80
81 double tpx[4], a[4], da[4];
82 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
83 complex_double* restrict coefs = spline->coefs;
84
85 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
86 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
87 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
88 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
89 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
90 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
91 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
92 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
93
94 intptr_t xs = spline->x_stride;
95
96 for (int n=0; n<spline->num_splines; n++) {
97 vals[n] = 0.0;
98 grads[n] = 0.0;
99 }
100
101 for (int i=0; i<4; i++) {
102 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs);
103 for (int n=0; n<spline->num_splines; n++) {
104 vals[n] += a[i] * coefs[n];
105 grads[n] += da[i] * coefs[n];
106 }
107 }
108
109 double dxInv = spline->x_grid.delta_inv;
110 for (int n=0; n<spline->num_splines; n++)
111 grads[n] *= dxInv;
112}
dAd
const double *restrict dAd
Definition bspline_data.cpp:188

References Ad, multi_UBspline_1d_z::coefs, dAd, Ugrid::delta_inv, multi_UBspline_1d_z::num_splines, restrict, Ugrid::start, multi_UBspline_1d_z::x_grid, and multi_UBspline_1d_z::x_stride.

◆ eval_multi_UBspline_1d_z_vgh()

void eval_multi_UBspline_1d_z_vgh ( multi_UBspline_1d_z * spline,
double x,
complex_double *restrict vals,
complex_double *restrict grads,
complex_double *restrict hess )

Definition at line 170 of file multi_bspline_eval_std_z.h.

175{
176 eval_multi_UBspline_1d_z_vgl (spline, x, vals, grads, hess);
177}
eval_multi_UBspline_1d_z_vgl
void eval_multi_UBspline_1d_z_vgl(multi_UBspline_1d_z *spline, double x, complex_double *restrict vals, complex_double *restrict grads, complex_double *restrict lapl)
Definition multi_bspline_eval_std_z.h:116

References eval_multi_UBspline_1d_z_vgl().

◆ eval_multi_UBspline_1d_z_vgl()

void eval_multi_UBspline_1d_z_vgl ( multi_UBspline_1d_z * spline,
double x,
complex_double *restrict vals,
complex_double *restrict grads,
complex_double *restrict lapl )

Definition at line 116 of file multi_bspline_eval_std_z.h.

121{
122 x -= spline->x_grid.start;
123 double ux = x*spline->x_grid.delta_inv;
124 double ipartx, tx;
125 tx = modf (ux, &ipartx); int ix = (int) ipartx;
126
127 double tpx[4], a[4], da[4], d2a[4];
128 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
129 complex_double* restrict coefs = spline->coefs;
130
131 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
132 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
133 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
134 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
135 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
136 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
137 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
138 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
139 d2a[0] = (d2Ad[ 0]*tpx[0] + d2Ad[ 1]*tpx[1] + d2Ad[ 2]*tpx[2] + d2Ad[ 3]*tpx[3]);
140 d2a[1] = (d2Ad[ 4]*tpx[0] + d2Ad[ 5]*tpx[1] + d2Ad[ 6]*tpx[2] + d2Ad[ 7]*tpx[3]);
141 d2a[2] = (d2Ad[ 8]*tpx[0] + d2Ad[ 9]*tpx[1] + d2Ad[10]*tpx[2] + d2Ad[11]*tpx[3]);
142 d2a[3] = (d2Ad[12]*tpx[0] + d2Ad[13]*tpx[1] + d2Ad[14]*tpx[2] + d2Ad[15]*tpx[3]);
143
144 intptr_t xs = spline->x_stride;
145
146 for (int n=0; n<spline->num_splines; n++) {
147 vals[n] = 0.0;
148 grads[n] = 0.0;
149 lapl[n] = 0.0;
150 }
151
152 for (int i=0; i<4; i++) {
153 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs);
154 for (int n=0; n<spline->num_splines; n++) {
155 vals[n] += a[i] * coefs[n];
156 grads[n] += da[i] * coefs[n];
157 lapl[n] += d2a[i] * coefs[n];
158 }
159 }
160
161 double dxInv = spline->x_grid.delta_inv;
162 for (int n=0; n<spline->num_splines; n++) {
163 grads[n] *= dxInv;
164 lapl [n] *= dxInv*dxInv;
165 }
166}
d2Ad
const double *restrict d2Ad
Definition bspline_data.cpp:195

References Ad, multi_UBspline_1d_z::coefs, d2Ad, dAd, Ugrid::delta_inv, multi_UBspline_1d_z::num_splines, restrict, Ugrid::start, multi_UBspline_1d_z::x_grid, and multi_UBspline_1d_z::x_stride.

◆ eval_multi_UBspline_2d_z()

void eval_multi_UBspline_2d_z ( multi_UBspline_2d_z * spline,
double x,
double y,
complex_double *restrict vals )

Definition at line 184 of file multi_bspline_eval_std_z.h.

187{
188 x -= spline->x_grid.start;
189 y -= spline->y_grid.start;
190 double ux = x*spline->x_grid.delta_inv;
191 double uy = y*spline->y_grid.delta_inv;
192 double ipartx, iparty, tx, ty;
193 tx = modf (ux, &ipartx); int ix = (int) ipartx;
194 ty = modf (uy, &iparty); int iy = (int) iparty;
195
196 double tpx[4], tpy[4], a[4], b[4];
197 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
198 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
199 complex_double* restrict coefs = spline->coefs;
200
201 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
202 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
203 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
204 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
205
206 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
207 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
208 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
209 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
210
211 intptr_t xs = spline->x_stride;
212 intptr_t ys = spline->y_stride;
213
214 for (int n=0; n<spline->num_splines; n++)
215 vals[n] = 0.0;
216
217 for (int i=0; i<4; i++)
218 for (int j=0; j<4; j++) {
219 double prefactor = a[i]*b[j];
220 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys);
221 for (int n=0; n<spline->num_splines; n++)
222 vals[n] += prefactor*coefs[n];
223 }
224}
resize_to_all_layers.y
y
Definition resize_to_all_layers.py:19
resize_to_all_layers.b
b
Definition resize_to_all_layers.py:26
multi_UBspline_2d_z::x_stride
intptr_t x_stride
Definition multi_bspline_structs.h:160
multi_UBspline_2d_z::x_grid
Ugrid x_grid
Definition multi_bspline_structs.h:161
multi_UBspline_2d_z::y_stride
intptr_t y_stride
Definition multi_bspline_structs.h:160
multi_UBspline_2d_z::coefs
complex_double *restrict coefs
Definition multi_bspline_structs.h:159
multi_UBspline_2d_z::num_splines
int num_splines
Definition multi_bspline_structs.h:163
multi_UBspline_2d_z::y_grid
Ugrid y_grid
Definition multi_bspline_structs.h:161

References Ad, multi_UBspline_2d_z::coefs, Ugrid::delta_inv, multi_UBspline_2d_z::num_splines, restrict, Ugrid::start, multi_UBspline_2d_z::x_grid, multi_UBspline_2d_z::x_stride, multi_UBspline_2d_z::y_grid, and multi_UBspline_2d_z::y_stride.

◆ eval_multi_UBspline_2d_z_vg()

void eval_multi_UBspline_2d_z_vg ( multi_UBspline_2d_z * spline,
double x,
double y,
complex_double *restrict vals,
complex_double *restrict grads )

Definition at line 228 of file multi_bspline_eval_std_z.h.

232{
233 x -= spline->x_grid.start;
234 y -= spline->y_grid.start;
235 double ux = x*spline->x_grid.delta_inv;
236 double uy = y*spline->y_grid.delta_inv;
237 double ipartx, iparty, tx, ty;
238 tx = modf (ux, &ipartx); int ix = (int) ipartx;
239 ty = modf (uy, &iparty); int iy = (int) iparty;
240
241 double tpx[4], tpy[4], a[4], b[4], da[4], db[4];
242 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
243 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
244 complex_double* restrict coefs = spline->coefs;
245
246 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
247 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
248 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
249 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
250 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
251 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
252 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
253 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
254
255 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
256 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
257 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
258 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
259 db[0] = (dAd[ 0]*tpy[0] + dAd[ 1]*tpy[1] + dAd[ 2]*tpy[2] + dAd[ 3]*tpy[3]);
260 db[1] = (dAd[ 4]*tpy[0] + dAd[ 5]*tpy[1] + dAd[ 6]*tpy[2] + dAd[ 7]*tpy[3]);
261 db[2] = (dAd[ 8]*tpy[0] + dAd[ 9]*tpy[1] + dAd[10]*tpy[2] + dAd[11]*tpy[3]);
262 db[3] = (dAd[12]*tpy[0] + dAd[13]*tpy[1] + dAd[14]*tpy[2] + dAd[15]*tpy[3]);
263
264 intptr_t xs = spline->x_stride;
265 intptr_t ys = spline->y_stride;
266
267 for (int n=0; n<spline->num_splines; n++) {
268 vals[n] = 0.0;
269 grads[2*n+0] = grads[2*n+1] = grads[2*n+2] = 0.0;
270 }
271
272 for (int i=0; i<4; i++)
273 for (int j=0; j<4; j++) {
274 double ab = a[i]*b[j];
275 double dab[2];
276 dab[0] = da[i]* b[j];
277 dab[1] = a[i]*db[j];
278
279 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys);
280 for (int n=0; n<spline->num_splines; n++) {
281 vals [n] += ab *coefs[n];
282 grads[2*n+0] += dab[0]*coefs[n];
283 grads[2*n+1] += dab[1]*coefs[n];
284 }
285 }
286
287 double dxInv = spline->x_grid.delta_inv;
288 double dyInv = spline->y_grid.delta_inv;
289 for (int n=0; n<spline->num_splines; n++) {
290 grads[2*n+0] *= dxInv;
291 grads[2*n+1] *= dyInv;
292 }
293}

References Ad, multi_UBspline_2d_z::coefs, dAd, Ugrid::delta_inv, multi_UBspline_2d_z::num_splines, restrict, Ugrid::start, multi_UBspline_2d_z::x_grid, multi_UBspline_2d_z::x_stride, multi_UBspline_2d_z::y_grid, and multi_UBspline_2d_z::y_stride.

◆ eval_multi_UBspline_2d_z_vgh()

void eval_multi_UBspline_2d_z_vgh ( multi_UBspline_2d_z * spline,
double x,
double y,
complex_double *restrict vals,
complex_double *restrict grads,
complex_double *restrict hess )

Definition at line 386 of file multi_bspline_eval_std_z.h.

391{
392 x -= spline->x_grid.start;
393 y -= spline->y_grid.start;
394 double ux = x*spline->x_grid.delta_inv;
395 double uy = y*spline->y_grid.delta_inv;
396 double ipartx, iparty, tx, ty;
397 tx = modf (ux, &ipartx); int ix = (int) ipartx;
398 ty = modf (uy, &iparty); int iy = (int) iparty;
399
400 double tpx[4], tpy[4], a[4], b[4], da[4], db[4], d2a[4], d2b[4];
401 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
402 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
403 complex_double* restrict coefs = spline->coefs;
404
405 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
406 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
407 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
408 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
409 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
410 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
411 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
412 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
413 d2a[0] = (d2Ad[ 0]*tpx[0] + d2Ad[ 1]*tpx[1] + d2Ad[ 2]*tpx[2] + d2Ad[ 3]*tpx[3]);
414 d2a[1] = (d2Ad[ 4]*tpx[0] + d2Ad[ 5]*tpx[1] + d2Ad[ 6]*tpx[2] + d2Ad[ 7]*tpx[3]);
415 d2a[2] = (d2Ad[ 8]*tpx[0] + d2Ad[ 9]*tpx[1] + d2Ad[10]*tpx[2] + d2Ad[11]*tpx[3]);
416 d2a[3] = (d2Ad[12]*tpx[0] + d2Ad[13]*tpx[1] + d2Ad[14]*tpx[2] + d2Ad[15]*tpx[3]);
417
418 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
419 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
420 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
421 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
422 db[0] = (dAd[ 0]*tpy[0] + dAd[ 1]*tpy[1] + dAd[ 2]*tpy[2] + dAd[ 3]*tpy[3]);
423 db[1] = (dAd[ 4]*tpy[0] + dAd[ 5]*tpy[1] + dAd[ 6]*tpy[2] + dAd[ 7]*tpy[3]);
424 db[2] = (dAd[ 8]*tpy[0] + dAd[ 9]*tpy[1] + dAd[10]*tpy[2] + dAd[11]*tpy[3]);
425 db[3] = (dAd[12]*tpy[0] + dAd[13]*tpy[1] + dAd[14]*tpy[2] + dAd[15]*tpy[3]);
426 d2b[0] = (d2Ad[ 0]*tpy[0] + d2Ad[ 1]*tpy[1] + d2Ad[ 2]*tpy[2] + d2Ad[ 3]*tpy[3]);
427 d2b[1] = (d2Ad[ 4]*tpy[0] + d2Ad[ 5]*tpy[1] + d2Ad[ 6]*tpy[2] + d2Ad[ 7]*tpy[3]);
428 d2b[2] = (d2Ad[ 8]*tpy[0] + d2Ad[ 9]*tpy[1] + d2Ad[10]*tpy[2] + d2Ad[11]*tpy[3]);
429 d2b[3] = (d2Ad[12]*tpy[0] + d2Ad[13]*tpy[1] + d2Ad[14]*tpy[2] + d2Ad[15]*tpy[3]);
430
431 intptr_t xs = spline->x_stride;
432 intptr_t ys = spline->y_stride;
433
434 for (int n=0; n<spline->num_splines; n++) {
435 vals[n] = 0.0;
436 grads[2*n+0] = grads[2*n+1] = 0.0;
437 for (int i=0; i<4; i++)
438 hess[4*n+i] = 0.0;
439 }
440
441 for (int i=0; i<4; i++)
442 for (int j=0; j<4; j++){
443 double ab = a[i]*b[j];
444 double dab[2], d2ab[3];
445 dab[0] = da[i]* b[j];
446 dab[1] = a[i]*db[j];
447 d2ab[0] = d2a[i] * b[j];
448 d2ab[1] = da[i] * db[j];
449 d2ab[2] = a[i] * d2b[j];
450
451 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys);
452 for (int n=0; n<spline->num_splines; n++) {
453 vals[n] += ab *coefs[n];
454 grads[2*n+0] += dab[0]*coefs[n];
455 grads[2*n+1] += dab[1]*coefs[n];
456 hess [4*n+0] += d2ab[0]*coefs[n];
457 hess [4*n+1] += d2ab[1]*coefs[n];
458 hess [4*n+3] += d2ab[2]*coefs[n];
459 }
460 }
461
462 double dxInv = spline->x_grid.delta_inv;
463 double dyInv = spline->y_grid.delta_inv;
464 for (int n=0; n<spline->num_splines; n++) {
465 grads[2*n+0] *= dxInv;
466 grads[2*n+1] *= dyInv;
467 hess[4*n+0] *= dxInv*dxInv;
468 hess[4*n+1] *= dxInv*dyInv;
469 hess[4*n+3] *= dyInv*dyInv;
470 // Copy hessian elements into lower half of 3x3 matrix
471 hess[4*n+2] = hess[4*n+1];
472 }
473}

References Ad, multi_UBspline_2d_z::coefs, d2Ad, dAd, Ugrid::delta_inv, multi_UBspline_2d_z::num_splines, restrict, Ugrid::start, multi_UBspline_2d_z::x_grid, multi_UBspline_2d_z::x_stride, multi_UBspline_2d_z::y_grid, and multi_UBspline_2d_z::y_stride.

◆ eval_multi_UBspline_2d_z_vgl()

void eval_multi_UBspline_2d_z_vgl ( multi_UBspline_2d_z * spline,
double x,
double y,
complex_double *restrict vals,
complex_double *restrict grads,
complex_double *restrict lapl )

Definition at line 296 of file multi_bspline_eval_std_z.h.

301{
302 x -= spline->x_grid.start;
303 y -= spline->y_grid.start;
304 double ux = x*spline->x_grid.delta_inv;
305 double uy = y*spline->y_grid.delta_inv;
306 double ipartx, iparty, tx, ty;
307 tx = modf (ux, &ipartx); int ix = (int) ipartx;
308 ty = modf (uy, &iparty); int iy = (int) iparty;
309
310 double tpx[4], tpy[4], a[4], b[4], da[4], db[4], d2a[4], d2b[4];
311 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
312 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
313 complex_double* restrict coefs = spline->coefs;
314
315 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
316 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
317 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
318 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
319 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
320 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
321 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
322 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
323 d2a[0] = (d2Ad[ 0]*tpx[0] + d2Ad[ 1]*tpx[1] + d2Ad[ 2]*tpx[2] + d2Ad[ 3]*tpx[3]);
324 d2a[1] = (d2Ad[ 4]*tpx[0] + d2Ad[ 5]*tpx[1] + d2Ad[ 6]*tpx[2] + d2Ad[ 7]*tpx[3]);
325 d2a[2] = (d2Ad[ 8]*tpx[0] + d2Ad[ 9]*tpx[1] + d2Ad[10]*tpx[2] + d2Ad[11]*tpx[3]);
326 d2a[3] = (d2Ad[12]*tpx[0] + d2Ad[13]*tpx[1] + d2Ad[14]*tpx[2] + d2Ad[15]*tpx[3]);
327
328 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
329 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
330 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
331 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
332 db[0] = (dAd[ 0]*tpy[0] + dAd[ 1]*tpy[1] + dAd[ 2]*tpy[2] + dAd[ 3]*tpy[3]);
333 db[1] = (dAd[ 4]*tpy[0] + dAd[ 5]*tpy[1] + dAd[ 6]*tpy[2] + dAd[ 7]*tpy[3]);
334 db[2] = (dAd[ 8]*tpy[0] + dAd[ 9]*tpy[1] + dAd[10]*tpy[2] + dAd[11]*tpy[3]);
335 db[3] = (dAd[12]*tpy[0] + dAd[13]*tpy[1] + dAd[14]*tpy[2] + dAd[15]*tpy[3]);
336 d2b[0] = (d2Ad[ 0]*tpy[0] + d2Ad[ 1]*tpy[1] + d2Ad[ 2]*tpy[2] + d2Ad[ 3]*tpy[3]);
337 d2b[1] = (d2Ad[ 4]*tpy[0] + d2Ad[ 5]*tpy[1] + d2Ad[ 6]*tpy[2] + d2Ad[ 7]*tpy[3]);
338 d2b[2] = (d2Ad[ 8]*tpy[0] + d2Ad[ 9]*tpy[1] + d2Ad[10]*tpy[2] + d2Ad[11]*tpy[3]);
339 d2b[3] = (d2Ad[12]*tpy[0] + d2Ad[13]*tpy[1] + d2Ad[14]*tpy[2] + d2Ad[15]*tpy[3]);
340
341 intptr_t xs = spline->x_stride;
342 intptr_t ys = spline->y_stride;
343
344 //complex_double lapl2[2*spline->num_splines];
345 complex_double* restrict lapl2 = spline->lapl2;
346 for (int n=0; n<spline->num_splines; n++) {
347 vals[n] = 0.0;
348 grads[2*n+0] = grads[2*n+1] = 0.0;
349 lapl2[2*n+0] = lapl2[2*n+1] = 0.0;
350 }
351
352 for (int i=0; i<4; i++)
353 for (int j=0; j<4; j++) {
354 double ab = a[i]*b[j];
355 double dab[2], d2ab[2];
356 dab[0] = da[i]* b[j];
357 dab[1] = a[i]*db[j];
358 d2ab[0] = d2a[i]* b[j];
359 d2ab[1] = a[i]*d2b[j];
360
361 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys);
362 for (int n=0; n<spline->num_splines; n++) {
363 vals[n] += ab *coefs[n];
364 grads[2*n+0] += dab[0]*coefs[n];
365 grads[2*n+1] += dab[1]*coefs[n];
366 lapl2[2*n+0] += d2ab[0]*coefs[n];
367 lapl2[2*n+1] += d2ab[1]*coefs[n];
368 }
369 }
370
371 double dxInv = spline->x_grid.delta_inv;
372 double dyInv = spline->y_grid.delta_inv;
373 for (int n=0; n<spline->num_splines; n++) {
374 grads[2*n+0] *= dxInv;
375 grads[2*n+1] *= dyInv;
376 lapl2[2*n+0] *= dxInv*dxInv;
377 lapl2[2*n+1] *= dyInv*dyInv;
378 lapl[n] = lapl2[2*n+0] + lapl2[2*n+1];
379 }
380}
multi_UBspline_2d_z::lapl2
complex_double *restrict lapl2
Definition multi_bspline_structs.h:165

References Ad, multi_UBspline_2d_z::coefs, d2Ad, dAd, Ugrid::delta_inv, multi_UBspline_2d_z::lapl2, multi_UBspline_2d_z::num_splines, restrict, Ugrid::start, multi_UBspline_2d_z::x_grid, multi_UBspline_2d_z::x_stride, multi_UBspline_2d_z::y_grid, and multi_UBspline_2d_z::y_stride.

◆ eval_multi_UBspline_3d_z()

void eval_multi_UBspline_3d_z ( multi_UBspline_3d_z * spline,
double x,
double y,
double z,
complex_double *restrict vals )

Definition at line 481 of file multi_bspline_eval_std_z.h.

484{
485 x -= spline->x_grid.start;
486 y -= spline->y_grid.start;
487 z -= spline->z_grid.start;
488 double ux = x*spline->x_grid.delta_inv;
489 double uy = y*spline->y_grid.delta_inv;
490 double uz = z*spline->z_grid.delta_inv;
491 double ipartx, iparty, ipartz, tx, ty, tz;
492 tx = modf (ux, &ipartx); int ix = (int) ipartx;
493 ty = modf (uy, &iparty); int iy = (int) iparty;
494 tz = modf (uz, &ipartz); int iz = (int) ipartz;
495
496 double tpx[4], tpy[4], tpz[4], a[4], b[4], c[4];
497 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
498 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
499 tpz[0] = tz*tz*tz; tpz[1] = tz*tz; tpz[2] = tz; tpz[3] = 1.0;
500 complex_double* restrict coefs = spline->coefs;
501
502 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
503 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
504 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
505 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
506
507 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
508 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
509 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
510 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
511
512 c[0] = (Ad[ 0]*tpz[0] + Ad[ 1]*tpz[1] + Ad[ 2]*tpz[2] + Ad[ 3]*tpz[3]);
513 c[1] = (Ad[ 4]*tpz[0] + Ad[ 5]*tpz[1] + Ad[ 6]*tpz[2] + Ad[ 7]*tpz[3]);
514 c[2] = (Ad[ 8]*tpz[0] + Ad[ 9]*tpz[1] + Ad[10]*tpz[2] + Ad[11]*tpz[3]);
515 c[3] = (Ad[12]*tpz[0] + Ad[13]*tpz[1] + Ad[14]*tpz[2] + Ad[15]*tpz[3]);
516
517 intptr_t xs = spline->x_stride;
518 intptr_t ys = spline->y_stride;
519 intptr_t zs = spline->z_stride;
520
521 for (int n=0; n<spline->num_splines; n++)
522 vals[n] = 0.0;
523
524 for (int i=0; i<4; i++)
525 for (int j=0; j<4; j++)
526 for (int k=0; k<4; k++) {
527 double prefactor = a[i]*b[j]*c[k];
528 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys + (iz+k)*zs);
529 for (int n=0; n<spline->num_splines; n++)
530 vals[n] += prefactor*coefs[n];
531 }
532}
multi_UBspline_3d_z::coefs
complex_double *restrict coefs
Definition multi_bspline_structs.h:172
multi_UBspline_3d_z::num_splines
int num_splines
Definition multi_bspline_structs.h:176
multi_UBspline_3d_z::x_grid
Ugrid x_grid
Definition multi_bspline_structs.h:174
multi_UBspline_3d_z::x_stride
intptr_t x_stride
Definition multi_bspline_structs.h:173
multi_UBspline_3d_z::z_stride
intptr_t z_stride
Definition multi_bspline_structs.h:173
multi_UBspline_3d_z::y_grid
Ugrid y_grid
Definition multi_bspline_structs.h:174
multi_UBspline_3d_z::z_grid
Ugrid z_grid
Definition multi_bspline_structs.h:174
multi_UBspline_3d_z::y_stride
intptr_t y_stride
Definition multi_bspline_structs.h:173

References Ad, multi_UBspline_3d_z::coefs, Ugrid::delta_inv, multi_UBspline_3d_z::num_splines, restrict, Ugrid::start, multi_UBspline_3d_z::x_grid, multi_UBspline_3d_z::x_stride, multi_UBspline_3d_z::y_grid, multi_UBspline_3d_z::y_stride, multi_UBspline_3d_z::z_grid, and multi_UBspline_3d_z::z_stride.

◆ eval_multi_UBspline_3d_z_vg()

void eval_multi_UBspline_3d_z_vg ( multi_UBspline_3d_z * spline,
double x,
double y,
double z,
complex_double *restrict vals,
complex_double *restrict grads )

Definition at line 536 of file multi_bspline_eval_std_z.h.

540{
541 x -= spline->x_grid.start;
542 y -= spline->y_grid.start;
543 z -= spline->z_grid.start;
544 double ux = x*spline->x_grid.delta_inv;
545 double uy = y*spline->y_grid.delta_inv;
546 double uz = z*spline->z_grid.delta_inv;
547 double ipartx, iparty, ipartz, tx, ty, tz;
548 tx = modf (ux, &ipartx); int ix = (int) ipartx;
549 ty = modf (uy, &iparty); int iy = (int) iparty;
550 tz = modf (uz, &ipartz); int iz = (int) ipartz;
551
552 double tpx[4], tpy[4], tpz[4], a[4], b[4], c[4],
553 da[4], db[4], dc[4];
554 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
555 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
556 tpz[0] = tz*tz*tz; tpz[1] = tz*tz; tpz[2] = tz; tpz[3] = 1.0;
557 complex_double* restrict coefs = spline->coefs;
558
559 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
560 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
561 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
562 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
563 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
564 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
565 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
566 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
567
568 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
569 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
570 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
571 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
572 db[0] = (dAd[ 0]*tpy[0] + dAd[ 1]*tpy[1] + dAd[ 2]*tpy[2] + dAd[ 3]*tpy[3]);
573 db[1] = (dAd[ 4]*tpy[0] + dAd[ 5]*tpy[1] + dAd[ 6]*tpy[2] + dAd[ 7]*tpy[3]);
574 db[2] = (dAd[ 8]*tpy[0] + dAd[ 9]*tpy[1] + dAd[10]*tpy[2] + dAd[11]*tpy[3]);
575 db[3] = (dAd[12]*tpy[0] + dAd[13]*tpy[1] + dAd[14]*tpy[2] + dAd[15]*tpy[3]);
576
577 c[0] = (Ad[ 0]*tpz[0] + Ad[ 1]*tpz[1] + Ad[ 2]*tpz[2] + Ad[ 3]*tpz[3]);
578 c[1] = (Ad[ 4]*tpz[0] + Ad[ 5]*tpz[1] + Ad[ 6]*tpz[2] + Ad[ 7]*tpz[3]);
579 c[2] = (Ad[ 8]*tpz[0] + Ad[ 9]*tpz[1] + Ad[10]*tpz[2] + Ad[11]*tpz[3]);
580 c[3] = (Ad[12]*tpz[0] + Ad[13]*tpz[1] + Ad[14]*tpz[2] + Ad[15]*tpz[3]);
581 dc[0] = (dAd[ 0]*tpz[0] + dAd[ 1]*tpz[1] + dAd[ 2]*tpz[2] + dAd[ 3]*tpz[3]);
582 dc[1] = (dAd[ 4]*tpz[0] + dAd[ 5]*tpz[1] + dAd[ 6]*tpz[2] + dAd[ 7]*tpz[3]);
583 dc[2] = (dAd[ 8]*tpz[0] + dAd[ 9]*tpz[1] + dAd[10]*tpz[2] + dAd[11]*tpz[3]);
584 dc[3] = (dAd[12]*tpz[0] + dAd[13]*tpz[1] + dAd[14]*tpz[2] + dAd[15]*tpz[3]);
585
586 intptr_t xs = spline->x_stride;
587 intptr_t ys = spline->y_stride;
588 intptr_t zs = spline->z_stride;
589
590 for (int n=0; n<spline->num_splines; n++) {
591 vals[n] = 0.0;
592 grads[3*n+0] = grads[3*n+1] = grads[3*n+2] = 0.0;
593 }
594
595 for (int i=0; i<4; i++)
596 for (int j=0; j<4; j++)
597 for (int k=0; k<4; k++) {
598 double abc = a[i]*b[j]*c[k];
599 double dabc[3];
600 dabc[0] = da[i]* b[j]* c[k];
601 dabc[1] = a[i]*db[j]* c[k];
602 dabc[2] = a[i]* b[j]*dc[k];
603
604 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys + (iz+k)*zs);
605 for (int n=0; n<spline->num_splines; n++) {
606 vals[n] += abc *coefs[n];
607 grads[3*n+0] += dabc[0]*coefs[n];
608 grads[3*n+1] += dabc[1]*coefs[n];
609 grads[3*n+2] += dabc[2]*coefs[n];
610 }
611 }
612
613 double dxInv = spline->x_grid.delta_inv;
614 double dyInv = spline->y_grid.delta_inv;
615 double dzInv = spline->z_grid.delta_inv;
616 for (int n=0; n<spline->num_splines; n++) {
617 grads[3*n+0] *= dxInv;
618 grads[3*n+1] *= dyInv;
619 grads[3*n+2] *= dzInv;
620 }
621}

References Ad, multi_UBspline_3d_z::coefs, dAd, Ugrid::delta_inv, multi_UBspline_3d_z::num_splines, restrict, Ugrid::start, multi_UBspline_3d_z::x_grid, multi_UBspline_3d_z::x_stride, multi_UBspline_3d_z::y_grid, multi_UBspline_3d_z::y_stride, multi_UBspline_3d_z::z_grid, and multi_UBspline_3d_z::z_stride.

◆ eval_multi_UBspline_3d_z_vgh()

void eval_multi_UBspline_3d_z_vgh ( multi_UBspline_3d_z * spline,
double x,
double y,
double z,
complex_double *restrict vals,
complex_double *restrict grads,
complex_double *restrict hess )

Definition at line 742 of file multi_bspline_eval_std_z.h.

747{
748 x -= spline->x_grid.start;
749 y -= spline->y_grid.start;
750 z -= spline->z_grid.start;
751 double ux = x*spline->x_grid.delta_inv;
752 double uy = y*spline->y_grid.delta_inv;
753 double uz = z*spline->z_grid.delta_inv;
754 double ipartx, iparty, ipartz, tx, ty, tz;
755 tx = modf (ux, &ipartx); int ix = (int) ipartx;
756 ty = modf (uy, &iparty); int iy = (int) iparty;
757 tz = modf (uz, &ipartz); int iz = (int) ipartz;
758
759 double tpx[4], tpy[4], tpz[4], a[4], b[4], c[4],
760 da[4], db[4], dc[4], d2a[4], d2b[4], d2c[4];
761 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
762 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
763 tpz[0] = tz*tz*tz; tpz[1] = tz*tz; tpz[2] = tz; tpz[3] = 1.0;
764 complex_double* restrict coefs = spline->coefs;
765
766 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
767 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
768 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
769 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
770 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
771 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
772 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
773 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
774 d2a[0] = (d2Ad[ 0]*tpx[0] + d2Ad[ 1]*tpx[1] + d2Ad[ 2]*tpx[2] + d2Ad[ 3]*tpx[3]);
775 d2a[1] = (d2Ad[ 4]*tpx[0] + d2Ad[ 5]*tpx[1] + d2Ad[ 6]*tpx[2] + d2Ad[ 7]*tpx[3]);
776 d2a[2] = (d2Ad[ 8]*tpx[0] + d2Ad[ 9]*tpx[1] + d2Ad[10]*tpx[2] + d2Ad[11]*tpx[3]);
777 d2a[3] = (d2Ad[12]*tpx[0] + d2Ad[13]*tpx[1] + d2Ad[14]*tpx[2] + d2Ad[15]*tpx[3]);
778
779 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
780 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
781 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
782 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
783 db[0] = (dAd[ 0]*tpy[0] + dAd[ 1]*tpy[1] + dAd[ 2]*tpy[2] + dAd[ 3]*tpy[3]);
784 db[1] = (dAd[ 4]*tpy[0] + dAd[ 5]*tpy[1] + dAd[ 6]*tpy[2] + dAd[ 7]*tpy[3]);
785 db[2] = (dAd[ 8]*tpy[0] + dAd[ 9]*tpy[1] + dAd[10]*tpy[2] + dAd[11]*tpy[3]);
786 db[3] = (dAd[12]*tpy[0] + dAd[13]*tpy[1] + dAd[14]*tpy[2] + dAd[15]*tpy[3]);
787 d2b[0] = (d2Ad[ 0]*tpy[0] + d2Ad[ 1]*tpy[1] + d2Ad[ 2]*tpy[2] + d2Ad[ 3]*tpy[3]);
788 d2b[1] = (d2Ad[ 4]*tpy[0] + d2Ad[ 5]*tpy[1] + d2Ad[ 6]*tpy[2] + d2Ad[ 7]*tpy[3]);
789 d2b[2] = (d2Ad[ 8]*tpy[0] + d2Ad[ 9]*tpy[1] + d2Ad[10]*tpy[2] + d2Ad[11]*tpy[3]);
790 d2b[3] = (d2Ad[12]*tpy[0] + d2Ad[13]*tpy[1] + d2Ad[14]*tpy[2] + d2Ad[15]*tpy[3]);
791
792 c[0] = (Ad[ 0]*tpz[0] + Ad[ 1]*tpz[1] + Ad[ 2]*tpz[2] + Ad[ 3]*tpz[3]);
793 c[1] = (Ad[ 4]*tpz[0] + Ad[ 5]*tpz[1] + Ad[ 6]*tpz[2] + Ad[ 7]*tpz[3]);
794 c[2] = (Ad[ 8]*tpz[0] + Ad[ 9]*tpz[1] + Ad[10]*tpz[2] + Ad[11]*tpz[3]);
795 c[3] = (Ad[12]*tpz[0] + Ad[13]*tpz[1] + Ad[14]*tpz[2] + Ad[15]*tpz[3]);
796 dc[0] = (dAd[ 0]*tpz[0] + dAd[ 1]*tpz[1] + dAd[ 2]*tpz[2] + dAd[ 3]*tpz[3]);
797 dc[1] = (dAd[ 4]*tpz[0] + dAd[ 5]*tpz[1] + dAd[ 6]*tpz[2] + dAd[ 7]*tpz[3]);
798 dc[2] = (dAd[ 8]*tpz[0] + dAd[ 9]*tpz[1] + dAd[10]*tpz[2] + dAd[11]*tpz[3]);
799 dc[3] = (dAd[12]*tpz[0] + dAd[13]*tpz[1] + dAd[14]*tpz[2] + dAd[15]*tpz[3]);
800 d2c[0] = (d2Ad[ 0]*tpz[0] + d2Ad[ 1]*tpz[1] + d2Ad[ 2]*tpz[2] + d2Ad[ 3]*tpz[3]);
801 d2c[1] = (d2Ad[ 4]*tpz[0] + d2Ad[ 5]*tpz[1] + d2Ad[ 6]*tpz[2] + d2Ad[ 7]*tpz[3]);
802 d2c[2] = (d2Ad[ 8]*tpz[0] + d2Ad[ 9]*tpz[1] + d2Ad[10]*tpz[2] + d2Ad[11]*tpz[3]);
803 d2c[3] = (d2Ad[12]*tpz[0] + d2Ad[13]*tpz[1] + d2Ad[14]*tpz[2] + d2Ad[15]*tpz[3]);
804
805 intptr_t xs = spline->x_stride;
806 intptr_t ys = spline->y_stride;
807 intptr_t zs = spline->z_stride;
808
809 for (int n=0; n<spline->num_splines; n++) {
810 vals[n] = 0.0;
811 grads[3*n+0] = grads[3*n+1] = grads[3*n+2] = 0.0;
812 for (int i=0; i<9; i++)
813 hess[9*n+i] = 0.0;
814 }
815
816 for (int i=0; i<4; i++)
817 for (int j=0; j<4; j++)
818 for (int k=0; k<4; k++) {
819 double abc = a[i]*b[j]*c[k];
820 double dabc[3], d2abc[6];
821 dabc[0] = da[i]* b[j]* c[k];
822 dabc[1] = a[i]*db[j]* c[k];
823 dabc[2] = a[i]* b[j]*dc[k];
824 d2abc[0] = d2a[i]* b[j]* c[k];
825 d2abc[1] = da[i]* db[j]* c[k];
826 d2abc[2] = da[i]* b[j]* dc[k];
827 d2abc[3] = a[i]*d2b[j]* c[k];
828 d2abc[4] = a[i]* db[j]* dc[k];
829 d2abc[5] = a[i]* b[j]*d2c[k];
830
831 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys + (iz+k)*zs);
832 for (int n=0; n<spline->num_splines; n++) {
833 vals[n] += abc *coefs[n];
834 grads[3*n+0] += dabc[0]*coefs[n];
835 grads[3*n+1] += dabc[1]*coefs[n];
836 grads[3*n+2] += dabc[2]*coefs[n];
837 hess [9*n+0] += d2abc[0]*coefs[n];
838 hess [9*n+1] += d2abc[1]*coefs[n];
839 hess [9*n+2] += d2abc[2]*coefs[n];
840 hess [9*n+4] += d2abc[3]*coefs[n];
841 hess [9*n+5] += d2abc[4]*coefs[n];
842 hess [9*n+8] += d2abc[5]*coefs[n];
843 }
844 }
845
846 double dxInv = spline->x_grid.delta_inv;
847 double dyInv = spline->y_grid.delta_inv;
848 double dzInv = spline->z_grid.delta_inv;
849 for (int n=0; n<spline->num_splines; n++) {
850 grads[3*n+0] *= dxInv;
851 grads[3*n+1] *= dyInv;
852 grads[3*n+2] *= dzInv;
853 hess[9*n+0] *= dxInv*dxInv;
854 hess[9*n+4] *= dyInv*dyInv;
855 hess[9*n+8] *= dzInv*dzInv;
856 hess[9*n+1] *= dxInv*dyInv;
857 hess[9*n+2] *= dxInv*dzInv;
858 hess[9*n+5] *= dyInv*dzInv;
859 // Copy hessian elements into lower half of 3x3 matrix
860 hess[9*n+3] = hess[9*n+1];
861 hess[9*n+6] = hess[9*n+2];
862 hess[9*n+7] = hess[9*n+5];
863 }
864}

References Ad, multi_UBspline_3d_z::coefs, d2Ad, dAd, Ugrid::delta_inv, multi_UBspline_3d_z::num_splines, restrict, Ugrid::start, multi_UBspline_3d_z::x_grid, multi_UBspline_3d_z::x_stride, multi_UBspline_3d_z::y_grid, multi_UBspline_3d_z::y_stride, multi_UBspline_3d_z::z_grid, and multi_UBspline_3d_z::z_stride.

◆ eval_multi_UBspline_3d_z_vgl()

void eval_multi_UBspline_3d_z_vgl ( multi_UBspline_3d_z * spline,
double x,
double y,
double z,
complex_double *restrict vals,
complex_double *restrict grads,
complex_double *restrict lapl )

Definition at line 626 of file multi_bspline_eval_std_z.h.

631{
632 x -= spline->x_grid.start;
633 y -= spline->y_grid.start;
634 z -= spline->z_grid.start;
635 double ux = x*spline->x_grid.delta_inv;
636 double uy = y*spline->y_grid.delta_inv;
637 double uz = z*spline->z_grid.delta_inv;
638 double ipartx, iparty, ipartz, tx, ty, tz;
639 tx = modf (ux, &ipartx); int ix = (int) ipartx;
640 ty = modf (uy, &iparty); int iy = (int) iparty;
641 tz = modf (uz, &ipartz); int iz = (int) ipartz;
642
643 double tpx[4], tpy[4], tpz[4], a[4], b[4], c[4],
644 da[4], db[4], dc[4], d2a[4], d2b[4], d2c[4];
645 tpx[0] = tx*tx*tx; tpx[1] = tx*tx; tpx[2] = tx; tpx[3] = 1.0;
646 tpy[0] = ty*ty*ty; tpy[1] = ty*ty; tpy[2] = ty; tpy[3] = 1.0;
647 tpz[0] = tz*tz*tz; tpz[1] = tz*tz; tpz[2] = tz; tpz[3] = 1.0;
648 complex_double* restrict coefs = spline->coefs;
649
650 a[0] = (Ad[ 0]*tpx[0] + Ad[ 1]*tpx[1] + Ad[ 2]*tpx[2] + Ad[ 3]*tpx[3]);
651 a[1] = (Ad[ 4]*tpx[0] + Ad[ 5]*tpx[1] + Ad[ 6]*tpx[2] + Ad[ 7]*tpx[3]);
652 a[2] = (Ad[ 8]*tpx[0] + Ad[ 9]*tpx[1] + Ad[10]*tpx[2] + Ad[11]*tpx[3]);
653 a[3] = (Ad[12]*tpx[0] + Ad[13]*tpx[1] + Ad[14]*tpx[2] + Ad[15]*tpx[3]);
654 da[0] = (dAd[ 0]*tpx[0] + dAd[ 1]*tpx[1] + dAd[ 2]*tpx[2] + dAd[ 3]*tpx[3]);
655 da[1] = (dAd[ 4]*tpx[0] + dAd[ 5]*tpx[1] + dAd[ 6]*tpx[2] + dAd[ 7]*tpx[3]);
656 da[2] = (dAd[ 8]*tpx[0] + dAd[ 9]*tpx[1] + dAd[10]*tpx[2] + dAd[11]*tpx[3]);
657 da[3] = (dAd[12]*tpx[0] + dAd[13]*tpx[1] + dAd[14]*tpx[2] + dAd[15]*tpx[3]);
658 d2a[0] = (d2Ad[ 0]*tpx[0] + d2Ad[ 1]*tpx[1] + d2Ad[ 2]*tpx[2] + d2Ad[ 3]*tpx[3]);
659 d2a[1] = (d2Ad[ 4]*tpx[0] + d2Ad[ 5]*tpx[1] + d2Ad[ 6]*tpx[2] + d2Ad[ 7]*tpx[3]);
660 d2a[2] = (d2Ad[ 8]*tpx[0] + d2Ad[ 9]*tpx[1] + d2Ad[10]*tpx[2] + d2Ad[11]*tpx[3]);
661 d2a[3] = (d2Ad[12]*tpx[0] + d2Ad[13]*tpx[1] + d2Ad[14]*tpx[2] + d2Ad[15]*tpx[3]);
662
663 b[0] = (Ad[ 0]*tpy[0] + Ad[ 1]*tpy[1] + Ad[ 2]*tpy[2] + Ad[ 3]*tpy[3]);
664 b[1] = (Ad[ 4]*tpy[0] + Ad[ 5]*tpy[1] + Ad[ 6]*tpy[2] + Ad[ 7]*tpy[3]);
665 b[2] = (Ad[ 8]*tpy[0] + Ad[ 9]*tpy[1] + Ad[10]*tpy[2] + Ad[11]*tpy[3]);
666 b[3] = (Ad[12]*tpy[0] + Ad[13]*tpy[1] + Ad[14]*tpy[2] + Ad[15]*tpy[3]);
667 db[0] = (dAd[ 0]*tpy[0] + dAd[ 1]*tpy[1] + dAd[ 2]*tpy[2] + dAd[ 3]*tpy[3]);
668 db[1] = (dAd[ 4]*tpy[0] + dAd[ 5]*tpy[1] + dAd[ 6]*tpy[2] + dAd[ 7]*tpy[3]);
669 db[2] = (dAd[ 8]*tpy[0] + dAd[ 9]*tpy[1] + dAd[10]*tpy[2] + dAd[11]*tpy[3]);
670 db[3] = (dAd[12]*tpy[0] + dAd[13]*tpy[1] + dAd[14]*tpy[2] + dAd[15]*tpy[3]);
671 d2b[0] = (d2Ad[ 0]*tpy[0] + d2Ad[ 1]*tpy[1] + d2Ad[ 2]*tpy[2] + d2Ad[ 3]*tpy[3]);
672 d2b[1] = (d2Ad[ 4]*tpy[0] + d2Ad[ 5]*tpy[1] + d2Ad[ 6]*tpy[2] + d2Ad[ 7]*tpy[3]);
673 d2b[2] = (d2Ad[ 8]*tpy[0] + d2Ad[ 9]*tpy[1] + d2Ad[10]*tpy[2] + d2Ad[11]*tpy[3]);
674 d2b[3] = (d2Ad[12]*tpy[0] + d2Ad[13]*tpy[1] + d2Ad[14]*tpy[2] + d2Ad[15]*tpy[3]);
675
676 c[0] = (Ad[ 0]*tpz[0] + Ad[ 1]*tpz[1] + Ad[ 2]*tpz[2] + Ad[ 3]*tpz[3]);
677 c[1] = (Ad[ 4]*tpz[0] + Ad[ 5]*tpz[1] + Ad[ 6]*tpz[2] + Ad[ 7]*tpz[3]);
678 c[2] = (Ad[ 8]*tpz[0] + Ad[ 9]*tpz[1] + Ad[10]*tpz[2] + Ad[11]*tpz[3]);
679 c[3] = (Ad[12]*tpz[0] + Ad[13]*tpz[1] + Ad[14]*tpz[2] + Ad[15]*tpz[3]);
680 dc[0] = (dAd[ 0]*tpz[0] + dAd[ 1]*tpz[1] + dAd[ 2]*tpz[2] + dAd[ 3]*tpz[3]);
681 dc[1] = (dAd[ 4]*tpz[0] + dAd[ 5]*tpz[1] + dAd[ 6]*tpz[2] + dAd[ 7]*tpz[3]);
682 dc[2] = (dAd[ 8]*tpz[0] + dAd[ 9]*tpz[1] + dAd[10]*tpz[2] + dAd[11]*tpz[3]);
683 dc[3] = (dAd[12]*tpz[0] + dAd[13]*tpz[1] + dAd[14]*tpz[2] + dAd[15]*tpz[3]);
684 d2c[0] = (d2Ad[ 0]*tpz[0] + d2Ad[ 1]*tpz[1] + d2Ad[ 2]*tpz[2] + d2Ad[ 3]*tpz[3]);
685 d2c[1] = (d2Ad[ 4]*tpz[0] + d2Ad[ 5]*tpz[1] + d2Ad[ 6]*tpz[2] + d2Ad[ 7]*tpz[3]);
686 d2c[2] = (d2Ad[ 8]*tpz[0] + d2Ad[ 9]*tpz[1] + d2Ad[10]*tpz[2] + d2Ad[11]*tpz[3]);
687 d2c[3] = (d2Ad[12]*tpz[0] + d2Ad[13]*tpz[1] + d2Ad[14]*tpz[2] + d2Ad[15]*tpz[3]);
688
689 intptr_t xs = spline->x_stride;
690 intptr_t ys = spline->y_stride;
691 intptr_t zs = spline->z_stride;
692
693 //complex_double lapl3[3*spline->num_splines];
694 complex_double* restrict lapl3 = spline->lapl3;
695 for (int n=0; n<spline->num_splines; n++) {
696 vals[n] = 0.0;
697 grads[3*n+0] = grads[3*n+1] = grads[3*n+2] = 0.0;
698 lapl3[3*n+0] = lapl3[3*n+1] = lapl3[3*n+2] = 0.0;
699 }
700
701 for (int i=0; i<4; i++)
702 for (int j=0; j<4; j++)
703 for (int k=0; k<4; k++) {
704 double abc = a[i]*b[j]*c[k];
705 double dabc[3], d2abc[3];
706 dabc[0] = da[i]* b[j]* c[k];
707 dabc[1] = a[i]*db[j]* c[k];
708 dabc[2] = a[i]* b[j]*dc[k];
709 d2abc[0] = d2a[i]* b[j]* c[k];
710 d2abc[1] = a[i]*d2b[j]* c[k];
711 d2abc[2] = a[i]* b[j]*d2c[k];
712
713 complex_double* restrict coefs = spline->coefs + ((ix+i)*xs + (iy+j)*ys + (iz+k)*zs);
714 for (int n=0; n<spline->num_splines; n++) {
715 vals[n] += abc *coefs[n];
716 grads[3*n+0] += dabc[0]*coefs[n];
717 grads[3*n+1] += dabc[1]*coefs[n];
718 grads[3*n+2] += dabc[2]*coefs[n];
719 lapl3[3*n+0] += d2abc[0]*coefs[n];
720 lapl3[3*n+1] += d2abc[1]*coefs[n];
721 lapl3[3*n+2] += d2abc[2]*coefs[n];
722 }
723 }
724
725 double dxInv = spline->x_grid.delta_inv;
726 double dyInv = spline->y_grid.delta_inv;
727 double dzInv = spline->z_grid.delta_inv;
728 for (int n=0; n<spline->num_splines; n++) {
729 grads[3*n+0] *= dxInv;
730 grads[3*n+1] *= dyInv;
731 grads[3*n+2] *= dzInv;
732 lapl3[3*n+0] *= dxInv*dxInv;
733 lapl3[3*n+1] *= dyInv*dyInv;
734 lapl3[3*n+2] *= dzInv*dzInv;
735 lapl[n] = lapl3[3*n+0] + lapl3[3*n+1] + lapl3[3*n+2];
736 }
737}
multi_UBspline_3d_z::lapl3
complex_double *restrict lapl3
Definition multi_bspline_structs.h:178

References Ad, multi_UBspline_3d_z::coefs, d2Ad, dAd, Ugrid::delta_inv, multi_UBspline_3d_z::lapl3, multi_UBspline_3d_z::num_splines, restrict, Ugrid::start, multi_UBspline_3d_z::x_grid, multi_UBspline_3d_z::x_stride, multi_UBspline_3d_z::y_grid, multi_UBspline_3d_z::y_stride, multi_UBspline_3d_z::z_grid, and multi_UBspline_3d_z::z_stride.