mirror of
https://gitlab.com/scemama/qmcchem.git
synced 2024-11-07 06:33:38 +01:00
127 lines
4.3 KiB
OCaml
127 lines
4.3 KiB
OCaml
open Core.Std
|
|
open Qptypes
|
|
|
|
type t =
|
|
| Property of Block.t
|
|
| Walkers of Compute_node.t * Pid.t * (float array) array
|
|
| Register of Compute_node.t * Pid.t
|
|
| Unregister of Compute_node.t * Pid.t
|
|
| Test
|
|
| GetWalkers of Strictly_positive_int.t
|
|
| Ezfio of string
|
|
|
|
|
|
let create = function
|
|
| [ "cpu" ; c ; pid ; b ; "1" ; v ] ->
|
|
let open Block in
|
|
Property
|
|
{ property = Property.Cpu;
|
|
value = Sample.of_float (Float.of_string v) ;
|
|
weight = Weight.of_float 1.;
|
|
compute_node = Compute_node.of_string c;
|
|
pid = Pid.of_string pid;
|
|
block_id = Block_id.of_int (Int.of_string b);
|
|
}
|
|
| [ "accep" ; c ; pid ; b ; "1" ; v ] ->
|
|
let open Block in
|
|
Property
|
|
{ property = Property.Accep;
|
|
value = Sample.of_float (Float.of_string v) ;
|
|
weight = Weight.of_float 1.;
|
|
compute_node = Compute_node.of_string c;
|
|
pid = Pid.of_string pid;
|
|
block_id = Block_id.of_int (Int.of_string b);
|
|
}
|
|
| [ prop ; c ; pid ; b ; w ; v ] ->
|
|
let open Block in
|
|
Property
|
|
{ property = Property.of_string prop;
|
|
value = Sample.of_float (Float.of_string v);
|
|
weight = Weight.of_float (Float.of_string w);
|
|
compute_node = Compute_node.of_string c;
|
|
pid = Pid.of_string pid;
|
|
block_id = Block_id.of_int (Int.of_string b);
|
|
}
|
|
| "elec_coord" :: c :: pid :: _ :: n ::walkers ->
|
|
begin
|
|
let walk_num =
|
|
Lazy.force Qputils.walk_num
|
|
and elec_num =
|
|
Lazy.force Qputils.elec_num
|
|
and n =
|
|
Int.of_string n
|
|
in
|
|
assert (n = List.length walkers);
|
|
assert (n = walk_num*(elec_num+1)*3);
|
|
let rec build_walker accu = function
|
|
| (0,tail) ->
|
|
let result =
|
|
List.rev accu
|
|
|> List.map ~f:Float.of_string
|
|
|> Array.of_list
|
|
in
|
|
(result, tail)
|
|
| (n,head::tail) ->
|
|
build_walker (head::accu) (n-1, tail)
|
|
| _ -> failwith "Bad walkers"
|
|
in
|
|
let rec build accu = function
|
|
| [] -> Array.of_list accu
|
|
| w ->
|
|
let (result, tail) =
|
|
build_walker [] (3*elec_num+3, w)
|
|
in
|
|
build (result::accu) tail
|
|
in
|
|
Walkers (Compute_node.of_string c, Pid.of_string pid, build [] walkers)
|
|
end
|
|
| [ "get_walkers" ; n ] -> GetWalkers (n |> Int.of_string |> Strictly_positive_int.of_int)
|
|
| [ "register" ; c ; pid ] -> Register (Compute_node.of_string c, Pid.of_string pid)
|
|
| [ "unregister" ; c ; pid ] -> Unregister (Compute_node.of_string c, Pid.of_string pid)
|
|
| [ "Test" ] -> Test
|
|
| [ "Ezfio" ; ezfio_msg ] -> Ezfio ezfio_msg
|
|
| prop :: c :: pid :: b :: d :: w :: l ->
|
|
let property =
|
|
Property.of_string prop
|
|
in
|
|
begin
|
|
assert (not (Property.is_scalar property));
|
|
let a =
|
|
Array.of_list l
|
|
|> Array.map ~f:Float.of_string
|
|
and dim =
|
|
Int.of_string d
|
|
in
|
|
assert (Array.length a = dim);
|
|
let open Block in
|
|
Property
|
|
{ property = property ;
|
|
value = Sample.of_float_array ~dim a;
|
|
weight = Weight.of_float (Float.of_string w);
|
|
compute_node = Compute_node.of_string c;
|
|
pid = Pid.of_string pid;
|
|
block_id = Block_id.of_int (Int.of_string b);
|
|
}
|
|
end
|
|
| l ->
|
|
begin
|
|
List.iter ~f:(Printf.printf ":%s:") l;
|
|
failwith "Message not understood"
|
|
end
|
|
|
|
|
|
let to_string = function
|
|
| Property b -> "Property : "^(Block.to_string b)
|
|
| Walkers (h,p,w) -> Printf.sprintf "Walkers : %s %s : %d walkers"
|
|
(Compute_node.to_string h) (Pid.to_string p)
|
|
(Array.length w)
|
|
| GetWalkers n -> Printf.sprintf "GetWalkers %d" (Strictly_positive_int.to_int n)
|
|
| Register (h,p) -> Printf.sprintf "Register : %s %s"
|
|
(Compute_node.to_string h) (Pid.to_string p)
|
|
| Unregister (h,p) -> Printf.sprintf "Unregister : %s %s"
|
|
(Compute_node.to_string h) (Pid.to_string p)
|
|
| Test -> "Test"
|
|
| Ezfio msg -> "Ezfio "^msg
|
|
|
|
|