<?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__ >= 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'>-></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'>-></span> <span class='hs-conid'>Int</span> <span class='hs-keyglyph'>-></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'>-></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'>-></span> <span class='hs-conid'>Ptr</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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'>-></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>