diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2024-09-11 13:07:33 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2024-09-11 13:07:47 +0200 |
| commit | b728b22248414c8319681a75f1c8e8cdf8da1fb2 (patch) | |
| tree | a17c1fe8b8d0cd9502d76262c7675e91413ea555 /src/Interpreter/Array.hs | |
| parent | 8942db88390ece8b1cf018ad723fedc6ae82cf64 (diff) | |
More WIP interpreter
Diffstat (limited to 'src/Interpreter/Array.hs')
| -rw-r--r-- | src/Interpreter/Array.hs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/Interpreter/Array.hs b/src/Interpreter/Array.hs index f358225..54e0791 100644 --- a/src/Interpreter/Array.hs +++ b/src/Interpreter/Array.hs @@ -34,6 +34,9 @@ arrayShape (Array sh _) = sh arraySize :: Array n t -> Int arraySize (Array sh _) = shapeSize sh +arrayIndexLinear :: Array n t -> Int -> t +arrayIndexLinear (Array _ v) i = v V.! i + arrayGenerateLinM :: Monad m => Shape n -> (Int -> m t) -> m (Array n t) arrayGenerateLinM sh f = Array sh <$> V.generateM (shapeSize sh) f |