pdorg_site/talks/cant-c-a-thing/code/wrapped_ocaml.ml

let matrix_mul x y =
  let x0 = Array.length x
  and y0 = Array.length y in
  let y1 = if y0 = 0 then 0 else Array.length y.(0) in
  let z = Array.make_matrix x0 y1 0 in
  for i = 0 to x0-1 do
    for j = 0 to y1-1 do
      for k = 0 to y0-1 do
        z.(i).(j) <- z.(i).(j) + x.(i).(k) * y.(k).(j)
      done
    done
  done;
  z

let new_matrix width height =
  Array.make_matrix width height 0

let matrix_assign m x y item =
  m.(x).(y) <- item

let matrix_retrieve m x y =
  m.(x).(y)


let _ = Callback.register "ocaml_new_matrix" new_matrix

let _ = Callback.register "ocaml_matrix_assign" matrix_assign

let _ = Callback.register "ocaml_matrix_retrieve" matrix_retrieve

let _ = Callback.register "ocaml_matrix_mul" matrix_mul