module Language.Parser.Ptera.Data.Alignable.Array where

import           Language.Parser.Ptera.Prelude

import qualified Data.Array                               as DataArray
import qualified Language.Parser.Ptera.Data.Alignable     as Alignable
import qualified Language.Parser.Ptera.Data.Alignable.Map as AlignableMap


type T = Array

newtype Array n a = Array (DataArray.Array Int a)
    deriving (Array n a -> Array n a -> Bool
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
forall k (n :: k) a. Eq a => Array n a -> Array n a -> Bool
/= :: Array n a -> Array n a -> Bool
$c/= :: forall k (n :: k) a. Eq a => Array n a -> Array n a -> Bool
== :: Array n a -> Array n a -> Bool
$c== :: forall k (n :: k) a. Eq a => Array n a -> Array n a -> Bool
Eq, Int -> Array n a -> ShowS
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
forall k (n :: k) a. Show a => Int -> Array n a -> ShowS
forall k (n :: k) a. Show a => [Array n a] -> ShowS
forall k (n :: k) a. Show a => Array n a -> String
showList :: [Array n a] -> ShowS
$cshowList :: forall k (n :: k) a. Show a => [Array n a] -> ShowS
show :: Array n a -> String
$cshow :: forall k (n :: k) a. Show a => Array n a -> String
showsPrec :: Int -> Array n a -> ShowS
$cshowsPrec :: forall k (n :: k) a. Show a => Int -> Array n a -> ShowS
Show, forall k (n :: k) a b. a -> Array n b -> Array n a
forall k (n :: k) a b. (a -> b) -> Array n a -> Array n b
forall a b. a -> Array n b -> Array n a
forall a b. (a -> b) -> Array n a -> Array n b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
<$ :: forall a b. a -> Array n b -> Array n a
$c<$ :: forall k (n :: k) a b. a -> Array n b -> Array n a
fmap :: forall a b. (a -> b) -> Array n a -> Array n b
$cfmap :: forall k (n :: k) a b. (a -> b) -> Array n a -> Array n b
Functor, forall a. Array n a -> Bool
forall k (n :: k) a. Eq a => a -> Array n a -> Bool
forall k (n :: k) a. Num a => Array n a -> a
forall k (n :: k) a. Ord a => Array n a -> a
forall k (n :: k) m. Monoid m => Array n m -> m
forall k (n :: k) a. Array n a -> Bool
forall k (n :: k) a. Array n a -> Int
forall k (n :: k) a. Array n a -> [a]
forall k (n :: k) a. (a -> a -> a) -> Array n a -> a
forall k (n :: k) m a. Monoid m => (a -> m) -> Array n a -> m
forall k (n :: k) b a. (b -> a -> b) -> b -> Array n a -> b
forall k (n :: k) a b. (a -> b -> b) -> b -> Array n a -> b
forall m a. Monoid m => (a -> m) -> Array n a -> m
forall a b. (a -> b -> b) -> b -> Array n a -> b
forall (t :: * -> *).
(forall m. Monoid m => t m -> m)
-> (forall m a. Monoid m => (a -> m) -> t a -> m)
-> (forall m a. Monoid m => (a -> m) -> t a -> m)
-> (forall a b. (a -> b -> b) -> b -> t a -> b)
-> (forall a b. (a -> b -> b) -> b -> t a -> b)
-> (forall b a. (b -> a -> b) -> b -> t a -> b)
-> (forall b a. (b -> a -> b) -> b -> t a -> b)
-> (forall a. (a -> a -> a) -> t a -> a)
-> (forall a. (a -> a -> a) -> t a -> a)
-> (forall a. t a -> [a])
-> (forall a. t a -> Bool)
-> (forall a. t a -> Int)
-> (forall a. Eq a => a -> t a -> Bool)
-> (forall a. Ord a => t a -> a)
-> (forall a. Ord a => t a -> a)
-> (forall a. Num a => t a -> a)
-> (forall a. Num a => t a -> a)
-> Foldable t
product :: forall a. Num a => Array n a -> a
$cproduct :: forall k (n :: k) a. Num a => Array n a -> a
sum :: forall a. Num a => Array n a -> a
$csum :: forall k (n :: k) a. Num a => Array n a -> a
minimum :: forall a. Ord a => Array n a -> a
$cminimum :: forall k (n :: k) a. Ord a => Array n a -> a
maximum :: forall a. Ord a => Array n a -> a
$cmaximum :: forall k (n :: k) a. Ord a => Array n a -> a
elem :: forall a. Eq a => a -> Array n a -> Bool
$celem :: forall k (n :: k) a. Eq a => a -> Array n a -> Bool
length :: forall a. Array n a -> Int
$clength :: forall k (n :: k) a. Array n a -> Int
null :: forall a. Array n a -> Bool
$cnull :: forall k (n :: k) a. Array n a -> Bool
toList :: forall a. Array n a -> [a]
$ctoList :: forall k (n :: k) a. Array n a -> [a]
foldl1 :: forall a. (a -> a -> a) -> Array n a -> a
$cfoldl1 :: forall k (n :: k) a. (a -> a -> a) -> Array n a -> a
foldr1 :: forall a. (a -> a -> a) -> Array n a -> a
$cfoldr1 :: forall k (n :: k) a. (a -> a -> a) -> Array n a -> a
foldl' :: forall b a. (b -> a -> b) -> b -> Array n a -> b
$cfoldl' :: forall k (n :: k) b a. (b -> a -> b) -> b -> Array n a -> b
foldl :: forall b a. (b -> a -> b) -> b -> Array n a -> b
$cfoldl :: forall k (n :: k) b a. (b -> a -> b) -> b -> Array n a -> b
foldr' :: forall a b. (a -> b -> b) -> b -> Array n a -> b
$cfoldr' :: forall k (n :: k) a b. (a -> b -> b) -> b -> Array n a -> b
foldr :: forall a b. (a -> b -> b) -> b -> Array n a -> b
$cfoldr :: forall k (n :: k) a b. (a -> b -> b) -> b -> Array n a -> b
foldMap' :: forall m a. Monoid m => (a -> m) -> Array n a -> m
$cfoldMap' :: forall k (n :: k) m a. Monoid m => (a -> m) -> Array n a -> m
foldMap :: forall m a. Monoid m => (a -> m) -> Array n a -> m
$cfoldMap :: forall k (n :: k) m a. Monoid m => (a -> m) -> Array n a -> m
fold :: forall m. Monoid m => Array n m -> m
$cfold :: forall k (n :: k) m. Monoid m => Array n m -> m
Foldable)

fromTotalMap :: Alignable.T n => n -> AlignableMap.T n a -> Array n a
fromTotalMap :: forall n a. T n => n -> T n a -> Array n a
fromTotalMap n
b T n a
m = forall {k} (n :: k) a. Array Int a -> Array n a
Array
    do forall i e. Ix i => (i, i) -> [(i, e)] -> Array i e
DataArray.array (Int
0, forall a. Enum a => a -> a
pred do coerce :: forall a b. Coercible a b => a -> b
coerce n
b) do coerce :: forall a b. Coercible a b => a -> b
coerce do forall n a. T n => Map n a -> [(n, a)]
AlignableMap.toAscList T n a
m

fromList :: forall n a. Alignable.T n => [a] -> Array n a
fromList :: forall n a. T n => [a] -> Array n a
fromList [a]
xs = forall {k} (n :: k) a. Array Int a -> Array n a
Array
    do forall i e. Ix i => (i, i) -> [e] -> Array i e
DataArray.listArray (Int
0, forall (t :: * -> *) a. Foldable t => t a -> Int
length [a]
xs forall a. Num a => a -> a -> a
- Int
1) [a]
xs

mapWithIx :: Alignable.T n => (n -> a -> a) -> Array n a -> Array n a
mapWithIx :: forall n a. T n => (n -> a -> a) -> Array n a -> Array n a
mapWithIx n -> a -> a
f (Array Array Int a
arr) = forall {k} (n :: k) a. Array Int a -> Array n a
Array
    do forall i e. Ix i => (i, i) -> [e] -> Array i e
DataArray.listArray
        do forall i e. Array i e -> (i, i)
DataArray.bounds Array Int a
arr
        do [ n -> a -> a
f (coerce :: forall a b. Coercible a b => a -> b
coerce Int
i) a
x | (Int
i, a
x) <- forall i e. Ix i => Array i e -> [(i, e)]
DataArray.assocs Array Int a
arr ]

index :: forall n a. Alignable.T n => Array n a -> n -> Maybe a
index :: forall n a. T n => Array n a -> n -> Maybe a
index Array n a
arr n
i = case forall i. Alignable i => i -> Int
Alignable.numIncrements n
i forall a. Ord a => a -> a -> Bool
< forall (t :: * -> *) a. Foldable t => t a -> Int
length Array n a
arr of
    Bool
False -> forall a. Maybe a
Nothing
    Bool
True  -> forall a. a -> Maybe a
Just do forall n a. T n => Array n a -> n -> a
forceIndex Array n a
arr n
i

forceIndex :: forall n a. Alignable.T n => Array n a -> n -> a
forceIndex :: forall n a. T n => Array n a -> n -> a
forceIndex = coerce :: forall a b. Coercible a b => a -> b
coerce do forall i e. Ix i => Array i e -> i -> e
(DataArray.!) @Int @a

assocs :: forall n a. Alignable.T n => Array n a -> [(n, a)]
assocs :: forall n a. T n => Array n a -> [(n, a)]
assocs = coerce :: forall a b. Coercible a b => a -> b
coerce do forall i e. Ix i => Array i e -> [(i, e)]
DataArray.assocs @Int @a