diff options
| author | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2025-12-01 01:44:05 +0100 |
|---|---|---|
| committer | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2025-12-01 16:06:57 +0100 |
| commit | 9560d0f26420409afd2230fb7e5e111eafcced06 (patch) | |
| tree | 5f64f9adfc521143bdf86be3128364aee4679c55 /src/Data/Array/Nested/Ranked.hs | |
| parent | a06c6416bab1639e5c3bd99b3c10de4dcf6c32f9 (diff) | |
Expose the unfolding of the indexing operations
Diffstat (limited to 'src/Data/Array/Nested/Ranked.hs')
| -rw-r--r-- | src/Data/Array/Nested/Ranked.hs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/Data/Array/Nested/Ranked.hs b/src/Data/Array/Nested/Ranked.hs index 2fbfdd8..bf35cc4 100644 --- a/src/Data/Array/Nested/Ranked.hs +++ b/src/Data/Array/Nested/Ranked.hs @@ -49,9 +49,11 @@ remptyArray = mtoRanked (memptyArray ZSX) rsize :: Elt a => Ranked n a -> Int rsize = shrSize . rshape +{-# INLINEABLE rindex #-} rindex :: Elt a => Ranked n a -> IIxR n -> a rindex (Ranked arr) idx = mindex arr (ixxFromIxR idx) +{-# INLINEABLE rindexPartial #-} rindexPartial :: forall n m a. Elt a => Ranked (n + m) a -> IIxR n -> Ranked m a rindexPartial (Ranked arr) idx = Ranked (mindexPartial @a @(Replicate n Nothing) @(Replicate m Nothing) |