2 -- The above warning supression flag is a temporary kludge.
3 -- While working on this module you are encouraged to remove it and fix
4 -- any warnings in the module. See
5 -- http://hackage.haskell.org/trac/ghc/wiki/WorkingConventions#Warnings
15 #include "HsVersions.h"
22 import StaticFlags (opt_Unregisterised)
25 -- Calculate the 'GlobalReg' or stack locations for function call
26 -- parameters as used by the Cmm calling convention.
29 = RegisterParam GlobalReg
32 type ArgumentFormat a = [(a, ParamLocation)]
34 assignArguments :: (a -> MachRep) -> [a] -> ArgumentFormat a
35 assignArguments f reps = assignments
37 (sizes, assignments) = unzip $ assignArguments' reps (negate (sum sizes)) availRegs
38 assignArguments' [] offset availRegs = []
39 assignArguments' (r:rs) offset availRegs =
40 (size,(r,assignment)):assignArguments' rs new_offset remaining
42 (assignment, new_offset, size, remaining) =
43 assign_reg (f r) offset availRegs
45 argumentsSize :: (a -> MachRep) -> [a] -> WordOff
46 argumentsSize f reps = maximum (0 : map arg_top args)
48 args = assignArguments f reps
50 arg_top (a, StackParam offset) = -offset
51 arg_top (_, RegisterParam _) = 0
53 -----------------------------------------------------------------------------
54 -- Local information about the registers available
56 type AvailRegs = ( [GlobalReg] -- available vanilla regs.
57 , [GlobalReg] -- floats
58 , [GlobalReg] -- doubles
59 , [GlobalReg] -- longs (int64 and word64)
62 -- Vanilla registers can contain pointers, Ints, Chars.
63 -- Floats and doubles have separate register supplies.
65 -- We take these register supplies from the *real* registers, i.e. those
66 -- that are guaranteed to map to machine registers.
68 useVanillaRegs | opt_Unregisterised = 0
69 | otherwise = mAX_Real_Vanilla_REG
70 useFloatRegs | opt_Unregisterised = 0
71 | otherwise = mAX_Real_Float_REG
72 useDoubleRegs | opt_Unregisterised = 0
73 | otherwise = mAX_Real_Double_REG
74 useLongRegs | opt_Unregisterised = 0
75 | otherwise = mAX_Real_Long_REG
77 availRegs = (regList VanillaReg useVanillaRegs,
78 regList FloatReg useFloatRegs,
79 regList DoubleReg useDoubleRegs,
80 regList LongReg useLongRegs)
82 regList f max = map f [1 .. max]
84 slot_size :: LocalReg -> Int
86 ((machRepByteWidth (localRegRep reg) - 1) `div` wORD_SIZE) + 1
88 slot_size' :: MachRep -> Int
89 slot_size' reg = ((machRepByteWidth reg - 1) `div` wORD_SIZE) + 1
91 assign_reg :: MachRep -> WordOff -> AvailRegs -> (ParamLocation, WordOff, WordOff, AvailRegs)
92 assign_reg I8 off (v:vs, fs, ds, ls) = (RegisterParam $ v, off, 0, (vs, fs, ds, ls))
93 assign_reg I16 off (v:vs, fs, ds, ls) = (RegisterParam $ v, off, 0, (vs, fs, ds, ls))
94 assign_reg I32 off (v:vs, fs, ds, ls) = (RegisterParam $ v, off, 0, (vs, fs, ds, ls))
95 assign_reg I64 off (vs, fs, ds, l:ls) = (RegisterParam $ l, off, 0, (vs, fs, ds, ls))
96 assign_reg I128 off _ = panic "I128 is not a supported register type"
97 assign_reg F32 off (vs, f:fs, ds, ls) = (RegisterParam $ f, off, 0, (vs, fs, ds, ls))
98 assign_reg F64 off (vs, fs, d:ds, ls) = (RegisterParam $ d, off, 0, (vs, fs, ds, ls))
99 assign_reg F80 off _ = panic "F80 is not a supported register type"
100 assign_reg reg off _ = (StackParam $ off, off + size, size, ([], [], [], [])) where size = slot_size' reg