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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
(* Planets: A simple 2-d celestial simulator
Copyright (C) 2001 Yaron M. Minsky
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*)
(* When run in a given directory, converts all the uni.[0-9a-z] files from
* the old to the new format.
*)
open Printf
open Tk
type body = { pos: float * float;
velocity: float * float;
radius: float;
color: color;
mass: float;
id: int;
}
type state = { mutable zoom: float;
mutable center: float * float;
mutable delta: float;
mutable bodies: body list;
}
let state = { zoom = 0.0;
center = (0.0, 0.0);
delta = 0.0;
bodies = [];
}
let load_universe filename =
try
let in_c = open_in_bin filename in
let nstate = ((Marshal.from_channel in_c):state) in
state.zoom <- nstate.zoom;
state.center <- nstate.center;
state.delta <- nstate.delta;
state.bodies <- nstate.bodies;
with
Sys_error str ->
print_string ("error opening file: " ^ str);
print_newline ();
()
(******************************************************************)
(******************************************************************)
let string_of_float x = sprintf "%.20e" x
let string_of_int x = sprintf "%d" x
let string_of_pair pair =
sprintf "(%s, %s)" (string_of_float (fst pair)) (string_of_float (snd pair))
let string_of_color color = match color with
`Color string -> string
| `Black -> "black"
| `Blue -> "blue"
| `Red -> "red"
| `White -> "white"
| `Green -> "green"
| `Yellow -> "yellow"
let write_body out_c body =
let indent = " " in
fprintf out_c "\nbody\n";
fprintf out_c "%spos %s\n" indent (string_of_pair body.pos);
fprintf out_c "%svelocity %s\n" indent (string_of_pair body.velocity);
fprintf out_c "%sradius %s\n" indent (string_of_float body.radius);
fprintf out_c "%smass %s\n" indent (string_of_float body.mass);
fprintf out_c "%scolor \"%s\"\n" indent (string_of_color body.color);
fprintf out_c "%sid %s\n" indent (string_of_int body.id)
let write_state out_c =
fprintf out_c "zoom %s\n" (string_of_float state.zoom);
fprintf out_c "center %s\n" (string_of_pair state.center);
fprintf out_c "delta %s\n" (string_of_float state.delta);
List.iter ~f:(write_body out_c) state.bodies;
close_out out_c
let write_state_file filename = write_state (open_out filename)
let get_files ~f dir =
let rec loop files =
try
let file = Unix.readdir dir in
if f file then loop (file::files)
else loop files
with
End_of_file -> files
in
loop []
let is_uni str =
let pat = Str.regexp "^uni\.[0-9a-z]$" in
Str.string_match ~pat str ~pos:0
let main () =
let dir = Unix.opendir(".") in
let files = get_files ~f:is_uni dir in
List.iter ~f:(fun filename ->
load_universe filename;
write_state_file filename)
files
let _ = if not !Sys.interactive then main ()
|
