mirror of
https://github.com/idris-lang/Idris2.git
synced 2024-12-23 19:54:50 +03:00
dbdf7dab3d
Following a fairly detailed discussion on slack, the feeling is generally that it's better to have a single interface. While precision is nice, it doesn't appear to buy us anything here. If that turns out to be wrong, or limiting somehow, we can revisit it later. Also: - it's easier for backend authors if the type of IO operations is slightly less restrictive. For example, if it's in HasIO, that limits alternative implementations, which might be awkward for some alternative back ends. - it's one less extra detail to learn. This is minor, but there needs to be a clear advantage if there's more detail to learn. - It is difficult to think of an underlying type that can't have a Monad instance (I have personally never encountered one - if they turns out to exist, again, we can revisit!)
81 lines
2.4 KiB
Idris
81 lines
2.4 KiB
Idris
module Data.IOArray
|
|
|
|
import Data.IOArray.Prims
|
|
import Data.List
|
|
|
|
export
|
|
record IOArray elem where
|
|
constructor MkIOArray
|
|
maxSize : Int
|
|
content : ArrayData (Maybe elem)
|
|
|
|
export
|
|
max : IOArray elem -> Int
|
|
max = maxSize
|
|
|
|
export
|
|
newArray : HasIO io => Int -> io (IOArray elem)
|
|
newArray size
|
|
= pure (MkIOArray size !(primIO (prim__newArray size Nothing)))
|
|
|
|
export
|
|
writeArray : HasIO io => IOArray elem -> Int -> elem -> io ()
|
|
writeArray arr pos el
|
|
= if pos < 0 || pos >= max arr
|
|
then pure ()
|
|
else primIO (prim__arraySet (content arr) pos (Just el))
|
|
|
|
export
|
|
readArray : HasIO io => IOArray elem -> Int -> io (Maybe elem)
|
|
readArray arr pos
|
|
= if pos < 0 || pos >= max arr
|
|
then pure Nothing
|
|
else primIO (prim__arrayGet (content arr) pos)
|
|
|
|
-- Make a new array of the given size with the elements copied from the
|
|
-- other array
|
|
export
|
|
newArrayCopy : HasIO io =>
|
|
(newsize : Int) -> IOArray elem -> io (IOArray elem)
|
|
newArrayCopy newsize arr
|
|
= do let newsize' = if newsize < max arr then max arr else newsize
|
|
arr' <- newArray newsize'
|
|
copyFrom (content arr) (content arr') (max arr - 1)
|
|
pure arr'
|
|
where
|
|
copyFrom : ArrayData (Maybe elem) ->
|
|
ArrayData (Maybe elem) ->
|
|
Int -> io ()
|
|
copyFrom old new pos
|
|
= if pos < 0
|
|
then pure ()
|
|
else do el <- primIO $ prim__arrayGet old pos
|
|
primIO $ prim__arraySet new pos el
|
|
assert_total (copyFrom old new (pos - 1))
|
|
|
|
export
|
|
toList : HasIO io => IOArray elem -> io (List (Maybe elem))
|
|
toList arr = iter 0 (max arr) []
|
|
where
|
|
iter : Int -> Int -> List (Maybe elem) -> io (List (Maybe elem))
|
|
iter pos end acc
|
|
= if pos >= end
|
|
then pure (reverse acc)
|
|
else do el <- readArray arr pos
|
|
assert_total (iter (pos + 1) end (el :: acc))
|
|
|
|
export
|
|
fromList : List (Maybe elem) -> IO (IOArray elem)
|
|
fromList ns
|
|
= do arr <- newArray (cast (length ns))
|
|
addToArray 0 ns arr
|
|
pure arr
|
|
where
|
|
addToArray : Int -> List (Maybe elem) -> IOArray elem -> IO ()
|
|
addToArray loc [] arr = pure ()
|
|
addToArray loc (Nothing :: ns) arr
|
|
= assert_total (addToArray (loc + 1) ns arr)
|
|
addToArray loc (Just el :: ns) arr
|
|
= do primIO $ prim__arraySet (content arr) loc (Just el)
|
|
assert_total (addToArray (loc + 1) ns arr)
|