comp-chem/cp2k
1
0
Fork 0
mirror of https://github.com/cp2k/cp2k.git synced 2026-07-29 06:35:28 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

grid: Handle corner case in general collocate

Browse source
This commit is contained in:
Ole Schütt 2020-09-24 08:40:43 +02:00 • committed by Ole Schütt
parent 62596e22f5
commit 50e47f6eaa
1 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/grid/ref/grid_ref_collocate.c
View file

@ -338,12 +338,14 @@ general_ci_to_grid(const int lp, const int j, const int jg, const int k,
const double exp_ab = exp(-zetp * (a + b));
const double exp_2a = exp(-zetp * 2.0 * a);
double exp_2ai = exp(-zetp * 2.0 * a * ismin);
double res = exp(-zetp * ((a * ismin + b) * ismin + c));
bool exp_is_invalid = true;
double exp_2ai = 0.0;
double exp_aiibic = 0.0;
for (int i = ismin; i <= ismax; i++) {
const int ig = map_i[i - index_min[0]];
if (ig < bounds_i[0] || bounds_i[1] < ig) {
exp_is_invalid = true;
continue;
}
@ -351,8 +353,14 @@ general_ci_to_grid(const int lp, const int j, const int jg, const int k,
const int grid_index =
kg * stride + jg * npts_local[0] + ig; // [kg, jg, ig]
if (exp_is_invalid) {
exp_2ai = exp(-zetp * 2.0 * a * i);
exp_aiibic = exp(-zetp * ((a * i + b) * i + c));
exp_is_invalid = false;
}
// polynomial terms
double dip = res; // exp(-zetp * ((a * i + b) * i + c));
double dip = exp_aiibic; // exp(-zetp * ((a * i + b) * i + c));
const double di = i - gp[0];
for (int il = 0; il <= lp; il++) {
grid[grid_index] += ci[il] * dip;
@ -360,7 +368,7 @@ general_ci_to_grid(const int lp, const int j, const int jg, const int k,
}
// update exponential term
res *= exp_2ai * exp_ab;
exp_aiibic *= exp_2ai * exp_ab;
exp_2ai *= exp_2a;
}
}