-- (c) The University of Glasgow 2006
-- (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
--
-- Storage manager representation of closures

{-# LANGUAGE CPP,GeneralizedNewtypeDeriving #-}

module GHC.Runtime.Heap.Layout (
        -- * Words and bytes
        WordOff, ByteOff,
        wordsToBytes, bytesToWordsRoundUp,
        roundUpToWords, roundUpTo,

        StgWord, fromStgWord, toStgWord,
        StgHalfWord, fromStgHalfWord, toStgHalfWord,
        halfWordSize, halfWordSizeInBits,

        -- * Closure representation
        SMRep(..), -- CmmInfo sees the rep; no one else does
        IsStatic,
        ClosureTypeInfo(..), ArgDescr(..), Liveness,
        ConstrDescription,

        -- ** Construction
        mkHeapRep, blackHoleRep, indStaticRep, mkStackRep, mkRTSRep, arrPtrsRep,
        smallArrPtrsRep, arrWordsRep,

        -- ** Predicates
        isStaticRep, isConRep, isThunkRep, isFunRep, isStaticNoCafCon,
        isStackRep,

        -- ** Size-related things
        heapClosureSizeW,
        fixedHdrSizeW, arrWordsHdrSize, arrWordsHdrSizeW, arrPtrsHdrSize,
        arrPtrsHdrSizeW, profHdrSize, thunkHdrSize, nonHdrSize, nonHdrSizeW,
        smallArrPtrsHdrSize, smallArrPtrsHdrSizeW, hdrSize, hdrSizeW,
        fixedHdrSize,

        -- ** RTS closure types
        rtsClosureType, rET_SMALL, rET_BIG,
        aRG_GEN, aRG_GEN_BIG,

        -- ** Arrays
        card, cardRoundUp, cardTableSizeB, cardTableSizeW
    ) where

import GHC.Prelude

import GHC.Types.Basic( ConTagZ )
import GHC.Driver.Session
import GHC.Platform
import GHC.Platform.Profile
import GHC.Data.FastString
import GHC.StgToCmm.Types

import GHC.Utils.Outputable
import GHC.Utils.Panic

import Data.Word
import Data.ByteString (ByteString)

{-
************************************************************************
*                                                                      *
                Words and bytes
*                                                                      *
************************************************************************
-}

-- | Byte offset, or byte count
type ByteOff = Int

-- | Round up the given byte count to the next byte count that's a
-- multiple of the machine's word size.
roundUpToWords :: Platform -> ByteOff -> ByteOff
roundUpToWords :: Platform -> ConTagZ -> ConTagZ
roundUpToWords Platform
platform ConTagZ
n = ConTagZ -> ConTagZ -> ConTagZ
roundUpTo ConTagZ
n (Platform -> ConTagZ
platformWordSizeInBytes Platform
platform)

-- | Round up @base@ to a multiple of @size@.
roundUpTo :: ByteOff -> ByteOff -> ByteOff
roundUpTo :: ConTagZ -> ConTagZ -> ConTagZ
roundUpTo ConTagZ
base ConTagZ
size = (ConTagZ
base forall a. Num a => a -> a -> a
+ (ConTagZ
size forall a. Num a => a -> a -> a
- ConTagZ
1)) forall a. Bits a => a -> a -> a
.&. (forall a. Bits a => a -> a
complement (ConTagZ
size forall a. Num a => a -> a -> a
- ConTagZ
1))

-- | Convert the given number of words to a number of bytes.
--
-- This function morally has type @WordOff -> ByteOff@, but uses @Num
-- a@ to allow for overloading.
wordsToBytes :: Num a => Platform -> a -> a
wordsToBytes :: forall a. Num a => Platform -> a -> a
wordsToBytes Platform
platform a
n = forall a b. (Integral a, Num b) => a -> b
fromIntegral (Platform -> ConTagZ
platformWordSizeInBytes Platform
platform) forall a. Num a => a -> a -> a
* a
n
{-# SPECIALIZE wordsToBytes :: Platform -> Int -> Int #-}
{-# SPECIALIZE wordsToBytes :: Platform -> Word -> Word #-}
{-# SPECIALIZE wordsToBytes :: Platform -> Integer -> Integer #-}

-- | First round the given byte count up to a multiple of the
-- machine's word size and then convert the result to words.
bytesToWordsRoundUp :: Platform -> ByteOff -> WordOff
bytesToWordsRoundUp :: Platform -> ConTagZ -> ConTagZ
bytesToWordsRoundUp Platform
platform ConTagZ
n = (ConTagZ
n forall a. Num a => a -> a -> a
+ ConTagZ
word_size forall a. Num a => a -> a -> a
- ConTagZ
1) forall a. Integral a => a -> a -> a
`quot` ConTagZ
word_size
 where word_size :: ConTagZ
word_size = Platform -> ConTagZ
platformWordSizeInBytes Platform
platform
-- StgWord is a type representing an StgWord on the target platform.
-- A Word64 is large enough to hold a Word for either a 32bit or 64bit platform
newtype StgWord = StgWord Word64
    deriving (StgWord -> StgWord -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: StgWord -> StgWord -> Bool
$c/= :: StgWord -> StgWord -> Bool
== :: StgWord -> StgWord -> Bool
$c== :: StgWord -> StgWord -> Bool
Eq, Eq StgWord
StgWord
ConTagZ -> StgWord
StgWord -> Bool
StgWord -> ConTagZ
StgWord -> Maybe ConTagZ
StgWord -> StgWord
StgWord -> ConTagZ -> Bool
StgWord -> ConTagZ -> StgWord
StgWord -> StgWord -> StgWord
forall a.
Eq a
-> (a -> a -> a)
-> (a -> a -> a)
-> (a -> a -> a)
-> (a -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> a
-> (ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> Bool)
-> (a -> Maybe ConTagZ)
-> (a -> ConTagZ)
-> (a -> Bool)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ -> a)
-> (a -> ConTagZ)
-> Bits a
popCount :: StgWord -> ConTagZ
$cpopCount :: StgWord -> ConTagZ
rotateR :: StgWord -> ConTagZ -> StgWord
$crotateR :: StgWord -> ConTagZ -> StgWord
rotateL :: StgWord -> ConTagZ -> StgWord
$crotateL :: StgWord -> ConTagZ -> StgWord
unsafeShiftR :: StgWord -> ConTagZ -> StgWord
$cunsafeShiftR :: StgWord -> ConTagZ -> StgWord
shiftR :: StgWord -> ConTagZ -> StgWord
$cshiftR :: StgWord -> ConTagZ -> StgWord
unsafeShiftL :: StgWord -> ConTagZ -> StgWord
$cunsafeShiftL :: StgWord -> ConTagZ -> StgWord
shiftL :: StgWord -> ConTagZ -> StgWord
$cshiftL :: StgWord -> ConTagZ -> StgWord
isSigned :: StgWord -> Bool
$cisSigned :: StgWord -> Bool
bitSize :: StgWord -> ConTagZ
$cbitSize :: StgWord -> ConTagZ
bitSizeMaybe :: StgWord -> Maybe ConTagZ
$cbitSizeMaybe :: StgWord -> Maybe ConTagZ
testBit :: StgWord -> ConTagZ -> Bool
$ctestBit :: StgWord -> ConTagZ -> Bool
complementBit :: StgWord -> ConTagZ -> StgWord
$ccomplementBit :: StgWord -> ConTagZ -> StgWord
clearBit :: StgWord -> ConTagZ -> StgWord
$cclearBit :: StgWord -> ConTagZ -> StgWord
setBit :: StgWord -> ConTagZ -> StgWord
$csetBit :: StgWord -> ConTagZ -> StgWord
bit :: ConTagZ -> StgWord
$cbit :: ConTagZ -> StgWord
zeroBits :: StgWord
$czeroBits :: StgWord
rotate :: StgWord -> ConTagZ -> StgWord
$crotate :: StgWord -> ConTagZ -> StgWord
shift :: StgWord -> ConTagZ -> StgWord
$cshift :: StgWord -> ConTagZ -> StgWord
complement :: StgWord -> StgWord
$ccomplement :: StgWord -> StgWord
xor :: StgWord -> StgWord -> StgWord
$cxor :: StgWord -> StgWord -> StgWord
.|. :: StgWord -> StgWord -> StgWord
$c.|. :: StgWord -> StgWord -> StgWord
.&. :: StgWord -> StgWord -> StgWord
$c.&. :: StgWord -> StgWord -> StgWord
Bits)

fromStgWord :: StgWord -> Integer
fromStgWord :: StgWord -> Integer
fromStgWord (StgWord Word64
i) = forall a. Integral a => a -> Integer
toInteger Word64
i

toStgWord :: Platform -> Integer -> StgWord
toStgWord :: Platform -> Integer -> StgWord
toStgWord Platform
platform Integer
i
    = case Platform -> PlatformWordSize
platformWordSize Platform
platform of
      -- These conversions mean that things like toStgWord (-1)
      -- do the right thing
      PlatformWordSize
PW4 -> Word64 -> StgWord
StgWord (forall a b. (Integral a, Num b) => a -> b
fromIntegral (forall a. Num a => Integer -> a
fromInteger Integer
i :: Word32))
      PlatformWordSize
PW8 -> Word64 -> StgWord
StgWord (forall a. Num a => Integer -> a
fromInteger Integer
i)

instance Outputable StgWord where
    ppr :: StgWord -> SDoc
ppr (StgWord Word64
i) = Integer -> SDoc
integer (forall a. Integral a => a -> Integer
toInteger Word64
i)

--

-- A Word32 is large enough to hold half a Word for either a 32bit or
-- 64bit platform
newtype StgHalfWord = StgHalfWord Word32
    deriving StgHalfWord -> StgHalfWord -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: StgHalfWord -> StgHalfWord -> Bool
$c/= :: StgHalfWord -> StgHalfWord -> Bool
== :: StgHalfWord -> StgHalfWord -> Bool
$c== :: StgHalfWord -> StgHalfWord -> Bool
Eq

fromStgHalfWord :: StgHalfWord -> Integer
fromStgHalfWord :: StgHalfWord -> Integer
fromStgHalfWord (StgHalfWord Word32
w) = forall a. Integral a => a -> Integer
toInteger Word32
w

toStgHalfWord :: Platform -> Integer -> StgHalfWord
toStgHalfWord :: Platform -> Integer -> StgHalfWord
toStgHalfWord Platform
platform Integer
i
    = case Platform -> PlatformWordSize
platformWordSize Platform
platform of
      -- These conversions mean that things like toStgHalfWord (-1)
      -- do the right thing
      PlatformWordSize
PW4 -> Word32 -> StgHalfWord
StgHalfWord (forall a b. (Integral a, Num b) => a -> b
fromIntegral (forall a. Num a => Integer -> a
fromInteger Integer
i :: Word16))
      PlatformWordSize
PW8 -> Word32 -> StgHalfWord
StgHalfWord (forall a. Num a => Integer -> a
fromInteger Integer
i :: Word32)

instance Outputable StgHalfWord where
    ppr :: StgHalfWord -> SDoc
ppr (StgHalfWord Word32
w) = Integer -> SDoc
integer (forall a. Integral a => a -> Integer
toInteger Word32
w)

-- | Half word size in bytes
halfWordSize :: Platform -> ByteOff
halfWordSize :: Platform -> ConTagZ
halfWordSize Platform
platform = Platform -> ConTagZ
platformWordSizeInBytes Platform
platform forall a. Integral a => a -> a -> a
`div` ConTagZ
2

halfWordSizeInBits :: Platform -> Int
halfWordSizeInBits :: Platform -> ConTagZ
halfWordSizeInBits Platform
platform = Platform -> ConTagZ
platformWordSizeInBits Platform
platform forall a. Integral a => a -> a -> a
`div` ConTagZ
2

{-
************************************************************************
*                                                                      *
\subsubsection[SMRep-datatype]{@SMRep@---storage manager representation}
*                                                                      *
************************************************************************
-}

-- | A description of the layout of a closure.  Corresponds directly
-- to the closure types in includes\/rts\/storage\/ClosureTypes.h.
data SMRep
  = HeapRep              -- GC routines consult sizes in info tbl
        IsStatic
        !WordOff         --  # ptr words
        !WordOff         --  # non-ptr words INCLUDING SLOP (see mkHeapRep below)
        ClosureTypeInfo  -- type-specific info

  | ArrayPtrsRep
        !WordOff        -- # ptr words
        !WordOff        -- # card table words

  | SmallArrayPtrsRep
        !WordOff        -- # ptr words

  | ArrayWordsRep
        !WordOff        -- # bytes expressed in words, rounded up

  | StackRep            -- Stack frame (RET_SMALL or RET_BIG)
        Liveness

  | RTSRep              -- The RTS needs to declare info tables with specific
        Int             -- type tags, so this form lets us override the default
        SMRep           -- tag for an SMRep.

-- | True \<=> This is a static closure.  Affects how we garbage-collect it.
-- Static closure have an extra static link field at the end.
-- Constructors do not have a static variant; see Note [static constructors]
type IsStatic = Bool

-- From an SMRep you can get to the closure type defined in
-- includes/rts/storage/ClosureTypes.h. Described by the function
-- rtsClosureType below.

data ClosureTypeInfo
  = Constr        ConTagZ ConstrDescription
  | Fun           FunArity ArgDescr
  | Thunk
  | ThunkSelector SelectorOffset
  | BlackHole
  | IndStatic

type ConstrDescription = ByteString -- result of dataConIdentity
type FunArity          = Int
type SelectorOffset    = Int

-----------------------------------------------------------------------------
-- Construction

mkHeapRep :: Profile -> IsStatic -> WordOff -> WordOff -> ClosureTypeInfo
          -> SMRep
mkHeapRep :: Profile -> Bool -> ConTagZ -> ConTagZ -> ClosureTypeInfo -> SMRep
mkHeapRep Profile
profile Bool
is_static ConTagZ
ptr_wds ConTagZ
nonptr_wds ClosureTypeInfo
cl_type_info
  = Bool -> ConTagZ -> ConTagZ -> ClosureTypeInfo -> SMRep
HeapRep Bool
is_static
            ConTagZ
ptr_wds
            (ConTagZ
nonptr_wds forall a. Num a => a -> a -> a
+ ConTagZ
slop_wds)
            ClosureTypeInfo
cl_type_info
  where
     slop_wds :: ConTagZ
slop_wds
      | Bool
is_static = ConTagZ
0
      | Bool
otherwise = forall a. Ord a => a -> a -> a
max ConTagZ
0 (Profile -> ConTagZ
minClosureSize Profile
profile forall a. Num a => a -> a -> a
- (ConTagZ
hdr_size forall a. Num a => a -> a -> a
+ ConTagZ
payload_size))

     hdr_size :: ConTagZ
hdr_size     = Profile -> ClosureTypeInfo -> ConTagZ
closureTypeHdrSize Profile
profile ClosureTypeInfo
cl_type_info
     payload_size :: ConTagZ
payload_size = ConTagZ
ptr_wds forall a. Num a => a -> a -> a
+ ConTagZ
nonptr_wds

mkRTSRep :: Int -> SMRep -> SMRep
mkRTSRep :: ConTagZ -> SMRep -> SMRep
mkRTSRep = ConTagZ -> SMRep -> SMRep
RTSRep

mkStackRep :: [Bool] -> SMRep
mkStackRep :: [Bool] -> SMRep
mkStackRep [Bool]
liveness = [Bool] -> SMRep
StackRep [Bool]
liveness

blackHoleRep :: SMRep
blackHoleRep :: SMRep
blackHoleRep = Bool -> ConTagZ -> ConTagZ -> ClosureTypeInfo -> SMRep
HeapRep Bool
False ConTagZ
0 ConTagZ
0 ClosureTypeInfo
BlackHole

indStaticRep :: SMRep
indStaticRep :: SMRep
indStaticRep = Bool -> ConTagZ -> ConTagZ -> ClosureTypeInfo -> SMRep
HeapRep Bool
True ConTagZ
1 ConTagZ
0 ClosureTypeInfo
IndStatic

arrPtrsRep :: Platform -> WordOff -> SMRep
arrPtrsRep :: Platform -> ConTagZ -> SMRep
arrPtrsRep Platform
platform ConTagZ
elems = ConTagZ -> ConTagZ -> SMRep
ArrayPtrsRep ConTagZ
elems (Platform -> ConTagZ -> ConTagZ
cardTableSizeW Platform
platform ConTagZ
elems)

smallArrPtrsRep :: WordOff -> SMRep
smallArrPtrsRep :: ConTagZ -> SMRep
smallArrPtrsRep ConTagZ
elems = ConTagZ -> SMRep
SmallArrayPtrsRep ConTagZ
elems

arrWordsRep :: Platform -> ByteOff -> SMRep
arrWordsRep :: Platform -> ConTagZ -> SMRep
arrWordsRep Platform
platform ConTagZ
bytes = ConTagZ -> SMRep
ArrayWordsRep (Platform -> ConTagZ -> ConTagZ
bytesToWordsRoundUp Platform
platform ConTagZ
bytes)

-----------------------------------------------------------------------------
-- Predicates

isStaticRep :: SMRep -> IsStatic
isStaticRep :: SMRep -> Bool
isStaticRep (HeapRep Bool
is_static ConTagZ
_ ConTagZ
_ ClosureTypeInfo
_) = Bool
is_static
isStaticRep (RTSRep ConTagZ
_ SMRep
rep)            = SMRep -> Bool
isStaticRep SMRep
rep
isStaticRep SMRep
_                         = Bool
False

isStackRep :: SMRep -> Bool
isStackRep :: SMRep -> Bool
isStackRep StackRep{}     = Bool
True
isStackRep (RTSRep ConTagZ
_ SMRep
rep) = SMRep -> Bool
isStackRep SMRep
rep
isStackRep SMRep
_              = Bool
False

isConRep :: SMRep -> Bool
isConRep :: SMRep -> Bool
isConRep (HeapRep Bool
_ ConTagZ
_ ConTagZ
_ Constr{}) = Bool
True
isConRep SMRep
_                        = Bool
False

isThunkRep :: SMRep -> Bool
isThunkRep :: SMRep -> Bool
isThunkRep (HeapRep Bool
_ ConTagZ
_ ConTagZ
_ ClosureTypeInfo
Thunk)           = Bool
True
isThunkRep (HeapRep Bool
_ ConTagZ
_ ConTagZ
_ ThunkSelector{}) = Bool
True
isThunkRep (HeapRep Bool
_ ConTagZ
_ ConTagZ
_ ClosureTypeInfo
BlackHole)       = Bool
True
isThunkRep (HeapRep Bool
_ ConTagZ
_ ConTagZ
_ ClosureTypeInfo
IndStatic)       = Bool
True
isThunkRep SMRep
_                               = Bool
False

isFunRep :: SMRep -> Bool
isFunRep :: SMRep -> Bool
isFunRep (HeapRep Bool
_ ConTagZ
_ ConTagZ
_ Fun{}) = Bool
True
isFunRep SMRep
_                     = Bool
False

isStaticNoCafCon :: SMRep -> Bool
-- This should line up exactly with CONSTR_NOCAF below
-- See Note [Static NoCaf constructors]
isStaticNoCafCon :: SMRep -> Bool
isStaticNoCafCon (HeapRep Bool
_ ConTagZ
0 ConTagZ
_ Constr{}) = Bool
True
isStaticNoCafCon SMRep
_                        = Bool
False


-----------------------------------------------------------------------------
-- Size-related things

fixedHdrSize :: Profile -> ByteOff
fixedHdrSize :: Profile -> ConTagZ
fixedHdrSize Profile
profile = forall a. Num a => Platform -> a -> a
wordsToBytes (Profile -> Platform
profilePlatform Profile
profile) (Profile -> ConTagZ
fixedHdrSizeW Profile
profile)

-- | Size of a closure header (StgHeader in includes\/rts\/storage\/Closures.h)
fixedHdrSizeW :: Profile -> WordOff
fixedHdrSizeW :: Profile -> ConTagZ
fixedHdrSizeW Profile
profile = PlatformConstants -> ConTagZ
pc_STD_HDR_SIZE (Profile -> PlatformConstants
profileConstants Profile
profile) forall a. Num a => a -> a -> a
+ Profile -> ConTagZ
profHdrSize Profile
profile

-- | Size of the profiling part of a closure header
-- (StgProfHeader in includes\/rts\/storage\/Closures.h)
profHdrSize :: Profile -> WordOff
profHdrSize :: Profile -> ConTagZ
profHdrSize Profile
profile =
   if Profile -> Bool
profileIsProfiling Profile
profile
      then PlatformConstants -> ConTagZ
pc_PROF_HDR_SIZE (Profile -> PlatformConstants
profileConstants Profile
profile)
      else ConTagZ
0

-- | The garbage collector requires that every closure is at least as
--   big as this.
minClosureSize :: Profile -> WordOff
minClosureSize :: Profile -> ConTagZ
minClosureSize Profile
profile
 = Profile -> ConTagZ
fixedHdrSizeW Profile
profile
   forall a. Num a => a -> a -> a
+ PlatformConstants -> ConTagZ
pc_MIN_PAYLOAD_SIZE (Profile -> PlatformConstants
profileConstants Profile
profile)

arrWordsHdrSize :: Profile -> ByteOff
arrWordsHdrSize :: Profile -> ConTagZ
arrWordsHdrSize Profile
profile
 = Profile -> ConTagZ
fixedHdrSize Profile
profile
   forall a. Num a => a -> a -> a
+ PlatformConstants -> ConTagZ
pc_SIZEOF_StgArrBytes_NoHdr (Profile -> PlatformConstants
profileConstants Profile
profile)

arrWordsHdrSizeW :: Profile -> WordOff
arrWordsHdrSizeW :: Profile -> ConTagZ
arrWordsHdrSizeW Profile
profile
 = Profile -> ConTagZ
fixedHdrSizeW Profile
profile
   forall a. Num a => a -> a -> a
+ (PlatformConstants -> ConTagZ
pc_SIZEOF_StgArrBytes_NoHdr (Profile -> PlatformConstants
profileConstants Profile
profile) forall a. Integral a => a -> a -> a
`quot`
      Platform -> ConTagZ
platformWordSizeInBytes (Profile -> Platform
profilePlatform Profile
profile))

arrPtrsHdrSize :: Profile -> ByteOff
arrPtrsHdrSize :: Profile -> ConTagZ
arrPtrsHdrSize Profile
profile
 = Profile -> ConTagZ
fixedHdrSize Profile
profile
   forall a. Num a => a -> a -> a
+ PlatformConstants -> ConTagZ
pc_SIZEOF_StgMutArrPtrs_NoHdr (Profile -> PlatformConstants
profileConstants Profile
profile)

arrPtrsHdrSizeW :: Profile -> WordOff
arrPtrsHdrSizeW :: Profile -> ConTagZ
arrPtrsHdrSizeW Profile
profile
 = Profile -> ConTagZ
fixedHdrSizeW Profile
profile
   forall a. Num a => a -> a -> a
+ (PlatformConstants -> ConTagZ
pc_SIZEOF_StgMutArrPtrs_NoHdr (Profile -> PlatformConstants
profileConstants Profile
profile) forall a. Integral a => a -> a -> a
`quot`
      Platform -> ConTagZ
platformWordSizeInBytes (Profile -> Platform
profilePlatform Profile
profile))

smallArrPtrsHdrSize :: Profile -> ByteOff
smallArrPtrsHdrSize :: Profile -> ConTagZ
smallArrPtrsHdrSize Profile
profile
 = Profile -> ConTagZ
fixedHdrSize Profile
profile
   forall a. Num a => a -> a -> a
+ PlatformConstants -> ConTagZ
pc_SIZEOF_StgSmallMutArrPtrs_NoHdr (Profile -> PlatformConstants
profileConstants Profile
profile)

smallArrPtrsHdrSizeW :: Profile -> WordOff
smallArrPtrsHdrSizeW :: Profile -> ConTagZ
smallArrPtrsHdrSizeW Profile
profile
 = Profile -> ConTagZ
fixedHdrSizeW Profile
profile
   forall a. Num a => a -> a -> a
+ (PlatformConstants -> ConTagZ
pc_SIZEOF_StgSmallMutArrPtrs_NoHdr (Profile -> PlatformConstants
profileConstants Profile
profile) forall a. Integral a => a -> a -> a
`quot`
      Platform -> ConTagZ
platformWordSizeInBytes (Profile -> Platform
profilePlatform Profile
profile))

-- Thunks have an extra header word on SMP, so the update doesn't
-- splat the payload.
thunkHdrSize :: Profile -> WordOff
thunkHdrSize :: Profile -> ConTagZ
thunkHdrSize Profile
profile = Profile -> ConTagZ
fixedHdrSizeW Profile
profile forall a. Num a => a -> a -> a
+ ConTagZ
smp_hdr
        where
         platform :: Platform
platform = Profile -> Platform
profilePlatform Profile
profile
         smp_hdr :: ConTagZ
smp_hdr  = PlatformConstants -> ConTagZ
pc_SIZEOF_StgSMPThunkHeader (Platform -> PlatformConstants
platformConstants Platform
platform) forall a. Integral a => a -> a -> a
`quot`
                         Platform -> ConTagZ
platformWordSizeInBytes Platform
platform

hdrSize :: Profile -> SMRep -> ByteOff
hdrSize :: Profile -> SMRep -> ConTagZ
hdrSize Profile
profile SMRep
rep = forall a. Num a => Platform -> a -> a
wordsToBytes (Profile -> Platform
profilePlatform Profile
profile) (Profile -> SMRep -> ConTagZ
hdrSizeW Profile
profile SMRep
rep)

hdrSizeW :: Profile -> SMRep -> WordOff
hdrSizeW :: Profile -> SMRep -> ConTagZ
hdrSizeW Profile
profile (HeapRep Bool
_ ConTagZ
_ ConTagZ
_ ClosureTypeInfo
ty)    = Profile -> ClosureTypeInfo -> ConTagZ
closureTypeHdrSize Profile
profile ClosureTypeInfo
ty
hdrSizeW Profile
profile (ArrayPtrsRep ConTagZ
_ ConTagZ
_)    = Profile -> ConTagZ
arrPtrsHdrSizeW Profile
profile
hdrSizeW Profile
profile (SmallArrayPtrsRep ConTagZ
_) = Profile -> ConTagZ
smallArrPtrsHdrSizeW Profile
profile
hdrSizeW Profile
profile (ArrayWordsRep ConTagZ
_)     = Profile -> ConTagZ
arrWordsHdrSizeW Profile
profile
hdrSizeW Profile
_ SMRep
_                           = forall a. String -> a
panic String
"GHC.Runtime.Heap.Layout.hdrSizeW"

nonHdrSize :: Platform -> SMRep -> ByteOff
nonHdrSize :: Platform -> SMRep -> ConTagZ
nonHdrSize Platform
platform SMRep
rep = forall a. Num a => Platform -> a -> a
wordsToBytes Platform
platform (SMRep -> ConTagZ
nonHdrSizeW SMRep
rep)

nonHdrSizeW :: SMRep -> WordOff
nonHdrSizeW :: SMRep -> ConTagZ
nonHdrSizeW (HeapRep Bool
_ ConTagZ
p ConTagZ
np ClosureTypeInfo
_) = ConTagZ
p forall a. Num a => a -> a -> a
+ ConTagZ
np
nonHdrSizeW (ArrayPtrsRep ConTagZ
elems ConTagZ
ct) = ConTagZ
elems forall a. Num a => a -> a -> a
+ ConTagZ
ct
nonHdrSizeW (SmallArrayPtrsRep ConTagZ
elems) = ConTagZ
elems
nonHdrSizeW (ArrayWordsRep ConTagZ
words) = ConTagZ
words
nonHdrSizeW (StackRep [Bool]
bs)      = forall (t :: * -> *) a. Foldable t => t a -> ConTagZ
length [Bool]
bs
nonHdrSizeW (RTSRep ConTagZ
_ SMRep
rep)     = SMRep -> ConTagZ
nonHdrSizeW SMRep
rep

-- | The total size of the closure, in words.
heapClosureSizeW :: Profile -> SMRep -> WordOff
heapClosureSizeW :: Profile -> SMRep -> ConTagZ
heapClosureSizeW Profile
profile SMRep
rep = case SMRep
rep of
   HeapRep Bool
_ ConTagZ
p ConTagZ
np ClosureTypeInfo
ty       -> Profile -> ClosureTypeInfo -> ConTagZ
closureTypeHdrSize Profile
profile ClosureTypeInfo
ty forall a. Num a => a -> a -> a
+ ConTagZ
p forall a. Num a => a -> a -> a
+ ConTagZ
np
   ArrayPtrsRep ConTagZ
elems ConTagZ
ct   -> Profile -> ConTagZ
arrPtrsHdrSizeW Profile
profile forall a. Num a => a -> a -> a
+ ConTagZ
elems forall a. Num a => a -> a -> a
+ ConTagZ
ct
   SmallArrayPtrsRep ConTagZ
elems -> Profile -> ConTagZ
smallArrPtrsHdrSizeW Profile
profile forall a. Num a => a -> a -> a
+ ConTagZ
elems
   ArrayWordsRep ConTagZ
words     -> Profile -> ConTagZ
arrWordsHdrSizeW Profile
profile forall a. Num a => a -> a -> a
+ ConTagZ
words
   SMRep
_                       -> forall a. String -> a
panic String
"GHC.Runtime.Heap.Layout.heapClosureSize"

closureTypeHdrSize :: Profile -> ClosureTypeInfo -> WordOff
closureTypeHdrSize :: Profile -> ClosureTypeInfo -> ConTagZ
closureTypeHdrSize Profile
profile ClosureTypeInfo
ty = case ClosureTypeInfo
ty of
                  ClosureTypeInfo
Thunk           -> Profile -> ConTagZ
thunkHdrSize Profile
profile
                  ThunkSelector{} -> Profile -> ConTagZ
thunkHdrSize Profile
profile
                  ClosureTypeInfo
BlackHole       -> Profile -> ConTagZ
thunkHdrSize Profile
profile
                  ClosureTypeInfo
IndStatic       -> Profile -> ConTagZ
thunkHdrSize Profile
profile
                  ClosureTypeInfo
_               -> Profile -> ConTagZ
fixedHdrSizeW Profile
profile
        -- All thunks use thunkHdrSize, even if they are non-updatable.
        -- this is because we don't have separate closure types for
        -- updatable vs. non-updatable thunks, so the GC can't tell the
        -- difference.  If we ever have significant numbers of non-
        -- updatable thunks, it might be worth fixing this.

-- ---------------------------------------------------------------------------
-- Arrays

-- | The byte offset into the card table of the card for a given element
card :: Platform -> Int -> Int
card :: Platform -> ConTagZ -> ConTagZ
card Platform
platform ConTagZ
i = ConTagZ
i forall a. Bits a => a -> ConTagZ -> a
`shiftR` PlatformConstants -> ConTagZ
pc_MUT_ARR_PTRS_CARD_BITS (Platform -> PlatformConstants
platformConstants Platform
platform)

-- | Convert a number of elements to a number of cards, rounding up
cardRoundUp :: Platform -> Int -> Int
cardRoundUp :: Platform -> ConTagZ -> ConTagZ
cardRoundUp Platform
platform ConTagZ
i =
  Platform -> ConTagZ -> ConTagZ
card Platform
platform (ConTagZ
i forall a. Num a => a -> a -> a
+ ((ConTagZ
1 forall a. Bits a => a -> ConTagZ -> a
`shiftL` PlatformConstants -> ConTagZ
pc_MUT_ARR_PTRS_CARD_BITS (Platform -> PlatformConstants
platformConstants Platform
platform)) forall a. Num a => a -> a -> a
- ConTagZ
1))

-- | The size of a card table, in bytes
cardTableSizeB :: Platform -> Int -> ByteOff
cardTableSizeB :: Platform -> ConTagZ -> ConTagZ
cardTableSizeB Platform
platform ConTagZ
elems = Platform -> ConTagZ -> ConTagZ
cardRoundUp Platform
platform ConTagZ
elems

-- | The size of a card table, in words
cardTableSizeW :: Platform -> Int -> WordOff
cardTableSizeW :: Platform -> ConTagZ -> ConTagZ
cardTableSizeW Platform
platform ConTagZ
elems =
  Platform -> ConTagZ -> ConTagZ
bytesToWordsRoundUp Platform
platform (Platform -> ConTagZ -> ConTagZ
cardTableSizeB Platform
platform ConTagZ
elems)

-----------------------------------------------------------------------------
-- deriving the RTS closure type from an SMRep

#include "../includes/rts/storage/ClosureTypes.h"
#include "../includes/rts/storage/FunTypes.h"
-- Defines CONSTR, CONSTR_1_0 etc

-- | Derives the RTS closure type from an 'SMRep'
rtsClosureType :: SMRep -> Int
rtsClosureType :: SMRep -> ConTagZ
rtsClosureType SMRep
rep
    = case SMRep
rep of
      RTSRep ConTagZ
ty SMRep
_ -> ConTagZ
ty

      -- See Note [static constructors]
      HeapRep Bool
_     ConTagZ
1 ConTagZ
0 Constr{} -> CONSTR_1_0
      HeapRep Bool
_     ConTagZ
0 ConTagZ
1 Constr{} -> CONSTR_0_1
      HeapRep Bool
_     ConTagZ
2 ConTagZ
0 Constr{} -> CONSTR_2_0
      HeapRep Bool
_     ConTagZ
1 ConTagZ
1 Constr{} -> CONSTR_1_1
      HeapRep Bool
_     ConTagZ
0 ConTagZ
2 Constr{} -> CONSTR_0_2
      HeapRep Bool
_     ConTagZ
0 ConTagZ
_ Constr{} -> CONSTR_NOCAF
           -- See Note [Static NoCaf constructors]
      HeapRep Bool
_     ConTagZ
_ ConTagZ
_ Constr{} -> CONSTR

      HeapRep Bool
False ConTagZ
1 ConTagZ
0 Fun{} -> FUN_1_0
      HeapRep Bool
False ConTagZ
0 ConTagZ
1 Fun{} -> FUN_0_1
      HeapRep Bool
False ConTagZ
2 ConTagZ
0 Fun{} -> FUN_2_0
      HeapRep Bool
False ConTagZ
1 ConTagZ
1 Fun{} -> FUN_1_1
      HeapRep Bool
False ConTagZ
0 ConTagZ
2 Fun{} -> FUN_0_2
      HeapRep Bool
False ConTagZ
_ ConTagZ
_ Fun{} -> FUN

      HeapRep Bool
False ConTagZ
1 ConTagZ
0 ClosureTypeInfo
Thunk -> THUNK_1_0
      HeapRep Bool
False ConTagZ
0 ConTagZ
1 ClosureTypeInfo
Thunk -> THUNK_0_1
      HeapRep Bool
False ConTagZ
2 ConTagZ
0 ClosureTypeInfo
Thunk -> THUNK_2_0
      HeapRep Bool
False ConTagZ
1 ConTagZ
1 ClosureTypeInfo
Thunk -> THUNK_1_1
      HeapRep Bool
False ConTagZ
0 ConTagZ
2 ClosureTypeInfo
Thunk -> THUNK_0_2
      HeapRep Bool
False ConTagZ
_ ConTagZ
_ ClosureTypeInfo
Thunk -> THUNK

      HeapRep Bool
False ConTagZ
_ ConTagZ
_ ThunkSelector{} ->  THUNK_SELECTOR

      HeapRep Bool
True ConTagZ
_ ConTagZ
_ Fun{}      -> FUN_STATIC
      HeapRep Bool
True ConTagZ
_ ConTagZ
_ ClosureTypeInfo
Thunk      -> THUNK_STATIC
      HeapRep Bool
False ConTagZ
_ ConTagZ
_ ClosureTypeInfo
BlackHole -> BLACKHOLE
      HeapRep Bool
False ConTagZ
_ ConTagZ
_ ClosureTypeInfo
IndStatic -> IND_STATIC

      SMRep
_ -> forall a. String -> a
panic String
"rtsClosureType"

-- We export these ones
rET_SMALL, rET_BIG, aRG_GEN, aRG_GEN_BIG :: Int
rET_SMALL :: ConTagZ
rET_SMALL   = RET_SMALL
rET_BIG :: ConTagZ
rET_BIG     = RET_BIG
aRG_GEN :: ConTagZ
aRG_GEN     = ARG_GEN
aRG_GEN_BIG :: ConTagZ
aRG_GEN_BIG = ARG_GEN_BIG

{-
Note [static constructors]
~~~~~~~~~~~~~~~~~~~~~~~~~~

We used to have a CONSTR_STATIC closure type, and each constructor had
two info tables: one with CONSTR (or CONSTR_1_0 etc.), and one with
CONSTR_STATIC.

This distinction was removed, because when copying a data structure
into a compact region, we must copy static constructors into the
compact region too.  If we didn't do this, we would need to track the
references from the compact region out to the static constructors,
because they might (indirectly) refer to CAFs.

Since static constructors will be copied to the heap, if we wanted to
use different info tables for static and dynamic constructors, we
would have to switch the info pointer when copying the constructor
into the compact region, which means we would need an extra field of
the static info table to point to the dynamic one.

However, since the distinction between static and dynamic closure
types is never actually needed (other than for assertions), we can
just drop the distinction and use the same info table for both.

The GC *does* need to distinguish between static and dynamic closures,
but it does this using the HEAP_ALLOCED() macro which checks whether
the address of the closure resides within the dynamic heap.
HEAP_ALLOCED() doesn't read the closure's info table.

Note [Static NoCaf constructors]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
If we know that a top-level binding 'x' is not Caffy (ie no CAFs are
reachable from 'x'), then a statically allocated constructor (Just x)
is also not Caffy, and the garbage collector need not follow its
argument fields.  Exploiting this would require two static info tables
for Just, for the two cases where the argument was Caffy or non-Caffy.

Currently we don't do this; instead we treat nullary constructors
as non-Caffy, and the others as potentially Caffy.


************************************************************************
*                                                                      *
             Pretty printing of SMRep and friends
*                                                                      *
************************************************************************
-}

instance Outputable ClosureTypeInfo where
   ppr :: ClosureTypeInfo -> SDoc
ppr = ClosureTypeInfo -> SDoc
pprTypeInfo

instance Outputable SMRep where
   ppr :: SMRep -> SDoc
ppr (HeapRep Bool
static ConTagZ
ps ConTagZ
nps ClosureTypeInfo
tyinfo)
     = SDoc -> ConTagZ -> SDoc -> SDoc
hang (SDoc
header SDoc -> SDoc -> SDoc
<+> SDoc
lbrace) ConTagZ
2 (forall a. Outputable a => a -> SDoc
ppr ClosureTypeInfo
tyinfo SDoc -> SDoc -> SDoc
<+> SDoc
rbrace)
     where
       header :: SDoc
header = String -> SDoc
text String
"HeapRep"
                SDoc -> SDoc -> SDoc
<+> if Bool
static then String -> SDoc
text String
"static" else SDoc
empty
                SDoc -> SDoc -> SDoc
<+> String -> ConTagZ -> SDoc
pp_n String
"ptrs" ConTagZ
ps SDoc -> SDoc -> SDoc
<+> String -> ConTagZ -> SDoc
pp_n String
"nonptrs" ConTagZ
nps
       pp_n :: String -> Int -> SDoc
       pp_n :: String -> ConTagZ -> SDoc
pp_n String
_ ConTagZ
0 = SDoc
empty
       pp_n String
s ConTagZ
n = ConTagZ -> SDoc
int ConTagZ
n SDoc -> SDoc -> SDoc
<+> String -> SDoc
text String
s

   ppr (ArrayPtrsRep ConTagZ
size ConTagZ
_) = String -> SDoc
text String
"ArrayPtrsRep" SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr ConTagZ
size

   ppr (SmallArrayPtrsRep ConTagZ
size) = String -> SDoc
text String
"SmallArrayPtrsRep" SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr ConTagZ
size

   ppr (ArrayWordsRep ConTagZ
words) = String -> SDoc
text String
"ArrayWordsRep" SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr ConTagZ
words

   ppr (StackRep [Bool]
bs) = String -> SDoc
text String
"StackRep" SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr [Bool]
bs

   ppr (RTSRep ConTagZ
ty SMRep
rep) = String -> SDoc
text String
"tag:" SDoc -> SDoc -> SDoc
<> forall a. Outputable a => a -> SDoc
ppr ConTagZ
ty SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr SMRep
rep

pprTypeInfo :: ClosureTypeInfo -> SDoc
pprTypeInfo :: ClosureTypeInfo -> SDoc
pprTypeInfo (Constr ConTagZ
tag ConstrDescription
descr)
  = String -> SDoc
text String
"Con" SDoc -> SDoc -> SDoc
<+>
    SDoc -> SDoc
braces ([SDoc] -> SDoc
sep [ String -> SDoc
text String
"tag:" SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr ConTagZ
tag
                , String -> SDoc
text String
"descr:" SDoc -> SDoc -> SDoc
<> String -> SDoc
text (forall a. Show a => a -> String
show ConstrDescription
descr) ])

pprTypeInfo (Fun ConTagZ
arity ArgDescr
args)
  = String -> SDoc
text String
"Fun" SDoc -> SDoc -> SDoc
<+>
    SDoc -> SDoc
braces ([SDoc] -> SDoc
sep [ String -> SDoc
text String
"arity:" SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr ConTagZ
arity
                , PtrString -> SDoc
ptext (String -> PtrString
sLit (String
"fun_type:")) SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr ArgDescr
args ])

pprTypeInfo (ThunkSelector ConTagZ
offset)
  = String -> SDoc
text String
"ThunkSel" SDoc -> SDoc -> SDoc
<+> forall a. Outputable a => a -> SDoc
ppr ConTagZ
offset

pprTypeInfo ClosureTypeInfo
Thunk     = String -> SDoc
text String
"Thunk"
pprTypeInfo ClosureTypeInfo
BlackHole = String -> SDoc
text String
"BlackHole"
pprTypeInfo ClosureTypeInfo
IndStatic = String -> SDoc
text String
"IndStatic"