QCaml/Basis/Basis.ml

41 lines
1.1 KiB
OCaml

type t = Contracted_shell.t array
(** Returns an array of the basis set per atom *)
let of_nuclei_and_general_basis n b =
Array.map (fun (e, center) ->
List.assoc e b
|> Array.map (fun (totAngMom, shell) ->
let expo = Array.map (fun General_basis.{exponent ; coefficient} ->
exponent) shell
and coef = Array.map (fun General_basis.{exponent ; coefficient} ->
coefficient) shell
in
Contracted_shell.create ~expo ~coef ~totAngMom ~center)
) n
|> Array.to_list
|> Array.concat
let to_string b =
let line ="
-----------------------------------------------------------------------
" in
"
Atomic Basis set
----------------
-----------------------------------------------------------------------
Angular Coordinates (Bohr) Exponents Coefficients
Momentum X Y Z
-----------------------------------------------------------------------
"
^
( Array.map (fun i ->
Contracted_shell.to_string i) b
|> Array.to_list
|> String.concat line
)
^ line