-
-
Save lehins/a83a5373a959d988f866b4f67fce82ae to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
{-# LANGUAGE GHCForeignImportPrim #-} | |
{-# LANGUAGE MagicHash #-} | |
{-# LANGUAGE UnboxedTuples #-} | |
{-# LANGUAGE UnliftedFFITypes #-} | |
module MutInt | |
( MutInt, | |
newMutInt, | |
getMutInt, | |
putMutInt, | |
) | |
where | |
import GHC.Exts | |
import GHC.Types | |
newtype MutInt | |
= MutInt Any | |
newMutInt :: IO MutInt | |
newMutInt = IO $ \s0 -> case stg_newMutInt# s0 of | |
(# s1, c #) -> (# s1, MutInt c #) | |
getMutInt :: MutInt -> IO Int | |
getMutInt (MutInt c) = IO $ \s0 -> case stg_getMutInt# s0 c of | |
(# s1, x #) -> (# s1, I# x #) | |
putMutInt :: MutInt -> Int -> IO () | |
putMutInt (MutInt c) (I# x) = IO $ \s0 -> case stg_putMutInt# s0 c x of | |
(# s1 #) -> (# s1, () #) | |
foreign import prim "stg_newMutIntzh" | |
stg_newMutInt# :: State# s -> (# State# s, Any #) | |
foreign import prim "stg_getMutIntzh" | |
stg_getMutInt# :: State# s -> Any -> (# State# s, Int# #) | |
foreign import prim "stg_putMutIntzh" | |
stg_putMutInt# :: State# s -> Any -> Int# -> (# State# s #) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#include "Cmm.h" | |
#define SIZEOF_StgMutInt (SIZEOF_StgHeader + WDS(1)) | |
INFO_TABLE(stg_MutInt, 0, 1, PRIM, "MutInt", "MutInt") | |
() | |
{ | |
return (); | |
} | |
stg_newMutIntzh () | |
{ | |
P_ c; | |
ALLOC_PRIM_ (SIZEOF_StgMutInt, stg_newMutIntzh); | |
c = Hp - SIZEOF_StgMutInt + WDS(1); | |
SET_HDR(c,stg_MutInt_info,CCCS); | |
return (c); | |
} | |
stg_getMutIntzh (P_ c) | |
{ | |
return (W_[c + SIZEOF_StgHeader]); | |
} | |
stg_putMutIntzh (P_ c, W_ x) | |
{ | |
W_[c + SIZEOF_StgHeader] = x; | |
return (); | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment