156 lines
5.5 KiB
C
156 lines
5.5 KiB
C
#ifndef EWALD_H
|
||
#define EWALD_H
|
||
#include <gsl/gsl_sf_result.h>
|
||
#include <gsl/gsl_errno.h>
|
||
#include <math.h> // for inlined lilgamma
|
||
#include <complex.h>
|
||
#include "qpms_types.h"
|
||
|
||
/*
|
||
* Implementation of two-dimensional lattice sum in three dimensions
|
||
* according to:
|
||
* [1] C.M. Linton, I. Thompson
|
||
* Journal of Computational Physics 228 (2009) 1815–1829
|
||
*/
|
||
|
||
/*
|
||
* N.B.!!! currently, the long-range parts are calculated not according to [1,(4.5)], but rather
|
||
* according to the spherical-harmonic-normalisation-independent formulation in my notes
|
||
* notes/ewald.lyx.
|
||
* Both parts of lattice sums are then calculated with the P_n^|m| e^imf substituted in place
|
||
* of Y_n^m (this is quite a weird normalisation especially for negative |m|, but it is consistent
|
||
* with the current implementation of the translation coefficients in translations.c;
|
||
* in the long run, it might make more sense to replace it everywhere with normalised
|
||
* Legendre polynomials).
|
||
*/
|
||
|
||
|
||
// Use this handler to ignore underflows of incomplete gamma.
|
||
gsl_error_handler_t IgnoreUnderflowsGSLErrorHandler;
|
||
|
||
|
||
/* Object holding the constant factors from [1, (4.5)] */
|
||
typedef struct {
|
||
qpms_l_t lMax;
|
||
qpms_y_t nelem_sc;
|
||
qpms_l_t *s1_jMaxes;
|
||
complex double **s1_constfacs; // indices [y][j] where j is same as in [1, (4.5)]
|
||
// TODO probably normalisation and equatorial legendre polynomials should be included, too
|
||
complex double *s1_constfacs_base; // internal pointer holding the memory for the constants
|
||
|
||
double *legendre0; /* now with GSL_SF_LEGENDRE_NONE normalisation, because this is what is
|
||
* what the multipliers from translations.c count with.
|
||
*/
|
||
// gsl_sf_legendre_t legendre0_type;
|
||
int legendre0_csphase; /* 1 or -1; csphase of the Legendre polynomials saved in legendre0.
|
||
This is because I dont't actually consider this fixed in
|
||
translations.c */
|
||
|
||
} qpms_ewald32_constants_t;
|
||
|
||
qpms_ewald32_constants_t *qpms_ewald32_constants_init(qpms_l_t lMax, int csphase);
|
||
void qpms_ewald32_constants_free(qpms_ewald32_constants_t *);
|
||
|
||
|
||
typedef struct { // as gsl_sf_result, but with complex val
|
||
complex double val;
|
||
double err;
|
||
} qpms_csf_result;
|
||
|
||
|
||
// [1, (A.9)]
|
||
static inline complex double lilgamma(double t) {
|
||
t = fabs(t);
|
||
if (t >= 1)
|
||
return sqrt(t*t - 1);
|
||
else
|
||
return -I * sqrt(1 - t*t);
|
||
}
|
||
|
||
|
||
// Incomplete Gamma function as series
|
||
// DLMF 8.7.3 (latter expression) for complex second argument
|
||
int cx_gamma_inc_series_e(double a, complex z, qpms_csf_result * result);
|
||
|
||
// Incomplete gamma for complex second argument
|
||
// if x is (almost) real, it just uses gsl_sf_gamma_inc_e
|
||
int complex_gamma_inc_e(double a, complex double x, qpms_csf_result *result);
|
||
|
||
|
||
// hypergeometric 2F2, used to calculate some errors
|
||
int hyperg_2F2_series(const double a, const double b, const double c, const double d,
|
||
const double x, gsl_sf_result *result);
|
||
|
||
#if 0
|
||
// The integral from (4.6); maybe should be static and not here.
|
||
int ewald32_sr_integral(double r, double k, double n, double eta, double *result, double *err, gsl_integration_workspace *workspace);
|
||
#endif
|
||
|
||
#include "lattices.h"
|
||
|
||
|
||
int ewald32_sigma0(complex double *result, double *err,
|
||
const qpms_ewald32_constants_t *c,
|
||
double eta, double k
|
||
);
|
||
|
||
// TODO make "compressed versions" where the (m+n)-odd terms (which are zero)
|
||
// are not included.
|
||
|
||
int ewald32_sigma_long_shiftedpoints (
|
||
complex double *target_sigmalr_y, // must be c->nelem_sc long
|
||
double *target_sigmalr_y_err, // must be c->nelem_sc long or NULL
|
||
const qpms_ewald32_constants_t *c,
|
||
double eta, double k, double unitcell_area,
|
||
size_t npoints, const point2d *Kpoints_plus_beta,
|
||
//point2d beta,
|
||
point2d particle_shift
|
||
);
|
||
int ewald32_sigma_long_points_and_shift (//NI
|
||
complex double *target_sigmalr_y, // must be c->nelem_sc long
|
||
double *target_sigmalr_y_err, // must be c->nelem_sc long or NULL
|
||
const qpms_ewald32_constants_t *c,
|
||
double eta, double k, double unitcell_area,
|
||
size_t npoints, const point2d *Kpoints,
|
||
point2d beta,
|
||
point2d particle_shift
|
||
);
|
||
int ewald32_sigma_long_shiftedpoints_rordered(//NI
|
||
complex double *target_sigmalr_y, // must be c->nelem_sc long
|
||
double *target_sigmalr_y_err, // must be c->nelem_sc long or NULL
|
||
const qpms_ewald32_constants_t *c,
|
||
double eta, double k, double unitcell_area,
|
||
const points2d_rordered_t *Kpoints_plus_beta_rordered,
|
||
point2d particle_shift
|
||
);
|
||
|
||
int ewald32_sigma_short_shiftedpoints(
|
||
complex double *target_sigmasr_y, // must be c->nelem_sc long
|
||
double *target_sigmasr_y_err, // must be c->nelem_sc long or NULL
|
||
const qpms_ewald32_constants_t *c, // N.B. not too useful here
|
||
double eta, double k,
|
||
size_t npoints, const point2d *Rpoints_plus_particle_shift,
|
||
point2d beta,
|
||
point2d particle_shift // used only in the very end to multiply it by the phase
|
||
);
|
||
int ewald32_sigma_short_points_and_shift(//NI
|
||
complex double *target_sigmasr_y, // must be c->nelem_sc long
|
||
double *target_sigmasr_y_err, // must be c->nelem_sc long or NULL
|
||
const qpms_ewald32_constants_t *c, // N.B. not too useful here
|
||
double eta, double k,
|
||
size_t npoints, const point2d *Rpoints,
|
||
point2d particle_shift
|
||
);
|
||
int ewald32_sigma_short_points_rordered(//NI
|
||
complex double *target_sigmasr_y, // must be c->nelem_sc long
|
||
double *target_sigmasr_y_err, // must be c->nelem_sc long or NULL
|
||
const qpms_ewald32_constants_t *c, // N.B. not too useful here
|
||
double eta, double k,
|
||
const points2d_rordered_t *Rpoints_plus_particle_shift_rordered,
|
||
point2d particle_shift // used only in the very end to multiply it by the phase
|
||
);
|
||
|
||
|
||
|
||
#endif //EWALD_H
|