haskell-effectful/effectful
1
1
Fork 0
mirror of https://github.com/haskell-effectful/effectful.git synced 2024-11-24 07:14:04 +03:00
Code Issues Projects Releases Wiki Activity

Slight improvement to getLocation

Browse Source
This commit is contained in:
Andrzej Rybczak 2022-03-09 16:51:23 +01:00
parent 3890b0a600
commit ce7112a309
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
effectful-core/src/Effectful/Internal/Env.hs
View File

@ -525,11 +525,11 @@ getLocation ix (Env fork size ref _) = do
-- specific effects for reinterpretation. -- specific effects for reinterpretation.
case fork of case fork of
NoFork -> pure $ Location i es NoFork -> pure $ Location i es
Forks _ baseIx lref forks -> do Forks _ baseIx lref forks
| i >= baseIx -> do
EnvRef _ les _ <- readIORef lref EnvRef _ les _ <- readIORef lref
if i >= baseIx pure $ Location (i - baseIx) les
then pure $ Location (i - baseIx) les | otherwise -> go es# i forks
else go es# i forks
where where
go :: SmallMutableArray# RealWorld Any go :: SmallMutableArray# RealWorld Any
-> Int -> Int
@ -537,8 +537,8 @@ getLocation ix (Env fork size ref _) = do
-> IO Location -> IO Location
go es# i = \case go es# i = \case
NoFork -> pure $ Location i (SmallMutableArray es#) NoFork -> pure $ Location i (SmallMutableArray es#)
Forks _ baseIx lref forks -> do Forks _ baseIx lref forks
| i >= baseIx -> do
EnvRef _ les _ <- readIORef lref EnvRef _ les _ <- readIORef lref
if i >= baseIx pure $ Location (i - baseIx) les
then pure $ Location (i - baseIx) les | otherwise -> go es# i forks
else go es# i forks