2 % (c) The AQUA Project, Glasgow University, 1996
4 \section[CmdLineOpts]{Things to do with command-line options}
7 #include "HsVersions.h"
9 module CmdLineOpts where
11 import PreludeGlaST -- bad bad bad boy, Will (_Array internals)
14 CHK_Ubiq() -- debugging consistency check
16 import Maybes ( assocMaybe, firstJust, maybeToBool, Maybe(..) )
17 import Util ( startsWith, panic, panic#, assertPanic )
20 A command-line {\em switch} is (generally) either on or off; e.g., the
21 ``verbose'' (-v) switch is either on or off. (The \tr{-G<group>}
22 switch is an exception; it's set to a string, or nothing.)
24 A list of {\em ToDo}s is things to be done in a particular part of
25 processing. A (fictitious) example for the Core-to-Core simplifier
26 might be: run the simplifier, then run the strictness analyser, then
27 run the simplifier again (three ``todos'').
29 There are three ``to-do processing centers'' at the moment. In the
30 main loop (\tr{main/Main.lhs}), in the Core-to-Core processing loop
31 (\tr{simplCore/SimplCore.lhs), and in the STG-to-STG processing loop
32 (\tr{simplStg/SimplStg.lhs}).
34 %************************************************************************
36 \subsection{Datatypes associated with command-line options}
38 %************************************************************************
42 = SwBool Bool -- on/off
43 | SwString FAST_STRING -- nothing or a String
44 | SwInt Int -- nothing or an Int
48 data CoreToDo -- These are diff core-to-core passes,
49 -- which may be invoked in any order,
50 -- as many times as you like.
52 = CoreDoSimplify -- The core-to-core simplifier.
53 (SimplifierSwitch -> SwitchResult)
54 -- Each run of the simplifier can take a different
55 -- set of simplifier-specific flags.
56 | CoreDoCalcInlinings1
57 | CoreDoCalcInlinings2
66 | CoreDoAutoCostCentres
67 | CoreDoFoldrBuildWorkerWrapper
68 | CoreDoFoldrBuildWWAnal
76 | StgDoMassageForProfiling -- should be (next to) last
77 -- There's also setStgVarInfo, but its absolute "lastness"
78 -- is so critical that it is hardwired in (no flag).
85 | SimplFloatLetsExposingWHNF
86 | SimplOkToFloatPrimOps
87 | SimplAlwaysFloatLetsFromLets
92 | SimplMayDeleteConjurableIds
93 | SimplPedanticBottoms -- see Simplifier for an explanation
94 | SimplDoArityExpand -- expand arity of bindings
95 | SimplDoFoldrBuild -- This is the per-simplification flag;
96 -- see also FoldrBuildOn, used elsewhere
98 | SimplDoInlineFoldrBuild
99 -- inline foldr/build (*after* f/b rule is used)
102 | SimplDoLambdaEtaExpansion
103 | SimplDoEtaReduction
105 | EssentialUnfoldingsOnly -- never mind the thresholds, only
106 -- do unfoldings that *must* be done
107 -- (to saturate constructors and primitives)
109 | ShowSimplifierProgress -- report counts on every interation
111 | MaxSimplifierIterations Int
113 | SimplUnfoldingUseThreshold Int -- per-simplification variants
114 | SimplUnfoldingCreationThreshold Int
116 | KeepSpecPragmaIds -- We normally *toss* Ids we can do without
119 | SimplNoLetFromCase -- used when turning off floating entirely
120 | SimplNoLetFromApp -- (for experimentation only) WDP 95/10
121 | SimplNoLetFromStrictLet
123 | SimplDontFoldBackAppend
124 -- we fold `foldr (:)' back into flip (++),
125 -- but we *don't* want to do it when compiling
126 -- List.hs, otherwise
127 -- xs ++ ys = foldr (:) ys xs
128 -- {- via our loopback -}
129 -- xs ++ ys = xs ++ ys
131 -- So only use this flag inside List.hs
132 -- (Sigh, what a HACK, Andy. WDP 96/01)
135 %************************************************************************
137 \subsection{Classifying command-line options}
139 %************************************************************************
142 lookup :: FAST_STRING -> Bool
143 lookup_int :: String -> Maybe Int
144 lookup_str :: String -> Maybe String
146 lookup sw = maybeToBool (assoc_opts sw)
148 lookup_str sw = firstJust (map (startsWith sw) unpacked_opts)
150 lookup_int sw = case (lookup_str sw) of
152 Just xx -> Just (read xx)
154 assoc_opts = assocMaybe [ (a, True) | a <- argv ]
155 unpacked_opts = map _UNPK_ argv
159 opt_AllDemanded = lookup SLIT("-fall-demanded")
160 opt_AllStrict = lookup SLIT("-fall-strict")
161 opt_AutoSccsOnAllToplevs = lookup SLIT("-fauto-sccs-on-all-toplevs")
162 opt_AutoSccsOnExportedToplevs = lookup SLIT("-fauto-sccs-on-exported-toplevs")
163 opt_AutoSccsOnIndividualCafs = lookup SLIT("-fauto-sccs-on-individual-cafs")
164 opt_D_dump_absC = lookup SLIT("-ddump-absC")
165 opt_D_dump_asm = lookup SLIT("-ddump-asm")
166 opt_D_dump_deforest = lookup SLIT("-ddump-deforest")
167 opt_D_dump_deriv = lookup SLIT("-ddump-deriv")
168 opt_D_dump_ds = lookup SLIT("-ddump-ds")
169 opt_D_dump_flatC = lookup SLIT("-ddump-flatC")
170 opt_D_dump_occur_anal = lookup SLIT("-ddump-occur-anal")
171 opt_D_dump_rdr = lookup SLIT("-ddump-rdr")
172 opt_D_dump_realC = lookup SLIT("-ddump-realC")
173 opt_D_dump_rn = lookup SLIT("-ddump-rn")
174 opt_D_dump_simpl = lookup SLIT("-ddump-simpl")
175 opt_D_dump_spec = lookup SLIT("-ddump-spec")
176 opt_D_dump_stg = lookup SLIT("-ddump-stg")
177 opt_D_dump_stranal = lookup SLIT("-ddump-stranal")
178 opt_D_dump_tc = lookup SLIT("-ddump-tc")
179 opt_D_show_passes = lookup SLIT("-dshow-passes")
180 opt_D_simplifier_stats = lookup SLIT("-dsimplifier-stats")
181 opt_D_source_stats = lookup SLIT("-dsource-stats")
182 opt_D_verbose_core2core = lookup SLIT("-dverbose-simpl")
183 opt_D_verbose_stg2stg = lookup SLIT("-dverbose-stg")
184 opt_DoCoreLinting = lookup SLIT("-dcore-lint")
185 opt_DoSemiTagging = lookup SLIT("-fsemi-tagging")
186 opt_DoTickyProfiling = lookup SLIT("-fticky-ticky")
187 opt_FoldrBuildOn = lookup SLIT("-ffoldr-build-on")
188 opt_FoldrBuildTrace = lookup SLIT("-ffoldr-build-trace")
189 opt_ForConcurrent = lookup SLIT("-fconcurrent")
190 opt_GranMacros = lookup SLIT("-fgransim")
191 opt_GlasgowExts = lookup SLIT("-fglasgow-exts")
192 opt_Haskell_1_3 = lookup SLIT("-fhaskell-1.3")
193 opt_HideBuiltinNames = lookup SLIT("-fhide-builtin-names")
194 opt_HideMostBuiltinNames = lookup SLIT("-fmin-builtin-names")
195 opt_IgnoreStrictnessPragmas = lookup SLIT("-fignore-strictness-pragmas")
196 opt_IrrefutableEverything = lookup SLIT("-firrefutable-everything")
197 opt_IrrefutableTuples = lookup SLIT("-firrefutable-tuples")
198 opt_WarnNameShadowing = lookup SLIT("-fwarn-name-shadowing")
199 opt_NumbersStrict = lookup SLIT("-fnumbers-strict")
200 opt_OmitBlackHoling = lookup SLIT("-dno-black-holing")
201 opt_OmitDefaultInstanceMethods = lookup SLIT("-fomit-default-instance-methods")
202 opt_OmitInterfacePragmas = lookup SLIT("-fomit-interface-pragmas")
203 opt_PprStyle_All = lookup SLIT("-dppr-all")
204 opt_PprStyle_Debug = lookup SLIT("-dppr-debug")
205 opt_PprStyle_User = lookup SLIT("-dppr-user")
206 opt_ReportWhyUnfoldingsDisallowed= lookup SLIT("-freport-disallowed-unfoldings")
207 opt_SccProfilingOn = lookup SLIT("-fscc-profiling")
208 opt_ShowImportSpecs = lookup SLIT("-fshow-import-specs")
209 opt_ShowPragmaNameErrs = lookup SLIT("-fshow-pragma-name-errs")
210 opt_SigsRequired = lookup SLIT("-fsignatures-required")
211 opt_SpecialiseAll = lookup SLIT("-fspecialise-all")
212 opt_SpecialiseImports = lookup SLIT("-fspecialise-imports")
213 opt_SpecialiseOverloaded = lookup SLIT("-fspecialise-overloaded")
214 opt_SpecialiseTrace = lookup SLIT("-ftrace-specialisation")
215 opt_SpecialiseUnboxed = lookup SLIT("-fspecialise-unboxed")
216 opt_StgDoLetNoEscapes = lookup SLIT("-flet-no-escape")
217 opt_Verbose = lookup SLIT("-v")
218 opt_CompilingPrelude = maybeToBool maybe_CompilingPrelude
219 maybe_CompilingPrelude = lookup_str "-fcompiling-prelude="
220 opt_SccGroup = lookup_str "-G="
221 opt_ProduceC = lookup_str "-C="
222 opt_ProduceS = lookup_str "-S="
223 opt_ProduceHi = lookup_str "-hifile=" -- the one to produce this time
224 opt_HiMap = lookup_str "-himap=" -- file saying where to look for .hi files
225 opt_EnsureSplittableC = lookup_str "-fglobalise-toplev-names="
226 opt_UnfoldingUseThreshold = lookup_int "-funfolding-use-threshold"
227 opt_UnfoldingCreationThreshold = lookup_int "-funfolding-creation-threshold"
228 opt_UnfoldingOverrideThreshold = lookup_int "-funfolding-override-threshold"
229 opt_ReturnInRegsThreshold = lookup_int "-freturn-in-regs-threshold"
231 opt_NoImplicitPrelude = lookup SLIT("-fno-implicit-prelude")
232 opt_IgnoreIfacePragmas = lookup SLIT("-fignore-interface-pragmas")
236 classifyOpts :: ([CoreToDo], -- Core-to-Core processing spec
237 [StgToDo]) -- STG-to-STG processing spec
239 classifyOpts = sep argv [] [] -- accumulators...
241 sep :: [FAST_STRING] -- cmd-line opts (input)
242 -> [CoreToDo] -> [StgToDo] -- to_do accumulators
243 -> ([CoreToDo], [StgToDo]) -- result
245 sep [] core_td stg_td -- all done!
246 = (reverse core_td, reverse stg_td)
248 # define CORE_TD(to_do) sep opts (to_do:core_td) stg_td
249 # define STG_TD(to_do) sep opts core_td (to_do:stg_td)
250 # define IGNORE_ARG() sep opts core_td stg_td
252 sep (opt1:opts) core_td stg_td
254 case (_UNPK_ opt1) of -- the non-"just match a string" options are at the end...
256 ',' : _ -> IGNORE_ARG() -- it is for the parser
258 "-fsimplify" -> -- gather up SimplifierSwitches specially...
259 simpl_sep opts [] core_td stg_td
261 "-fcalc-inlinings1"-> CORE_TD(CoreDoCalcInlinings1)
262 "-fcalc-inlinings2"-> CORE_TD(CoreDoCalcInlinings2)
263 "-ffloat-inwards" -> CORE_TD(CoreDoFloatInwards)
264 "-ffull-laziness" -> CORE_TD(CoreDoFullLaziness)
265 "-fliberate-case" -> CORE_TD(CoreLiberateCase)
266 "-fprint-core" -> CORE_TD(CoreDoPrintCore)
267 "-fstatic-args" -> CORE_TD(CoreDoStaticArgs)
268 "-fstrictness" -> CORE_TD(CoreDoStrictness)
269 "-fspecialise" -> CORE_TD(CoreDoSpecialising)
270 "-fdeforest" -> CORE_TD(CoreDoDeforest)
271 "-fadd-auto-sccs" -> CORE_TD(CoreDoAutoCostCentres)
272 "-ffoldr-build-worker-wrapper" -> CORE_TD(CoreDoFoldrBuildWorkerWrapper)
273 "-ffoldr-build-ww-anal" -> CORE_TD(CoreDoFoldrBuildWWAnal)
275 "-fstg-static-args" -> STG_TD(StgDoStaticArgs)
276 "-fupdate-analysis" -> STG_TD(StgDoUpdateAnalysis)
277 "-dstg-stats" -> STG_TD(D_stg_stats)
278 "-flambda-lift" -> STG_TD(StgDoLambdaLift)
279 "-fmassage-stg-for-profiling" -> STG_TD(StgDoMassageForProfiling)
281 _ -> -- NB: the driver is really supposed to handle bad options
286 simpl_sep :: [FAST_STRING] -- cmd-line opts (input)
287 -> [SimplifierSwitch] -- simplifier-switch accumulator
288 -> [CoreToDo] -> [StgToDo] -- to_do accumulators
289 -> ([CoreToDo], [StgToDo]) -- result
291 -- "simpl_sep" tailcalls "sep" once it's seen one set
292 -- of SimplifierSwitches for a CoreDoSimplify.
295 simpl_sep input@[] simpl_sw core_td stg_td
296 = panic "simpl_sep []"
299 -- The SimplifierSwitches should be delimited by "(" and ")".
301 simpl_sep (opt1:opts) simpl_sw core_td stg_td
302 = case (_UNPK_ opt1) of
303 "(" -> ASSERT (null simpl_sw)
304 simpl_sep opts [] core_td stg_td
306 this_simpl = CoreDoSimplify (isAmongSimpl simpl_sw)
308 sep opts (this_simpl : core_td) stg_td
310 # define SIMPL_SW(sw) simpl_sep opts (sw:simpl_sw) core_td stg_td
312 -- the non-"just match a string" options are at the end...
313 "-fshow-simplifier-progress" -> SIMPL_SW(ShowSimplifierProgress)
314 "-fcode-duplication-ok" -> SIMPL_SW(SimplOkToDupCode)
315 "-ffloat-lets-exposing-whnf" -> SIMPL_SW(SimplFloatLetsExposingWHNF)
316 "-ffloat-primops-ok" -> SIMPL_SW(SimplOkToFloatPrimOps)
317 "-falways-float-lets-from-lets" -> SIMPL_SW(SimplAlwaysFloatLetsFromLets)
318 "-fdo-case-elim" -> SIMPL_SW(SimplDoCaseElim)
319 "-fdo-eta-reduction" -> SIMPL_SW(SimplDoEtaReduction)
320 "-fdo-lambda-eta-expansion" -> SIMPL_SW(SimplDoLambdaEtaExpansion)
321 "-fdo-foldr-build" -> SIMPL_SW(SimplDoFoldrBuild)
322 "-fdo-not-fold-back-append" -> SIMPL_SW(SimplDontFoldBackAppend)
323 "-fdo-arity-expand" -> SIMPL_SW(SimplDoArityExpand)
324 "-fdo-inline-foldr-build" -> SIMPL_SW(SimplDoInlineFoldrBuild)
325 "-freuse-con" -> SIMPL_SW(SimplReuseCon)
326 "-fcase-of-case" -> SIMPL_SW(SimplCaseOfCase)
327 "-flet-to-case" -> SIMPL_SW(SimplLetToCase)
328 "-fpedantic-bottoms" -> SIMPL_SW(SimplPedanticBottoms)
329 "-fkeep-spec-pragma-ids" -> SIMPL_SW(KeepSpecPragmaIds)
330 "-fkeep-unused-bindings" -> SIMPL_SW(KeepUnusedBindings)
331 "-fmay-delete-conjurable-ids" -> SIMPL_SW(SimplMayDeleteConjurableIds)
332 "-fessential-unfoldings-only" -> SIMPL_SW(EssentialUnfoldingsOnly)
333 "-fignore-inline-pragma" -> SIMPL_SW(IgnoreINLINEPragma)
334 "-fno-let-from-case" -> SIMPL_SW(SimplNoLetFromCase)
335 "-fno-let-from-app" -> SIMPL_SW(SimplNoLetFromApp)
336 "-fno-let-from-strict-let" -> SIMPL_SW(SimplNoLetFromStrictLet)
338 o | starts_with_msi -> SIMPL_SW(MaxSimplifierIterations (read after_msi))
339 | starts_with_suut -> SIMPL_SW(SimplUnfoldingUseThreshold (read after_suut))
340 | starts_with_suct -> SIMPL_SW(SimplUnfoldingCreationThreshold (read after_suct))
342 maybe_suut = startsWith "-fsimpl-uf-use-threshold" o
343 maybe_suct = startsWith "-fsimpl-uf-creation-threshold" o
344 maybe_msi = startsWith "-fmax-simplifier-iterations" o
345 starts_with_suut = maybeToBool maybe_suut
346 starts_with_suct = maybeToBool maybe_suct
347 starts_with_msi = maybeToBool maybe_msi
348 (Just after_suut) = maybe_suut
349 (Just after_suct) = maybe_suct
350 (Just after_msi) = maybe_msi
352 _ -> -- NB: the driver is really supposed to handle bad options
353 simpl_sep opts simpl_sw core_td stg_td
356 %************************************************************************
358 \subsection{Switch ordering}
360 %************************************************************************
362 In spite of the @Produce*@ and @SccGroup@ constructors, these things
363 behave just like enumeration types.
366 instance Eq SimplifierSwitch where
367 a == b = tagOf_SimplSwitch a _EQ_ tagOf_SimplSwitch b
369 instance Ord SimplifierSwitch where
370 a < b = tagOf_SimplSwitch a _LT_ tagOf_SimplSwitch b
371 a <= b = tagOf_SimplSwitch a _LE_ tagOf_SimplSwitch b
373 tagOf_SimplSwitch SimplOkToDupCode =(ILIT(0) :: FAST_INT)
374 tagOf_SimplSwitch SimplFloatLetsExposingWHNF = ILIT(1)
375 tagOf_SimplSwitch SimplOkToFloatPrimOps = ILIT(2)
376 tagOf_SimplSwitch SimplAlwaysFloatLetsFromLets = ILIT(3)
377 tagOf_SimplSwitch SimplDoCaseElim = ILIT(4)
378 tagOf_SimplSwitch SimplReuseCon = ILIT(5)
379 tagOf_SimplSwitch SimplCaseOfCase = ILIT(6)
380 tagOf_SimplSwitch SimplLetToCase = ILIT(7)
381 tagOf_SimplSwitch SimplMayDeleteConjurableIds = ILIT(9)
382 tagOf_SimplSwitch SimplPedanticBottoms = ILIT(10)
383 tagOf_SimplSwitch SimplDoArityExpand = ILIT(11)
384 tagOf_SimplSwitch SimplDoFoldrBuild = ILIT(12)
385 tagOf_SimplSwitch SimplDoInlineFoldrBuild = ILIT(14)
386 tagOf_SimplSwitch IgnoreINLINEPragma = ILIT(15)
387 tagOf_SimplSwitch SimplDoLambdaEtaExpansion = ILIT(16)
388 tagOf_SimplSwitch SimplDoEtaReduction = ILIT(18)
389 tagOf_SimplSwitch EssentialUnfoldingsOnly = ILIT(19)
390 tagOf_SimplSwitch ShowSimplifierProgress = ILIT(20)
391 tagOf_SimplSwitch (MaxSimplifierIterations _) = ILIT(21)
392 tagOf_SimplSwitch (SimplUnfoldingUseThreshold _) = ILIT(22)
393 tagOf_SimplSwitch (SimplUnfoldingCreationThreshold _) = ILIT(23)
394 tagOf_SimplSwitch KeepSpecPragmaIds = ILIT(24)
395 tagOf_SimplSwitch KeepUnusedBindings = ILIT(25)
396 tagOf_SimplSwitch SimplNoLetFromCase = ILIT(26)
397 tagOf_SimplSwitch SimplNoLetFromApp = ILIT(27)
398 tagOf_SimplSwitch SimplNoLetFromStrictLet = ILIT(28)
399 tagOf_SimplSwitch SimplDontFoldBackAppend = ILIT(29)
400 -- If you add anything here, be sure to change lAST_SIMPL_SWITCH_TAG, too!
402 tagOf_SimplSwitch _ = panic# "tagOf_SimplSwitch"
404 lAST_SIMPL_SWITCH_TAG = IBOX(tagOf_SimplSwitch SimplDontFoldBackAppend)
407 %************************************************************************
409 \subsection{Switch lookup}
411 %************************************************************************
414 isAmongSimpl :: [SimplifierSwitch] -> SimplifierSwitch -> SwitchResult
416 isAmongSimpl on_switches
418 tidied_on_switches = foldl rm_dups [] on_switches
420 sw_tbl :: Array Int SwitchResult
422 sw_tbl = (array (0, lAST_SIMPL_SWITCH_TAG) -- bounds...
426 all_undefined = [ i := SwBool False | i <- [0 .. lAST_SIMPL_SWITCH_TAG ] ]
428 defined_elems = map mk_assoc_elem tidied_on_switches
430 -- (avoid some unboxing, bounds checking, and other horrible things:)
431 case sw_tbl of { _Array bounds_who_needs_'em stuff ->
433 case (indexArray# stuff (tagOf_SimplSwitch switch)) of
437 mk_assoc_elem k@(MaxSimplifierIterations lvl) = IBOX(tagOf_SimplSwitch k) := SwInt lvl
438 mk_assoc_elem k@(SimplUnfoldingUseThreshold i) = IBOX(tagOf_SimplSwitch k) := SwInt i
439 mk_assoc_elem k@(SimplUnfoldingCreationThreshold i) = IBOX(tagOf_SimplSwitch k) := SwInt i
441 mk_assoc_elem k = IBOX(tagOf_SimplSwitch k) := SwBool True -- I'm here, Mom!
443 -- cannot have duplicates if we are going to use the array thing
445 rm_dups switches_so_far switch
446 = if switch `is_elem` switches_so_far
448 else switch : switches_so_far
450 sw `is_elem` [] = False
451 sw `is_elem` (s:ss) = (tagOf_SimplSwitch sw) _EQ_ (tagOf_SimplSwitch s)
455 %************************************************************************
457 \subsection{Misc functions for command-line options}
459 %************************************************************************
463 switchIsOn :: (switch -> SwitchResult) -> switch -> Bool
465 switchIsOn lookup_fn switch
466 = case (lookup_fn switch) of
467 SwBool False -> False
470 stringSwitchSet :: (switch -> SwitchResult)
471 -> (FAST_STRING -> switch)
474 stringSwitchSet lookup_fn switch
475 = case (lookup_fn (switch (panic "stringSwitchSet"))) of
476 SwString str -> Just str
479 intSwitchSet :: (switch -> SwitchResult)
483 intSwitchSet lookup_fn switch
484 = case (lookup_fn (switch (panic "intSwitchSet"))) of
485 SwInt int -> Just int