Shim out writeCharArray# when bounds checking is on

This commit is contained in:
Andrew Martin 2022-08-01 16:43:59 -04:00
parent 3e8e438109
commit 43a2049168
3 changed files with 10 additions and 1 deletions

View file

@ -1,8 +1,12 @@
{-# language MagicHash #-}
{-# language UnboxedTuples #-}
module Op
( writeCharArray#
) where
import GHC.Exts ((<#),(>=#))
import GHC.Int (Int(I#))
import qualified GHC.Exts as Exts
writeCharArray# :: Exts.MutableByteArray# s -> Exts.Int# -> Exts.Char# -> Exts.State# s -> Exts.State# s
@ -11,4 +15,4 @@ writeCharArray# arr i v st = case i <# 0# of
_ -> case Exts.getSizeofMutableByteArray# arr st of
(# st', sz #) -> case i >=# sz of
1# -> error ("writeCharArray#: index " ++ show (I# i) ++ " >= length " ++ show (I# sz))
_ -> Exts.writeInt8Array# arr i (Exts.intToInt8# v) st'
_ -> Exts.writeCharArray# arr i v st'