7d552e4595
git-svn-id: https://swig.svn.sourceforge.net/svnroot/swig/trunk/SWIG@6646 626c5289-ae23-0410-ae9c-e8d60b6d4f22
87 lines
2.3 KiB
OCaml
87 lines
2.3 KiB
OCaml
(* Plot a 3D Function *)
|
|
|
|
(* Use the wrapped GIFPlot library *)
|
|
open Swig
|
|
open Gifplot
|
|
open Int32
|
|
|
|
(* Here is the function to plot *)
|
|
let func x y =
|
|
5.0 *.
|
|
(cos (2.0 *. (sqrt (x *. x) +. (y *. y)))) *.
|
|
(exp (-0.3 *. (sqrt (x *. x) +. (y *. y))))
|
|
|
|
(* Here are some plotting parameters *)
|
|
|
|
let xmin = -5.0
|
|
let xmax = 5.0
|
|
let ymin = -5.0
|
|
let ymax = 5.0
|
|
let zmin = -5.0
|
|
let zmax = 5.0
|
|
|
|
(* Grid resolution *)
|
|
let nxpoints = 60
|
|
let nypoints = 60
|
|
|
|
let cmap = _new_ColorMap (C_string "cmap")
|
|
let frame = _new_FrameBuffer (C_list [ C_int 500 ;
|
|
C_int 500 ])
|
|
let _ = _FrameBuffer_clear (C_list [ frame ; _BLACK C_void ])
|
|
|
|
let p2 = _new_Plot3D (C_list [ frame ;
|
|
C_float xmin ; C_float ymin ; C_float zmin ;
|
|
C_float xmax ; C_float ymax ; C_float zmax ])
|
|
let _ = _Plot3D_lookat (C_list [ p2 ; C_float (2.0 *. (zmax -. zmin)) ])
|
|
let _ = _Plot3D_autoperspective (C_list [ p2 ; C_float 40.0 ])
|
|
let _ = _Plot3D_rotu (C_list [ p2 ; C_float 60.0 ])
|
|
let _ = _Plot3D_rotr (C_list [ p2 ; C_float 30.0 ])
|
|
let _ = _Plot3D_rotd (C_list [ p2 ; C_float 10.0 ])
|
|
|
|
let drawsolid () =
|
|
begin
|
|
_Plot3D_clear (C_list [ p2 ; _BLACK C_void ]) ;
|
|
_Plot3D_start p2 ;
|
|
let dx = ((xmax -. xmin) /. (float_of_int nxpoints))
|
|
and dy = ((ymax -. ymin) /. (float_of_int nypoints))
|
|
and cscale = (240.0 /. (zmax -. zmin)) in
|
|
let rec x_loop x i =
|
|
if i < nxpoints then
|
|
begin
|
|
let rec y_loop y j =
|
|
begin
|
|
if j < nypoints then
|
|
let z1 = func x y
|
|
and z2 = func (x +. dx) y
|
|
and z3 = func (x +. dx) (y +. dy)
|
|
and z4 = func x (y +. dy) in
|
|
let c1 = cscale *. (z1 -. zmin)
|
|
and c2 = cscale *. (z2 -. zmin)
|
|
and c3 = cscale *. (z3 -. zmin)
|
|
and c4 = cscale *. (z4 -. zmin) in
|
|
let cc = (c1 +. c2 +. c3 +. c4) /. 4.0 in
|
|
let c = (max (min (int_of_float cc) 239) 0) in
|
|
_Plot3D_solidquad
|
|
(C_list (p2 ::
|
|
(List.map
|
|
(fun x -> C_float x)
|
|
[ x ; y ; z1 ;
|
|
(x +. dx) ; y ; z2 ;
|
|
(x +. dx) ; (y +. dy) ; z3 ;
|
|
x ; (y +. dx) ; z4 ;
|
|
(float_of_int (c + 16)) ]))) ;
|
|
y_loop (y +. dy) (j + 1)
|
|
end in
|
|
begin
|
|
y_loop ymin 0 ;
|
|
x_loop (x +. dx) (i + 1)
|
|
end
|
|
end in
|
|
x_loop xmin 0
|
|
end
|
|
|
|
let _ = print_endline "Making a nice 3D plot..."
|
|
let _ = drawsolid ()
|
|
|
|
let _ = _FrameBuffer_writeGIF (C_list [ frame ; cmap ; C_string "image.gif" ])
|
|
let _ = print_endline "Write image.gif"
|