X-Git-Url: https://git.xandkar.net/?p=cellular-automata.git;a=blobdiff_plain;f=polymorphic-life%2F001%2Fsrc%2Fpolymorphic_life.ml;h=a667bc4b4a23c875d15965f52b41d11850d8e60a;hp=e3c233dfb77da370f8d394d8ef37f572f71c028d;hb=4d49c95e96b6a688ecbc54bea64b810c9611b400;hpb=8c93b722cd62dccd61959609c96ce57e53dbe3e3 diff --git a/polymorphic-life/001/src/polymorphic_life.ml b/polymorphic-life/001/src/polymorphic_life.ml index e3c233d..a667bc4 100644 --- a/polymorphic-life/001/src/polymorphic_life.ml +++ b/polymorphic-life/001/src/polymorphic_life.ml @@ -1,8 +1,57 @@ open Core.Std +module type MATRIX = sig + type 'a t + + val create : rows:int -> cols:int -> data:'a -> 'a t + + val get : 'a t -> row:int -> col:int -> 'a + + val set : 'a t -> row:int -> col:int -> data:'a -> unit + + val map : 'a t -> f:(row:int -> col:int -> data:'a -> 'b) -> 'b t + + val iter : 'a t -> f:(row:int -> col:int -> data:'a -> unit) -> unit +end + +module Matrix : MATRIX = struct + type 'a t = 'a array array + + let create ~rows ~cols ~data = + Array.make_matrix ~dimx:rows ~dimy:cols data + + let iter t ~f = + Array.iteri t ~f:( + fun row cols -> + Array.iteri cols ~f:( + fun col data -> + f ~row ~col ~data + ) + ) + + let map t ~f = + Array.mapi t ~f:( + fun row cols -> + Array.mapi cols ~f:( + fun col data -> + f ~row ~col ~data + ) + ) + + let get t ~row ~col = + t.(row).(col) + + let set t ~row ~col ~data = + t.(row).(col) <- data +end + + let main () = - printf "Hi!\n" + let pool = Matrix.create ~rows:5 ~cols:5 ~data:() in + Matrix.iter pool ~f:( + fun ~row ~col ~data:() -> printf "R: %d, K: %d\n" row col + ) let () = main ()