Buffers for scanning string input stored in external arrays.
\begin{code}
+
+{-# OPTIONS -fno-prune-tydecls #-}
+-- Don't really understand this!
+-- ERROR: Can't see the data constructor(s) for _ccall_/_casm_ argument;
+-- type: ForeignObj(try compiling with -fno-prune-tydecls ..)
+
+
module StringBuffer
(
StringBuffer,
lexemeToBuffer, -- :: StringBuffer -> StringBuffer
FastString,
- _ByteArray
+ ByteArray
) where
-import Ubiq
-import PreludeGlaST
-import PreludeGlaMisc
+#include "HsVersions.h"
+
+import GlaExts
+import Addr ( Addr(..) )
+import Foreign
+import ST
+import IO ( openFile, hFileSize, hClose, IOMode(..) )
+
+#if __GLASGOW_HASKELL__ < 301
+import IOBase ( IOError(..), IOErrorType(..) )
+import IOHandle ( readHandle, writeHandle, filePtr )
+import PackBase ( unpackCStringBA )
+#else
+import PrelIOBase ( IOError(..), IOErrorType(..) )
+import PrelHandle ( readHandle, writeHandle, filePtr )
+import PrelPack ( unpackCStringBA )
+#endif
+
import PrimPacked
import FastString
-import HandleHack
-
+import Char (isDigit)
\end{code}
\begin{code}
\end{code}
\begin{code}
+instance Text StringBuffer where
+ showsPrec _ s = showString ""
+\end{code}
+\begin{code}
hGetStringBuffer :: FilePath -> IO StringBuffer
hGetStringBuffer fname =
--- _trace ("Renamer: opening " ++ fname)
+-- trace ("Renamer: opening " ++ fname) $
openFile fname ReadMode >>= \ hndl ->
hFileSize hndl >>= \ len@(J# _ _ d#) ->
let len_i = fromInteger len in
-- Allocate an array for system call to store its bytes into.
-- ToDo: make it robust
--- _trace (show (len_i::Int)+1)
- (_casm_ `` %r=(char *)malloc(sizeof(char)*(int)%0); '' (len_i::Int)) `thenPrimIO` \ arr@(A# a#) ->
+-- trace (show ((len_i::Int)+1)) $
+ _casm_ `` %r=(char *)malloc(sizeof(char)*(int)%0); '' (len_i::Int) >>= \ arr@(A# a#) ->
if addr2Int# a# ==# 0# then
- failWith (UserError ("hGetStringBuffer: Could not allocate "++show len_i ++ " bytes"))
+ failWith MkIOError(hndl,UserError,("hGetStringBuffer: Could not allocate "++show len_i ++ " bytes"))
else
--- _casm_ `` %r=NULL; '' `thenPrimIO` \ free_p ->
--- makeForeignObj arr free_p `thenPrimIO` \ fo@(_ForeignObj fo#) ->
- _readHandle hndl >>= \ _hndl ->
- _writeHandle hndl _hndl >>
- let ptr = _filePtr _hndl in
- _ccall_ fread arr (1::Int) len_i ptr `thenPrimIO` \ (I# read#) ->
--- _trace ("DEBUG: opened " ++ fname ++ show (I# read#)) $
+-- _casm_ `` %r=NULL; '' >>= \ free_p ->
+-- makeForeignObj arr free_p >>= \ fo@(_ForeignObj fo#) ->
+ readHandle hndl >>= \ hndl_ ->
+ writeHandle hndl hndl_ >>
+ let ptr = filePtr hndl_ in
+ _ccall_ fread arr (1::Int) len_i ptr >>= \ (I# read#) ->
+-- trace ("DEBUG: opened " ++ fname ++ show (I# read#)) $
hClose hndl >>
if read# ==# 0# then -- EOF or other error
- failWith (UserError "hGetStringBuffer: EOF reached or some other error")
+ failWith MkIOError(hndl,UserError,"hGetStringBuffer: EOF reached or some other error")
else
-- Add a sentinel NUL
- _casm_ `` ((char *)%0)[(int)%1]=(char)0; '' arr (I# (read# -# 1#)) `thenPrimIO` \ () ->
+ _casm_ `` ((char *)%0)[(int)%1]=(char)0; '' arr (I# (read# -# 1#)) >>= \ () ->
return (StringBuffer a# read# 0# 0#)
freeStringBuffer :: StringBuffer -> IO ()
freeStringBuffer (StringBuffer a# _ _ _) =
- _casm_ `` free((char *)%0); '' (A# a#) `thenPrimIO` \ () ->
- return ()
+ _casm_ `` free((char *)%0); '' (A# a#)
unsafeWriteBuffer :: StringBuffer -> Int# -> Char# -> StringBuffer
unsafeWriteBuffer s@(StringBuffer a _ _ _) i# ch# =
- unsafePerformPrimIO (
- _casm_ `` ((char *)%0)[(int)%1]=(char)%2; '' (A# a) (I# i#) (C# ch#) `thenPrimIO` \ () ->
- returnPrimIO s)
+ unsafePerformIO (
+ _casm_ `` ((char *)%0)[(int)%1]=(char)%2; '' (A# a) (I# i#) (C# ch#) >>= \ () ->
+ return s
+ )
\end{code}
loop c# =
case indexCharOffAddr# fo c# of
ch# | pred (C# ch#) -> StringBuffer fo l# c# c#
+ | ch# `eqChar#` '\NUL'# && c# >=# l# -> StringBuffer fo l# l# l# -- EOB, return immediately.
| otherwise -> loop (c# +# 1#)
stepOverLexeme :: StringBuffer -> StringBuffer
loop c# =
case indexCharOffAddr# fo c# of
ch# | pred (C# ch#) -> loop (c# +# 1#)
+ | ch# `eqChar#` '\NUL'# && c# >=# l# -> StringBuffer fo l# l# l# -- EOB, return immediately.
| otherwise -> StringBuffer fo l# s# c#
loop acc# c# =
case indexCharOffAddr# fo c# of
ch# | isDigit (C# ch#) -> loop (acc# *# 10# +# (ord# ch# -# ord# '0'#)) (c# +# 1#)
+ | ch# `eqChar#` '\NUL'# && c# >=# l# -> (I# acc#, StringBuffer fo l# l# l#) -- EOB, return immediately.
| otherwise -> (I# acc#,StringBuffer fo l# s# c#)
case indexCharOffAddr# fo c# of
'\"'# ->
case indexCharOffAddr# fo (c# -# 1#) of
- '\\'# -> --escaped, false alarm.
- loop (c# +# 1#)
+ '\\'# ->
+ -- looks like an escaped something or other to me,
+ -- better count the number of "\\"s that are immediately
+ -- preceeding to decide if the " is escaped.
+ let
+ odd_slashes flg i# =
+ case indexCharOffAddr# fo i# of
+ '\\'# -> odd_slashes (not flg) (i# -# 1#)
+ _ -> flg
+ in
+ if odd_slashes True (c# -# 2#) then
+ -- odd number, " is ecaped.
+ loop (c# +# 1#)
+ else -- a real end of string delimiter after all.
+ StringBuffer fo l# s# c#
_ -> StringBuffer fo l# s# c#
+ '\NUL'# ->
+ if c# >=# l# then -- hit sentinel, this doesn't look too good..
+ StringBuffer fo l# l# l#
+ else
+ loop (c# +# 1#)
_ -> loop (c# +# 1#)
case indexCharOffAddr# fo c# of
'\''# ->
case indexCharOffAddr# fo (c# -# 1#) of
- '\\'# -> --escaped, false alarm.
- loop (c# +# 1#)
+ '\\'# ->
+ case indexCharOffAddr# fo (c# -# 2#) of
+ '\\'# -> -- end of char
+ StringBuffer fo l# s# c#
+ _ -> loop (c# +# 1#) -- false alarm
_ -> StringBuffer fo l# s# c#
+ '\NUL'# ->
+ if c# >=# l# then -- hit sentinel, this doesn't look too good..
+ StringBuffer fo l# l# l#
+ else
+ loop (c# +# 1#)
_ -> loop (c# +# 1#)
untilChar# :: StringBuffer -> Char# -> StringBuffer
if start_pos# ==# current# then
""
else
- byteArrayToString (copySubStr (A# fo) (I# start_pos#) (I# (current# -# start_pos#)))
-
+ unpackCStringBA (copySubStr (A# fo) (I# start_pos#) (I# (current# -# start_pos#)))
lexemeToByteArray :: StringBuffer -> _ByteArray Int
lexemeToByteArray (StringBuffer fo _ start_pos# current#) =
projects / ghc-hetmet.git / blobdiff