aboutsummaryrefslogtreecommitdiff
path: root/src/Interpreter/Rep.hs
diff options
context:
space:
mode:
authorTom Smeding <tom@tomsmeding.com>2025-11-10 21:49:45 +0100
committerTom Smeding <tom@tomsmeding.com>2025-11-10 21:50:25 +0100
commit174af2ba568de66e0d890825b8bda930b8e7bb96 (patch)
tree5a20f52662e87ff7cf6a6bef5db0713aa6c7884e /src/Interpreter/Rep.hs
parent92bca235e3aaa287286b6af082d3fce585825a35 (diff)
Move module hierarchy under CHAD.
Diffstat (limited to 'src/Interpreter/Rep.hs')
-rw-r--r--src/Interpreter/Rep.hs105
1 files changed, 0 insertions, 105 deletions
diff --git a/src/Interpreter/Rep.hs b/src/Interpreter/Rep.hs
deleted file mode 100644
index 1682303..0000000
--- a/src/Interpreter/Rep.hs
+++ /dev/null
@@ -1,105 +0,0 @@
-{-# LANGUAGE DataKinds #-}
-{-# LANGUAGE ScopedTypeVariables #-}
-{-# LANGUAGE TypeApplications #-}
-{-# LANGUAGE TypeFamilies #-}
-{-# LANGUAGE UndecidableInstances #-}
-module Interpreter.Rep where
-
-import Control.DeepSeq
-import Data.Coerce (coerce)
-import Data.List (intersperse, intercalate)
-import Data.Foldable (toList)
-import Data.IORef
-import GHC.Exts (withDict)
-
-import Array
-import AST
-import AST.Pretty
-import Data
-
-
-type family Rep t where
- Rep TNil = ()
- Rep (TPair a b) = (Rep a, Rep b)
- Rep (TEither a b) = Either (Rep a) (Rep b)
- Rep (TLEither a b) = Maybe (Either (Rep a) (Rep b))
- Rep (TMaybe t) = Maybe (Rep t)
- Rep (TArr n t) = Array n (Rep t)
- Rep (TScal sty) = ScalRep sty
- Rep (TAccum t) = RepAc t
-
--- Mutable, represents monoid types t.
-type family RepAc t where
- RepAc TNil = ()
- RepAc (TPair a b) = (RepAc a, RepAc b)
- RepAc (TLEither a b) = IORef (Maybe (Either (RepAc a) (RepAc b)))
- RepAc (TMaybe t) = IORef (Maybe (RepAc t))
- RepAc (TArr n t) = Array n (RepAc t)
- RepAc (TScal sty) = IORef (ScalRep sty)
-
-newtype Value t = Value { unValue :: Rep t }
-
-liftV :: (Rep a -> Rep b) -> Value a -> Value b
-liftV f (Value x) = Value (f x)
-
-liftV2 :: (Rep a -> Rep b -> Rep c) -> Value a -> Value b -> Value c
-liftV2 f (Value x) (Value y) = Value (f x y)
-
-vPair :: Value a -> Value b -> Value (TPair a b)
-vPair = liftV2 (,)
-
-vUnpair :: Value (TPair a b) -> (Value a, Value b)
-vUnpair (Value (x, y)) = (Value x, Value y)
-
-showValue :: Int -> STy t -> Rep t -> ShowS
-showValue _ STNil () = showString "()"
-showValue _ (STPair a b) (x, y) = showString "(" . showValue 0 a x . showString "," . showValue 0 b y . showString ")"
-showValue d (STEither a _) (Left x) = showParen (d > 10) $ showString "Inl " . showValue 11 a x
-showValue d (STEither _ b) (Right y) = showParen (d > 10) $ showString "Inr " . showValue 11 b y
-showValue _ (STLEither _ _) Nothing = showString "LNil"
-showValue d (STLEither a _) (Just (Left x)) = showParen (d > 10) $ showString "LInl " . showValue 11 a x
-showValue d (STLEither _ b) (Just (Right y)) = showParen (d > 10) $ showString "LInr " . showValue 11 b y
-showValue _ (STMaybe _) Nothing = showString "Nothing"
-showValue d (STMaybe t) (Just x) = showParen (d > 10) $ showString "Just " . showValue 11 t x
-showValue d (STArr _ t) arr = showParen (d > 10) $
- showString "arrayFromList " . showsPrec 11 (arrayShape arr)
- . showString " ["
- . foldr (.) id (intersperse (showString ",") $ map (showValue 0 t) (toList arr))
- . showString "]"
-showValue d (STScal sty) x = case sty of
- STF32 -> showsPrec d x
- STF64 -> showsPrec d x
- STI32 -> showsPrec d x
- STI64 -> showsPrec d x
- STBool -> showsPrec d x
-showValue _ (STAccum t) _ = showString $ "<accumulator for " ++ ppSMTy 0 t ++ ">"
-
-showEnv :: SList STy env -> SList Value env -> String
-showEnv = \env vals -> "[" ++ intercalate ", " (showEntries env vals) ++ "]"
- where
- showEntries :: SList STy env -> SList Value env -> [String]
- showEntries SNil SNil = []
- showEntries (t `SCons` env) (Value x `SCons` xs) = showValue 0 t x "" : showEntries env xs
-
-rnfRep :: STy t -> Rep t -> ()
-rnfRep STNil () = ()
-rnfRep (STPair a b) (x, y) = rnfRep a x `seq` rnfRep b y
-rnfRep (STEither a _) (Left x) = rnfRep a x
-rnfRep (STEither _ b) (Right y) = rnfRep b y
-rnfRep (STLEither _ _) Nothing = ()
-rnfRep (STLEither a _) (Just (Left x)) = rnfRep a x
-rnfRep (STLEither _ b) (Just (Right y)) = rnfRep b y
-rnfRep (STMaybe _) Nothing = ()
-rnfRep (STMaybe t) (Just x) = rnfRep t x
-rnfRep (STArr (_ :: SNat n) (t :: STy t2)) arr =
- withDict @(KnownTy t2) t $ rnf (coerce @(Array n (Rep t2)) @(Array n (Value t2)) arr)
-rnfRep (STScal t) x = case t of
- STI32 -> rnf x
- STI64 -> rnf x
- STF32 -> rnf x
- STF64 -> rnf x
- STBool -> rnf x
-rnfRep STAccum{} _ = error "Cannot rnf accumulators"
-
-instance KnownTy t => NFData (Value t) where
- rnf (Value x) = rnfRep (knownTy @t) x