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_CompilingPrelude = lookup SLIT("-prelude")
165 opt_D_dump_absC = lookup SLIT("-ddump-absC")
166 opt_D_dump_asm = lookup SLIT("-ddump-asm")
167 opt_D_dump_deforest = lookup SLIT("-ddump-deforest")
168 opt_D_dump_deriv = lookup SLIT("-ddump-deriv")
169 opt_D_dump_ds = lookup SLIT("-ddump-ds")
170 opt_D_dump_flatC = lookup SLIT("-ddump-flatC")
171 opt_D_dump_occur_anal = lookup SLIT("-ddump-occur-anal")
172 opt_D_dump_rdr = lookup SLIT("-ddump-rdr")
173 opt_D_dump_realC = lookup SLIT("-ddump-realC")
174 opt_D_dump_rn = lookup SLIT("-ddump-rn")
175 opt_D_dump_simpl = lookup SLIT("-ddump-simpl")
176 opt_D_dump_spec = lookup SLIT("-ddump-spec")
177 opt_D_dump_stg = lookup SLIT("-ddump-stg")
178 opt_D_dump_stranal = lookup SLIT("-ddump-stranal")
179 opt_D_dump_tc = lookup SLIT("-ddump-tc")
180 opt_D_show_passes = lookup SLIT("-dshow-passes")
181 opt_D_simplifier_stats = lookup SLIT("-dsimplifier-stats")
182 opt_D_source_stats = lookup SLIT("-dsource-stats")
183 opt_D_verbose_core2core = lookup SLIT("-dverbose-simpl")
184 opt_D_verbose_stg2stg = lookup SLIT("-dverbose-stg")
185 opt_DoCoreLinting = lookup SLIT("-dcore-lint")
186 opt_DoSemiTagging = lookup SLIT("-fsemi-tagging")
187 opt_DoTickyProfiling = lookup SLIT("-fticky-ticky")
188 opt_EmitArityChecks = lookup SLIT("-darity-checks")
189 opt_FoldrBuildOn = lookup SLIT("-ffoldr-build-on")
190 opt_FoldrBuildTrace = lookup SLIT("-ffoldr-build-trace")
191 opt_ForConcurrent = lookup SLIT("-fconcurrent")
192 opt_GlasgowExts = lookup SLIT("-fglasgow-exts")
193 opt_Haskell_1_3 = lookup SLIT("-fhaskell-1.3")
194 opt_HideBuiltinNames = lookup SLIT("-fhide-builtin-names")
195 opt_HideMostBuiltinNames = lookup SLIT("-fmin-builtin-names")
196 opt_IgnoreStrictnessPragmas = lookup SLIT("-fignore-strictness-pragmas")
197 opt_IrrefutableEverything = lookup SLIT("-firrefutable-everything")
198 opt_IrrefutableTuples = lookup SLIT("-firrefutable-tuples")
199 opt_WarnNameShadowing = lookup SLIT("-fwarn-name-shadowing")
200 opt_NumbersStrict = lookup SLIT("-fnumbers-strict")
201 opt_OmitBlackHoling = lookup SLIT("-dno-black-holing")
202 opt_OmitDefaultInstanceMethods = lookup SLIT("-fomit-default-instance-methods")
203 opt_OmitInterfacePragmas = lookup SLIT("-fomit-interface-pragmas")
204 opt_OmitReexportedInstances = lookup SLIT("-fomit-reexported-instances")
205 opt_PprStyle_All = lookup SLIT("-dppr-all")
206 opt_PprStyle_Debug = lookup SLIT("-dppr-debug")
207 opt_PprStyle_User = lookup SLIT("-dppr-user")
208 opt_ReportWhyUnfoldingsDisallowed= lookup SLIT("-freport-disallowed-unfoldings")
209 opt_SccProfilingOn = lookup SLIT("-fscc-profiling")
210 opt_ShowImportSpecs = lookup SLIT("-fshow-import-specs")
211 opt_ShowPragmaNameErrs = lookup SLIT("-fshow-pragma-name-errs")
212 opt_SigsRequired = lookup SLIT("-fsignatures-required")
213 opt_SpecialiseAll = lookup SLIT("-fspecialise-all")
214 opt_SpecialiseImports = lookup SLIT("-fspecialise-imports")
215 opt_SpecialiseOverloaded = lookup SLIT("-fspecialise-overloaded")
216 opt_SpecialiseTrace = lookup SLIT("-ftrace-specialisation")
217 opt_SpecialiseUnboxed = lookup SLIT("-fspecialise-unboxed")
218 opt_StgDoLetNoEscapes = lookup SLIT("-flet-no-escape")
219 opt_Verbose = lookup SLIT("-v")
220 opt_AsmTarget = lookup_str "-fasm="
221 opt_SccGroup = lookup_str "-G="
222 opt_ProduceC = lookup_str "-C="
223 opt_ProduceS = lookup_str "-S="
224 opt_ProduceHi = lookup_str "-hifile="
225 opt_ProduceHu = lookup_str "-hufile="
226 opt_EnsureSplittableC = lookup_str "-fglobalise-toplev-names="
227 opt_UnfoldingUseThreshold = lookup_int "-funfolding-use-threshold"
228 opt_UnfoldingCreationThreshold = lookup_int "-funfolding-creation-threshold"
229 opt_UnfoldingOverrideThreshold = lookup_int "-funfolding-override-threshold"
230 opt_ReturnInRegsThreshold = lookup_int "-freturn-in-regs-threshold"
232 opt_NoImplicitPrelude = lookup SLIT("-fno-implicit-prelude")
233 opt_IgnoreIfacePragmas = lookup SLIT("-fignore-interface-pragmas")
235 opt_HiSuffix = case (lookup_str "-hisuffix=") of { Nothing -> ".hi" ; Just x -> x }
236 opt_SysHiSuffix = case (lookup_str "-syshisuffix=") of { Nothing -> ".hi" ; Just x -> x }
238 opt_HiDirList = get_dir_list "-i="
239 opt_SysHiDirList = get_dir_list "-j="
242 = case (lookup_str tag) of
243 Nothing -> [{-no dirs to search???-}]
244 Just xs -> colon_split xs "" [] -- character and dir accumulators, both reversed...
246 colon_split [] cacc dacc = reverse (reverse cacc : dacc)
247 colon_split (':' : xs) cacc dacc = colon_split xs "" (reverse cacc : dacc)
248 colon_split ( x : xs) cacc dacc = colon_split xs (x : cacc) dacc
250 -- -hisuf, -hisuf-prelude
251 -- -fno-implicit-prelude
252 -- -fignore-interface-pragmas
253 -- importdirs and sysimport dirs
257 classifyOpts :: ([CoreToDo], -- Core-to-Core processing spec
258 [StgToDo]) -- STG-to-STG processing spec
260 classifyOpts = sep argv [] [] -- accumulators...
262 sep :: [FAST_STRING] -- cmd-line opts (input)
263 -> [CoreToDo] -> [StgToDo] -- to_do accumulators
264 -> ([CoreToDo], [StgToDo]) -- result
266 sep [] core_td stg_td -- all done!
267 = (reverse core_td, reverse stg_td)
269 # define CORE_TD(to_do) sep opts (to_do:core_td) stg_td
270 # define STG_TD(to_do) sep opts core_td (to_do:stg_td)
271 # define IGNORE_ARG() sep opts core_td stg_td
273 sep (opt1:opts) core_td stg_td
275 case (_UNPK_ opt1) of -- the non-"just match a string" options are at the end...
277 ',' : _ -> IGNORE_ARG() -- it is for the parser
279 "-fsimplify" -> -- gather up SimplifierSwitches specially...
280 simpl_sep opts [] core_td stg_td
282 "-fcalc-inlinings1"-> CORE_TD(CoreDoCalcInlinings1)
283 "-fcalc-inlinings2"-> CORE_TD(CoreDoCalcInlinings2)
284 "-ffloat-inwards" -> CORE_TD(CoreDoFloatInwards)
285 "-ffull-laziness" -> CORE_TD(CoreDoFullLaziness)
286 "-fliberate-case" -> CORE_TD(CoreLiberateCase)
287 "-fprint-core" -> CORE_TD(CoreDoPrintCore)
288 "-fstatic-args" -> CORE_TD(CoreDoStaticArgs)
289 "-fstrictness" -> CORE_TD(CoreDoStrictness)
290 "-fspecialise" -> CORE_TD(CoreDoSpecialising)
291 "-fdeforest" -> CORE_TD(CoreDoDeforest)
292 "-fadd-auto-sccs" -> CORE_TD(CoreDoAutoCostCentres)
293 "-ffoldr-build-worker-wrapper" -> CORE_TD(CoreDoFoldrBuildWorkerWrapper)
294 "-ffoldr-build-ww-anal" -> CORE_TD(CoreDoFoldrBuildWWAnal)
296 "-fstg-static-args" -> STG_TD(StgDoStaticArgs)
297 "-fupdate-analysis" -> STG_TD(StgDoUpdateAnalysis)
298 "-dstg-stats" -> STG_TD(D_stg_stats)
299 "-flambda-lift" -> STG_TD(StgDoLambdaLift)
300 "-fmassage-stg-for-profiling" -> STG_TD(StgDoMassageForProfiling)
302 _ -> -- NB: the driver is really supposed to handle bad options
307 simpl_sep :: [FAST_STRING] -- cmd-line opts (input)
308 -> [SimplifierSwitch] -- simplifier-switch accumulator
309 -> [CoreToDo] -> [StgToDo] -- to_do accumulators
310 -> ([CoreToDo], [StgToDo]) -- result
312 -- "simpl_sep" tailcalls "sep" once it's seen one set
313 -- of SimplifierSwitches for a CoreDoSimplify.
316 simpl_sep input@[] simpl_sw core_td stg_td
317 = panic "simpl_sep []"
320 -- The SimplifierSwitches should be delimited by "(" and ")".
322 simpl_sep (opt1:opts) simpl_sw core_td stg_td
323 = case (_UNPK_ opt1) of
324 "(" -> ASSERT (null simpl_sw)
325 simpl_sep opts [] core_td stg_td
327 this_simpl = CoreDoSimplify (isAmongSimpl simpl_sw)
329 sep opts (this_simpl : core_td) stg_td
331 # define SIMPL_SW(sw) simpl_sep opts (sw:simpl_sw) core_td stg_td
333 -- the non-"just match a string" options are at the end...
334 "-fshow-simplifier-progress" -> SIMPL_SW(ShowSimplifierProgress)
335 "-fcode-duplication-ok" -> SIMPL_SW(SimplOkToDupCode)
336 "-ffloat-lets-exposing-whnf" -> SIMPL_SW(SimplFloatLetsExposingWHNF)
337 "-ffloat-primops-ok" -> SIMPL_SW(SimplOkToFloatPrimOps)
338 "-falways-float-lets-from-lets" -> SIMPL_SW(SimplAlwaysFloatLetsFromLets)
339 "-fdo-case-elim" -> SIMPL_SW(SimplDoCaseElim)
340 "-fdo-eta-reduction" -> SIMPL_SW(SimplDoEtaReduction)
341 "-fdo-lambda-eta-expansion" -> SIMPL_SW(SimplDoLambdaEtaExpansion)
342 "-fdo-foldr-build" -> SIMPL_SW(SimplDoFoldrBuild)
343 "-fdo-not-fold-back-append" -> SIMPL_SW(SimplDontFoldBackAppend)
344 "-fdo-arity-expand" -> SIMPL_SW(SimplDoArityExpand)
345 "-fdo-inline-foldr-build" -> SIMPL_SW(SimplDoInlineFoldrBuild)
346 "-freuse-con" -> SIMPL_SW(SimplReuseCon)
347 "-fcase-of-case" -> SIMPL_SW(SimplCaseOfCase)
348 "-flet-to-case" -> SIMPL_SW(SimplLetToCase)
349 "-fpedantic-bottoms" -> SIMPL_SW(SimplPedanticBottoms)
350 "-fkeep-spec-pragma-ids" -> SIMPL_SW(KeepSpecPragmaIds)
351 "-fkeep-unused-bindings" -> SIMPL_SW(KeepUnusedBindings)
352 "-fmay-delete-conjurable-ids" -> SIMPL_SW(SimplMayDeleteConjurableIds)
353 "-fessential-unfoldings-only" -> SIMPL_SW(EssentialUnfoldingsOnly)
354 "-fignore-inline-pragma" -> SIMPL_SW(IgnoreINLINEPragma)
355 "-fno-let-from-case" -> SIMPL_SW(SimplNoLetFromCase)
356 "-fno-let-from-app" -> SIMPL_SW(SimplNoLetFromApp)
357 "-fno-let-from-strict-let" -> SIMPL_SW(SimplNoLetFromStrictLet)
359 o | starts_with_msi -> SIMPL_SW(MaxSimplifierIterations (read after_msi))
360 | starts_with_suut -> SIMPL_SW(SimplUnfoldingUseThreshold (read after_suut))
361 | starts_with_suct -> SIMPL_SW(SimplUnfoldingCreationThreshold (read after_suct))
363 maybe_suut = startsWith "-fsimpl-uf-use-threshold" o
364 maybe_suct = startsWith "-fsimpl-uf-creation-threshold" o
365 maybe_msi = startsWith "-fmax-simplifier-iterations" o
366 starts_with_suut = maybeToBool maybe_suut
367 starts_with_suct = maybeToBool maybe_suct
368 starts_with_msi = maybeToBool maybe_msi
369 (Just after_suut) = maybe_suut
370 (Just after_suct) = maybe_suct
371 (Just after_msi) = maybe_msi
373 _ -> -- NB: the driver is really supposed to handle bad options
374 simpl_sep opts simpl_sw core_td stg_td
377 %************************************************************************
379 \subsection{Switch ordering}
381 %************************************************************************
383 In spite of the @Produce*@ and @SccGroup@ constructors, these things
384 behave just like enumeration types.
387 instance Eq SimplifierSwitch where
388 a == b = tagOf_SimplSwitch a _EQ_ tagOf_SimplSwitch b
390 instance Ord SimplifierSwitch where
391 a < b = tagOf_SimplSwitch a _LT_ tagOf_SimplSwitch b
392 a <= b = tagOf_SimplSwitch a _LE_ tagOf_SimplSwitch b
394 tagOf_SimplSwitch SimplOkToDupCode =(ILIT(0) :: FAST_INT)
395 tagOf_SimplSwitch SimplFloatLetsExposingWHNF = ILIT(1)
396 tagOf_SimplSwitch SimplOkToFloatPrimOps = ILIT(2)
397 tagOf_SimplSwitch SimplAlwaysFloatLetsFromLets = ILIT(3)
398 tagOf_SimplSwitch SimplDoCaseElim = ILIT(4)
399 tagOf_SimplSwitch SimplReuseCon = ILIT(5)
400 tagOf_SimplSwitch SimplCaseOfCase = ILIT(6)
401 tagOf_SimplSwitch SimplLetToCase = ILIT(7)
402 tagOf_SimplSwitch SimplMayDeleteConjurableIds = ILIT(9)
403 tagOf_SimplSwitch SimplPedanticBottoms = ILIT(10)
404 tagOf_SimplSwitch SimplDoArityExpand = ILIT(11)
405 tagOf_SimplSwitch SimplDoFoldrBuild = ILIT(12)
406 tagOf_SimplSwitch SimplDoInlineFoldrBuild = ILIT(14)
407 tagOf_SimplSwitch IgnoreINLINEPragma = ILIT(15)
408 tagOf_SimplSwitch SimplDoLambdaEtaExpansion = ILIT(16)
409 tagOf_SimplSwitch SimplDoEtaReduction = ILIT(18)
410 tagOf_SimplSwitch EssentialUnfoldingsOnly = ILIT(19)
411 tagOf_SimplSwitch ShowSimplifierProgress = ILIT(20)
412 tagOf_SimplSwitch (MaxSimplifierIterations _) = ILIT(21)
413 tagOf_SimplSwitch (SimplUnfoldingUseThreshold _) = ILIT(22)
414 tagOf_SimplSwitch (SimplUnfoldingCreationThreshold _) = ILIT(23)
415 tagOf_SimplSwitch KeepSpecPragmaIds = ILIT(24)
416 tagOf_SimplSwitch KeepUnusedBindings = ILIT(25)
417 tagOf_SimplSwitch SimplNoLetFromCase = ILIT(26)
418 tagOf_SimplSwitch SimplNoLetFromApp = ILIT(27)
419 tagOf_SimplSwitch SimplNoLetFromStrictLet = ILIT(28)
420 tagOf_SimplSwitch SimplDontFoldBackAppend = ILIT(29)
421 -- If you add anything here, be sure to change lAST_SIMPL_SWITCH_TAG, too!
423 tagOf_SimplSwitch _ = panic# "tagOf_SimplSwitch"
425 lAST_SIMPL_SWITCH_TAG = IBOX(tagOf_SimplSwitch SimplDontFoldBackAppend)
428 %************************************************************************
430 \subsection{Switch lookup}
432 %************************************************************************
435 isAmongSimpl :: [SimplifierSwitch] -> SimplifierSwitch -> SwitchResult
437 isAmongSimpl on_switches
439 tidied_on_switches = foldl rm_dups [] on_switches
441 sw_tbl :: Array Int SwitchResult
443 sw_tbl = (array (0, lAST_SIMPL_SWITCH_TAG) -- bounds...
447 all_undefined = [ i := SwBool False | i <- [0 .. lAST_SIMPL_SWITCH_TAG ] ]
449 defined_elems = map mk_assoc_elem tidied_on_switches
451 -- (avoid some unboxing, bounds checking, and other horrible things:)
452 case sw_tbl of { _Array bounds_who_needs_'em stuff ->
454 case (indexArray# stuff (tagOf_SimplSwitch switch)) of
458 mk_assoc_elem k@(MaxSimplifierIterations lvl) = IBOX(tagOf_SimplSwitch k) := SwInt lvl
459 mk_assoc_elem k@(SimplUnfoldingUseThreshold i) = IBOX(tagOf_SimplSwitch k) := SwInt i
460 mk_assoc_elem k@(SimplUnfoldingCreationThreshold i) = IBOX(tagOf_SimplSwitch k) := SwInt i
462 mk_assoc_elem k = IBOX(tagOf_SimplSwitch k) := SwBool True -- I'm here, Mom!
464 -- cannot have duplicates if we are going to use the array thing
466 rm_dups switches_so_far switch
467 = if switch `is_elem` switches_so_far
469 else switch : switches_so_far
471 sw `is_elem` [] = False
472 sw `is_elem` (s:ss) = (tagOf_SimplSwitch sw) _EQ_ (tagOf_SimplSwitch s)
476 %************************************************************************
478 \subsection{Misc functions for command-line options}
480 %************************************************************************
484 switchIsOn :: (switch -> SwitchResult) -> switch -> Bool
486 switchIsOn lookup_fn switch
487 = case (lookup_fn switch) of
488 SwBool False -> False
491 stringSwitchSet :: (switch -> SwitchResult)
492 -> (FAST_STRING -> switch)
495 stringSwitchSet lookup_fn switch
496 = case (lookup_fn (switch (panic "stringSwitchSet"))) of
497 SwString str -> Just str
500 intSwitchSet :: (switch -> SwitchResult)
504 intSwitchSet lookup_fn switch
505 = case (lookup_fn (switch (panic "intSwitchSet"))) of
506 SwInt int -> Just int