compiler/cmm/CmmCallConv.hs

   1 module CmmCallConv (
   2   ParamLocation(..),
   3   ArgumentFormat,
   4   assignArguments,
   5   argumentsSize,
   6 ) where
   7
   8 #include "HsVersions.h"
   9
  10 import Cmm
  11 import MachOp
  12 import SMRep
  13
  14 import Constants
  15 import StaticFlags (opt_Unregisterised)
  16 import Panic
  17
  18 -- Calculate the 'GlobalReg' or stack locations for function call
  19 -- parameters as used by the Cmm calling convention.
  20
  21 data ParamLocation
  22   = RegisterParam GlobalReg
  23   | StackParam WordOff
  24
  25 type ArgumentFormat a = [(a, ParamLocation)]
  26
  27 assignArguments :: (a -> MachRep) -> [a] -> ArgumentFormat a
  28 assignArguments f reps = assignments
  29     where
  30       (sizes, assignments) = unzip $ assignArguments' reps (negate (sum sizes)) availRegs
  31       assignArguments' [] offset availRegs = []
  32       assignArguments' (r:rs) offset availRegs =
  33           (size,(r,assignment)):assignArguments' rs new_offset remaining
  34           where
  35             (assignment, new_offset, size, remaining) =
  36                 assign_reg (f r) offset availRegs
  37
  38 argumentsSize :: (a -> MachRep) -> [a] -> WordOff
  39 argumentsSize f reps = maximum (0 : map arg_top args)
  40     where
  41       args = assignArguments f reps
  42
  43       arg_top (a, StackParam offset) = -offset
  44       arg_top (_, RegisterParam _) = 0
  45
  46 -----------------------------------------------------------------------------
  47 -- Local information about the registers available
  48
  49 type AvailRegs = ( [GlobalReg]   -- available vanilla regs.
  50                  , [GlobalReg]   -- floats
  51                  , [GlobalReg]   -- doubles
  52                  , [GlobalReg]   -- longs (int64 and word64)
  53                  )
  54
  55 -- Vanilla registers can contain pointers, Ints, Chars.
  56 -- Floats and doubles have separate register supplies.
  57 --
  58 -- We take these register supplies from the *real* registers, i.e. those
  59 -- that are guaranteed to map to machine registers.
  60
  61 useVanillaRegs | opt_Unregisterised = 0
  62                | otherwise          = mAX_Real_Vanilla_REG
  63 useFloatRegs   | opt_Unregisterised = 0
  64                | otherwise          = mAX_Real_Float_REG
  65 useDoubleRegs  | opt_Unregisterised = 0
  66                | otherwise          = mAX_Real_Double_REG
  67 useLongRegs    | opt_Unregisterised = 0
  68                | otherwise          = mAX_Real_Long_REG
  69
  70 availRegs = (regList VanillaReg useVanillaRegs,
  71              regList FloatReg useFloatRegs,
  72              regList DoubleReg useDoubleRegs,
  73              regList LongReg useLongRegs)
  74     where
  75       regList f max = map f [1 .. max]
  76
  77 slot_size :: LocalReg -> Int
  78 slot_size reg =
  79     ((machRepByteWidth (localRegRep reg) - 1) `div` wORD_SIZE) + 1
  80
  81 slot_size' :: MachRep -> Int
  82 slot_size' reg = ((machRepByteWidth reg - 1) `div` wORD_SIZE) + 1
  83
  84 assign_reg :: MachRep -> WordOff -> AvailRegs -> (ParamLocation, WordOff, WordOff, AvailRegs)
  85 assign_reg I8  off (v:vs, fs, ds, ls) = (RegisterParam $ v, off, 0, (vs, fs, ds, ls))
  86 assign_reg I16 off (v:vs, fs, ds, ls) = (RegisterParam $ v, off, 0, (vs, fs, ds, ls))
  87 assign_reg I32 off (v:vs, fs, ds, ls) = (RegisterParam $ v, off, 0, (vs, fs, ds, ls))
  88 assign_reg I64 off (vs, fs, ds, l:ls) = (RegisterParam $ l, off, 0, (vs, fs, ds, ls))
  89 assign_reg I128 off _                 = panic "I128 is not a supported register type"
  90 assign_reg F32 off (vs, f:fs, ds, ls) = (RegisterParam $ f, off, 0, (vs, fs, ds, ls))
  91 assign_reg F64 off (vs, fs, d:ds, ls) = (RegisterParam $ d, off, 0, (vs, fs, ds, ls))
  92 assign_reg F80 off _                  = panic "F80 is not a supported register type"
  93 assign_reg reg off _                  = (StackParam $ off, off + size, size, ([], [], [], [])) where size = slot_size' reg