X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2Futils%2FPrimPacked.lhs;h=f2d034dcee619b86e0b92fd5fb78763130cdaf86;hb=2a2efb720c0fdc06fe749f96f284b00b30f8f3f7;hp=f7f0201b6ee8ab99b1945e38c832da3564a62bff;hpb=4be8b8b3051b020ef813176b75c21d11ebaab1d9;p=ghc-hetmet.git diff --git a/ghc/compiler/utils/PrimPacked.lhs b/ghc/compiler/utils/PrimPacked.lhs index f7f0201..f2d034d 100644 --- a/ghc/compiler/utils/PrimPacked.lhs +++ b/ghc/compiler/utils/PrimPacked.lhs @@ -11,18 +11,15 @@ subsystem, mostly. {-# OPTIONS -optc-DNON_POSIX_SOURCE #-} module PrimPacked ( - Ptr(..), nullPtr, writeCharOffPtr, plusAddr#, - BA(..), MBA(..), + Ptr(..), nullPtr, plusAddr#, + BA(..), packString, -- :: String -> (Int, BA) unpackNBytesBA, -- :: BA -> Int -> [Char] strLength, -- :: Ptr CChar -> Int copyPrefixStr, -- :: Addr# -> Int -> BA - copySubStr, -- :: Addr# -> Int -> Int -> BA copySubStrBA, -- :: BA -> Int -> Int -> BA eqStrPrefix, -- :: Addr# -> ByteArray# -> Int# -> Bool - eqCharStrPrefix, -- :: Addr# -> Addr# -> Int# -> Bool eqStrPrefixBA, -- :: ByteArray# -> ByteArray# -> Int# -> Int# -> Bool - eqCharStrPrefixBA, -- :: Addr# -> ByteArray# -> Int# -> Int# -> Bool ) where -- This #define suppresses the "import FastString" that @@ -70,11 +67,6 @@ nullPtr = Ptr (int2Addr# 0#) plusAddr# :: Addr# -> Int# -> Addr# plusAddr# a# i# = int2Addr# (addr2Int# a# +# i#) #endif - --- more compatibility: in 5.00+ we would use the Storable class for this, --- but 4.08 doesn't have it. -writeCharOffPtr (Ptr a#) (I# i#) (C# c#) = IO $ \s# -> - case writeCharOffAddr# a# i# c# s# of { s# -> (# s#, () #) } \end{code} Wrapper types for bytearrays @@ -154,9 +146,11 @@ Copying out a substring, assume a 0-indexed string: (and positive lengths, thank you). \begin{code} +#ifdef UNUSED copySubStr :: Addr# -> Int -> Int -> BA copySubStr a# (I# start#) length = copyPrefixStr (a# `plusAddr#` start#) length +#endif copySubStrBA :: BA -> Int -> Int -> BA copySubStrBA (BA barr#) (I# start#) len@(I# length#) = ba @@ -191,7 +185,7 @@ freeze_ps_array :: MBA s -> Int# -> ST s BA #if __GLASGOW_HASKELL__ < 411 #define NEW_BYTE_ARRAY newCharArray# #else -#define NEW_BYTE_ARRAY newByteArray# +#define NEW_BYTE_ARRAY newPinnedByteArray# #endif new_ps_array size = ST $ \ s -> @@ -214,49 +208,58 @@ Compare two equal-length strings for equality: \begin{code} eqStrPrefix :: Addr# -> ByteArray# -> Int# -> Bool eqStrPrefix a# barr# len# = - unsafePerformIO $ do + inlinePerformIO $ do x <- memcmp_ba a# barr# (I# len#) return (x == 0) --- unused??? +#ifdef UNUSED eqCharStrPrefix :: Addr# -> Addr# -> Int# -> Bool eqCharStrPrefix a1# a2# len# = - unsafePerformIO $ do + inlinePerformIO $ do x <- memcmp a1# a2# (I# len#) return (x == 0) +#endif eqStrPrefixBA :: ByteArray# -> ByteArray# -> Int# -> Int# -> Bool eqStrPrefixBA b1# b2# start# len# = - unsafePerformIO $ do + inlinePerformIO $ do x <- memcmp_baoff_ba b2# (I# start#) b1# (I# len#) return (x == 0) +#ifdef UNUSED eqCharStrPrefixBA :: Addr# -> ByteArray# -> Int# -> Int# -> Bool eqCharStrPrefixBA a# b2# start# len# = - unsafePerformIO $ do + inlinePerformIO $ do x <- memcmp_baoff b2# (I# start#) a# (I# len#) return (x == 0) +#endif \end{code} \begin{code} +-- Just like unsafePerformIO, but we inline it. This is safe when +-- there are no side effects, and improves performance. +{-# INLINE inlinePerformIO #-} +inlinePerformIO :: IO a -> a +inlinePerformIO (IO m) = case m realWorld# of (# _, r #) -> r + #if __GLASGOW_HASKELL__ <= 408 strLength (Ptr a#) = ghc_strlen a# -foreign import ccall "ghc_strlen" unsafe +foreign import ccall unsafe "ghc_strlen" ghc_strlen :: Addr# -> Int #else -foreign import ccall "ghc_strlen" unsafe +foreign import ccall unsafe "ghc_strlen" strLength :: Ptr () -> Int #endif -foreign import ccall "ghc_memcmp" unsafe +foreign import ccall unsafe "ghc_memcmp" memcmp :: Addr# -> Addr# -> Int -> IO Int -foreign import ccall "ghc_memcmp" unsafe +foreign import ccall unsafe "ghc_memcmp" memcmp_ba :: Addr# -> ByteArray# -> Int -> IO Int -foreign import ccall "ghc_memcmp_off" unsafe +foreign import ccall unsafe "ghc_memcmp_off" memcmp_baoff :: ByteArray# -> Int -> Addr# -> Int -> IO Int -foreign import ccall "ghc_memcmp_off" unsafe +foreign import ccall unsafe "ghc_memcmp_off" memcmp_baoff_ba :: ByteArray# -> Int -> ByteArray# -> Int -> IO Int \end{code}