X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=GHC%2FST.lhs;h=cdccb18204d1637287b1bfe366d49a1576a40bd4;hb=0425e42d530ad34fbef2529a20ba326dbd4109b7;hp=f98b33d73da3e169f4345db1ad9638ef705496e1;hpb=7f1f4e7a695c402ddd3a1dc2cc7114e649a78ebc;p=haskell-directory.git diff --git a/GHC/ST.lhs b/GHC/ST.lhs index f98b33d..cdccb18 100644 --- a/GHC/ST.lhs +++ b/GHC/ST.lhs @@ -1,14 +1,20 @@ -% ------------------------------------------------------------------------------ -% $Id: ST.lhs,v 1.1 2001/06/28 14:15:03 simonmar Exp $ -% -% (c) The University of Glasgow, 1992-2000 -% - -\section[GHC.ST]{The @ST@ monad} - \begin{code} -{-# OPTIONS -fno-implicit-prelude #-} - +{-# OPTIONS_GHC -fno-implicit-prelude #-} +----------------------------------------------------------------------------- +-- | +-- Module : GHC.ST +-- Copyright : (c) The University of Glasgow, 1992-2002 +-- License : see libraries/base/LICENSE +-- +-- Maintainer : cvs-ghc@haskell.org +-- Stability : internal +-- Portability : non-portable (GHC Extensions) +-- +-- The 'ST' Monad. +-- +----------------------------------------------------------------------------- + +-- #hide module GHC.ST where import GHC.Base @@ -28,6 +34,18 @@ The state-transformer monad proper. By default the monad is strict; too many people got bitten by space leaks when it was lazy. \begin{code} +-- | The strict state-transformer monad. +-- A computation of type @'ST' s a@ transforms an internal state indexed +-- by @s@, and returns a value of type @a@. +-- The @s@ parameter is either +-- +-- * an unstantiated type variable (inside invocations of 'runST'), or +-- +-- * 'RealWorld' (inside invocations of 'Control.Monad.ST.stToIO'). +-- +-- It serves to keep the internal states of different invocations +-- of 'runST' separate from each other and from invocations of +-- 'Control.Monad.ST.stToIO'. newtype ST s a = ST (STRep s a) type STRep s a = State# s -> (# State# s, a #) @@ -40,14 +58,14 @@ instance Monad (ST s) where {-# INLINE return #-} {-# INLINE (>>) #-} {-# INLINE (>>=) #-} - return x = ST $ \ s -> (# s, x #) - m >> k = m >>= \ _ -> k + return x = ST (\ s -> (# s, x #)) + m >> k = m >>= \ _ -> k (ST m) >>= k - = ST $ \ s -> + = ST (\ s -> case (m s) of { (# new_s, r #) -> case (k r) of { ST k2 -> - (k2 new_s) }} + (k2 new_s) }}) data STret s a = STret (State# s) a @@ -65,6 +83,9 @@ unsafeInterleaveST (ST m) = ST ( \ s -> (# s, r #) ) +-- | Allow the result of a state transformer computation to be used (lazily) +-- inside the computation. +-- Note that if @f@ is strict, @'fixST' f = _|_@. fixST :: (a -> ST s a) -> ST s a fixST k = ST $ \ s -> let ans = liftST (k r) s @@ -114,13 +135,17 @@ All calls to @f@ will share a {\em single} array! End SLPJ 95/04. -- The INLINE prevents runSTRep getting inlined in *this* module -- so that it is still visible when runST is inlined in an importing -- module. Regrettably delicate. runST is behaving like a wrapper. + +-- | Return the value computed by a state transformer computation. +-- The @forall@ ensures that the internal state used by the 'ST' +-- computation is inaccessible to the rest of the program. runST :: (forall s. ST s a) -> a runST st = runSTRep (case st of { ST st_rep -> st_rep }) -- I'm only letting runSTRep be inlined right at the end, in particular *after* full laziness --- That's what the "INLINE 100" says. +-- That's what the "INLINE [0]" says. -- SLPJ Apr 99 -{-# INLINE 100 runSTRep #-} +{-# INLINE [0] runSTRep #-} runSTRep :: (forall s. STRep s a) -> a runSTRep st_rep = case st_rep realWorld# of (# _, r #) -> r