\section{Fast strings}
Compact representations of character strings with
-unique identifiers.
+unique identifiers (hash-cons'ish).
\begin{code}
+#include "HsVersions.h"
+
module FastString
(
FastString(..), -- not abstract, for now.
tagCmpFS -- :: FastString -> FastString -> _CMP_TAG
) where
+#if __GLASGOW_HASKELL__ <= 201
import PreludeGlaST
import PreludeGlaMisc
import HandleHack
+import Ubiq
+#else
+import GlaExts
+import Foreign
+import IOBase
+import IOHandle
+import ST
+import STBase
+import {-# SOURCE #-} Unique ( mkUniqueGrimily, Unique, Uniquable(..) )
+#if __GLASGOW_HASKELL__ == 202
+import PrelBase ( Char (..) )
+#endif
+#if __GLASGOW_HASKELL__ >= 206
+import PackBase
+#endif
+#endif
import PrimPacked
-import Ubiq
#define hASH_TBL_SIZE 993
uniqueOf (I# i#) = mkUniqueGrimily i#
instance Text FastString where
- readsPrec p = error "readsPrec: FastString: ToDo"
showsPrec p ps@(FastString u# _ _) r = showsPrec p (unpackFS ps) r
showsPrec p ps r = showsPrec p (unpackFS ps) r
nullFastString (CharStr _ l#) = l# ==# 0#
unpackFS :: FastString -> String
+#if defined(__GLASGOW_HASKELL__) && __GLASGOW_HASKELL__ <= 205
unpackFS (FastString _ l# ba#) = byteArrayToString (_ByteArray (0,I# l#) ba#)
+#else
+unpackFS (FastString _ l# ba#) = unpackCStringBA# l# ba#
+#endif
unpackFS (CharStr addr len#) =
unpack 0#
where
newArray (0::Int,hASH_TBL_SIZE) [] `thenPrimIO` \ (_MutableArray _ arr#) ->
newVar (FastStringTable 0# arr#))
-lookupTbl :: FastStringTable -> Int# -> [FastString]
+lookupTbl :: FastStringTable -> Int# -> PrimIO [FastString]
lookupTbl (FastStringTable _ arr#) i# =
- unsafePerformPrimIO ( \ (S# s#) ->
- case readArray# arr# i# s# of { StateAndPtr# s2# r ->
- (r, S# s2#) } )
+ MkST ( \ (S# s#) ->
+ case readArray# arr# i# s# of { StateAndPtr# s2# r ->
+ (r, S# s2#) })
updTbl :: FastStringTableVar -> FastStringTable -> Int# -> [FastString] -> PrimIO ()
-updTbl (_MutableArray _ var#) (FastStringTable uid# arr#) i# ls (S# s#) =
+updTbl (_MutableArray _ var#) (FastStringTable uid# arr#) i# ls =
+ MkST ( \ (S# s#) ->
case writeArray# arr# i# ls s# of { s2# ->
case writeArray# var# 0# (FastStringTable (uid# +# 1#) arr#) s2# of { s3# ->
- ((), S# s3#) }}
+ ((), S# s3#) }})
mkFastString# :: Addr# -> Int# -> FastString
mkFastString# a# len# =
h = hashStr a# len#
in
-- _trace ("hashed: "++show (I# h)) $
- case lookupTbl ft h of
+ lookupTbl ft h `thenPrimIO` \ lookup_result ->
+ case lookup_result of
[] ->
-- no match, add it to table by copying out the
-- the string into a ByteArray
mkFastSubStringFO# :: ForeignObj# -> Int# -> Int# -> FastString
mkFastSubStringFO# fo# start# len# =
unsafePerformPrimIO (
- readVar string_table `thenPrimIO` \ ft@(FastStringTable uid# tbl#) ->
- let h = hashSubStrFO fo# start# len# in
- case lookupTbl ft h of
+ readVar string_table `thenPrimIO` \ ft@(FastStringTable uid# tbl#) ->
+ let
+ h = hashSubStrFO fo# start# len#
+ in
+ lookupTbl ft h `thenPrimIO` \ lookup_result ->
+ case lookup_result of
[] ->
-- no match, add it to table by copying out the
-- the string into a ByteArray
mkFastSubStringBA# :: ByteArray# -> Int# -> Int# -> FastString
mkFastSubStringBA# barr# start# len# =
unsafePerformPrimIO (
- readVar string_table `thenPrimIO` \ ft@(FastStringTable uid# tbl#) ->
- let h = hashSubStrBA barr# start# len# in
- -- _trace ("hashed(b): "++show (I# h)) $
- case lookupTbl ft h of
+ readVar string_table `thenPrimIO` \ ft@(FastStringTable uid# tbl#) ->
+ let
+ h = hashSubStrBA barr# start# len#
+ in
+-- _trace ("hashed(b): "++show (I# h)) $
+ lookupTbl ft h `thenPrimIO` \ lookup_result ->
+ case lookup_result of
[] ->
-- no match, add it to table by copying out the
-- the string into a ByteArray
mkFastString :: String -> FastString
mkFastString str =
+#if defined(__GLASGOW_HASKELL__) && __GLASGOW_HASKELL__ <= 205
case stringToByteArray str of
+#else
+ case packString str of
+#endif
(_ByteArray (_,I# len#) frozen#) ->
mkFastSubStringBA# frozen# 0# len#
{- 0-indexed array, len# == index to one beyond end of string,
1# -> ((ord# c0 *# 631#) +# len#) `remInt#` hASH_TBL_SIZE#
2# -> ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# len#) `remInt#` hASH_TBL_SIZE#
_ -> ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# (ord# c2 *# 43#) +# len#) `remInt#` hASH_TBL_SIZE#
-{- Currently UNUSED:
- if len# ==# 0# then
- 0#
- else
- ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# (ord# c2 *# 43#) +# len#)
- `remInt#` hASH_TBL_SIZE#
--}
where
c0 = indexCharOffAddr# a# 0#
- c1 = indexCharOffAddr# a# 1# --(len# `quotInt#` 2# -# 1#)
- c2 = indexCharOffAddr# a# 2# --(len# -# 1#)
+ c1 = indexCharOffAddr# a# (len# `quotInt#` 2# -# 1#)
+ c2 = indexCharOffAddr# a# (len# -# 1#)
+{-
+ c1 = indexCharOffAddr# a# 1#
+ c2 = indexCharOffAddr# a# 2#
+-}
hashSubStrFO :: ForeignObj# -> Int# -> Int# -> Int#
- -- use the Addr to produce a hash value between 0 & m (inclusive)
+ -- use the FO to produce a hash value between 0 & m (inclusive)
hashSubStrFO fo# start# len# =
case len# of
0# -> 0#
1# -> ((ord# c0 *# 631#) +# len#) `remInt#` hASH_TBL_SIZE#
2# -> ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# len#) `remInt#` hASH_TBL_SIZE#
_ -> ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# (ord# c2 *# 43#) +# len#) `remInt#` hASH_TBL_SIZE#
-{-
- if len# ==# 0# then
- 0#
- else
- ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# (ord# c2 *# 43#) +# len#)
- `remInt#` hASH_TBL_SIZE#
--}
where
c0 = indexCharOffFO# fo# 0#
- c1 = indexCharOffFO# fo# 1# --(len# `quotInt#` 2# -# 1#)
- c2 = indexCharOffFO# fo# 2# --(len# -# 1#)
+ c1 = indexCharOffFO# fo# (len# `quotInt#` 2# -# 1#)
+ c2 = indexCharOffFO# fo# (len# -# 1#)
+
+-- c1 = indexCharOffFO# fo# 1#
+-- c2 = indexCharOffFO# fo# 2#
hashSubStrBA :: ByteArray# -> Int# -> Int# -> Int#
- -- use the Addr to produce a hash value between 0 & m (inclusive)
+ -- use the byte array to produce a hash value between 0 & m (inclusive)
hashSubStrBA ba# start# len# =
case len# of
0# -> 0#
1# -> ((ord# c0 *# 631#) +# len#) `remInt#` hASH_TBL_SIZE#
2# -> ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# len#) `remInt#` hASH_TBL_SIZE#
_ -> ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# (ord# c2 *# 43#) +# len#) `remInt#` hASH_TBL_SIZE#
-{-
- if len# ==# 0# then
- 0#
- else
- ((ord# c0 *# 631#) +# (ord# c1 *# 217#) +# (ord# c2 *# 43#) +# len#)
- `remInt#` hASH_TBL_SIZE#
--}
where
c0 = indexCharArray# ba# 0#
- c1 = indexCharArray# ba# 1# --(len# `quotInt#` 2# -# 1#)
- c2 = indexCharArray# ba# 2# --(len# -# 1#)
+ c1 = indexCharArray# ba# (len# `quotInt#` 2# -# 1#)
+ c2 = indexCharArray# ba# (len# -# 1#)
+
+-- c1 = indexCharArray# ba# 1#
+-- c2 = indexCharArray# ba# 2#
\end{code}
else _GT
))
where
+ bottom :: (Int,Int)
bottom = error "tagCmp"
tagCmpFS (CharStr bs1 len1) (CharStr bs2 len2)
= unsafePerformPrimIO (
else _GT
))
where
- ba1 = _ByteArray (error "") bs1
+ ba1 = _ByteArray ((error "")::(Int,Int)) bs1
ba2 = A# bs2
tagCmpFS a@(CharStr _ _) b@(FastString _ _ _)
@fwrite@).
\begin{code}
+#if __GLASGOW_HASKELL__ >= 201
+#define _ErrorHandle IOBase.ErrorHandle
+#define _ReadHandle IOBase.ReadHandle
+#define _ClosedHandle IOBase.ClosedHandle
+#define _SemiClosedHandle IOBase.SemiClosedHandle
+#define _constructError IOBase.constructError
+#define _filePtr IOHandle.filePtr
+#define failWith fail
+#endif
+
hPutFS :: Handle -> FastString -> IO ()
hPutFS handle (FastString _ l# ba#) =
if l# ==# 0# then
failWith ioError
_ClosedHandle ->
_writeHandle handle htype >>
- failWith (IllegalOperation "handle is closed")
+ failWith MkIOError(handle,IllegalOperation,"handle is closed")
_SemiClosedHandle _ _ ->
_writeHandle handle htype >>
- failWith (IllegalOperation "handle is closed")
+ failWith MkIOError(handle,IllegalOperation,"handle is closed")
_ReadHandle _ _ _ ->
_writeHandle handle htype >>
- failWith (IllegalOperation "handle is not open for writing")
+ failWith MkIOError(handle,IllegalOperation,"handle is not open for writing")
other ->
let fp = _filePtr htype in
-- here we go..
- _ccall_ writeFile (_ByteArray (error "") ba#) fp (I# l#) `thenPrimIO` \rc ->
+ _ccall_ writeFile (_ByteArray ((error "")::(Int,Int)) ba#) fp (I# l#) `CCALL_THEN` \rc ->
if rc==0 then
return ()
else
- _constructError "hPutFS" `thenPrimIO` \ err ->
+ _constructError "hPutFS" `CCALL_THEN` \ err ->
failWith err
hPutFS handle (CharStr a# l#) =
if l# ==# 0# then
failWith ioError
_ClosedHandle ->
_writeHandle handle htype >>
- failWith (IllegalOperation "handle is closed")
+ failWith MkIOError(handle,IllegalOperation,"handle is closed")
_SemiClosedHandle _ _ ->
_writeHandle handle htype >>
- failWith (IllegalOperation "handle is closed")
+ failWith MkIOError(handle,IllegalOperation,"handle is closed")
_ReadHandle _ _ _ ->
_writeHandle handle htype >>
- failWith (IllegalOperation "handle is not open for writing")
+ failWith MkIOError(handle,IllegalOperation,"handle is not open for writing")
other ->
let fp = _filePtr htype in
-- here we go..
- _ccall_ writeFile (A# a#) fp (I# l#) `thenPrimIO` \rc ->
+ _ccall_ writeFile (A# a#) fp (I# l#) `CCALL_THEN` \rc ->
if rc==0 then
return ()
else
- _constructError "hPutFS" `thenPrimIO` \ err ->
+ _constructError "hPutFS" `CCALL_THEN` \ err ->
failWith err
--ToDo: avoid silly code duplic.
projects / ghc-hetmet.git / blobdiff