{-# LANGUAGE DuplicateRecordFields #-}
module IR.Flat
    ( File(..)
    , CodeGenReq(..)
    , Node(..)
    , Node'(..)
    , Field(..)
    , Method(..)
    , Variant(..)
    , Union(..)
    ) where

import Data.Word

import qualified IR.Common as Common
import qualified IR.Name   as Name

type Brand = Common.ListBrand Node

data CodeGenReq = CodeGenReq
    { allNodes :: [Node]
    , reqFiles :: [File]
    }
    deriving(Show, Eq)

data File = File
    { nodes    :: [Node]
    , fileId   :: !Word64
    , fileName :: FilePath
    }
    deriving(Show, Eq)

data Node = Node
    { name       :: Name.CapnpQ
    , nodeId     :: !Word64
    , union_     :: Node'
    , typeParams :: [Common.TypeParamRef Node]
    }
    deriving(Show, Eq)

data Node'
    = Enum [Name.UnQ]
    | Struct
        { fields        :: [Field]
        -- ^ The struct's fields, excluding an anonymous union, if any.
        , isGroup       :: !Bool
        , dataWordCount :: !Word16
        , pointerCount  :: !Word16
        , union         :: Maybe Union
        -- ^ The struct's anonymous union, if any.
        }
    | Interface
        { methods :: [Method]
        , supers  :: [Common.InterfaceType Brand Node]
        }
    | Constant
        { value :: Common.Value Brand Node
        }
    | Other
    deriving(Show, Eq)


data Method = Method
    { name       :: Name.UnQ
    , paramType  :: Common.CompositeType Brand Node
    , resultType :: Common.CompositeType Brand Node
    }
    deriving(Show, Eq)


data Union = Union
    { tagOffset :: !Word32
    , variants  :: [Variant]
    }
    deriving(Show, Eq)

data Field = Field
    { fieldName    :: Name.CapnpQ
    , fieldLocType :: Common.FieldLocType Brand Node
    }
    deriving(Show, Eq)

data Variant = Variant
    { tagValue :: !Word16
    , field    :: Field
    -- ^ The field's name is really the name of the variant.
    }
    deriving(Show, Eq)