mirror of
https://gitlab.com/scemama/QCaml.git
synced 2024-12-22 20:33:36 +01:00
Added nuclei.org
This commit is contained in:
parent
a4582ffb67
commit
f45423a9ba
@ -1,8 +1,21 @@
|
|||||||
|
(* [[file:../nuclei.org::*Type][Type:2]] *)
|
||||||
open Common
|
open Common
|
||||||
open Xyz_ast
|
|
||||||
|
|
||||||
type t = (Element.t * Coordinate.t) array
|
type t = (Element.t * Coordinate.t) array
|
||||||
|
open Xyz_ast
|
||||||
|
(* Type:2 ends here *)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
(* | ~of_xyz_string~ | Create from a string, in xyz format |
|
||||||
|
* | ~of_xyz_file~ | Create from a file, in xyz format |
|
||||||
|
* | ~of_zmt_string~ | Create from a string, in z-matrix format |
|
||||||
|
* | ~of_zmt_file~ | Create from a file, in z-matrix format |
|
||||||
|
* | ~to_string~ | Transform to a string, for printing |
|
||||||
|
* | ~of_filename~ | Detects the type of file (xyz, z-matrix) and reads the file | *)
|
||||||
|
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Conversion][Conversion:2]] *)
|
||||||
let of_xyz_lexbuf lexbuf =
|
let of_xyz_lexbuf lexbuf =
|
||||||
let data =
|
let data =
|
||||||
Xyz_parser.input Nuclei_lexer.read_all lexbuf
|
Xyz_parser.input Nuclei_lexer.read_all lexbuf
|
||||||
@ -41,9 +54,9 @@ let of_zmt_string buffer =
|
|||||||
Zmatrix.of_string buffer
|
Zmatrix.of_string buffer
|
||||||
|> Zmatrix.to_xyz
|
|> Zmatrix.to_xyz
|
||||||
|> Array.map (fun (e,x,y,z) ->
|
|> Array.map (fun (e,x,y,z) ->
|
||||||
(e, Coordinate.(angstrom_to_bohr @@ make_angstrom { x ; y ; z} ))
|
(e, Coordinate.(angstrom_to_bohr @@ make_angstrom { x ; y ; z} ))
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
let of_zmt_file filename =
|
let of_zmt_file filename =
|
||||||
let ic = open_in filename in
|
let ic = open_in filename in
|
||||||
@ -56,11 +69,11 @@ let of_zmt_file filename =
|
|||||||
List.rev accu
|
List.rev accu
|
||||||
|> String.concat "\n"
|
|> String.concat "\n"
|
||||||
in aux []
|
in aux []
|
||||||
|> of_zmt_string
|
|> of_zmt_string
|
||||||
|
|
||||||
|
|
||||||
let to_string atoms =
|
let to_string atoms =
|
||||||
"
|
"
|
||||||
Nuclear Coordinates (Angstrom)
|
Nuclear Coordinates (Angstrom)
|
||||||
------------------------------
|
------------------------------
|
||||||
|
|
||||||
@ -69,27 +82,50 @@ let to_string atoms =
|
|||||||
Number X Y Z
|
Number X Y Z
|
||||||
-----------------------------------------------------------------------
|
-----------------------------------------------------------------------
|
||||||
" ^
|
" ^
|
||||||
(Array.mapi (fun i (e, coord) ->
|
(Array.mapi (fun i (e, coord) ->
|
||||||
let open Coordinate in
|
let open Coordinate in
|
||||||
let coord =
|
let coord =
|
||||||
bohr_to_angstrom coord
|
bohr_to_angstrom coord
|
||||||
in
|
in
|
||||||
Printf.sprintf " %5d %5d %5s %12.6f %12.6f %12.6f"
|
Printf.sprintf " %5d %5d %5s %12.6f %12.6f %12.6f"
|
||||||
(i+1) (Element.to_int e) (Element.to_string e)
|
(i+1) (Element.to_int e) (Element.to_string e)
|
||||||
coord.x coord.y coord.z
|
coord.x coord.y coord.z
|
||||||
) atoms
|
) atoms
|
||||||
|> Array.to_list
|
|> Array.to_list
|
||||||
|> String.concat "\n" ) ^
|
|> String.concat "\n" ) ^
|
||||||
"
|
"
|
||||||
-----------------------------------------------------------------------
|
-----------------------------------------------------------------------
|
||||||
|
|
||||||
"
|
"
|
||||||
|
|
||||||
|
|
||||||
let of_filename filename =
|
let of_filename filename =
|
||||||
of_xyz_file filename
|
of_xyz_file filename
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
let to_xyz_string t =
|
||||||
|
[ string_of_int (Array.length t) ; "" ] @
|
||||||
|
( Array.map (fun (e, coord) ->
|
||||||
|
let open Coordinate in
|
||||||
|
let coord =
|
||||||
|
bohr_to_angstrom coord
|
||||||
|
in
|
||||||
|
Printf.sprintf " %5s %12.6f %12.6f %12.6f"
|
||||||
|
(Element.to_string e) coord.x coord.y coord.z
|
||||||
|
) t
|
||||||
|
|> Array.to_list )
|
||||||
|
|> String.concat "\n"
|
||||||
|
(* Conversion:2 ends here *)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
(* | ~repulsion~ | Nuclear repulsion energy, in atomic units |
|
||||||
|
* | ~charge~ | Sum of the charges of the nuclei |
|
||||||
|
* | ~small_core~ | Number of core electrons in the small core model |
|
||||||
|
* | ~large_core~ | Number of core electrons in the large core model | *)
|
||||||
|
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Query][Query:2]] *)
|
||||||
let repulsion nuclei =
|
let repulsion nuclei =
|
||||||
let get_charge e =
|
let get_charge e =
|
||||||
Element.to_charge e
|
Element.to_charge e
|
||||||
@ -100,10 +136,10 @@ let repulsion nuclei =
|
|||||||
Array.fold_left (fun accu (e2, coord2) ->
|
Array.fold_left (fun accu (e2, coord2) ->
|
||||||
let r = Coordinate.(norm (coord1 |- coord2)) in
|
let r = Coordinate.(norm (coord1 |- coord2)) in
|
||||||
if r > 0. then
|
if r > 0. then
|
||||||
accu +. 0.5 *. (get_charge e2) *. (get_charge e1) /. r
|
accu +. 0.5 *. (get_charge e2) *. (get_charge e1) /. r
|
||||||
else accu
|
else accu
|
||||||
) 0. nuclei
|
) 0. nuclei
|
||||||
) 0. nuclei
|
) 0. nuclei
|
||||||
|
|
||||||
|
|
||||||
let charge nuclei =
|
let charge nuclei =
|
||||||
@ -112,36 +148,15 @@ let charge nuclei =
|
|||||||
|> Charge.of_int
|
|> Charge.of_int
|
||||||
|
|
||||||
|
|
||||||
let to_xyz_string t =
|
|
||||||
[ string_of_int (Array.length t) ; "" ] @
|
|
||||||
( Array.map (fun (e, coord) ->
|
|
||||||
let open Coordinate in
|
|
||||||
let coord =
|
|
||||||
bohr_to_angstrom coord
|
|
||||||
in
|
|
||||||
Printf.sprintf " %5s %12.6f %12.6f %12.6f"
|
|
||||||
(Element.to_string e) coord.x coord.y coord.z
|
|
||||||
) t
|
|
||||||
|> Array.to_list )
|
|
||||||
|> String.concat "\n"
|
|
||||||
|
|
||||||
|
|
||||||
let to_t2_string t =
|
|
||||||
[ "# nAt nEl nCore nRyd" ;
|
|
||||||
Printf.sprintf " %d %d %d 0" (Array.length t)
|
|
||||||
(Array.fold_left (+) 0 (Array.map (fun (e,_) -> Element.to_int e) t) )
|
|
||||||
(2 * Array.length t);
|
|
||||||
"# Znuc x y z" ]
|
|
||||||
@ (Array.map (fun (e, coord) ->
|
|
||||||
let open Coordinate in
|
|
||||||
Printf.sprintf " %5f %12.6f %12.6f %12.6f"
|
|
||||||
(Element.to_int e |> float_of_int) coord.x coord.y coord.z
|
|
||||||
) t
|
|
||||||
|> Array.to_list )
|
|
||||||
|> String.concat "\n"
|
|
||||||
|
|
||||||
|
|
||||||
let small_core a =
|
let small_core a =
|
||||||
Array.fold_left (fun accu (e,_) -> accu + (Element.small_core e)) 0 a
|
Array.fold_left (fun accu (e,_) -> accu + (Element.small_core e)) 0 a
|
||||||
|
|
||||||
|
|
||||||
|
let large_core a =
|
||||||
|
Array.fold_left (fun accu (e,_) -> accu + (Element.large_core e)) 0 a
|
||||||
|
(* Query:2 ends here *)
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Printers][Printers:2]] *)
|
||||||
|
let pp ppf t =
|
||||||
|
Format.fprintf ppf "@[%s@]" (to_string t)
|
||||||
|
(* Printers:2 ends here *)
|
||||||
|
|||||||
@ -1,39 +1,42 @@
|
|||||||
(** Data structure for the molecular geometry, represented as an array
|
(* Type
|
||||||
of tuples ({!Element.t}, {!Coordinate.t}).
|
*
|
||||||
*)
|
* #+NAME: types *)
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::types][types]] *)
|
||||||
open Common
|
open Common
|
||||||
|
|
||||||
type t = (Element.t * Coordinate.t) array
|
type t = (Element.t * Coordinate.t) array
|
||||||
|
(* types ends here *)
|
||||||
|
|
||||||
|
(* Conversion *)
|
||||||
|
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Conversion][Conversion:1]] *)
|
||||||
val of_xyz_string : string -> t
|
val of_xyz_string : string -> t
|
||||||
(** Create from a string, in [xyz] format. *)
|
val to_xyz_string : t -> string
|
||||||
|
val of_xyz_file : string -> t
|
||||||
val of_xyz_file : string -> t
|
|
||||||
(** Create from a file, in [xyz] format. *)
|
|
||||||
|
|
||||||
val of_zmt_string : string -> t
|
val of_zmt_string : string -> t
|
||||||
(** Create from a string, in z-matrix format. *)
|
val of_zmt_file : string -> t
|
||||||
|
|
||||||
val of_zmt_file : string -> t
|
val to_string : t -> string
|
||||||
(** Create from a file, in z-matrix format. *)
|
|
||||||
|
|
||||||
val to_string : t -> string
|
|
||||||
(** Transform to a string, for printing. *)
|
|
||||||
|
|
||||||
val of_filename : string -> t
|
val of_filename : string -> t
|
||||||
(** Detects the type of file ([xyz], z-matrix) and reads the file. *)
|
(* Conversion:1 ends here *)
|
||||||
|
|
||||||
|
(* Query *)
|
||||||
|
|
||||||
|
|
||||||
val repulsion : t -> float
|
(* [[file:../nuclei.org::*Query][Query:1]] *)
|
||||||
(** Nuclear repulsion energy, in atomic units. *)
|
val repulsion : t -> float
|
||||||
|
val charge : t -> Charge.t
|
||||||
val charge : t -> Charge.t
|
|
||||||
(** Sum of the charges of the nuclei. *)
|
|
||||||
|
|
||||||
val small_core : t -> int
|
val small_core : t -> int
|
||||||
(** Number of core electrons in the small core model. *)
|
val large_core : t -> int
|
||||||
|
(* Query:1 ends here *)
|
||||||
|
|
||||||
val to_xyz_string : t -> string
|
(* Printers *)
|
||||||
val to_t2_string : t -> string
|
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Printers][Printers:1]] *)
|
||||||
|
val pp : Format.formatter -> t -> unit
|
||||||
|
(* Printers:1 ends here *)
|
||||||
|
|||||||
@ -1,3 +1,10 @@
|
|||||||
|
(* Lexer
|
||||||
|
*
|
||||||
|
* =nuclei_lexer.mll= contains the description of the lexemes used in
|
||||||
|
* an xyz file. *)
|
||||||
|
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Lexer][Lexer:1]] *)
|
||||||
{
|
{
|
||||||
open Xyz_parser
|
open Xyz_parser
|
||||||
}
|
}
|
||||||
@ -6,8 +13,8 @@ let eol = ['\n']
|
|||||||
let white = [' ' '\t']+
|
let white = [' ' '\t']+
|
||||||
let word = [^' ' '\t' '\n']+
|
let word = [^' ' '\t' '\n']+
|
||||||
let letter = ['A'-'Z' 'a'-'z']
|
let letter = ['A'-'Z' 'a'-'z']
|
||||||
let integer = ['0'-'9']+
|
let integer = ['0'-'9']+
|
||||||
let real = '-'? (integer '.' integer | integer '.' | '.' integer) (['e' 'E'] ('+'|'-')? integer)?
|
let real = '-'? (integer '.' integer | integer '.' | '.' integer) (['e' 'E'] ('+'|'-')? integer)?
|
||||||
|
|
||||||
|
|
||||||
rule read_all = parse
|
rule read_all = parse
|
||||||
@ -38,3 +45,4 @@ rule read_all = parse
|
|||||||
done;
|
done;
|
||||||
*)
|
*)
|
||||||
}
|
}
|
||||||
|
(* Lexer:1 ends here *)
|
||||||
|
|||||||
@ -1,6 +1,10 @@
|
|||||||
(** When an [xyz] file is read by the [Xyz_parser.mly], it is converted into
|
|
||||||
an {!xyz_file} data structure. *)
|
|
||||||
|
|
||||||
|
|
||||||
|
(* When an xyz file is read by =xyz_parser.mly=, it is converted into
|
||||||
|
* an ~xyz_file~ data structure. *)
|
||||||
|
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Parser][Parser:2]] *)
|
||||||
open Common
|
open Common
|
||||||
|
|
||||||
type nucleus =
|
type nucleus =
|
||||||
@ -15,4 +19,4 @@ type xyz_file =
|
|||||||
file_title : string ;
|
file_title : string ;
|
||||||
nuclei : nucleus list ;
|
nuclei : nucleus list ;
|
||||||
}
|
}
|
||||||
|
(* Parser:2 ends here *)
|
||||||
|
|||||||
@ -1,5 +1,3 @@
|
|||||||
/* Parses nuclear coordinates in xyz format */
|
|
||||||
|
|
||||||
%{
|
%{
|
||||||
open Common
|
open Common
|
||||||
|
|
||||||
@ -84,5 +82,3 @@ atoms_list:
|
|||||||
| atoms_list SPACE INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT EOL { output_of_int $5 $7 $9 $3 :: $1 }
|
| atoms_list SPACE INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT EOL { output_of_int $5 $7 $9 $3 :: $1 }
|
||||||
| atoms_list SPACE INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT SPACE EOL { output_of_int $5 $7 $9 $3 :: $1 }
|
| atoms_list SPACE INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT SPACE EOL { output_of_int $5 $7 $9 $3 :: $1 }
|
||||||
;
|
;
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
393
particles/nuclei.org
Normal file
393
particles/nuclei.org
Normal file
@ -0,0 +1,393 @@
|
|||||||
|
#+begin_src elisp tangle: no :results none :exports none
|
||||||
|
(setq pwd (file-name-directory buffer-file-name))
|
||||||
|
(setq name (file-name-nondirectory (substring buffer-file-name 0 -4)))
|
||||||
|
(setq lib (concat pwd "lib/"))
|
||||||
|
(setq testdir (concat pwd "test/"))
|
||||||
|
(setq mli (concat lib name ".mli"))
|
||||||
|
(setq ml (concat lib name ".ml"))
|
||||||
|
(setq test-ml (concat testdir name ".ml"))
|
||||||
|
(org-babel-tangle)
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
* Nuclei
|
||||||
|
:PROPERTIES:
|
||||||
|
:header-args: :noweb yes :comments both
|
||||||
|
:END:
|
||||||
|
|
||||||
|
** Type
|
||||||
|
|
||||||
|
#+NAME: types
|
||||||
|
#+begin_src ocaml :tangle (eval mli)
|
||||||
|
open Common
|
||||||
|
|
||||||
|
type t = (Element.t * Coordinate.t) array
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval ml) :exports none
|
||||||
|
<<types>>
|
||||||
|
open Xyz_ast
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
** xyz file lexer/parser
|
||||||
|
|
||||||
|
*** Lexer
|
||||||
|
|
||||||
|
=nuclei_lexer.mll= contains the description of the lexemes used in
|
||||||
|
an xyz file.
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle lib/nuclei_lexer.mll :export none
|
||||||
|
{
|
||||||
|
open Xyz_parser
|
||||||
|
}
|
||||||
|
|
||||||
|
let eol = ['\n']
|
||||||
|
let white = [' ' '\t']+
|
||||||
|
let word = [^' ' '\t' '\n']+
|
||||||
|
let letter = ['A'-'Z' 'a'-'z']
|
||||||
|
let integer = ['0'-'9']+
|
||||||
|
let real = '-'? (integer '.' integer | integer '.' | '.' integer) (['e' 'E'] ('+'|'-')? integer)?
|
||||||
|
|
||||||
|
|
||||||
|
rule read_all = parse
|
||||||
|
| eof { EOF }
|
||||||
|
| eol { EOL }
|
||||||
|
| white as w { SPACE w }
|
||||||
|
| integer as i { INTEGER (int_of_string i) }
|
||||||
|
| real as f { FLOAT (float_of_string f) }
|
||||||
|
| word as w { WORD w }
|
||||||
|
|
||||||
|
|
||||||
|
{
|
||||||
|
(* DEBUG
|
||||||
|
let () =
|
||||||
|
let ic = open_in "h2o.xyz" in
|
||||||
|
let lexbuf = Lexing.from_channel ic in
|
||||||
|
while true do
|
||||||
|
let s =
|
||||||
|
match read_all lexbuf with
|
||||||
|
| EOL -> "EOL"
|
||||||
|
| SPACE w -> "SPACE("^w^")"
|
||||||
|
| INTEGER i -> "INTEGER("^(string_of_int i)^")"
|
||||||
|
| FLOAT f -> "FLOAT("^(string_of_float f)^")"
|
||||||
|
| WORD w -> "WORD("^w^")"
|
||||||
|
| EOF -> "EOF"
|
||||||
|
in
|
||||||
|
print_endline s
|
||||||
|
done;
|
||||||
|
*)
|
||||||
|
}
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
*** Parser
|
||||||
|
|
||||||
|
=xyz_parser.mly= parses nuclear coordinates in xyz format.
|
||||||
|
#+begin_src ocaml :tangle lib/xyz_parser.mly :export none :comments none
|
||||||
|
%{
|
||||||
|
open Common
|
||||||
|
|
||||||
|
let make_angstrom x y z =
|
||||||
|
Coordinate.(make_angstrom {
|
||||||
|
x ; y ; z
|
||||||
|
})
|
||||||
|
|
||||||
|
let output_of f x y z =
|
||||||
|
let a = make_angstrom x y z in
|
||||||
|
fun e ->
|
||||||
|
{
|
||||||
|
Xyz_ast.
|
||||||
|
element = f e;
|
||||||
|
coord = a ;
|
||||||
|
}
|
||||||
|
|
||||||
|
let output_of_string = output_of Element.of_string
|
||||||
|
let output_of_int = output_of Element.of_int
|
||||||
|
|
||||||
|
%}
|
||||||
|
|
||||||
|
%token EOL
|
||||||
|
%token <string> SPACE
|
||||||
|
%token <string> WORD
|
||||||
|
%token <int> INTEGER
|
||||||
|
%token <float> FLOAT
|
||||||
|
%token EOF
|
||||||
|
|
||||||
|
%start input
|
||||||
|
%type <Xyz_ast.xyz_file> input
|
||||||
|
|
||||||
|
%% /* Grammar rules and actions follow */
|
||||||
|
|
||||||
|
input:
|
||||||
|
| integer title atoms_xyz {
|
||||||
|
{
|
||||||
|
number_of_atoms = $1;
|
||||||
|
file_title = $2;
|
||||||
|
nuclei = $3;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
;
|
||||||
|
|
||||||
|
|
||||||
|
integer:
|
||||||
|
| INTEGER EOL { $1 }
|
||||||
|
| INTEGER SPACE EOL { $1 }
|
||||||
|
| SPACE INTEGER EOL { $2 }
|
||||||
|
| SPACE INTEGER SPACE EOL { $2 }
|
||||||
|
;
|
||||||
|
|
||||||
|
title:
|
||||||
|
| title_list EOL { $1 }
|
||||||
|
;
|
||||||
|
|
||||||
|
text:
|
||||||
|
| WORD { $1 }
|
||||||
|
| SPACE { $1 }
|
||||||
|
| FLOAT { (string_of_float $1)}
|
||||||
|
| INTEGER { (string_of_int $1)}
|
||||||
|
;
|
||||||
|
|
||||||
|
title_list:
|
||||||
|
| { "" }
|
||||||
|
| title_list text { ($1 ^ $2) }
|
||||||
|
;
|
||||||
|
|
||||||
|
atoms_xyz:
|
||||||
|
| atoms_list EOL { List.rev $1 }
|
||||||
|
| atoms_list EOF { List.rev $1 }
|
||||||
|
;
|
||||||
|
|
||||||
|
atoms_list:
|
||||||
|
| { [] }
|
||||||
|
| atoms_list WORD SPACE FLOAT SPACE FLOAT SPACE FLOAT EOL { output_of_string $4 $6 $8 $2 :: $1 }
|
||||||
|
| atoms_list WORD SPACE FLOAT SPACE FLOAT SPACE FLOAT SPACE EOL { output_of_string $4 $6 $8 $2 :: $1 }
|
||||||
|
| atoms_list INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT EOL { output_of_int $4 $6 $8 $2 :: $1 }
|
||||||
|
| atoms_list INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT SPACE EOL { output_of_int $4 $6 $8 $2 :: $1 }
|
||||||
|
| atoms_list SPACE WORD SPACE FLOAT SPACE FLOAT SPACE FLOAT EOL { output_of_string $5 $7 $9 $3 :: $1 }
|
||||||
|
| atoms_list SPACE WORD SPACE FLOAT SPACE FLOAT SPACE FLOAT SPACE EOL { output_of_string $5 $7 $9 $3 :: $1 }
|
||||||
|
| atoms_list SPACE INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT EOL { output_of_int $5 $7 $9 $3 :: $1 }
|
||||||
|
| atoms_list SPACE INTEGER SPACE FLOAT SPACE FLOAT SPACE FLOAT SPACE EOL { output_of_int $5 $7 $9 $3 :: $1 }
|
||||||
|
;
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
When an xyz file is read by =xyz_parser.mly=, it is converted into
|
||||||
|
an ~xyz_file~ data structure.
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle lib/xyz_ast.mli
|
||||||
|
open Common
|
||||||
|
|
||||||
|
type nucleus =
|
||||||
|
{
|
||||||
|
element: Element.t ;
|
||||||
|
coord : Coordinate.angstrom Coordinate.point;
|
||||||
|
}
|
||||||
|
|
||||||
|
type xyz_file =
|
||||||
|
{
|
||||||
|
number_of_atoms : int ;
|
||||||
|
file_title : string ;
|
||||||
|
nuclei : nucleus list ;
|
||||||
|
}
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
** Conversion
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval mli)
|
||||||
|
val of_xyz_string : string -> t
|
||||||
|
val to_xyz_string : t -> string
|
||||||
|
val of_xyz_file : string -> t
|
||||||
|
|
||||||
|
val of_zmt_string : string -> t
|
||||||
|
val of_zmt_file : string -> t
|
||||||
|
|
||||||
|
val to_string : t -> string
|
||||||
|
|
||||||
|
val of_filename : string -> t
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
| ~of_xyz_string~ | Create from a string, in xyz format |
|
||||||
|
| ~of_xyz_file~ | Create from a file, in xyz format |
|
||||||
|
| ~of_zmt_string~ | Create from a string, in z-matrix format |
|
||||||
|
| ~of_zmt_file~ | Create from a file, in z-matrix format |
|
||||||
|
| ~to_string~ | Transform to a string, for printing |
|
||||||
|
| ~of_filename~ | Detects the type of file (xyz, z-matrix) and reads the file |
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval ml) :exports none
|
||||||
|
let of_xyz_lexbuf lexbuf =
|
||||||
|
let data =
|
||||||
|
Xyz_parser.input Nuclei_lexer.read_all lexbuf
|
||||||
|
in
|
||||||
|
|
||||||
|
let len = List.length data.nuclei in
|
||||||
|
if len <> data.number_of_atoms then
|
||||||
|
Printf.sprintf "Error: expected %d atoms but %d read"
|
||||||
|
data.number_of_atoms len
|
||||||
|
|> failwith;
|
||||||
|
|
||||||
|
List.map (fun nucleus ->
|
||||||
|
nucleus.element, Coordinate.angstrom_to_bohr nucleus.coord
|
||||||
|
) data.nuclei
|
||||||
|
|> Array.of_list
|
||||||
|
|
||||||
|
|
||||||
|
let of_xyz_string input_string =
|
||||||
|
Lexing.from_string input_string
|
||||||
|
|> of_xyz_lexbuf
|
||||||
|
|
||||||
|
|
||||||
|
let of_xyz_file filename =
|
||||||
|
let ic = open_in filename in
|
||||||
|
let lexbuf =
|
||||||
|
Lexing.from_channel ic
|
||||||
|
in
|
||||||
|
let result =
|
||||||
|
of_xyz_lexbuf lexbuf
|
||||||
|
in
|
||||||
|
close_in ic;
|
||||||
|
result
|
||||||
|
|
||||||
|
|
||||||
|
let of_zmt_string buffer =
|
||||||
|
Zmatrix.of_string buffer
|
||||||
|
|> Zmatrix.to_xyz
|
||||||
|
|> Array.map (fun (e,x,y,z) ->
|
||||||
|
(e, Coordinate.(angstrom_to_bohr @@ make_angstrom { x ; y ; z} ))
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
let of_zmt_file filename =
|
||||||
|
let ic = open_in filename in
|
||||||
|
let rec aux accu =
|
||||||
|
try
|
||||||
|
let line = input_line ic in
|
||||||
|
aux (line::accu)
|
||||||
|
with End_of_file ->
|
||||||
|
close_in ic;
|
||||||
|
List.rev accu
|
||||||
|
|> String.concat "\n"
|
||||||
|
in aux []
|
||||||
|
|> of_zmt_string
|
||||||
|
|
||||||
|
|
||||||
|
let to_string atoms =
|
||||||
|
"
|
||||||
|
Nuclear Coordinates (Angstrom)
|
||||||
|
------------------------------
|
||||||
|
|
||||||
|
-----------------------------------------------------------------------
|
||||||
|
Center Atomic Element Coordinates (Angstroms)
|
||||||
|
Number X Y Z
|
||||||
|
-----------------------------------------------------------------------
|
||||||
|
" ^
|
||||||
|
(Array.mapi (fun i (e, coord) ->
|
||||||
|
let open Coordinate in
|
||||||
|
let coord =
|
||||||
|
bohr_to_angstrom coord
|
||||||
|
in
|
||||||
|
Printf.sprintf " %5d %5d %5s %12.6f %12.6f %12.6f"
|
||||||
|
(i+1) (Element.to_int e) (Element.to_string e)
|
||||||
|
coord.x coord.y coord.z
|
||||||
|
) atoms
|
||||||
|
|> Array.to_list
|
||||||
|
|> String.concat "\n" ) ^
|
||||||
|
"
|
||||||
|
-----------------------------------------------------------------------
|
||||||
|
|
||||||
|
"
|
||||||
|
|
||||||
|
|
||||||
|
let of_filename filename =
|
||||||
|
of_xyz_file filename
|
||||||
|
|
||||||
|
|
||||||
|
let to_xyz_string t =
|
||||||
|
[ string_of_int (Array.length t) ; "" ] @
|
||||||
|
( Array.map (fun (e, coord) ->
|
||||||
|
let open Coordinate in
|
||||||
|
let coord =
|
||||||
|
bohr_to_angstrom coord
|
||||||
|
in
|
||||||
|
Printf.sprintf " %5s %12.6f %12.6f %12.6f"
|
||||||
|
(Element.to_string e) coord.x coord.y coord.z
|
||||||
|
) t
|
||||||
|
|> Array.to_list )
|
||||||
|
|> String.concat "\n"
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
** Query
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval mli)
|
||||||
|
val repulsion : t -> float
|
||||||
|
val charge : t -> Charge.t
|
||||||
|
val small_core : t -> int
|
||||||
|
val large_core : t -> int
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
| ~repulsion~ | Nuclear repulsion energy, in atomic units |
|
||||||
|
| ~charge~ | Sum of the charges of the nuclei |
|
||||||
|
| ~small_core~ | Number of core electrons in the small core model |
|
||||||
|
| ~large_core~ | Number of core electrons in the large core model |
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval ml) :exports none
|
||||||
|
let repulsion nuclei =
|
||||||
|
let get_charge e =
|
||||||
|
Element.to_charge e
|
||||||
|
|> Charge.to_float
|
||||||
|
in
|
||||||
|
Array.fold_left ( fun accu (e1, coord1) ->
|
||||||
|
accu +.
|
||||||
|
Array.fold_left (fun accu (e2, coord2) ->
|
||||||
|
let r = Coordinate.(norm (coord1 |- coord2)) in
|
||||||
|
if r > 0. then
|
||||||
|
accu +. 0.5 *. (get_charge e2) *. (get_charge e1) /. r
|
||||||
|
else accu
|
||||||
|
) 0. nuclei
|
||||||
|
) 0. nuclei
|
||||||
|
|
||||||
|
|
||||||
|
let charge nuclei =
|
||||||
|
Array.fold_left (fun accu (e, _) -> accu + Charge.to_int (Element.to_charge e) )
|
||||||
|
0 nuclei
|
||||||
|
|> Charge.of_int
|
||||||
|
|
||||||
|
|
||||||
|
let small_core a =
|
||||||
|
Array.fold_left (fun accu (e,_) -> accu + (Element.small_core e)) 0 a
|
||||||
|
|
||||||
|
|
||||||
|
let large_core a =
|
||||||
|
Array.fold_left (fun accu (e,_) -> accu + (Element.large_core e)) 0 a
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
** Printers
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval mli)
|
||||||
|
val pp : Format.formatter -> t -> unit
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval ml) :exports none
|
||||||
|
let pp ppf t =
|
||||||
|
Format.fprintf ppf "@[%s@]" (to_string t)
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
** Tests
|
||||||
|
|
||||||
|
#+begin_src ocaml :tangle (eval test-ml) :exports none
|
||||||
|
open Common
|
||||||
|
open Particles
|
||||||
|
open Alcotest
|
||||||
|
|
||||||
|
let wd = Common.Qcaml.root ^ Filename.dir_sep ^ "test"
|
||||||
|
|
||||||
|
let test_xyz molecule length repulsion charge core =
|
||||||
|
let xyz = Nuclei.of_xyz_file (wd^Filename.dir_sep^molecule^".xyz") in
|
||||||
|
check int "length" length (Array.length xyz);
|
||||||
|
check (float 1.e-4) "repulsion" repulsion (Nuclei.repulsion xyz);
|
||||||
|
check int "charge" charge (Charge.to_int @@ Nuclei.charge xyz);
|
||||||
|
check int "small_core" core (Nuclei.small_core xyz);
|
||||||
|
()
|
||||||
|
|
||||||
|
let tests = [
|
||||||
|
"caffeine", `Quick, (fun () -> test_xyz "caffeine" 24 917.0684 102 28);
|
||||||
|
"water", `Quick, (fun () -> test_xyz "water" 3 9.19497 10 2);
|
||||||
|
]
|
||||||
|
#+end_src
|
||||||
|
|
||||||
@ -1,3 +1,7 @@
|
|||||||
|
(* Tests *)
|
||||||
|
|
||||||
|
|
||||||
|
(* [[file:../nuclei.org::*Tests][Tests:1]] *)
|
||||||
open Common
|
open Common
|
||||||
open Particles
|
open Particles
|
||||||
open Alcotest
|
open Alcotest
|
||||||
@ -16,3 +20,4 @@ let tests = [
|
|||||||
"caffeine", `Quick, (fun () -> test_xyz "caffeine" 24 917.0684 102 28);
|
"caffeine", `Quick, (fun () -> test_xyz "caffeine" 24 917.0684 102 28);
|
||||||
"water", `Quick, (fun () -> test_xyz "water" 3 9.19497 10 2);
|
"water", `Quick, (fun () -> test_xyz "water" 3 9.19497 10 2);
|
||||||
]
|
]
|
||||||
|
(* Tests:1 ends here *)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user