(*  * OWL - an OCaml numerical library for scientific computing  * Copyright (c) 2016 Liang Wang <liang.wang@cl.cam.ac.uk>  *)

(** Define the types shared by various modules *)

open Bigarray open Ctypes

(** configure the logger *)

let _ = Log.color_on (); Log.(set_log_level INFO) (* struct definition for real dense matrix *) module Dense_real = struct   type int_array = (int64, int64_elt, c_layout) Array1.t   type elt_array = (float, float64_elt, c_layout) Array2.t   type mblk_struct   let mblk_struct : mblk_struct structure typ = structure "gsl_block_struct"     let msize = field mblk_struct "size" int64_t     let mdata = field mblk_struct "data" (ptr double)   let () = seal mblk_struct   

(** structure definition for vector, refer to gsl_vector_double.h *)

  type vec_struct   let vec_struct : vec_struct structure typ = structure "gsl_vector"     let vsize  = field vec_struct "size" int64_t     let stride = field vec_struct "stride" int64_t     let vdata  = field vec_struct "data" (ptr double)     let vblock = field vec_struct "block" (ptr mblk_struct)     let vowner = field vec_struct "owner" int64_t   let () = seal vec_struct   

(** structure definition for dense matrix, refer to gsl_matrix_double.h *)

  type mat_struct   let mat_struct : mat_struct structure typ = structure "gsl_matrix"     let size1 = field mat_struct "size1" int64_t     let size2 = field mat_struct "size2" int64_t     let tda   = field mat_struct "tda" int64_t     let data  = field mat_struct "data" (ptr double)     let block = field mat_struct "block" (ptr mblk_struct)     let owner = field mat_struct "owner" int64_t   let () = seal mat_struct   

(** define the vector record *)

  type vec_record = {     mutable vsize  : int;            (* size of a vector *)     mutable stride : int;            (* stride of a vector *)     mutable vdata  : elt_array;      (* actual data of a vector *)     mutable vptr   : vec_struct Ctypes_static.structure Ctypes_static.ptr;     (* pointer to a vector's memory *)   } end (* struct definition for complex dense matrix *) module Dense_complex = struct   type int_array = (int64, int64_elt, c_layout) Array1.t   type elt_array = (Complex.t, complex64_elt, c_layout) Array2.t   type mblk_struct   let mblk_struct : mblk_struct structure typ = structure "gsl_block_complex_struct"     let msize = field mblk_struct "size" int64_t     let mdata = field mblk_struct "data" (ptr complex64)   let () = seal mblk_struct   

(** structure definition for vector, refer to gsl_vector_double.h *)

  type vec_struct   let vec_struct : vec_struct structure typ = structure "gsl_vector_complex"     let vsize  = field vec_struct "size" int64_t     let stride = field vec_struct "stride" int64_t     let vdata  = field vec_struct "data" (ptr complex64)     let vblock = field vec_struct "block" (ptr mblk_struct)     let vowner = field vec_struct "owner" int64_t   let () = seal vec_struct   

(** structure definition for dense matrix, refer to gsl_matrix_double.h *)

  type mat_struct   let mat_struct : mat_struct structure typ = structure "gsl_matrix_complex"     let size1 = field mat_struct "size1" int64_t     let size2 = field mat_struct "size2" int64_t     let tda   = field mat_struct "tda" int64_t     let data  = field mat_struct "data" (ptr complex64)     let block = field mat_struct "block" (ptr mblk_struct)     let owner = field mat_struct "owner" int64_t   let () = seal mat_struct   

(** define the vector record *)

  type vec_record = {     mutable vsize  : int;            (* size of a vector *)     mutable stride : int;            (* stride of a vector *)     mutable vdata  : elt_array;      (* actual data of a vector *)     mutable vptr   : vec_struct Ctypes_static.structure Ctypes_static.ptr;     (* pointer to a vector's memory *)   } end (* struct definition for real sparse matrix *) module Sparse_real = struct   type int_array = (int64, int64_elt, c_layout) Array1.t   type elt_array = (float, float64_elt, c_layout) Array1.t   type spmat_struct   let spmat_struct : spmat_struct structure typ = structure "gsl_spmatrix"     let sp_size1 = field spmat_struct "size1" int64_t     let sp_size2 = field spmat_struct "size2" int64_t     let sp_i     = field spmat_struct "i" (ptr int64_t)     let sp_data  = field spmat_struct "data" (ptr double)     let sp_p     = field spmat_struct "p" (ptr int64_t)     let sp_nzmax = field spmat_struct "nzmax" int64_t     let sp_nz    = field spmat_struct "nz" int64_t     let sp_tree  = field spmat_struct "tree_data" (ptr void)     let sp_work  = field spmat_struct "work" (ptr void)     let sp_type  = field spmat_struct "sptype" int64_t   let () = seal spmat_struct   

(** record definition for sparse matrix *)

  type spmat_record = {     mutable m   : int;           (* number of rows *)     mutable n   : int;           (* number of columns *)     mutable i   : int_array;     (* i index, meaning depends on the matrix format *)     mutable d   : elt_array;     (* where data actually stored *)     mutable p   : int_array;     (* p index, meaning depends on the matrix format *)     mutable nz  : int;           (* total number of non-zero elements. BE CAREFUL! *)     (* tree missing *)     (* work missing *)     mutable typ : int;           (* sparse matrix format, 0:triplet; 1:CCS; 2:CRS *)     mutable ptr : spmat_struct Ctypes_static.structure Ctypes_static.ptr;     (* pointer to the sparse metrix *)   } end (* struct definition for complex sparse matrix *) module Sparse_complex = struct   type int_array = int array   type elt_array = (Complex.t, complex64_elt, c_layout) Array1.t   

(** record definition for sparse matrix *)

  type spmat_record = {     mutable m   : int;           (* number of rows *)     mutable n   : int;           (* number of columns *)     mutable i   : int_array;     (* i index, meaning depends on the matrix format *)     mutable d   : elt_array;     (* where data actually stored *)     mutable p   : int_array;     (* p index, meaning depends on the matrix format *)     mutable nz  : int;           (* total number of non-zero elements *)     mutable typ : int;           (* sparse matrix format, 0:triplet; 1:CCS; 2:CRS *)     mutable h   : (int, int) Hashtbl.t   } end (* ends here *)