mirror of
https://github.com/LCPQ/quantum_package
synced 2024-08-25 05:51:46 +02:00
94 lines
2.0 KiB
OCaml
94 lines
2.0 KiB
OCaml
open Qptypes;;
|
|
open Qputils;;
|
|
open Core.Std;;
|
|
|
|
module Hartree_fock : sig
|
|
type t =
|
|
{ n_it_scf_max : Strictly_positive_int.t;
|
|
thresh_scf : Threshold.t;
|
|
} with sexp
|
|
;;
|
|
val read : unit -> t
|
|
val to_string : t -> string
|
|
val to_rst : t -> Rst_string.t
|
|
val of_rst : Rst_string.t -> t
|
|
end = struct
|
|
type t =
|
|
{ n_it_scf_max : Strictly_positive_int.t;
|
|
thresh_scf : Threshold.t;
|
|
} with sexp
|
|
;;
|
|
|
|
let get_default = Qpackage.get_ezfio_default "hartree_fock";;
|
|
|
|
let read_n_it_scf_max () =
|
|
if not (Ezfio.has_hartree_fock_n_it_scf_max ()) then
|
|
get_default "n_it_scf_max"
|
|
|> Int.of_string
|
|
|> Ezfio.set_hartree_fock_n_it_scf_max
|
|
;
|
|
Ezfio.get_hartree_fock_n_it_scf_max ()
|
|
|> Strictly_positive_int.of_int
|
|
;;
|
|
|
|
let read_thresh_scf() =
|
|
if not (Ezfio.has_hartree_fock_thresh_scf()) then
|
|
get_default "thresh_scf"
|
|
|> Float.of_string
|
|
|> Ezfio.set_hartree_fock_thresh_scf
|
|
;
|
|
Ezfio.get_hartree_fock_thresh_scf ()
|
|
|> Threshold.of_float ;;
|
|
|
|
|
|
let read () =
|
|
{ n_it_scf_max = read_n_it_scf_max ();
|
|
thresh_scf = read_thresh_scf ();
|
|
}
|
|
;;
|
|
|
|
let to_string b =
|
|
Printf.sprintf "
|
|
n_it_scf_max = %s
|
|
thresh_scf = %s
|
|
"
|
|
(Strictly_positive_int.to_string b.n_it_scf_max)
|
|
(Threshold.to_string b.thresh_scf)
|
|
;;
|
|
|
|
let to_rst b =
|
|
Printf.sprintf "
|
|
Max number of SCF iterations ::
|
|
|
|
n_it_scf_max = %s
|
|
|
|
SCF convergence criterion (on energy) ::
|
|
|
|
thresh_scf = %s
|
|
|
|
"
|
|
(Strictly_positive_int.to_string b.n_it_scf_max)
|
|
(Threshold.to_string b.thresh_scf)
|
|
|> Rst_string.of_string
|
|
;;
|
|
|
|
let of_rst s =
|
|
let s = Rst_string.to_string s
|
|
|> String.split ~on:'\n'
|
|
|> List.filter ~f:(fun line ->
|
|
String.contains line '=')
|
|
|> List.map ~f:(fun line ->
|
|
"("^(
|
|
String.tr line ~target:'=' ~replacement:' '
|
|
)^")" )
|
|
|> String.concat
|
|
in
|
|
Sexp.of_string ("("^s^")")
|
|
|> t_of_sexp
|
|
;;
|
|
|
|
|
|
end
|
|
|
|
|