{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE UndecidableInstances #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}
module Terminal.Game.Layer.Object.IO where
import Terminal.Game.Utils
import Terminal.Game.Layer.Object.Interface
import Terminal.Game.Layer.Object.Primitive
import Terminal.Game.Plane
import qualified Control.Concurrent as CC
import qualified Control.Monad as CM
import qualified Control.Monad.Catch as MC
import qualified Control.Monad.Trans as T
import qualified Data.List.Split as LS
import qualified System.Clock as SC
import qualified System.Console.ANSI as CA
import qualified System.Console.Terminal.Size as TS
import qualified System.IO as SI
instance {-# OVERLAPS #-} (Monad m, T.MonadIO m) => MonadInput m where
startEvents :: Integer -> m InputHandle
startEvents Integer
tps = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO forall a b. (a -> b) -> a -> b
$ Integer -> IO InputHandle
startIOInput Integer
tps
pollEvents :: MVar [Event] -> m [Event]
pollEvents MVar [Event]
ve = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO forall a b. (a -> b) -> a -> b
$ forall a. MVar a -> a -> IO a
CC.swapMVar MVar [Event]
ve []
stopEvents :: [ThreadId] -> m ()
stopEvents [ThreadId]
ts = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO forall a b. (a -> b) -> a -> b
$ [ThreadId] -> IO ()
stopEventsIO [ThreadId]
ts
areEventsOver :: m Bool
areEventsOver = forall (m :: * -> *) a. Monad m => a -> m a
return Bool
False
startIOInput :: TPS -> IO InputHandle
startIOInput :: Integer -> IO InputHandle
startIOInput Integer
tps =
Handle -> BufferMode -> IO ()
SI.hSetBuffering Handle
SI.stdin BufferMode
SI.NoBuffering forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
Handle -> BufferMode -> IO ()
SI.hSetBuffering Handle
SI.stdout BufferMode
SI.NoBuffering forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
Handle -> Bool -> IO ()
SI.hSetEcho Handle
SI.stdin Bool
False forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
forall a. a -> IO (MVar a)
CC.newMVar [] forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \MVar [Event]
ve ->
Integer -> IO Integer
getTimeTick Integer
tps forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \Integer
it ->
IO () -> IO ThreadId
CC.forkIO (MVar [Event] -> Integer -> Integer -> IO ()
addTick MVar [Event]
ve Integer
tps Integer
it) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \ThreadId
te ->
IO () -> IO ThreadId
CC.forkIO (MVar [Event] -> IO ()
addKeypress MVar [Event]
ve) forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \ThreadId
tk ->
forall (m :: * -> *) a. Monad m => a -> m a
return (MVar [Event] -> [ThreadId] -> InputHandle
InputHandle MVar [Event]
ve [ThreadId
te, ThreadId
tk])
type Elapsed = Integer
getTimeTick :: TPS -> IO Elapsed
getTimeTick :: Integer -> IO Integer
getTimeTick Integer
tps =
forall (m :: * -> *). MonadTimer m => m Integer
getTime forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \Integer
tm ->
let ns :: Integer
ns = Integer
10 forall a b. (Num a, Integral b) => a -> b -> a
^ (Integer
9 :: Integer)
t1 :: Integer
t1 = forall a. Integral a => a -> a -> a
quot Integer
ns Integer
tps in
forall (m :: * -> *) a. Monad m => a -> m a
return (forall a. Integral a => a -> a -> a
quot Integer
tm Integer
t1)
addTick :: CC.MVar [Event] -> TPS -> Elapsed -> IO ()
addTick :: MVar [Event] -> Integer -> Integer -> IO ()
addTick MVar [Event]
ve Integer
tps Integer
el =
Integer -> IO Integer
getTimeTick Integer
tps forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \Integer
t ->
forall (m :: * -> *) a. Applicative m => Int -> m a -> m ()
CM.replicateM_ (forall a b. (Integral a, Num b) => a -> b
fromIntegral forall a b. (a -> b) -> a -> b
$ Integer
tforall a. Num a => a -> a -> a
-Integer
el)
(MVar [Event] -> Event -> IO ()
addEvent MVar [Event]
ve Event
Tick) forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
forall (m :: * -> *). MonadTimer m => Integer -> m ()
sleepABit Integer
tps forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
MVar [Event] -> Integer -> Integer -> IO ()
addTick MVar [Event]
ve Integer
tps Integer
t
addKeypress :: CC.MVar [Event] -> IO ()
addKeypress :: MVar [Event] -> IO ()
addKeypress MVar [Event]
ve =
IO Char
inputCharTerminal forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \Char
c ->
MVar [Event] -> Event -> IO ()
addEvent MVar [Event]
ve (Char -> Event
KeyPress Char
c) forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
MVar [Event] -> IO ()
addKeypress MVar [Event]
ve
addEvent :: CC.MVar [Event] -> Event -> IO ()
addEvent :: MVar [Event] -> Event -> IO ()
addEvent MVar [Event]
ve Event
e = MVar [Event] -> IO ()
vf MVar [Event]
ve
where
vf :: MVar [Event] -> IO ()
vf MVar [Event]
d = forall a. MVar a -> (a -> IO a) -> IO ()
CC.modifyMVar_ MVar [Event]
d (forall (m :: * -> *) a. Monad m => a -> m a
return forall b c a. (b -> c) -> (a -> b) -> a -> c
. (forall a. [a] -> [a] -> [a]
++[Event
e]))
stopEventsIO :: [CC.ThreadId] -> IO ()
stopEventsIO :: [ThreadId] -> IO ()
stopEventsIO [ThreadId]
ts = forall (t :: * -> *) (m :: * -> *) a b.
(Foldable t, Monad m) =>
(a -> m b) -> t a -> m ()
mapM_ ThreadId -> IO ()
CC.killThread [ThreadId]
ts
instance {-# OVERLAPS #-} (Monad m, T.MonadIO m) => MonadTimer m where
getTime :: m Integer
getTime = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO forall a b. (a -> b) -> a -> b
$ TimeSpec -> Integer
SC.toNanoSecs forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Clock -> IO TimeSpec
SC.getTime Clock
SC.Monotonic
sleepABit :: Integer -> m ()
sleepABit Integer
tps = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO forall a b. (a -> b) -> a -> b
$
Int -> IO ()
CC.threadDelay (forall a b. (Integral a, Num b) => a -> b
fromIntegral forall a b. (a -> b) -> a -> b
$ forall a. Integral a => a -> a -> a
quot Integer
oneTickSec (Integer
tpsforall a. Num a => a -> a -> a
*Integer
10))
instance {-# OVERLAPS #-}
(Monad m, T.MonadIO m, MC.MonadMask m, MC.MonadThrow m) =>
MonadException m where
cleanUpErr :: forall a b. m a -> m b -> m a
cleanUpErr m a
m m b
c = forall (m :: * -> *) a b. MonadMask m => m a -> m b -> m a
MC.finally m a
m m b
c
throwExc :: forall a. ATGException -> m a
throwExc ATGException
t = forall (m :: * -> *) e a. (MonadThrow m, Exception e) => e -> m a
MC.throwM ATGException
t
instance {-# OVERLAPS #-} (Monad m, T.MonadIO m) => MonadDisplay m where
setupDisplay :: m ()
setupDisplay = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO IO ()
initPart
clearDisplay :: m ()
clearDisplay = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO IO ()
clearScreen
displaySize :: m (Maybe Dimensions)
displaySize = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO IO (Maybe Dimensions)
displaySizeIO
blitPlane :: Maybe Plane -> Plane -> m ()
blitPlane Maybe Plane
mp Plane
p = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO (Maybe Plane -> Plane -> IO ()
blitPlaneIO Maybe Plane
mp Plane
p)
shutdownDisplay :: m ()
shutdownDisplay = forall (m :: * -> *) a. MonadIO m => IO a -> m a
T.liftIO IO ()
cleanAndExit
displaySizeIO :: IO (Maybe Dimensions)
displaySizeIO :: IO (Maybe Dimensions)
displaySizeIO =
forall n. Integral n => IO (Maybe (Window n))
TS.size forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \Maybe (Window Int)
ts ->
IO Bool
isWin32Console forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \Bool
bw ->
forall (m :: * -> *) a. Monad m => a -> m a
return (forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (Bool -> Window Int -> Dimensions
f Bool
bw) Maybe (Window Int)
ts)
where
f :: Bool -> TS.Window Int -> Dimensions
f :: Bool -> Window Int -> Dimensions
f Bool
wbw (TS.Window Int
h Int
w) =
let h' :: Int
h' | Bool
wbw = Int
h forall a. Num a => a -> a -> a
- Int
1
| Bool
otherwise = Int
h
in (Int
w, Int
h')
blitPlaneIO :: Maybe Plane -> Plane -> IO ()
blitPlaneIO :: Maybe Plane -> Plane -> IO ()
blitPlaneIO Maybe Plane
mpo Plane
pn =
let
(Int
pw, Int
ph) = Plane -> Dimensions
planeSize Plane
pn
bp :: Plane
bp = Int -> Int -> Plane
blankPlane Int
pw Int
ph
po :: Plane
po = Plane -> Plane -> Dimensions -> Plane
pastePlane (forall b a. b -> (a -> b) -> Maybe a -> b
maybe Plane
bp forall a. a -> a
id Maybe Plane
mpo) Plane
bp (Int
1, Int
1)
in
let pn' :: Plane
pn' = Plane -> Plane -> Dimensions -> Plane
pastePlane Plane
pn Plane
bp (Int
1, Int
1)
in
[SGR] -> IO ()
CA.setSGR [SGR
CA.Reset] forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
Plane -> Plane -> IO ()
blitMap Plane
po Plane
pn'
initPart :: IO ()
initPart :: IO ()
initPart =
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
CM.unless Bool
CC.rtsSupportsBoundThreads
(forall a. HasCallStack => [Char] -> a
error [Char]
errMes) forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
IO ()
CA.hideCursor forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
[Char] -> IO TextEncoding
SI.mkTextEncoding [Char]
"UTF-8//TRANSLIT" forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \TextEncoding
te ->
Handle -> TextEncoding -> IO ()
SI.hSetEncoding Handle
SI.stdout TextEncoding
te forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
IO ()
clearScreen
where
errMes :: [Char]
errMes = [[Char]] -> [Char]
unlines
[[Char]
"\nError: you *must* compile this program with -threaded!",
[Char]
"Just add",
[Char]
"",
[Char]
" ghc-options: -threaded",
[Char]
"",
[Char]
"in your .cabal file (executable section) and you will be fine!"]
clearScreen :: IO ()
clearScreen :: IO ()
clearScreen = Int -> Int -> IO ()
CA.setCursorPosition Int
0 Int
0 forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
[SGR] -> IO ()
CA.setSGR [SGR
CA.Reset] forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
forall (m :: * -> *).
(MonadDisplay m, MonadException m) =>
m Dimensions
displaySizeErr forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \(Int
w, Int
h) ->
forall (m :: * -> *) a. Applicative m => Int -> m a -> m ()
CM.replicateM_ (forall a b. (Integral a, Num b) => a -> b
fromIntegral forall a b. (a -> b) -> a -> b
$ Int
wforall a. Num a => a -> a -> a
*Int
h) (Char -> IO ()
putChar Char
' ')
cleanAndExit :: IO ()
cleanAndExit :: IO ()
cleanAndExit = [SGR] -> IO ()
CA.setSGR [SGR
CA.Reset] forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
IO ()
CA.clearScreen forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
Int -> Int -> IO ()
CA.setCursorPosition Int
0 Int
0 forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
IO ()
CA.showCursor
blitMap :: Plane -> Plane -> IO ()
blitMap :: Plane -> Plane -> IO ()
blitMap Plane
po Plane
pn =
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
CM.when (Plane -> Dimensions
planeSize Plane
po forall a. Eq a => a -> a -> Bool
/= Plane -> Dimensions
planeSize Plane
pn)
(forall a. HasCallStack => [Char] -> a
error [Char]
"blitMap: different plane sizes") forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
Int -> Int -> IO ()
CA.setCursorPosition Int
0 Int
0 forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
Dimensions -> [[Cell]] -> [[Cell]] -> IO ()
blitToTerminal (Int
0, Int
0) (Plane -> [[Cell]]
orderedCells Plane
po) (Plane -> [[Cell]]
orderedCells Plane
pn)
orderedCells :: Plane -> [[Cell]]
orderedCells :: Plane -> [[Cell]]
orderedCells Plane
p = forall e. Int -> [e] -> [[e]]
LS.chunksOf (forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
w) [Cell]
cells
where
cells :: [Cell]
cells = forall a b. (a -> b) -> [a] -> [b]
map forall a b. (a, b) -> b
snd forall a b. (a -> b) -> a -> b
$ Plane -> [(Dimensions, Cell)]
assocsPlane Plane
p
(Int
w, Int
_) = Plane -> Dimensions
planeSize Plane
p
blitToTerminal :: Coords -> [[Cell]] -> [[Cell]] -> IO ()
blitToTerminal :: Dimensions -> [[Cell]] -> [[Cell]] -> IO ()
blitToTerminal (Int
rr, Int
rc) [[Cell]]
ocs [[Cell]]
ncs = forall (t :: * -> *) (m :: * -> *) b a.
(Foldable t, Monad m) =>
(b -> a -> m b) -> b -> t a -> m ()
CM.foldM_ Int -> [(Cell, Cell)] -> IO Int
blitLine Int
rr [[(Cell, Cell)]]
oldNew
where
oldNew :: [[(Cell, Cell)]]
oldNew :: [[(Cell, Cell)]]
oldNew = forall a b c. (a -> b -> c) -> [a] -> [b] -> [c]
zipWith forall a b. [a] -> [b] -> [(a, b)]
zip [[Cell]]
ocs [[Cell]]
ncs
blitLine :: Row -> [(Cell, Cell)] -> IO Row
blitLine :: Int -> [(Cell, Cell)] -> IO Int
blitLine Int
pr [(Cell, Cell)]
ccs =
forall (t :: * -> *) (m :: * -> *) b a.
(Foldable t, Monad m) =>
(b -> a -> m b) -> b -> t a -> m ()
CM.foldM_ Int -> (Cell, Cell) -> IO Int
blitCell Int
0 [(Cell, Cell)]
ccs forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
let wr :: Int
wr = Int
pr forall a. Num a => a -> a -> a
+ Int
1 in
Int -> Int -> IO ()
CA.setCursorPosition (forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
wr)
(forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
rc) forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
forall (m :: * -> *) a. Monad m => a -> m a
return Int
wr
blitCell :: Int -> (Cell, Cell) -> IO Int
blitCell :: Int -> (Cell, Cell) -> IO Int
blitCell Int
k (Cell
clo, Cell
cln)
| Cell
cln forall a. Eq a => a -> a -> Bool
== Cell
clo = forall (m :: * -> *) a. Monad m => a -> m a
return (Int
kforall a. Num a => a -> a -> a
+Int
1)
| Bool
otherwise = Int -> IO Int
moveIf Int
k forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \Int
k' ->
Cell -> IO ()
putCellStyle Cell
cln forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
forall (m :: * -> *) a. Monad m => a -> m a
return Int
k'
moveIf :: Int -> IO Int
moveIf :: Int -> IO Int
moveIf Int
k | Int
k forall a. Eq a => a -> a -> Bool
== Int
0 = forall (m :: * -> *) a. Monad m => a -> m a
return Int
k
| Bool
otherwise = Int -> IO ()
CA.cursorForward Int
k forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
forall (m :: * -> *) a. Monad m => a -> m a
return Int
0
putCellStyle :: Cell -> IO ()
putCellStyle :: Cell -> IO ()
putCellStyle Cell
c = [SGR] -> IO ()
CA.setSGR ([SGR
CA.Reset] forall a. [a] -> [a] -> [a]
++ [SGR]
sgrb forall a. [a] -> [a] -> [a]
++ [SGR]
sgrr forall a. [a] -> [a] -> [a]
++ [SGR]
sgrc) forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>>
Char -> IO ()
putChar (Cell -> Char
cellChar Cell
c)
where
sgrb :: [SGR]
sgrb | Cell -> Bool
isBold Cell
c = [ConsoleIntensity -> SGR
CA.SetConsoleIntensity ConsoleIntensity
CA.BoldIntensity]
| Bool
otherwise = []
sgrr :: [SGR]
sgrr | Cell -> Bool
isReversed Cell
c = [Bool -> SGR
CA.SetSwapForegroundBackground Bool
True]
| Bool
otherwise = []
sgrc :: [SGR]
sgrc | Just (ANSIColorInfo (Color
k, ColorIntensity
i)) <- Cell -> Maybe ColorInfo
cellColor Cell
c = [ConsoleLayer -> ColorIntensity -> Color -> SGR
CA.SetColor ConsoleLayer
CA.Foreground ColorIntensity
i Color
k]
| Just (RGBColorInfo Colour Float
k) <- Cell -> Maybe ColorInfo
cellColor Cell
c = [ConsoleLayer -> Colour Float -> SGR
CA.SetRGBColor ConsoleLayer
CA.Foreground Colour Float
k]
| Just (PaletteColorInfo Word8
k) <- Cell -> Maybe ColorInfo
cellColor Cell
c = [ConsoleLayer -> Word8 -> SGR
CA.SetPaletteColor ConsoleLayer
CA.Foreground Word8
k]
| Bool
otherwise = []
oneTickSec :: Integer
oneTickSec :: Integer
oneTickSec = Integer
10 forall a b. (Num a, Integral b) => a -> b -> a
^ (Integer
6 :: Integer)