ocaml/otherlibs/graph/graphics.ml

229 lines
7.2 KiB
OCaml

(***********************************************************************)
(* *)
(* Objective Caml *)
(* *)
(* Xavier Leroy, projet Cristal, INRIA Rocquencourt *)
(* *)
(* Copyright 1996 Institut National de Recherche en Informatique et *)
(* en Automatique. All rights reserved. This file is distributed *)
(* under the terms of the GNU Library General Public License, with *)
(* the special exception on linking described in file ../../LICENSE. *)
(* *)
(***********************************************************************)
(* $Id$ *)
exception Graphic_failure of string
(* Initializations *)
let _ =
Callback.register_exception "Graphics.Graphic_failure" (Graphic_failure "")
external raw_open_graph: string -> unit = "caml_gr_open_graph"
external raw_close_graph: unit -> unit = "caml_gr_close_graph"
external sigio_signal: unit -> int = "caml_gr_sigio_signal"
external sigio_handler: int -> unit = "caml_gr_sigio_handler"
let unix_open_graph arg =
Sys.set_signal (sigio_signal()) (Sys.Signal_handle sigio_handler);
raw_open_graph arg
let unix_close_graph () =
Sys.set_signal (sigio_signal()) Sys.Signal_ignore;
raw_close_graph ()
let (open_graph, close_graph) =
match Sys.os_type with
| "Unix" | "Cygwin" -> (unix_open_graph, unix_close_graph)
| "Win32" -> (raw_open_graph, raw_close_graph)
| "MacOS" -> (raw_open_graph, raw_close_graph)
| _ -> invalid_arg ("Graphics: unknown OS type: " ^ Sys.os_type)
external set_window_title : string -> unit = "caml_gr_set_window_title"
external clear_graph : unit -> unit = "caml_gr_clear_graph"
external size_x : unit -> int = "caml_gr_size_x"
external size_y : unit -> int = "caml_gr_size_y"
(* Double-buffering *)
external display_mode : bool -> unit = "caml_gr_display_mode"
external remember_mode : bool -> unit = "caml_gr_remember_mode"
external synchronize : unit -> unit = "caml_gr_synchronize"
let auto_synchronize = function
| true -> display_mode true; remember_mode true; synchronize ()
| false -> display_mode false; remember_mode true
;;
(* Colors *)
type color = int
let rgb r g b = (r lsl 16) + (g lsl 8) + b
external set_color : color -> unit = "caml_gr_set_color"
let black = 0x000000
and white = 0xFFFFFF
and red = 0xFF0000
and green = 0x00FF00
and blue = 0x0000FF
and yellow = 0xFFFF00
and cyan = 0x00FFFF
and magenta = 0xFF00FF
let background = white
and foreground = black
(* Drawing *)
external plot : int -> int -> unit = "caml_gr_plot"
let plots points =
for i = 0 to Array.length points - 1 do
let (x, y) = points.(i) in
plot x y;
done
;;
external point_color : int -> int -> color = "caml_gr_point_color"
external moveto : int -> int -> unit = "caml_gr_moveto"
external current_x : unit -> int = "caml_gr_current_x"
external current_y : unit -> int = "caml_gr_current_y"
let current_point () = current_x (), current_y ()
external lineto : int -> int -> unit = "caml_gr_lineto"
let rlineto x y = lineto (current_x () + x) (current_y () + y)
let rmoveto x y = moveto (current_x () + x) (current_y () + y)
external draw_rect : int -> int -> int -> int -> unit = "caml_gr_draw_rect"
let draw_poly, draw_poly_line =
let dodraw close_flag points =
if Array.length points > 0 then begin
let (savex, savey) = current_point () in
moveto (fst points.(0)) (snd points.(0));
for i = 1 to Array.length points - 1 do
let (x, y) = points.(i) in
lineto x y;
done;
if close_flag then lineto (fst points.(0)) (snd points.(0));
moveto savex savey;
end;
in dodraw true, dodraw false
;;
let draw_segments segs =
let (savex, savey) = current_point () in
for i = 0 to Array.length segs - 1 do
let (x1, y1, x2, y2) = segs.(i) in
moveto x1 y1;
lineto x2 y2;
done;
moveto savex savey;
;;
external draw_arc : int -> int -> int -> int -> int -> int -> unit
= "caml_gr_draw_arc" "caml_gr_draw_arc_nat"
let draw_ellipse x y rx ry = draw_arc x y rx ry 0 360
let draw_circle x y r = draw_arc x y r r 0 360
external set_line_width : int -> unit = "caml_gr_set_line_width"
external fill_rect : int -> int -> int -> int -> unit = "caml_gr_fill_rect"
external fill_poly : (int * int) array -> unit = "caml_gr_fill_poly"
external fill_arc : int -> int -> int -> int -> int -> int -> unit
= "caml_gr_fill_arc" "caml_gr_fill_arc_nat"
let fill_ellipse x y rx ry = fill_arc x y rx ry 0 360
let fill_circle x y r = fill_arc x y r r 0 360
(* Text *)
external draw_char : char -> unit = "caml_gr_draw_char"
external draw_string : string -> unit = "caml_gr_draw_string"
external set_font : string -> unit = "caml_gr_set_font"
external set_text_size : int -> unit = "caml_gr_set_text_size"
external text_size : string -> int * int = "caml_gr_text_size"
(* Images *)
type image
let transp = -1
external make_image : color array array -> image = "caml_gr_make_image"
external dump_image : image -> color array array = "caml_gr_dump_image"
external draw_image : image -> int -> int -> unit = "caml_gr_draw_image"
external create_image : int -> int -> image = "caml_gr_create_image"
external blit_image : image -> int -> int -> unit = "caml_gr_blit_image"
let get_image x y w h =
let image = create_image w h in
blit_image image x y;
image
(* Events *)
type status =
{ mouse_x : int;
mouse_y : int;
button : bool;
keypressed : bool;
key : char }
type event =
Button_down
| Button_up
| Key_pressed
| Mouse_motion
| Poll
external wait_next_event : event list -> status = "caml_gr_wait_event"
let mouse_pos () =
let e = wait_next_event [Poll] in (e.mouse_x, e.mouse_y)
let button_down () =
let e = wait_next_event [Poll] in e.button
let read_key () =
let e = wait_next_event [Key_pressed] in e.key
let key_pressed () =
let e = wait_next_event [Poll] in e.keypressed
(*** Sound *)
external sound : int -> int -> unit = "caml_gr_sound"
(* Splines *)
let add (x1, y1) (x2, y2) = (x1 +. x2, y1 +. y2)
and sub (x1, y1) (x2, y2) = (x1 -. x2, y1 -. y2)
and middle (x1, y1) (x2, y2) = ((x1 +. x2) /. 2.0, (y1 +. y2) /. 2.0)
and area (x1, y1) (x2, y2) = abs_float (x1 *. y2 -. x2 *. y1)
and norm (x1, y1) = sqrt (x1 *. x1 +. y1 *. y1);;
let test a b c d =
let v = sub d a in
let s = norm v in
area v (sub a b) <= s && area v (sub a c) <= s;;
let spline a b c d =
let rec spl accu a b c d =
if test a b c d then d :: accu else
let a' = middle a b
and o = middle b c in
let b' = middle a' o
and d' = middle c d in
let c' = middle o d' in
let i = middle b' c' in
spl (spl accu a a' b' i) i c' d' d in
spl [a] a b c d;;
let curveto b c (x, y as d) =
let float_point (x, y) = (float_of_int x, float_of_int y) in
let round f = int_of_float (f +. 0.5) in
let int_point (x, y) = (round x, round y) in
let points =
spline
(float_point (current_point ()))
(float_point b) (float_point c) (float_point d) in
draw_poly_line
(Array.of_list (List.map int_point points));
moveto x y;;