Sophie

Sophie

distrib > Fedora > 14 > x86_64 > by-pkgid > 8d1ef08c9e0d44c69764afc615a03d0d > files > 1719

ghc-ghc-devel-6.12.3-5.fc14.i686.rpm

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html>
<head>
<!-- Generated by HsColour, http://www.cs.york.ac.uk/fp/darcs/hscolour/ -->
<title>utils/FastMutInt.lhs</title>
<link type='text/css' rel='stylesheet' href='hscolour.css' />
</head>
<body>
\begin{code}
<pre><a name="line-1"></a><span class='hs-comment'>{-# OPTIONS -cpp #-}</span>
<a name="line-2"></a><span class='hs-comment'>{-# OPTIONS_GHC -O #-}</span>
<a name="line-3"></a><span class='hs-comment'>-- We always optimise this, otherwise performance of a non-optimised</span>
<a name="line-4"></a><span class='hs-comment'>-- compiler is severely affected</span>
<a name="line-5"></a>
<a name="line-6"></a><span class='hs-comment'>--</span>
<a name="line-7"></a><span class='hs-comment'>-- (c) The University of Glasgow 2002-2006</span>
<a name="line-8"></a><span class='hs-comment'>--</span>
<a name="line-9"></a><span class='hs-comment'>-- Unboxed mutable Ints</span>
<a name="line-10"></a>
<a name="line-11"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>FastMutInt</span><span class='hs-layout'>(</span>
<a name="line-12"></a>	<span class='hs-conid'>FastMutInt</span><span class='hs-layout'>,</span> <span class='hs-varid'>newFastMutInt</span><span class='hs-layout'>,</span>
<a name="line-13"></a>	<span class='hs-varid'>readFastMutInt</span><span class='hs-layout'>,</span> <span class='hs-varid'>writeFastMutInt</span><span class='hs-layout'>,</span>
<a name="line-14"></a>
<a name="line-15"></a>	<span class='hs-conid'>FastMutPtr</span><span class='hs-layout'>,</span> <span class='hs-varid'>newFastMutPtr</span><span class='hs-layout'>,</span>
<a name="line-16"></a>	<span class='hs-varid'>readFastMutPtr</span><span class='hs-layout'>,</span> <span class='hs-varid'>writeFastMutPtr</span>
<a name="line-17"></a>  <span class='hs-layout'>)</span> <span class='hs-keyword'>where</span>
<a name="line-18"></a>
<a name="line-19"></a><span class='hs-cpp'>#ifdef __GLASGOW_HASKELL__</span>
<a name="line-20"></a>
<a name="line-21"></a><span class='hs-cpp'>#include "../includes/MachDeps.h"</span>
<a name="line-22"></a><span class='hs-cpp'>#ifndef SIZEOF_HSINT</span>
<a name="line-23"></a><span class='hs-cpp'>#define SIZEOF_HSINT  INT_SIZE_IN_BYTES</span>
<a name="line-24"></a><span class='hs-cpp'>#endif</span>
<a name="line-25"></a>
<a name="line-26"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>GHC</span><span class='hs-varop'>.</span><span class='hs-conid'>Base</span>
<a name="line-27"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>GHC</span><span class='hs-varop'>.</span><span class='hs-conid'>Ptr</span>
<a name="line-28"></a>
<a name="line-29"></a><span class='hs-cpp'>#if __GLASGOW_HASKELL__ &gt;= 611</span>
<a name="line-30"></a><span class='hs-comment'>-- import GHC.IO ( IO(..) )</span>
<a name="line-31"></a><span class='hs-cpp'>#else</span>
<a name="line-32"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>GHC</span><span class='hs-varop'>.</span><span class='hs-conid'>IOBase</span> <span class='hs-layout'>(</span> <span class='hs-conid'>IO</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span> <span class='hs-layout'>)</span>
<a name="line-33"></a><span class='hs-cpp'>#endif</span>
<a name="line-34"></a>
<a name="line-35"></a><span class='hs-cpp'>#else /* ! __GLASGOW_HASKELL__ */</span>
<a name="line-36"></a>
<a name="line-37"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>IORef</span>
<a name="line-38"></a>
<a name="line-39"></a><span class='hs-cpp'>#endif</span>
<a name="line-40"></a>
<a name="line-41"></a><a name="newFastMutInt"></a><span class='hs-definition'>newFastMutInt</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>IO</span> <span class='hs-conid'>FastMutInt</span>
<a name="line-42"></a><a name="readFastMutInt"></a><span class='hs-definition'>readFastMutInt</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>IO</span> <span class='hs-conid'>Int</span>
<a name="line-43"></a><a name="writeFastMutInt"></a><span class='hs-definition'>writeFastMutInt</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Int</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>IO</span> <span class='hs-conid'>()</span>
<a name="line-44"></a>
<a name="line-45"></a><a name="newFastMutPtr"></a><span class='hs-definition'>newFastMutPtr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>IO</span> <span class='hs-conid'>FastMutPtr</span>
<a name="line-46"></a><a name="readFastMutPtr"></a><span class='hs-definition'>readFastMutPtr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>IO</span> <span class='hs-layout'>(</span><span class='hs-conid'>Ptr</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span>
<a name="line-47"></a><a name="writeFastMutPtr"></a><span class='hs-definition'>writeFastMutPtr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Ptr</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>IO</span> <span class='hs-conid'>()</span>
</pre>\end{code}

\begin{code}
<pre><a name="line-1"></a><span class='hs-cpp'>#</span><span class='hs-varid'>ifdef</span> <span class='hs-sel'>__GLASGOW_HASKELL__</span>
<a name="line-2"></a><a name="FastMutInt"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-layout'>(</span><span class='hs-conid'>MutableByteArray</span><span class='hs-cpp'>#</span> <span class='hs-conid'>RealWorld</span><span class='hs-layout'>)</span>
<a name="line-3"></a>
<a name="line-4"></a><a name="newFastMutInt"></a><span class='hs-definition'>newFastMutInt</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>IO</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-5"></a>  <span class='hs-keyword'>case</span> <span class='hs-varid'>newByteArray</span><span class='hs-cpp'>#</span> <span class='hs-varid'>size</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span> <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-varid'>arr</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-6"></a>  <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-varid'>arr</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-7"></a>  <span class='hs-keyword'>where</span> <span class='hs-varop'>!</span><span class='hs-layout'>(</span><span class='hs-conid'>I</span><span class='hs-cpp'>#</span> <span class='hs-varid'>size</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>SIZEOF_HSINT</span>
<a name="line-8"></a>
<a name="line-9"></a><a name="readFastMutInt"></a><span class='hs-definition'>readFastMutInt</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutInt</span> <span class='hs-varid'>arr</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>IO</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-10"></a>  <span class='hs-keyword'>case</span> <span class='hs-varid'>readIntArray</span><span class='hs-cpp'>#</span> <span class='hs-varid'>arr</span> <span class='hs-num'>0</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span> <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-varid'>i</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-11"></a>  <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-conid'>I</span><span class='hs-cpp'>#</span> <span class='hs-varid'>i</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-12"></a>
<a name="line-13"></a><a name="writeFastMutInt"></a><span class='hs-definition'>writeFastMutInt</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutInt</span> <span class='hs-varid'>arr</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>I</span><span class='hs-cpp'>#</span> <span class='hs-varid'>i</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>IO</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-14"></a>  <span class='hs-keyword'>case</span> <span class='hs-varid'>writeIntArray</span><span class='hs-cpp'>#</span> <span class='hs-varid'>arr</span> <span class='hs-num'>0</span><span class='hs-cpp'>#</span> <span class='hs-varid'>i</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span> <span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-15"></a>  <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-conid'>()</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-16"></a>
<a name="line-17"></a><a name="FastMutPtr"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-layout'>(</span><span class='hs-conid'>MutableByteArray</span><span class='hs-cpp'>#</span> <span class='hs-conid'>RealWorld</span><span class='hs-layout'>)</span>
<a name="line-18"></a>
<a name="line-19"></a><a name="newFastMutPtr"></a><span class='hs-definition'>newFastMutPtr</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>IO</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-20"></a>  <span class='hs-keyword'>case</span> <span class='hs-varid'>newByteArray</span><span class='hs-cpp'>#</span> <span class='hs-varid'>size</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span> <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-varid'>arr</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-21"></a>  <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-varid'>arr</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-22"></a>  <span class='hs-keyword'>where</span> <span class='hs-varop'>!</span><span class='hs-layout'>(</span><span class='hs-conid'>I</span><span class='hs-cpp'>#</span> <span class='hs-varid'>size</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>SIZEOF_VOID_P</span>
<a name="line-23"></a>
<a name="line-24"></a><a name="readFastMutPtr"></a><span class='hs-definition'>readFastMutPtr</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutPtr</span> <span class='hs-varid'>arr</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>IO</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-25"></a>  <span class='hs-keyword'>case</span> <span class='hs-varid'>readAddrArray</span><span class='hs-cpp'>#</span> <span class='hs-varid'>arr</span> <span class='hs-num'>0</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span> <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-varid'>i</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-26"></a>  <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-conid'>Ptr</span> <span class='hs-varid'>i</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-27"></a>
<a name="line-28"></a><a name="writeFastMutPtr"></a><span class='hs-definition'>writeFastMutPtr</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutPtr</span> <span class='hs-varid'>arr</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>Ptr</span> <span class='hs-varid'>i</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>IO</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-29"></a>  <span class='hs-keyword'>case</span> <span class='hs-varid'>writeAddrArray</span><span class='hs-cpp'>#</span> <span class='hs-varid'>arr</span> <span class='hs-num'>0</span><span class='hs-cpp'>#</span> <span class='hs-varid'>i</span> <span class='hs-varid'>s</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span> <span class='hs-varid'>s</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-30"></a>  <span class='hs-layout'>(</span><span class='hs-cpp'>#</span> <span class='hs-varid'>s</span><span class='hs-layout'>,</span> <span class='hs-conid'>()</span> <span class='hs-cpp'>#</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-31"></a><span class='hs-cpp'>#else /* ! __GLASGOW_HASKELL__ */</span>
<a name="line-32"></a><span class='hs-comment'>--maybe someday we could use</span>
<a name="line-33"></a><span class='hs-varop'>--http://haskell.org/haskellwiki/Library/ArrayRef</span>
<a name="line-34"></a><a name="FastMutInt"></a><span class='hs-comment'>--which has an implementation of IOURefs</span>
<a name="line-35"></a><a name="FastMutInt"></a><span class='hs-comment'>--that is unboxed in GHC and just strict in all other compilers...</span>
<a name="line-36"></a><a name="FastMutInt"></a><span class='hs-keyword'>newtype</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-layout'>(</span><span class='hs-conid'>IORef</span> <span class='hs-conid'>Int</span><span class='hs-layout'>)</span>
<a name="line-37"></a>
<a name="line-38"></a><span class='hs-comment'>-- If any default value was chosen, it surely would be 0,</span>
<a name="line-39"></a><span class='hs-comment'>-- so we will use that since IORef requires a default value.</span>
<a name="line-40"></a><span class='hs-comment'>-- Or maybe it would be more interesting to package an error,</span>
<a name="line-41"></a><span class='hs-comment'>-- assuming nothing relies on being able to read a bogus Int?</span>
<a name="line-42"></a><span class='hs-comment'>-- That could interfere with its strictness for smart optimizers</span>
<a name="line-43"></a><span class='hs-comment'>-- (are they allowed to optimize a 'newtype' that way?) ...</span>
<a name="line-44"></a><span class='hs-comment'>-- Well, maybe that can be added (in DEBUG?) later.</span>
<a name="line-45"></a><span class='hs-definition'>newFastMutInt</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>fmap</span> <span class='hs-conid'>FastMutInt</span> <span class='hs-layout'>(</span><span class='hs-varid'>newIORef</span> <span class='hs-num'>0</span><span class='hs-layout'>)</span>
<a name="line-46"></a>
<a name="line-47"></a><span class='hs-definition'>readFastMutInt</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutInt</span> <span class='hs-varid'>ioRefInt</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>readIORef</span> <span class='hs-varid'>ioRefInt</span>
<a name="line-48"></a>
<a name="line-49"></a><span class='hs-comment'>-- FastMutInt is strict in the value it contains.</span>
<a name="line-50"></a><span class='hs-definition'>writeFastMutInt</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutInt</span> <span class='hs-varid'>ioRefInt</span><span class='hs-layout'>)</span> <span class='hs-varid'>i</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>i</span> <span class='hs-varop'>`seq`</span> <span class='hs-varid'>writeIORef</span> <span class='hs-varid'>ioRefInt</span> <span class='hs-varid'>i</span>
<a name="line-51"></a>
<a name="line-52"></a>
<a name="line-53"></a><a name="FastMutPtr"></a><span class='hs-keyword'>newtype</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-layout'>(</span><span class='hs-conid'>IORef</span> <span class='hs-layout'>(</span><span class='hs-conid'>Ptr</span> <span class='hs-conid'>()</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-54"></a>
<a name="line-55"></a><span class='hs-comment'>-- If any default value was chosen, it surely would be 0,</span>
<a name="line-56"></a><span class='hs-comment'>-- so we will use that since IORef requires a default value.</span>
<a name="line-57"></a><span class='hs-comment'>-- Or maybe it would be more interesting to package an error,</span>
<a name="line-58"></a><span class='hs-comment'>-- assuming nothing relies on being able to read a bogus Ptr?</span>
<a name="line-59"></a><span class='hs-comment'>-- That could interfere with its strictness for smart optimizers</span>
<a name="line-60"></a><span class='hs-comment'>-- (are they allowed to optimize a 'newtype' that way?) ...</span>
<a name="line-61"></a><span class='hs-comment'>-- Well, maybe that can be added (in DEBUG?) later.</span>
<a name="line-62"></a><span class='hs-definition'>newFastMutPtr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>fmap</span> <span class='hs-conid'>FastMutPtr</span> <span class='hs-layout'>(</span><span class='hs-varid'>newIORef</span> <span class='hs-layout'>(</span><span class='hs-varid'>castPtr</span> <span class='hs-varid'>nullPtr</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-63"></a>
<a name="line-64"></a><span class='hs-definition'>readFastMutPtr</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutPtr</span> <span class='hs-varid'>ioRefPtr</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>readIORef</span> <span class='hs-varid'>ioRefPtr</span>
<a name="line-65"></a>
<a name="line-66"></a><span class='hs-comment'>-- FastMutPtr is strict in the value it contains.</span>
<a name="line-67"></a><span class='hs-definition'>writeFastMutPtr</span> <span class='hs-layout'>(</span><span class='hs-conid'>FastMutPtr</span> <span class='hs-varid'>ioRefPtr</span><span class='hs-layout'>)</span> <span class='hs-varid'>i</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>i</span> <span class='hs-varop'>`seq`</span> <span class='hs-varid'>writeIORef</span> <span class='hs-varid'>ioRefPtr</span> <span class='hs-varid'>i</span>
<a name="line-68"></a><span class='hs-cpp'>#endif</span>
</pre>\end{code}

</body>
</html>