2022-01-06 00:51:42 +01:00
|
|
|
#+TITLE: Atomic Orbitals
|
2021-04-30 01:26:19 +02:00
|
|
|
#+SETUPFILE: ../tools/theme.setup
|
2021-04-09 11:26:04 +02:00
|
|
|
#+INCLUDE: ../tools/lib.org
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
* Introduction
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
The atomic basis set is defined as a list of shells. Each shell $s$ is
|
|
|
|
centered on a nucleus $A$, possesses a given angular momentum $l$ and a
|
|
|
|
radial function $R_s$. The radial function is a linear combination of
|
2022-01-05 12:26:11 +01:00
|
|
|
/primitive/ functions that can be of type Slater ($p=1$) or
|
2021-04-01 01:19:33 +02:00
|
|
|
Gaussian ($p=2$):
|
|
|
|
|
|
|
|
\[
|
|
|
|
R_s(\mathbf{r}) = \mathcal{N}_s |\mathbf{r}-\mathbf{R}_A|^{n_s}
|
2021-06-03 18:26:00 +02:00
|
|
|
\sum_{k=1}^{N_{\text{prim}}} a_{ks}\, f_{ks}
|
2021-04-30 01:26:19 +02:00
|
|
|
\exp \left( - \gamma_{ks} | \mathbf{r}-\mathbf{R}_A | ^p \right).
|
2021-04-01 01:19:33 +02:00
|
|
|
\]
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
In the case of Gaussian functions, $n_s$ is always zero. The
|
2021-09-06 07:45:10 +02:00
|
|
|
normalization factor $\mathcal{N}_s$ ensures that all the functions of
|
|
|
|
the shell are normalized (integrate) to unity. Usually, basis sets are
|
|
|
|
given a combination of normalized primitives, so the normalization
|
2021-06-03 18:26:00 +02:00
|
|
|
coefficients of the primitives, $f_{ks}$, need also to be provided.
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
Atomic orbitals (AOs) are defined as
|
|
|
|
|
|
|
|
\[
|
2021-06-03 18:26:00 +02:00
|
|
|
\chi_i (\mathbf{r}) = \mathcal{M}_i\, P_{\eta(i)}(\mathbf{r})\, R_{\theta(i)} (\mathbf{r})
|
2021-04-01 01:19:33 +02:00
|
|
|
\]
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-09-06 07:45:10 +02:00
|
|
|
where $\theta(i)$ returns the shell on which the AO is expanded, and
|
|
|
|
$\eta(i)$ denotes which angular function is chosen and $P$ are the
|
|
|
|
generating functions of the given angular momentum $\eta(i)$. Here,
|
|
|
|
the parameter $\mathcal{M}_i$ is an extra parameter which allows the
|
|
|
|
normalization of the different functions of the same shell to be
|
2021-06-03 18:26:00 +02:00
|
|
|
different, as in GAMESS for example.
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-06-03 18:26:00 +02:00
|
|
|
In this section we describe first how the basis set is stored in the
|
|
|
|
context, and then we present the kernels used to compute the values,
|
2021-04-01 01:19:33 +02:00
|
|
|
gradients and Laplacian of the atomic basis functions.
|
|
|
|
|
|
|
|
* Headers :noexport:
|
2021-04-30 01:26:19 +02:00
|
|
|
#+begin_src elisp :noexport :results none
|
2021-04-16 00:57:08 +02:00
|
|
|
(org-babel-lob-ingest "../tools/lib.org")
|
2021-05-09 02:12:38 +02:00
|
|
|
#+end_src
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-10-14 21:40:14 +02:00
|
|
|
#+begin_src c :tangle (eval h_private_func)
|
|
|
|
#ifndef QMCKL_AO_HPF
|
|
|
|
#define QMCKL_AO_HPF
|
2022-03-21 18:32:39 +01:00
|
|
|
#include "qmckl_blas_private_type.h"
|
2021-10-14 21:40:14 +02:00
|
|
|
|
|
|
|
#+end_src
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
#+begin_src c :tangle (eval h_private_type)
|
|
|
|
#ifndef QMCKL_AO_HPT
|
|
|
|
#define QMCKL_AO_HPT
|
2021-05-09 02:12:38 +02:00
|
|
|
|
2021-04-21 01:56:47 +02:00
|
|
|
#include <stdbool.h>
|
2022-02-27 23:31:52 +01:00
|
|
|
#include <stdio.h>
|
2022-03-21 18:32:39 +01:00
|
|
|
#include "qmckl_blas_private_type.h"
|
2021-04-01 01:19:33 +02:00
|
|
|
#+end_src
|
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
#+begin_src f90 :tangle (eval f) :noweb yes
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
|
|
#include "config.h"
|
|
|
|
#endif
|
|
|
|
#+end_src
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
#+begin_src c :tangle (eval c_test) :noweb yes
|
|
|
|
#include "qmckl.h"
|
2021-05-11 16:41:03 +02:00
|
|
|
#include "assert.h"
|
2021-05-10 10:05:50 +02:00
|
|
|
#ifdef HAVE_CONFIG_H
|
2021-05-11 16:41:03 +02:00
|
|
|
#include "config.h"
|
2021-05-09 02:12:38 +02:00
|
|
|
#endif
|
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
#include <stdio.h>
|
2021-06-10 22:57:59 +02:00
|
|
|
#include <math.h>
|
2022-02-16 15:14:41 +01:00
|
|
|
#include <string.h>
|
|
|
|
|
2021-06-03 22:34:23 +02:00
|
|
|
#include "chbrclf.h"
|
2021-07-11 23:20:40 +02:00
|
|
|
#include "qmckl_ao_private_func.h"
|
2021-06-03 22:34:23 +02:00
|
|
|
|
2021-05-11 16:41:03 +02:00
|
|
|
int main() {
|
|
|
|
qmckl_context context;
|
|
|
|
context = qmckl_context_create();
|
2021-04-01 01:19:33 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval c)
|
2021-05-10 10:05:50 +02:00
|
|
|
#ifdef HAVE_CONFIG_H
|
2021-05-10 10:41:59 +02:00
|
|
|
#include "config.h"
|
2021-05-09 02:12:38 +02:00
|
|
|
#endif
|
|
|
|
|
2021-05-10 10:05:50 +02:00
|
|
|
#ifdef HAVE_STDINT_H
|
2021-04-01 01:19:33 +02:00
|
|
|
#include <stdint.h>
|
2021-05-10 10:05:50 +02:00
|
|
|
#elif HAVE_INTTYPES_H
|
|
|
|
#include <inttypes.h>
|
|
|
|
#endif
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
2021-04-21 01:56:47 +02:00
|
|
|
#include <stdbool.h>
|
2021-04-01 01:19:33 +02:00
|
|
|
#include <assert.h>
|
2022-02-16 00:21:37 +01:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <math.h>
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-05-11 13:57:23 +02:00
|
|
|
#include "qmckl.h"
|
2021-04-01 01:19:33 +02:00
|
|
|
#include "qmckl_context_private_type.h"
|
|
|
|
#include "qmckl_memory_private_type.h"
|
2022-03-21 18:32:39 +01:00
|
|
|
#include "qmckl_blas_private_type.h"
|
2021-04-01 01:19:33 +02:00
|
|
|
#include "qmckl_memory_private_func.h"
|
2021-06-10 22:57:59 +02:00
|
|
|
#include "qmckl_ao_private_type.h"
|
|
|
|
#include "qmckl_ao_private_func.h"
|
2022-06-29 13:31:58 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
#+end_src
|
|
|
|
|
2024-01-11 14:33:40 +01:00
|
|
|
#+NAME:fortran_cutoff
|
|
|
|
#+begin_src fortran
|
|
|
|
cutoff = qmckl_get_numprec_precision(context)
|
|
|
|
cutoff = dlog(2.d0**(cutoff-2))
|
|
|
|
#+end_src
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
* Context
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
** Constant data
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
The following arrays are stored in the context, and need to be set
|
|
|
|
when initializing the library:
|
|
|
|
|
|
|
|
#+NAME: constant_data
|
2023-03-13 17:06:41 +01:00
|
|
|
|---------------------------+-----------------------+----------------------------------------------------------------------|
|
|
|
|
| Variable | Type | Description |
|
|
|
|
|---------------------------+-----------------------+----------------------------------------------------------------------|
|
|
|
|
| ~type~ | ~char~ | Gaussian (~'G'~) or Slater (~'S'~) |
|
|
|
|
| ~shell_num~ | ~int64_t~ | Number of shells |
|
|
|
|
| ~prim_num~ | ~int64_t~ | Total number of primitives |
|
|
|
|
| ~nucleus_index~ | ~int64_t[nucl_num]~ | Index of the first shell of each nucleus |
|
|
|
|
| ~nucleus_shell_num~ | ~int64_t[nucl_num]~ | Number of shells per nucleus |
|
|
|
|
| ~shell_ang_mom~ | ~int32_t[shell_num]~ | Angular momentum of each shell |
|
|
|
|
| ~shell_prim_num~ | ~int64_t[shell_num]~ | Number of primitives in each shell |
|
|
|
|
| ~shell_prim_index~ | ~int64_t[shell_num]~ | Address of the first primitive of each shell in the ~EXPONENT~ array |
|
|
|
|
| ~shell_factor~ | ~double[shell_num]~ | Normalization factor for each shell |
|
|
|
|
| ~exponent~ | ~double[prim_num]~ | Array of exponents |
|
|
|
|
| ~coefficient~ | ~double[prim_num]~ | Array of coefficients |
|
|
|
|
| ~prim_factor~ | ~double[prim_num]~ | Normalization factors of the primtives |
|
|
|
|
| ~ao_num~ | ~int64_t~ | Number of AOs |
|
|
|
|
| ~ao_cartesian~ | ~bool~ | If true, use polynomials. Otherwise, use spherical harmonics |
|
|
|
|
| ~ao_factor~ | ~double[ao_num]~ | Normalization factor of the AO |
|
|
|
|
|---------------------------+-----------------------+----------------------------------------------------------------------|
|
|
|
|
|
|
|
|
The following data is computed when the basis is finalized:
|
|
|
|
|
2023-11-30 12:56:06 +01:00
|
|
|
|---------------------------+-----------------------+-------------------------------------------------------------------------------------------------|
|
|
|
|
| Variable | Type | Description |
|
|
|
|
|---------------------------+-----------------------+-------------------------------------------------------------------------------------------------|
|
|
|
|
| ~nucleus_prim_index~ | ~int64_t[nucl_num+1]~ | Index of the first primitive of each nucleus |
|
|
|
|
| ~nucleus_max_ang_mom~ | ~int32_t[nucl_num]~ | Maximum angular momentum of each nucleus |
|
|
|
|
| ~coefficient_normalized~~ | ~double[prim_num]~ | Normalized array of coefficients |
|
|
|
|
| ~ao_ang_mom~ | ~int32_t[ao_num]~ | Angular momentum of the shell to which the AO belongs |
|
|
|
|
| ~ao_nucl~ | ~int64_t[ao_num]~ | Nucleus on which the AO is centered |
|
|
|
|
| ~nucleus_range~ | ~double[nucl_num]~ | Used to compute the distance beyond which all Gaussian AOs are zero, depending on the precision |
|
|
|
|
|---------------------------+-----------------------+-------------------------------------------------------------------------------------------------|
|
2022-01-05 15:56:25 +01:00
|
|
|
|
|
|
|
For H_2 with the following basis set,
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+NAME: basis
|
|
|
|
#+BEGIN_EXAMPLE
|
2021-04-01 01:19:33 +02:00
|
|
|
HYDROGEN
|
|
|
|
S 5
|
2022-01-05 12:26:11 +01:00
|
|
|
1 3.387000E+01 6.068000E-03
|
|
|
|
2 5.095000E+00 4.530800E-02
|
|
|
|
3 1.159000E+00 2.028220E-01
|
|
|
|
4 3.258000E-01 5.039030E-01
|
|
|
|
5 1.027000E-01 3.834210E-01
|
2021-04-01 01:19:33 +02:00
|
|
|
S 1
|
2022-01-05 12:26:11 +01:00
|
|
|
1 3.258000E-01 1.000000E+00
|
2021-04-01 01:19:33 +02:00
|
|
|
S 1
|
2022-01-05 12:26:11 +01:00
|
|
|
1 1.027000E-01 1.000000E+00
|
2021-04-01 01:19:33 +02:00
|
|
|
P 1
|
2022-01-05 12:26:11 +01:00
|
|
|
1 1.407000E+00 1.000000E+00
|
2021-04-01 01:19:33 +02:00
|
|
|
P 1
|
2022-01-05 12:26:11 +01:00
|
|
|
1 3.880000E-01 1.000000E+00
|
2021-04-01 01:19:33 +02:00
|
|
|
D 1
|
2022-01-05 12:26:11 +01:00
|
|
|
1 1.057000E+00 1.0000000
|
2022-01-05 15:56:25 +01:00
|
|
|
#+END_EXAMPLE
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
we have:
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+NAME: params
|
|
|
|
#+BEGIN_EXAMPLE
|
2021-04-01 01:19:33 +02:00
|
|
|
type = 'G'
|
|
|
|
shell_num = 12
|
|
|
|
prim_num = 20
|
2023-05-02 14:24:30 +02:00
|
|
|
ao_num = 30
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-10 00:10:19 +02:00
|
|
|
nucleus_index = [0 , 6]
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-03 18:26:00 +02:00
|
|
|
shell_ang_mom = [0, 0, 0, 1, 1, 2, 0, 0, 0, 1, 1, 2]
|
2021-04-01 01:19:33 +02:00
|
|
|
shell_factor = [ 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.]
|
|
|
|
shell_prim_num = [5, 1, 1, 1, 1, 1, 5, 1, 1, 1, 1, 1]
|
2021-06-10 00:10:19 +02:00
|
|
|
shell_prim_index = [0 , 5 , 6 , 7 , 8 , 9 , 10, 15, 16, 17, 18, 19]
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-03 18:26:00 +02:00
|
|
|
exponent = [ 33.87, 5.095, 1.159, 0.3258, 0.1027, 0.3258, 0.1027, 1.407,
|
|
|
|
0.388, 1.057, 33.87, 5.095, 1.159, 0.3258, 0.1027, 0.3258, 0.1027, 1.407,
|
|
|
|
0.388, 1.057]
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-03 18:26:00 +02:00
|
|
|
coefficient = [ 0.006068, 0.045308, 0.202822, 0.503903, 0.383421, 1.0, 1.0,
|
|
|
|
1.0, 1.0, 1.0, 0.006068, 0.045308, 0.202822, 0.503903, 0.383421, 1.0, 1.0,
|
|
|
|
1.0, 1.0, 1.0]
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-03 18:26:00 +02:00
|
|
|
prim_factor = [ 1.0006253235944540e+01, 2.4169531573445120e+00, 7.9610924849766440e-01
|
|
|
|
3.0734305383061117e-01, 1.2929684417481876e-01, 3.0734305383061117e-01,
|
|
|
|
1.2929684417481876e-01, 2.1842769845268308e+00, 4.3649547399719840e-01,
|
|
|
|
1.8135965626177861e+00, 1.0006253235944540e+01, 2.4169531573445120e+00,
|
|
|
|
7.9610924849766440e-01, 3.0734305383061117e-01, 1.2929684417481876e-01,
|
|
|
|
3.0734305383061117e-01, 1.2929684417481876e-01, 2.1842769845268308e+00,
|
|
|
|
4.3649547399719840e-01, 1.8135965626177861e+00 ]
|
2022-01-05 15:56:25 +01:00
|
|
|
#+END_EXAMPLE
|
|
|
|
|
|
|
|
A scalar variable ~$V$~ present in this table can be set or get by
|
2022-01-06 02:28:13 +01:00
|
|
|
calling the functions:
|
2022-01-05 15:56:25 +01:00
|
|
|
|
|
|
|
#+NAME: template_scalar_c
|
|
|
|
#+begin_src C :tangle no
|
|
|
|
qmckl_exit_code qmckl_set_ao_basis_$V$ ( qmckl_context context,
|
|
|
|
const $type_of_V$ $V$);
|
|
|
|
|
|
|
|
qmckl_exit_code qmckl_get_ao_basis_$V$ ( const qmckl_context context,
|
|
|
|
$type_of_V$ const $V$);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle no
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_$V$ (context, $V$) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
$f_type_of_V$ , intent(in) , value :: $V$
|
|
|
|
end function qmckl_set_ao_basis_$V$
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_$V$ (context, $V$) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
$f_type_of_V$ , intent(out) :: $V$
|
|
|
|
end function qmckl_get_ao_basis_$V$
|
|
|
|
end interface
|
|
|
|
#+end_src
|
|
|
|
For array variables, use the rule:
|
|
|
|
|
|
|
|
#+NAME: template_array_c
|
|
|
|
#+begin_src C :tangle no
|
|
|
|
qmckl_exit_code qmckl_set_ao_basis_$V$ ( qmckl_context context,
|
|
|
|
const $type_of_V$ $V$,
|
|
|
|
const int64_t size_max);
|
|
|
|
|
|
|
|
qmckl_exit_code qmckl_get_ao_basis_$V$ ( const qmckl_context context,
|
|
|
|
$type_of_V$ const $V$,
|
|
|
|
const int64_t size_max);
|
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src f90 :tangle no
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_$V$ (context, &
|
|
|
|
$V$, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
$f_type_of_V$ , intent(in) , value :: $V$
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_$V$
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_$V$ (context, &
|
|
|
|
$V$, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
$f_type_of_V$ , intent(out) :: $V$
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_get_ao_basis_$V$
|
|
|
|
end interface
|
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Data structure :noexport:
|
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_private_type) :noweb yes
|
2021-04-01 01:19:33 +02:00
|
|
|
typedef struct qmckl_ao_basis_struct {
|
2022-03-21 18:32:39 +01:00
|
|
|
int64_t shell_num;
|
|
|
|
int64_t prim_num;
|
|
|
|
int64_t ao_num;
|
|
|
|
int64_t * restrict nucleus_index;
|
|
|
|
int64_t * restrict nucleus_shell_num;
|
|
|
|
int32_t * restrict shell_ang_mom;
|
|
|
|
int64_t * restrict shell_prim_num;
|
|
|
|
int64_t * restrict shell_prim_index;
|
|
|
|
double * restrict shell_factor;
|
|
|
|
double * restrict exponent;
|
|
|
|
double * restrict coefficient;
|
|
|
|
double * restrict prim_factor;
|
|
|
|
double * restrict ao_factor;
|
2023-03-13 17:06:41 +01:00
|
|
|
int64_t * restrict ao_nucl;
|
|
|
|
int32_t * restrict ao_ang_mom;
|
2022-03-21 18:32:39 +01:00
|
|
|
|
|
|
|
int64_t * restrict nucleus_prim_index;
|
|
|
|
double * restrict coefficient_normalized;
|
|
|
|
int32_t * restrict nucleus_max_ang_mom;
|
|
|
|
double * restrict nucleus_range;
|
|
|
|
double * restrict primitive_vgl;
|
|
|
|
uint64_t primitive_vgl_date;
|
|
|
|
double * restrict shell_vgl;
|
|
|
|
uint64_t shell_vgl_date;
|
|
|
|
double * restrict ao_vgl;
|
|
|
|
uint64_t ao_vgl_date;
|
2022-05-10 19:18:19 +02:00
|
|
|
double * restrict ao_value;
|
|
|
|
uint64_t ao_value_date;
|
2022-03-21 18:32:39 +01:00
|
|
|
|
|
|
|
int32_t uninitialized;
|
|
|
|
bool provided;
|
|
|
|
bool ao_cartesian;
|
|
|
|
char type;
|
2022-02-25 20:39:20 +01:00
|
|
|
#ifdef HAVE_HPC
|
|
|
|
<<HPC_struct>>
|
|
|
|
#endif
|
2021-04-01 01:19:33 +02:00
|
|
|
} qmckl_ao_basis_struct;
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
The ~uninitialized~ integer contains one bit set to one for each
|
|
|
|
initialization function which has not been called. It becomes equal
|
|
|
|
to zero after all initialization functions have been called. The
|
|
|
|
struct is then initialized and ~provided == true~.
|
|
|
|
Some values are initialized by default, and are not concerned by
|
|
|
|
this mechanism.
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_private_func)
|
2021-06-10 22:57:59 +02:00
|
|
|
qmckl_exit_code qmckl_init_ao_basis(qmckl_context context);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c)
|
2021-06-10 22:57:59 +02:00
|
|
|
qmckl_exit_code qmckl_init_ao_basis(qmckl_context context) {
|
|
|
|
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_init_ao_basis",
|
|
|
|
NULL);
|
2021-06-10 22:57:59 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2021-06-10 22:57:59 +02:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
2021-07-09 00:45:17 +02:00
|
|
|
ctx->ao_basis.uninitialized = (1 << 14) - 1;
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
/* Default values */
|
2021-07-09 00:45:17 +02:00
|
|
|
ctx->ao_basis.ao_cartesian = true;
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Initialization functions
|
2022-01-05 19:22:16 +01:00
|
|
|
|
|
|
|
~size_max~ is the dimension of the input array, which should be
|
|
|
|
equal of larger than the value given in the table of section [[Context]].
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
**** C interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+NAME:pre2
|
|
|
|
#+begin_src c :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
2022-07-08 09:15:17 +02:00
|
|
|
QMCKL_NULL_CONTEXT,
|
2022-07-08 09:24:33 +02:00
|
|
|
"qmckl_set_ao_*",
|
2022-01-05 12:26:11 +01:00
|
|
|
NULL);
|
2022-07-08 09:24:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-07-08 09:15:17 +02:00
|
|
|
|
2022-07-09 11:34:38 +02:00
|
|
|
if (mask != 0 && !(ctx->ao_basis.uninitialized & mask)) {
|
2022-07-08 09:15:17 +02:00
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALREADY_SET,
|
2022-07-08 09:24:33 +02:00
|
|
|
"qmckl_set_ao_*",
|
2022-07-08 09:15:17 +02:00
|
|
|
NULL);
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+NAME:post2
|
|
|
|
#+begin_src c :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.uninitialized &= ~mask;
|
|
|
|
ctx->ao_basis.provided = (ctx->ao_basis.uninitialized == 0);
|
|
|
|
if (ctx->ao_basis.provided) {
|
|
|
|
qmckl_exit_code rc_ = qmckl_finalize_basis(context);
|
|
|
|
if (rc_ != QMCKL_SUCCESS) return rc_;
|
|
|
|
}
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
To set the basis set, all the following functions need to be
|
|
|
|
called.
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_type (qmckl_context context,
|
|
|
|
const char basis_type);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_type(qmckl_context context,
|
|
|
|
const char basis_type)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-07-08 09:15:17 +02:00
|
|
|
int32_t mask = 1;
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<pre2>>
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
if (basis_type != 'G' && basis_type != 'S') {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_type",
|
2021-10-14 16:32:58 +02:00
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.type = basis_type;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_num (qmckl_context context,
|
|
|
|
const int64_t shell_num);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_num (qmckl_context context,
|
2022-01-06 02:28:13 +01:00
|
|
|
const int64_t shell_num)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-07-08 09:15:17 +02:00
|
|
|
int32_t mask = 1 << 1;
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<pre2>>
|
|
|
|
|
|
|
|
if (shell_num <= 0) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"shell_num <= 0");
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t prim_num = ctx->ao_basis.prim_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (0L < prim_num && prim_num < shell_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"shell_num > prim_num");
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.shell_num = shell_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_prim_num (qmckl_context context,
|
|
|
|
const int64_t prim_num);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_prim_num (qmckl_context context,
|
|
|
|
const int64_t prim_num)
|
2022-01-06 02:28:13 +01:00
|
|
|
{
|
2022-07-08 09:15:17 +02:00
|
|
|
int32_t mask = 1 << 2;
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<pre2>>
|
|
|
|
|
|
|
|
if (prim_num <= 0) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"prim_num must be positive");
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t shell_num = ctx->ao_basis.shell_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_num <= 0L) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"shell_num is not set");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (prim_num < shell_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"prim_num < shell_num");
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.prim_num = prim_num;
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-07-08 09:15:17 +02:00
|
|
|
qmckl_set_ao_basis_nucleus_shell_num (qmckl_context context,
|
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-07-08 09:15:17 +02:00
|
|
|
qmckl_set_ao_basis_nucleus_shell_num (qmckl_context context,
|
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const int64_t size_max)
|
2022-01-06 02:28:13 +01:00
|
|
|
{
|
2022-07-08 09:15:17 +02:00
|
|
|
int32_t mask = 1 << 3;
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<pre2>>
|
2021-06-10 00:10:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t nucl_num = ctx->nucleus.num;
|
|
|
|
|
|
|
|
if (nucl_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
2022-07-08 09:15:17 +02:00
|
|
|
"qmckl_set_ao_basis_nucleus_shell_num",
|
|
|
|
"shell_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < nucl_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-07-08 09:15:17 +02:00
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_set_ao_basis_nucleus_shell_num",
|
2022-01-05 12:26:11 +01:00
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
if (ctx->ao_basis.nucleus_shell_num != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.nucleus_shell_num);
|
2022-01-05 12:26:11 +01:00
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
2022-07-08 09:15:17 +02:00
|
|
|
"qmckl_set_ao_basis_nucleus_shell_num",
|
2022-01-05 12:26:11 +01:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = nucl_num * sizeof(int64_t);
|
|
|
|
int64_t* new_array = (int64_t*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
2022-07-08 09:15:17 +02:00
|
|
|
"qmckl_set_ao_basis_nucleus_shell_num",
|
2022-01-05 12:26:11 +01:00
|
|
|
NULL);
|
2021-06-10 00:10:19 +02:00
|
|
|
}
|
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
memcpy(new_array, nucleus_shell_num, mem_info.size);
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
ctx->ao_basis.nucleus_shell_num = new_array;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-07-08 09:15:17 +02:00
|
|
|
qmckl_set_ao_basis_nucleus_index (qmckl_context context,
|
|
|
|
const int64_t* nucleus_index,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-07-08 09:15:17 +02:00
|
|
|
qmckl_set_ao_basis_nucleus_index (qmckl_context context,
|
|
|
|
const int64_t* nucleus_index,
|
|
|
|
const int64_t size_max)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-07-08 09:15:17 +02:00
|
|
|
int32_t mask = 1 << 4;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t nucl_num = ctx->nucleus.num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (nucl_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
2022-07-08 09:15:17 +02:00
|
|
|
"qmckl_set_ao_basis_nucleus_index",
|
|
|
|
"nucl_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < nucl_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-07-08 09:15:17 +02:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_nucleus_index",
|
2022-01-05 12:26:11 +01:00
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
if (ctx->ao_basis.nucleus_index != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.nucleus_index);
|
2022-01-05 12:26:11 +01:00
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
2022-07-08 09:15:17 +02:00
|
|
|
"qmckl_set_ao_basis_nucleus_index",
|
2022-01-05 12:26:11 +01:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = nucl_num * sizeof(int64_t);
|
|
|
|
int64_t* new_array = (int64_t*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
2022-07-08 09:15:17 +02:00
|
|
|
"qmckl_set_ao_basis_nucleus_index",
|
2022-01-05 12:26:11 +01:00
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
memcpy(new_array, nucleus_index, mem_info.size);
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
ctx->ao_basis.nucleus_index = new_array;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_ang_mom (qmckl_context context,
|
|
|
|
const int32_t* shell_ang_mom,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_ang_mom (qmckl_context context,
|
|
|
|
const int32_t* shell_ang_mom,
|
2022-01-06 02:28:13 +01:00
|
|
|
const int64_t size_max)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 5;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t shell_num = ctx->ao_basis.shell_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_num == 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_shell_ang_mom",
|
|
|
|
"shell_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < shell_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_shell_ang_mom",
|
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.shell_ang_mom != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.shell_ang_mom);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_shell_ang_mom",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = shell_num * sizeof(int32_t);
|
|
|
|
int32_t * new_array = (int32_t*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_shell_ang_mom",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, shell_ang_mom, mem_info.size);
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.shell_ang_mom = new_array;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_prim_num (qmckl_context context,
|
|
|
|
const int64_t* shell_prim_num,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_prim_num (qmckl_context context,
|
|
|
|
const int64_t* shell_prim_num,
|
|
|
|
const int64_t size_max)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 6;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t shell_num = ctx->ao_basis.shell_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_shell_prim_num",
|
|
|
|
"shell_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < shell_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_shell_prim_num",
|
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.shell_prim_num != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.shell_prim_num);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_shell_prim_num",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = shell_num * sizeof(int64_t);
|
|
|
|
int64_t* new_array = (int64_t*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_shell_prim_num",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, shell_prim_num, mem_info.size);
|
|
|
|
|
|
|
|
ctx->ao_basis.shell_prim_num = new_array;
|
|
|
|
|
|
|
|
<<post2>>
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_prim_index (qmckl_context context,
|
|
|
|
const int64_t* shell_prim_index,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_prim_index (qmckl_context context,
|
|
|
|
const int64_t* shell_prim_index,
|
|
|
|
const int64_t size_max)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 7;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t shell_num = ctx->ao_basis.shell_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_shell_prim_index",
|
|
|
|
"shell_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < shell_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_shell_prim_index",
|
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.shell_prim_index != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.shell_prim_index);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_shell_prim_index",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = shell_num * sizeof(int64_t);
|
|
|
|
int64_t* new_array = (int64_t*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_shell_prim_index",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, shell_prim_index, mem_info.size);
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.shell_prim_index = new_array;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-03-21 18:32:39 +01:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_factor (qmckl_context context,
|
|
|
|
const double* shell_factor,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_shell_factor (qmckl_context context,
|
|
|
|
const double* shell_factor,
|
|
|
|
const int64_t size_max)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 8;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t shell_num = ctx->ao_basis.shell_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_shell_factor",
|
|
|
|
"shell_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < shell_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_shell_factor",
|
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.shell_factor != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.shell_factor);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_shell_factor",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = shell_num * sizeof(double);
|
|
|
|
double* new_array = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_shell_factor",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, shell_factor, mem_info.size);
|
|
|
|
|
|
|
|
ctx->ao_basis.shell_factor = new_array;
|
|
|
|
|
|
|
|
<<post2>>
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-03-21 18:32:39 +01:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_exponent (qmckl_context context,
|
|
|
|
const double* exponent,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_exponent (qmckl_context context,
|
|
|
|
const double* exponent,
|
|
|
|
const int64_t size_max)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 9;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t prim_num = ctx->ao_basis.prim_num;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (prim_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_exponent",
|
|
|
|
"prim_num is not set");
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < prim_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_exponent",
|
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.exponent != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.exponent);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_exponent",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = prim_num * sizeof(double);
|
|
|
|
double* new_array = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_exponent",
|
|
|
|
NULL);
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, exponent, mem_info.size);
|
|
|
|
|
|
|
|
ctx->ao_basis.exponent = new_array;
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_coefficient (qmckl_context context,
|
|
|
|
const double* coefficient,
|
2021-10-14 16:32:58 +02:00
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_coefficient (qmckl_context context,
|
|
|
|
const double* coefficient,
|
2021-10-14 16:32:58 +02:00
|
|
|
const int64_t size_max)
|
|
|
|
{
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 10;
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t prim_num = ctx->ao_basis.prim_num;
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (prim_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_coefficient",
|
|
|
|
"prim_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < prim_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_coefficient",
|
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.coefficient != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.coefficient);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_coefficient",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = prim_num * sizeof(double);
|
|
|
|
double* new_array = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_coefficient",
|
|
|
|
NULL);
|
2021-06-03 18:26:00 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, coefficient, mem_info.size);
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.coefficient = new_array;
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_prim_factor (qmckl_context context,
|
|
|
|
const double* prim_factor,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_prim_factor (qmckl_context context,
|
|
|
|
const double* prim_factor,
|
|
|
|
const int64_t size_max)
|
2021-10-14 16:32:58 +02:00
|
|
|
{
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 11;
|
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t prim_num = ctx->ao_basis.prim_num;
|
|
|
|
|
|
|
|
if (prim_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_prim_factor",
|
|
|
|
"prim_num is not set");
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < prim_num) {
|
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_prim_factor",
|
|
|
|
"input array too small");
|
|
|
|
}
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.prim_factor != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.prim_factor);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_prim_factor",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = prim_num * sizeof(double);
|
|
|
|
double* new_array = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_prim_factor",
|
2021-10-14 16:32:58 +02:00
|
|
|
NULL);
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, prim_factor, mem_info.size);
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
ctx->ao_basis.prim_factor = new_array;
|
|
|
|
|
|
|
|
<<post2>>
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-07-09 00:45:17 +02:00
|
|
|
|
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_set_ao_basis_ao_num (qmckl_context context,
|
|
|
|
const int64_t ao_num);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_set_ao_basis_ao_num (qmckl_context context,
|
|
|
|
const int64_t ao_num)
|
|
|
|
{
|
|
|
|
int32_t mask = 1 << 12;
|
|
|
|
|
|
|
|
<<pre2>>
|
|
|
|
|
|
|
|
if (ao_num <= 0) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"ao_num must be positive");
|
|
|
|
}
|
|
|
|
|
|
|
|
const int64_t shell_num = ctx->ao_basis.shell_num;
|
|
|
|
if (shell_num <= 0L) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"shell_num is not set");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ao_num < shell_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_set_ao_basis_shell_num",
|
|
|
|
"ao_num < shell_num");
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_num = ao_num;
|
|
|
|
|
|
|
|
<<post2>>
|
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_set_ao_basis_ao_factor (qmckl_context context,
|
|
|
|
const double* ao_factor,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-10-14 16:32:58 +02:00
|
|
|
qmckl_exit_code
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_set_ao_basis_ao_factor (qmckl_context context,
|
|
|
|
const double* ao_factor,
|
2021-10-14 16:32:58 +02:00
|
|
|
const int64_t size_max)
|
2022-01-06 02:28:13 +01:00
|
|
|
{
|
2021-07-09 00:45:17 +02:00
|
|
|
int32_t mask = 1 << 13;
|
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
<<pre2>>
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
const int64_t ao_num = ctx->ao_basis.ao_num;
|
|
|
|
|
|
|
|
if (ao_num <= 0L) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_FAILURE,
|
|
|
|
"qmckl_set_ao_basis_ao_factor",
|
|
|
|
"ao_num is not set");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ao_num) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_set_ao_basis_ao_factor",
|
|
|
|
"input array too small");
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ctx->ao_basis.ao_factor != NULL) {
|
|
|
|
qmckl_exit_code rc = qmckl_free(context, ctx->ao_basis.ao_factor);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc,
|
|
|
|
"qmckl_set_ao_basis_ao_factor",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ao_num * sizeof(double);
|
|
|
|
double* new_array = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (new_array == NULL) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_set_ao_basis_ao_factor",
|
|
|
|
NULL);
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
memcpy(new_array, ao_factor, mem_info.size);
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_factor = new_array;
|
|
|
|
|
|
|
|
<<post2>>
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_set_ao_basis_cartesian (qmckl_context context,
|
|
|
|
const bool cartesian);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_set_ao_basis_cartesian (qmckl_context context,
|
|
|
|
const bool cartesian)
|
|
|
|
{
|
2022-07-08 09:15:17 +02:00
|
|
|
int32_t mask = 1;
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
<<pre2>>
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
ctx->ao_basis.ao_cartesian = cartesian;
|
2022-01-05 12:26:11 +01:00
|
|
|
|
|
|
|
<<post2>>
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
**** Fortran interface
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_type (context, &
|
|
|
|
basis_type) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
character(c_char) , intent(in) , value :: basis_type
|
|
|
|
end function qmckl_set_ao_basis_type
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_shell_num(context, &
|
|
|
|
num) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: num
|
|
|
|
end function qmckl_set_ao_basis_shell_num
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_prim_num(context, &
|
|
|
|
num) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: num
|
|
|
|
end function qmckl_set_ao_basis_prim_num
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_nucleus_index(context, &
|
|
|
|
idx, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) :: idx(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_nucleus_index
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_nucleus_shell_num(context, &
|
|
|
|
shell_num, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) :: shell_num(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_nucleus_shell_num
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_shell_ang_mom(context, &
|
|
|
|
shell_ang_mom, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int32_t) , intent(in) :: shell_ang_mom(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_shell_ang_mom
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_shell_prim_num(context, &
|
|
|
|
shell_prim_num, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) :: shell_prim_num(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_shell_prim_num
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_shell_prim_index(context, &
|
|
|
|
shell_prim_index, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) :: shell_prim_index(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_shell_prim_index
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_shell_factor(context, &
|
|
|
|
shell_factor, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
real (c_double) , intent(in) :: shell_factor(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_shell_factor
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_exponent(context, &
|
|
|
|
exponent, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
real (c_double) , intent(in) :: exponent(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_exponent
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_coefficient(context, &
|
|
|
|
coefficient, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
real (c_double) , intent(in) :: coefficient(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_coefficient
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_prim_factor(context, &
|
|
|
|
prim_factor, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
real (c_double) , intent(in) :: prim_factor(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_prim_factor
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_ao_num(context, &
|
|
|
|
num) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: num
|
|
|
|
end function qmckl_set_ao_basis_ao_num
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_cartesian(context, &
|
|
|
|
cartesian) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
logical (c_bool) , intent(in) , value :: cartesian
|
|
|
|
end function qmckl_set_ao_basis_cartesian
|
|
|
|
end interface
|
|
|
|
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_set_ao_basis_ao_factor(context, &
|
|
|
|
ao_factor, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
|
|
|
real (c_double) , intent(in) :: ao_factor(*)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_set_ao_basis_ao_factor
|
|
|
|
end interface
|
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Access functions
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
~size_max~ is the dimension of the input array, which should be
|
2022-01-05 19:22:16 +01:00
|
|
|
equal of larger than the value given in the table of section [[Context]].
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
**** C interface
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_type (const qmckl_context context,
|
2022-01-06 02:28:13 +01:00
|
|
|
char* const basis_type);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_type (const qmckl_context context,
|
|
|
|
char* const basis_type)
|
|
|
|
{
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_type",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1;
|
|
|
|
|
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_type",
|
|
|
|
NULL);
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.type != (char) 0);
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
basis_type[0] = ctx->ao_basis.type;
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_num (const qmckl_context context,
|
|
|
|
int64_t* const shell_num);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_num (const qmckl_context context,
|
|
|
|
int64_t* const shell_num)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_shell_factor",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 1;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_shell_num",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.shell_num > (int64_t) 0);
|
|
|
|
,*shell_num = ctx->ao_basis.shell_num;
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_prim_num (const qmckl_context context,
|
|
|
|
int64_t* const prim_num);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_prim_num (const qmckl_context context,
|
|
|
|
int64_t* const prim_num)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_prim_num",
|
|
|
|
NULL);
|
2021-06-10 00:10:19 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-06-10 00:10:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 2;
|
2021-06-10 00:10:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-06-10 00:10:19 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_prim_num",
|
2021-06-10 00:10:19 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.prim_num > (int64_t) 0);
|
2021-06-10 00:10:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
,*prim_num = ctx->ao_basis.prim_num;
|
|
|
|
return QMCKL_SUCCESS;
|
2021-06-10 00:10:19 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-06-10 00:10:19 +02:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_nucleus_shell_num (const qmckl_context context,
|
|
|
|
int64_t* const nucleus_shell_num,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_nucleus_shell_num (const qmckl_context context,
|
|
|
|
int64_t* const nucleus_shell_num,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-06-10 00:10:19 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_nucleus_shell_num",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 3;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_nucleus_shell_num",
|
2021-04-01 01:19:33 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (nucleus_shell_num == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_nucleus_shell_num",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->nucleus.num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_nucleus_shell_num",
|
|
|
|
"Array too small. Expected nucl_num");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.nucleus_shell_num != NULL);
|
|
|
|
memcpy(nucleus_shell_num, ctx->ao_basis.nucleus_shell_num,
|
|
|
|
(size_t) ctx->nucleus.num * sizeof(int64_t));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_nucleus_index (const qmckl_context context,
|
|
|
|
int64_t* const nucleus_index,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_nucleus_index (const qmckl_context context,
|
|
|
|
int64_t* const nucleus_index,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_nucleus_index",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 4;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_nucleus_index",
|
2021-04-01 01:19:33 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (nucleus_index == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_nucleus_index",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->nucleus.num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_nucleus_index",
|
|
|
|
"Array too small. Expected shell_num");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.nucleus_index != NULL);
|
|
|
|
memcpy(nucleus_index, ctx->ao_basis.nucleus_index,
|
|
|
|
(size_t) ctx->nucleus.num * sizeof(int64_t));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_ang_mom (const qmckl_context context,
|
|
|
|
int32_t* const shell_ang_mom,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_ang_mom (const qmckl_context context,
|
|
|
|
int32_t* const shell_ang_mom,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_shell_ang_mom",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 5;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_shell_ang_mom",
|
2021-04-01 01:19:33 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_ang_mom == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_shell_ang_mom",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.shell_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_shell_ang_mom",
|
|
|
|
"Array too small. Expected shell_num");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.shell_ang_mom != NULL);
|
|
|
|
memcpy(shell_ang_mom, ctx->ao_basis.shell_ang_mom,
|
|
|
|
(size_t) ctx->ao_basis.shell_num * sizeof(int32_t));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_prim_num (const qmckl_context context,
|
|
|
|
int64_t* const shell_prim_num,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_prim_num (const qmckl_context context,
|
|
|
|
int64_t* const shell_prim_num,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_num",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 6;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_num",
|
2021-04-01 01:19:33 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_prim_num == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_num",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.shell_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_num",
|
|
|
|
"Array too small. Expected shell_num");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.shell_prim_num != NULL);
|
|
|
|
memcpy(shell_prim_num, ctx->ao_basis.shell_prim_num,
|
|
|
|
(size_t) ctx->ao_basis.shell_num * sizeof(int64_t));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_prim_index (const qmckl_context context,
|
|
|
|
int64_t* const shell_prim_index,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_prim_index (const qmckl_context context,
|
|
|
|
int64_t* const shell_prim_index,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_index",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 7;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_index",
|
2021-04-01 01:19:33 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_prim_index == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_index",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.shell_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_shell_prim_index",
|
|
|
|
"Array too small. Expected shell_num");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.shell_prim_index != NULL);
|
|
|
|
memcpy(shell_prim_index, ctx->ao_basis.shell_prim_index,
|
|
|
|
(size_t) ctx->ao_basis.shell_num * sizeof(int64_t));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_factor (const qmckl_context context,
|
|
|
|
double* const shell_factor,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_factor (const qmckl_context context,
|
|
|
|
double* const shell_factor,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_shell_factor",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 8;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_shell_factor",
|
2021-04-01 01:19:33 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (shell_factor == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_shell_factor",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.shell_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_shell_factor",
|
|
|
|
"Array too small. Expected shell_num");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.shell_factor != NULL);
|
|
|
|
memcpy(shell_factor, ctx->ao_basis.shell_factor,
|
|
|
|
(size_t) ctx->ao_basis.shell_num * sizeof(double));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_exponent (const qmckl_context context,
|
|
|
|
double* const exponent,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_exponent (const qmckl_context context,
|
|
|
|
double* const exponent,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_exponent",
|
|
|
|
NULL);
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 9;
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-04-01 01:19:33 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_exponent",
|
2021-04-01 01:19:33 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (exponent == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_exponent",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.prim_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_exponent",
|
|
|
|
"Array too small. Expected prim_num");
|
|
|
|
}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.exponent != NULL);
|
|
|
|
memcpy(exponent, ctx->ao_basis.exponent,
|
|
|
|
(size_t) ctx->ao_basis.prim_num * sizeof(double));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_coefficient (const qmckl_context context,
|
|
|
|
double* const coefficient,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_coefficient (const qmckl_context context,
|
|
|
|
double* const coefficient,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-06-03 18:26:00 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_coefficient",
|
|
|
|
NULL);
|
2021-06-03 18:26:00 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 10;
|
|
|
|
|
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_coefficient",
|
|
|
|
NULL);
|
2021-06-03 18:26:00 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (coefficient == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_coefficient",
|
|
|
|
"NULL pointer");
|
|
|
|
}
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.prim_num) {
|
2021-06-03 18:26:00 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_coefficient",
|
|
|
|
"Array too small. Expected prim_num");
|
2021-06-03 18:26:00 +02:00
|
|
|
}
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.coefficient != NULL);
|
|
|
|
memcpy(coefficient, ctx->ao_basis.coefficient,
|
|
|
|
(size_t) ctx->ao_basis.prim_num * sizeof(double));
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2021-06-03 18:26:00 +02:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_prim_factor (const qmckl_context context,
|
|
|
|
double* const prim_factor,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-06-03 18:26:00 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_prim_factor (const qmckl_context context,
|
|
|
|
double* const prim_factor,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-07-09 00:45:17 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_prim_factor",
|
|
|
|
NULL);
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
int32_t mask = 1 << 11;
|
|
|
|
|
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_prim_factor",
|
|
|
|
NULL);
|
2021-10-14 16:32:58 +02:00
|
|
|
}
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (prim_factor == NULL) {
|
2021-07-09 00:45:17 +02:00
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_get_ao_basis_prim_factor",
|
|
|
|
"NULL pointer");
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.prim_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_prim_factor",
|
|
|
|
"Array too small. Expected prim_num");
|
|
|
|
}
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.prim_factor != NULL);
|
|
|
|
memcpy(prim_factor, ctx->ao_basis.prim_factor,
|
|
|
|
(size_t) ctx->ao_basis.prim_num * sizeof(double));
|
|
|
|
return QMCKL_SUCCESS;
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-07-09 00:45:17 +02:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_num (const qmckl_context context,
|
|
|
|
int64_t* const ao_num);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_num (const qmckl_context context,
|
|
|
|
int64_t* const ao_num)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-07-09 00:45:17 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_ao_num",
|
|
|
|
NULL);
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 12;
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
2021-07-09 00:45:17 +02:00
|
|
|
return qmckl_failwith( context,
|
2022-01-05 12:26:11 +01:00
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_ao_num",
|
2021-07-09 00:45:17 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.ao_num > (int64_t) 0);
|
2021-07-09 00:45:17 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
*ao_num = ctx->ao_basis.ao_num;
|
|
|
|
return QMCKL_SUCCESS;
|
2021-07-09 00:45:17 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-06-10 00:10:19 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_factor (const qmckl_context context,
|
|
|
|
double* const ao_factor,
|
|
|
|
const int64_t size_max);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-06-10 00:10:19 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_factor (const qmckl_context context,
|
|
|
|
double* const ao_factor,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
2021-06-10 00:10:19 +02:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
2021-10-14 16:32:58 +02:00
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
2022-01-05 12:26:11 +01:00
|
|
|
"qmckl_get_ao_basis_ao_factor",
|
2021-10-14 16:32:58 +02:00
|
|
|
NULL);
|
2021-06-10 00:10:19 +02:00
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2021-06-10 00:10:19 +02:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
int32_t mask = 1 << 13;
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if ( (ctx->ao_basis.uninitialized & mask) != 0) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_get_ao_basis_ao_factor",
|
|
|
|
NULL);
|
2021-06-10 22:57:59 +02:00
|
|
|
}
|
2021-07-23 12:27:18 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (ao_factor == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_2,
|
|
|
|
"qmckl_get_ao_basis_ao_factor",
|
|
|
|
"NULL pointer");
|
2021-06-10 22:57:59 +02:00
|
|
|
}
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (size_max < ctx->ao_basis.ao_num) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_ao_factor",
|
|
|
|
"Array too small. Expected ao_num");
|
|
|
|
}
|
2021-07-08 19:20:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx->ao_basis.ao_factor != NULL);
|
|
|
|
memcpy(ao_factor, ctx->ao_basis.ao_factor, ctx->ao_basis.ao_num * sizeof(double));
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-07-23 12:27:18 +02:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
When all the data for the AOs have been provided, the following
|
|
|
|
function returns ~true~.
|
2021-07-23 12:27:18 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func)
|
2022-01-05 12:26:11 +01:00
|
|
|
bool qmckl_ao_basis_provided (const qmckl_context context);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 12:26:11 +01:00
|
|
|
bool qmckl_ao_basis_provided(const qmckl_context context) {
|
2021-07-08 19:20:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return false;
|
|
|
|
}
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 12:26:11 +01:00
|
|
|
assert (ctx != NULL);
|
2021-07-08 19:20:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
return ctx->ao_basis.provided;
|
2021-06-10 00:10:19 +02:00
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
**** Fortran interface
|
2021-06-23 23:44:20 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_type (context, &
|
|
|
|
basis_type) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
character(c_char) , intent(out) :: basis_type
|
|
|
|
end function qmckl_get_ao_basis_type
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_shell_num(context, &
|
|
|
|
num) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(out) :: num
|
|
|
|
end function qmckl_get_ao_basis_shell_num
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_prim_num(context, &
|
|
|
|
num) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(out) :: num
|
|
|
|
end function qmckl_get_ao_basis_prim_num
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_nucleus_shell_num(context, &
|
|
|
|
shell_num, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(out) :: shell_num(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_nucleus_shell_num
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_nucleus_index(context, &
|
|
|
|
idx, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(out) :: idx(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_nucleus_index
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_shell_ang_mom(context, &
|
|
|
|
shell_ang_mom, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int32_t) , intent(out) :: shell_ang_mom(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_shell_ang_mom
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_shell_prim_num(context, &
|
|
|
|
shell_prim_num, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(out) :: shell_prim_num(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_shell_prim_num
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_shell_prim_index(context, &
|
|
|
|
shell_prim_index, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(out) :: shell_prim_index(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_shell_prim_index
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_shell_factor(context, &
|
|
|
|
shell_factor, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
real (c_double) , intent(out) :: shell_factor(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_shell_factor
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_exponent(context, &
|
|
|
|
exponent, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
real (c_double) , intent(out) :: exponent(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_exponent
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_coefficient(context, &
|
|
|
|
coefficient, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
real (c_double) , intent(out) :: coefficient(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_coefficient
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-06-23 23:44:20 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_prim_factor(context, &
|
|
|
|
prim_factor, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-06-23 23:44:20 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
real (c_double) , intent(out) :: prim_factor(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_prim_factor
|
2021-07-19 09:26:05 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-07-19 09:26:05 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_ao_num(context, &
|
|
|
|
num) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-07-19 09:26:05 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(out) :: num
|
|
|
|
end function qmckl_get_ao_basis_ao_num
|
2021-07-19 09:26:05 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-07-19 09:26:05 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_cartesian(context, &
|
|
|
|
cartesian) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-07-19 09:26:05 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
logical (c_bool) , intent(out) :: cartesian
|
|
|
|
end function qmckl_get_ao_basis_cartesian
|
2021-07-19 09:26:05 +02:00
|
|
|
end interface
|
2022-01-05 12:26:11 +01:00
|
|
|
|
2021-07-19 09:26:05 +02:00
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_ao_factor(context, &
|
|
|
|
ao_factor, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-07-19 09:26:05 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2022-01-05 15:56:25 +01:00
|
|
|
real (c_double) , intent(out) :: ao_factor(*)
|
2022-01-05 12:26:11 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
2022-01-05 15:56:25 +01:00
|
|
|
end function qmckl_get_ao_basis_ao_factor
|
2021-06-23 23:44:20 +02:00
|
|
|
end interface
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Test :noexport:
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :tangle (eval c_test) :exports none :exports none
|
2021-06-10 22:57:59 +02:00
|
|
|
const int64_t nucl_num = chbrclf_nucl_num;
|
|
|
|
const double* nucl_charge = chbrclf_charge;
|
|
|
|
const double* nucl_coord = &(chbrclf_nucl_coord[0][0]);
|
|
|
|
|
|
|
|
qmckl_exit_code rc;
|
|
|
|
rc = qmckl_set_nucleus_num (context, nucl_num);
|
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
|
2022-01-23 16:18:46 +01:00
|
|
|
rc = qmckl_set_nucleus_coord (context, 'T', &(nucl_coord[0]), 3*nucl_num);
|
2021-06-10 22:57:59 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
|
2022-01-23 16:18:46 +01:00
|
|
|
rc = qmckl_set_nucleus_charge(context, nucl_charge, nucl_num);
|
2021-06-10 22:57:59 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
|
|
|
|
assert(qmckl_nucleus_provided(context));
|
|
|
|
|
|
|
|
|
2021-07-09 00:45:17 +02:00
|
|
|
const int64_t shell_num = chbrclf_shell_num;
|
|
|
|
const int64_t prim_num = chbrclf_prim_num;
|
|
|
|
const int64_t ao_num = chbrclf_ao_num;
|
2021-06-10 00:10:19 +02:00
|
|
|
const int64_t * nucleus_index = &(chbrclf_basis_nucleus_index[0]);
|
|
|
|
const int64_t * nucleus_shell_num = &(chbrclf_basis_nucleus_shell_num[0]);
|
2021-06-03 22:34:23 +02:00
|
|
|
const int32_t * shell_ang_mom = &(chbrclf_basis_shell_ang_mom[0]);
|
|
|
|
const int64_t * shell_prim_num = &(chbrclf_basis_shell_prim_num[0]);
|
|
|
|
const int64_t * shell_prim_index = &(chbrclf_basis_shell_prim_index[0]);
|
|
|
|
const double * shell_factor = &(chbrclf_basis_shell_factor[0]);
|
|
|
|
const double * exponent = &(chbrclf_basis_exponent[0]);
|
|
|
|
const double * coefficient = &(chbrclf_basis_coefficient[0]);
|
|
|
|
const double * prim_factor = &(chbrclf_basis_prim_factor[0]);
|
2021-07-09 00:45:17 +02:00
|
|
|
const double * ao_factor = &(chbrclf_basis_ao_factor[0]);
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
const char typ = 'G';
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2021-06-03 22:34:23 +02:00
|
|
|
rc = qmckl_set_ao_basis_type (context, typ);
|
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2021-06-03 22:34:23 +02:00
|
|
|
rc = qmckl_set_ao_basis_shell_num (context, shell_num);
|
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2021-06-03 22:34:23 +02:00
|
|
|
rc = qmckl_set_ao_basis_prim_num (context, prim_num);
|
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_nucleus_index (context, nucleus_index, nucl_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-07-08 09:15:17 +02:00
|
|
|
rc = qmckl_set_ao_basis_nucleus_index (context, nucleus_index, nucl_num);
|
|
|
|
assert(rc == QMCKL_ALREADY_SET);
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_nucleus_shell_num (context, nucleus_shell_num, nucl_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_shell_ang_mom (context, shell_ang_mom, shell_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_shell_factor (context, shell_factor, shell_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_shell_prim_num (context, shell_prim_num, shell_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_shell_prim_index (context, shell_prim_index, shell_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_exponent (context, exponent, prim_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_coefficient (context, coefficient, prim_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
assert(!qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_prim_factor (context, prim_factor, prim_num);
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
2021-07-09 00:45:17 +02:00
|
|
|
|
|
|
|
rc = qmckl_set_ao_basis_ao_num(context, ao_num);
|
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
rc = qmckl_set_ao_basis_ao_factor (context, ao_factor, ao_num);
|
2021-07-09 00:45:17 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-06-03 22:34:23 +02:00
|
|
|
assert(qmckl_ao_basis_provided(context));
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
int64_t shell_num_test ;
|
|
|
|
int64_t prim_num_test ;
|
|
|
|
int64_t ao_num_test ;
|
|
|
|
int64_t * nucleus_index_test ;
|
|
|
|
int64_t * nucleus_shell_num_test;
|
|
|
|
int32_t * shell_ang_mom_test ;
|
|
|
|
int64_t * shell_prim_num_test ;
|
|
|
|
int64_t * shell_prim_index_test ;
|
|
|
|
double * shell_factor_test ;
|
|
|
|
double * exponent_test ;
|
|
|
|
double * coefficient_test ;
|
|
|
|
double * prim_factor_test ;
|
|
|
|
double * ao_factor_test ;
|
|
|
|
char typ_test ;
|
|
|
|
|
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
rc = qmckl_get_ao_basis_type (context, &typ_test);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-07-11 23:20:40 +02:00
|
|
|
assert(typ == typ_test);
|
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
rc = qmckl_get_ao_basis_shell_num (context, &shell_num_test);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-07-11 23:20:40 +02:00
|
|
|
assert(shell_num == shell_num_test);
|
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
rc = qmckl_get_ao_basis_prim_num (context, &prim_num_test);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-07-11 23:20:40 +02:00
|
|
|
assert(prim_num == prim_num_test);
|
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
nucleus_index_test = (int64_t*) malloc (nucl_num * sizeof(int64_t));
|
|
|
|
rc = qmckl_get_ao_basis_nucleus_index (context, nucleus_index_test, nucl_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < nucl_num ; ++i) {
|
|
|
|
assert(nucleus_index_test[i] == nucleus_index[i]);
|
|
|
|
}
|
2021-10-14 16:32:58 +02:00
|
|
|
free(nucleus_index_test);
|
|
|
|
|
|
|
|
nucleus_shell_num_test = (int64_t*) malloc ( nucl_num * sizeof(int64_t));
|
|
|
|
rc = qmckl_get_ao_basis_nucleus_shell_num (context, nucleus_shell_num_test, nucl_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
|
|
|
for (int64_t i=0 ; i < nucl_num ; ++i) {
|
|
|
|
assert(nucleus_shell_num_test[i] == nucleus_shell_num[i]);
|
|
|
|
}
|
2022-07-08 17:43:11 +02:00
|
|
|
free(nucleus_shell_num_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
shell_ang_mom_test = (int32_t*) malloc ( shell_num * sizeof(int32_t));
|
|
|
|
rc = qmckl_get_ao_basis_shell_ang_mom (context, shell_ang_mom_test, shell_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < shell_num ; ++i) {
|
|
|
|
assert(shell_ang_mom_test[i] == shell_ang_mom[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(shell_ang_mom_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
shell_factor_test = (double*) malloc ( shell_num * sizeof(double));
|
|
|
|
rc = qmckl_get_ao_basis_shell_factor (context, shell_factor_test, shell_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < shell_num ; ++i) {
|
|
|
|
assert(shell_factor_test[i] == shell_factor[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(shell_factor_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
shell_prim_num_test = (int64_t*) malloc ( shell_num * sizeof(int64_t));
|
|
|
|
rc = qmckl_get_ao_basis_shell_prim_num (context, shell_prim_num_test, shell_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < shell_num ; ++i) {
|
|
|
|
assert(shell_prim_num_test[i] == shell_prim_num[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(shell_prim_num_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
shell_prim_index_test = (int64_t*) malloc ( shell_num * sizeof(int64_t));
|
|
|
|
rc = qmckl_get_ao_basis_shell_prim_index (context, shell_prim_index_test, shell_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < shell_num ; ++i) {
|
|
|
|
assert(shell_prim_index_test[i] == shell_prim_index[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(shell_prim_index_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
exponent_test = (double*) malloc ( prim_num * sizeof(double));
|
|
|
|
rc = qmckl_get_ao_basis_exponent(context, exponent_test, prim_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < prim_num ; ++i) {
|
|
|
|
assert(exponent_test[i] == exponent[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(exponent_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
coefficient_test = (double*) malloc ( prim_num * sizeof(double));
|
|
|
|
rc = qmckl_get_ao_basis_coefficient(context, coefficient_test, prim_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < prim_num ; ++i) {
|
|
|
|
assert(coefficient_test[i] == coefficient[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(coefficient_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
prim_factor_test = (double*) malloc ( prim_num * sizeof(double));
|
|
|
|
rc = qmckl_get_ao_basis_prim_factor (context, prim_factor_test, prim_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < prim_num ; ++i) {
|
|
|
|
assert(prim_factor_test[i] == prim_factor[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(prim_factor_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
rc = qmckl_get_ao_basis_ao_num(context, &ao_num_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
assert(ao_num == ao_num_test);
|
|
|
|
|
2021-10-14 16:32:58 +02:00
|
|
|
ao_factor_test = (double*) malloc ( ao_num * sizeof(double));
|
|
|
|
rc = qmckl_get_ao_basis_ao_factor (context, ao_factor_test, ao_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
for (int64_t i=0 ; i < ao_num ; ++i) {
|
|
|
|
assert(ao_factor_test[i] == ao_factor[i]);
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
free(ao_factor_test);
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
** Computed data
|
|
|
|
|
|
|
|
The following data is computed as described in the next sections:
|
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
|----------------------+-----------------------------------+----------------------------------------------------------------------------------------------|
|
|
|
|
| Variable | Type | Description |
|
|
|
|
|----------------------+-----------------------------------+----------------------------------------------------------------------------------------------|
|
2022-02-15 16:42:47 +01:00
|
|
|
| ~primitive_vgl~ | ~double[point_num][5][prim_num]~ | Value, gradients, Laplacian of the primitives at current positions |
|
2022-01-26 17:06:51 +01:00
|
|
|
| ~primitive_vgl_date~ | ~uint64_t~ | Last modification date of Value, gradients, Laplacian of the primitives at current positions |
|
2022-02-11 16:07:25 +01:00
|
|
|
| ~shell_vgl~ | ~double[point_num][5][shell_num]~ | Value, gradients, Laplacian of the primitives at current positions |
|
2022-01-26 17:06:51 +01:00
|
|
|
| ~shell_vgl_date~ | ~uint64_t~ | Last modification date of Value, gradients, Laplacian of the AOs at current positions |
|
2022-05-10 19:18:19 +02:00
|
|
|
| ~ao_vgl~ | ~double[point_num][5][ao_num]~ | Value, gradients, Laplacian of the AOs at current positions |
|
2022-01-26 17:06:51 +01:00
|
|
|
| ~ao_vgl_date~ | ~uint64_t~ | Last modification date of Value, gradients, Laplacian of the AOs at current positions |
|
2022-05-10 19:18:19 +02:00
|
|
|
| ~ao_value~ | ~double[point_num][ao_num]~ | Values of the the AOs at current positions |
|
|
|
|
| ~ao_value_date~ | ~uint64_t~ | Last modification date of the values of the AOs at current positions |
|
2022-03-28 16:53:36 +02:00
|
|
|
|----------------------+-----------------------------------+----------------------------------------------------------------------------------------------|
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
*** After initialization
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
When the basis set is completely entered, extra data structures may be
|
|
|
|
computed to accelerate the calculations. The primitives within each
|
|
|
|
contraction are sorted in ascending order of their exponents, such
|
|
|
|
that as soon as a primitive is zero all the following functions
|
|
|
|
vanish. Also, it is possible to compute a nuclear radius beyond which
|
|
|
|
all the primitives are zero up to the numerical accuracy defined in
|
|
|
|
the context.
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_private_func) :noweb yes :exports none
|
|
|
|
qmckl_exit_code qmckl_finalize_basis (qmckl_context context);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code qmckl_finalize_basis(qmckl_context context) {
|
2022-01-05 15:56:25 +01:00
|
|
|
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
2022-01-05 19:22:16 +01:00
|
|
|
"qmckl_finalize_basis",
|
2022-01-05 15:56:25 +01:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 15:56:25 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
int64_t nucl_num = 0;
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
qmckl_exit_code rc = qmckl_get_nucleus_num(context, &nucl_num);
|
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
/* nucleus_prim_index */
|
|
|
|
{
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = (ctx->nucleus.num + (int64_t) 1) * sizeof(int64_t);
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
ctx->ao_basis.nucleus_prim_index = (int64_t*) qmckl_malloc(context, mem_info);
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
if (ctx->ao_basis.nucleus_prim_index == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"ao_basis.nucleus_prim_index",
|
|
|
|
NULL);
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
for (int64_t i=0 ; i<nucl_num ; ++i) {
|
|
|
|
int64_t shell_idx = ctx->ao_basis.nucleus_index[i];
|
|
|
|
ctx->ao_basis.nucleus_prim_index[i] = ctx->ao_basis.shell_prim_index[shell_idx];
|
|
|
|
}
|
|
|
|
ctx->ao_basis.nucleus_prim_index[nucl_num] = ctx->ao_basis.prim_num;
|
2022-01-05 15:56:25 +01:00
|
|
|
}
|
|
|
|
|
2023-03-13 17:06:41 +01:00
|
|
|
/* ao_ang_mom */
|
|
|
|
{
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
2023-03-14 14:59:14 +01:00
|
|
|
mem_info.size = ctx->ao_basis.ao_num * sizeof(int32_t);
|
2023-03-13 17:06:41 +01:00
|
|
|
|
|
|
|
ctx->ao_basis.ao_ang_mom = (int32_t*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (ctx->ao_basis.ao_ang_mom == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"ao_basis.ao_ang_mom",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2023-03-14 14:59:14 +01:00
|
|
|
mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->ao_basis.ao_num * sizeof(int64_t);
|
2023-03-13 17:06:41 +01:00
|
|
|
ctx->ao_basis.ao_nucl = (int64_t*) qmckl_malloc(context, mem_info);
|
|
|
|
|
2023-03-14 14:59:14 +01:00
|
|
|
if (ctx->ao_basis.ao_nucl == NULL) {
|
2023-03-13 17:06:41 +01:00
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"ao_basis.ao_nucl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2023-03-14 14:59:14 +01:00
|
|
|
int64_t lstart[32];
|
|
|
|
for (int32_t l=0 ; l<32 ; ++l) {
|
|
|
|
lstart[l] = l*(l+1)*(l+2)/6;
|
|
|
|
}
|
2023-09-11 17:05:41 +02:00
|
|
|
|
2023-03-13 17:06:41 +01:00
|
|
|
int64_t ao_idx = 0;
|
|
|
|
for (int64_t inucl=0 ; inucl<nucl_num ; ++inucl) {
|
|
|
|
const int64_t ishell_start = ctx->ao_basis.nucleus_index[inucl];
|
|
|
|
const int64_t ishell_end = ctx->ao_basis.nucleus_index[inucl] + ctx->ao_basis.nucleus_shell_num[inucl];
|
|
|
|
for (int64_t ishell = ishell_start ; ishell < ishell_end ; ++ishell) {
|
|
|
|
const int l = ctx->ao_basis.shell_ang_mom[ishell];
|
2023-03-14 14:59:14 +01:00
|
|
|
assert (l<32);
|
|
|
|
for (int m=lstart[l] ; m < lstart[l+1]; m++) {
|
2023-03-13 17:06:41 +01:00
|
|
|
ctx->ao_basis.ao_ang_mom[ao_idx] = l;
|
|
|
|
ctx->ao_basis.ao_nucl[ao_idx] = inucl;
|
|
|
|
++ao_idx;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-03-14 14:59:14 +01:00
|
|
|
assert( ao_idx == ctx->ao_basis.ao_num );
|
2023-03-13 17:06:41 +01:00
|
|
|
}
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
/* Normalize coefficients */
|
|
|
|
{
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->ao_basis.prim_num * sizeof(double);
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
ctx->ao_basis.coefficient_normalized = (double *) qmckl_malloc(context, mem_info);
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
if (ctx->ao_basis.coefficient_normalized == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"ao_basis.coefficient_normalized",
|
|
|
|
NULL);
|
|
|
|
}
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
for (int64_t ishell=0 ; ishell < ctx->ao_basis.shell_num ; ++ishell) {
|
|
|
|
for (int64_t iprim=ctx->ao_basis.shell_prim_index[ishell] ;
|
|
|
|
iprim < ctx->ao_basis.shell_prim_index[ishell]+ctx->ao_basis.shell_prim_num[ishell] ;
|
|
|
|
++iprim) {
|
|
|
|
ctx->ao_basis.coefficient_normalized[iprim] =
|
|
|
|
ctx->ao_basis.coefficient[iprim] * ctx->ao_basis.prim_factor[iprim] *
|
|
|
|
ctx->ao_basis.shell_factor[ishell];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* Find max angular momentum on each nucleus */
|
|
|
|
{
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->nucleus.num * sizeof(int32_t);
|
|
|
|
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom = (int32_t *) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (ctx->ao_basis.nucleus_max_ang_mom == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"ao_basis.nucleus_max_ang_mom",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
for (int64_t inucl=0 ; inucl < nucl_num ; ++inucl) {
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom[inucl] = 0;
|
|
|
|
for (int64_t ishell=ctx->ao_basis.nucleus_index[inucl] ;
|
|
|
|
ishell < ctx->ao_basis.nucleus_index[inucl] + ctx->ao_basis.nucleus_shell_num[inucl] ;
|
|
|
|
++ishell) {
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom[inucl] =
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom[inucl] > ctx->ao_basis.shell_ang_mom[ishell] ?
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom[inucl] : ctx->ao_basis.shell_ang_mom[ishell] ;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
|
2023-11-30 12:56:06 +01:00
|
|
|
/*
|
|
|
|
// Find distance beyond which all AOs are zero using computed Gaussians.
|
2022-01-05 19:22:16 +01:00
|
|
|
{
|
2023-11-30 12:50:06 +01:00
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->nucleus.num * 53 * sizeof(double);
|
|
|
|
|
|
|
|
ctx->ao_basis.nucleus_range = (double *) qmckl_malloc(context, mem_info);
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
if (ctx->ao_basis.nucleus_range == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"ao_basis.nucleus_range",
|
|
|
|
NULL);
|
|
|
|
}
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
for (int64_t i=0 ; i<ctx->nucleus.num * 53 ; ++i) {
|
|
|
|
ctx->ao_basis.nucleus_range[i] = 50.;
|
|
|
|
}
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
if (ctx->ao_basis.type == 'G') {
|
|
|
|
rc = qmckl_compute_nucleus_range_gaussian(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->ao_basis.prim_num,
|
|
|
|
nucl_num,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_prim_index,
|
|
|
|
ctx->ao_basis.shell_prim_num,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.exponent,
|
|
|
|
ctx->ao_basis.coefficient_normalized,
|
|
|
|
ctx->ao_basis.nucleus_range);
|
2022-01-05 19:22:16 +01:00
|
|
|
}
|
|
|
|
}
|
2023-11-30 12:56:06 +01:00
|
|
|
*/
|
|
|
|
|
|
|
|
/* Find distance beyond which all AOs are zero.
|
|
|
|
The distance is obtained by sqrt(-log(epsilon)*range) */
|
|
|
|
{
|
|
|
|
if (ctx->ao_basis.type == 'G') {
|
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->nucleus.num * sizeof(double);
|
|
|
|
|
|
|
|
ctx->ao_basis.nucleus_range = (double *) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (ctx->ao_basis.nucleus_range == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"ao_basis.nucleus_range",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
for (int64_t inucl=0 ; inucl < ctx->nucleus.num ; ++inucl) {
|
|
|
|
ctx->ao_basis.nucleus_range[inucl] = 0.;
|
|
|
|
for (int64_t ishell=ctx->ao_basis.nucleus_index[inucl] ;
|
|
|
|
ishell < ctx->ao_basis.nucleus_index[inucl] + ctx->ao_basis.nucleus_shell_num[inucl] ;
|
|
|
|
++ishell) {
|
|
|
|
for (int64_t iprim=ctx->ao_basis.shell_prim_index[ishell] ;
|
|
|
|
iprim < ctx->ao_basis.shell_prim_index[ishell] + ctx->ao_basis.shell_prim_num[ishell] ;
|
|
|
|
++iprim) {
|
|
|
|
double range = 1./ctx->ao_basis.exponent[iprim];
|
|
|
|
ctx->ao_basis.nucleus_range[inucl] =
|
|
|
|
ctx->ao_basis.nucleus_range[inucl] > range ?
|
|
|
|
ctx->ao_basis.nucleus_range[inucl] : range;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2022-02-25 20:39:20 +01:00
|
|
|
|
|
|
|
#ifdef HAVE_HPC
|
|
|
|
rc = qmckl_finalize_basis_hpc(context);
|
2022-11-17 16:54:49 +01:00
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
2022-02-25 20:39:20 +01:00
|
|
|
#endif
|
|
|
|
|
2022-11-17 16:54:49 +01:00
|
|
|
return qmckl_context_touch(context);
|
2022-01-05 19:22:16 +01:00
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
2022-02-27 23:31:52 +01:00
|
|
|
*** TODO HPC-specific data structures
|
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
|
2022-02-27 11:18:26 +01:00
|
|
|
For faster access, we provide extra arrays for the shell information as:
|
2022-03-21 18:32:39 +01:00
|
|
|
- $C_{psa}$ = =coef_per_nucleus[inucl][ishell][iprim]=
|
|
|
|
- $\gamma_{pa}$ =expo_per_nucleus[inucl][iprim]=
|
2022-02-25 20:39:20 +01:00
|
|
|
|
|
|
|
such that the computation of the radial parts can be vectorized.
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
Exponents are sorted in increasing order to exit loops quickly,
|
2022-02-27 11:18:26 +01:00
|
|
|
and only unique exponents are kept. This also allows to pack the
|
|
|
|
exponents to enable vectorization of exponentials.
|
2022-02-25 20:39:20 +01:00
|
|
|
|
|
|
|
The computation of the radial part is made as
|
|
|
|
\[
|
|
|
|
R_{sa} = \sum_p C_{psa} \gamma_{pa}
|
|
|
|
\]
|
|
|
|
which is a matrix-vector product.
|
2022-03-21 18:32:39 +01:00
|
|
|
|
|
|
|
#+NAME:HPC_struct
|
2022-02-27 11:18:26 +01:00
|
|
|
#+begin_src c :comments org :exports none
|
2022-02-25 20:39:20 +01:00
|
|
|
/* HPC specific data structures */
|
2022-03-21 18:32:39 +01:00
|
|
|
int32_t* restrict prim_num_per_nucleus;
|
|
|
|
qmckl_tensor coef_per_nucleus;
|
|
|
|
qmckl_matrix expo_per_nucleus;
|
2022-02-25 20:39:20 +01:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_private_func) :exports none
|
2022-02-28 22:28:11 +01:00
|
|
|
#ifdef HAVE_HPC
|
2022-02-25 20:39:20 +01:00
|
|
|
qmckl_exit_code qmckl_finalize_basis_hpc (qmckl_context context);
|
2022-02-28 22:28:11 +01:00
|
|
|
#endif
|
2022-02-25 20:39:20 +01:00
|
|
|
#+end_src
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :exports none
|
2022-02-27 23:31:52 +01:00
|
|
|
/* Data structure for sorting */
|
|
|
|
struct combined {
|
|
|
|
double expo;
|
|
|
|
int64_t index;
|
|
|
|
};
|
|
|
|
|
|
|
|
/* Comparison function */
|
|
|
|
int compare_basis( const void * l, const void * r )
|
|
|
|
{
|
2022-03-21 18:32:39 +01:00
|
|
|
const struct combined * restrict _l= (const struct combined *)l;
|
|
|
|
const struct combined * restrict _r= (const struct combined *)r;
|
|
|
|
if( _l->expo > _r->expo ) return 1;
|
2022-02-27 23:31:52 +01:00
|
|
|
if( _l->expo < _r->expo ) return -1;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2022-02-28 22:28:11 +01:00
|
|
|
#ifdef HAVE_HPC
|
2022-02-25 20:39:20 +01:00
|
|
|
qmckl_exit_code qmckl_finalize_basis_hpc (qmckl_context context)
|
|
|
|
{
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-02-27 23:31:52 +01:00
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
|
|
|
|
mem_info.size = ctx->nucleus.num * sizeof(int32_t);
|
|
|
|
ctx->ao_basis.prim_num_per_nucleus = (int32_t*) qmckl_malloc(context, mem_info);
|
2022-02-25 20:39:20 +01:00
|
|
|
|
2022-02-27 23:31:52 +01:00
|
|
|
/* Find max number of primitives per nucleus */
|
2022-02-25 20:39:20 +01:00
|
|
|
int64_t shell_max = 0;
|
|
|
|
int64_t prim_max = 0;
|
|
|
|
int64_t nucl_num = ctx->nucleus.num;
|
|
|
|
for (int inucl=0 ; inucl < nucl_num ; ++inucl) {
|
|
|
|
shell_max = ctx->ao_basis.nucleus_shell_num[inucl] > shell_max ?
|
2022-02-27 23:31:52 +01:00
|
|
|
ctx->ao_basis.nucleus_shell_num[inucl] : shell_max;
|
2022-02-25 20:39:20 +01:00
|
|
|
|
|
|
|
int64_t prim_num = 0;
|
|
|
|
const int64_t ishell_start = ctx->ao_basis.nucleus_index[inucl];
|
|
|
|
const int64_t ishell_end = ctx->ao_basis.nucleus_index[inucl] + ctx->ao_basis.nucleus_shell_num[inucl];
|
|
|
|
for (int64_t ishell = ishell_start ; ishell < ishell_end ; ++ishell) {
|
|
|
|
prim_num += ctx->ao_basis.shell_prim_num[ishell];
|
|
|
|
}
|
|
|
|
prim_max = prim_num > prim_max ?
|
2022-02-27 23:31:52 +01:00
|
|
|
prim_num : prim_max;
|
|
|
|
ctx->ao_basis.prim_num_per_nucleus[inucl] = prim_num;
|
2022-02-25 20:39:20 +01:00
|
|
|
}
|
|
|
|
|
2022-02-27 23:31:52 +01:00
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
int64_t size[3] = { prim_max, shell_max, nucl_num };
|
|
|
|
ctx->ao_basis.coef_per_nucleus = qmckl_tensor_alloc( context, 3, size );
|
|
|
|
ctx->ao_basis.coef_per_nucleus = qmckl_tensor_set(ctx->ao_basis.coef_per_nucleus, 0.);
|
|
|
|
|
|
|
|
ctx->ao_basis.expo_per_nucleus = qmckl_matrix_alloc( context, prim_max, nucl_num );
|
|
|
|
ctx->ao_basis.expo_per_nucleus = qmckl_matrix_set(ctx->ao_basis.expo_per_nucleus, 0.);
|
|
|
|
|
2022-02-27 23:31:52 +01:00
|
|
|
struct combined expo[prim_max];
|
2022-02-25 20:39:20 +01:00
|
|
|
double coef[shell_max][prim_max];
|
2022-02-27 23:31:52 +01:00
|
|
|
double newcoef[prim_max];
|
|
|
|
for (int64_t inucl=0 ; inucl < nucl_num ; ++inucl) {
|
2022-02-25 20:39:20 +01:00
|
|
|
memset(expo, 0, sizeof(expo));
|
|
|
|
memset(coef, 0, sizeof(expo));
|
|
|
|
|
|
|
|
int64_t idx = 0;
|
|
|
|
const int64_t ishell_start = ctx->ao_basis.nucleus_index[inucl];
|
|
|
|
const int64_t ishell_end = ctx->ao_basis.nucleus_index[inucl] + ctx->ao_basis.nucleus_shell_num[inucl];
|
|
|
|
for (int64_t ishell = ishell_start ; ishell < ishell_end ; ++ishell) {
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
const int64_t iprim_start = ctx->ao_basis.shell_prim_index[ishell];
|
|
|
|
const int64_t iprim_end = ctx->ao_basis.shell_prim_index[ishell] + ctx->ao_basis.shell_prim_num[ishell];
|
|
|
|
for (int64_t iprim = iprim_start ; iprim < iprim_end ; ++iprim) {
|
2022-02-27 23:31:52 +01:00
|
|
|
expo[idx].expo = ctx->ao_basis.exponent[iprim];
|
|
|
|
expo[idx].index = idx;
|
|
|
|
idx += 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Sort exponents */
|
|
|
|
qsort( expo, (size_t) idx, sizeof(struct combined), compare_basis );
|
|
|
|
|
|
|
|
idx = 0;
|
|
|
|
int64_t idx2 = 0;
|
|
|
|
for (int64_t ishell = ishell_start ; ishell < ishell_end ; ++ishell) {
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-02-27 23:31:52 +01:00
|
|
|
memset(newcoef, 0, sizeof(newcoef));
|
|
|
|
const int64_t iprim_start = ctx->ao_basis.shell_prim_index[ishell];
|
|
|
|
const int64_t iprim_end = ctx->ao_basis.shell_prim_index[ishell] + ctx->ao_basis.shell_prim_num[ishell];
|
|
|
|
for (int64_t iprim = iprim_start ; iprim < iprim_end ; ++iprim) {
|
2022-03-21 18:32:39 +01:00
|
|
|
newcoef[idx] = ctx->ao_basis.coefficient_normalized[iprim];
|
2022-02-27 23:31:52 +01:00
|
|
|
idx += 1;
|
|
|
|
}
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t i=0 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-27 23:31:52 +01:00
|
|
|
idx2 = expo[i].index;
|
|
|
|
coef[ishell - ishell_start][i] = newcoef[idx2];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Apply ordering to coefficients */
|
|
|
|
|
|
|
|
/* Remove duplicates */
|
|
|
|
int64_t newidx[prim_max];
|
|
|
|
int64_t idxmax = 0;
|
|
|
|
idx = 0;
|
|
|
|
newidx[0] = 0;
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t i=1 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-27 23:31:52 +01:00
|
|
|
if (expo[i].expo != expo[i-1].expo) {
|
2022-02-25 20:39:20 +01:00
|
|
|
idx += 1;
|
|
|
|
}
|
2022-02-27 23:31:52 +01:00
|
|
|
newidx[i] = idx;
|
|
|
|
}
|
|
|
|
idxmax = idx;
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t j=0 ; j<ishell_end-ishell_start ; ++j) {
|
2022-02-27 23:31:52 +01:00
|
|
|
memset(newcoef, 0, sizeof(newcoef));
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t i=0 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-27 23:31:52 +01:00
|
|
|
newcoef[newidx[i]] += coef[j][i];
|
|
|
|
}
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t i=0 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-27 23:31:52 +01:00
|
|
|
coef[j][i] = newcoef[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t i=0 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-27 23:31:52 +01:00
|
|
|
expo[newidx[i]].expo = expo[i].expo;
|
2022-02-25 20:39:20 +01:00
|
|
|
}
|
2022-02-27 23:31:52 +01:00
|
|
|
ctx->ao_basis.prim_num_per_nucleus[inucl] = idxmax+1;
|
2022-02-25 20:39:20 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t i=0 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-27 23:31:52 +01:00
|
|
|
qmckl_mat( ctx->ao_basis.expo_per_nucleus, i, inucl ) = expo[i].expo;
|
2022-02-25 20:39:20 +01:00
|
|
|
}
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int32_t j=0 ; j<ishell_end-ishell_start ; ++j) {
|
|
|
|
for (int32_t i=0 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-25 20:39:20 +01:00
|
|
|
qmckl_ten3( ctx->ao_basis.coef_per_nucleus, i, j, inucl ) = coef[j][i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
/*
|
|
|
|
for (int32_t i=0 ; i<ctx->ao_basis.prim_num_per_nucleus[inucl] ; ++i) {
|
2022-02-27 23:31:52 +01:00
|
|
|
printf("%4ld %4ld %15.5e | ", inucl, i, qmckl_mat( ctx->ao_basis.expo_per_nucleus, i, inucl ));
|
|
|
|
for (int64_t j=0 ; j<ishell_end-ishell_start ; ++j) {
|
|
|
|
printf("%8.5f ", qmckl_ten3( ctx->ao_basis.coef_per_nucleus, i, j, inucl ));
|
|
|
|
}
|
|
|
|
printf("\n");
|
|
|
|
}
|
|
|
|
printf("\n");
|
|
|
|
*/
|
2022-02-25 20:39:20 +01:00
|
|
|
}
|
2022-02-27 23:31:52 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-02-28 22:28:11 +01:00
|
|
|
#endif
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-02-25 20:39:20 +01:00
|
|
|
#+end_src
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
*** Access functions
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_func) :noweb yes
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_primitive_vgl (qmckl_context context,
|
|
|
|
double* const primitive_vgl,
|
|
|
|
const int64_t size_max);
|
|
|
|
#+end_src
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
Returns the array of values, gradients an Laplacian of primitive
|
|
|
|
basis functions evaluated at the current coordinates.
|
|
|
|
See section [[Computation of primitives]].
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_primitive_vgl (qmckl_context context,
|
|
|
|
double* const primitive_vgl,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_primitive_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-01-08 15:36:07 +01:00
|
|
|
if (size_max <= 0) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_primitive_vgl",
|
|
|
|
"size_max <= 0");
|
|
|
|
}
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
qmckl_exit_code rc;
|
|
|
|
|
|
|
|
rc = qmckl_provide_ao_basis_primitive_vgl(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 19:22:16 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
int64_t sze = ctx->ao_basis.prim_num * 5 * ctx->point.num;
|
2022-01-05 19:22:16 +01:00
|
|
|
if (size_max < sze) {
|
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 19:22:16 +01:00
|
|
|
"qmckl_get_ao_basis_primitive_vgl",
|
|
|
|
"input array too small");
|
|
|
|
}
|
2022-01-08 15:36:07 +01:00
|
|
|
memcpy(primitive_vgl, ctx->ao_basis.primitive_vgl, (size_t) sze * sizeof(double));
|
2022-01-05 19:22:16 +01:00
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
|
|
|
#+end_src
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_primitive_vgl &
|
|
|
|
(context, primitive_vgl, size_max) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 19:22:16 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2023-09-11 17:05:41 +02:00
|
|
|
real(c_double), intent(out) :: primitive_vgl(*)
|
2022-01-05 19:22:16 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function
|
|
|
|
end interface
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_func) :noweb yes
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_vgl (qmckl_context context,
|
|
|
|
double* const shell_vgl,
|
|
|
|
const int64_t size_max);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
Returns the array of values, gradients an Laplacian of contracted shells
|
|
|
|
evaluated at the current coordinates. See section [[Computation of shells]].
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_shell_vgl (qmckl_context context,
|
|
|
|
double* const shell_vgl,
|
2022-01-06 02:28:13 +01:00
|
|
|
const int64_t size_max)
|
2022-01-05 19:22:16 +01:00
|
|
|
{
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_shell_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_exit_code rc;
|
|
|
|
|
|
|
|
rc = qmckl_provide_ao_basis_shell_vgl(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 19:22:16 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
int64_t sze = ctx->ao_basis.shell_num * 5 * ctx->point.num;
|
2022-01-05 19:22:16 +01:00
|
|
|
if (size_max < sze) {
|
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 19:22:16 +01:00
|
|
|
"qmckl_get_ao_basis_shell_vgl",
|
|
|
|
"input array too small");
|
|
|
|
}
|
2022-01-08 15:36:07 +01:00
|
|
|
memcpy(shell_vgl, ctx->ao_basis.shell_vgl, (size_t)sze * sizeof(double));
|
2022-01-05 19:22:16 +01:00
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
2022-01-05 15:56:25 +01:00
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_shell_vgl &
|
|
|
|
(context, shell_vgl, size_max) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 15:56:25 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2023-09-11 17:05:41 +02:00
|
|
|
real(c_double), intent(out) :: shell_vgl(*)
|
2022-01-05 15:56:25 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function
|
|
|
|
end interface
|
|
|
|
#+end_src
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_func) :noweb yes
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_vgl (qmckl_context context,
|
|
|
|
double* const ao_vgl,
|
|
|
|
const int64_t size_max);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
Returns the array of values, gradients an Laplacian of the atomic orbitals
|
|
|
|
evaluated at the current coordinates.
|
|
|
|
See section [[Combining radial and polynomial parts]].
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_vgl (qmckl_context context,
|
|
|
|
double* const ao_vgl,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
2022-01-06 02:28:13 +01:00
|
|
|
"qmckl_get_ao_basis_ao_vgl",
|
2022-01-05 19:22:16 +01:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_exit_code rc;
|
|
|
|
|
2022-07-11 13:42:39 +02:00
|
|
|
rc = qmckl_provide_ao_basis_ao_vgl(context);
|
2022-01-05 19:22:16 +01:00
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 19:22:16 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
int64_t sze = ctx->ao_basis.ao_num * 5 * ctx->point.num;
|
2022-01-05 19:22:16 +01:00
|
|
|
if (size_max < sze) {
|
|
|
|
return qmckl_failwith( context,
|
2022-01-08 15:36:07 +01:00
|
|
|
QMCKL_INVALID_ARG_3,
|
2022-01-05 19:22:16 +01:00
|
|
|
"qmckl_get_ao_basis_ao_vgl",
|
|
|
|
"input array too small");
|
|
|
|
}
|
2022-01-08 15:36:07 +01:00
|
|
|
memcpy(ao_vgl, ctx->ao_basis.ao_vgl, (size_t) sze * sizeof(double));
|
2022-01-05 19:22:16 +01:00
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_ao_vgl (context, &
|
|
|
|
ao_vgl, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-01-05 19:22:16 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2023-09-11 17:05:41 +02:00
|
|
|
real(c_double), intent(out) :: ao_vgl(*)
|
2022-01-05 19:22:16 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_get_ao_basis_ao_vgl
|
|
|
|
end interface
|
|
|
|
#+end_src
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
Uses the given array to compute the VGL.
|
2022-02-18 01:24:37 +01:00
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_func) :noweb yes
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_vgl_inplace (qmckl_context context,
|
|
|
|
double* const ao_vgl,
|
|
|
|
const int64_t size_max);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_vgl_inplace (qmckl_context context,
|
|
|
|
double* const ao_vgl,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_ao_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_exit_code rc;
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-02-18 01:24:37 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
int64_t sze = ctx->ao_basis.ao_num * 5 * ctx->point.num;
|
|
|
|
if (size_max < sze) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_ao_vgl",
|
|
|
|
"input array too small");
|
|
|
|
}
|
|
|
|
|
|
|
|
rc = qmckl_context_touch(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
|
|
|
double* old_array = ctx->ao_basis.ao_vgl;
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_vgl = ao_vgl;
|
|
|
|
|
2022-07-11 13:42:39 +02:00
|
|
|
rc = qmckl_provide_ao_basis_ao_vgl(context);
|
2022-02-18 01:24:37 +01:00
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_vgl = old_array;
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_ao_vgl_inplace (context, &
|
|
|
|
ao_vgl, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-02-18 01:24:37 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2023-09-11 17:05:41 +02:00
|
|
|
real(c_double), intent(out) :: ao_vgl(*)
|
2022-02-18 01:24:37 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_get_ao_basis_ao_vgl_inplace
|
|
|
|
end interface
|
|
|
|
#+end_src
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2022-06-11 10:57:58 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+begin_src c :comments org :tangle (eval h_func) :noweb yes
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_value (qmckl_context context,
|
|
|
|
double* const ao_value,
|
|
|
|
const int64_t size_max);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
Returns the array of values of the atomic orbitals evaluated at
|
|
|
|
the current coordinates. See section [[Combining radial and polynomial parts]].
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_value (qmckl_context context,
|
|
|
|
double* const ao_value,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_ao_value",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_exit_code rc;
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
rc = qmckl_provide_ao_basis_ao_value(context);
|
2022-05-10 19:18:19 +02:00
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
int64_t sze = ctx->ao_basis.ao_num * ctx->point.num;
|
|
|
|
if (size_max < sze) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_ao_value",
|
|
|
|
"input array too small");
|
|
|
|
}
|
|
|
|
memcpy(ao_value, ctx->ao_basis.ao_value, (size_t) sze * sizeof(double));
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_ao_value (context, &
|
|
|
|
ao_value, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-05-10 19:18:19 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2023-09-11 17:05:41 +02:00
|
|
|
real(c_double) , intent(out) :: ao_value(*)
|
2022-05-10 19:18:19 +02:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_get_ao_basis_ao_value
|
|
|
|
end interface
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
Uses the given array to compute the value.
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval h_func) :noweb yes
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_value_inplace (qmckl_context context,
|
|
|
|
double* const ao_value,
|
|
|
|
const int64_t size_max);
|
|
|
|
#+end_src
|
2022-06-11 10:57:58 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_get_ao_basis_ao_value_inplace (qmckl_context context,
|
|
|
|
double* const ao_value,
|
|
|
|
const int64_t size_max)
|
|
|
|
{
|
|
|
|
|
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_get_ao_basis_ao_value",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_exit_code rc;
|
|
|
|
|
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
int64_t sze = ctx->ao_basis.ao_num * ctx->point.num;
|
|
|
|
if (size_max < sze) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_ARG_3,
|
|
|
|
"qmckl_get_ao_basis_ao_value",
|
|
|
|
"input array too small");
|
|
|
|
}
|
|
|
|
|
|
|
|
rc = qmckl_context_touch(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
|
|
|
double* old_array = ctx->ao_basis.ao_value;
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_value = ao_value;
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
rc = qmckl_provide_ao_basis_ao_value(context);
|
2022-05-10 19:18:19 +02:00
|
|
|
if (rc != QMCKL_SUCCESS) return rc;
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_value = old_array;
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
|
|
|
integer(c_int32_t) function qmckl_get_ao_basis_ao_value_inplace (context, &
|
|
|
|
ao_value, size_max) bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-05-10 19:18:19 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
integer (c_int64_t) , intent(in) , value :: context
|
2023-09-11 17:05:41 +02:00
|
|
|
real(c_double) , intent(out) :: ao_value(*)
|
2022-05-10 19:18:19 +02:00
|
|
|
integer (c_int64_t) , intent(in) , value :: size_max
|
|
|
|
end function qmckl_get_ao_basis_ao_value_inplace
|
|
|
|
end interface
|
|
|
|
#+end_src
|
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
* Radial part
|
2021-07-23 12:27:18 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
** General functions for Gaussian basis functions
|
2021-04-18 15:10:55 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
~qmckl_ao_gaussian_vgl~ computes the values, gradients and
|
|
|
|
Laplacians at a given point of ~n~ Gaussian functions centered at
|
|
|
|
the same point:
|
|
|
|
|
|
|
|
\[ v_i = \exp(-a_i |X-R|^2) \]
|
|
|
|
\[ \nabla_x v_i = -2 a_i (X_x - R_x) v_i \]
|
|
|
|
\[ \nabla_y v_i = -2 a_i (X_y - R_y) v_i \]
|
|
|
|
\[ \nabla_z v_i = -2 a_i (X_z - R_z) v_i \]
|
|
|
|
\[ \Delta v_i = a_i (4 |X-R|^2 a_i - 6) v_i \]
|
|
|
|
|
|
|
|
|--------------+------------------+------------------------------------------------------|
|
|
|
|
| Variable | Type | Description |
|
|
|
|
|--------------+------------------+------------------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | Global state |
|
|
|
|
| ~X(3)~ | ~double[3]~ | Array containing the coordinates of the points |
|
|
|
|
| ~R(3)~ | ~double[3]~ | Array containing the x,y,z coordinates of the center |
|
|
|
|
| ~n~ | ~int64_t~ | Number of computed Gaussians |
|
|
|
|
| ~A(n)~ | ~double[n]~ | Exponents of the Gaussians |
|
|
|
|
| ~VGL(ldv,5)~ | ~double[5][ldv]~ | Value, gradients and Laplacian of the Gaussians |
|
|
|
|
| ~ldv~ | ~int64_t~ | Leading dimension of array ~VGL~ |
|
|
|
|
|--------------+------------------+------------------------------------------------------|
|
|
|
|
|
|
|
|
Requirements:
|
|
|
|
|
|
|
|
- ~context~ \ne 0
|
|
|
|
- ~n~ > 0
|
|
|
|
- ~ldv~ >= 5
|
|
|
|
- ~A(i)~ > 0 for all ~i~
|
|
|
|
- ~X~ is allocated with at least $3 \times 8$ bytes
|
|
|
|
- ~R~ is allocated with at least $3 \times 8$ bytes
|
|
|
|
- ~A~ is allocated with at least $n \times 8$ bytes
|
|
|
|
- ~VGL~ is allocated with at least $n \times 5 \times 8$ bytes
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval h_func)
|
2021-06-10 22:57:59 +02:00
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_ao_gaussian_vgl(const qmckl_context context,
|
|
|
|
const double *X,
|
|
|
|
const double *R,
|
|
|
|
const int64_t *n,
|
|
|
|
const int64_t *A,
|
|
|
|
const double *VGL,
|
|
|
|
const int64_t ldv);
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src f90 :tangle (eval f)
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_ao_gaussian_vgl(context, X, R, n, A, VGL, ldv) &
|
|
|
|
bind(C) result(info)
|
|
|
|
use qmckl_constants
|
2021-04-01 01:19:33 +02:00
|
|
|
implicit none
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context) , intent(in) , value :: context
|
|
|
|
real (c_double) , intent(in) :: X(3), R(3)
|
|
|
|
integer (c_int64_t) , intent(in) , value :: n
|
2024-02-23 16:37:31 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: ldv
|
2023-09-22 16:41:43 +02:00
|
|
|
real (c_double) , intent(in) :: A(n)
|
|
|
|
real (c_double) , intent(out) :: VGL(ldv,5)
|
|
|
|
integer (qmckl_exit_code) :: info
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
integer*8 :: i,j
|
2023-08-24 09:59:57 +02:00
|
|
|
double precision :: Y(3), r2, t, u, v
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
info = QMCKL_SUCCESS
|
|
|
|
|
|
|
|
if (context == QMCKL_NULL_CONTEXT) then
|
|
|
|
info = QMCKL_INVALID_CONTEXT
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
if (n <= 0) then
|
|
|
|
info = QMCKL_INVALID_ARG_4
|
2021-04-01 01:19:33 +02:00
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
if (ldv < n) then
|
|
|
|
info = QMCKL_INVALID_ARG_7
|
2021-04-01 01:19:33 +02:00
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
do i=1,3
|
|
|
|
Y(i) = X(i) - R(i)
|
2021-04-01 01:19:33 +02:00
|
|
|
end do
|
2021-06-10 22:57:59 +02:00
|
|
|
r2 = Y(1)*Y(1) + Y(2)*Y(2) + Y(3)*Y(3)
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
do i=1,n
|
|
|
|
VGL(i,1) = dexp(-A(i) * r2)
|
|
|
|
end do
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
do i=1,n
|
|
|
|
VGL(i,5) = A(i) * VGL(i,1)
|
|
|
|
end do
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
t = -2.d0 * ( X(1) - R(1) )
|
|
|
|
u = -2.d0 * ( X(2) - R(2) )
|
|
|
|
v = -2.d0 * ( X(3) - R(3) )
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
do i=1,n
|
|
|
|
VGL(i,2) = t * VGL(i,5)
|
|
|
|
VGL(i,3) = u * VGL(i,5)
|
|
|
|
VGL(i,4) = v * VGL(i,5)
|
|
|
|
end do
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
t = 4.d0 * r2
|
|
|
|
do i=1,n
|
|
|
|
VGL(i,5) = (t * A(i) - 6.d0) * VGL(i,5)
|
|
|
|
end do
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
end function qmckl_ao_gaussian_vgl
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src f90 :tangle (eval fh_func) :exports none
|
2021-06-10 22:57:59 +02:00
|
|
|
interface
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_ao_gaussian_vgl(context, &
|
|
|
|
X, R, n, A, VGL, ldv) bind(C) result(info)
|
|
|
|
use qmckl_constants
|
|
|
|
integer (qmckl_context) , intent(in) , value :: context
|
2021-06-10 22:57:59 +02:00
|
|
|
integer (c_int64_t) , intent(in) , value :: ldv
|
|
|
|
integer (c_int64_t) , intent(in) , value :: n
|
|
|
|
real (c_double) , intent(in) :: X(3), R(3), A(n)
|
|
|
|
real (c_double) , intent(out) :: VGL(ldv,5)
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_exit_code) :: info
|
2021-06-10 22:57:59 +02:00
|
|
|
end function qmckl_ao_gaussian_vgl
|
|
|
|
end interface
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-04-16 00:57:08 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Test :noexport:
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval f_test)
|
2023-09-22 16:41:43 +02:00
|
|
|
function test_qmckl_ao_gaussian_vgl(context) bind(C)
|
2021-04-01 01:19:33 +02:00
|
|
|
use qmckl
|
|
|
|
implicit none
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_context), intent(in), value :: context
|
|
|
|
integer(qmckl_exit_code) :: test_qmckl_ao_gaussian_vgl
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
integer*8 :: n, ldv, j, i
|
2022-02-25 16:30:16 +01:00
|
|
|
double precision :: X(3), R(3), Y(3), r2, z
|
2021-06-10 22:57:59 +02:00
|
|
|
double precision, allocatable :: VGL(:,:), A(:)
|
|
|
|
double precision :: epsilon
|
|
|
|
|
2022-02-25 16:30:16 +01:00
|
|
|
epsilon = 3.d0 * qmckl_get_numprec_epsilon(context)
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
X = (/ 1.1 , 2.2 , 3.3 /)
|
|
|
|
R = (/ 0.1 , 1.2 , -2.3 /)
|
|
|
|
Y(:) = X(:) - R(:)
|
|
|
|
r2 = Y(1)**2 + Y(2)**2 + Y(3)**2
|
|
|
|
|
|
|
|
n = 10;
|
|
|
|
ldv = 100;
|
|
|
|
|
|
|
|
allocate (A(n), VGL(ldv,5))
|
|
|
|
do i=1,n
|
|
|
|
A(i) = 0.0013 * dble(ishft(1,i))
|
|
|
|
end do
|
|
|
|
|
2021-10-13 15:01:23 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
test_qmckl_ao_gaussian_vgl = &
|
|
|
|
qmckl_ao_gaussian_vgl(context, X, R, n, A, VGL, ldv)
|
|
|
|
if (test_qmckl_ao_gaussian_vgl /= 0) return
|
|
|
|
|
|
|
|
test_qmckl_ao_gaussian_vgl = -1
|
|
|
|
|
|
|
|
do i=1,n
|
|
|
|
test_qmckl_ao_gaussian_vgl = -11
|
2022-02-25 16:30:16 +01:00
|
|
|
z = dabs(1.d0 - VGL(i,1) / (dexp(-A(i) * r2)) )
|
|
|
|
if ( z > epsilon ) then
|
|
|
|
print *, z, epsilon
|
|
|
|
return
|
|
|
|
end if
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
test_qmckl_ao_gaussian_vgl = -12
|
2022-02-25 16:30:16 +01:00
|
|
|
z = dabs(1.d0 - VGL(i,2) / (&
|
|
|
|
-2.d0 * A(i) * Y(1) * dexp(-A(i) * r2) ))
|
|
|
|
if ( z > epsilon ) then
|
|
|
|
print *, z, epsilon
|
|
|
|
return
|
|
|
|
end if
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
test_qmckl_ao_gaussian_vgl = -13
|
2022-02-25 16:30:16 +01:00
|
|
|
z = dabs(1.d0 - VGL(i,3) / (&
|
|
|
|
-2.d0 * A(i) * Y(2) * dexp(-A(i) * r2) ))
|
|
|
|
if ( z > epsilon ) then
|
|
|
|
print *, z, epsilon
|
|
|
|
return
|
|
|
|
end if
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
test_qmckl_ao_gaussian_vgl = -14
|
2022-02-25 16:30:16 +01:00
|
|
|
z = dabs(1.d0 - VGL(i,4) / (&
|
|
|
|
-2.d0 * A(i) * Y(3) * dexp(-A(i) * r2) ))
|
|
|
|
if ( z > epsilon ) then
|
|
|
|
print *, z, epsilon
|
|
|
|
return
|
|
|
|
end if
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
test_qmckl_ao_gaussian_vgl = -15
|
2022-02-25 16:30:16 +01:00
|
|
|
z = dabs(1.d0 - VGL(i,5) / (&
|
|
|
|
A(i) * (4.d0*r2*A(i) - 6.d0) * dexp(-A(i) * r2) ))
|
|
|
|
if ( z > epsilon ) then
|
|
|
|
print *, z, epsilon
|
|
|
|
return
|
|
|
|
end if
|
2021-06-10 22:57:59 +02:00
|
|
|
end do
|
|
|
|
|
|
|
|
test_qmckl_ao_gaussian_vgl = 0
|
2021-08-30 17:59:11 +02:00
|
|
|
|
2021-10-13 15:01:23 +02:00
|
|
|
deallocate(VGL)
|
2021-06-10 22:57:59 +02:00
|
|
|
end function test_qmckl_ao_gaussian_vgl
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src c :tangle (eval c_test) :exports none
|
|
|
|
int test_qmckl_ao_gaussian_vgl(qmckl_context context);
|
|
|
|
assert(0 == test_qmckl_ao_gaussian_vgl(context));
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
** TODO General functions for Slater basis functions :noexport:
|
|
|
|
** TODO General functions for Radial functions on a grid :noexport:
|
2021-07-08 19:20:19 +02:00
|
|
|
** Computation of primitives
|
2022-01-05 15:56:25 +01:00
|
|
|
:PROPERTIES:
|
|
|
|
:Name: qmckl_compute_ao_basis_primitive_gaussian_vgl
|
|
|
|
:CRetType: qmckl_exit_code
|
|
|
|
:FRetType: qmckl_exit_code
|
|
|
|
:END:
|
|
|
|
|
|
|
|
#+NAME: qmckl_ao_basis_primitive_gaussian_vgl_args
|
2022-01-26 17:06:51 +01:00
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|----------------------+----------------------------------+--------+--------------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~prim_num~ | ~int64_t~ | in | Number of primitives |
|
|
|
|
| ~point_num~ | ~int64_t~ | in | Number of points considered |
|
|
|
|
| ~nucl_num~ | ~int64_t~ | in | Number of nuclei |
|
2023-09-22 16:41:43 +02:00
|
|
|
| ~nucleus_prim_index~ | ~int64_t[nucl_num+1]~ | in | Index of the 1st primitive of each nucleus |
|
2022-01-26 17:06:51 +01:00
|
|
|
| ~coord~ | ~double[3][point_num]~ | in | Coordinates |
|
|
|
|
| ~nucl_coord~ | ~double[3][nucl_num]~ | in | Nuclear coordinates |
|
|
|
|
| ~expo~ | ~double[prim_num]~ | in | Exponents of the primitives |
|
2022-02-11 16:07:25 +01:00
|
|
|
| ~primitive_vgl~ | ~double[point_num][5][prim_num]~ | out | Value, gradients and Laplacian of the primitives |
|
2022-01-05 19:22:16 +01:00
|
|
|
|
|
|
|
#+CALL: generate_c_header(table=qmckl_ao_basis_primitive_gaussian_vgl_args,rettyp=get_value("CRetType"),fname="qmckl_compute_ao_basis_primitive_gaussian_vgl"))
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_compute_ao_basis_primitive_gaussian_vgl (
|
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t prim_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
const int64_t point_num,
|
2022-01-05 19:22:16 +01:00
|
|
|
const int64_t nucl_num,
|
|
|
|
const int64_t* nucleus_prim_index,
|
2022-01-26 17:06:51 +01:00
|
|
|
const double* coord,
|
2022-01-05 19:22:16 +01:00
|
|
|
const double* nucl_coord,
|
|
|
|
const double* expo,
|
2023-11-29 02:10:20 +01:00
|
|
|
double* const primitive_vgl );
|
2022-01-05 19:22:16 +01:00
|
|
|
#+end_src
|
2021-06-10 22:57:59 +02:00
|
|
|
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+begin_src f90 :comments org :tangle (eval f) :noweb yes
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_compute_ao_basis_primitive_gaussian_vgl &
|
|
|
|
(context, prim_num, point_num, nucl_num, nucleus_prim_index, coord, nucl_coord, expo, primitive_vgl) &
|
|
|
|
bind(C) result(info)
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
use qmckl_constants
|
2023-11-29 02:10:20 +01:00
|
|
|
|
2024-01-11 14:33:40 +01:00
|
|
|
use qmckl, only: qmckl_get_numprec_precision
|
2022-01-05 15:56:25 +01:00
|
|
|
implicit none
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: prim_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: point_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: nucl_num
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_prim_index(nucl_num+1)
|
|
|
|
real (c_double ) , intent(in) :: coord(point_num,3)
|
|
|
|
real (c_double ) , intent(in) :: nucl_coord(nucl_num,3)
|
|
|
|
real (c_double ) , intent(in) :: expo(prim_num)
|
|
|
|
real (c_double ) , intent(out) :: primitive_vgl(prim_num,5,point_num)
|
|
|
|
integer(qmckl_exit_code) :: info
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
integer*8 :: inucl, iprim, ipoint
|
2022-01-05 15:56:25 +01:00
|
|
|
double precision :: x, y, z, two_a, ar2, r2, v, cutoff
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
info = QMCKL_SUCCESS
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
! Don't compute exponentials when the result will be almost zero.
|
2024-01-11 14:33:40 +01:00
|
|
|
<<fortran_cutoff>>
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
do inucl=1,nucl_num
|
|
|
|
! C is zero-based, so shift bounds by one
|
|
|
|
do iprim = nucleus_prim_index(inucl)+1, nucleus_prim_index(inucl+1)
|
2022-01-26 17:06:51 +01:00
|
|
|
do ipoint = 1, point_num
|
|
|
|
x = coord(ipoint,1) - nucl_coord(inucl,1)
|
|
|
|
y = coord(ipoint,2) - nucl_coord(inucl,2)
|
|
|
|
z = coord(ipoint,3) - nucl_coord(inucl,3)
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
r2 = x*x + y*y + z*z
|
|
|
|
ar2 = expo(iprim)*r2
|
|
|
|
if (ar2 > cutoff) cycle
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
v = dexp(-ar2)
|
|
|
|
two_a = -2.d0 * expo(iprim) * v
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
primitive_vgl(iprim, 1, ipoint) = v
|
|
|
|
primitive_vgl(iprim, 2, ipoint) = two_a * x
|
|
|
|
primitive_vgl(iprim, 3, ipoint) = two_a * y
|
|
|
|
primitive_vgl(iprim, 4, ipoint) = two_a * z
|
|
|
|
primitive_vgl(iprim, 5, ipoint) = two_a * (3.d0 - 2.d0*ar2)
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
end do
|
|
|
|
end do
|
|
|
|
end do
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
end function qmckl_compute_ao_basis_primitive_gaussian_vgl
|
2022-01-05 19:22:16 +01:00
|
|
|
#+end_src
|
2022-01-05 15:56:25 +01:00
|
|
|
|
|
|
|
|
|
|
|
*** Provide :noexport:
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+CALL: write_provider_header( group="ao_basis", data="primitive_vgl" )
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval h_private_func) :noweb yes :export none
|
2022-01-05 15:56:25 +01:00
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_primitive_vgl(qmckl_context context);
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+CALL: write_provider_pre( group="ao_basis", data="primitive_vgl", dimension="ctx->ao_basis.prim_num * 5 * ctx->point.num")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
2022-01-05 15:56:25 +01:00
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_primitive_vgl(qmckl_context context)
|
|
|
|
{
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_exit_code rc = QMCKL_SUCCESS;
|
2022-07-09 13:11:20 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
2022-07-09 13:11:20 +02:00
|
|
|
"qmckl_provide_ao_basis_primitive_vgl",
|
2022-01-05 15:56:25 +01:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 15:56:25 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
if (!ctx->ao_basis.provided) {
|
|
|
|
return qmckl_failwith( context,
|
2021-06-10 22:57:59 +02:00
|
|
|
QMCKL_NOT_PROVIDED,
|
2022-07-09 13:11:20 +02:00
|
|
|
"qmckl_provide_ao_basis_primitive_vgl",
|
2021-06-10 22:57:59 +02:00
|
|
|
NULL);
|
|
|
|
}
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
/* Compute if necessary */
|
2022-01-26 17:06:51 +01:00
|
|
|
if (ctx->point.date > ctx->ao_basis.primitive_vgl_date) {
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->ao_basis.prim_num * 5 * ctx->point.num * sizeof(double);
|
|
|
|
|
|
|
|
if (ctx->ao_basis.primitive_vgl != NULL) {
|
|
|
|
qmckl_memory_info_struct mem_info_test = qmckl_memory_info_struct_zero;
|
|
|
|
rc = qmckl_get_malloc_info(context, ctx->ao_basis.primitive_vgl, &mem_info_test);
|
|
|
|
|
|
|
|
/* if rc != QMCKL_SUCCESS, we are maybe in an _inplace function because the
|
|
|
|
memory was not allocated with qmckl_malloc */
|
|
|
|
|
|
|
|
if ((rc == QMCKL_SUCCESS) && (mem_info_test.size != mem_info.size)) {
|
2022-07-09 13:11:20 +02:00
|
|
|
rc = qmckl_free(context, ctx->ao_basis.primitive_vgl);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
ctx->ao_basis.primitive_vgl = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
/* Allocate array */
|
|
|
|
if (ctx->ao_basis.primitive_vgl == NULL) {
|
|
|
|
|
|
|
|
double* primitive_vgl = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (primitive_vgl == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_ao_basis_primitive_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
ctx->ao_basis.primitive_vgl = primitive_vgl;
|
|
|
|
}
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2021-06-10 22:57:59 +02:00
|
|
|
if (ctx->ao_basis.type == 'G') {
|
|
|
|
rc = qmckl_compute_ao_basis_primitive_gaussian_vgl(context,
|
|
|
|
ctx->ao_basis.prim_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
ctx->point.num,
|
2021-06-10 22:57:59 +02:00
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->ao_basis.nucleus_prim_index,
|
2022-01-26 17:06:51 +01:00
|
|
|
ctx->point.coord.data,
|
2022-01-23 16:18:46 +01:00
|
|
|
ctx->nucleus.coord.data,
|
2021-06-10 22:57:59 +02:00
|
|
|
ctx->ao_basis.exponent,
|
|
|
|
ctx->ao_basis.primitive_vgl);
|
|
|
|
} else {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_FAILURE,
|
|
|
|
"compute_ao_basis_primitive_vgl",
|
|
|
|
"Not yet implemented");
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
}
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+CALL: write_provider_post( group="ao_basis", data="shell_vgl" )
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
2021-06-10 22:57:59 +02:00
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
ctx->ao_basis.shell_vgl_date = ctx->date;
|
2021-06-10 22:57:59 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Test :noexport:
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
#+begin_src python :results output :exports none :exports none
|
2021-06-10 22:57:59 +02:00
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
def f(a,x,y):
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
return np.exp( -a*(np.linalg.norm(x-y))**2 )
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
def df(a,x,y,n):
|
|
|
|
h0 = 1.e-6
|
|
|
|
if n == 1: h = np.array([h0,0.,0.])
|
|
|
|
elif n == 2: h = np.array([0.,h0,0.])
|
|
|
|
elif n == 3: h = np.array([0.,0.,h0])
|
|
|
|
return ( f(a,x+h,y) - f(a,x-h,y) ) / (2.*h0)
|
|
|
|
|
|
|
|
def d2f(a,x,y,n):
|
|
|
|
h0 = 1.e-6
|
|
|
|
if n == 1: h = np.array([h0,0.,0.])
|
|
|
|
elif n == 2: h = np.array([0.,h0,0.])
|
|
|
|
elif n == 3: h = np.array([0.,0.,h0])
|
|
|
|
return ( f(a,x+h,y) - 2.*f(a,x,y) + f(a,x-h,y) ) / h0**2
|
|
|
|
|
|
|
|
def lf(a,x,y):
|
|
|
|
return d2f(a,x,y,1) + d2f(a,x,y,2) + d2f(a,x,y,3)
|
|
|
|
|
|
|
|
elec_26_w1 = np.array( [ 1.49050402641, 2.90106987953, -1.05920815468 ] )
|
|
|
|
nucl_1 = np.array( [ 1.096243353458458e+00, 8.907054016973815e-01, 7.777092280258892e-01 ] )
|
|
|
|
nucl_2 = np.array( [ 1.168459237342663e+00, 1.125660720053393e+00, 2.833370314829343e+00 ] )
|
|
|
|
|
2021-09-23 11:16:54 +02:00
|
|
|
#double prim_vgl[prim_num][5][elec_num];
|
2021-06-10 22:57:59 +02:00
|
|
|
a = 0.9059; x = elec_26_w1 ; y = nucl_1
|
2021-09-23 11:16:54 +02:00
|
|
|
print ( "[7][0][26] : %e"% f(a,x,y))
|
|
|
|
print ( "[7][1][26] : %e"% df(a,x,y,1))
|
|
|
|
print ( "[7][2][26] : %e"% df(a,x,y,2))
|
|
|
|
print ( "[7][3][26] : %e"% df(a,x,y,3))
|
|
|
|
print ( "[7][4][26] : %e"% lf(a,x,y))
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+RESULTS:
|
2021-09-23 11:16:54 +02:00
|
|
|
: [7][0][26] : 1.050157e-03
|
|
|
|
: [7][1][26] : -7.501497e-04
|
|
|
|
: [7][2][26] : -3.825069e-03
|
|
|
|
: [7][3][26] : 3.495056e-03
|
|
|
|
: [7][4][26] : 2.040013e-02
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+begin_src c :tangle (eval c_test) :exports none
|
2021-06-22 23:33:09 +02:00
|
|
|
{
|
2022-01-23 16:18:46 +01:00
|
|
|
#define walk_num 1 // chbrclf_walk_num
|
2021-06-10 22:57:59 +02:00
|
|
|
#define elec_num chbrclf_elec_num
|
|
|
|
#define prim_num chbrclf_prim_num
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
int64_t elec_up_num = chbrclf_elec_up_num;
|
|
|
|
int64_t elec_dn_num = chbrclf_elec_dn_num;
|
|
|
|
double* elec_coord = &(chbrclf_elec_coord[0][0][0]);
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
rc = qmckl_set_electron_num (context, elec_up_num, elec_dn_num);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
assert(qmckl_electron_provided(context));
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2023-09-14 11:00:24 +02:00
|
|
|
int64_t point_num = elec_num;
|
2022-08-07 14:57:10 +02:00
|
|
|
|
|
|
|
rc = qmckl_set_point(context, 'N', point_num, elec_coord, point_num*3);
|
2022-01-05 19:22:16 +01:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
double prim_vgl[point_num][5][prim_num];
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
rc = qmckl_get_ao_basis_primitive_vgl(context, &(prim_vgl[0][0][0]),
|
2022-08-07 14:57:10 +02:00
|
|
|
(int64_t) 5*point_num*prim_num );
|
2022-01-05 19:22:16 +01:00
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-06-10 22:57:59 +02:00
|
|
|
|
2022-06-15 23:21:31 +02:00
|
|
|
printf("prim_vgl[26][0][7] = %e\n",prim_vgl[26][0][7]);
|
2022-02-11 16:07:25 +01:00
|
|
|
assert( fabs(prim_vgl[26][0][7] - ( 1.0501570432064878E-003)) < 1.e-14 );
|
2022-06-15 23:21:31 +02:00
|
|
|
printf("prim_vgl[26][1][7] = %e\n",prim_vgl[26][1][7]);
|
2022-02-11 16:07:25 +01:00
|
|
|
assert( fabs(prim_vgl[26][1][7] - (-7.5014974095310560E-004)) < 1.e-14 );
|
2022-06-15 23:21:31 +02:00
|
|
|
printf("prim_vgl[26][2][7] = %e\n",prim_vgl[26][2][7]);
|
2022-02-11 16:07:25 +01:00
|
|
|
assert( fabs(prim_vgl[26][2][7] - (-3.8250692897610380E-003)) < 1.e-14 );
|
2022-06-15 23:21:31 +02:00
|
|
|
printf("prim_vgl[26][3][7] = %e\n",prim_vgl[26][3][7]);
|
2022-02-11 16:07:25 +01:00
|
|
|
assert( fabs(prim_vgl[26][3][7] - ( 3.4950559194080275E-003)) < 1.e-14 );
|
2022-06-15 23:21:31 +02:00
|
|
|
printf("prim_vgl[26][4][7] = %e\n",prim_vgl[26][4][7]);
|
2022-02-11 16:07:25 +01:00
|
|
|
assert( fabs(prim_vgl[26][4][7] - ( 2.0392163767356572E-002)) < 1.e-14 );
|
2021-10-13 15:01:23 +02:00
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
}
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+end_src
|
2021-06-14 12:53:38 +02:00
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
** Computation of shells
|
|
|
|
:PROPERTIES:
|
|
|
|
:Name: qmckl_compute_ao_basis_shell_gaussian_vgl
|
|
|
|
:CRetType: qmckl_exit_code
|
|
|
|
:FRetType: qmckl_exit_code
|
|
|
|
:END:
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+NAME: qmckl_ao_basis_shell_gaussian_vgl_args
|
2022-01-26 17:06:51 +01:00
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|---------------------+-----------------------------------+--------+----------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~prim_num~ | ~int64_t~ | in | Number of primitives |
|
|
|
|
| ~shell_num~ | ~int64_t~ | in | Number of shells |
|
|
|
|
| ~point_num~ | ~int64_t~ | in | Number of points |
|
|
|
|
| ~nucl_num~ | ~int64_t~ | in | Number of nuclei |
|
|
|
|
| ~nucleus_shell_num~ | ~int64_t[nucl_num]~ | in | Number of shells for each nucleus |
|
|
|
|
| ~nucleus_index~ | ~int64_t[nucl_num]~ | in | Index of the 1st shell of each nucleus |
|
2022-02-14 19:11:37 +01:00
|
|
|
| ~nucleus_range~ | ~double[nucl_num]~ | in | Range of the nucleus |
|
2022-01-26 17:06:51 +01:00
|
|
|
| ~shell_prim_index~ | ~int64_t[shell_num]~ | in | Index of the 1st primitive of each shell |
|
|
|
|
| ~shell_prim_num~ | ~int64_t[shell_num]~ | in | Number of primitives per shell |
|
|
|
|
| ~coord~ | ~double[3][point_num]~ | in | Coordinates |
|
|
|
|
| ~nucl_coord~ | ~double[3][nucl_num]~ | in | Nuclear coordinates |
|
|
|
|
| ~expo~ | ~double[prim_num]~ | in | Exponents of the primitives |
|
|
|
|
| ~coef_normalized~ | ~double[prim_num]~ | in | Coefficients of the primitives |
|
2022-02-11 16:07:25 +01:00
|
|
|
| ~shell_vgl~ | ~double[point_num][5][shell_num]~ | out | Value, gradients and Laplacian of the shells |
|
2022-01-05 15:56:25 +01:00
|
|
|
|
|
|
|
#+CALL: generate_c_header(table=qmckl_ao_basis_shell_gaussian_vgl_args,rettyp=get_value("CRetType"),fname="qmckl_compute_ao_basis_shell_gaussian_vgl"))
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_compute_ao_basis_shell_gaussian_vgl (
|
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t prim_num,
|
|
|
|
const int64_t shell_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
const int64_t point_num,
|
2022-01-05 15:56:25 +01:00
|
|
|
const int64_t nucl_num,
|
2022-01-05 19:22:16 +01:00
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const int64_t* nucleus_index,
|
2022-02-14 19:11:37 +01:00
|
|
|
const double* nucleus_range,
|
2022-01-05 19:22:16 +01:00
|
|
|
const int64_t* shell_prim_index,
|
|
|
|
const int64_t* shell_prim_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
const double* coord,
|
2022-01-05 19:22:16 +01:00
|
|
|
const double* nucl_coord,
|
|
|
|
const double* expo,
|
|
|
|
const double* coef_normalized,
|
2023-11-29 02:10:20 +01:00
|
|
|
double* const shell_vgl );
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-10-13 15:01:23 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+begin_src f90 :comments org :tangle (eval f) :noweb yes
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_compute_ao_basis_shell_gaussian_vgl( &
|
2022-01-26 17:06:51 +01:00
|
|
|
context, prim_num, shell_num, point_num, nucl_num, &
|
2022-02-14 19:11:37 +01:00
|
|
|
nucleus_shell_num, nucleus_index, nucleus_range, &
|
|
|
|
shell_prim_index, shell_prim_num, coord, nucl_coord, &
|
|
|
|
expo, coef_normalized, shell_vgl) &
|
2023-09-22 16:41:43 +02:00
|
|
|
bind(C) result(info)
|
|
|
|
|
|
|
|
use qmckl_constants
|
2024-01-11 14:33:40 +01:00
|
|
|
use qmckl, only: qmckl_get_numprec_precision
|
2023-09-22 16:41:43 +02:00
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
implicit none
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: prim_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: shell_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: point_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: nucl_num
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_shell_num(nucl_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_index(nucl_num)
|
|
|
|
real (c_double ) , intent(in) :: nucleus_range(nucl_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: shell_prim_index(shell_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: shell_prim_num(shell_num)
|
|
|
|
real (c_double ) , intent(in) :: coord(point_num,3)
|
|
|
|
real (c_double ) , intent(in) :: nucl_coord(nucl_num,3)
|
|
|
|
real (c_double ) , intent(in) :: expo(prim_num)
|
|
|
|
real (c_double ) , intent(in) :: coef_normalized(prim_num)
|
|
|
|
real (c_double ) , intent(out) :: shell_vgl(shell_num,5,point_num)
|
|
|
|
integer(qmckl_exit_code) :: info
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
integer*8 :: inucl, iprim, ipoint, ishell
|
2022-01-05 19:22:16 +01:00
|
|
|
integer*8 :: ishell_start, ishell_end
|
|
|
|
integer*8 :: iprim_start , iprim_end
|
2021-06-22 23:33:09 +02:00
|
|
|
double precision :: x, y, z, two_a, ar2, r2, v, cutoff
|
|
|
|
|
|
|
|
info = QMCKL_SUCCESS
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
! Don't compute exponentials when the result will be almost zero.
|
2024-01-11 14:33:40 +01:00
|
|
|
<<fortran_cutoff>>
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-02-14 19:11:37 +01:00
|
|
|
do ipoint = 1, point_num
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2022-02-14 19:11:37 +01:00
|
|
|
do inucl=1,nucl_num
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
x = coord(ipoint,1) - nucl_coord(inucl,1)
|
|
|
|
y = coord(ipoint,2) - nucl_coord(inucl,2)
|
|
|
|
z = coord(ipoint,3) - nucl_coord(inucl,3)
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2021-09-23 11:16:54 +02:00
|
|
|
r2 = x*x + y*y + z*z
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2023-11-30 12:56:06 +01:00
|
|
|
if (r2 > cutoff*nucleus_range(inucl)) then
|
2022-02-14 19:11:37 +01:00
|
|
|
cycle
|
|
|
|
end if
|
|
|
|
|
|
|
|
! C is zero-based, so shift bounds by one
|
|
|
|
ishell_start = nucleus_index(inucl) + 1
|
|
|
|
ishell_end = nucleus_index(inucl) + nucleus_shell_num(inucl)
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
do ishell=ishell_start, ishell_end
|
2021-07-08 19:20:19 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
shell_vgl(ishell, 1, ipoint) = 0.d0
|
|
|
|
shell_vgl(ishell, 2, ipoint) = 0.d0
|
|
|
|
shell_vgl(ishell, 3, ipoint) = 0.d0
|
|
|
|
shell_vgl(ishell, 4, ipoint) = 0.d0
|
|
|
|
shell_vgl(ishell, 5, ipoint) = 0.d0
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
iprim_start = shell_prim_index(ishell) + 1
|
|
|
|
iprim_end = shell_prim_index(ishell) + shell_prim_num(ishell)
|
|
|
|
|
|
|
|
do iprim = iprim_start, iprim_end
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2021-09-23 11:16:54 +02:00
|
|
|
ar2 = expo(iprim)*r2
|
|
|
|
if (ar2 > cutoff) then
|
|
|
|
cycle
|
|
|
|
end if
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2021-09-23 11:16:54 +02:00
|
|
|
v = coef_normalized(iprim) * dexp(-ar2)
|
|
|
|
two_a = -2.d0 * expo(iprim) * v
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
shell_vgl(ishell, 1, ipoint) = &
|
|
|
|
shell_vgl(ishell, 1, ipoint) + v
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
shell_vgl(ishell, 2, ipoint) = &
|
|
|
|
shell_vgl(ishell, 2, ipoint) + two_a * x
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
shell_vgl(ishell, 3, ipoint) = &
|
|
|
|
shell_vgl(ishell, 3, ipoint) + two_a * y
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
shell_vgl(ishell, 4, ipoint) = &
|
|
|
|
shell_vgl(ishell, 4, ipoint) + two_a * z
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
shell_vgl(ishell, 5, ipoint) = &
|
|
|
|
shell_vgl(ishell, 5, ipoint) + two_a * (3.d0 - 2.d0*ar2)
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
end do
|
2021-09-23 11:16:54 +02:00
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
end do
|
|
|
|
end do
|
2021-09-23 11:16:54 +02:00
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
end do
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
end function qmckl_compute_ao_basis_shell_gaussian_vgl
|
2022-01-05 19:22:16 +01:00
|
|
|
#+end_src
|
2021-06-22 23:33:09 +02:00
|
|
|
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Provide :noexport:
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+CALL: write_provider_header( group="ao_basis", data="shell_vgl" )
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval h_private_func) :noweb yes :export none
|
2022-01-05 15:56:25 +01:00
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_shell_vgl(qmckl_context context);
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+CALL: write_provider_pre( group="ao_basis", data="shell_vgl", dimension="ctx->ao_basis.shell_num * 5 * ctx->point.num")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
2022-01-05 15:56:25 +01:00
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_shell_vgl(qmckl_context context)
|
|
|
|
{
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_exit_code rc = QMCKL_SUCCESS;
|
2022-07-09 13:11:20 +02:00
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
|
|
|
"qmckl_provide_ao_basis_shell_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 15:56:25 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
if (!ctx->ao_basis.provided) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_NOT_PROVIDED,
|
|
|
|
"qmckl_provide_ao_basis_shell_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Compute if necessary */
|
2022-01-26 17:06:51 +01:00
|
|
|
if (ctx->point.date > ctx->ao_basis.shell_vgl_date) {
|
2022-01-05 15:56:25 +01:00
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->ao_basis.shell_num * 5 * ctx->point.num * sizeof(double);
|
|
|
|
|
|
|
|
if (ctx->ao_basis.shell_vgl != NULL) {
|
|
|
|
qmckl_memory_info_struct mem_info_test = qmckl_memory_info_struct_zero;
|
|
|
|
rc = qmckl_get_malloc_info(context, ctx->ao_basis.shell_vgl, &mem_info_test);
|
|
|
|
|
|
|
|
/* if rc != QMCKL_SUCCESS, we are maybe in an _inplace function because the
|
|
|
|
memory was not allocated with qmckl_malloc */
|
|
|
|
|
|
|
|
if ((rc == QMCKL_SUCCESS) && (mem_info_test.size != mem_info.size)) {
|
2022-07-09 13:11:20 +02:00
|
|
|
rc = qmckl_free(context, ctx->ao_basis.shell_vgl);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
ctx->ao_basis.shell_vgl = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
/* Allocate array */
|
|
|
|
if (ctx->ao_basis.shell_vgl == NULL) {
|
|
|
|
|
|
|
|
double* shell_vgl = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (shell_vgl == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_ao_basis_shell_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
ctx->ao_basis.shell_vgl = shell_vgl;
|
|
|
|
}
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-01-05 15:56:25 +01:00
|
|
|
if (ctx->ao_basis.type == 'G') {
|
|
|
|
rc = qmckl_compute_ao_basis_shell_gaussian_vgl(context,
|
2022-01-05 19:22:16 +01:00
|
|
|
ctx->ao_basis.prim_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
ctx->point.num,
|
2022-01-05 19:22:16 +01:00
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
2023-11-30 12:56:06 +01:00
|
|
|
ctx->ao_basis.nucleus_range,
|
2022-01-05 19:22:16 +01:00
|
|
|
ctx->ao_basis.shell_prim_index,
|
|
|
|
ctx->ao_basis.shell_prim_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
ctx->point.coord.data,
|
2022-01-23 16:18:46 +01:00
|
|
|
ctx->nucleus.coord.data,
|
2022-01-05 19:22:16 +01:00
|
|
|
ctx->ao_basis.exponent,
|
|
|
|
ctx->ao_basis.coefficient_normalized,
|
|
|
|
ctx->ao_basis.shell_vgl);
|
2022-01-05 15:56:25 +01:00
|
|
|
} else {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_FAILURE,
|
|
|
|
"compute_ao_basis_shell_vgl",
|
|
|
|
"Not yet implemented");
|
|
|
|
}
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
|
|
|
#+CALL: write_provider_post( group="ao_basis", data="shell_vgl" )
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
2022-01-05 15:56:25 +01:00
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx->ao_basis.shell_vgl_date = ctx->date;
|
|
|
|
}
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
2022-01-05 15:56:25 +01:00
|
|
|
|
|
|
|
*** Test :noexport:
|
|
|
|
|
2021-06-22 23:33:09 +02:00
|
|
|
#+begin_src python :results output :exports none
|
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
def f(a,x,y):
|
Integration of Verificarlo CI tests (#1)
* comment
* Update distance test code
The distance test has been updated to the latest version, with a first attempt at using vfc_probes inside it
* Functional implementation of vfc_probes in the distance tests
This commit has the first functional vfc_ci tests. Verificarlo tests
should be written over the existing tests, and they can be enabled with
the following configure command:
QMCKL_DEVEL=1 ./configure --prefix=$PWD/_install --enable-maintainer-mode --enable-vfc_ci CC="verificarlo-f -Mpreprocess -D VFC_CI" FC="verificarlo-f -Mpreprocess -D VFC_CI" --host=x86_64
The --enable-vfc_ci flag will trigger the linking of the vfc_ci
library. Moreover, as of now, the "-Mpreprocess" and "-D VFC_CI" flags
have to be specified directly here. There is probably an appropriate
macro to place those flags into but I couldn't find it yet, and could
only manage to build the tests this way.
When the VFC_CI preprocessor is defined, somme additional code to
register and export the test probes can be executed (see
qmckl_distance.org).
As of now, the tests are built as normal, even though they are expected
to fail :
make all
make check
From there, the test_qmckl_distance (and potentially the others)
executable can be called at will. This will typically be done
automatically by vfc_ci, but one could manually execute the executable
by defining the following env variables :
VFC_PROBES_OUTPUT="test.csv" VFC_BACKENDS="libinterflop_ieee.so"
depending on the export file and the Verificarlo backend to be used.
The next steps will be to define more tests such as this one, and to
integrate them into a Verificarlo CI workflow (by writing a
vfc_tests_config.json file and using the automatic CI setup
command).
* Error in FOrtran interface fixed
* Added missing Fortran interfaces
* Modify distance test and install process integration
All probes are now ignored using only the preprocessor (instead
of checking for a facultative argument) in the distance test.
Moreover,preprocessing can now be enabled correctly using FCFLAGS
(the issue seemed to come from the order of the arguments passed
to gfortran/verificarlo-f with the preprocessor arg having to come
first).
* Add vfc_probes to AO tests
vfc_probes have been added to qmckl_ao.org in the same way as
qmckl_distance.org, which means that it can be enabled or disabled at
compile time using the --enable-vfc_ci option.
qmckl_distance.org has been slightly modified with a better indentation,
and configure.ac now adds the "-D VFC_CI" flag to CFLAGS when vfc_ci is
enabled.
Co-authored-by: Anthony Scemama <scemama@irsamc.ups-tlse.fr>
2021-07-07 13:42:42 +02:00
|
|
|
return np.sum( [c * np.exp( -b*(np.linalg.norm(x-y))**2) for b,c in a] )
|
2021-06-22 23:33:09 +02:00
|
|
|
|
|
|
|
def df(a,x,y,n):
|
|
|
|
h0 = 1.e-6
|
|
|
|
if n == 1: h = np.array([h0,0.,0.])
|
|
|
|
elif n == 2: h = np.array([0.,h0,0.])
|
|
|
|
elif n == 3: h = np.array([0.,0.,h0])
|
|
|
|
return ( f(a,x+h,y) - f(a,x-h,y) ) / (2.*h0)
|
|
|
|
|
|
|
|
def d2f(a,x,y,n):
|
|
|
|
h0 = 1.e-6
|
|
|
|
if n == 1: h = np.array([h0,0.,0.])
|
|
|
|
elif n == 2: h = np.array([0.,h0,0.])
|
|
|
|
elif n == 3: h = np.array([0.,0.,h0])
|
|
|
|
return ( f(a,x+h,y) - 2.*f(a,x,y) + f(a,x-h,y) ) / h0**2
|
|
|
|
|
|
|
|
def lf(a,x,y):
|
|
|
|
return d2f(a,x,y,1) + d2f(a,x,y,2) + d2f(a,x,y,3)
|
|
|
|
|
|
|
|
elec_26_w1 = np.array( [ 1.49050402641, 2.90106987953, -1.05920815468 ] )
|
|
|
|
elec_15_w2 = np.array( [ -2.20180344582,-1.9113150239, 2.2193744778600002 ] )
|
|
|
|
nucl_1 = np.array( [ 1.096243353458458e+00, 8.907054016973815e-01, 7.777092280258892e-01 ] )
|
|
|
|
nucl_2 = np.array( [ 1.168459237342663e+00, 1.125660720053393e+00, 2.833370314829343e+00 ] )
|
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
#double prim_vgl[prim_num][5][point_num];
|
2021-06-22 23:33:09 +02:00
|
|
|
x = elec_26_w1 ; y = nucl_1
|
2021-07-11 23:20:40 +02:00
|
|
|
a = [( 8.236000E+03, -1.130000E-04 * 6.1616545431994848e+02 ),
|
|
|
|
( 1.235000E+03, -8.780000E-04 * 1.4847738511079908e+02 ),
|
|
|
|
( 2.808000E+02, -4.540000E-03 * 4.8888635917437597e+01 ),
|
|
|
|
( 7.927000E+01, -1.813300E-02 * 1.8933972232608955e+01 ),
|
|
|
|
( 2.559000E+01, -5.576000E-02 * 8.1089160941724145e+00 ),
|
|
|
|
( 8.997000E+00, -1.268950E-01 * 3.7024003863155635e+00 ),
|
|
|
|
( 3.319000E+00, -1.703520E-01 * 1.7525302846177560e+00 ),
|
|
|
|
( 9.059000E-01, 1.403820E-01 * 6.6179013183966806e-01 ),
|
|
|
|
( 3.643000E-01, 5.986840E-01 * 3.3419848027174592e-01 ),
|
|
|
|
( 1.285000E-01, 3.953890E-01 * 1.5296336817449557e-01 )]
|
|
|
|
|
2021-09-23 11:16:54 +02:00
|
|
|
print ( "[1][0][26] : %25.15e"% f(a,x,y))
|
|
|
|
print ( "[1][1][26] : %25.15e"% df(a,x,y,1))
|
|
|
|
print ( "[1][2][26] : %25.15e"% df(a,x,y,2))
|
|
|
|
print ( "[1][3][26] : %25.15e"% df(a,x,y,3))
|
|
|
|
print ( "[1][4][26] : %25.15e"% lf(a,x,y))
|
2021-06-22 23:33:09 +02:00
|
|
|
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+RESULTS:
|
2021-09-23 11:16:54 +02:00
|
|
|
: [1][0][26] : 3.564393437193867e-02
|
|
|
|
: [1][1][26] : -6.030177988891605e-03
|
|
|
|
: [1][2][26] : -3.074832579871845e-02
|
|
|
|
: [1][3][26] : 2.809546963133958e-02
|
|
|
|
: [1][4][26] : 1.903338597841753e-02
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+begin_src c :tangle (eval c_test) :exports none
|
2021-06-22 23:33:09 +02:00
|
|
|
{
|
|
|
|
#define shell_num chbrclf_shell_num
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
double* elec_coord = &(chbrclf_elec_coord[0][0][0]);
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
assert(qmckl_electron_provided(context));
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2023-09-14 11:00:24 +02:00
|
|
|
int64_t point_num = elec_num;
|
2022-08-07 14:57:10 +02:00
|
|
|
rc = qmckl_set_point(context, 'N', point_num, elec_coord, point_num*3);
|
2022-01-05 19:22:16 +01:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
2021-06-22 23:33:09 +02:00
|
|
|
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
double shell_vgl[point_num][5][shell_num];
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
rc = qmckl_get_ao_basis_shell_vgl(context, &(shell_vgl[0][0][0]),
|
2022-08-07 14:57:10 +02:00
|
|
|
(int64_t) 5*point_num*shell_num);
|
2022-01-05 19:22:16 +01:00
|
|
|
assert (rc == QMCKL_SUCCESS);
|
2021-06-22 23:33:09 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
printf(" shell_vgl[26][0][1] %25.15e\n", shell_vgl[26][0][1]);
|
|
|
|
printf(" shell_vgl[26][1][1] %25.15e\n", shell_vgl[26][1][1]);
|
|
|
|
printf(" shell_vgl[26][2][1] %25.15e\n", shell_vgl[26][2][1]);
|
|
|
|
printf(" shell_vgl[26][3][1] %25.15e\n", shell_vgl[26][3][1]);
|
|
|
|
printf(" shell_vgl[26][4][1] %25.15e\n", shell_vgl[26][4][1]);
|
2021-10-13 15:01:23 +02:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
assert( fabs(shell_vgl[26][0][1] - ( 3.564393437193868e-02)) < 1.e-14 );
|
|
|
|
assert( fabs(shell_vgl[26][1][1] - (-6.030177987072189e-03)) < 1.e-14 );
|
|
|
|
assert( fabs(shell_vgl[26][2][1] - (-3.074832579537582e-02)) < 1.e-14 );
|
|
|
|
assert( fabs(shell_vgl[26][3][1] - ( 2.809546963519935e-02)) < 1.e-14 );
|
|
|
|
assert( fabs(shell_vgl[26][4][1] - ( 1.896046117183968e-02)) < 1.e-14 );
|
2021-10-13 15:01:23 +02:00
|
|
|
|
|
|
|
}
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
#+end_src
|
2022-01-06 02:28:13 +01:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
* Polynomial part
|
2021-07-11 23:20:40 +02:00
|
|
|
|
|
|
|
Going from the atomic basis set to AOs implies a systematic
|
|
|
|
construction of all the angular functions of each shell. We
|
|
|
|
consider two cases for the angular functions: the real-valued
|
|
|
|
spherical harmonics, and the polynomials in Cartesian coordinates.
|
|
|
|
In the case of spherical harmonics, the AOs are ordered in
|
|
|
|
increasing magnetic quantum number ($-l \le m \le l$), and in the
|
|
|
|
case of polynomials we choose the canonical ordering, i.e
|
|
|
|
|
|
|
|
\begin{eqnarray}
|
|
|
|
p & : & p_x, p_y, p_z \nonumber \\
|
|
|
|
d & : & d_{xx}, d_{xy}, d_{xz}, d_{yy}, d_{yz}, d_{zz} \nonumber \\
|
|
|
|
f & : & f_{xxx}, f_{xxy}, f_{xxz}, f_{xyy}, f_{xyz}, f_{xzz}, f_{yyy}, f_{yyz}, f_{yzz}, f_{zzz} \nonumber \\
|
|
|
|
{\rm etc.} \nonumber
|
|
|
|
\end{eqnarray}
|
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
** General functions for Powers of $x-X_i$
|
|
|
|
:PROPERTIES:
|
|
|
|
:Name: qmckl_ao_power
|
|
|
|
:CRetType: qmckl_exit_code
|
|
|
|
:FRetType: qmckl_exit_code
|
|
|
|
:END:
|
|
|
|
|
|
|
|
The ~qmckl_ao_power~ function computes all the powers of the ~n~
|
|
|
|
input data up to the given maximum value given in input for each of
|
|
|
|
the $n$ points:
|
|
|
|
|
|
|
|
\[ P_{ik} = X_i^k \]
|
|
|
|
|
|
|
|
#+NAME: qmckl_ao_power_args
|
2022-01-05 15:56:25 +01:00
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|-----------+-----------------+--------+---------------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~n~ | int64_t | in | Number of values |
|
|
|
|
| ~X~ | double[n] | in | Array containing the input values |
|
|
|
|
| ~LMAX~ | int32_t[n] | in | Array containing the maximum power for each value |
|
|
|
|
| ~P~ | double[n][ldp] | out | Array containing all the powers of ~X~ |
|
|
|
|
| ~ldp~ | int64_t | in | Leading dimension of array ~P~ |
|
|
|
|
|
|
|
|
Requirements:
|
|
|
|
|
|
|
|
- ~context~ is not ~QMCKL_NULL_CONTEXT~
|
|
|
|
- ~n~ > 0
|
|
|
|
- ~X~ is allocated with at least $n \times 8$ bytes
|
|
|
|
- ~LMAX~ is allocated with at least $n \times 4$ bytes
|
|
|
|
- ~P~ is allocated with at least $n \times \max_i \text{LMAX}_i \times 8$ bytes
|
|
|
|
- ~LDP~ >= $\max_i$ ~LMAX[i]~
|
|
|
|
|
|
|
|
#+CALL: generate_c_header(table=qmckl_ao_power_args,rettyp=get_value("CRetType"),fname="qmckl_ao_power")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
2022-01-05 19:22:16 +01:00
|
|
|
qmckl_exit_code qmckl_ao_power (
|
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t n,
|
|
|
|
const double* X,
|
|
|
|
const int32_t* LMAX,
|
|
|
|
double* const P,
|
2023-11-29 02:10:20 +01:00
|
|
|
const int64_t ldp );
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval f)
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_ao_power(context, n, X, LMAX, P, ldp) &
|
|
|
|
bind(C) result(info)
|
|
|
|
use qmckl_constants
|
2021-06-10 22:57:59 +02:00
|
|
|
implicit none
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: n
|
2024-02-23 11:56:28 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: ldp
|
2023-09-22 16:41:43 +02:00
|
|
|
real (c_double ) , intent(in) :: X(n)
|
|
|
|
integer (c_int32_t) , intent(in) :: LMAX(n)
|
|
|
|
real (c_double ) , intent(out) :: P(ldp,n)
|
|
|
|
|
|
|
|
integer(qmckl_exit_code) :: info
|
|
|
|
integer(c_int64_t) :: i,k
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
info = QMCKL_SUCCESS
|
|
|
|
|
|
|
|
if (context == QMCKL_NULL_CONTEXT) then
|
|
|
|
info = QMCKL_INVALID_CONTEXT
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (n <= ldp) then
|
|
|
|
info = QMCKL_INVALID_ARG_2
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
k = MAXVAL(LMAX)
|
|
|
|
if (LDP < k) then
|
|
|
|
info = QMCKL_INVALID_ARG_6
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (k <= 0) then
|
|
|
|
info = QMCKL_INVALID_ARG_4
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
do i=1,n
|
|
|
|
P(1,i) = X(i)
|
|
|
|
do k=2,LMAX(i)
|
|
|
|
P(k,i) = P(k-1,i) * X(i)
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
end function qmckl_ao_power
|
2022-01-05 15:56:25 +01:00
|
|
|
#+end_src
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
#+CALL: generate_f_interface(table=qmckl_ao_power_args,rettyp=get_value("CRetType"),fname="qmckl_ao_power")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_exit_code) function qmckl_ao_power &
|
2021-06-10 22:57:59 +02:00
|
|
|
(context, n, X, LMAX, P, ldp) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use :: qmckl_constants
|
2021-06-10 22:57:59 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
2021-06-10 22:57:59 +02:00
|
|
|
integer (c_int64_t) , intent(in) , value :: n
|
2024-02-23 11:56:28 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: ldp
|
2022-01-05 19:22:16 +01:00
|
|
|
real (c_double ) , intent(in) :: X(n)
|
|
|
|
integer (c_int32_t) , intent(in) :: LMAX(n)
|
|
|
|
real (c_double ) , intent(out) :: P(ldp,n)
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
end function qmckl_ao_power
|
|
|
|
end interface
|
|
|
|
#+end_src
|
|
|
|
|
2022-01-05 15:56:25 +01:00
|
|
|
*** Test :noexport:
|
2021-06-10 22:57:59 +02:00
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval f_test)
|
2023-11-29 02:10:20 +01:00
|
|
|
function test_qmckl_ao_power(context) bind(C)
|
2021-06-10 22:57:59 +02:00
|
|
|
use qmckl
|
|
|
|
implicit none
|
|
|
|
|
|
|
|
integer(qmckl_context), intent(in), value :: context
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_exit_code) :: test_qmckl_ao_power
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-04-30 01:26:19 +02:00
|
|
|
integer*8 :: n, LDP
|
|
|
|
integer, allocatable :: LMAX(:)
|
2021-04-01 01:19:33 +02:00
|
|
|
double precision, allocatable :: X(:), P(:,:)
|
|
|
|
integer*8 :: i,j
|
|
|
|
double precision :: epsilon
|
|
|
|
|
2024-01-11 14:33:40 +01:00
|
|
|
epsilon = qmckl_get_numprec_precision(context)
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
n = 100;
|
|
|
|
LDP = 10;
|
|
|
|
|
|
|
|
allocate(X(n), P(LDP,n), LMAX(n))
|
|
|
|
|
|
|
|
do j=1,n
|
|
|
|
X(j) = -5.d0 + 0.1d0 * dble(j)
|
|
|
|
LMAX(j) = 1 + int(mod(j, 5),4)
|
|
|
|
end do
|
|
|
|
|
2021-04-30 01:26:19 +02:00
|
|
|
test_qmckl_ao_power = qmckl_ao_power(context, n, X, LMAX, P, LDP)
|
2021-04-01 01:19:33 +02:00
|
|
|
if (test_qmckl_ao_power /= QMCKL_SUCCESS) return
|
|
|
|
|
|
|
|
test_qmckl_ao_power = QMCKL_FAILURE
|
|
|
|
|
|
|
|
do j=1,n
|
|
|
|
do i=1,LMAX(j)
|
|
|
|
if ( X(j)**i == 0.d0 ) then
|
|
|
|
if ( P(i,j) /= 0.d0) return
|
|
|
|
else
|
|
|
|
if ( dabs(1.d0 - P(i,j) / (X(j)**i)) > epsilon ) return
|
|
|
|
end if
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
|
|
|
|
test_qmckl_ao_power = QMCKL_SUCCESS
|
|
|
|
deallocate(X,P,LMAX)
|
|
|
|
end function test_qmckl_ao_power
|
2021-04-16 00:57:08 +02:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-04-16 00:57:08 +02:00
|
|
|
#+begin_src c :tangle (eval c_test) :exports none
|
2022-01-05 15:56:25 +01:00
|
|
|
int test_qmckl_ao_power(qmckl_context context);
|
|
|
|
assert(0 == test_qmckl_ao_power(context));
|
2021-04-16 00:57:08 +02:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-06-10 22:57:59 +02:00
|
|
|
** General functions for Value, Gradient and Laplacian of a polynomial
|
2021-04-16 00:57:08 +02:00
|
|
|
:PROPERTIES:
|
|
|
|
:Name: qmckl_ao_polynomial_vgl
|
|
|
|
:CRetType: qmckl_exit_code
|
|
|
|
:FRetType: qmckl_exit_code
|
|
|
|
:END:
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
A polynomial is centered on a nucleus $\mathbf{R}_i$
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
\[
|
2021-04-30 01:26:19 +02:00
|
|
|
P_l(\mathbf{r},\mathbf{R}_i) = (x-X_i)^a (y-Y_i)^b (z-Z_i)^c
|
2021-04-01 01:19:33 +02:00
|
|
|
\]
|
|
|
|
|
|
|
|
The gradients with respect to electron coordinates are
|
2021-04-30 01:26:19 +02:00
|
|
|
|
|
|
|
\begin{eqnarray*}
|
2021-04-01 01:19:33 +02:00
|
|
|
\frac{\partial }{\partial x} P_l\left(\mathbf{r},\mathbf{R}_i \right) &
|
|
|
|
= & a (x-X_i)^{a-1} (y-Y_i)^b (z-Z_i)^c \\
|
|
|
|
\frac{\partial }{\partial y} P_l\left(\mathbf{r},\mathbf{R}_i \right) &
|
|
|
|
= & b (x-X_i)^a (y-Y_i)^{b-1} (z-Z_i)^c \\
|
|
|
|
\frac{\partial }{\partial z} P_l\left(\mathbf{r},\mathbf{R}_i \right) &
|
|
|
|
= & c (x-X_i)^a (y-Y_i)^b (z-Z_i)^{c-1} \\
|
2021-04-30 01:26:19 +02:00
|
|
|
\end{eqnarray*}
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
and the Laplacian is
|
2021-04-30 01:26:19 +02:00
|
|
|
|
|
|
|
\begin{eqnarray*}
|
|
|
|
\left( \frac{\partial }{\partial x^2} +
|
|
|
|
\frac{\partial }{\partial y^2} +
|
2021-04-01 01:19:33 +02:00
|
|
|
\frac{\partial }{\partial z^2} \right) P_l
|
2021-04-30 01:26:19 +02:00
|
|
|
\left(\mathbf{r},\mathbf{R}_i \right) & = &
|
2021-04-01 01:19:33 +02:00
|
|
|
a(a-1) (x-X_i)^{a-2} (y-Y_i)^b (z-Z_i)^c + \\
|
2023-11-29 19:53:03 +01:00
|
|
|
&& b(b-1) (x-X_i)^a (y-Y_i)^{b-2} (z-Z_i)^c + \\
|
|
|
|
&& c(c-1) (x-X_i)^a (y-Y_i)^b (z-Z_i)^{c-2}.
|
2021-04-01 01:19:33 +02:00
|
|
|
\end{eqnarray*}
|
|
|
|
|
|
|
|
~qmckl_ao_polynomial_vgl~ computes the values, gradients and
|
|
|
|
Laplacians at a given point in space, of all polynomials with an
|
|
|
|
angular momentum up to ~lmax~.
|
|
|
|
|
2021-04-16 00:57:08 +02:00
|
|
|
#+NAME: qmckl_ao_polynomial_vgl_args
|
2022-01-05 19:22:16 +01:00
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|-----------+-------------------+--------+------------------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~X~ | ~double[3]~ | in | Array containing the coordinates of the points |
|
|
|
|
| ~R~ | ~double[3]~ | in | Array containing the x,y,z coordinates of the center |
|
|
|
|
| ~lmax~ | ~int32_t~ | in | Maximum angular momentum |
|
|
|
|
| ~n~ | ~int64_t~ | inout | Number of computed polynomials |
|
|
|
|
| ~L~ | ~int32_t[n][ldl]~ | out | Contains a,b,c for all ~n~ results |
|
|
|
|
| ~ldl~ | ~int64_t~ | in | Leading dimension of ~L~ |
|
|
|
|
| ~VGL~ | ~double[n][ldv]~ | out | Value, gradients and Laplacian of the polynomials |
|
|
|
|
| ~ldv~ | ~int64_t~ | in | Leading dimension of array ~VGL~ |
|
|
|
|
|
|
|
|
Requirements:
|
|
|
|
|
|
|
|
- ~context~ \ne ~QMCKL_NULL_CONTEXT~
|
2021-04-16 00:57:08 +02:00
|
|
|
- ~n~ > 0
|
|
|
|
- ~lmax~ >= 0
|
|
|
|
- ~ldl~ >= 3
|
|
|
|
- ~ldv~ >= 5
|
|
|
|
- ~X~ is allocated with at least $3 \times 8$ bytes
|
|
|
|
- ~R~ is allocated with at least $3 \times 8$ bytes
|
|
|
|
- ~n~ >= ~(lmax+1)(lmax+2)(lmax+3)/6~
|
|
|
|
- ~L~ is allocated with at least $3 \times n \times 4$ bytes
|
|
|
|
- ~VGL~ is allocated with at least $5 \times n \times 8$ bytes
|
|
|
|
- On output, ~n~ should be equal to ~(lmax+1)(lmax+2)(lmax+3)/6~
|
|
|
|
- On output, the powers are given in the following order (l=a+b+c):
|
|
|
|
- Increasing values of ~l~
|
|
|
|
- Within a given value of ~l~, alphabetical order of the
|
|
|
|
string made by a*"x" + b*"y" + c*"z" (in Python notation).
|
|
|
|
For example, with a=0, b=2 and c=1 the string is "yyz"
|
|
|
|
|
|
|
|
#+CALL: generate_c_header(table=qmckl_ao_polynomial_vgl_args,rettyp=get_value("CRetType"),fname=get_value("Name"))
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_ao_polynomial_vgl (
|
|
|
|
const qmckl_context context,
|
|
|
|
const double* X,
|
|
|
|
const double* R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* const VGL,
|
2023-11-29 02:10:20 +01:00
|
|
|
const int64_t ldv );
|
2021-04-16 00:57:08 +02:00
|
|
|
#+end_src
|
|
|
|
|
2022-02-16 01:12:42 +01:00
|
|
|
#+CALL: generate_c_header(table=qmckl_ao_polynomial_vgl_args,rettyp=get_value("CRetType"),fname="qmckl_ao_polynomial_vgl_doc")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_ao_polynomial_vgl_doc (
|
|
|
|
const qmckl_context context,
|
|
|
|
const double* X,
|
|
|
|
const double* R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* const VGL,
|
2023-11-29 02:10:20 +01:00
|
|
|
const int64_t ldv );
|
2022-02-16 01:12:42 +01:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval c) :comments org
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_ao_polynomial_vgl (const qmckl_context context,
|
|
|
|
const double* X,
|
|
|
|
const double* R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* const VGL,
|
|
|
|
const int64_t ldv )
|
|
|
|
{
|
|
|
|
#ifdef HAVE_HPC
|
|
|
|
//return qmckl_ao_polynomial_vgl_hpc (context, X, R, lmax, n, L, ldl, VGL, ldv);
|
2023-09-22 16:41:43 +02:00
|
|
|
return qmckl_ao_polynomial_vgl_doc
|
2022-02-16 01:12:42 +01:00
|
|
|
#else
|
2023-09-22 16:41:43 +02:00
|
|
|
return qmckl_ao_polynomial_vgl_doc
|
2022-02-19 19:24:18 +01:00
|
|
|
#endif
|
2023-09-22 16:41:43 +02:00
|
|
|
(context, X, R, lmax, n, L, ldl, VGL, ldv);
|
2022-02-16 01:12:42 +01:00
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
2021-04-16 00:57:08 +02:00
|
|
|
#+begin_src f90 :tangle (eval f)
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_ao_polynomial_vgl_doc (context, &
|
|
|
|
X, R, lmax, n, L, ldl, VGL, ldv) &
|
|
|
|
bind(C) result(info)
|
|
|
|
use qmckl_constants
|
2021-04-01 01:19:33 +02:00
|
|
|
implicit none
|
2023-09-22 16:41:43 +02:00
|
|
|
|
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
real (c_double ) , intent(in) :: X(3)
|
|
|
|
real (c_double ) , intent(in) :: R(3)
|
|
|
|
integer (c_int32_t) , intent(in) , value :: lmax
|
|
|
|
integer (c_int64_t) , intent(inout) :: n
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldl
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldv
|
2024-02-23 16:37:31 +01:00
|
|
|
integer (c_int32_t) , intent(out) :: L(ldl,(lmax+1)*(lmax+2)*(lmax+3)/6)
|
|
|
|
real (c_double ) , intent(out) :: VGL(ldv,(lmax+1)*(lmax+2)*(lmax+3)/6)
|
2023-09-22 16:41:43 +02:00
|
|
|
|
|
|
|
integer(qmckl_exit_code) :: info
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
integer*8 :: i,j
|
|
|
|
integer :: a,b,c,d
|
2023-08-24 09:59:57 +02:00
|
|
|
double precision :: Y(3)
|
|
|
|
double precision :: pows(-2:lmax,3)
|
2021-04-01 01:19:33 +02:00
|
|
|
double precision :: xy, yz, xz
|
|
|
|
double precision :: da, db, dc, dd
|
|
|
|
|
|
|
|
info = 0
|
|
|
|
|
|
|
|
if (context == QMCKL_NULL_CONTEXT) then
|
|
|
|
info = QMCKL_INVALID_CONTEXT
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (lmax < 0) then
|
|
|
|
info = QMCKL_INVALID_ARG_4
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (ldl < 3) then
|
|
|
|
info = QMCKL_INVALID_ARG_7
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (ldv < 5) then
|
|
|
|
info = QMCKL_INVALID_ARG_9
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
|
2024-05-08 13:56:32 +02:00
|
|
|
! The shift below is such that polynomials will not make the AO equal to zero at the nodes of the orbitals
|
2021-04-01 01:19:33 +02:00
|
|
|
do i=1,3
|
2024-11-04 16:44:28 +01:00
|
|
|
Y(i) = (X(i) - R(i)) + 1.d-20
|
2021-04-01 01:19:33 +02:00
|
|
|
end do
|
|
|
|
|
|
|
|
if (lmax == 0) then
|
|
|
|
VGL(1,1) = 1.d0
|
2023-08-24 09:59:57 +02:00
|
|
|
VGL(2,1) = 0.d0
|
|
|
|
VGL(3,1) = 0.d0
|
|
|
|
VGL(4,1) = 0.d0
|
|
|
|
VGL(5,1) = 0.d0
|
|
|
|
l(1,1) = 0
|
|
|
|
l(2,1) = 0
|
|
|
|
l(3,1) = 0
|
2021-04-01 01:19:33 +02:00
|
|
|
n=1
|
|
|
|
else if (lmax > 0) then
|
|
|
|
pows(-2:0,1:3) = 1.d0
|
|
|
|
do i=1,lmax
|
2021-04-30 01:26:19 +02:00
|
|
|
pows(i,1) = pows(i-1,1) * Y(1)
|
|
|
|
pows(i,2) = pows(i-1,2) * Y(2)
|
|
|
|
pows(i,3) = pows(i-1,3) * Y(3)
|
2021-04-01 01:19:33 +02:00
|
|
|
end do
|
|
|
|
|
|
|
|
VGL(1:5,1:4) = 0.d0
|
2023-08-24 09:59:57 +02:00
|
|
|
VGL(1,1) = 1.d0
|
2022-02-09 18:06:46 +01:00
|
|
|
VGL(1,2) = pows(1,1)
|
|
|
|
VGL(2,2) = 1.d0
|
|
|
|
VGL(1,3) = pows(1,2)
|
|
|
|
VGL(3,3) = 1.d0
|
|
|
|
VGL(1,4) = pows(1,3)
|
|
|
|
VGL(4,4) = 1.d0
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2023-08-24 09:59:57 +02:00
|
|
|
l (1:3,1:4) = 0
|
|
|
|
l (1,2) = 1
|
|
|
|
l (2,3) = 1
|
|
|
|
l (3,4) = 1
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
n=4
|
|
|
|
endif
|
|
|
|
|
|
|
|
! l>=2
|
|
|
|
dd = 2.d0
|
|
|
|
do d=2,lmax
|
|
|
|
da = dd
|
|
|
|
do a=d,0,-1
|
|
|
|
db = dd-da
|
|
|
|
do b=d-a,0,-1
|
|
|
|
c = d - a - b
|
|
|
|
dc = dd - da - db
|
|
|
|
n = n+1
|
|
|
|
|
|
|
|
l(1,n) = a
|
|
|
|
l(2,n) = b
|
|
|
|
l(3,n) = c
|
|
|
|
|
|
|
|
xy = pows(a,1) * pows(b,2)
|
|
|
|
yz = pows(b,2) * pows(c,3)
|
|
|
|
xz = pows(a,1) * pows(c,3)
|
|
|
|
|
2022-02-09 18:06:46 +01:00
|
|
|
VGL(1,n) = xy * pows(c,3)
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
xy = dc * xy
|
|
|
|
xz = db * xz
|
|
|
|
yz = da * yz
|
|
|
|
|
2022-02-09 18:06:46 +01:00
|
|
|
VGL(2,n) = pows(a-1,1) * yz
|
|
|
|
VGL(3,n) = pows(b-1,2) * xz
|
|
|
|
VGL(4,n) = pows(c-1,3) * xy
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2022-02-09 18:06:46 +01:00
|
|
|
VGL(5,n) = &
|
2021-04-01 01:19:33 +02:00
|
|
|
(da-1.d0) * pows(a-2,1) * yz + &
|
|
|
|
(db-1.d0) * pows(b-2,2) * xz + &
|
|
|
|
(dc-1.d0) * pows(c-2,3) * xy
|
|
|
|
|
|
|
|
db = db - 1.d0
|
|
|
|
end do
|
|
|
|
da = da - 1.d0
|
|
|
|
end do
|
|
|
|
dd = dd + 1.d0
|
|
|
|
end do
|
|
|
|
|
|
|
|
info = QMCKL_SUCCESS
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
end function qmckl_ao_polynomial_vgl_doc
|
2022-02-16 01:12:42 +01:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+CALL: generate_f_interface(table=qmckl_ao_polynomial_vgl_args,rettyp=get_value("FRetType"),fname="qmckl_ao_polynomial_vgl_doc" )
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_exit_code) function qmckl_ao_polynomial_vgl_doc &
|
2022-02-16 01:12:42 +01:00
|
|
|
(context, X, R, lmax, n, L, ldl, VGL, ldv) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-02-16 01:12:42 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
2022-02-16 01:12:42 +01:00
|
|
|
real (c_double ) , intent(in) :: X(3)
|
|
|
|
real (c_double ) , intent(in) :: R(3)
|
|
|
|
integer (c_int32_t) , intent(in) , value :: lmax
|
|
|
|
integer (c_int64_t) , intent(inout) :: n
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldl
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldv
|
2024-02-23 11:56:28 +01:00
|
|
|
integer (c_int32_t) , intent(out) :: L(ldl,n)
|
|
|
|
real (c_double ) , intent(out) :: VGL(ldv,n)
|
2022-02-16 01:12:42 +01:00
|
|
|
|
|
|
|
end function qmckl_ao_polynomial_vgl_doc
|
|
|
|
end interface
|
2021-04-16 00:57:08 +02:00
|
|
|
#+end_src
|
|
|
|
|
2022-02-16 01:12:42 +01:00
|
|
|
#+CALL: generate_f_interface(table=qmckl_ao_polynomial_vgl_args,rettyp=get_value("FRetType"),fname="qmckl_ao_polynomial_vgl" )
|
2021-04-16 00:57:08 +02:00
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_exit_code) function qmckl_ao_polynomial_vgl &
|
2021-04-16 00:57:08 +02:00
|
|
|
(context, X, R, lmax, n, L, ldl, VGL, ldv) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2021-04-16 00:57:08 +02:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
2021-04-16 00:57:08 +02:00
|
|
|
real (c_double ) , intent(in) :: X(3)
|
|
|
|
real (c_double ) , intent(in) :: R(3)
|
|
|
|
integer (c_int32_t) , intent(in) , value :: lmax
|
2024-02-23 11:56:28 +01:00
|
|
|
integer (c_int64_t) , intent(inout) :: n
|
2021-04-16 00:57:08 +02:00
|
|
|
integer (c_int64_t) , intent(in) , value :: ldl
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldv
|
2024-02-23 11:56:28 +01:00
|
|
|
integer (c_int32_t) , intent(out) :: L(ldl,n)
|
|
|
|
real (c_double ) , intent(out) :: VGL(ldv,n)
|
2021-04-16 00:57:08 +02:00
|
|
|
|
|
|
|
end function qmckl_ao_polynomial_vgl
|
|
|
|
end interface
|
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2022-02-16 01:12:42 +01:00
|
|
|
#+CALL: generate_c_header(table=qmckl_ao_polynomial_vgl_args,rettyp=get_value("CRetType"),fname="qmckl_ao_polynomial_transp_vgl")
|
2022-02-11 16:07:25 +01:00
|
|
|
|
2022-02-16 01:12:42 +01:00
|
|
|
#+RESULTS:
|
2022-02-11 16:07:25 +01:00
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_ao_polynomial_transp_vgl (
|
|
|
|
const qmckl_context context,
|
|
|
|
const double* X,
|
|
|
|
const double* R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* const VGL,
|
2023-11-29 02:10:20 +01:00
|
|
|
const int64_t ldv );
|
2022-02-16 01:12:42 +01:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+CALL: generate_c_header(table=qmckl_ao_polynomial_vgl_args,rettyp=get_value("CRetType"),fname="qmckl_ao_polynomial_transp_vgl_doc")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_ao_polynomial_transp_vgl_doc (
|
|
|
|
const qmckl_context context,
|
|
|
|
const double* X,
|
|
|
|
const double* R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* const VGL,
|
2023-11-29 02:10:20 +01:00
|
|
|
const int64_t ldv );
|
2022-02-16 01:12:42 +01:00
|
|
|
#+end_src
|
|
|
|
|
2022-10-13 11:48:40 +02:00
|
|
|
# #+CALL: generate_c_header(table=qmckl_ao_polynomial_vgl_args,rettyp=get_value("CRetType"),fname="qmckl_ao_polynomial_transp_vgl_hpc")
|
2022-02-16 15:14:41 +01:00
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :tangle (eval h_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_ao_polynomial_transp_vgl_hpc (
|
|
|
|
const qmckl_context context,
|
|
|
|
const double* X,
|
|
|
|
const double* R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* const VGL,
|
2022-02-19 19:24:18 +01:00
|
|
|
const int64_t ldv );
|
2022-02-16 15:14:41 +01:00
|
|
|
#+end_src
|
|
|
|
|
2022-02-16 01:12:42 +01:00
|
|
|
#+begin_src c :tangle (eval c) :comments org
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_ao_polynomial_transp_vgl (const qmckl_context context,
|
|
|
|
const double* X,
|
|
|
|
const double* R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* const VGL,
|
|
|
|
const int64_t ldv )
|
|
|
|
{
|
|
|
|
#ifdef HAVE_HPC
|
2023-09-22 16:41:43 +02:00
|
|
|
return qmckl_ao_polynomial_transp_vgl_hpc
|
2022-02-16 01:12:42 +01:00
|
|
|
#else
|
2023-09-22 16:41:43 +02:00
|
|
|
return qmckl_ao_polynomial_transp_vgl_doc
|
2022-02-19 19:24:18 +01:00
|
|
|
#endif
|
2023-09-22 16:41:43 +02:00
|
|
|
(context, X, R, lmax, n, L, ldl, VGL, ldv);
|
2022-02-16 01:12:42 +01:00
|
|
|
}
|
2022-02-11 16:07:25 +01:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval f)
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_ao_polynomial_transp_vgl_doc (context, &
|
|
|
|
X, R, lmax, n, L, ldl, VGL, ldv) &
|
|
|
|
bind(C) result(info)
|
|
|
|
|
|
|
|
use qmckl_constants
|
2022-02-11 16:07:25 +01:00
|
|
|
implicit none
|
2023-09-22 16:41:43 +02:00
|
|
|
|
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
real (c_double ) , intent(in) :: X(3)
|
|
|
|
real (c_double ) , intent(in) :: R(3)
|
|
|
|
integer (c_int32_t) , intent(in) , value :: lmax
|
|
|
|
integer (c_int64_t) , intent(inout) :: n
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldl
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldv
|
2024-02-23 16:37:31 +01:00
|
|
|
integer (c_int32_t) , intent(out) :: L(ldl,(lmax+1)*(lmax+2)*(lmax+3)/6)
|
|
|
|
real (c_double ) , intent(out) :: VGL(ldv,5)
|
2023-09-22 16:41:43 +02:00
|
|
|
|
|
|
|
integer(qmckl_exit_code) :: info
|
2022-02-11 16:07:25 +01:00
|
|
|
|
|
|
|
integer*8 :: i,j
|
|
|
|
integer :: a,b,c,d
|
|
|
|
real*8 :: Y(3)
|
2022-02-14 19:11:37 +01:00
|
|
|
real*8 :: pows(-2:21,3) ! lmax < 22
|
2022-02-11 16:07:25 +01:00
|
|
|
double precision :: xy, yz, xz
|
|
|
|
double precision :: da, db, dc, dd
|
|
|
|
|
|
|
|
info = 0
|
|
|
|
|
|
|
|
if (context == QMCKL_NULL_CONTEXT) then
|
|
|
|
info = QMCKL_INVALID_CONTEXT
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (lmax < 0) then
|
|
|
|
info = QMCKL_INVALID_ARG_4
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (ldl < 3) then
|
|
|
|
info = QMCKL_INVALID_ARG_7
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
if (ldv < (lmax+1)*(lmax+2)*(lmax+3)/6) then
|
|
|
|
info = QMCKL_INVALID_ARG_9
|
|
|
|
return
|
|
|
|
endif
|
|
|
|
|
|
|
|
|
2022-02-14 19:11:37 +01:00
|
|
|
if (lmax > 0) then
|
|
|
|
|
|
|
|
do i=1,3
|
|
|
|
Y(i) = X(i) - R(i)
|
|
|
|
end do
|
2022-02-11 16:07:25 +01:00
|
|
|
pows(-2:0,1:3) = 1.d0
|
|
|
|
do i=1,lmax
|
|
|
|
pows(i,1) = pows(i-1,1) * Y(1)
|
|
|
|
pows(i,2) = pows(i-1,2) * Y(2)
|
|
|
|
pows(i,3) = pows(i-1,3) * Y(3)
|
|
|
|
end do
|
|
|
|
|
|
|
|
l (1:3,1:4) = 0
|
|
|
|
VGL(1:4,1:5) = 0.d0
|
|
|
|
|
|
|
|
VGL(1 ,1 ) = 1.d0
|
|
|
|
|
|
|
|
l (1,2) = 1
|
2022-02-16 15:14:41 +01:00
|
|
|
VGL(2,1) = Y(1)
|
2022-02-11 16:07:25 +01:00
|
|
|
VGL(2,2) = 1.d0
|
|
|
|
|
|
|
|
l (2,3) = 1
|
2022-02-16 15:14:41 +01:00
|
|
|
VGL(3,1) = Y(2)
|
2022-02-11 16:07:25 +01:00
|
|
|
VGL(3,3) = 1.d0
|
|
|
|
|
|
|
|
l (3,4) = 1
|
2022-02-16 15:14:41 +01:00
|
|
|
VGL(4,1) = Y(3)
|
2022-02-11 16:07:25 +01:00
|
|
|
VGL(4,4) = 1.d0
|
|
|
|
|
|
|
|
n=4
|
2022-02-19 19:24:18 +01:00
|
|
|
else
|
2022-02-14 19:11:37 +01:00
|
|
|
VGL(1,1) = 1.d0
|
|
|
|
VGL(1,2:5) = 0.d0
|
|
|
|
l(1:3,1) = 0
|
|
|
|
n=1
|
|
|
|
return
|
2022-02-11 16:07:25 +01:00
|
|
|
endif
|
|
|
|
|
|
|
|
! l>=2
|
|
|
|
dd = 2.d0
|
|
|
|
do d=2,lmax
|
|
|
|
da = dd
|
|
|
|
do a=d,0,-1
|
|
|
|
db = dd-da
|
|
|
|
do b=d-a,0,-1
|
|
|
|
c = d - a - b
|
|
|
|
dc = dd - da - db
|
|
|
|
n = n+1
|
|
|
|
|
|
|
|
xy = pows(a,1) * pows(b,2)
|
|
|
|
yz = pows(b,2) * pows(c,3)
|
|
|
|
xz = pows(a,1) * pows(c,3)
|
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
l(1,n) = a
|
|
|
|
l(2,n) = b
|
|
|
|
l(3,n) = c
|
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
VGL(n,1) = xy * pows(c,3)
|
|
|
|
|
|
|
|
xy = dc * xy
|
|
|
|
xz = db * xz
|
|
|
|
yz = da * yz
|
|
|
|
|
|
|
|
VGL(n,2) = pows(a-1,1) * yz
|
|
|
|
VGL(n,3) = pows(b-1,2) * xz
|
|
|
|
VGL(n,4) = pows(c-1,3) * xy
|
|
|
|
|
|
|
|
VGL(n,5) = &
|
|
|
|
(da-1.d0) * pows(a-2,1) * yz + &
|
|
|
|
(db-1.d0) * pows(b-2,2) * xz + &
|
|
|
|
(dc-1.d0) * pows(c-2,3) * xy
|
|
|
|
|
|
|
|
db = db - 1.d0
|
|
|
|
end do
|
|
|
|
da = da - 1.d0
|
|
|
|
end do
|
|
|
|
dd = dd + 1.d0
|
|
|
|
end do
|
|
|
|
|
|
|
|
info = QMCKL_SUCCESS
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
end function qmckl_ao_polynomial_transp_vgl_doc
|
2022-02-11 16:07:25 +01:00
|
|
|
#+end_src
|
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
#+begin_src c :tangle (eval c) :comments org
|
2022-10-13 11:48:40 +02:00
|
|
|
static inline qmckl_exit_code
|
|
|
|
qmckl_ao_polynomial_transp_vgl_hpc_inline (const qmckl_context context,
|
2022-03-21 18:32:39 +01:00
|
|
|
const double* restrict X,
|
|
|
|
const double* restrict R,
|
2022-02-16 15:14:41 +01:00
|
|
|
const int32_t lmax,
|
2022-05-06 11:30:21 +02:00
|
|
|
int64_t* n,
|
2022-03-21 18:32:39 +01:00
|
|
|
int32_t* restrict const L,
|
2022-02-16 15:14:41 +01:00
|
|
|
const int64_t ldl,
|
2022-03-21 18:32:39 +01:00
|
|
|
double* restrict const VGL,
|
2022-02-16 15:14:41 +01:00
|
|
|
const int64_t ldv )
|
|
|
|
{
|
2022-04-05 11:03:38 +02:00
|
|
|
const qmckl_context_struct* ctx = (qmckl_context_struct*) context;
|
2022-02-16 15:14:41 +01:00
|
|
|
assert (ctx != NULL && X != NULL && R != NULL && n != NULL && L != NULL && VGL != NULL);
|
|
|
|
if (lmax < 0) return QMCKL_INVALID_ARG_4;
|
|
|
|
if (ldl < 3) return QMCKL_INVALID_ARG_7;
|
|
|
|
|
|
|
|
int32_t m;
|
2024-05-08 13:56:32 +02:00
|
|
|
const double shift=1.e-20;
|
2022-02-16 15:14:41 +01:00
|
|
|
|
|
|
|
switch (lmax) {
|
|
|
|
case 0:
|
|
|
|
{
|
|
|
|
if (ldv < 1) return QMCKL_INVALID_ARG_9;
|
2022-02-16 19:40:14 +01:00
|
|
|
L[0] = 0; L[1] = 0; L[2] = 0;
|
2022-02-16 15:14:41 +01:00
|
|
|
|
2022-02-16 19:49:05 +01:00
|
|
|
VGL[0 ] = 1.0;
|
|
|
|
VGL[ldv ] = 0.0;
|
|
|
|
VGL[ldv<<1 ] = 0.0;
|
2022-02-19 19:24:18 +01:00
|
|
|
VGL[(ldv<<1)+ldv] = 0.0;
|
2022-02-16 19:49:05 +01:00
|
|
|
VGL[ldv<<2 ] = 0.0;
|
2022-02-16 19:40:14 +01:00
|
|
|
m=1;
|
2022-02-16 15:14:41 +01:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
case 1:
|
|
|
|
{
|
|
|
|
if (ldv < 4) return QMCKL_INVALID_ARG_9;
|
|
|
|
|
2022-02-16 19:40:14 +01:00
|
|
|
if (ldl == 3) {
|
|
|
|
|
|
|
|
const int32_t ltmp[12] = {0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1};
|
|
|
|
for (int i=0 ; i<12 ; ++i)
|
|
|
|
L[i] = ltmp[i];
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 19:40:14 +01:00
|
|
|
} else {
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
int32_t* restrict const l[4] = {L, L+ldl, L+(ldl<<1), L+ldl+(ldl<<1)};
|
2022-02-16 19:40:14 +01:00
|
|
|
l[0][0] = 0; l[0][1] = 0; l[0][2] = 0;
|
|
|
|
l[1][0] = 1; l[1][1] = 0; l[1][2] = 0;
|
|
|
|
l[2][0] = 0; l[2][1] = 1; l[2][2] = 0;
|
|
|
|
l[3][0] = 0; l[3][1] = 0; l[3][2] = 1;
|
2022-02-16 15:14:41 +01:00
|
|
|
}
|
|
|
|
|
2022-02-16 19:40:14 +01:00
|
|
|
|
|
|
|
if (ldv == 4) {
|
|
|
|
|
2024-11-04 16:44:28 +01:00
|
|
|
const double tmp[20] = {1.0, (X[0]-R[0])+shift, (X[1]-R[1])+shift, (X[2]-R[2])+shift,
|
2022-02-17 12:36:16 +01:00
|
|
|
0.0, 1.0, 0.0, 0.0,
|
|
|
|
0.0, 0.0, 1.0, 0.0,
|
|
|
|
0.0, 0.0, 0.0, 1.0,
|
|
|
|
0.0, 0.0, 0.0, 0.0};
|
|
|
|
|
2022-02-16 19:40:14 +01:00
|
|
|
for (int i=0 ; i<20 ; ++i)
|
|
|
|
VGL[i] = tmp[i];
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
double* restrict const vgl1 = VGL;
|
|
|
|
double* restrict const vgl2 = VGL + ldv;
|
|
|
|
double* restrict const vgl3 = VGL + (ldv << 1);
|
|
|
|
double* restrict const vgl4 = VGL + ldv + (ldv << 1);
|
|
|
|
double* restrict const vgl5 = VGL + (ldv << 2);
|
2022-02-16 19:40:14 +01:00
|
|
|
|
|
|
|
for (int32_t k=0 ; k<4 ; ++k) {
|
|
|
|
vgl2[k] = 0.0;
|
|
|
|
vgl3[k] = 0.0;
|
|
|
|
vgl4[k] = 0.0;
|
|
|
|
vgl5[k] = 0.0;
|
|
|
|
}
|
|
|
|
vgl1[0] = 1.0;
|
2024-11-04 16:44:28 +01:00
|
|
|
vgl1[1] = (X[0]-R[0])+shift;
|
|
|
|
vgl1[2] = (X[1]-R[1])+shift;
|
|
|
|
vgl1[3] = (X[2]-R[2])+shift;
|
2022-02-16 19:40:14 +01:00
|
|
|
vgl2[1] = 1.0;
|
|
|
|
vgl3[2] = 1.0;
|
|
|
|
vgl4[3] = 1.0;
|
2022-02-16 15:14:41 +01:00
|
|
|
}
|
|
|
|
m=4;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case 2:
|
|
|
|
{
|
|
|
|
if (ldv < 10) return QMCKL_INVALID_ARG_9;
|
2022-02-16 19:40:14 +01:00
|
|
|
if (ldl == 3) {
|
2022-02-19 19:24:18 +01:00
|
|
|
const int32_t ltmp[30] = {0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1,
|
2022-02-16 19:40:14 +01:00
|
|
|
2, 0, 0, 1, 1, 0, 1, 0, 1, 0, 2, 0,
|
|
|
|
0, 1, 1, 0, 0, 2};
|
|
|
|
for (int i=0 ; i<30 ; ++i)
|
|
|
|
L[i] = ltmp[i];
|
|
|
|
|
|
|
|
} else {
|
2022-03-21 18:32:39 +01:00
|
|
|
int32_t* restrict l[10];
|
2022-02-16 19:40:14 +01:00
|
|
|
for (int32_t i=0 ; i<10 ; ++i) {
|
|
|
|
l[i] = L + i*ldl;
|
|
|
|
}
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 19:40:14 +01:00
|
|
|
l[0][0] = 0; l[0][1] = 0; l[0][2] = 0;
|
|
|
|
l[1][0] = 1; l[1][1] = 0; l[1][2] = 0;
|
|
|
|
l[2][0] = 0; l[2][1] = 1; l[2][2] = 0;
|
2022-02-19 19:24:18 +01:00
|
|
|
l[3][0] = 0; l[3][1] = 0; l[3][2] = 1;
|
2022-02-16 19:40:14 +01:00
|
|
|
l[4][0] = 2; l[4][1] = 0; l[4][2] = 0;
|
|
|
|
l[5][0] = 1; l[5][1] = 1; l[5][2] = 0;
|
|
|
|
l[6][0] = 1; l[6][1] = 0; l[6][2] = 1;
|
|
|
|
l[7][0] = 0; l[7][1] = 2; l[7][2] = 0;
|
|
|
|
l[8][0] = 0; l[8][1] = 1; l[8][2] = 1;
|
|
|
|
l[9][0] = 0; l[9][1] = 0; l[9][2] = 2;
|
|
|
|
}
|
|
|
|
|
2022-02-19 19:24:18 +01:00
|
|
|
const double Y[3] = { X[0]-R[0],
|
|
|
|
X[1]-R[1],
|
2022-02-16 15:14:41 +01:00
|
|
|
X[2]-R[2] };
|
|
|
|
|
2022-02-16 19:40:14 +01:00
|
|
|
if (ldv == 50) {
|
|
|
|
const double tmp[50] = {
|
2024-05-08 13:56:32 +02:00
|
|
|
1.0, Y[0]+shift, Y[1]+shift, Y[2]+shift,
|
|
|
|
Y[0] * Y[0]+shift, Y[0] * Y[1]+shift, Y[0] * Y[2]+shift,
|
|
|
|
Y[1] * Y[1]+shift, Y[1] * Y[2]+shift, Y[2] * Y[2]+shift,
|
|
|
|
0.0, 1.0, 0.0, 0.0,
|
|
|
|
Y[0] + Y[0], Y[1], Y[2], 0.0, 0.0, 0.0,
|
2022-02-16 19:40:14 +01:00
|
|
|
0.0, 0.0, 1.0, 0.0, 0.0,
|
|
|
|
Y[0], 0.0, Y[1] + Y[1], Y[2], 0.0,
|
|
|
|
0.0, 0.0, 0.0, 1.0, 0.0,
|
|
|
|
0.0, Y[0], 0.0, Y[1], Y[2] + Y[2],
|
|
|
|
0.0, 0.0, 0.0, 0.0, 2.0,
|
|
|
|
0.0, 0.0, 2.0, 0., 2.0 };
|
|
|
|
for (int i=0 ; i<50 ; ++i)
|
|
|
|
VGL[i] = tmp[i];
|
|
|
|
} else {
|
2022-03-21 18:32:39 +01:00
|
|
|
double* restrict const vgl1 = VGL;
|
|
|
|
double* restrict const vgl2 = VGL + ldv;
|
|
|
|
double* restrict const vgl3 = VGL + (ldv << 1);
|
|
|
|
double* restrict const vgl4 = VGL + 3*ldv;
|
|
|
|
double* restrict const vgl5 = VGL + (ldv << 2);
|
2022-02-16 19:40:14 +01:00
|
|
|
|
2024-05-08 13:56:32 +02:00
|
|
|
vgl1[0] = 1.0 ; vgl1[1] = Y[0]+shift ; vgl1[2] = Y[1]+shift;
|
|
|
|
vgl1[3] = Y[2]+shift ; vgl1[4] = Y[0]*Y[0]+shift; vgl1[5] = Y[0]*Y[1]+shift;
|
|
|
|
vgl1[6] = Y[0]*Y[2]+shift; vgl1[7] = Y[1]*Y[1]+shift; vgl1[8] = Y[1]*Y[2]+shift;
|
|
|
|
vgl1[9] = Y[2]*Y[2]+shift;
|
2022-02-16 19:40:14 +01:00
|
|
|
|
|
|
|
vgl2[0] = 0.0 ; vgl2[1] = 1.0 ; vgl2[2] = 0.0 ;
|
|
|
|
vgl2[3] = 0.0 ; vgl2[4] = Y[0]+Y[0]; vgl2[5] = Y[1];
|
|
|
|
vgl2[6] = Y[2]; vgl2[7] = 0.0 ; vgl2[8] = 0.0 ;
|
|
|
|
vgl2[9] = 0.0 ;
|
|
|
|
|
|
|
|
vgl3[0] = 0.0; vgl3[1] = 0.0 ; vgl3[2] = 1.0 ;
|
|
|
|
vgl3[3] = 0.0; vgl3[4] = 0.0 ; vgl3[5] = Y[0];
|
|
|
|
vgl3[6] = 0.0; vgl3[7] = Y[1]+Y[1]; vgl3[8] = Y[2];
|
|
|
|
vgl3[9] = 0.0;
|
|
|
|
|
|
|
|
vgl4[0] = 0.0 ; vgl4[1] = 0.0; vgl4[2] = 0.0 ;
|
|
|
|
vgl4[3] = 1.0 ; vgl4[4] = 0.0; vgl4[5] = 0.0 ;
|
|
|
|
vgl4[6] = Y[0] ; vgl4[7] = 0.0; vgl4[8] = Y[1];
|
|
|
|
vgl4[9] = Y[2]+Y[2];
|
|
|
|
|
|
|
|
vgl5[0] = 0.0; vgl5[1] = 0.0; vgl5[2] = 0.0;
|
|
|
|
vgl5[3] = 0.0; vgl5[4] = 2.0; vgl5[5] = 0.0;
|
|
|
|
vgl5[6] = 0.0; vgl5[7] = 2.0; vgl5[8] = 0.0;
|
|
|
|
vgl5[9] = 2.0;
|
2022-02-19 19:24:18 +01:00
|
|
|
}
|
2022-02-16 19:40:14 +01:00
|
|
|
m=10;
|
2022-02-16 15:14:41 +01:00
|
|
|
break;
|
|
|
|
}
|
2022-02-19 19:24:18 +01:00
|
|
|
default:
|
2022-02-16 15:14:41 +01:00
|
|
|
{
|
|
|
|
const int32_t size_max = (lmax+1)*(lmax+2)*(lmax+3)/6;
|
|
|
|
if (ldv < size_max) return QMCKL_INVALID_ARG_9;
|
2022-02-16 19:40:14 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
double* restrict const vgl1 = VGL;
|
|
|
|
double* restrict const vgl2 = VGL + ldv;
|
|
|
|
double* restrict const vgl3 = VGL + (ldv<<1);
|
|
|
|
double* restrict const vgl4 = VGL + ldv + (ldv<<1);
|
|
|
|
double* restrict const vgl5 = VGL + (ldv<<2);
|
2022-02-19 19:24:18 +01:00
|
|
|
|
|
|
|
const double Y[3] = { X[0]-R[0],
|
|
|
|
X[1]-R[1],
|
2024-05-08 13:56:32 +02:00
|
|
|
X[2]-R[2]};
|
2022-02-16 19:40:14 +01:00
|
|
|
|
2022-02-19 19:24:18 +01:00
|
|
|
assert(size_max > lmax+3);
|
2022-02-16 15:14:41 +01:00
|
|
|
double pows[3][size_max];
|
2022-02-19 19:24:18 +01:00
|
|
|
|
|
|
|
for (int32_t i=0 ; i<3 ; ++i) {
|
2022-02-16 15:14:41 +01:00
|
|
|
pows[0][i] = 1.0;
|
|
|
|
pows[1][i] = 1.0;
|
|
|
|
pows[2][i] = 1.0;
|
|
|
|
}
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
for (int32_t i=3 ; i<=lmax+2 ; ++i) {
|
|
|
|
pows[0][i] = pows[0][i-1] * Y[0];
|
|
|
|
pows[1][i] = pows[1][i-1] * Y[1];
|
|
|
|
pows[2][i] = pows[2][i-1] * Y[2];
|
|
|
|
}
|
2022-02-19 19:24:18 +01:00
|
|
|
|
|
|
|
int32_t* l[size_max];
|
2022-02-16 15:14:41 +01:00
|
|
|
for (int32_t i=0 ; i<size_max ; ++i) {
|
2022-02-19 19:24:18 +01:00
|
|
|
l[i] = &(L[i*ldl]);
|
2022-02-16 15:14:41 +01:00
|
|
|
}
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
for (int32_t i=0 ; i<4 ; ++i) {
|
|
|
|
l[i][0] = 0;
|
|
|
|
l[i][1] = 0;
|
|
|
|
l[i][2] = 0;
|
|
|
|
}
|
|
|
|
l[1][0] = 1;
|
|
|
|
l[2][1] = 1;
|
|
|
|
l[3][2] = 1;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
for (int32_t k=0 ; k<4 ; ++k) {
|
|
|
|
vgl2[k] = 0.0;
|
|
|
|
vgl3[k] = 0.0;
|
|
|
|
vgl4[k] = 0.0;
|
|
|
|
vgl5[k] = 0.0;
|
|
|
|
}
|
|
|
|
vgl1[0] = 1.0;
|
2024-05-08 13:56:32 +02:00
|
|
|
vgl1[1] = Y[0]+shift;
|
|
|
|
vgl1[2] = Y[1]+shift;
|
|
|
|
vgl1[3] = Y[2]+shift;
|
2022-02-16 15:14:41 +01:00
|
|
|
|
|
|
|
vgl2[1] = 1.0;
|
|
|
|
vgl3[2] = 1.0;
|
|
|
|
vgl4[3] = 1.0;
|
|
|
|
m=4;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
double dd = 2.0;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
for (int32_t d=2 ; d<= lmax ; ++d) {
|
|
|
|
double da = dd;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
for (int32_t a=d ; a>=0 ; --a) {
|
|
|
|
double db = dd-da;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
for (int32_t b=d-a ; b>=0 ; --b) {
|
2022-02-19 19:24:18 +01:00
|
|
|
const int32_t c = d - a - b;
|
2022-02-16 15:14:41 +01:00
|
|
|
const double dc = dd - da - db;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
double xy = pows[0][a+2] * pows[1][b+2];
|
|
|
|
double yz = pows[1][b+2] * pows[2][c+2];
|
|
|
|
double xz = pows[0][a+2] * pows[2][c+2];
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
l[m][0] = a;
|
|
|
|
l[m][1] = b;
|
|
|
|
l[m][2] = c;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2024-05-08 13:56:32 +02:00
|
|
|
vgl1[m] = xy * pows[2][c+2]+shift;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
xy *= dc;
|
|
|
|
xz *= db;
|
|
|
|
yz *= da;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
vgl2[m] = pows[0][a+1] * yz;
|
|
|
|
vgl3[m] = pows[1][b+1] * xz;
|
|
|
|
vgl4[m] = pows[2][c+1] * xy;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
vgl5[m] = (da-1.) * pows[0][a] * yz +
|
|
|
|
(db-1.) * pows[1][b] * xz +
|
|
|
|
(dc-1.) * pows[2][c] * xy;
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 15:14:41 +01:00
|
|
|
db -= 1.0;
|
|
|
|
++m;
|
|
|
|
}
|
|
|
|
da -= 1.0;
|
|
|
|
}
|
|
|
|
dd += 1.0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
,*n = m;
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-10-13 11:48:40 +02:00
|
|
|
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_ao_polynomial_transp_vgl_hpc (const qmckl_context context,
|
|
|
|
const double* restrict X,
|
|
|
|
const double* restrict R,
|
|
|
|
const int32_t lmax,
|
|
|
|
int64_t* n,
|
|
|
|
int32_t* restrict const L,
|
|
|
|
const int64_t ldl,
|
|
|
|
double* restrict const VGL,
|
|
|
|
const int64_t ldv )
|
|
|
|
{
|
|
|
|
return qmckl_ao_polynomial_transp_vgl_hpc_inline (context,
|
|
|
|
X, R, lmax, n, L, ldl, VGL, ldv );
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
#+end_src
|
|
|
|
|
2022-02-16 01:12:42 +01:00
|
|
|
|
|
|
|
#+begin_src f90 :tangle (eval fh_func) :comments org :exports none
|
|
|
|
interface
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_exit_code) function qmckl_ao_polynomial_transp_vgl &
|
2022-02-16 01:12:42 +01:00
|
|
|
(context, X, R, lmax, n, L, ldl, VGL, ldv) &
|
|
|
|
bind(C)
|
2024-02-23 12:06:13 +01:00
|
|
|
use qmckl_constants
|
2022-02-16 01:12:42 +01:00
|
|
|
import
|
|
|
|
implicit none
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
2022-02-16 01:12:42 +01:00
|
|
|
real (c_double ) , intent(in) :: X(3)
|
|
|
|
real (c_double ) , intent(in) :: R(3)
|
|
|
|
integer (c_int32_t) , intent(in) , value :: lmax
|
|
|
|
integer (c_int64_t) , intent(inout) :: n
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ldl
|
2022-02-11 16:07:25 +01:00
|
|
|
integer (c_int64_t) , intent(in) , value :: ldv
|
2024-02-23 11:56:28 +01:00
|
|
|
integer (c_int32_t) , intent(out) :: L(ldl,n)
|
|
|
|
real (c_double ) , intent(out) :: VGL(ldv,5)
|
2022-02-11 16:07:25 +01:00
|
|
|
|
|
|
|
end function qmckl_ao_polynomial_transp_vgl
|
2022-02-16 01:12:42 +01:00
|
|
|
end interface
|
2022-02-11 16:07:25 +01:00
|
|
|
#+end_src
|
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
*** Test :noexport:
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-04-16 00:57:08 +02:00
|
|
|
#+begin_src f90 :tangle (eval f_test)
|
2023-11-29 02:10:20 +01:00
|
|
|
function test_qmckl_ao_polynomial_vgl(context) bind(C)
|
2021-04-01 01:19:33 +02:00
|
|
|
use qmckl
|
|
|
|
implicit none
|
|
|
|
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_context), intent(in), value :: context
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
integer :: lmax, d, i
|
|
|
|
integer, allocatable :: L(:,:)
|
|
|
|
integer*8 :: n, ldl, ldv, j
|
|
|
|
double precision :: X(3), R(3), Y(3)
|
|
|
|
double precision, allocatable :: VGL(:,:)
|
|
|
|
double precision :: w
|
|
|
|
double precision :: epsilon
|
2023-09-22 16:41:43 +02:00
|
|
|
integer(qmckl_exit_code) :: test_qmckl_ao_polynomial_vgl
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2024-01-11 14:33:40 +01:00
|
|
|
epsilon = qmckl_get_numprec_precision(context)
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
X = (/ 1.1 , 2.2 , 3.3 /)
|
|
|
|
R = (/ 0.1 , 1.2 , -2.3 /)
|
|
|
|
Y(:) = X(:) - R(:)
|
|
|
|
|
|
|
|
lmax = 4;
|
|
|
|
ldl = 3;
|
|
|
|
ldv = 100;
|
|
|
|
|
|
|
|
d = (lmax+1)*(lmax+2)*(lmax+3)/6
|
|
|
|
|
|
|
|
allocate (L(ldl,d), VGL(ldv,d))
|
|
|
|
|
|
|
|
test_qmckl_ao_polynomial_vgl = &
|
|
|
|
qmckl_ao_polynomial_vgl(context, X, R, lmax, n, L, ldl, VGL, ldv)
|
|
|
|
|
|
|
|
if (test_qmckl_ao_polynomial_vgl /= QMCKL_SUCCESS) return
|
|
|
|
if (n /= d) return
|
|
|
|
|
|
|
|
do j=1,n
|
|
|
|
test_qmckl_ao_polynomial_vgl = QMCKL_FAILURE
|
|
|
|
do i=1,3
|
|
|
|
if (L(i,j) < 0) return
|
|
|
|
end do
|
|
|
|
test_qmckl_ao_polynomial_vgl = QMCKL_FAILURE
|
|
|
|
if (dabs(1.d0 - VGL(1,j) / (&
|
|
|
|
Y(1)**L(1,j) * Y(2)**L(2,j) * Y(3)**L(3,j) &
|
|
|
|
)) > epsilon ) return
|
|
|
|
|
|
|
|
test_qmckl_ao_polynomial_vgl = QMCKL_FAILURE
|
|
|
|
if (L(1,j) < 1) then
|
|
|
|
if (VGL(2,j) /= 0.d0) return
|
|
|
|
else
|
|
|
|
if (dabs(1.d0 - VGL(2,j) / (&
|
|
|
|
L(1,j) * Y(1)**(L(1,j)-1) * Y(2)**L(2,j) * Y(3)**L(3,j) &
|
|
|
|
)) > epsilon ) return
|
|
|
|
end if
|
|
|
|
|
|
|
|
test_qmckl_ao_polynomial_vgl = QMCKL_FAILURE
|
|
|
|
if (L(2,j) < 1) then
|
|
|
|
if (VGL(3,j) /= 0.d0) return
|
|
|
|
else
|
|
|
|
if (dabs(1.d0 - VGL(3,j) / (&
|
|
|
|
L(2,j) * Y(1)**L(1,j) * Y(2)**(L(2,j)-1) * Y(3)**L(3,j) &
|
|
|
|
)) > epsilon ) return
|
|
|
|
end if
|
|
|
|
|
|
|
|
test_qmckl_ao_polynomial_vgl = QMCKL_FAILURE
|
|
|
|
if (L(3,j) < 1) then
|
|
|
|
if (VGL(4,j) /= 0.d0) return
|
|
|
|
else
|
|
|
|
if (dabs(1.d0 - VGL(4,j) / (&
|
|
|
|
L(3,j) * Y(1)**L(1,j) * Y(2)**L(2,j) * Y(3)**(L(3,j)-1) &
|
|
|
|
)) > epsilon ) return
|
|
|
|
end if
|
|
|
|
|
|
|
|
test_qmckl_ao_polynomial_vgl = QMCKL_FAILURE
|
|
|
|
w = 0.d0
|
|
|
|
if (L(1,j) > 1) then
|
2021-04-30 01:26:19 +02:00
|
|
|
w = w + L(1,j) * (L(1,j)-1) * Y(1)**(L(1,j)-2) * Y(2)**L(2,j) * Y(3)**L(3,j)
|
2021-04-01 01:19:33 +02:00
|
|
|
end if
|
|
|
|
if (L(2,j) > 1) then
|
2021-04-30 01:26:19 +02:00
|
|
|
w = w + L(2,j) * (L(2,j)-1) * Y(1)**L(1,j) * Y(2)**(L(2,j)-2) * Y(3)**L(3,j)
|
2021-04-01 01:19:33 +02:00
|
|
|
end if
|
|
|
|
if (L(3,j) > 1) then
|
2021-04-30 01:26:19 +02:00
|
|
|
w = w + L(3,j) * (L(3,j)-1) * Y(1)**L(1,j) * Y(2)**L(2,j) * Y(3)**(L(3,j)-2)
|
2021-04-01 01:19:33 +02:00
|
|
|
end if
|
2023-08-23 14:24:33 +02:00
|
|
|
if (w /= 0.d0) then
|
|
|
|
if (dabs(1.d0 - VGL(5,j) / w) > epsilon ) return
|
|
|
|
endif
|
2021-04-01 01:19:33 +02:00
|
|
|
end do
|
|
|
|
|
|
|
|
test_qmckl_ao_polynomial_vgl = QMCKL_SUCCESS
|
|
|
|
|
|
|
|
deallocate(L,VGL)
|
|
|
|
end function test_qmckl_ao_polynomial_vgl
|
2021-04-16 00:57:08 +02:00
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
2021-04-16 00:57:08 +02:00
|
|
|
#+begin_src c :tangle (eval c_test)
|
2022-02-16 15:14:41 +01:00
|
|
|
int test_qmckl_ao_polynomial_vgl(qmckl_context context);
|
|
|
|
assert(0 == test_qmckl_ao_polynomial_vgl(context));
|
|
|
|
|
|
|
|
double X[3] = { 1.1, 2.2, 3.3 };
|
|
|
|
double R[3] = { 0.2, 1.1, 3.0 };
|
2022-02-19 19:24:18 +01:00
|
|
|
int32_t ldv[8] = {1, 4, 10, 20, 35, 56, 84, 120};
|
|
|
|
for (int32_t ldl=3 ; ldl<=5 ; ++ldl) {
|
2022-02-16 19:40:14 +01:00
|
|
|
int64_t n;
|
2022-02-19 19:24:18 +01:00
|
|
|
int32_t L0[200][ldl];
|
|
|
|
int32_t L1[200][ldl];
|
2022-02-16 19:40:14 +01:00
|
|
|
printf("ldl=%d\n", ldl);
|
2022-02-19 19:24:18 +01:00
|
|
|
for (int32_t lmax=0 ; lmax<=7 ; lmax++) {
|
2022-02-16 19:40:14 +01:00
|
|
|
double VGL0[5][ldv[lmax]];
|
|
|
|
double VGL1[5][ldv[lmax]];
|
|
|
|
memset(&L0[0][0], 0, sizeof(L0));
|
|
|
|
memset(&L1[0][0], 0, sizeof(L1));
|
|
|
|
memset(&VGL0[0][0], 0, sizeof(VGL0));
|
|
|
|
memset(&VGL1[0][0], 0, sizeof(VGL1));
|
|
|
|
rc = qmckl_ao_polynomial_transp_vgl_doc (context, X, R, lmax, &n, &(L0[0][0]), ldl, &(VGL0[0][0]), ldv[lmax]);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
rc = qmckl_ao_polynomial_transp_vgl_hpc (context, X, R, lmax, &n, &(L1[0][0]), ldl, &(VGL1[0][0]), ldv[lmax]);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
printf("lmax=%d\n", lmax);
|
|
|
|
for (int32_t l=0 ; l<n ; ++l) {
|
|
|
|
for (int32_t k=0 ; k<3 ; ++k) {
|
|
|
|
printf("L[%d][%d] = %d %d\n", l, k, L0[l][k], L1[l][k]);
|
|
|
|
assert( L0[l][k] == L1[l][k] );
|
|
|
|
}
|
|
|
|
}
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-02-16 19:40:14 +01:00
|
|
|
for (int32_t k=0 ; k<5 ; ++k) {
|
|
|
|
for (int32_t l=0 ; l<n ; ++l) {
|
2022-02-25 13:57:13 +01:00
|
|
|
printf("VGL[%d][%d] = %e %e %e\n", k, l, VGL0[k][l], VGL1[k][l], VGL0[k][l]-VGL1[k][l]);
|
|
|
|
assert( fabs(1.-(VGL0[k][l]+1.e-100)/(VGL1[k][l]+1.e-100)) < 1.e-15 );
|
2022-02-16 19:40:14 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2022-02-16 15:14:41 +01:00
|
|
|
}
|
2021-04-16 00:57:08 +02:00
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-18 15:10:55 +02:00
|
|
|
* Combining radial and polynomial parts
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
** Determination of nucleus ranges
|
|
|
|
|
|
|
|
The range of a nucleus is defined as the distance beyond which all
|
|
|
|
the AOs are zero, up to a given precision. For all possible numbers
|
|
|
|
of bits of precision (1-53), a range is given.
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval h_private_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_compute_nucleus_range_gaussian(
|
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t ao_num,
|
|
|
|
const int64_t shell_num,
|
|
|
|
const int64_t prim_num,
|
|
|
|
const int64_t nucl_num,
|
|
|
|
const double* nucl_coord,
|
|
|
|
const int64_t* nucleus_index,
|
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const int32_t* nucleus_max_ang_mom,
|
|
|
|
const int64_t* shell_prim_index,
|
|
|
|
const int64_t* shell_prim_num,
|
|
|
|
const int32_t* shell_ang_mom,
|
|
|
|
const double* ao_factor,
|
|
|
|
const double* ao_expo,
|
|
|
|
const double* coefficient_normalized,
|
|
|
|
double* const nucleus_range);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src f90 :comments org :tangle (eval f) :noweb yes
|
|
|
|
function qmckl_compute_nucleus_range_gaussian(context, &
|
|
|
|
ao_num, shell_num, prim_num, nucl_num, &
|
|
|
|
nucl_coord, nucleus_index, nucleus_shell_num, &
|
|
|
|
nucleus_max_ang_mom, shell_prim_index, shell_prim_num, shell_ang_mom, &
|
|
|
|
ao_factor, expo, coef_normalized, nucleus_range) &
|
|
|
|
bind(C) result(info)
|
|
|
|
use qmckl_constants
|
2024-01-11 14:33:40 +01:00
|
|
|
use qmckl, only: qmckl_ao_polynomial_vgl, qmckl_get_numprec_precision
|
2023-11-30 12:50:06 +01:00
|
|
|
implicit none
|
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ao_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: shell_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: prim_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: nucl_num
|
|
|
|
real (c_double ) , intent(in) :: nucl_coord(nucl_num,3)
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_index(nucl_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_shell_num(nucl_num)
|
|
|
|
integer (c_int32_t) , intent(in) :: nucleus_max_ang_mom(nucl_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: shell_prim_index(shell_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: shell_prim_num(shell_num)
|
|
|
|
integer (c_int32_t) , intent(in) :: shell_ang_mom(shell_num)
|
|
|
|
real (c_double ) , intent(in) :: ao_factor(ao_num)
|
|
|
|
real (c_double ) , intent(in) :: expo(prim_num)
|
|
|
|
real (c_double ) , intent(in) :: coef_normalized(prim_num)
|
|
|
|
real (c_double ) , intent(out) :: nucleus_range(nucl_num,53)
|
|
|
|
|
|
|
|
integer(qmckl_exit_code) :: info
|
|
|
|
|
|
|
|
double precision :: e_coord(3), n_coord(3)
|
|
|
|
integer*8 :: n_poly, iprim, iprim_start, iprim_end
|
|
|
|
integer :: l, il, k, iprecision
|
|
|
|
integer*8 :: ipoint, inucl, ishell
|
|
|
|
integer*8 :: ishell_start, ishell_end
|
|
|
|
integer :: lstart(0:20)
|
|
|
|
double precision :: shell_vgl(3)
|
|
|
|
double precision :: x, y, z, r2, ar2, two_a
|
|
|
|
double precision :: vmax, cutoff, v
|
|
|
|
|
|
|
|
double precision, allocatable :: poly_vgl(:,:), ao_vgl(:,:)
|
|
|
|
integer , allocatable :: powers(:,:), ao_index(:)
|
|
|
|
|
|
|
|
allocate(poly_vgl(5,ao_num), powers(3,ao_num), ao_index(ao_num))
|
|
|
|
|
|
|
|
! Pre-computed data
|
|
|
|
do l=0,20
|
|
|
|
lstart(l) = l*(l+1)*(l+2)/6 +1
|
|
|
|
end do
|
|
|
|
|
|
|
|
k=1
|
|
|
|
do inucl=1,nucl_num
|
|
|
|
ishell_start = nucleus_index(inucl) + 1
|
|
|
|
ishell_end = nucleus_index(inucl) + nucleus_shell_num(inucl)
|
|
|
|
do ishell = ishell_start, ishell_end
|
|
|
|
l = shell_ang_mom(ishell)
|
|
|
|
ao_index(ishell) = k
|
|
|
|
k = k + lstart(l+1) - lstart(l)
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
info = QMCKL_SUCCESS
|
|
|
|
|
|
|
|
n_coord(1) = 0.d0
|
|
|
|
n_coord(2) = 0.d0
|
|
|
|
n_coord(3) = 0.d0
|
|
|
|
e_coord(2) = 0.d0
|
|
|
|
e_coord(3) = 0.d0
|
|
|
|
|
|
|
|
nucleus_range = 50.d0
|
|
|
|
|
|
|
|
do inucl=1,nucl_num
|
|
|
|
|
|
|
|
x = 50.d0
|
|
|
|
do iprecision = 53,2,-1
|
|
|
|
|
|
|
|
cutoff = 1.d0 / ( 2.d0** (iprecision-2) )
|
|
|
|
|
|
|
|
vmax = 0.d0
|
|
|
|
do while ( (vmax < cutoff) .and. (x > 0.d0) )
|
|
|
|
x = x - .1d0
|
|
|
|
vmax = 0.d0
|
|
|
|
e_coord(1) = x
|
|
|
|
r2 = x*x
|
|
|
|
|
|
|
|
! Compute polynomials
|
|
|
|
info = qmckl_ao_polynomial_vgl(context, e_coord, n_coord, &
|
|
|
|
nucleus_max_ang_mom(inucl), n_poly, powers, 3_8, &
|
|
|
|
poly_vgl, 5_8)
|
|
|
|
|
|
|
|
! Loop over shells
|
|
|
|
ishell_start = nucleus_index(inucl) + 1
|
|
|
|
ishell_end = nucleus_index(inucl) + nucleus_shell_num(inucl)
|
|
|
|
do ishell = ishell_start, ishell_end
|
|
|
|
shell_vgl(1) = 0.d0
|
|
|
|
shell_vgl(2) = 0.d0
|
|
|
|
shell_vgl(3) = 0.d0
|
|
|
|
|
|
|
|
iprim_start = shell_prim_index(ishell) + 1
|
|
|
|
iprim_end = shell_prim_index(ishell) + shell_prim_num(ishell)
|
|
|
|
|
|
|
|
do iprim = iprim_start, iprim_end
|
|
|
|
|
|
|
|
ar2 = expo(iprim)*r2
|
|
|
|
|
|
|
|
v = coef_normalized(iprim) * dexp(-ar2)
|
|
|
|
two_a = -2.d0 * expo(iprim) * v
|
|
|
|
|
|
|
|
shell_vgl(1) = shell_vgl(1) + v
|
|
|
|
shell_vgl(2) = shell_vgl(2) + two_a * x
|
|
|
|
shell_vgl(3) = shell_vgl(3) + two_a * (3.d0 - 2.d0*ar2)
|
|
|
|
|
|
|
|
end do
|
|
|
|
|
|
|
|
k = ao_index(ishell)
|
|
|
|
l = shell_ang_mom(ishell)
|
|
|
|
do il = lstart(l), lstart(l+1)-1
|
|
|
|
vmax = max(vmax, poly_vgl(1,il) * shell_vgl(1) * ao_factor(k))
|
|
|
|
vmax = max(vmax, ( poly_vgl(2,il) * shell_vgl(1) + poly_vgl(1,il) * shell_vgl(2) ) * ao_factor(k))
|
|
|
|
vmax = max(vmax, ( poly_vgl(5,il) * shell_vgl(3) + poly_vgl(1,il) * shell_vgl(3) + &
|
|
|
|
2.d0 * (poly_vgl(2,il) * shell_vgl(2) ) ) * ao_factor(k) )
|
|
|
|
|
|
|
|
k = k+1
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
nucleus_range(inucl,iprecision) = x
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
deallocate(poly_vgl, powers)
|
|
|
|
end function qmckl_compute_nucleus_range_gaussian
|
|
|
|
#+end_src
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
** Values only
|
2021-07-11 23:20:40 +02:00
|
|
|
:PROPERTIES:
|
2022-05-10 19:18:19 +02:00
|
|
|
:Name: qmckl_compute_ao_value
|
2021-07-11 23:20:40 +02:00
|
|
|
:CRetType: qmckl_exit_code
|
|
|
|
:FRetType: qmckl_exit_code
|
|
|
|
:END:
|
2022-05-10 19:18:19 +02:00
|
|
|
*** Unoptimized version
|
|
|
|
#+NAME: qmckl_ao_value_args_doc
|
2023-11-28 17:00:39 +01:00
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|-----------------------+-----------------------------------+--------+----------------------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~ao_num~ | ~int64_t~ | in | Number of AOs |
|
|
|
|
| ~shell_num~ | ~int64_t~ | in | Number of shells |
|
|
|
|
| ~point_num~ | ~int64_t~ | in | Number of points |
|
|
|
|
| ~nucl_num~ | ~int64_t~ | in | Number of nuclei |
|
|
|
|
| ~coord~ | ~double[3][point_num]~ | in | Coordinates |
|
|
|
|
| ~nucl_coord~ | ~double[3][nucl_num]~ | in | Nuclear coordinates |
|
|
|
|
| ~nucleus_index~ | ~int64_t[nucl_num]~ | in | Index of the 1st shell of each nucleus |
|
|
|
|
| ~nucleus_shell_num~ | ~int64_t[nucl_num]~ | in | Number of shells per nucleus |
|
|
|
|
| ~nucleus_range~ | ~double[nucl_num]~ | in | Range beyond which all is zero |
|
|
|
|
| ~nucleus_max_ang_mom~ | ~int32_t[nucl_num]~ | in | Maximum angular momentum per nucleus |
|
|
|
|
| ~shell_ang_mom~ | ~int32_t[shell_num]~ | in | Angular momentum of each shell |
|
|
|
|
| ~ao_factor~ | ~double[ao_num]~ | in | Normalization factor of the AOs |
|
|
|
|
| ~shell_vgl~ | ~double[point_num][5][shell_num]~ | in | Value, gradients and Laplacian of the shells |
|
|
|
|
| ~ao_value~ | ~double[point_num][ao_num]~ | out | Values of the AOs |
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
#+begin_src f90 :comments org :tangle (eval f) :noweb yes
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_compute_ao_value_doc(context, &
|
2022-01-26 17:06:51 +01:00
|
|
|
ao_num, shell_num, point_num, nucl_num, &
|
|
|
|
coord, nucl_coord, nucleus_index, nucleus_shell_num, &
|
2021-07-11 23:20:40 +02:00
|
|
|
nucleus_range, nucleus_max_ang_mom, shell_ang_mom, &
|
2022-05-10 19:18:19 +02:00
|
|
|
ao_factor, shell_vgl, ao_value) &
|
2023-09-22 16:41:43 +02:00
|
|
|
bind(C) result(info)
|
|
|
|
use qmckl_constants
|
2024-01-11 14:33:40 +01:00
|
|
|
use qmckl, only: qmckl_ao_polynomial_vgl, qmckl_get_numprec_precision
|
2021-07-11 23:20:40 +02:00
|
|
|
implicit none
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ao_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: shell_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: point_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: nucl_num
|
|
|
|
real (c_double ) , intent(in) :: coord(point_num,3)
|
|
|
|
real (c_double ) , intent(in) :: nucl_coord(nucl_num,3)
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_index(nucl_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_shell_num(nucl_num)
|
|
|
|
real (c_double ) , intent(in) :: nucleus_range(nucl_num)
|
|
|
|
integer (c_int32_t) , intent(in) :: nucleus_max_ang_mom(nucl_num)
|
|
|
|
integer (c_int32_t) , intent(in) :: shell_ang_mom(shell_num)
|
|
|
|
real (c_double ) , intent(in) :: ao_factor(ao_num)
|
|
|
|
real (c_double ) , intent(in) :: shell_vgl(shell_num,5,point_num)
|
|
|
|
real (c_double ) , intent(out) :: ao_value(ao_num,point_num)
|
|
|
|
|
|
|
|
integer(qmckl_exit_code) :: info
|
2021-07-11 23:20:40 +02:00
|
|
|
|
|
|
|
double precision :: e_coord(3), n_coord(3)
|
|
|
|
integer*8 :: n_poly
|
2022-02-11 16:07:25 +01:00
|
|
|
integer :: l, il, k
|
2022-01-26 17:06:51 +01:00
|
|
|
integer*8 :: ipoint, inucl, ishell
|
2022-01-05 19:22:16 +01:00
|
|
|
integer*8 :: ishell_start, ishell_end
|
2021-07-11 23:20:40 +02:00
|
|
|
integer :: lstart(0:20)
|
|
|
|
double precision :: x, y, z, r2
|
|
|
|
double precision :: cutoff
|
|
|
|
|
|
|
|
double precision, allocatable :: poly_vgl(:,:)
|
2022-02-14 19:11:37 +01:00
|
|
|
integer , allocatable :: powers(:,:), ao_index(:)
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-02-14 19:11:37 +01:00
|
|
|
allocate(poly_vgl(5,ao_num), powers(3,ao_num), ao_index(ao_num))
|
2021-07-11 23:20:40 +02:00
|
|
|
|
|
|
|
! Pre-computed data
|
2022-02-11 16:07:25 +01:00
|
|
|
do l=0,20
|
|
|
|
lstart(l) = l*(l+1)*(l+2)/6 +1
|
2021-07-11 23:20:40 +02:00
|
|
|
end do
|
|
|
|
|
2022-02-14 19:11:37 +01:00
|
|
|
k=1
|
|
|
|
do inucl=1,nucl_num
|
|
|
|
ishell_start = nucleus_index(inucl) + 1
|
|
|
|
ishell_end = nucleus_index(inucl) + nucleus_shell_num(inucl)
|
|
|
|
do ishell = ishell_start, ishell_end
|
|
|
|
l = shell_ang_mom(ishell)
|
|
|
|
ao_index(ishell) = k
|
|
|
|
k = k + lstart(l+1) - lstart(l)
|
|
|
|
end do
|
|
|
|
end do
|
2021-07-11 23:20:40 +02:00
|
|
|
info = QMCKL_SUCCESS
|
2021-07-23 12:27:18 +02:00
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
! Don't compute polynomials when the radial part is zero.
|
2024-01-11 14:33:40 +01:00
|
|
|
<<fortran_cutoff>>
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
do ipoint = 1, point_num
|
|
|
|
e_coord(1) = coord(ipoint,1)
|
|
|
|
e_coord(2) = coord(ipoint,2)
|
|
|
|
e_coord(3) = coord(ipoint,3)
|
2023-08-24 10:14:37 +02:00
|
|
|
ao_value(:,ipoint) = 0.d0
|
2021-09-23 11:16:54 +02:00
|
|
|
do inucl=1,nucl_num
|
2022-02-19 19:24:18 +01:00
|
|
|
n_coord(1) = nucl_coord(inucl,1)
|
|
|
|
n_coord(2) = nucl_coord(inucl,2)
|
|
|
|
n_coord(3) = nucl_coord(inucl,3)
|
2021-09-23 11:16:54 +02:00
|
|
|
|
2022-01-26 17:06:51 +01:00
|
|
|
! Test if the point is in the range of the nucleus
|
2021-09-23 11:16:54 +02:00
|
|
|
x = e_coord(1) - n_coord(1)
|
|
|
|
y = e_coord(2) - n_coord(2)
|
|
|
|
z = e_coord(3) - n_coord(3)
|
|
|
|
|
2022-02-14 19:11:37 +01:00
|
|
|
r2 = x*x + y*y + z*z
|
2021-09-23 11:16:54 +02:00
|
|
|
|
2023-11-30 12:56:06 +01:00
|
|
|
if (r2 > cutoff*nucleus_range(inucl)) then
|
2021-09-23 11:16:54 +02:00
|
|
|
cycle
|
|
|
|
end if
|
|
|
|
|
2022-02-19 19:24:18 +01:00
|
|
|
! Compute polynomials
|
2023-09-22 16:41:43 +02:00
|
|
|
info = qmckl_ao_polynomial_vgl(context, e_coord, n_coord, &
|
2022-02-11 16:07:25 +01:00
|
|
|
nucleus_max_ang_mom(inucl), n_poly, powers, 3_8, &
|
|
|
|
poly_vgl, 5_8)
|
|
|
|
|
|
|
|
! Loop over shells
|
|
|
|
ishell_start = nucleus_index(inucl) + 1
|
|
|
|
ishell_end = nucleus_index(inucl) + nucleus_shell_num(inucl)
|
|
|
|
do ishell = ishell_start, ishell_end
|
2022-02-14 19:11:37 +01:00
|
|
|
k = ao_index(ishell)
|
2022-02-11 16:07:25 +01:00
|
|
|
l = shell_ang_mom(ishell)
|
|
|
|
do il = lstart(l), lstart(l+1)-1
|
|
|
|
! Value
|
2022-05-10 19:18:19 +02:00
|
|
|
ao_value(k,ipoint) = &
|
2022-02-11 16:07:25 +01:00
|
|
|
poly_vgl(1,il) * shell_vgl(ishell,1,ipoint) * ao_factor(k)
|
|
|
|
k = k+1
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
end do
|
2022-02-09 18:06:46 +01:00
|
|
|
|
2022-02-11 16:07:25 +01:00
|
|
|
deallocate(poly_vgl, powers)
|
2023-09-22 16:41:43 +02:00
|
|
|
end function qmckl_compute_ao_value_doc
|
2022-05-10 19:18:19 +02:00
|
|
|
#+end_src
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
*** HPC version
|
|
|
|
#+NAME: qmckl_ao_value_args_hpc_gaussian
|
2023-11-28 17:00:39 +01:00
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|-----------------------+-----------------------------+--------+----------------------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~ao_num~ | ~int64_t~ | in | Number of AOs |
|
|
|
|
| ~shell_num~ | ~int64_t~ | in | Number of shells |
|
|
|
|
| ~prim_num~ | ~int64_t~ | in | Number of primitives |
|
|
|
|
| ~point_num~ | ~int64_t~ | in | Number of points |
|
|
|
|
| ~nucl_num~ | ~int64_t~ | in | Number of nuclei |
|
|
|
|
| ~coord~ | ~double[3][point_num]~ | in | Coordinates |
|
|
|
|
| ~nucl_coord~ | ~double[3][nucl_num]~ | in | Nuclear coordinates |
|
|
|
|
| ~nucleus_index~ | ~int64_t[nucl_num]~ | in | Index of the 1st shell of each nucleus |
|
|
|
|
| ~nucleus_shell_num~ | ~int64_t[nucl_num]~ | in | Number of shells per nucleus |
|
|
|
|
| ~nucleus_range~ | ~double[nucl_num]~ | in | Range beyond which all is zero |
|
|
|
|
| ~nucleus_max_ang_mom~ | ~int32_t[nucl_num]~ | in | Maximum angular momentum per nucleus |
|
|
|
|
| ~shell_ang_mom~ | ~int32_t[shell_num]~ | in | Angular momentum of each shell |
|
|
|
|
| ~shell_prim_index~ | ~int64_t[shell_num]~ | in | Index of the 1st primitive of each shell |
|
|
|
|
| ~shell_prim_num~ | ~int64_t[shell_num]~ | in | Number of primitives per shell |
|
|
|
|
| ~ao_factor~ | ~double[ao_num]~ | in | Normalization factor of the AOs |
|
|
|
|
| ~ao_expo~ | ~double[prim_num]~ | in | Value, gradients and Laplacian of the shells |
|
|
|
|
| ~coef_normalized~ | ~double[prim_num]~ | in | Value, gradients and Laplacian of the shells |
|
|
|
|
| ~ao_value~ | ~double[point_num][ao_num]~ | out | Values of the AOs |
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2022-08-14 16:09:58 +02:00
|
|
|
#+NAME:ao_value_constants
|
|
|
|
#+begin_src c :exports none
|
|
|
|
int32_t lstart[32] __attribute__((aligned(64)));
|
2022-02-19 19:24:18 +01:00
|
|
|
for (int32_t l=0 ; l<32 ; ++l) {
|
2022-02-16 00:21:37 +01:00
|
|
|
lstart[l] = l*(l+1)*(l+2)/6;
|
|
|
|
}
|
2022-02-11 16:07:25 +01:00
|
|
|
|
2022-08-14 16:09:58 +02:00
|
|
|
int64_t ao_index[shell_num+1] __attribute__((aligned(64)));
|
2022-02-16 00:21:37 +01:00
|
|
|
int64_t size_max = 0;
|
2022-03-21 18:32:39 +01:00
|
|
|
int64_t prim_max = 0;
|
|
|
|
int64_t shell_max = 0;
|
2022-02-19 19:24:18 +01:00
|
|
|
{
|
2022-03-21 18:32:39 +01:00
|
|
|
int64_t k=0;
|
|
|
|
for (int inucl=0 ; inucl < nucl_num ; ++inucl) {
|
|
|
|
prim_max = prim_num_per_nucleus[inucl] > prim_max ?
|
|
|
|
prim_num_per_nucleus[inucl] : prim_max;
|
|
|
|
shell_max = nucleus_shell_num[inucl] > shell_max ?
|
|
|
|
nucleus_shell_num[inucl] : shell_max;
|
|
|
|
const int64_t ishell_start = nucleus_index[inucl];
|
|
|
|
const int64_t ishell_end = nucleus_index[inucl] + nucleus_shell_num[inucl];
|
|
|
|
for (int64_t ishell = ishell_start ; ishell < ishell_end ; ++ishell) {
|
|
|
|
const int l = shell_ang_mom[ishell];
|
|
|
|
ao_index[ishell] = k;
|
|
|
|
k += lstart[l+1] - lstart[l];
|
|
|
|
size_max = size_max < lstart[l+1] ? lstart[l+1] : size_max;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
ao_index[shell_num] = ao_num+1;
|
2022-02-16 00:21:37 +01:00
|
|
|
}
|
2022-02-11 16:07:25 +01:00
|
|
|
|
2023-11-28 17:00:39 +01:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
|
|
|
assert (ctx != NULL);
|
|
|
|
|
2022-02-16 00:21:37 +01:00
|
|
|
/* Don't compute polynomials when the radial part is zero. */
|
2023-11-28 17:00:39 +01:00
|
|
|
const int precision = ctx->numprec.precision;
|
|
|
|
const double cutoff = log( (double) ( ((uint64_t) 1) << (precision-2) ) );
|
2023-11-30 01:17:18 +01:00
|
|
|
const bool double_precision = precision > 22;
|
2022-02-11 16:07:25 +01:00
|
|
|
|
2022-08-14 16:09:58 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
#ifdef HAVE_HPC
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_compute_ao_value_hpc_gaussian (const qmckl_context context,
|
|
|
|
const int64_t ao_num,
|
|
|
|
const int64_t shell_num,
|
|
|
|
const int32_t* restrict prim_num_per_nucleus,
|
|
|
|
const int64_t point_num,
|
|
|
|
const int64_t nucl_num,
|
|
|
|
const double* restrict coord,
|
|
|
|
const double* restrict nucl_coord,
|
|
|
|
const int64_t* restrict nucleus_index,
|
|
|
|
const int64_t* restrict nucleus_shell_num,
|
|
|
|
const double* nucleus_range,
|
|
|
|
const int32_t* restrict nucleus_max_ang_mom,
|
|
|
|
const int32_t* restrict shell_ang_mom,
|
|
|
|
const double* restrict ao_factor,
|
|
|
|
const qmckl_matrix expo_per_nucleus,
|
|
|
|
const qmckl_tensor coef_per_nucleus,
|
|
|
|
double* restrict const ao_value )
|
|
|
|
{
|
|
|
|
<<ao_value_constants>>
|
|
|
|
|
2022-02-16 00:21:37 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
2022-10-13 11:48:40 +02:00
|
|
|
#pragma omp parallel if (point_num > 16)
|
2022-02-16 00:21:37 +01:00
|
|
|
#endif
|
|
|
|
{
|
2022-02-17 15:37:57 +01:00
|
|
|
qmckl_exit_code rc;
|
2022-08-14 16:09:58 +02:00
|
|
|
double ar2[prim_max] __attribute__((aligned(64)));
|
2023-11-28 17:00:39 +01:00
|
|
|
float ar2_sp[prim_max] __attribute__((aligned(64)));
|
2022-08-14 16:09:58 +02:00
|
|
|
int32_t powers[3*size_max] __attribute__((aligned(64)));
|
|
|
|
double poly_vgl[5*size_max] __attribute__((aligned(64)));
|
2022-02-16 00:21:37 +01:00
|
|
|
|
2022-08-14 16:09:58 +02:00
|
|
|
double exp_mat[prim_max] __attribute__((aligned(64)));
|
2023-11-28 17:00:39 +01:00
|
|
|
float exp_mat_sp[prim_max] __attribute__((aligned(64)));
|
2022-08-14 16:09:58 +02:00
|
|
|
double ce_mat[shell_max] __attribute__((aligned(64)));
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2023-11-29 02:10:20 +01:00
|
|
|
int32_t (*coef_mat_sparse_idx)[shell_max][prim_max+1] __attribute__((aligned(64))) =
|
2023-09-13 13:23:16 +02:00
|
|
|
malloc(nucl_num * sizeof (*coef_mat_sparse_idx));
|
|
|
|
|
|
|
|
double (*coef_mat_sparse)[shell_max][prim_max+1] __attribute__((aligned(64))) =
|
|
|
|
malloc(nucl_num * sizeof (*coef_mat_sparse));
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int i=0 ; i<nucl_num ; ++i) {
|
|
|
|
for (int j=0 ; j<shell_max; ++j) {
|
2022-10-11 17:46:30 +02:00
|
|
|
int l=1;
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int k=0 ; k<prim_max; ++k) {
|
2022-10-11 17:46:30 +02:00
|
|
|
const double c = qmckl_ten3(coef_per_nucleus,k, j, i);
|
|
|
|
if (c != 0.) {
|
|
|
|
coef_mat_sparse_idx[i][j][l] = k;
|
|
|
|
coef_mat_sparse[i][j][l] = c;
|
|
|
|
l+=1;
|
|
|
|
}
|
2022-03-21 18:32:39 +01:00
|
|
|
}
|
2022-10-11 17:46:30 +02:00
|
|
|
coef_mat_sparse_idx[i][j][0] = l;
|
2022-03-21 18:32:39 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-02-16 00:21:37 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
2024-02-07 17:04:18 +01:00
|
|
|
#pragma omp for schedule(guided)
|
2022-02-16 00:21:37 +01:00
|
|
|
#endif
|
|
|
|
for (int64_t ipoint=0 ; ipoint < point_num ; ++ipoint) {
|
2024-02-06 22:27:17 +01:00
|
|
|
|
2024-02-07 17:06:27 +01:00
|
|
|
/*
|
2024-02-06 22:39:52 +01:00
|
|
|
double* const ptr = &ao_value[ipoint*ao_num];
|
|
|
|
for (size_t k=0 ; k<ao_num ; ++k) {
|
|
|
|
ptr[k] = 0.;
|
|
|
|
}
|
2024-05-08 13:56:32 +02:00
|
|
|
,*/
|
2024-02-06 22:39:52 +01:00
|
|
|
|
2024-02-07 17:06:27 +01:00
|
|
|
memset(&ao_value[ipoint*ao_num], 0, ao_num*sizeof(double));
|
2024-02-06 22:27:17 +01:00
|
|
|
|
2022-08-14 16:09:58 +02:00
|
|
|
const double e_coord[3] __attribute__((aligned(64))) =
|
|
|
|
{ coord[ipoint],
|
|
|
|
coord[ipoint + point_num],
|
|
|
|
coord[ipoint + 2*point_num] };
|
2023-09-11 17:05:41 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int64_t inucl=0 ; inucl < nucl_num ; ++inucl) {
|
2022-08-14 16:09:58 +02:00
|
|
|
const double n_coord[3] __attribute__((aligned(64))) =
|
|
|
|
{ nucl_coord[inucl],
|
|
|
|
nucl_coord[inucl + nucl_num],
|
|
|
|
nucl_coord[inucl + 2*nucl_num] };
|
2023-09-11 17:05:41 +02:00
|
|
|
|
2024-07-11 13:53:10 +02:00
|
|
|
/* Shift to avoid having exact zeros at the nodes of the AOs */
|
2024-05-08 13:56:32 +02:00
|
|
|
const double shift = 1.e-20;
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
const double x = e_coord[0] - n_coord[0];
|
|
|
|
const double y = e_coord[1] - n_coord[1];
|
|
|
|
const double z = e_coord[2] - n_coord[2];
|
2022-02-14 19:11:37 +01:00
|
|
|
|
2023-11-28 17:00:39 +01:00
|
|
|
const double x2 = x*x;
|
|
|
|
const double y2 = y*y;
|
|
|
|
const double z2 = z*z;
|
|
|
|
const double r2 = x2 + y2 + z2;
|
2022-02-14 19:11:37 +01:00
|
|
|
|
2024-05-08 13:56:32 +02:00
|
|
|
/* Test if the point is in the range of the nucleus */
|
2023-11-30 12:56:06 +01:00
|
|
|
if (r2 > cutoff * nucleus_range[inucl]) {
|
2022-03-21 18:32:39 +01:00
|
|
|
continue;
|
|
|
|
}
|
2022-02-14 19:11:37 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
int64_t n_poly;
|
|
|
|
switch (nucleus_max_ang_mom[inucl]) {
|
|
|
|
case 0:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case 1:
|
2022-08-14 16:09:58 +02:00
|
|
|
poly_vgl[0] = 1.;
|
2022-05-10 19:18:19 +02:00
|
|
|
poly_vgl[1] = x;
|
|
|
|
poly_vgl[2] = y;
|
|
|
|
poly_vgl[3] = z;
|
2022-03-21 18:32:39 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
case 2:
|
2022-08-14 16:09:58 +02:00
|
|
|
poly_vgl[0] = 1.;
|
2024-05-08 13:56:32 +02:00
|
|
|
poly_vgl[1] = x+shift;
|
|
|
|
poly_vgl[2] = y+shift;
|
|
|
|
poly_vgl[3] = z+shift;
|
2023-11-28 17:00:39 +01:00
|
|
|
poly_vgl[4] = x2;
|
2022-05-10 19:18:19 +02:00
|
|
|
poly_vgl[5] = x*y;
|
|
|
|
poly_vgl[6] = x*z;
|
2023-11-28 17:00:39 +01:00
|
|
|
poly_vgl[7] = y2;
|
2022-05-10 19:18:19 +02:00
|
|
|
poly_vgl[8] = y*z;
|
2023-11-28 17:00:39 +01:00
|
|
|
poly_vgl[9] = z2;
|
2022-03-21 18:32:39 +01:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
2022-10-13 11:48:40 +02:00
|
|
|
rc = qmckl_ao_polynomial_transp_vgl_hpc_inline(context, e_coord, n_coord,
|
2023-11-28 17:00:39 +01:00
|
|
|
nucleus_max_ang_mom[inucl],
|
|
|
|
&n_poly, powers, (int64_t) 3,
|
|
|
|
poly_vgl, size_max);
|
2022-03-21 18:32:39 +01:00
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
break;
|
|
|
|
}
|
2022-02-17 12:36:16 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
/* Compute all exponents */
|
2022-02-17 12:36:16 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
int64_t nidx = 0;
|
2023-11-28 17:00:39 +01:00
|
|
|
if (double_precision) {
|
|
|
|
|
|
|
|
for (int64_t iprim = 0 ; iprim < prim_num_per_nucleus[inucl] ; ++iprim) {
|
|
|
|
const double v = qmckl_mat(expo_per_nucleus, iprim, inucl) * r2;
|
|
|
|
if (v <= cutoff) {
|
2023-11-29 01:18:15 +01:00
|
|
|
ar2[iprim] = v;
|
2023-11-28 17:00:39 +01:00
|
|
|
++nidx;
|
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2023-11-28 17:00:39 +01:00
|
|
|
for (int64_t iprim = 0 ; iprim < nidx ; ++iprim) {
|
2023-11-29 01:18:15 +01:00
|
|
|
exp_mat[iprim] = exp(-ar2[iprim]);
|
2023-11-28 17:00:39 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
for (int64_t iprim = 0 ; iprim < prim_num_per_nucleus[inucl] ; ++iprim) {
|
|
|
|
const double v = qmckl_mat(expo_per_nucleus, iprim, inucl) * r2;
|
|
|
|
if (v <= cutoff) {
|
2023-11-29 01:18:15 +01:00
|
|
|
ar2_sp[iprim] = (float) v;
|
2023-11-28 17:00:39 +01:00
|
|
|
++nidx;
|
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2023-11-28 17:00:39 +01:00
|
|
|
for (int64_t iprim = 0 ; iprim < nidx ; ++iprim) {
|
2023-11-29 01:18:15 +01:00
|
|
|
exp_mat_sp[iprim] = expf(-ar2_sp[iprim]);
|
2023-11-28 17:00:39 +01:00
|
|
|
}
|
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2023-11-28 17:00:39 +01:00
|
|
|
for (int64_t iprim = 0 ; iprim < nidx ; ++iprim) {
|
|
|
|
exp_mat[iprim] = exp_mat_sp[iprim];
|
2022-03-21 18:32:39 +01:00
|
|
|
}
|
2022-02-17 12:36:16 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
}
|
2022-02-19 19:24:18 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int i=0 ; i<nucleus_shell_num[inucl] ; ++i) {
|
2022-08-14 16:09:58 +02:00
|
|
|
ce_mat[i] = 0.;
|
2022-10-13 11:31:44 +02:00
|
|
|
const int32_t* restrict idx = &(coef_mat_sparse_idx[inucl][i][0]);
|
|
|
|
const double* restrict v = &(coef_mat_sparse[inucl][i][0]);
|
2022-10-11 17:46:30 +02:00
|
|
|
for (int l=1 ; l<coef_mat_sparse_idx[inucl][i][0]; ++l) {
|
|
|
|
const int k = idx[l];
|
|
|
|
if (k >= nidx) break;
|
2024-01-30 11:31:07 +01:00
|
|
|
ce_mat[i] = ce_mat[i] + v[l] * exp_mat[k];
|
2022-02-27 23:31:52 +01:00
|
|
|
}
|
2022-03-21 18:32:39 +01:00
|
|
|
}
|
2022-02-27 23:31:52 +01:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
const int64_t ishell_start = nucleus_index[inucl];
|
|
|
|
const int64_t ishell_end = nucleus_index[inucl] + nucleus_shell_num[inucl];
|
|
|
|
|
2023-11-28 17:00:39 +01:00
|
|
|
const int64_t ipoint_ao_num = ipoint*ao_num;
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int64_t ishell = ishell_start ; ishell < ishell_end ; ++ishell) {
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
const double s1 = ce_mat[ishell-ishell_start];
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
if (s1 == 0.0) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
const int64_t k = ao_index[ishell];
|
2023-11-28 17:00:39 +01:00
|
|
|
double* restrict const ao_value_1 = ao_value + ipoint_ao_num + k;
|
2022-03-21 18:32:39 +01:00
|
|
|
|
|
|
|
const int32_t l = shell_ang_mom[ishell];
|
|
|
|
const int32_t n = lstart[l+1]-lstart[l];
|
|
|
|
|
|
|
|
double* restrict poly_vgl_1 = NULL;
|
|
|
|
if (nidx > 0) {
|
|
|
|
const double* restrict f = ao_factor + k;
|
|
|
|
const int64_t idx = lstart[l];
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
poly_vgl_1 = &(poly_vgl[idx]);
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
switch (n) {
|
2022-08-14 16:09:58 +02:00
|
|
|
case 1:
|
2022-05-10 19:18:19 +02:00
|
|
|
ao_value_1[0] = s1 * f[0];
|
2022-03-21 18:32:39 +01:00
|
|
|
break;
|
2022-08-14 16:09:58 +02:00
|
|
|
case 3:
|
2022-02-25 16:30:16 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
2022-03-21 18:32:39 +01:00
|
|
|
#pragma omp simd
|
2022-02-25 16:30:16 +01:00
|
|
|
#endif
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int il=0 ; il<3 ; ++il) {
|
2022-05-10 19:18:19 +02:00
|
|
|
ao_value_1[il] = poly_vgl_1[il] * s1 * f[il];
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case(6):
|
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
|
|
|
for (int il=0 ; il<6 ; ++il) {
|
|
|
|
ao_value_1[il] = poly_vgl_1[il] * s1 * f[il];
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
#ifdef HAVE_OPENMP
|
2022-08-14 16:09:58 +02:00
|
|
|
#pragma omp simd
|
2022-05-10 19:18:19 +02:00
|
|
|
#endif
|
|
|
|
for (int il=0 ; il<n ; ++il) {
|
|
|
|
ao_value_1[il] = poly_vgl_1[il] * s1 * f[il];
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2023-09-13 13:23:16 +02:00
|
|
|
free(coef_mat_sparse_idx);
|
|
|
|
free(coef_mat_sparse);
|
2022-05-10 19:18:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
*** Interfaces
|
|
|
|
# #+CALL: generate_c_header(table=qmckl_ao_value_args_doc,rettyp=get_value("CRetType"),fname="qmckl_compute_ao_value"))
|
|
|
|
# (Commented because the header needs to go into h_private_func)
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval h_private_func) :comments org
|
|
|
|
qmckl_exit_code qmckl_compute_ao_value_doc (
|
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t ao_num,
|
|
|
|
const int64_t shell_num,
|
|
|
|
const int64_t point_num,
|
|
|
|
const int64_t nucl_num,
|
|
|
|
const double* coord,
|
|
|
|
const double* nucl_coord,
|
|
|
|
const int64_t* nucleus_index,
|
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const double* nucleus_range,
|
|
|
|
const int32_t* nucleus_max_ang_mom,
|
|
|
|
const int32_t* shell_ang_mom,
|
|
|
|
const double* ao_factor,
|
|
|
|
const double* shell_vgl,
|
|
|
|
double* const ao_value );
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval h_private_func) :comments org
|
|
|
|
#ifdef HAVE_HPC
|
|
|
|
qmckl_exit_code qmckl_compute_ao_value_hpc_gaussian (
|
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t ao_num,
|
|
|
|
const int64_t shell_num,
|
|
|
|
const int32_t* prim_num_per_nucleus,
|
|
|
|
const int64_t point_num,
|
|
|
|
const int64_t nucl_num,
|
|
|
|
const double* coord,
|
|
|
|
const double* nucl_coord,
|
|
|
|
const int64_t* nucleus_index,
|
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const double* nucleus_range,
|
|
|
|
const int32_t* nucleus_max_ang_mom,
|
|
|
|
const int32_t* shell_ang_mom,
|
|
|
|
const double* ao_factor,
|
|
|
|
const qmckl_matrix expo_per_nucleus,
|
|
|
|
const qmckl_tensor coef_per_nucleus,
|
|
|
|
double* const ao_value );
|
|
|
|
#endif
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
**** Provide :noexport:
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+CALL: write_provider_header( group="ao_basis", data="ao_value" )
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval h_private_func) :noweb yes :export none
|
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_ao_value(qmckl_context context);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+CALL: write_provider_pre( group="ao_basis", data="ao_value", dimension="ctx->ao_basis.ao_num * ctx->point.num")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_ao_value(qmckl_context context)
|
2022-05-10 19:18:19 +02:00
|
|
|
{
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_exit_code rc = QMCKL_SUCCESS;
|
2022-07-09 13:11:20 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
2022-07-09 13:11:20 +02:00
|
|
|
"qmckl_provide_ao_basis_ao_value",
|
2022-05-10 19:18:19 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
if (!ctx->ao_basis.provided) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_NOT_PROVIDED,
|
2022-07-09 13:11:20 +02:00
|
|
|
"qmckl_provide_ao_basis_ao_value",
|
2022-05-10 19:18:19 +02:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Compute if necessary */
|
|
|
|
if (ctx->point.date > ctx->ao_basis.ao_value_date) {
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->ao_basis.ao_num * ctx->point.num * sizeof(double);
|
|
|
|
|
|
|
|
if (ctx->ao_basis.ao_value != NULL) {
|
|
|
|
qmckl_memory_info_struct mem_info_test = qmckl_memory_info_struct_zero;
|
|
|
|
rc = qmckl_get_malloc_info(context, ctx->ao_basis.ao_value, &mem_info_test);
|
|
|
|
|
|
|
|
/* if rc != QMCKL_SUCCESS, we are maybe in an _inplace function because the
|
|
|
|
memory was not allocated with qmckl_malloc */
|
|
|
|
|
|
|
|
if ((rc == QMCKL_SUCCESS) && (mem_info_test.size != mem_info.size)) {
|
2022-07-09 13:11:20 +02:00
|
|
|
rc = qmckl_free(context, ctx->ao_basis.ao_value);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
ctx->ao_basis.ao_value = NULL;
|
|
|
|
}
|
2022-05-10 19:18:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Allocate array */
|
|
|
|
if (ctx->ao_basis.ao_value == NULL) {
|
|
|
|
|
|
|
|
double* ao_value = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (ao_value == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_ao_basis_ao_value",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
ctx->ao_basis.ao_value = ao_value;
|
|
|
|
}
|
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-05-10 19:18:19 +02:00
|
|
|
if (ctx->ao_basis.ao_vgl_date == ctx->point.date) {
|
|
|
|
|
|
|
|
// ao_vgl has been computed at this step: Just copy the data.
|
|
|
|
|
|
|
|
double * v = &(ctx->ao_basis.ao_value[0]);
|
|
|
|
double * vgl = &(ctx->ao_basis.ao_vgl[0]);
|
|
|
|
for (int i=0 ; i<ctx->point.num ; ++i) {
|
|
|
|
for (int k=0 ; k<ctx->ao_basis.ao_num ; ++k) {
|
|
|
|
v[k] = vgl[k];
|
|
|
|
}
|
|
|
|
v += ctx->ao_basis.ao_num;
|
|
|
|
vgl += ctx->ao_basis.ao_num * 5;
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
#ifdef HAVE_HPC
|
|
|
|
if (ctx->ao_basis.type == 'G') {
|
|
|
|
rc = qmckl_compute_ao_value_hpc_gaussian(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->ao_basis.prim_num_per_nucleus,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
2023-11-30 12:56:06 +01:00
|
|
|
ctx->ao_basis.nucleus_range,
|
2022-05-10 19:18:19 +02:00
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.expo_per_nucleus,
|
|
|
|
ctx->ao_basis.coef_per_nucleus,
|
|
|
|
ctx->ao_basis.ao_value);
|
|
|
|
/*
|
|
|
|
} else if (ctx->ao_basis.type == 'S') {
|
|
|
|
rc = qmck_compute_ao_value_hpc_slater(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->ao_basis.prim_num,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
|
|
|
ctx->ao_basis.nucleus_range,
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.shell_prim_index,
|
|
|
|
ctx->ao_basis.shell_prim_num,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.exponent,
|
|
|
|
ctx->ao_basis.coefficient_normalized,
|
|
|
|
ctx->ao_basis.ao_value);
|
|
|
|
,*/
|
|
|
|
} else {
|
2022-07-09 13:11:20 +02:00
|
|
|
/* Provide required data */
|
|
|
|
rc = qmckl_provide_ao_basis_shell_vgl(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc, "qmckl_provide_ao_basis_shell_vgl", NULL);
|
|
|
|
}
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
rc = qmckl_compute_ao_value_doc(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
2023-11-30 12:56:06 +01:00
|
|
|
ctx->ao_basis.nucleus_range,
|
2022-05-10 19:18:19 +02:00
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.shell_vgl,
|
|
|
|
ctx->ao_basis.ao_value);
|
|
|
|
}
|
|
|
|
#else
|
2022-07-09 13:11:20 +02:00
|
|
|
/* Provide required data */
|
|
|
|
rc = qmckl_provide_ao_basis_shell_vgl(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc, "qmckl_provide_ao_basis_shell_vgl", NULL);
|
|
|
|
}
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
rc = qmckl_compute_ao_value_doc(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
2023-11-30 12:56:06 +01:00
|
|
|
ctx->ao_basis.nucleus_range,
|
2022-05-10 19:18:19 +02:00
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.shell_vgl,
|
|
|
|
ctx->ao_basis.ao_value);
|
|
|
|
#endif
|
2022-07-09 13:11:20 +02:00
|
|
|
}
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+CALL: write_provider_post( group="ao_basis", data="ao_value" )
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2022-07-09 13:11:20 +02:00
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return rc;
|
2022-05-10 19:18:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_value_date = ctx->date;
|
|
|
|
}
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-07-09 13:11:20 +02:00
|
|
|
#+end_src
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
**** Test :noexport:
|
|
|
|
|
|
|
|
#+begin_src python :results output :exports none
|
|
|
|
import numpy as np
|
|
|
|
from math import sqrt
|
|
|
|
|
|
|
|
h0 = 1.e-4
|
|
|
|
def f(a,x,y):
|
|
|
|
return np.sum( [c * np.exp( -b*(np.linalg.norm(x-y))**2) for b,c in a] )
|
|
|
|
|
|
|
|
elec_26_w1 = np.array( [ 1.49050402641, 2.90106987953, -1.05920815468 ] )
|
|
|
|
elec_15_w2 = np.array( [ -2.20180344582,-1.9113150239, 2.2193744778600002 ] )
|
|
|
|
nucl_1 = np.array( [ -2.302574592081335e+00, -3.542027060505035e-01, -5.334129934317614e-02] )
|
|
|
|
|
|
|
|
#double ao_value[prim_num][5][elec_num];
|
|
|
|
x = elec_26_w1 ; y = nucl_1
|
|
|
|
a = [( 4.0382999999999998e+02, 1.4732000000000000e-03 * 5.9876577632594533e+04),
|
|
|
|
( 1.2117000000000000e+02, 1.2672500000000000e-02 * 7.2836806319891484e+03),
|
|
|
|
( 4.6344999999999999e+01, 5.8045100000000002e-02 * 1.3549226646722386e+03),
|
|
|
|
( 1.9721000000000000e+01, 1.7051030000000000e-01 * 3.0376315094739988e+02),
|
|
|
|
( 8.8623999999999992e+00, 3.1859579999999998e-01 * 7.4924579607137730e+01),
|
|
|
|
( 3.9962000000000000e+00, 3.8450230000000002e-01 * 1.8590543353806009e+01),
|
|
|
|
( 1.7636000000000001e+00, 2.7377370000000001e-01 * 4.4423176930919421e+00),
|
|
|
|
( 7.0618999999999998e-01, 7.4396699999999996e-02 * 8.9541051939952665e-01)]
|
|
|
|
|
|
|
|
norm = sqrt(3.)
|
|
|
|
# x^2 * g(r)
|
|
|
|
print ( "[26][0][219] : %25.15e"%(fx(a,x,y)) )
|
|
|
|
print ( "[26][1][219] : %25.15e"%(df(a,x,y,1)) )
|
|
|
|
print ( "[26][2][219] : %25.15e"%(df(a,x,y,2)) )
|
|
|
|
print ( "[26][3][219] : %25.15e"%(df(a,x,y,3)) )
|
|
|
|
print ( "[26][4][219] : %25.15e"%(lf(a,x,y)) )
|
|
|
|
|
|
|
|
print ( "[26][0][220] : %25.15e"%(norm*f(a,x,y) * (x[0] - y[0]) * (x[1] - y[1]) ))
|
|
|
|
print ( "[26][1][220] : %25.15e"%(norm*df(a,x,y,1)* (x[0] - y[0]) * (x[1] - y[1]) + norm*f(a,x,y) * (x[1] - y[1])) )
|
|
|
|
|
|
|
|
print ( "[26][0][221] : %25.15e"%(norm*f(a,x,y) * (x[0] - y[0]) * (x[2] - y[2])) )
|
|
|
|
print ( "[26][1][221] : %25.15e"%(norm*df(a,x,y,1)* (x[0] - y[0]) * (x[2] - y[2]) + norm*f(a,x,y) * (x[2] - y[2])) )
|
|
|
|
|
|
|
|
print ( "[26][0][222] : %25.15e"%(f(a,x,y) * (x[1] - y[1]) * (x[1] - y[1])) )
|
|
|
|
print ( "[26][1][222] : %25.15e"%(df(a,x,y,1)* (x[1] - y[1]) * (x[1] - y[1])) )
|
|
|
|
|
|
|
|
print ( "[26][0][223] : %25.15e"%(norm*f(a,x,y) * (x[1] - y[1]) * (x[2] - y[2])) )
|
|
|
|
print ( "[26][1][223] : %25.15e"%(norm*df(a,x,y,1)* (x[1] - y[1]) * (x[2] - y[2])) )
|
|
|
|
|
|
|
|
print ( "[26][0][224] : %25.15e"%(f(a,x,y) * (x[2] - y[2]) * (x[2] - y[2])) )
|
|
|
|
print ( "[26][1][224] : %25.15e"%(df(a,x,y,1)* (x[2] - y[2]) * (x[2] - y[2])) )
|
|
|
|
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval c_test) :exports none
|
|
|
|
{
|
|
|
|
#define shell_num chbrclf_shell_num
|
|
|
|
#define ao_num chbrclf_ao_num
|
|
|
|
|
|
|
|
double* elec_coord = &(chbrclf_elec_coord[0][0][0]);
|
|
|
|
|
|
|
|
assert(qmckl_electron_provided(context));
|
|
|
|
|
2023-09-14 11:00:24 +02:00
|
|
|
int64_t point_num = elec_num;
|
2022-08-07 14:57:10 +02:00
|
|
|
rc = qmckl_set_point(context, 'N', point_num, elec_coord, point_num*3);
|
2022-05-10 19:18:19 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
double ao_value[point_num][ao_num];
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
rc = qmckl_get_ao_basis_ao_value(context, &(ao_value[0][0]),
|
2022-08-07 14:57:10 +02:00
|
|
|
(int64_t) point_num*ao_num);
|
2022-05-10 19:18:19 +02:00
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
|
|
|
printf("\n");
|
|
|
|
printf(" ao_value ao_value[26][219] %25.15e\n", ao_value[26][219]);
|
|
|
|
printf(" ao_value ao_value[26][220] %25.15e\n", ao_value[26][220]);
|
|
|
|
printf(" ao_value ao_value[26][221] %25.15e\n", ao_value[26][221]);
|
|
|
|
printf(" ao_value ao_value[26][222] %25.15e\n", ao_value[26][222]);
|
|
|
|
printf(" ao_value ao_value[26][223] %25.15e\n", ao_value[26][223]);
|
|
|
|
printf(" ao_value ao_value[26][224] %25.15e\n", ao_value[26][224]);
|
|
|
|
printf("\n");
|
|
|
|
|
2022-10-11 17:46:30 +02:00
|
|
|
printf("%e %e\n", ao_value[26][219], 1.020298798341620e-08);
|
2022-05-10 19:18:19 +02:00
|
|
|
assert( fabs(ao_value[26][219] - ( 1.020298798341620e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_value[26][220] - ( 1.516643537739178e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_value[26][221] - ( -4.686370882518819e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_value[26][222] - ( 7.514816980753531e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_value[26][223] - ( -4.021908374204471e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_value[26][224] - ( 7.175045873560788e-10)) < 1.e-14 );
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
** Value, gradients, Laplacian
|
|
|
|
:PROPERTIES:
|
|
|
|
:Name: qmckl_compute_ao_vgl
|
|
|
|
:CRetType: qmckl_exit_code
|
|
|
|
:FRetType: qmckl_exit_code
|
|
|
|
:END:
|
2023-03-13 15:32:35 +01:00
|
|
|
*** Reference version
|
2022-05-10 19:18:19 +02:00
|
|
|
#+NAME: qmckl_ao_vgl_args_doc
|
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|-----------------------+-----------------------------------+--------+----------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~ao_num~ | ~int64_t~ | in | Number of AOs |
|
|
|
|
| ~shell_num~ | ~int64_t~ | in | Number of shells |
|
|
|
|
| ~point_num~ | ~int64_t~ | in | Number of points |
|
|
|
|
| ~nucl_num~ | ~int64_t~ | in | Number of nuclei |
|
|
|
|
| ~coord~ | ~double[3][point_num]~ | in | Coordinates |
|
|
|
|
| ~nucl_coord~ | ~double[3][nucl_num]~ | in | Nuclear coordinates |
|
|
|
|
| ~nucleus_index~ | ~int64_t[nucl_num]~ | in | Index of the 1st shell of each nucleus |
|
|
|
|
| ~nucleus_shell_num~ | ~int64_t[nucl_num]~ | in | Number of shells per nucleus |
|
|
|
|
| ~nucleus_range~ | ~double[nucl_num]~ | in | Range beyond which all is zero |
|
|
|
|
| ~nucleus_max_ang_mom~ | ~int32_t[nucl_num]~ | in | Maximum angular momentum per nucleus |
|
|
|
|
| ~shell_ang_mom~ | ~int32_t[shell_num]~ | in | Angular momentum of each shell |
|
|
|
|
| ~ao_factor~ | ~double[ao_num]~ | in | Normalization factor of the AOs |
|
|
|
|
| ~shell_vgl~ | ~double[point_num][5][shell_num]~ | in | Value, gradients and Laplacian of the shells |
|
|
|
|
| ~ao_vgl~ | ~double[point_num][5][ao_num]~ | out | Value, gradients and Laplacian of the AOs |
|
|
|
|
|
|
|
|
#+begin_src f90 :comments org :tangle (eval f) :noweb yes
|
2023-09-22 16:41:43 +02:00
|
|
|
function qmckl_compute_ao_vgl_doc(context, &
|
2022-05-10 19:18:19 +02:00
|
|
|
ao_num, shell_num, point_num, nucl_num, &
|
|
|
|
coord, nucl_coord, nucleus_index, nucleus_shell_num, &
|
|
|
|
nucleus_range, nucleus_max_ang_mom, shell_ang_mom, &
|
|
|
|
ao_factor, shell_vgl, ao_vgl) &
|
2023-09-22 16:41:43 +02:00
|
|
|
bind(C) result(info)
|
|
|
|
use qmckl_constants
|
2024-01-11 14:33:40 +01:00
|
|
|
use qmckl, only : qmckl_ao_polynomial_vgl, qmckl_get_numprec_precision
|
2022-05-10 19:18:19 +02:00
|
|
|
implicit none
|
2023-09-22 16:41:43 +02:00
|
|
|
integer (qmckl_context), intent(in) , value :: context
|
|
|
|
integer (c_int64_t) , intent(in) , value :: ao_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: shell_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: point_num
|
|
|
|
integer (c_int64_t) , intent(in) , value :: nucl_num
|
|
|
|
real (c_double ) , intent(in) :: coord(point_num,3)
|
|
|
|
real (c_double ) , intent(in) :: nucl_coord(nucl_num,3)
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_index(nucl_num)
|
|
|
|
integer (c_int64_t) , intent(in) :: nucleus_shell_num(nucl_num)
|
|
|
|
real (c_double ) , intent(in) :: nucleus_range(nucl_num)
|
|
|
|
integer (c_int32_t) , intent(in) :: nucleus_max_ang_mom(nucl_num)
|
|
|
|
integer (c_int32_t) , intent(in) :: shell_ang_mom(shell_num)
|
|
|
|
real (c_double ) , intent(in) :: ao_factor(ao_num)
|
|
|
|
real (c_double ) , intent(in) :: shell_vgl(shell_num,5,point_num)
|
|
|
|
real (c_double ) , intent(out) :: ao_vgl(ao_num,5,point_num)
|
|
|
|
integer(qmckl_exit_code) :: info
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
double precision :: e_coord(3), n_coord(3)
|
|
|
|
integer*8 :: n_poly
|
|
|
|
integer :: l, il, k
|
|
|
|
integer*8 :: ipoint, inucl, ishell
|
|
|
|
integer*8 :: ishell_start, ishell_end
|
|
|
|
integer :: lstart(0:20)
|
|
|
|
double precision :: x, y, z, r2
|
|
|
|
double precision :: cutoff
|
|
|
|
|
|
|
|
double precision, allocatable :: poly_vgl(:,:)
|
|
|
|
integer , allocatable :: powers(:,:), ao_index(:)
|
|
|
|
|
|
|
|
allocate(poly_vgl(5,ao_num), powers(3,ao_num), ao_index(ao_num))
|
|
|
|
|
|
|
|
! Pre-computed data
|
|
|
|
do l=0,20
|
|
|
|
lstart(l) = l*(l+1)*(l+2)/6 +1
|
|
|
|
end do
|
|
|
|
|
|
|
|
k=1
|
|
|
|
do inucl=1,nucl_num
|
|
|
|
ishell_start = nucleus_index(inucl) + 1
|
|
|
|
ishell_end = nucleus_index(inucl) + nucleus_shell_num(inucl)
|
|
|
|
do ishell = ishell_start, ishell_end
|
|
|
|
l = shell_ang_mom(ishell)
|
|
|
|
ao_index(ishell) = k
|
|
|
|
k = k + lstart(l+1) - lstart(l)
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
info = QMCKL_SUCCESS
|
|
|
|
|
|
|
|
! Don't compute polynomials when the radial part is zero.
|
2024-01-11 14:33:40 +01:00
|
|
|
<<fortran_cutoff>>
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
do ipoint = 1, point_num
|
|
|
|
e_coord(1) = coord(ipoint,1)
|
|
|
|
e_coord(2) = coord(ipoint,2)
|
|
|
|
e_coord(3) = coord(ipoint,3)
|
2023-08-24 09:59:57 +02:00
|
|
|
ao_vgl(:,:,ipoint) = 0.d0
|
2022-05-10 19:18:19 +02:00
|
|
|
do inucl=1,nucl_num
|
|
|
|
n_coord(1) = nucl_coord(inucl,1)
|
|
|
|
n_coord(2) = nucl_coord(inucl,2)
|
|
|
|
n_coord(3) = nucl_coord(inucl,3)
|
|
|
|
|
|
|
|
! Test if the point is in the range of the nucleus
|
|
|
|
x = e_coord(1) - n_coord(1)
|
|
|
|
y = e_coord(2) - n_coord(2)
|
|
|
|
z = e_coord(3) - n_coord(3)
|
|
|
|
|
|
|
|
r2 = x*x + y*y + z*z
|
|
|
|
|
2023-11-30 12:56:06 +01:00
|
|
|
if (r2 > cutoff*nucleus_range(inucl)) then
|
2022-05-10 19:18:19 +02:00
|
|
|
cycle
|
|
|
|
end if
|
|
|
|
|
|
|
|
! Compute polynomials
|
2023-09-22 16:41:43 +02:00
|
|
|
info = qmckl_ao_polynomial_vgl(context, e_coord, n_coord, &
|
2022-05-10 19:18:19 +02:00
|
|
|
nucleus_max_ang_mom(inucl), n_poly, powers, 3_8, &
|
|
|
|
poly_vgl, 5_8)
|
|
|
|
|
|
|
|
! Loop over shells
|
|
|
|
ishell_start = nucleus_index(inucl) + 1
|
|
|
|
ishell_end = nucleus_index(inucl) + nucleus_shell_num(inucl)
|
|
|
|
do ishell = ishell_start, ishell_end
|
|
|
|
k = ao_index(ishell)
|
|
|
|
l = shell_ang_mom(ishell)
|
|
|
|
do il = lstart(l), lstart(l+1)-1
|
|
|
|
! Value
|
|
|
|
ao_vgl(k,1,ipoint) = &
|
|
|
|
poly_vgl(1,il) * shell_vgl(ishell,1,ipoint) * ao_factor(k)
|
|
|
|
|
|
|
|
! Grad_x
|
|
|
|
ao_vgl(k,2,ipoint) = ( &
|
|
|
|
poly_vgl(2,il) * shell_vgl(ishell,1,ipoint) + &
|
|
|
|
poly_vgl(1,il) * shell_vgl(ishell,2,ipoint) &
|
|
|
|
) * ao_factor(k)
|
|
|
|
|
|
|
|
! Grad_y
|
|
|
|
ao_vgl(k,3,ipoint) = ( &
|
|
|
|
poly_vgl(3,il) * shell_vgl(ishell,1,ipoint) + &
|
|
|
|
poly_vgl(1,il) * shell_vgl(ishell,3,ipoint) &
|
|
|
|
) * ao_factor(k)
|
|
|
|
|
|
|
|
! Grad_z
|
|
|
|
ao_vgl(k,4,ipoint) = ( &
|
|
|
|
poly_vgl(4,il) * shell_vgl(ishell,1,ipoint) + &
|
|
|
|
poly_vgl(1,il) * shell_vgl(ishell,4,ipoint) &
|
|
|
|
) * ao_factor(k)
|
|
|
|
|
|
|
|
! Lapl_z
|
|
|
|
ao_vgl(k,5,ipoint) = ( &
|
|
|
|
poly_vgl(5,il) * shell_vgl(ishell,1,ipoint) + &
|
|
|
|
poly_vgl(1,il) * shell_vgl(ishell,5,ipoint) + &
|
|
|
|
2.d0 * ( &
|
|
|
|
poly_vgl(2,il) * shell_vgl(ishell,2,ipoint) + &
|
|
|
|
poly_vgl(3,il) * shell_vgl(ishell,3,ipoint) + &
|
|
|
|
poly_vgl(4,il) * shell_vgl(ishell,4,ipoint) ) &
|
|
|
|
) * ao_factor(k)
|
|
|
|
|
|
|
|
k = k+1
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
end do
|
|
|
|
|
|
|
|
deallocate(poly_vgl, powers)
|
2023-09-22 16:41:43 +02:00
|
|
|
end function qmckl_compute_ao_vgl_doc
|
2022-05-10 19:18:19 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
*** HPC version
|
|
|
|
#+NAME: qmckl_ao_vgl_args_hpc_gaussian
|
2023-11-30 12:50:06 +01:00
|
|
|
| Variable | Type | In/Out | Description |
|
|
|
|
|-----------------------+--------------------------------+--------+-------------------------------------------|
|
|
|
|
| ~context~ | ~qmckl_context~ | in | Global state |
|
|
|
|
| ~ao_num~ | ~int64_t~ | in | Number of AOs |
|
|
|
|
| ~shell_num~ | ~int64_t~ | in | Number of shells |
|
|
|
|
| ~prim_num~ | ~int64_t~ | in | Number of primitives |
|
|
|
|
| ~point_num~ | ~int64_t~ | in | Number of points |
|
|
|
|
| ~nucl_num~ | ~int64_t~ | in | Number of nuclei |
|
|
|
|
| ~coord~ | ~double[3][point_num]~ | in | Coordinates |
|
|
|
|
| ~nucl_coord~ | ~double[3][nucl_num]~ | in | Nuclear coordinates |
|
|
|
|
| ~nucleus_index~ | ~int64_t[nucl_num]~ | in | Index of the 1st shell of each nucleus |
|
|
|
|
| ~nucleus_shell_num~ | ~int64_t[nucl_num]~ | in | Number of shells per nucleus |
|
|
|
|
| ~nucleus_range~ | ~double[nucl_num]~ | in | Range beyond which all is zero |
|
|
|
|
| ~nucleus_max_ang_mom~ | ~int32_t[nucl_num]~ | in | Maximum angular momentum per nucleus |
|
|
|
|
| ~shell_ang_mom~ | ~int32_t[shell_num]~ | in | Angular momentum of each shell |
|
|
|
|
| ~shell_prim_index~ | ~int64_t[shell_num]~ | in | Index of the 1st primitive of each shell |
|
|
|
|
| ~shell_prim_num~ | ~int64_t[shell_num]~ | in | Number of primitives per shell |
|
|
|
|
| ~ao_factor~ | ~double[ao_num]~ | in | Normalization factor of the AOs |
|
|
|
|
| ~ao_expo~ | ~double[prim_num]~ | in | Exponents of the primitives |
|
|
|
|
| ~coef_normalized~ | ~double[prim_num]~ | in | Normalized coefficients of the primitives |
|
|
|
|
| ~ao_vgl~ | ~double[point_num][5][ao_num]~ | out | Value, gradients and Laplacian of the AOs |
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
|
|
|
#ifdef HAVE_HPC
|
|
|
|
qmckl_exit_code
|
|
|
|
qmckl_compute_ao_vgl_hpc_gaussian (
|
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t ao_num,
|
|
|
|
const int64_t shell_num,
|
|
|
|
const int32_t* restrict prim_num_per_nucleus,
|
|
|
|
const int64_t point_num,
|
|
|
|
const int64_t nucl_num,
|
|
|
|
const double* restrict coord,
|
|
|
|
const double* restrict nucl_coord,
|
|
|
|
const int64_t* restrict nucleus_index,
|
|
|
|
const int64_t* restrict nucleus_shell_num,
|
|
|
|
const double* nucleus_range,
|
|
|
|
const int32_t* restrict nucleus_max_ang_mom,
|
|
|
|
const int32_t* restrict shell_ang_mom,
|
|
|
|
const double* restrict ao_factor,
|
|
|
|
const qmckl_matrix expo_per_nucleus,
|
|
|
|
const qmckl_tensor coef_per_nucleus,
|
|
|
|
double* restrict const ao_vgl )
|
|
|
|
{
|
2022-10-11 17:46:30 +02:00
|
|
|
<<ao_value_constants>>
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
#ifdef HAVE_OPENMP
|
2022-10-13 11:53:49 +02:00
|
|
|
#pragma omp parallel if (point_num > 16)
|
2022-05-10 19:18:19 +02:00
|
|
|
#endif
|
|
|
|
{
|
|
|
|
qmckl_exit_code rc;
|
2022-06-29 13:31:58 +02:00
|
|
|
double ar2[prim_max] __attribute__((aligned(64)));
|
2023-11-28 17:00:39 +01:00
|
|
|
float ar2_sp[prim_max] __attribute__((aligned(64)));
|
2022-06-29 13:31:58 +02:00
|
|
|
int32_t powers[3*size_max] __attribute__((aligned(64)));
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2022-06-29 13:31:58 +02:00
|
|
|
double exp_mat[prim_max][8] __attribute__((aligned(64))) ;
|
2023-11-28 17:00:39 +01:00
|
|
|
float exp_mat_sp[prim_max] __attribute__((aligned(64)));
|
2022-06-29 13:31:58 +02:00
|
|
|
double ce_mat[shell_max][8] __attribute__((aligned(64))) ;
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2023-11-29 02:10:20 +01:00
|
|
|
int32_t (*coef_mat_sparse_idx)[shell_max][prim_max+1] __attribute__((aligned(64))) =
|
2023-09-13 13:23:16 +02:00
|
|
|
malloc(nucl_num * sizeof (*coef_mat_sparse_idx));
|
|
|
|
|
|
|
|
double (*coef_mat_sparse)[shell_max][prim_max+1] __attribute__((aligned(64))) =
|
|
|
|
malloc(nucl_num * sizeof (*coef_mat_sparse));
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
for (int i=0 ; i<nucl_num ; ++i) {
|
|
|
|
for (int j=0 ; j<shell_max; ++j) {
|
2022-10-11 17:46:30 +02:00
|
|
|
int l=1;
|
2022-05-10 19:18:19 +02:00
|
|
|
for (int k=0 ; k<prim_max; ++k) {
|
2022-10-11 17:46:30 +02:00
|
|
|
const double c = qmckl_ten3(coef_per_nucleus,k, j, i);
|
|
|
|
if (c != 0.) {
|
|
|
|
coef_mat_sparse_idx[i][j][l] = k;
|
|
|
|
coef_mat_sparse[i][j][l] = c;
|
|
|
|
l+=1;
|
|
|
|
}
|
2022-05-10 19:18:19 +02:00
|
|
|
}
|
2022-10-11 17:46:30 +02:00
|
|
|
coef_mat_sparse_idx[i][j][0] = l;
|
2022-05-10 19:18:19 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-14 16:09:58 +02:00
|
|
|
double poly_vgl_l1[4][4] __attribute__((aligned(64))) =
|
2022-10-11 17:46:30 +02:00
|
|
|
{{1.0, 0.0, 0.0, 0.0},
|
|
|
|
{0.0, 1.0, 0.0, 0.0},
|
|
|
|
{0.0, 0.0, 1.0, 0.0},
|
|
|
|
{0.0, 0.0, 0.0, 1.0}};
|
2022-08-14 16:09:58 +02:00
|
|
|
double poly_vgl_l2[5][10]__attribute__((aligned(64))) =
|
2022-10-11 17:46:30 +02:00
|
|
|
{{1., 0., 0., 0., 0., 0., 0., 0., 0., 0.},
|
|
|
|
{0., 1., 0., 0., 0., 0., 0., 0., 0., 0.},
|
|
|
|
{0., 0., 1., 0., 0., 0., 0., 0., 0., 0.},
|
|
|
|
{0., 0., 0., 1., 0., 0., 0., 0., 0., 0.},
|
|
|
|
{0., 0., 0., 0., 2., 0., 0., 2., 0., 2.}};
|
2022-08-14 16:09:58 +02:00
|
|
|
double poly_vgl[5][size_max] __attribute__((aligned(64)));
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp for
|
|
|
|
#endif
|
|
|
|
for (int64_t ipoint=0 ; ipoint < point_num ; ++ipoint) {
|
2024-02-06 22:27:17 +01:00
|
|
|
|
2024-02-07 17:06:27 +01:00
|
|
|
/*
|
2024-02-06 22:39:52 +01:00
|
|
|
double* const ptr = &ao_vgl[ipoint*5*ao_num];
|
|
|
|
for (size_t k=0 ; k<5*ao_num ; ++k) {
|
|
|
|
ptr[k] = 0.;
|
|
|
|
}
|
2024-05-08 13:56:32 +02:00
|
|
|
,*/
|
2024-02-06 22:39:52 +01:00
|
|
|
|
2024-02-07 17:06:27 +01:00
|
|
|
memset(&ao_vgl[ipoint*ao_num*5], 0, 5*ao_num*sizeof(double));
|
2024-02-06 22:27:17 +01:00
|
|
|
|
2022-06-29 13:31:58 +02:00
|
|
|
const double e_coord[3] __attribute__((aligned(64))) =
|
2022-08-14 16:09:58 +02:00
|
|
|
{ coord[ipoint],
|
|
|
|
coord[ipoint + point_num],
|
|
|
|
coord[ipoint + 2*point_num] };
|
2023-09-11 17:05:41 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
for (int64_t inucl=0 ; inucl < nucl_num ; ++inucl) {
|
2022-06-29 13:31:58 +02:00
|
|
|
const double n_coord[3] __attribute__((aligned(64))) =
|
2022-08-14 16:09:58 +02:00
|
|
|
{ nucl_coord[inucl],
|
|
|
|
nucl_coord[inucl + nucl_num],
|
|
|
|
nucl_coord[inucl + 2*nucl_num] };
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2024-05-08 13:56:32 +02:00
|
|
|
const double shift = 1.e-20;
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
const double x = e_coord[0] - n_coord[0];
|
|
|
|
const double y = e_coord[1] - n_coord[1];
|
|
|
|
const double z = e_coord[2] - n_coord[2];
|
|
|
|
|
2023-11-28 17:00:39 +01:00
|
|
|
const double x2 = x*x;
|
|
|
|
const double y2 = y*y;
|
|
|
|
const double z2 = z*z;
|
|
|
|
const double r2 = x2 + y2 + z2;
|
2022-05-10 19:18:19 +02:00
|
|
|
|
2024-05-08 13:56:32 +02:00
|
|
|
/* Test if the point is in the range of the nucleus */
|
2023-11-30 12:56:06 +01:00
|
|
|
if (r2 > cutoff * nucleus_range[inucl]) {
|
2022-05-10 19:18:19 +02:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
int64_t n_poly;
|
|
|
|
switch (nucleus_max_ang_mom[inucl]) {
|
|
|
|
case 0:
|
|
|
|
break;
|
|
|
|
|
|
|
|
case 1:
|
2024-05-08 13:56:32 +02:00
|
|
|
poly_vgl_l1[0][1] = x+shift;
|
|
|
|
poly_vgl_l1[0][2] = y+shift;
|
|
|
|
poly_vgl_l1[0][3] = z+shift;
|
2022-05-10 19:18:19 +02:00
|
|
|
break;
|
|
|
|
|
|
|
|
case 2:
|
2024-05-08 13:56:32 +02:00
|
|
|
poly_vgl_l2[0][1] = x+shift;
|
|
|
|
poly_vgl_l2[0][2] = y+shift;
|
|
|
|
poly_vgl_l2[0][3] = z+shift;
|
|
|
|
poly_vgl_l2[0][4] = x2+shift;
|
|
|
|
poly_vgl_l2[0][5] = x*y+shift;
|
|
|
|
poly_vgl_l2[0][6] = x*z+shift;
|
|
|
|
poly_vgl_l2[0][7] = y2+shift;
|
|
|
|
poly_vgl_l2[0][8] = y*z+shift;
|
|
|
|
poly_vgl_l2[0][9] = z2+shift;
|
2022-05-10 19:18:19 +02:00
|
|
|
poly_vgl_l2[1][4] = x+x;
|
|
|
|
poly_vgl_l2[1][5] = y;
|
|
|
|
poly_vgl_l2[1][6] = z;
|
|
|
|
poly_vgl_l2[2][5] = x;
|
|
|
|
poly_vgl_l2[2][7] = y+y;
|
|
|
|
poly_vgl_l2[2][8] = z;
|
|
|
|
poly_vgl_l2[3][6] = x;
|
|
|
|
poly_vgl_l2[3][8] = y;
|
|
|
|
poly_vgl_l2[3][9] = z+z;
|
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
2022-10-13 11:48:40 +02:00
|
|
|
rc = qmckl_ao_polynomial_transp_vgl_hpc_inline(context, e_coord, n_coord,
|
2023-11-28 17:00:39 +01:00
|
|
|
nucleus_max_ang_mom[inucl],
|
|
|
|
&n_poly, powers, (int64_t) 3,
|
|
|
|
&(poly_vgl[0][0]), size_max);
|
2022-05-10 19:18:19 +02:00
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Compute all exponents */
|
|
|
|
|
|
|
|
int64_t nidx = 0;
|
2023-11-28 17:00:39 +01:00
|
|
|
|
|
|
|
if (double_precision) {
|
|
|
|
|
|
|
|
for (int64_t iprim = 0 ; iprim < prim_num_per_nucleus[inucl] ; ++iprim) {
|
|
|
|
const double v = qmckl_mat(expo_per_nucleus, iprim, inucl) * r2;
|
|
|
|
if (v <= cutoff) {
|
|
|
|
ar2[iprim] = v;
|
|
|
|
++nidx;
|
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2023-11-28 17:00:39 +01:00
|
|
|
for (int64_t iprim = 0 ; iprim < nidx ; ++iprim) {
|
|
|
|
exp_mat[iprim][0] = exp(-ar2[iprim]);
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
for (int64_t iprim = 0 ; iprim < prim_num_per_nucleus[inucl] ; ++iprim) {
|
|
|
|
const double v = qmckl_mat(expo_per_nucleus, iprim, inucl) * r2;
|
|
|
|
if (v <= cutoff) {
|
2023-11-29 01:18:15 +01:00
|
|
|
ar2[iprim] = v;
|
2023-11-28 17:00:39 +01:00
|
|
|
ar2_sp[iprim] = (float) v;
|
|
|
|
++nidx;
|
|
|
|
} else {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2023-11-28 17:00:39 +01:00
|
|
|
for (int64_t iprim = 0 ; iprim < nidx ; ++iprim) {
|
|
|
|
exp_mat_sp[iprim] = expf(-ar2_sp[iprim]);
|
|
|
|
}
|
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2023-11-28 17:00:39 +01:00
|
|
|
for (int64_t iprim = 0 ; iprim < nidx ; ++iprim) {
|
|
|
|
exp_mat[iprim][0] = exp_mat_sp[iprim];
|
2022-05-10 19:18:19 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
2022-08-14 16:09:58 +02:00
|
|
|
|
2024-02-14 11:11:50 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2022-05-10 19:18:19 +02:00
|
|
|
for (int64_t iprim = 0 ; iprim < nidx ; ++iprim) {
|
|
|
|
double f = qmckl_mat(expo_per_nucleus, iprim, inucl) * exp_mat[iprim][0];
|
|
|
|
f = -f-f;
|
|
|
|
exp_mat[iprim][1] = f * x;
|
|
|
|
exp_mat[iprim][2] = f * y;
|
|
|
|
exp_mat[iprim][3] = f * z;
|
|
|
|
exp_mat[iprim][4] = f * (3.0 - 2.0 * ar2[iprim]);
|
|
|
|
}
|
|
|
|
|
2022-06-14 22:30:33 +02:00
|
|
|
|
2022-06-29 13:31:58 +02:00
|
|
|
/* --- */
|
2022-06-14 22:30:33 +02:00
|
|
|
|
2022-10-11 17:46:30 +02:00
|
|
|
// if (do_sparse) {
|
|
|
|
for (int i=0 ; i<nucleus_shell_num[inucl] ; ++i) {
|
2022-08-14 16:09:58 +02:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2022-10-11 17:46:30 +02:00
|
|
|
for (int j=0 ; j<8 ; ++j) {
|
|
|
|
ce_mat[i][j] = 0.;
|
|
|
|
}
|
2022-10-13 11:31:44 +02:00
|
|
|
const int32_t* restrict idx = &(coef_mat_sparse_idx[inucl][i][0]);
|
|
|
|
const double* restrict v = &(coef_mat_sparse[inucl][i][0]);
|
2022-10-11 17:46:30 +02:00
|
|
|
for (int l=1 ; l<idx[0]; ++l) {
|
|
|
|
const int k = idx[l];
|
|
|
|
if (k >= nidx) break;
|
2022-08-14 16:09:58 +02:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
2022-10-11 17:46:30 +02:00
|
|
|
for (int j=0 ; j<8 ; ++j) {
|
2024-01-30 11:31:07 +01:00
|
|
|
ce_mat[i][j] = ce_mat[i][j] + v[l] * exp_mat[k][j];
|
2022-06-29 13:31:58 +02:00
|
|
|
}
|
|
|
|
}
|
2022-05-10 19:18:19 +02:00
|
|
|
}
|
|
|
|
|
2022-10-11 17:46:30 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
const int64_t ishell_start = nucleus_index[inucl];
|
|
|
|
const int64_t ishell_end = nucleus_index[inucl] + nucleus_shell_num[inucl];
|
|
|
|
|
2023-11-28 17:00:39 +01:00
|
|
|
const int64_t ipoint_5_ao_num = ipoint*5*ao_num;
|
2022-05-10 19:18:19 +02:00
|
|
|
for (int64_t ishell = ishell_start ; ishell < ishell_end ; ++ishell) {
|
|
|
|
|
|
|
|
const double s1 = ce_mat[ishell-ishell_start][0];
|
|
|
|
|
|
|
|
const int64_t k = ao_index[ishell];
|
2023-11-28 17:00:39 +01:00
|
|
|
double* restrict const ao_vgl_1 = ao_vgl + ipoint_5_ao_num + k;
|
2022-05-10 19:18:19 +02:00
|
|
|
|
|
|
|
const int32_t l = shell_ang_mom[ishell];
|
|
|
|
const int32_t n = lstart[l+1]-lstart[l];
|
|
|
|
|
|
|
|
double* restrict const ao_vgl_2 = ao_vgl_1 + ao_num;
|
|
|
|
double* restrict const ao_vgl_3 = ao_vgl_1 + (ao_num<<1);
|
|
|
|
double* restrict const ao_vgl_4 = ao_vgl_1 + (ao_num<<1) + ao_num;
|
|
|
|
double* restrict const ao_vgl_5 = ao_vgl_1 + (ao_num<<2);
|
|
|
|
|
2022-06-15 23:21:31 +02:00
|
|
|
if (s1 == 0.0) {
|
2023-11-28 17:00:39 +01:00
|
|
|
/*
|
2022-06-15 23:21:31 +02:00
|
|
|
for (int64_t il=0 ; il<n ; ++il) {
|
|
|
|
ao_vgl_1[il] = 0.0;
|
|
|
|
ao_vgl_2[il] = 0.0;
|
|
|
|
ao_vgl_3[il] = 0.0;
|
|
|
|
ao_vgl_4[il] = 0.0;
|
|
|
|
ao_vgl_5[il] = 0.0;
|
|
|
|
}
|
2024-05-08 13:56:32 +02:00
|
|
|
,*/
|
2022-06-15 23:21:31 +02:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
const double s2 = ce_mat[ishell-ishell_start][1];
|
|
|
|
const double s3 = ce_mat[ishell-ishell_start][2];
|
|
|
|
const double s4 = ce_mat[ishell-ishell_start][3];
|
|
|
|
const double s5 = ce_mat[ishell-ishell_start][4];
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
double* restrict poly_vgl_1 = NULL;
|
|
|
|
double* restrict poly_vgl_2 = NULL;
|
|
|
|
double* restrict poly_vgl_3 = NULL;
|
|
|
|
double* restrict poly_vgl_4 = NULL;
|
|
|
|
double* restrict poly_vgl_5 = NULL;
|
|
|
|
if (nidx > 0) {
|
|
|
|
const double* restrict f = ao_factor + k;
|
|
|
|
const int64_t idx = lstart[l];
|
|
|
|
|
|
|
|
switch (nucleus_max_ang_mom[inucl]) {
|
|
|
|
case 0:
|
|
|
|
break;
|
|
|
|
case 1:
|
|
|
|
poly_vgl_1 = &(poly_vgl_l1[0][idx]);
|
|
|
|
poly_vgl_2 = &(poly_vgl_l1[1][idx]);
|
|
|
|
poly_vgl_3 = &(poly_vgl_l1[2][idx]);
|
|
|
|
poly_vgl_4 = &(poly_vgl_l1[3][idx]);
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
poly_vgl_1 = &(poly_vgl_l2[0][idx]);
|
|
|
|
poly_vgl_2 = &(poly_vgl_l2[1][idx]);
|
|
|
|
poly_vgl_3 = &(poly_vgl_l2[2][idx]);
|
|
|
|
poly_vgl_4 = &(poly_vgl_l2[3][idx]);
|
|
|
|
poly_vgl_5 = &(poly_vgl_l2[4][idx]);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
poly_vgl_1 = &(poly_vgl[0][idx]);
|
|
|
|
poly_vgl_2 = &(poly_vgl[1][idx]);
|
|
|
|
poly_vgl_3 = &(poly_vgl[2][idx]);
|
|
|
|
poly_vgl_4 = &(poly_vgl[3][idx]);
|
|
|
|
poly_vgl_5 = &(poly_vgl[4][idx]);
|
|
|
|
}
|
|
|
|
switch (n) {
|
2022-08-14 16:09:58 +02:00
|
|
|
case 1:
|
2022-05-10 19:18:19 +02:00
|
|
|
ao_vgl_1[0] = s1 * f[0];
|
|
|
|
ao_vgl_2[0] = s2 * f[0];
|
|
|
|
ao_vgl_3[0] = s3 * f[0];
|
|
|
|
ao_vgl_4[0] = s4 * f[0];
|
2024-01-11 14:33:40 +01:00
|
|
|
ao_vgl_5[0] = s5 * f[0];
|
2022-05-10 19:18:19 +02:00
|
|
|
break;
|
2022-08-14 16:09:58 +02:00
|
|
|
case 3:
|
2022-05-10 19:18:19 +02:00
|
|
|
#ifdef HAVE_OPENMP
|
|
|
|
#pragma omp simd
|
|
|
|
#endif
|
|
|
|
for (int il=0 ; il<3 ; ++il) {
|
|
|
|
ao_vgl_1[il] = poly_vgl_1[il] * s1 * f[il];
|
|
|
|
ao_vgl_2[il] = (poly_vgl_2[il] * s1 + poly_vgl_1[il] * s2) * f[il];
|
|
|
|
ao_vgl_3[il] = (poly_vgl_3[il] * s1 + poly_vgl_1[il] * s3) * f[il];
|
|
|
|
ao_vgl_4[il] = (poly_vgl_4[il] * s1 + poly_vgl_1[il] * s4) * f[il];
|
|
|
|
ao_vgl_5[il] = (poly_vgl_1[il] * s5 +
|
|
|
|
2.0*(poly_vgl_2[il] * s2 +
|
2022-03-21 18:32:39 +01:00
|
|
|
poly_vgl_3[il] * s3 +
|
|
|
|
poly_vgl_4[il] * s4 )) * f[il];
|
|
|
|
}
|
|
|
|
break;
|
2022-08-14 16:09:58 +02:00
|
|
|
case 6:
|
2022-02-25 16:30:16 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
2022-03-21 18:32:39 +01:00
|
|
|
#pragma omp simd
|
2022-02-25 16:30:16 +01:00
|
|
|
#endif
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int il=0 ; il<6 ; ++il) {
|
|
|
|
ao_vgl_1[il] = poly_vgl_1[il] * s1 * f[il];
|
|
|
|
ao_vgl_2[il] = (poly_vgl_2[il] * s1 + poly_vgl_1[il] * s2) * f[il];
|
|
|
|
ao_vgl_3[il] = (poly_vgl_3[il] * s1 + poly_vgl_1[il] * s3) * f[il];
|
|
|
|
ao_vgl_4[il] = (poly_vgl_4[il] * s1 + poly_vgl_1[il] * s4) * f[il];
|
|
|
|
ao_vgl_5[il] = (poly_vgl_5[il] * s1 + poly_vgl_1[il] * s5 +
|
|
|
|
2.0*(poly_vgl_2[il] * s2 +
|
|
|
|
poly_vgl_3[il] * s3 +
|
|
|
|
poly_vgl_4[il] * s4 )) * f[il];
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
2022-02-25 16:30:16 +01:00
|
|
|
#ifdef HAVE_OPENMP
|
2022-06-15 23:21:31 +02:00
|
|
|
#pragma omp simd
|
2022-02-25 16:30:16 +01:00
|
|
|
#endif
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int il=0 ; il<n ; ++il) {
|
|
|
|
ao_vgl_1[il] = poly_vgl_1[il] * s1 * f[il];
|
|
|
|
ao_vgl_2[il] = (poly_vgl_2[il] * s1 + poly_vgl_1[il] * s2) * f[il];
|
|
|
|
ao_vgl_3[il] = (poly_vgl_3[il] * s1 + poly_vgl_1[il] * s3) * f[il];
|
|
|
|
ao_vgl_4[il] = (poly_vgl_4[il] * s1 + poly_vgl_1[il] * s4) * f[il];
|
|
|
|
ao_vgl_5[il] = (poly_vgl_5[il] * s1 + poly_vgl_1[il] * s5 +
|
|
|
|
2.0*(poly_vgl_2[il] * s2 +
|
|
|
|
poly_vgl_3[il] * s3 +
|
|
|
|
poly_vgl_4[il] * s4 )) * f[il];
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
2023-11-28 17:00:39 +01:00
|
|
|
/*
|
2024-02-23 11:56:28 +01:00
|
|
|
} else {
|
2022-03-21 18:32:39 +01:00
|
|
|
for (int64_t il=0 ; il<n ; ++il) {
|
|
|
|
ao_vgl_1[il] = 0.0;
|
|
|
|
ao_vgl_2[il] = 0.0;
|
|
|
|
ao_vgl_3[il] = 0.0;
|
|
|
|
ao_vgl_4[il] = 0.0;
|
|
|
|
ao_vgl_5[il] = 0.0;
|
|
|
|
}
|
2024-05-08 13:56:32 +02:00
|
|
|
,*/
|
2022-02-16 00:21:37 +01:00
|
|
|
}
|
2022-03-21 18:32:39 +01:00
|
|
|
}
|
|
|
|
}
|
2022-02-16 00:21:37 +01:00
|
|
|
}
|
2023-09-13 13:23:16 +02:00
|
|
|
free(coef_mat_sparse_idx);
|
|
|
|
free(coef_mat_sparse);
|
2022-02-16 00:21:37 +01:00
|
|
|
}
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2022-02-16 00:21:37 +01:00
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-03-01 14:17:38 +01:00
|
|
|
#endif
|
2022-05-10 19:18:19 +02:00
|
|
|
#+end_src
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
*** Interfaces
|
|
|
|
# #+CALL: generate_c_header(table=qmckl_ao_vgl_args_doc,rettyp=get_value("CRetType"),fname="qmckl_compute_ao_vgl"))
|
|
|
|
# (Commented because the header needs to go into h_private_func)
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+begin_src c :tangle (eval h_private_func) :comments org
|
2022-02-14 19:11:37 +01:00
|
|
|
qmckl_exit_code qmckl_compute_ao_vgl_doc (
|
2021-07-11 23:20:40 +02:00
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t ao_num,
|
|
|
|
const int64_t shell_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
const int64_t point_num,
|
2021-07-11 23:20:40 +02:00
|
|
|
const int64_t nucl_num,
|
2022-01-26 17:06:51 +01:00
|
|
|
const double* coord,
|
2021-07-11 23:20:40 +02:00
|
|
|
const double* nucl_coord,
|
|
|
|
const int64_t* nucleus_index,
|
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const double* nucleus_range,
|
|
|
|
const int32_t* nucleus_max_ang_mom,
|
|
|
|
const int32_t* shell_ang_mom,
|
2021-07-13 09:54:43 +02:00
|
|
|
const double* ao_factor,
|
2021-07-11 23:20:40 +02:00
|
|
|
const double* shell_vgl,
|
2021-07-23 12:27:18 +02:00
|
|
|
double* const ao_vgl );
|
2022-05-10 19:18:19 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :tangle (eval h_private_func) :comments org
|
2022-03-01 14:17:38 +01:00
|
|
|
#ifdef HAVE_HPC
|
2022-02-19 19:24:18 +01:00
|
|
|
qmckl_exit_code qmckl_compute_ao_vgl_hpc_gaussian (
|
2022-02-14 19:11:37 +01:00
|
|
|
const qmckl_context context,
|
|
|
|
const int64_t ao_num,
|
|
|
|
const int64_t shell_num,
|
2022-03-21 18:32:39 +01:00
|
|
|
const int32_t* prim_num_per_nucleus,
|
2022-02-14 19:11:37 +01:00
|
|
|
const int64_t point_num,
|
|
|
|
const int64_t nucl_num,
|
|
|
|
const double* coord,
|
|
|
|
const double* nucl_coord,
|
|
|
|
const int64_t* nucleus_index,
|
|
|
|
const int64_t* nucleus_shell_num,
|
|
|
|
const double* nucleus_range,
|
|
|
|
const int32_t* nucleus_max_ang_mom,
|
|
|
|
const int32_t* shell_ang_mom,
|
|
|
|
const double* ao_factor,
|
2022-03-21 18:32:39 +01:00
|
|
|
const qmckl_matrix expo_per_nucleus,
|
|
|
|
const qmckl_tensor coef_per_nucleus,
|
2022-02-14 19:11:37 +01:00
|
|
|
double* const ao_vgl );
|
2022-03-01 14:17:38 +01:00
|
|
|
#endif
|
2022-05-10 19:18:19 +02:00
|
|
|
#+end_src
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
**** Provide :noexport:
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2022-07-11 13:42:39 +02:00
|
|
|
#+CALL: write_provider_header( group="ao_basis", data="ao_vgl" )
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2022-07-11 13:42:39 +02:00
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval h_private_func) :noweb yes :export none
|
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_ao_vgl(qmckl_context context);
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+CALL: write_provider_pre( group="ao_basis", data="ao_vgl", dimension="ctx->ao_basis.ao_num * 5 * ctx->point.num")
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
|
|
|
qmckl_exit_code qmckl_provide_ao_basis_ao_vgl(qmckl_context context)
|
2022-01-05 19:22:16 +01:00
|
|
|
{
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_exit_code rc = QMCKL_SUCCESS;
|
2022-07-11 13:42:39 +02:00
|
|
|
|
2022-01-05 19:22:16 +01:00
|
|
|
if (qmckl_context_check(context) == QMCKL_NULL_CONTEXT) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_INVALID_CONTEXT,
|
2022-07-11 13:42:39 +02:00
|
|
|
"qmckl_provide_ao_basis_ao_vgl",
|
2022-01-05 19:22:16 +01:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
2022-04-05 11:03:38 +02:00
|
|
|
qmckl_context_struct* const ctx = (qmckl_context_struct*) context;
|
2022-01-05 19:22:16 +01:00
|
|
|
assert (ctx != NULL);
|
|
|
|
|
|
|
|
if (!ctx->ao_basis.provided) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_NOT_PROVIDED,
|
2022-07-11 13:42:39 +02:00
|
|
|
"qmckl_provide_ao_basis_ao_vgl",
|
2022-01-05 19:22:16 +01:00
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Compute if necessary */
|
2022-01-26 17:06:51 +01:00
|
|
|
if (ctx->point.date > ctx->ao_basis.ao_vgl_date) {
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
qmckl_memory_info_struct mem_info = qmckl_memory_info_struct_zero;
|
|
|
|
mem_info.size = ctx->ao_basis.ao_num * 5 * ctx->point.num * sizeof(double);
|
|
|
|
|
|
|
|
if (ctx->ao_basis.ao_vgl != NULL) {
|
|
|
|
qmckl_memory_info_struct mem_info_test = qmckl_memory_info_struct_zero;
|
|
|
|
rc = qmckl_get_malloc_info(context, ctx->ao_basis.ao_vgl, &mem_info_test);
|
|
|
|
|
|
|
|
/* if rc != QMCKL_SUCCESS, we are maybe in an _inplace function because the
|
|
|
|
memory was not allocated with qmckl_malloc */
|
|
|
|
|
|
|
|
if ((rc == QMCKL_SUCCESS) && (mem_info_test.size != mem_info.size)) {
|
2022-07-11 13:42:39 +02:00
|
|
|
rc = qmckl_free(context, ctx->ao_basis.ao_vgl);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
ctx->ao_basis.ao_vgl = NULL;
|
|
|
|
}
|
2022-01-05 19:22:16 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Allocate array */
|
|
|
|
if (ctx->ao_basis.ao_vgl == NULL) {
|
|
|
|
|
|
|
|
double* ao_vgl = (double*) qmckl_malloc(context, mem_info);
|
|
|
|
|
|
|
|
if (ao_vgl == NULL) {
|
|
|
|
return qmckl_failwith( context,
|
|
|
|
QMCKL_ALLOCATION_FAILED,
|
|
|
|
"qmckl_ao_basis_ao_vgl",
|
|
|
|
NULL);
|
|
|
|
}
|
|
|
|
ctx->ao_basis.ao_vgl = ao_vgl;
|
|
|
|
}
|
2022-07-11 13:42:39 +02:00
|
|
|
|
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :exports none
|
2022-02-14 19:11:37 +01:00
|
|
|
#ifdef HAVE_HPC
|
2023-11-30 12:50:06 +01:00
|
|
|
if (ctx->ao_basis.type == 'G') {
|
|
|
|
rc = qmckl_compute_ao_vgl_hpc_gaussian(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->ao_basis.prim_num_per_nucleus,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
2023-11-30 12:56:06 +01:00
|
|
|
ctx->ao_basis.nucleus_range,
|
2023-11-30 12:50:06 +01:00
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.expo_per_nucleus,
|
|
|
|
ctx->ao_basis.coef_per_nucleus,
|
|
|
|
ctx->ao_basis.ao_vgl);
|
|
|
|
|
|
|
|
/* DEBUG
|
|
|
|
rc = qmckl_provide_ao_basis_shell_vgl(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc, "qmckl_provide_ao_basis_shell_vgl", NULL);
|
|
|
|
}
|
2022-06-15 23:21:31 +02:00
|
|
|
int64_t K= ctx->ao_basis.ao_num * 5 * ctx->point.num;
|
|
|
|
double* check = malloc(K*sizeof(double));
|
2023-11-30 12:50:06 +01:00
|
|
|
rc = qmckl_compute_ao_vgl_doc(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
|
|
|
ctx->ao_basis.nucleus_range,
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.shell_vgl,
|
|
|
|
check);
|
2022-06-15 23:21:31 +02:00
|
|
|
for (int64_t i=0 ; i<K ; ++i) {
|
2023-11-30 12:50:06 +01:00
|
|
|
if (fabs(check[i] - ctx->ao_basis.ao_vgl[i]) > 1.e-10) {
|
|
|
|
int a, b, c;
|
|
|
|
a = i/(ctx->ao_basis.ao_num*5);
|
|
|
|
b = (i-a*ctx->ao_basis.ao_num*5)/ctx->ao_basis.ao_num;
|
|
|
|
c = (i-a*ctx->ao_basis.ao_num*5 -b*ctx->ao_basis.ao_num);
|
|
|
|
printf("%d: %d, %d, %d, %e %e\n", i, a, b, c, check[i], ctx->ao_basis.ao_vgl[i]);
|
2022-06-15 23:21:31 +02:00
|
|
|
}
|
2023-11-30 12:50:06 +01:00
|
|
|
}
|
|
|
|
,*/
|
2022-06-15 23:21:31 +02:00
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
/*
|
2022-02-19 19:24:18 +01:00
|
|
|
} else if (ctx->ao_basis.type == 'S') {
|
2023-11-30 12:50:06 +01:00
|
|
|
rc = qmck_compute_ao_vgl_hpc_slater(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->ao_basis.prim_num,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
|
|
|
ctx->ao_basis.nucleus_range,
|
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.shell_prim_index,
|
|
|
|
ctx->ao_basis.shell_prim_num,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.exponent,
|
|
|
|
ctx->ao_basis.coefficient_normalized,
|
|
|
|
ctx->ao_basis.ao_vgl);
|
|
|
|
,*/
|
|
|
|
} else {
|
|
|
|
/* Provide required data */
|
2022-07-11 13:42:39 +02:00
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
rc = qmckl_provide_ao_basis_shell_vgl(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc, "qmckl_provide_ao_basis_shell_vgl", NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
rc = qmckl_compute_ao_vgl_doc(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
2023-11-30 12:56:06 +01:00
|
|
|
ctx->ao_basis.nucleus_range,
|
2023-11-30 12:50:06 +01:00
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.shell_vgl,
|
|
|
|
ctx->ao_basis.ao_vgl);
|
|
|
|
}
|
2022-02-14 19:11:37 +01:00
|
|
|
#else
|
2023-11-30 12:50:06 +01:00
|
|
|
rc = qmckl_provide_ao_basis_shell_vgl(context);
|
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return qmckl_failwith( context, rc, "qmckl_provide_ao_basis_shell_vgl", NULL);
|
|
|
|
}
|
2022-07-11 13:42:39 +02:00
|
|
|
|
2023-11-30 12:50:06 +01:00
|
|
|
rc = qmckl_compute_ao_vgl_doc(context,
|
|
|
|
ctx->ao_basis.ao_num,
|
|
|
|
ctx->ao_basis.shell_num,
|
|
|
|
ctx->point.num,
|
|
|
|
ctx->nucleus.num,
|
|
|
|
ctx->point.coord.data,
|
|
|
|
ctx->nucleus.coord.data,
|
|
|
|
ctx->ao_basis.nucleus_index,
|
|
|
|
ctx->ao_basis.nucleus_shell_num,
|
2023-11-30 12:56:06 +01:00
|
|
|
ctx->ao_basis.nucleus_range,
|
2023-11-30 12:50:06 +01:00
|
|
|
ctx->ao_basis.nucleus_max_ang_mom,
|
|
|
|
ctx->ao_basis.shell_ang_mom,
|
|
|
|
ctx->ao_basis.ao_factor,
|
|
|
|
ctx->ao_basis.shell_vgl,
|
|
|
|
ctx->ao_basis.ao_vgl);
|
2022-02-14 19:11:37 +01:00
|
|
|
#endif
|
2022-07-11 13:42:39 +02:00
|
|
|
#+end_src
|
|
|
|
|
|
|
|
#+CALL: write_provider_post( group="ao_basis", data="ao_vgl" )
|
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
#+begin_src c :comments org :tangle (eval c) :noweb yes :export none
|
2022-01-05 19:22:16 +01:00
|
|
|
if (rc != QMCKL_SUCCESS) {
|
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx->ao_basis.ao_vgl_date = ctx->date;
|
|
|
|
}
|
|
|
|
|
|
|
|
return QMCKL_SUCCESS;
|
|
|
|
}
|
2022-07-11 13:42:39 +02:00
|
|
|
#+end_src
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
**** Test :noexport:
|
2022-01-05 19:22:16 +01:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+begin_src python :results output :exports none
|
2021-07-11 23:20:40 +02:00
|
|
|
import numpy as np
|
2021-09-23 11:16:54 +02:00
|
|
|
from math import sqrt
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
h0 = 1.e-4
|
2021-07-11 23:20:40 +02:00
|
|
|
def f(a,x,y):
|
2021-07-23 12:27:18 +02:00
|
|
|
return np.sum( [c * np.exp( -b*(np.linalg.norm(x-y))**2) for b,c in a] )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
def fx(a,x,y):
|
|
|
|
return f(a,x,y) * (x[0] - y[0])**2
|
|
|
|
|
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
def df(a,x,y,n):
|
2022-03-21 18:32:39 +01:00
|
|
|
if n == 1: h = np.array([h0,0.,0.])
|
|
|
|
elif n == 2: h = np.array([0.,h0,0.])
|
|
|
|
elif n == 3: h = np.array([0.,0.,h0])
|
|
|
|
return ( fx(a,x+h,y) - fx(a,x-h,y) ) / (2.*h0)
|
|
|
|
# return np.sum( [-2.*b * c * np.exp( -b*(np.linalg.norm(x-y))**2) for b,c in a] ) * (x-y)[n-1]
|
2021-07-11 23:20:40 +02:00
|
|
|
|
|
|
|
def d2f(a,x,y,n):
|
|
|
|
if n == 1: h = np.array([h0,0.,0.])
|
|
|
|
elif n == 2: h = np.array([0.,h0,0.])
|
|
|
|
elif n == 3: h = np.array([0.,0.,h0])
|
2022-03-21 18:32:39 +01:00
|
|
|
return ( fx(a,x+h,y) - 2.*fx(a,x,y) + fx(a,x-h,y) ) / h0**2
|
2022-06-11 10:57:58 +02:00
|
|
|
# return np.sum( [( (2.*b*(x-y)[n-1])**2 -2.*b ) * c * np.exp( -b*(np.linalg.norm(x-y))**2) for b,c in a] )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
|
|
|
def lf(a,x,y):
|
2022-06-11 10:57:58 +02:00
|
|
|
# return np.sum( [( (2.*b*np.linalg.norm(x-y))**2 -6.*b ) * c * np.exp( -b*(np.linalg.norm(x-y))**2) for b,c in a] )
|
2021-07-11 23:20:40 +02:00
|
|
|
return d2f(a,x,y,1) + d2f(a,x,y,2) + d2f(a,x,y,3)
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
|
2021-07-11 23:20:40 +02:00
|
|
|
elec_26_w1 = np.array( [ 1.49050402641, 2.90106987953, -1.05920815468 ] )
|
|
|
|
elec_15_w2 = np.array( [ -2.20180344582,-1.9113150239, 2.2193744778600002 ] )
|
|
|
|
nucl_1 = np.array( [ -2.302574592081335e+00, -3.542027060505035e-01, -5.334129934317614e-02] )
|
|
|
|
|
2021-09-23 11:16:54 +02:00
|
|
|
#double ao_vgl[prim_num][5][elec_num];
|
2021-07-11 23:20:40 +02:00
|
|
|
x = elec_26_w1 ; y = nucl_1
|
2022-03-21 18:32:39 +01:00
|
|
|
a = [( 4.0382999999999998e+02, 1.4732000000000000e-03 * 5.9876577632594533e+04),
|
|
|
|
( 1.2117000000000000e+02, 1.2672500000000000e-02 * 7.2836806319891484e+03),
|
|
|
|
( 4.6344999999999999e+01, 5.8045100000000002e-02 * 1.3549226646722386e+03),
|
|
|
|
( 1.9721000000000000e+01, 1.7051030000000000e-01 * 3.0376315094739988e+02),
|
|
|
|
( 8.8623999999999992e+00, 3.1859579999999998e-01 * 7.4924579607137730e+01),
|
|
|
|
( 3.9962000000000000e+00, 3.8450230000000002e-01 * 1.8590543353806009e+01),
|
|
|
|
( 1.7636000000000001e+00, 2.7377370000000001e-01 * 4.4423176930919421e+00),
|
|
|
|
( 7.0618999999999998e-01, 7.4396699999999996e-02 * 8.9541051939952665e-01)]
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-09-23 11:16:54 +02:00
|
|
|
norm = sqrt(3.)
|
2022-03-21 18:32:39 +01:00
|
|
|
# x^2 * g(r)
|
|
|
|
print ( "[26][0][219] : %25.15e"%(fx(a,x,y)) )
|
|
|
|
print ( "[26][1][219] : %25.15e"%(df(a,x,y,1)) )
|
|
|
|
print ( "[26][2][219] : %25.15e"%(df(a,x,y,2)) )
|
|
|
|
print ( "[26][3][219] : %25.15e"%(df(a,x,y,3)) )
|
|
|
|
print ( "[26][4][219] : %25.15e"%(lf(a,x,y)) )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
print ( "[26][0][220] : %25.15e"%(norm*f(a,x,y) * (x[0] - y[0]) * (x[1] - y[1]) ))
|
|
|
|
print ( "[26][1][220] : %25.15e"%(norm*df(a,x,y,1)* (x[0] - y[0]) * (x[1] - y[1]) + norm*f(a,x,y) * (x[1] - y[1])) )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
print ( "[26][0][221] : %25.15e"%(norm*f(a,x,y) * (x[0] - y[0]) * (x[2] - y[2])) )
|
|
|
|
print ( "[26][1][221] : %25.15e"%(norm*df(a,x,y,1)* (x[0] - y[0]) * (x[2] - y[2]) + norm*f(a,x,y) * (x[2] - y[2])) )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
print ( "[26][0][222] : %25.15e"%(f(a,x,y) * (x[1] - y[1]) * (x[1] - y[1])) )
|
|
|
|
print ( "[26][1][222] : %25.15e"%(df(a,x,y,1)* (x[1] - y[1]) * (x[1] - y[1])) )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
print ( "[26][0][223] : %25.15e"%(norm*f(a,x,y) * (x[1] - y[1]) * (x[2] - y[2])) )
|
|
|
|
print ( "[26][1][223] : %25.15e"%(norm*df(a,x,y,1)* (x[1] - y[1]) * (x[2] - y[2])) )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
print ( "[26][0][224] : %25.15e"%(f(a,x,y) * (x[2] - y[2]) * (x[2] - y[2])) )
|
|
|
|
print ( "[26][1][224] : %25.15e"%(df(a,x,y,1)* (x[2] - y[2]) * (x[2] - y[2])) )
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+end_src
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+RESULTS:
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+begin_src c :tangle (eval c_test) :exports none
|
2021-07-11 23:20:40 +02:00
|
|
|
{
|
|
|
|
#define shell_num chbrclf_shell_num
|
|
|
|
#define ao_num chbrclf_ao_num
|
|
|
|
|
|
|
|
double* elec_coord = &(chbrclf_elec_coord[0][0][0]);
|
|
|
|
|
|
|
|
assert(qmckl_electron_provided(context));
|
|
|
|
|
2023-09-14 11:00:24 +02:00
|
|
|
int64_t point_num = elec_num;
|
2022-08-07 14:57:10 +02:00
|
|
|
rc = qmckl_set_point(context, 'N', point_num, elec_coord, point_num*3);
|
2021-07-11 23:20:40 +02:00
|
|
|
assert(rc == QMCKL_SUCCESS);
|
|
|
|
|
|
|
|
|
2022-08-07 14:57:10 +02:00
|
|
|
double ao_vgl[point_num][5][ao_num];
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2022-01-06 02:28:13 +01:00
|
|
|
rc = qmckl_get_ao_basis_ao_vgl(context, &(ao_vgl[0][0][0]),
|
2022-08-07 14:57:10 +02:00
|
|
|
(int64_t) 5*point_num*ao_num);
|
2021-07-11 23:20:40 +02:00
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
|
|
|
printf("\n");
|
2022-03-21 18:32:39 +01:00
|
|
|
printf(" ao_vgl ao_vgl[26][0][219] %25.15e\n", ao_vgl[26][0][219]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][1][219] %25.15e\n", ao_vgl[26][1][219]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][2][219] %25.15e\n", ao_vgl[26][2][219]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][3][219] %25.15e\n", ao_vgl[26][3][219]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][4][219] %25.15e\n", ao_vgl[26][4][219]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][0][220] %25.15e\n", ao_vgl[26][0][220]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][1][220] %25.15e\n", ao_vgl[26][1][220]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][2][220] %25.15e\n", ao_vgl[26][2][220]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][3][220] %25.15e\n", ao_vgl[26][3][220]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][4][220] %25.15e\n", ao_vgl[26][4][220]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][0][221] %25.15e\n", ao_vgl[26][0][221]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][1][221] %25.15e\n", ao_vgl[26][1][221]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][2][221] %25.15e\n", ao_vgl[26][2][221]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][3][221] %25.15e\n", ao_vgl[26][3][221]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][4][221] %25.15e\n", ao_vgl[26][4][221]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][0][222] %25.15e\n", ao_vgl[26][0][222]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][1][222] %25.15e\n", ao_vgl[26][1][222]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][2][222] %25.15e\n", ao_vgl[26][2][222]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][3][222] %25.15e\n", ao_vgl[26][3][222]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][4][222] %25.15e\n", ao_vgl[26][4][222]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][0][223] %25.15e\n", ao_vgl[26][0][223]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][1][223] %25.15e\n", ao_vgl[26][1][223]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][2][223] %25.15e\n", ao_vgl[26][2][223]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][3][223] %25.15e\n", ao_vgl[26][3][223]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][4][223] %25.15e\n", ao_vgl[26][4][223]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][0][224] %25.15e\n", ao_vgl[26][0][224]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][1][224] %25.15e\n", ao_vgl[26][1][224]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][2][224] %25.15e\n", ao_vgl[26][2][224]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][3][224] %25.15e\n", ao_vgl[26][3][224]);
|
|
|
|
printf(" ao_vgl ao_vgl[26][4][224] %25.15e\n", ao_vgl[26][4][224]);
|
2021-07-11 23:20:40 +02:00
|
|
|
printf("\n");
|
|
|
|
|
2022-03-21 18:32:39 +01:00
|
|
|
assert( fabs(ao_vgl[26][0][219] - ( 1.020298798341620e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][1][219] - ( -4.928035238010602e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][2][219] - ( -4.691009312035986e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][3][219] - ( 1.449504046436699e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][4][219] - ( 4.296442111843973e-07)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][0][220] - ( 1.516643537739178e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][1][220] - ( -7.725221462603871e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][2][220] - ( -6.507140835104833e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][3][220] - ( 2.154644255710413e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][4][220] - ( 6.365449359656352e-07)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][0][221] - ( -4.686370882518819e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][1][221] - ( 2.387064067626827e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][2][221] - ( 2.154644255710412e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][3][221] - ( -1.998731863512374e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][4][221] - ( -1.966899656441993e-07)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][0][222] - ( 7.514816980753531e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][1][222] - ( -4.025889138635182e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][2][222] - ( -2.993372555126361e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][3][222] - ( 1.067604670272904e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][4][222] - ( 3.168199650002648e-07)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][0][223] - ( -4.021908374204471e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][1][223] - ( 2.154644255710413e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][2][223] - ( 1.725594944732276e-08)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][3][223] - ( -1.715339357718333e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][4][223] - ( -1.688020516893476e-07)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][0][224] - ( 7.175045873560788e-10)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][1][224] - ( -3.843864637762753e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][2][224] - ( -3.298857850451910e-09)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][3][224] - ( -4.073047518790881e-10)) < 1.e-14 );
|
|
|
|
assert( fabs(ao_vgl[26][4][224] - ( 3.153244195820293e-08)) < 1.e-14 );
|
|
|
|
|
2021-10-13 15:01:23 +02:00
|
|
|
}
|
|
|
|
|
2022-05-10 19:18:19 +02:00
|
|
|
#+end_src
|
2021-07-11 23:20:40 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
* End of files :noexport:
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
#+begin_src c :tangle (eval h_private_type)
|
|
|
|
#endif
|
|
|
|
#+end_src
|
|
|
|
|
2021-10-14 21:40:14 +02:00
|
|
|
#+begin_src c :tangle (eval h_private_func)
|
|
|
|
#endif
|
|
|
|
#+end_src
|
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
*** Test
|
|
|
|
#+begin_src c :tangle (eval c_test)
|
2021-05-11 16:41:03 +02:00
|
|
|
rc = qmckl_context_destroy(context);
|
|
|
|
assert (rc == QMCKL_SUCCESS);
|
|
|
|
|
|
|
|
return 0;
|
2021-04-01 01:19:33 +02:00
|
|
|
}
|
|
|
|
#+end_src
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-10-14 21:40:14 +02:00
|
|
|
*** Compute file names
|
2021-04-30 01:26:19 +02:00
|
|
|
#+begin_src emacs-lisp
|
|
|
|
; The following is required to compute the file names
|
|
|
|
|
|
|
|
(setq pwd (file-name-directory buffer-file-name))
|
|
|
|
(setq name (file-name-nondirectory (substring buffer-file-name 0 -4)))
|
|
|
|
(setq f (concat pwd name "_f.f90"))
|
|
|
|
(setq fh (concat pwd name "_fh.f90"))
|
|
|
|
(setq c (concat pwd name ".c"))
|
|
|
|
(setq h (concat name ".h"))
|
|
|
|
(setq h_private (concat name "_private.h"))
|
|
|
|
(setq c_test (concat pwd "test_" name ".c"))
|
|
|
|
(setq f_test (concat pwd "test_" name "_f.f90"))
|
|
|
|
|
|
|
|
; Minted
|
|
|
|
(require 'ox-latex)
|
|
|
|
(setq org-latex-listings 'minted)
|
|
|
|
(add-to-list 'org-latex-packages-alist '("" "listings"))
|
|
|
|
(add-to-list 'org-latex-packages-alist '("" "color"))
|
|
|
|
|
|
|
|
#+end_src
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
#+RESULTS:
|
|
|
|
| | color |
|
|
|
|
| | listings |
|
2021-04-30 01:26:19 +02:00
|
|
|
|
2021-04-01 01:19:33 +02:00
|
|
|
|
|
|
|
# -*- mode: org -*-
|
|
|
|
# vim: syntax=c
|
2021-10-14 16:32:58 +02:00
|
|
|
|
2022-01-05 12:26:11 +01:00
|
|
|
* TODO [0/1] Missing features :noexport:
|
|
|
|
- [ ] Error messages to tell what is missing when initializing
|