forked from ocaml/ocaml
/
ocamlcp.ml
183 lines (173 loc) · 7.23 KB
/
ocamlcp.ml
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
(**************************************************************************)
(* *)
(* OCaml *)
(* *)
(* Xavier Leroy, projet Cristal, INRIA Rocquencourt *)
(* *)
(* Copyright 1998 Institut National de Recherche en Informatique et *)
(* en Automatique. *)
(* *)
(* All rights reserved. This file is distributed under the terms of *)
(* the GNU Lesser General Public License version 2.1, with the *)
(* special exception on linking described in the file LICENSE. *)
(* *)
(**************************************************************************)
open Printf
let compargs = ref ([] : string list)
let profargs = ref ([] : string list)
let toremove = ref ([] : string list)
let option opt () = compargs := opt :: !compargs
let option_with_arg opt arg =
compargs := (Filename.quote arg) :: opt :: !compargs
;;
let option_with_int opt arg =
compargs := (string_of_int arg) :: opt :: !compargs
;;
let make_archive = ref false;;
let with_impl = ref false;;
let with_intf = ref false;;
let with_mli = ref false;;
let with_ml = ref false;;
let process_file filename =
if Filename.check_suffix filename ".ml" then with_ml := true;
if Filename.check_suffix filename ".mli" then with_mli := true;
compargs := (Filename.quote filename) :: !compargs
;;
let usage = "Usage: ocamlcp <options> <files>\noptions are:"
let incompatible o =
fprintf stderr "ocamlcp: profiling is incompatible with the %s option\n" o;
exit 2
module Options = Main_args.Make_bytecomp_options (struct
let _a () = make_archive := true; option "-a" ()
let _absname = option "-absname"
let _annot = option "-annot"
let _binannot = option "-bin-annot"
let _c = option "-c"
let _cc s = option_with_arg "-cc" s
let _cclib s = option_with_arg "-cclib" s
let _ccopt s = option_with_arg "-ccopt" s
let _config = option "-config"
let _config_var s = option_with_arg "-config-var" s
let _compat_32 = option "-compat-32"
let _custom = option "-custom"
let _dllib = option_with_arg "-dllib"
let _dllpath = option_with_arg "-dllpath"
let _dtypes = option "-dtypes"
let _for_pack = option_with_arg "-for-pack"
let _g = option "-g"
let _i = option "-i"
let _I s = option_with_arg "-I" s
let _impl s = with_impl := true; option_with_arg "-impl" s
let _intf s = with_intf := true; option_with_arg "-intf" s
let _intf_suffix s = option_with_arg "-intf-suffix" s
let _keep_docs = option "-keep-docs"
let _no_keep_docs = option "-no-keep-docs"
let _keep_locs = option "-keep-locs"
let _no_keep_locs = option "-no-keep-locs"
let _labels = option "-labels"
let _linkall = option "-linkall"
let _make_runtime = option "-make-runtime"
let _alias_deps = option "-alias-deps"
let _no_alias_deps = option "-no-alias-deps"
let _app_funct = option "-app-funct"
let _no_app_funct = option "-no-app-funct"
let _no_check_prims = option "-no-check-prims"
let _noassert = option "-noassert"
let _nolabels = option "-nolabels"
let _noautolink = option "-noautolink"
let _nostdlib = option "-nostdlib"
let _o s = option_with_arg "-o" s
let _opaque = option "-opaque"
let _open s = option_with_arg "-open" s
let _output_obj = option "-output-obj"
let _output_complete_obj = option "-output-complete-obj"
let _pack = option "-pack"
let _plugin = option_with_arg "-plugin"
let _pp _s = incompatible "-pp"
let _ppx _s = incompatible "-ppx"
let _principal = option "-principal"
let _no_principal = option "-no-principal"
let _rectypes = option "-rectypes"
let _no_rectypes = option "-no-rectypes"
let _runtime_variant s = option_with_arg "-runtime-variant" s
let _safe_string = option "-safe-string"
let _short_paths = option "-short-paths"
let _strict_sequence = option "-strict-sequence"
let _no_strict_sequence = option "-no-strict-sequence"
let _strict_formats = option "-strict-formats"
let _no_strict_formats = option "-no-strict-formats"
let _thread () = option "-thread" ()
let _vmthread () = option "-vmthread" ()
let _unboxed_types = option "-unboxed-types"
let _no_unboxed_types = option "-no-unboxed-types"
let _unsafe = option "-unsafe"
let _unsafe_string = option "-unsafe-string"
let _use_prims s = option_with_arg "-use-prims" s
let _use_runtime s = option_with_arg "-use-runtime" s
let _v = option "-v"
let _version = option "-version"
let _vnum = option "-vnum"
let _verbose = option "-verbose"
let _w = option_with_arg "-w"
let _warn_error = option_with_arg "-warn-error"
let _warn_help = option "-warn-help"
let _color s = option_with_arg "-color" s
let _where = option "-where"
let _nopervasives = option "-nopervasives"
let _match_context_rows n = option_with_int "-match-context-rows" n
let _dno_unique_ids = option "-dno-unique-ids"
let _dunique_ids = option "-dunique-ids"
let _dsource = option "-dsource"
let _dparsetree = option "-dparsetree"
let _dtypedtree = option "-dtypedtree"
let _drawlambda = option "-drawlambda"
let _dlambda = option "-dlambda"
let _dflambda = option "-dflambda"
let _dinstr = option "-dinstr"
let _dcamlprimc = option "-dcamlprimc"
let _dtimings = option "-dtimings"
let _dprofile = option "-dprofile"
let _args = Arg.read_arg
let _args0 = Arg.read_arg0
let anonymous = process_file
end);;
let add_profarg s =
profargs := (Filename.quote s) :: "-m" :: !profargs
;;
let optlist =
("-P", Arg.String add_profarg,
"[afilmt] Profile constructs specified by argument (default fm):\n\
\032 a Everything\n\
\032 f Function calls and method calls\n\
\032 i if ... then ... else\n\
\032 l while and for loops\n\
\032 m match ... with\n\
\032 t try ... with")
:: ("-p", Arg.String add_profarg, "[afilmt] Same as option -P")
:: Options.list
in
Arg.parse_expand optlist process_file usage;
if !with_impl && !with_intf then begin
fprintf stderr "ocamlcp cannot deal with both \"-impl\" and \"-intf\"\n";
fprintf stderr "please compile interfaces and implementations separately\n";
exit 2;
end else if !with_impl && !with_mli then begin
fprintf stderr "ocamlcp cannot deal with both \"-impl\" and .mli files\n";
fprintf stderr "please compile interfaces and implementations separately\n";
exit 2;
end else if !with_intf && !with_ml then begin
fprintf stderr "ocamlcp cannot deal with both \"-intf\" and .ml files\n";
fprintf stderr "please compile interfaces and implementations separately\n";
exit 2;
end;
if !with_impl then profargs := "-impl" :: !profargs;
if !with_intf then profargs := "-intf" :: !profargs;
let status =
Sys.command
(Printf.sprintf "ocamlc -pp \"ocamlprof -instrument %s\" %s %s"
(String.concat " " (List.rev !profargs))
(if !make_archive then "" else "profiling.cmo")
(String.concat " " (List.rev !compargs)))
in
exit status
;;