[project @ 2000-04-13 19:31:05 by panne]
[ghc-hetmet.git] / ghc / compiler / utils / PrimPacked.lhs
index 1021645..aa38d6a 100644 (file)
@@ -1,5 +1,5 @@
 %
-% (c) The GRASP/AQUA Project, Glasgow University, 1997
+% (c) The GRASP/AQUA Project, Glasgow University, 1997-1998
 %
 \section{Basic ops on packed representations}
 
@@ -22,8 +22,7 @@ module PrimPacked
         eqCharStrPrefixBA,  -- :: Addr# -> ByteArray# -> Int# -> Int# -> Bool
         eqStrPrefixFO,      -- :: ForeignObj# -> ByteArray# -> Int# -> Int# -> Bool
 
-        addrOffset#,        -- :: Addr# -> Int# -> Addr# 
-        indexCharOffFO#     -- :: ForeignObj# -> Int# -> Char#
+        addrOffset#         -- :: Addr# -> Int# -> Addr# 
        ) where
 
 -- This #define suppresses the "import FastString" that
@@ -32,13 +31,26 @@ module PrimPacked
 #include "HsVersions.h"
 
 import GlaExts
-import Addr    ( Addr(..) )
-import GHC
-import ArrBase
+import PrelAddr        ( Addr(..) )
 import ST
+import Foreign
+-- ForeignObj is now exported abstractly.
+#if __GLASGOW_HASKELL__ >= 303
+import PrelForeign   ( ForeignObj(..) )
+#endif
+
+#if __GLASGOW_HASKELL__ < 301
+import ArrBase         ( StateAndMutableByteArray#(..), 
+                         StateAndByteArray#(..) )
 import STBase
-import IOBase  ( ForeignObj(..) )
-import PackBase ( unpackCStringBA, packString )
+#elif __GLASGOW_HASKELL__ < 400
+import PrelArr         ( StateAndMutableByteArray#(..), 
+                         StateAndByteArray#(..) )
+import PrelST
+#else
+import PrelST
+#endif
+
 \end{code} 
 
 Return the length of a @\\NUL@ terminated character string:
@@ -50,7 +62,7 @@ strLength a =
     _ccall_ strlen a  >>= \ len@(I# _) ->
     return len
  )
-
+{-# NOINLINE strLength #-}
 \end{code}
 
 Copying a char string prefix into a byte array,
@@ -122,21 +134,10 @@ copySubStrFO (ForeignObj fo) (I# start#) len@(I# length#) =
       = write_ps_array arr_in# idx (chr# 0#) `seqStrictlyST`
        returnStrictlyST ()
       | otherwise
-      = case (indexCharOffFO# fo (idx +# start#)) of { ch ->
+      = case (indexCharOffForeignObj# fo (idx +# start#)) of { ch ->
        write_ps_array arr_in# idx ch `seqStrictlyST`
        fill_in arr_in# (idx +# 1#) }
 
-{- ToDo: add FO primitives.. -}
-#if defined(__GLASGOW_HASKELL__) && __GLASGOW_HASKELL__ <=205
-indexCharOffFO# :: ForeignObj# -> Int# -> Char#
-indexCharOffFO# fo# i# = 
-  case unsafePerformIO (_casm_ ``%r=(char)*((char *)%0 + (int)%1); '' (ForeignObj fo#) (I# i#)) of
-    C# c -> c
-#else
-indexCharOffFO# :: ForeignObj# -> Int# -> Char#
-indexCharOffFO# fo i = indexCharOffForeignObj# fo i
-#endif
-
 -- step on (char *) pointer by x units.
 addrOffset# :: Addr# -> Int# -> Addr# 
 addrOffset# a# i# =
@@ -144,7 +145,11 @@ addrOffset# a# i# =
     A# a -> a
 
 copySubStrBA :: ByteArray Int -> Int -> Int -> ByteArray Int
+#if __GLASGOW_HASKELL__ >= 405
+copySubStrBA (ByteArray _ _ barr#) (I# start#) len@(I# length#) =
+#else
 copySubStrBA (ByteArray _ barr#) (I# start#) len@(I# length#) =
+#endif
  runST (
   {- allocate an array that will hold the string
     (not forgetting the NUL at the end)
@@ -177,19 +182,47 @@ write_ps_array    :: MutableByteArray s Int -> Int# -> Char# -> ST s ()
 freeze_ps_array :: MutableByteArray s Int -> Int# -> ST s (ByteArray Int)
 
 new_ps_array size = ST $ \ s ->
+#if __GLASGOW_HASKELL__ < 400
     case (newCharArray# size s)          of { StateAndMutableByteArray# s2# barr# ->
     STret s2# (MutableByteArray bot barr#) }
+#elif __GLASGOW_HASKELL__ < 405
+    case (newCharArray# size s)          of { (# s2#, barr# #) ->
+    (# s2#, MutableByteArray bot barr# #) }
+#else
+    case (newCharArray# size s)          of { (# s2#, barr# #) ->
+    (# s2#, MutableByteArray bot bot barr# #) }
+#endif
   where
     bot = error "new_ps_array"
 
+#if __GLASGOW_HASKELL__ < 400
 write_ps_array (MutableByteArray _ barr#) n ch = ST $ \ s# ->
     case writeCharArray# barr# n ch s# of { s2#   ->
     STret s2# () }
+#elif __GLASGOW_HASKELL__ < 405
+write_ps_array (MutableByteArray _ barr#) n ch = ST $ \ s# ->
+    case writeCharArray# barr# n ch s# of { s2#   ->
+    (# s2#, () #) }
+#else
+write_ps_array (MutableByteArray _ _ barr#) n ch = ST $ \ s# ->
+    case writeCharArray# barr# n ch s# of { s2#   ->
+    (# s2#, () #) }
+#endif
 
 -- same as unsafeFreezeByteArray
+#if __GLASGOW_HASKELL__ < 400
 freeze_ps_array (MutableByteArray _ arr#) len# = ST $ \ s# ->
     case unsafeFreezeByteArray# arr# s# of { StateAndByteArray# s2# frozen# ->
     STret s2# (ByteArray (0,I# len#) frozen#) }
+#elif __GLASGOW_HASKELL__ < 405
+freeze_ps_array (MutableByteArray _ arr#) len# = ST $ \ s# ->
+    case unsafeFreezeByteArray# arr# s# of { (# s2#, frozen# #) ->
+    (# s2#, ByteArray (0,I# len#) frozen# #) }
+#else
+freeze_ps_array (MutableByteArray _ _ arr#) len# = ST $ \ s# ->
+    case unsafeFreezeByteArray# arr# s# of { (# s2#, frozen# #) ->
+    (# s2#, ByteArray 0 (I# len#) frozen# #) }
+#endif
 \end{code}
 
 
@@ -199,46 +232,71 @@ Compare two equal-length strings for equality:
 eqStrPrefix :: Addr# -> ByteArray# -> Int# -> Bool
 eqStrPrefix a# barr# len# = 
   unsafePerformIO (
-   _ccall_ strncmp (A# a#) (ByteArray bottom barr#) (I# len#) >>= \ (I# x#) ->
+#if __GLASGOW_HASKELL__ < 405
+   _ccall_ strncmp (A# a#) (ByteArray bot barr#) (I# len#) >>= \ (I# x#) ->
+#else
+   _ccall_ strncmp (A# a#) (ByteArray bot bot barr#) (I# len#) >>= \ (I# x#) ->
+#endif
    return (x# ==# 0#))
   where
-   bottom :: (Int,Int)
-   bottom = error "eqStrPrefix"
+#if __GLASGOW_HASKELL__ < 405
+   bot :: (Int,Int)
+#else
+   bot :: Int
+#endif
+   bot = error "eqStrPrefix"
 
 eqCharStrPrefix :: Addr# -> Addr# -> Int# -> Bool
 eqCharStrPrefix a1# a2# len# = 
   unsafePerformIO (
    _ccall_ strncmp (A# a1#) (A# a2#) (I# len#) >>= \ (I# x#) ->
    return (x# ==# 0#))
-  where
-   bottom :: (Int,Int)
-   bottom = error "eqStrPrefix"
 
 eqStrPrefixBA :: ByteArray# -> ByteArray# -> Int# -> Int# -> Bool
 eqStrPrefixBA b1# b2# start# len# = 
   unsafePerformIO (
    _casm_ ``%r=(int)strncmp((char *)%0+(int)%1,%2,%3); '' 
-         (ByteArray bottom b2#) 
+#if __GLASGOW_HASKELL__ < 405
+         (ByteArray bot b2#)
+#else
+         (ByteArray bot bot b2#) 
+#endif 
          (I# start#) 
-          (ByteArray bottom b1#) 
+#if __GLASGOW_HASKELL__ < 405
+          (ByteArray bot b1#) 
+#else
+          (ByteArray bot bot b1#) 
+#endif
           (I# len#)                  >>= \ (I# x#) ->
    return (x# ==# 0#))
   where
-   bottom :: (Int,Int)
-   bottom = error "eqStrPrefixBA"
+#if __GLASGOW_HASKELL__ < 405
+   bot :: (Int,Int)
+#else
+   bot :: Int
+#endif
+   bot = error "eqStrPrefixBA"
 
 eqCharStrPrefixBA :: Addr# -> ByteArray# -> Int# -> Int# -> Bool
 eqCharStrPrefixBA a# b2# start# len# = 
   unsafePerformIO (
    _casm_ ``%r=(int)strncmp((char *)%0+(int)%1,%2,%3); '' 
-         (ByteArray bottom b2#) 
+#if __GLASGOW_HASKELL__ < 405
+         (ByteArray bot b2#) 
+#else
+         (ByteArray bot bot b2#) 
+#endif
          (I# start#) 
           (A# a#)
           (I# len#)                  >>= \ (I# x#) ->
    return (x# ==# 0#))
   where
-   bottom :: (Int,Int)
-   bottom = error "eqCharStrPrefixBA"
+#if __GLASGOW_HASKELL__ < 405
+   bot :: (Int,Int)
+#else
+   bot :: Int
+#endif
+   bot = error "eqCharStrPrefixBA"
 
 eqStrPrefixFO :: ForeignObj# -> ByteArray# -> Int# -> Int# -> Bool
 eqStrPrefixFO fo# barr# start# len# = 
@@ -246,21 +304,18 @@ eqStrPrefixFO fo# barr# start# len# =
    _casm_ ``%r=(int)strncmp((char *)%0+(int)%1,%2,%3); '' 
          (ForeignObj fo#) 
          (I# start#) 
-          (ByteArray bottom barr#) 
+#if __GLASGOW_HASKELL__ < 405
+          (ByteArray bot barr#) 
+#else
+          (ByteArray bot bot barr#) 
+#endif
           (I# len#)                  >>= \ (I# x#) ->
    return (x# ==# 0#))
   where
-   bottom :: (Int,Int)
-   bottom = error "eqStrPrefixFO"
-\end{code}
-
-\begin{code}
-byteArrayToString :: ByteArray Int -> String
-byteArrayToString = unpackCStringBA
-\end{code}
-
-
-\begin{code}
-stringToByteArray :: String -> (ByteArray Int)
-stringToByteArray = packString
+#if __GLASGOW_HASKELL__ < 405
+   bot :: (Int,Int)
+#else
+   bot :: Int
+#endif
+   bot = error "eqStrPrefixFO"
 \end{code}