luau/prototyping/FFI/Data/Vector.agda
2022-02-08 18:34:41 -06:00

47 lines
1.7 KiB
Agda

module FFI.Data.Vector where
open import Agda.Builtin.Int using (Int; pos; negsuc)
open import Agda.Builtin.Nat using (Nat)
open import FFI.Data.Bool using (Bool; false; true)
open import FFI.Data.HaskellInt using (HaskellInt; haskellIntToInt; intToHaskellInt)
open import FFI.Data.Maybe using (Maybe; just; nothing)
{-# FOREIGN GHC import qualified Data.Vector #-}
postulate Vector : Set Set
{-# POLARITY Vector ++ #-}
{-# COMPILE GHC Vector = type Data.Vector.Vector #-}
postulate
empty : {A} (Vector A)
null : {A} (Vector A) Bool
unsafeHead : {A} (Vector A) A
unsafeTail : {A} (Vector A) (Vector A)
hLength : {A} (Vector A) HaskellInt
hLookup : {A} (Vector A) HaskellInt (Maybe A)
snoc : {A} (Vector A) A (Vector A)
{-# COMPILE GHC empty = \_ -> Data.Vector.empty #-}
{-# COMPILE GHC null = \_ -> Data.Vector.null #-}
{-# COMPILE GHC unsafeHead = \_ -> Data.Vector.unsafeHead #-}
{-# COMPILE GHC unsafeTail = \_ -> Data.Vector.unsafeTail #-}
{-# COMPILE GHC hLength = \_ -> Data.Vector.length #-}
{-# COMPILE GHC hLookup = \_ -> (Data.Vector.!?) #-}
{-# COMPILE GHC snoc = \_ -> Data.Vector.snoc #-}
head : {A} (Vector A) (Maybe A)
head vec with null vec
head vec | false = just (unsafeHead vec)
head vec | true = nothing
tail : {A} (Vector A) Vector A
tail vec with null vec
tail vec | false = unsafeTail vec
tail vec | true = empty
length : {A} (Vector A) Nat
length vec with haskellIntToInt(hLength vec)
length vec | pos n = n
length vec | negsuc n = 0
lookup : {A} (Vector A) Nat Maybe A
lookup vec n = hLookup vec (intToHaskellInt (pos n))