{-# language BangPatterns #-}
{-# language DataKinds #-}
{-# language MagicHash #-}
{-# language TypeFamilies #-}
{-# language DuplicateRecordFields #-}

module Data.Bytes.Types
  ( Bytes(..)
  , Bytes#(..)
  , MutableBytes(..)
  , UnmanagedBytes(..)
  , BytesN(..)
  , ByteArrayN(..)
  ) where

import Data.Bytes.Internal (Bytes(..))
import Data.Primitive (ByteArray(..),MutableByteArray(..))
import Data.Primitive.Addr (Addr)
import GHC.TypeNats (Nat)
import UnliftedBytes (Bytes#(..))

-- | A slice of a 'ByteArray' whose compile-time-known length is represented
-- by a phantom type variable. Consumers of this data constructor must be
-- careful to preserve the expected invariant.
data BytesN (n :: Nat) = BytesN
  { BytesN n -> ByteArray
array :: {-# UNPACK #-} !ByteArray
  , BytesN n -> Int
offset :: {-# UNPACK #-} !Int
  }

-- | A 'ByteArray' whose compile-time-known length is represented
-- by a phantom type variable. Consumers of this data constructor must be
-- careful to preserve the expected invariant.
newtype ByteArrayN (n :: Nat) = ByteArrayN
  { ByteArrayN n -> ByteArray
array :: ByteArray
  }

-- | A slice of a 'MutableByteArray'.
data MutableBytes s = MutableBytes
  { MutableBytes s -> MutableByteArray s
array :: {-# UNPACK #-} !(MutableByteArray s)
  , MutableBytes s -> Int
offset :: {-# UNPACK #-} !Int
  , MutableBytes s -> Int
length :: {-# UNPACK #-} !Int
  }

-- | A slice of unmanaged memory.
data UnmanagedBytes = UnmanagedBytes
  { UnmanagedBytes -> Addr
address :: {-# UNPACK #-} !Addr
  , UnmanagedBytes -> Int
length :: {-# UNPACK #-} !Int
  }