viewcvs.cgi/driver/run.ml

let () = State.close ();;

let dump = ref None
let src  = ref []

let specs =
  [ "-dump", Arg.String (fun s -> dump := Some s), 
    " specify filename for persistency";
    "-quiet", Arg.Set Cduce.quiet, 
    "suppress normal output (typing, results)"
 ]

let () =
  Arg.parse specs (fun s -> src := s :: !src) 
    "cduce [options] [script]\n\nOptions:"

let ppf = 
  if !Cduce.quiet then Format.formatter_of_buffer (Buffer.create 1023)
  else Format.std_formatter
let ppf_err = Format.err_formatter

let do_file s =
  let (src, chan) = 
    if s = "" then (`Stream, stdin) else (`File s, open_in s) in
  Location.push_source src;
  let input = Stream.of_channel chan in
  if Stream.peek input = Some '#' then
    (
      let rec count n =
        match Stream.next input with
          | '\n' -> n
          | _ -> count (n + 1) in
      Wlexer.set_delta_loc (count 1)
    );
  let ok = Cduce.run ppf ppf_err input in
  if s <> "" then close_in chan;
  if not ok then (Format.fprintf ppf_err "@."; exit 1)

  
let main () =
  (match !dump with
     | Some f ->
         (try 
            Format.fprintf ppf "Restoring state: ";
            let chan = open_in_bin f in
            let s = Marshal.from_channel chan in
            close_in chan;
            State.set s;
            Format.fprintf ppf "done ...@."
          with Sys_error _ ->  
            Format.fprintf ppf "failed ...@.")
     | None -> ());
  (match !src with
     | [] -> 
         Format.fprintf ppf "No script specified; using stdin ...@.";
         do_file ""
     | l -> List.iter do_file l);
  (match !dump with
     | Some f ->
         Format.fprintf ppf "Saving state ...@\n";
         let s = State.get () in
         let chan = open_out_bin f in
         Marshal.to_channel chan s [ Marshal.Closures ];
         close_out chan
     | None -> ())


let () = main (); Types.print_stat ppf_err
1	let () = State.close ();;
2
3	let dump = ref None
4	let src = ref []
5
6	let specs =
7	[ "-dump", Arg.String (fun s -> dump := Some s),
8	" specify filename for persistency";
9	"-quiet", Arg.Set Cduce.quiet,
10	"suppress normal output (typing, results)"
11	]
12
13	let () =
14	Arg.parse specs (fun s -> src := s :: !src)
15	"cduce [options] [script]\n\nOptions:"
16
17	let ppf =
18	if !Cduce.quiet then Format.formatter_of_buffer (Buffer.create 1023)
19	else Format.std_formatter
20	let ppf_err = Format.err_formatter
21
22	let do_file s =
23	let (src, chan) =
24	if s = "" then (`Stream, stdin) else (`File s, open_in s) in
25	Location.push_source src;
26	let input = Stream.of_channel chan in
27	if Stream.peek input = Some '#' then
28	(
29	let rec count n =
30	match Stream.next input with
31	\| '\n' -> n
32	\| _ -> count (n + 1) in
33	Wlexer.set_delta_loc (count 1)
34	);
35	let ok = Cduce.run ppf ppf_err input in
36	if s <> "" then close_in chan;
37	if not ok then (Format.fprintf ppf_err "@."; exit 1)
38
39
40
41	let main () =
42	(match !dump with
43	\| Some f ->
44	(try
45	Format.fprintf ppf "Restoring state: ";
46	let chan = open_in_bin f in
47	let s = Marshal.from_channel chan in
48	close_in chan;
49	State.set s;
50	Format.fprintf ppf "done ...@."
51	with Sys_error _ ->
52	Format.fprintf ppf "failed ...@.")
53	\| None -> ());
54	(match !src with
55	\| [] ->
56	Format.fprintf ppf "No script specified; using stdin ...@.";
57	do_file ""
58	\| l -> List.iter do_file l);
59	(match !dump with
60	\| Some f ->
61	Format.fprintf ppf "Saving state ...@\n";
62	let s = State.get () in
63	let chan = open_out_bin f in
64	Marshal.to_channel chan s [ Marshal.Closures ];
65	close_out chan
66	\| None -> ())
67
68
69
70	let () = main (); Types.print_stat ppf_err