Fix missing import in finiterectlat-scatter.py
[qpms.git] / qpms / vswf.c
blob1d0e349249688a0ff0bbb6730ada5f81c4410ce6
1 #include <math.h>
2 #include <gsl/gsl_math.h>
3 #include "assert_cython_workaround.h"
4 #include "vswf.h"
5 #include "indexing.h"
6 #include "translations.h" // TODO move qpms_sph_bessel_fill elsewhere
7 #include "qpms_specfunc.h"
8 #include <stdlib.h>
9 #include <string.h>
10 #include "qpms_error.h"
11 #include "normalisation.h"
14 qpms_vswf_set_spec_t *qpms_vswf_set_spec_init() {
15 qpms_vswf_set_spec_t *s = calloc(1,sizeof(qpms_vswf_set_spec_t));
16 if (s == NULL) return NULL; // TODO warn
17 // The rest are zeroes automatically because of calloc:
18 s->lMax_L = -1;
19 return s;
22 #define MAX(x,y) (((x)<(y))?(y):(x))
24 qpms_errno_t qpms_vswf_set_spec_append(qpms_vswf_set_spec_t *s, const qpms_uvswfi_t u) {
25 qpms_l_t l;
26 qpms_m_t m;
27 qpms_vswf_type_t t;
28 if (qpms_uvswfi2tmn(u, &t, &m, &l)!=QPMS_SUCCESS) return QPMS_ERROR; // TODO WARN
29 if (s->n + 1 > s->capacity) {
30 size_t newcap = (s->capacity < 32) ? 32 : 2*s->capacity;
31 qpms_uvswfi_t *newmem = realloc(s->ilist, newcap * sizeof(qpms_uvswfi_t));
32 if (newmem == NULL) return QPMS_ENOMEM; // TODO WARN
33 s->capacity = newcap;
34 s->ilist = newmem;
36 s->ilist[s->n] = u;
37 ++s->n;
38 switch(t) {
39 case QPMS_VSWF_ELECTRIC:
40 s->lMax_N = MAX(s->lMax_N, l);
41 break;
42 case QPMS_VSWF_MAGNETIC:
43 s->lMax_M = MAX(s->lMax_M, l);
44 break;
45 case QPMS_VSWF_LONGITUDINAL:
46 s->lMax_L = MAX(s->lMax_L, l);
47 break;
48 default:
49 QPMS_WTF;
51 s->lMax = MAX(s->lMax, l);
52 return QPMS_SUCCESS;
55 bool qpms_vswf_set_spec_isidentical(const qpms_vswf_set_spec_t *a,
56 const qpms_vswf_set_spec_t *b) {
57 if (a == b) return true;
58 if (a->norm != b->norm) return false;
59 if (a->n != b->n) return false;
60 for (size_t i = 0; i < a->n; ++i)
61 if (a->ilist[i] != b->ilist[i])
62 return false;
63 return true;
66 qpms_vswf_set_spec_t *qpms_vswf_set_spec_copy(const qpms_vswf_set_spec_t *or){
67 qpms_vswf_set_spec_t *c;
68 QPMS_CRASHING_MALLOC(c, sizeof(*c));
69 *c = *or;
70 c->ilist = malloc(sizeof(qpms_uvswfi_t) * c->n);
71 memcpy(c->ilist, or->ilist, sizeof(qpms_uvswfi_t)*c->n);
72 c->capacity = c->n;
73 return c;
76 qpms_vswf_set_spec_t *qpms_vswf_set_spec_from_lMax(qpms_l_t lMax,
77 qpms_normalisation_t norm) {
78 qpms_vswf_set_spec_t *c;
79 QPMS_CRASHING_MALLOC(c, sizeof(*c));
80 c->n = c->capacity = 2 * qpms_lMax2nelem(lMax);
81 c->ilist = malloc(sizeof(qpms_uvswfi_t) * c->capacity);
82 size_t i = 0;
83 for (int it = 0; it < 2; ++it)
84 for (qpms_l_t n = 1; n <= lMax; ++n)
85 for (qpms_m_t m = -n; m <= n; ++m)
86 c->ilist[i++] =
87 qpms_tmn2uvswfi(it ? QPMS_VSWF_MAGNETIC : QPMS_VSWF_ELECTRIC, m, n);
88 c->norm = norm;
89 c->lMax = c->lMax_M = c->lMax_N = lMax;
90 c->lMax_L = -1;
91 return c;
94 void qpms_vswf_set_spec_free(qpms_vswf_set_spec_t *s) {
95 if(s) free(s->ilist);
96 free(s);
99 struct bspec_reindex_pair {
100 qpms_uvswfi_t ui;
101 size_t i_orig;
104 static int cmp_bspec_reindex_pair(const void *aa, const void *bb) {
105 const struct bspec_reindex_pair *a = aa, *b = bb;
106 if (a->ui < b->ui) return -1;
107 else if (a->ui == b->ui) return 0;
108 else return 1;
111 size_t *qpms_vswf_set_reindex(const qpms_vswf_set_spec_t *small, const qpms_vswf_set_spec_t *big) {
112 QPMS_UNTESTED;
113 struct bspec_reindex_pair *small_pairs, *big_pairs;
114 size_t *r;
115 QPMS_CRASHING_MALLOC(small_pairs, sizeof(struct bspec_reindex_pair) * small->n);
116 QPMS_CRASHING_MALLOC(big_pairs, sizeof(struct bspec_reindex_pair) * big->n);
117 QPMS_CRASHING_MALLOC(r, sizeof(size_t) * small->n);
118 for(size_t i = 0; i < small->n; ++i) {
119 small_pairs[i].ui = small->ilist[i];
120 small_pairs[i].i_orig = i;
122 for(size_t i = 0 ; i < big->n; ++i) {
123 big_pairs[i].ui = big->ilist[i];
124 big_pairs[i].i_orig = i;
126 qsort(small_pairs, small->n, sizeof(struct bspec_reindex_pair), cmp_bspec_reindex_pair);
127 qsort(big_pairs, big->n, sizeof(struct bspec_reindex_pair), cmp_bspec_reindex_pair);
129 size_t bi = 0;
130 for(size_t si = 0; si < small->n; ++si) {
131 while(big_pairs[bi].ui < small_pairs[si].ui)
132 ++bi;
133 if(big_pairs[bi].ui == small_pairs[si].ui)
134 r[small_pairs[si].i_orig] = big_pairs[bi].i_orig;
135 else
136 r[small_pairs[si].i_orig] = ~(size_t)0;
139 free(small_pairs);
140 free(big_pairs);
141 return r;
144 csphvec_t qpms_vswf_single_el_csph(qpms_m_t m, qpms_l_t l, csph_t kdlj,
145 qpms_bessel_t btyp, qpms_normalisation_t norm) {
146 lmcheck(l,m);
147 csphvec_t N;
148 complex double *bessel;
149 QPMS_CRASHING_MALLOC(bessel,(l+1)*sizeof(complex double));
150 QPMS_ENSURE_SUCCESS(qpms_sph_bessel_fill(btyp, l, kdlj.r, bessel));
151 qpms_pitau_t pt = qpms_pitau_get(kdlj.theta, l, qpms_normalisation_t_csphase(norm));
153 complex double eimf = qpms_spharm_azimuthal_part(norm, m, kdlj.phi);
154 complex double d_eimf_dmf = qpms_spharm_azimuthal_part_derivative_div_m(norm, m, kdlj.phi);
155 qpms_y_t y = qpms_mn2y(m,l);
157 N.rc = l*(l+1) * pt.leg[y] * bessel[l] / kdlj.r * eimf;
158 complex double besselfac = bessel[l-1] - l * bessel[l] / kdlj.r;
159 N.thetac = pt.tau[y] * besselfac * eimf;
160 N.phic = pt.pi[y] * besselfac * d_eimf_dmf;
162 N = csphvec_scale(qpms_normalisation_factor_N_noCS(norm, l, m), N);
164 qpms_pitau_free(pt);
165 free(bessel);
166 return N;
169 csphvec_t qpms_vswf_single_mg_csph(qpms_m_t m, qpms_l_t l, csph_t kdlj,
170 qpms_bessel_t btyp, qpms_normalisation_t norm) {
171 lmcheck(l,m);
172 csphvec_t M;
173 complex double *bessel;
174 QPMS_CRASHING_MALLOC(bessel,(l+1)*sizeof(complex double));
175 QPMS_ENSURE_SUCCESS(qpms_sph_bessel_fill(btyp, l, kdlj.r, bessel));
176 qpms_pitau_t pt = qpms_pitau_get(kdlj.theta, l, qpms_normalisation_t_csphase(norm));
177 complex double eimf = qpms_spharm_azimuthal_part(norm, m, kdlj.phi);
178 complex double d_eimf_dmf = qpms_spharm_azimuthal_part_derivative_div_m(norm, m, kdlj.phi);
179 qpms_y_t y = qpms_mn2y(m,l);
181 M.rc = 0.;
182 M.thetac = pt.pi[y] * bessel[l] * d_eimf_dmf;
183 M.phic = -pt.tau[y] * bessel[l] * eimf;
185 M = csphvec_scale(qpms_normalisation_factor_M_noCS(norm, l, m), M);
187 qpms_pitau_free(pt);
188 free(bessel);
189 return M;
192 csphvec_t qpms_vswf_single_el(qpms_m_t m, qpms_l_t l, sph_t kdlj,
193 qpms_bessel_t btyp, qpms_normalisation_t norm) {
194 return qpms_vswf_single_el_csph(m, l, sph2csph(kdlj), btyp, norm);
197 csphvec_t qpms_vswf_single_mg(qpms_m_t m, qpms_l_t l, sph_t kdlj,
198 qpms_bessel_t btyp, qpms_normalisation_t norm) {
199 return qpms_vswf_single_mg_csph(m, l, sph2csph(kdlj), btyp, norm);
202 qpms_vswfset_sph_t *qpms_vswfset_make(qpms_l_t lMax, sph_t kdlj,
203 qpms_bessel_t btyp, qpms_normalisation_t norm) {
204 qpms_vswfset_sph_t *res = malloc(sizeof(qpms_vswfset_sph_t));
205 res->lMax = lMax;
206 qpms_y_t nelem = qpms_lMax2nelem(lMax);
207 QPMS_CRASHING_MALLOC(res->el, sizeof(csphvec_t)*nelem);
208 QPMS_CRASHING_MALLOC(res->mg, sizeof(csphvec_t)*nelem);
209 QPMS_ENSURE_SUCCESS(qpms_vswf_fill(NULL, res->mg, res->el, lMax, kdlj, btyp, norm));
210 return res;
213 void qpms_vswfset_sph_pfree(qpms_vswfset_sph_t *w) {
214 assert(NULL != w && NULL != w->el && NULL != w->mg);
215 free(w->el);
216 free(w->mg);
217 free(w);
220 csphvec_t qpms_vswf_L00(csph_t kr, qpms_bessel_t btyp,
221 qpms_normalisation_t norm) {
222 QPMS_UNTESTED;
223 // CHECKME Is it OK to ignore norm?? (Is L_0^0 the same for all conventions?)
224 complex double bessel0;
225 QPMS_ENSURE_SUCCESS(qpms_sph_bessel_fill(btyp, 0, kr.r, &bessel0));
226 csphvec_t result = {0.25 * M_2_SQRTPI * bessel0, 0, 0};
227 return result;
230 qpms_errno_t qpms_vswf_fill_csph(csphvec_t *const longtarget,
231 csphvec_t * const mgtarget, csphvec_t * const eltarget, qpms_l_t lMax,
232 csph_t kr, qpms_bessel_t btyp, const qpms_normalisation_t norm) {
233 assert(lMax >= 1);
234 complex double *bessel = malloc((lMax+1)*sizeof(complex double));
235 qpms_errno_t bessel_retval = qpms_sph_bessel_fill(btyp, lMax, kr.r, bessel);
236 QPMS_ENSURE_SUCCESS_OR(bessel_retval, QPMS_ESING);
237 qpms_pitau_t pt = qpms_pitau_get(kr.theta, lMax, qpms_normalisation_t_csphase(norm));
238 complex double const *pbes = bessel + 1; // starting from l = 1
239 double const *pleg = pt.leg;
240 double const *ppi = pt.pi;
241 double const *ptau = pt.tau;
242 csphvec_t *plong = longtarget, *pmg = mgtarget, *pel = eltarget;
243 for(qpms_l_t l = 1; l <= lMax; ++l) {
244 complex double besfac;
245 complex double besderfac;
246 if (kr.r) {
247 besfac = *pbes / kr.r;
248 } else {
249 besfac = (1 == l) ? 1/3. : 0;
251 besderfac = *(pbes-1) - l * besfac;
252 for(qpms_m_t m = -l; m <= l; ++m) {
253 complex double eimf = qpms_spharm_azimuthal_part(norm, m, kr.phi);
254 complex double d_eimf_dmf = qpms_spharm_azimuthal_part_derivative_div_m(norm, m, kr.phi);
255 if (longtarget) { QPMS_UNTESTED;
256 double longfac = sqrt(l*(l+1));
257 plong->rc = // FATAL FIXME: I get wrong result here for plane wave re-expansion
258 // whenever kr.r > 0 (for waves with longitudinal component, ofcoz)
259 /*(*(pbes-1) - (l+1)/kr.r* *pbes)*/
260 (besderfac-besfac)
261 * (*pleg) * longfac * eimf;
262 plong->thetac = *ptau * besfac * longfac * eimf;
263 plong->phic = *ppi * besfac * longfac * d_eimf_dmf;
264 *plong = csphvec_scale(qpms_normalisation_factor_L_noCS(norm, l, m), *plong);
265 ++plong;
267 if (eltarget) {
268 pel->rc = l*(l+1) * (*pleg) * besfac * eimf;
269 pel->thetac = *ptau * besderfac * eimf;
270 pel->phic = *ppi * besderfac * d_eimf_dmf;
271 *pel = csphvec_scale(qpms_normalisation_factor_N_noCS(norm, l, m), *pel);
272 ++pel;
274 if (mgtarget) {
275 pmg->rc = 0.;
276 pmg->thetac = *ppi * (*pbes) * d_eimf_dmf;
277 pmg->phic = - *ptau * (*pbes) * eimf;
278 *pmg = csphvec_scale(qpms_normalisation_factor_M_noCS(norm, l, m), *pmg);
279 ++pmg;
281 ++pleg; ++ppi; ++ptau;
283 ++pbes;
285 free(bessel);
286 qpms_pitau_free(pt);
287 return bessel_retval;
290 qpms_errno_t qpms_vswf_fill(csphvec_t *const longtarget,
291 csphvec_t * const mgtarget, csphvec_t * const eltarget, qpms_l_t lMax,
292 sph_t kr, qpms_bessel_t btyp, qpms_normalisation_t norm) {
293 csph_t krc = {kr.r, kr.theta, kr.phi};
294 return qpms_vswf_fill_csph(longtarget, mgtarget, eltarget, lMax,
295 krc, btyp, norm);
298 // consistency check: this should give the same results as the above function (up to rounding errors)
299 qpms_errno_t qpms_vswf_fill_alternative(csphvec_t *const longtarget, csphvec_t * const mgtarget, csphvec_t * const eltarget,
300 qpms_l_t lMax, sph_t kr,
301 qpms_bessel_t btyp, qpms_normalisation_t norm) {
302 assert(lMax >= 1);
303 complex double *bessel = malloc((lMax+1)*sizeof(complex double));
304 QPMS_ENSURE_SUCCESS(qpms_sph_bessel_fill(btyp, lMax, kr.r, bessel));
305 complex double const *pbes = bessel + 1; // starting from l = 1
307 qpms_y_t nelem = qpms_lMax2nelem(lMax);
308 csphvec_t *a;
309 QPMS_CRASHING_MALLOC(a, 3*nelem*sizeof(csphvec_t))
310 csphvec_t * const a1 = a, * const a2 = a1 + nelem, * const a3 = a2 + 2 * nelem;
311 QPMS_ENSURE_SUCCESS(qpms_vecspharm_fill(a1, a2, a3, lMax, kr, norm));
312 const csphvec_t *p1 = a1;
313 const csphvec_t *p2 = a2;
314 const csphvec_t *p3 = a3;
316 csphvec_t *plong = longtarget, *pmg = mgtarget, *pel = eltarget;
317 for(qpms_l_t l = 1; l <= lMax; ++l) {
318 complex double besfac = *pbes / kr.r;
319 complex double besderfac = *(pbes-1) - l * besfac;
320 double sqrtlfac = sqrt(l*(l+1));
321 for(qpms_m_t m = -l; m <= l; ++m) {
322 if (longtarget) {
323 complex double L2Nfac = qpms_normalisation_factor_L_noCS(norm, l, m)
324 / qpms_normalisation_factor_N_noCS(norm, l, m);
325 *plong = csphvec_add(csphvec_scale(besderfac-besfac, *p3),
326 csphvec_scale(sqrtlfac * besfac, *p2));
327 *plong = csphvec_scale(L2Nfac, *plong);
328 ++plong;
330 if (eltarget) {
331 *pel = csphvec_add(csphvec_scale(besderfac, *p2),
332 csphvec_scale(sqrtlfac * besfac, *p3));
333 ++pel;
335 if (mgtarget) {
336 *pmg = csphvec_scale(*pbes, *p1);
337 ++pmg;
339 ++p1; ++p2; ++p3;
341 ++pbes;
343 free(a);
344 free(bessel);
345 return QPMS_SUCCESS;
348 qpms_errno_t qpms_vecspharm_fill(csphvec_t *const a1target, csphvec_t *const a2target, csphvec_t *const a3target,
349 qpms_l_t lMax, sph_t dir, qpms_normalisation_t norm) {
350 assert(lMax >= 1);
351 qpms_pitau_t pt = qpms_pitau_get(dir.theta, lMax, qpms_normalisation_t_csphase(norm));
352 double const *pleg = pt.leg;
353 double const *ppi = pt.pi;
354 double const *ptau = pt.tau;
355 csphvec_t *p1 = a1target, *p2 = a2target, *p3 = a3target;
356 for (qpms_l_t l = 1; l <= lMax; ++l) {
357 for(qpms_m_t m = -l; m <= l; ++m) {
358 const complex double Mfac = qpms_normalisation_factor_M_noCS(norm, l, m);
359 const complex double Nfac = qpms_normalisation_factor_N_noCS(norm, l, m);
360 const complex double eimf = qpms_spharm_azimuthal_part(norm, m, dir.phi);
361 const complex double deimf_dmf = qpms_spharm_azimuthal_part_derivative_div_m(norm, m, dir.phi);
362 if (a1target) {
363 p1->rc = 0;
364 p1->thetac = *ppi * deimf_dmf * Mfac;
365 p1->phic = -*ptau * eimf * Mfac;
366 ++p1;
368 if (a2target) {
369 p2->rc = 0;
370 p2->thetac = *ptau * eimf * Nfac;
371 p2->phic = *ppi * deimf_dmf * Nfac;
372 ++p2;
374 if (a3target) {
375 p3->rc = sqrt(l*(l+1)) * (*pleg) * eimf * Nfac;
376 p3->thetac = 0;
377 p3->phic = 0;
378 ++p3;
380 ++pleg; ++ppi; ++ptau;
383 qpms_pitau_free(pt);
384 return QPMS_SUCCESS;
387 qpms_errno_t qpms_vecspharm_dual_fill(csphvec_t *const a1target, csphvec_t *const a2target, csphvec_t *const a3target,
388 qpms_l_t lMax, sph_t dir, qpms_normalisation_t norm) {
389 #if 1
390 return qpms_vecspharm_fill(a1target, a2target, a3target, lMax, dir,
391 qpms_normalisation_dual(norm));
392 #else
393 assert(lMax >= 1);
394 qpms_pitau_t pt = qpms_pitau_get(dir.theta, lMax, norm);
395 double const *pleg = pt.leg;
396 double const *ppi = pt.pi;
397 double const *ptau = pt.tau;
398 csphvec_t *p1 = a1target, *p2 = a2target, *p3 = a3target;
399 for(qpms_l_t l = 1; l <= lMax; ++l) {
400 for(qpms_m_t m = -l; m <= l; ++m) {
401 double normfac = 1./qpms_normalisation_t_factor_abssquare(norm, l, m); // factor w.r.t. Kristensson
402 complex double eimf = cexp(m * dir.phi * I);
403 if (a1target) {
404 p1->rc = 0;
405 p1->thetac = conj(*ppi * normfac * I * eimf);
406 p1->phic = conj(-*ptau * normfac * eimf);
407 ++p1;
409 if (a2target) {
410 p2->rc = 0;
411 p2->thetac = conj(*ptau * normfac * eimf);
412 p2->phic = conj(*ppi * normfac * I * eimf);
413 ++p2;
415 if (a3target) {
416 p3->rc = conj(sqrt(l*(l+1)) * (*pleg) * normfac * eimf);
417 p3->thetac = 0;
418 p3->phic = 0;
419 ++p3;
421 ++pleg; ++ppi; ++ptau;
424 qpms_pitau_free(pt);
425 return QPMS_SUCCESS;
426 #endif
430 static inline complex double ipowl(qpms_l_t l) {
431 switch(l % 4) {
432 case 0: return 1;
433 break;
434 case 1: return I;
435 break;
436 case 2: return -1;
437 break;
438 case 3: return -I;
439 break;
440 default: QPMS_WTF;
442 QPMS_WTF;
445 qpms_errno_t qpms_planewave2vswf_fill_sph(sph_t wavedir, csphvec_t amplitude,
446 complex double *target_longcoeff, complex double *target_mgcoeff,
447 complex double *target_elcoeff, qpms_l_t lMax, qpms_normalisation_t norm) {
448 qpms_y_t nelem = qpms_lMax2nelem(lMax);
449 csphvec_t * const dual_A1 = malloc(3*nelem*sizeof(csphvec_t)), *const dual_A2 = dual_A1 + nelem,
450 * const dual_A3 = dual_A2 + nelem;
451 QPMS_ENSURE_SUCCESS(qpms_vecspharm_dual_fill(dual_A1, dual_A2, dual_A3, lMax, wavedir, norm))
452 const csphvec_t *pA1 = dual_A1, *pA2 = dual_A2, *pA3 = dual_A3;
453 complex double *plong = target_longcoeff, *pmg = target_mgcoeff, *pel = target_elcoeff;
454 for (qpms_l_t l = 1; l <= lMax; ++l) {
455 complex double prefac1 = 4 * M_PI * ipowl(l);
456 complex double prefac23 = - 4 * M_PI * ipowl(l+1);
457 for (qpms_m_t m = -l; m <= l; ++m) {
458 if(target_longcoeff) *plong = prefac23 * csphvec_dotnc(*pA3, amplitude);
459 if(target_mgcoeff) *pmg = prefac1 * csphvec_dotnc(*pA1, amplitude);
460 if(target_elcoeff) *pel = prefac23 * csphvec_dotnc(*pA2, amplitude);
461 ++pA1; ++pA2; ++pA3; ++plong; ++pmg; ++pel;
465 free(dual_A1);
466 return QPMS_SUCCESS;
469 qpms_errno_t qpms_planewave2vswf_fill_cart(cart3_t wavedir_cart /*allow complex k?*/, ccart3_t amplitude_cart,
470 complex double * const longcoeff, complex double * const mgcoeff,
471 complex double * const elcoeff, qpms_l_t lMax, qpms_normalisation_t norm)
474 sph_t wavedir_sph = cart2sph(wavedir_cart);
475 csphvec_t amplitude_sphvec = ccart2csphvec(amplitude_cart, wavedir_sph);
476 return qpms_planewave2vswf_fill_sph(wavedir_sph, amplitude_sphvec,
477 longcoeff, mgcoeff, elcoeff, lMax, norm);
480 qpms_errno_t qpms_incfield_planewave(complex double *target, const qpms_vswf_set_spec_t *bspec,
481 const cart3_t evalpoint, const void *args, bool add) {
482 QPMS_UNTESTED;
483 const qpms_incfield_planewave_params_t *p = args;
485 const ccart3_t k_cart = p->use_cartesian ? p->k.cart : csph2ccart(p->k.sph);
486 const complex double phase = ccart3_dotnc(k_cart, cart32ccart3(evalpoint));
487 if(cimag(phase))
488 QPMS_INCOMPLETE_IMPLEMENTATION("Complex-valued wave vector not implemented correctly; cf. docs.");
489 const complex double phasefac = cexp(I*phase);
491 // Throw away the imaginary component; TODO handle it correctly
492 const sph_t k_sph = csph2sph(p->use_cartesian ? ccart2csph(p->k.cart) : p->k.sph);
493 const csphvec_t E_sph = csphvec_scale(phasefac,
494 p->use_cartesian ? ccart2csphvec(p->E.cart, k_sph) : p->E.sph);
496 complex double *lc = NULL, *mc = NULL, *nc = NULL;
497 const qpms_y_t nelem = qpms_lMax2nelem(bspec->lMax);
498 if (bspec->lMax_L > 0) QPMS_CRASHING_MALLOC(lc, nelem * sizeof(complex double));
499 if (bspec->lMax_M > 0) QPMS_CRASHING_MALLOC(mc, nelem * sizeof(complex double));
500 if (bspec->lMax_N > 0) QPMS_CRASHING_MALLOC(nc, nelem * sizeof(complex double));
502 qpms_errno_t retval = qpms_planewave2vswf_fill_sph(k_sph, E_sph, lc, mc, nc,
503 bspec->lMax, bspec->norm);
505 if (!add) memset(target, 0, bspec->n * sizeof(complex double));
506 for (size_t i = 0; i < bspec->n; ++i) {
507 const qpms_uvswfi_t ui = bspec->ilist[i];
508 if (ui == QPMS_UI_L00) // for l = 0 the coefficient is zero due to symmetry (for real wave vector)
509 target[i] = 0;
510 else {
511 qpms_vswf_type_t t; qpms_y_t y;
512 QPMS_ENSURE_SUCCESS(qpms_uvswfi2ty_l(ui, &t, &y));
513 switch(t) {
514 case QPMS_VSWF_ELECTRIC:
515 target[i] += nc[y];
516 break;
517 case QPMS_VSWF_MAGNETIC:
518 target[i] += mc[y];
519 break;
520 case QPMS_VSWF_LONGITUDINAL:
521 target[i] += lc[y];
522 break;
523 default:
524 QPMS_WTF;
528 free(lc); free(mc); free(nc);
529 return retval;
533 csphvec_t qpms_eval_vswf_csph(csph_t kr,
534 complex double * const lc, complex double *const mc, complex double *const ec,
535 qpms_l_t lMax, qpms_bessel_t btyp, qpms_normalisation_t norm)
537 qpms_y_t nelem = qpms_lMax2nelem(lMax);
538 csphvec_t lsum, msum, esum, lcomp, mcomp, ecomp;
539 csphvec_kahaninit(&lsum, &lcomp);
540 csphvec_kahaninit(&msum, &mcomp);
541 csphvec_kahaninit(&esum, &ecomp);
542 csphvec_t *lset = NULL, *mset = NULL, *eset = NULL;
543 if(lc) lset = malloc(nelem * sizeof(csphvec_t));
544 if(mc) mset = malloc(nelem * sizeof(csphvec_t));
545 if(ec) eset = malloc(nelem * sizeof(csphvec_t));
546 qpms_vswf_fill_csph(lset, mset, eset, lMax, kr, btyp, norm);
547 if(lc) for(qpms_y_t y = 0; y < nelem; ++y)
548 csphvec_kahanadd(&lsum, &lcomp, csphvec_scale(lc[y], lset[y]));
549 if(mc) for(qpms_y_t y = 0; y < nelem; ++y)
550 csphvec_kahanadd(&msum, &mcomp, csphvec_scale(mc[y], mset[y]));
551 if(ec) for(qpms_y_t y = 0; y < nelem; ++y)
552 csphvec_kahanadd(&esum, &ecomp, csphvec_scale(ec[y], eset[y]));
553 if(lc) free(lset);
554 if(mc) free(mset);
555 if(ec) free(eset);
556 //return csphvec_add(esum, csphvec_add(msum, lsum));
557 csphvec_kahanadd(&esum, &ecomp, msum);
558 csphvec_kahanadd(&esum, &ecomp, lsum);
559 return esum;
562 csphvec_t qpms_eval_vswf(sph_t kr,
563 complex double * const lc, complex double *const mc, complex double *const ec,
564 qpms_l_t lMax, qpms_bessel_t btyp, qpms_normalisation_t norm) {
565 csph_t krc = {kr.r, kr.theta, kr.phi};
566 return qpms_eval_vswf_csph(krc, lc, mc, ec, lMax, btyp, norm);
569 qpms_errno_t qpms_uvswf_fill(csphvec_t *const target, const qpms_vswf_set_spec_t *bspec,
570 csph_t kr, qpms_bessel_t btyp) {
571 QPMS_UNTESTED;
572 QPMS_ENSURE(target, "Target array pointer must not be NULL.");
573 csphvec_t *el = NULL, *mg = NULL, *lg = NULL;
574 const qpms_y_t nelem = qpms_lMax2nelem(bspec->lMax);
575 if (bspec->lMax_L > 0)
576 QPMS_CRASHING_MALLOC(lg, nelem * sizeof(csphvec_t));
577 if (bspec->lMax_M > 0)
578 QPMS_CRASHING_MALLOC(mg, nelem * sizeof(csphvec_t));
579 if (bspec->lMax_N > 0)
580 QPMS_CRASHING_MALLOC(el, nelem * sizeof(csphvec_t));
581 qpms_errno_t retval = qpms_vswf_fill_csph(lg, mg, el, bspec->lMax, kr, btyp, bspec->norm);
582 for (size_t i = 0; i < bspec->n; ++i) {
583 const qpms_uvswfi_t ui = bspec->ilist[i];
584 if (ui == QPMS_UI_L00) // l = 0 longitudinal wave must be calculated separately.
585 target[i] = qpms_vswf_L00(kr, btyp, bspec->norm);
586 else {
587 qpms_vswf_type_t t; qpms_y_t y;
588 QPMS_ENSURE_SUCCESS(qpms_uvswfi2ty_l(ui, &t, &y));
589 switch(t) {
590 case QPMS_VSWF_ELECTRIC:
591 target[i] = el[y];
592 break;
593 case QPMS_VSWF_MAGNETIC:
594 target[i] = mg[y];
595 break;
596 case QPMS_VSWF_LONGITUDINAL:
597 target[i] = lg[y];
598 break;
599 default:
600 QPMS_WTF;
604 free(lg);
605 free(mg);
606 free(el);
607 return retval;
611 csphvec_t qpms_eval_uvswf(const qpms_vswf_set_spec_t *bspec,
612 const complex double *coeffs, const csph_t kr,
613 const qpms_bessel_t btyp) {
614 QPMS_UNTESTED;
615 complex double *cM = NULL, *cN = NULL, *cL = NULL, cL00 = 0;
616 if (bspec->lMax_L > 0)
617 QPMS_CRASHING_CALLOC(cL, bspec->n, sizeof(complex double));
618 if (bspec->lMax_M > 0)
619 QPMS_CRASHING_CALLOC(cM, bspec->n, sizeof(complex double));
620 if (bspec->lMax_N > 0)
621 QPMS_CRASHING_CALLOC(cN, bspec->n, sizeof(complex double));
622 for (size_t i = 0; i < bspec->n; ++i) {
623 if (bspec->ilist[i] == 0) // L00, needs special care
624 cL00 = coeffs[i];
625 else {
626 qpms_vswf_type_t t;
627 qpms_y_t y;
628 QPMS_ENSURE_SUCCESS(qpms_uvswfi2ty_l(bspec->ilist[i], &t, &y));
629 switch(t) {
630 case QPMS_VSWF_LONGITUDINAL:
631 QPMS_ASSERT(cL);
632 cL[y] = coeffs[i];
633 break;
634 case QPMS_VSWF_MAGNETIC:
635 QPMS_ASSERT(cM);
636 cM[y] = coeffs[i];
637 break;
638 case QPMS_VSWF_ELECTRIC:
639 QPMS_ASSERT(cN);
640 cN[y] = coeffs[i];
641 break;
642 default:
643 QPMS_WTF;
647 csphvec_t result = qpms_eval_vswf_csph(kr, cL, cM, cN, bspec->lMax, btyp, bspec->norm);
648 free(cM); free(cN); free(cL);
649 if(cL00)
650 result = csphvec_add(result,
651 csphvec_scale(cL00, qpms_vswf_L00(kr, btyp, bspec->norm)));
652 return result;