open Core;;
open Qptypes;;

type t = (Symmetry.Xyz.t * Gto.t * Nucl_number.t ) list [@@deriving sexp]

let of_basis b =
  let rec do_work accu = function
    | [] -> accu
    | (g,n)::tail ->
        begin
          let new_accu = 
            Symmetry.Xyz.of_symmetry g.Gto.sym 
            |> List.rev_map ~f:(fun x-> (x,g,n)) 
          in
          do_work (new_accu@accu) tail
        end
  in
  do_work [] b
  |> List.rev
;;

let to_basis b =
  let rec do_work accu = function
  | [] -> List.rev accu
  | (s,g,n)::tail -> 
    let first_sym = 
      Symmetry.Xyz.of_symmetry g.Gto.sym
      |> List.hd_exn
    in
    let new_accu = 
      if ( s = first_sym ) then
        (g,n)::accu
      else
        accu
    in
    do_work new_accu tail
  in
  do_work [] b 
;;

let to_string b =
  let middle = List.map ~f:(fun (x,y,z) -> 
     "( "^((Int.to_string (Nucl_number.to_int z)))^", "^
     (Symmetry.Xyz.to_string x)^", "^(Gto.to_string y)
     ^" )"
  ) b
  |> String.concat ~sep:",\n"
  in "("^middle^")"
;;

include To_md5;;
let to_md5 = to_md5 sexp_of_t
;;