ocaml/driver/optmain.ml

74 lines
3.0 KiB
OCaml

open Clflags
let process_file name =
if Filename.check_suffix name ".ml" then begin
Optcompile.implementation name;
objfiles := (Filename.chop_suffix name ".ml" ^ ".cmx") :: !objfiles
end
else if Filename.check_suffix name ".mli" then
Optcompile.interface name
else if Filename.check_suffix name ".cmx"
or Filename.check_suffix name ".cmxa" then
objfiles := name :: !objfiles
else if Filename.check_suffix name ".o"
or Filename.check_suffix name ".a" then
ccobjs := name :: !ccobjs
else if Filename.check_suffix name ".c" then begin
Optcompile.c_file name;
ccobjs := (Filename.chop_suffix (Filename.basename name) ".c" ^ ".o")
:: !ccobjs
end
else
raise(Arg.Bad("don't know what to do with " ^ name))
let print_version_number () =
print_string "The Caml Special Light native-code compiler, version ";
print_string Config.version;
print_newline()
let main () =
try
Arg.parse
["-I", Arg.String(fun dir -> include_dirs := dir :: !include_dirs);
"-c", Arg.Unit(fun () -> compile_only := true);
"-S", Arg.Unit(fun s -> assembler_only := true; compile_only := true);
"-o", Arg.String(fun s -> exec_name := s; archive_name := s);
"-i", Arg.Unit(fun () -> print_types := true);
"-a", Arg.Unit(fun () -> make_archive := true);
"-fast", Arg.Unit(fun () -> fast := true);
"-nopervasives", Arg.Unit(fun () -> nopervasives := true);
"-ccopt", Arg.String(fun s -> ccopts := s :: !ccopts);
"-cclib", Arg.String(fun s -> ccobjs := ("-l" ^ s) :: !ccobjs);
"-l", Arg.String(fun s -> ccobjs := s :: !ccobjs);
"-dlambda", Arg.Unit(fun () -> dump_lambda := true);
"-dcmm", Arg.Unit(fun () -> Clflags.dump_cmm := true);
"-dsel", Arg.Unit(fun () -> Clflags.dump_selection := true);
"-dlive", Arg.Unit(fun () -> Clflags.dump_live := true;
Printmach.print_live := true);
"-dspill", Arg.Unit(fun () -> Clflags.dump_spill := true);
"-dsplit", Arg.Unit(fun () -> Clflags.dump_split := true);
"-dinterf", Arg.Unit(fun () -> Clflags.dump_interf := true);
"-dprefer", Arg.Unit(fun () -> Clflags.dump_prefer := true);
"-dalloc", Arg.Unit(fun () -> Clflags.dump_regalloc := true);
"-dreload", Arg.Unit(fun () -> Clflags.dump_reload := true);
"-dlinear", Arg.Unit(fun () -> Clflags.dump_linear := true);
"-dstartup", Arg.Unit(fun () -> Clflags.keep_startup_file := true);
"-v", Arg.Unit print_version_number;
"-", Arg.String process_file]
process_file;
if !make_archive then begin
Optcompile.init_path();
Asmlibrarian.create_archive (List.rev !objfiles) !archive_name
end
else if not !compile_only & !objfiles <> [] then begin
Optcompile.init_path();
Asmlink.link (List.rev !objfiles)
end;
exit 0
with x ->
Format.set_formatter_output stderr;
Opterrors.report_error x;
exit 2
let _ = Printexc.catch main ()