toki/numidr
1
0
Fork 0
Code Issues Pull requests Activity

Add operator form for unsafe indexing

Browse source
This commit is contained in:
Kiana Sheibani 2022-08-04 15:13:00 -04:00
parent 1ebf0dcbe9
commit 7916e10aef
Signed by: toki
GPG key ID: 6CB106C25E86A9F7
3 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/Data/NumIdr/Array/Array.idr
View file

@ -243,6 +243,7 @@ array v = MkArray COrder (calcStrides COrder s) s (fromList $ collapse v)
infixl 10 !!
infixl 10 !?
infixl 10 !#
infixl 11 !!..
infixl 11 !?..
@ -395,6 +396,7 @@ export
elements : Array {rk} s a -> Vect (product s) a
elements (MkArray _ sts sh p) =
let elems = map (flip index p . getLocation' sts) (getAllCoords' sh)
-- TODO: prove that the number of elements is `product s`
in assert_total $ case toVect (product sh) elems of Just v => v