Source file combi.ml

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
open Bigarray

type t = {
  n    : int;
  k    : int;
  data : (int, int_elt, c_layout) Bigarray.Array1.t
}

external _init_first : t -> unit = "ml_gsl_combination_init_first"
external _init_last  : t -> unit = "ml_gsl_combination_init_last"

let make n k =
  let c = { n; k; data = Array1.create int c_layout k } in begin
    _init_first c;
    c
  end

let to_array { data; _ } =
  let len = Array1.dim data in
  Array.init len (Array1.get data)

external prev : t -> unit = "ml_gsl_combination_prev"
external next : t -> unit = "ml_gsl_combination_next"

external _valid : t -> bool = "ml_gsl_combination_valid"

let valid c =
  try _valid c
  with Error.Gsl_exn (Error.FAILURE, _) -> false