<?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>Codec/Zlib/Enum.hs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> <pre><a name="line-1"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>Codec</span><span class='hs-varop'>.</span><span class='hs-conid'>Zlib</span><span class='hs-varop'>.</span><span class='hs-conid'>Enum</span> <span class='hs-layout'>(</span> <a name="line-2"></a> <span class='hs-comment'>-- * Enumeratees</span> <a name="line-3"></a> <span class='hs-varid'>compress</span><span class='hs-layout'>,</span> <span class='hs-varid'>decompress</span><span class='hs-layout'>,</span> <span class='hs-varid'>gzip</span><span class='hs-layout'>,</span> <span class='hs-varid'>ungzip</span><span class='hs-layout'>,</span> <a name="line-4"></a> <span class='hs-comment'>-- * Re-exported from zlib-bindings</span> <a name="line-5"></a> <span class='hs-conid'>WindowBits</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-varid'>defaultWindowBits</span> <a name="line-6"></a><span class='hs-layout'>)</span> <span class='hs-keyword'>where</span> <a name="line-7"></a> <a name="line-8"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Codec</span><span class='hs-varop'>.</span><span class='hs-conid'>Zlib</span> <a name="line-9"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Enumerator</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>E</span> <a name="line-10"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Enumerator</span><span class='hs-varop'>.</span><span class='hs-conid'>List</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>EL</span> <a name="line-11"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span><span class='hs-varop'>.</span><span class='hs-conid'>Trans</span><span class='hs-varop'>.</span><span class='hs-conid'>Class</span> <span class='hs-layout'>(</span><span class='hs-varid'>lift</span><span class='hs-layout'>)</span> <a name="line-12"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span><span class='hs-varop'>.</span><span class='hs-conid'>IO</span><span class='hs-varop'>.</span><span class='hs-conid'>Class</span> <span class='hs-layout'>(</span><span class='hs-conid'>MonadIO</span><span class='hs-layout'>,</span> <span class='hs-varid'>liftIO</span><span class='hs-layout'>)</span> <a name="line-13"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>ByteString</span> <span class='hs-layout'>(</span><span class='hs-conid'>ByteString</span><span class='hs-layout'>)</span> <a name="line-14"></a> <a name="line-15"></a><a name="gzip"></a><span class='hs-comment'>-- | Gzip compression with default parameters.</span> <a name="line-16"></a><span class='hs-definition'>gzip</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>MonadIO</span> <span class='hs-varid'>m</span> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>Enumeratee</span> <span class='hs-conid'>ByteString</span> <span class='hs-conid'>ByteString</span> <span class='hs-varid'>m</span> <span class='hs-varid'>a</span> <a name="line-17"></a><span class='hs-definition'>gzip</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>compress</span> <span class='hs-num'>1</span> <span class='hs-layout'>(</span><span class='hs-conid'>WindowBits</span> <span class='hs-num'>31</span><span class='hs-layout'>)</span> <a name="line-18"></a> <a name="line-19"></a><a name="ungzip"></a><span class='hs-comment'>-- | Gzip decompression with default parameters.</span> <a name="line-20"></a><span class='hs-definition'>ungzip</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>MonadIO</span> <span class='hs-varid'>m</span> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>Enumeratee</span> <span class='hs-conid'>ByteString</span> <span class='hs-conid'>ByteString</span> <span class='hs-varid'>m</span> <span class='hs-varid'>a</span> <a name="line-21"></a><span class='hs-definition'>ungzip</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>decompress</span> <span class='hs-layout'>(</span><span class='hs-conid'>WindowBits</span> <span class='hs-num'>31</span><span class='hs-layout'>)</span> <a name="line-22"></a> <a name="line-23"></a><span class='hs-comment'>-- |</span> <a name="line-24"></a><span class='hs-comment'>-- Decompress (inflate) a stream of 'ByteString's. For example:</span> <a name="line-25"></a><span class='hs-comment'>--</span> <a name="line-26"></a><span class='hs-comment'>-- > run $ enumFile "test.z" $$ decompress defaultWindowBits $$ printChunks True</span> <a name="line-27"></a> <a name="line-28"></a><a name="decompress"></a><span class='hs-definition'>decompress</span> <a name="line-29"></a> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>MonadIO</span> <span class='hs-varid'>m</span> <a name="line-30"></a> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>WindowBits</span> <span class='hs-comment'>-- ^ Zlib parameter (see the zlib-bindings package as well as the zlib C library)</span> <a name="line-31"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Enumeratee</span> <span class='hs-conid'>ByteString</span> <span class='hs-conid'>ByteString</span> <span class='hs-varid'>m</span> <span class='hs-varid'>a</span> <a name="line-32"></a><span class='hs-definition'>decompress</span> <span class='hs-varid'>config</span> <span class='hs-varid'>inner</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-33"></a> <span class='hs-varid'>inf</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftIO</span> <span class='hs-varop'>$</span> <span class='hs-varid'>initInflate</span> <span class='hs-varid'>config</span> <a name="line-34"></a> <span class='hs-varid'>decompress'</span> <span class='hs-varid'>inf</span> <span class='hs-varid'>inner</span> <a name="line-35"></a> <a name="line-36"></a><a name="decompress'"></a><span class='hs-definition'>decompress'</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>MonadIO</span> <span class='hs-varid'>m</span> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>Inflate</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Enumeratee</span> <span class='hs-conid'>ByteString</span> <span class='hs-conid'>ByteString</span> <span class='hs-varid'>m</span> <span class='hs-varid'>b</span> <a name="line-37"></a><span class='hs-definition'>decompress'</span> <span class='hs-varid'>inf</span> <span class='hs-layout'>(</span><span class='hs-conid'>Continue</span> <span class='hs-varid'>k</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-38"></a> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-conid'>EL</span><span class='hs-varop'>.</span><span class='hs-varid'>head</span> <a name="line-39"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>x</span> <span class='hs-keyword'>of</span> <a name="line-40"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-41"></a> <span class='hs-varid'>chunk</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftIO</span> <span class='hs-varop'>$</span> <span class='hs-varid'>finishInflate</span> <span class='hs-varid'>inf</span> <a name="line-42"></a> <span class='hs-varid'>lift</span> <span class='hs-varop'>$</span> <span class='hs-varid'>runIteratee</span> <span class='hs-varop'>$</span> <span class='hs-varid'>k</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Chunks</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>chunk</span><span class='hs-keyglyph'>]</span> <a name="line-43"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>bs</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-44"></a> <span class='hs-varid'>chunks</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftIO</span> <span class='hs-varop'>$</span> <span class='hs-varid'>withInflateInput</span> <span class='hs-varid'>inf</span> <span class='hs-varid'>bs</span> <span class='hs-varop'>$</span> <span class='hs-varid'>callback</span> <a name="line-45"></a> <span class='hs-varid'>step</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>lift</span> <span class='hs-varop'>$</span> <span class='hs-varid'>runIteratee</span> <span class='hs-varop'>$</span> <span class='hs-varid'>k</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Chunks</span> <span class='hs-varid'>chunks</span> <a name="line-46"></a> <span class='hs-varid'>decompress'</span> <span class='hs-varid'>inf</span> <span class='hs-varid'>step</span> <a name="line-47"></a><span class='hs-definition'>decompress'</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>step</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-varid'>step</span> <a name="line-48"></a> <a name="line-49"></a><span class='hs-comment'>-- |</span> <a name="line-50"></a><span class='hs-comment'>-- Compress (deflate) a stream of 'ByteString's. The 'WindowBits' also control</span> <a name="line-51"></a><span class='hs-comment'>-- the format (zlib vs. gzip).</span> <a name="line-52"></a> <a name="line-53"></a><a name="compress"></a><span class='hs-definition'>compress</span> <a name="line-54"></a> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>MonadIO</span> <span class='hs-varid'>m</span> <a name="line-55"></a> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>Int</span> <span class='hs-comment'>-- ^ Compression level</span> <a name="line-56"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>WindowBits</span> <span class='hs-comment'>-- ^ Zlib parameter (see the zlib-bindings package as well as the zlib C library)</span> <a name="line-57"></a> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Enumeratee</span> <span class='hs-conid'>ByteString</span> <span class='hs-conid'>ByteString</span> <span class='hs-varid'>m</span> <span class='hs-varid'>a</span> <a name="line-58"></a><span class='hs-definition'>compress</span> <span class='hs-varid'>level</span> <span class='hs-varid'>config</span> <span class='hs-varid'>inner</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-59"></a> <span class='hs-varid'>def</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftIO</span> <span class='hs-varop'>$</span> <span class='hs-varid'>initDeflate</span> <span class='hs-varid'>level</span> <span class='hs-varid'>config</span> <a name="line-60"></a> <span class='hs-varid'>compress'</span> <span class='hs-varid'>def</span> <span class='hs-varid'>inner</span> <a name="line-61"></a> <a name="line-62"></a><a name="compress'"></a><span class='hs-definition'>compress'</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>MonadIO</span> <span class='hs-varid'>m</span> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>Deflate</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Enumeratee</span> <span class='hs-conid'>ByteString</span> <span class='hs-conid'>ByteString</span> <span class='hs-varid'>m</span> <span class='hs-varid'>b</span> <a name="line-63"></a><span class='hs-definition'>compress'</span> <span class='hs-varid'>def</span> <span class='hs-layout'>(</span><span class='hs-conid'>Continue</span> <span class='hs-varid'>k</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-64"></a> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-conid'>EL</span><span class='hs-varop'>.</span><span class='hs-varid'>head</span> <a name="line-65"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>x</span> <span class='hs-keyword'>of</span> <a name="line-66"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-67"></a> <span class='hs-varid'>chunks</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftIO</span> <span class='hs-varop'>$</span> <span class='hs-varid'>finishDeflate</span> <span class='hs-varid'>def</span> <span class='hs-varop'>$</span> <span class='hs-varid'>callback</span> <a name="line-68"></a> <span class='hs-varid'>lift</span> <span class='hs-varop'>$</span> <span class='hs-varid'>runIteratee</span> <span class='hs-varop'>$</span> <span class='hs-varid'>k</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Chunks</span> <span class='hs-varid'>chunks</span> <a name="line-69"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>bs</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-70"></a> <span class='hs-varid'>chunks</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>liftIO</span> <span class='hs-varop'>$</span> <span class='hs-varid'>withDeflateInput</span> <span class='hs-varid'>def</span> <span class='hs-varid'>bs</span> <span class='hs-varop'>$</span> <span class='hs-varid'>callback</span> <a name="line-71"></a> <span class='hs-varid'>step</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>lift</span> <span class='hs-varop'>$</span> <span class='hs-varid'>runIteratee</span> <span class='hs-varop'>$</span> <span class='hs-varid'>k</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Chunks</span> <span class='hs-varid'>chunks</span> <a name="line-72"></a> <span class='hs-varid'>compress'</span> <span class='hs-varid'>def</span> <span class='hs-varid'>step</span> <a name="line-73"></a><span class='hs-definition'>compress'</span> <span class='hs-keyword'>_</span> <span class='hs-varid'>step</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-varid'>step</span> <a name="line-74"></a> <a name="line-75"></a><a name="callback"></a><span class='hs-definition'>callback</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Monad</span> <span class='hs-varid'>m</span> <span class='hs-keyglyph'>=></span> <span class='hs-varid'>m</span> <span class='hs-layout'>(</span><span class='hs-conid'>Maybe</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>m</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span> <a name="line-76"></a><span class='hs-definition'>callback</span> <span class='hs-varid'>pop</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>go</span> <span class='hs-varid'>id</span> <span class='hs-keyword'>where</span> <a name="line-77"></a> <span class='hs-varid'>go</span> <span class='hs-varid'>front</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <a name="line-78"></a> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>pop</span> <a name="line-79"></a> <span class='hs-keyword'>case</span> <span class='hs-varid'>x</span> <span class='hs-keyword'>of</span> <a name="line-80"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>front</span> <span class='hs-conid'>[]</span> <a name="line-81"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>y</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>go</span> <span class='hs-layout'>(</span><span class='hs-varid'>front</span> <span class='hs-varop'>.</span> <span class='hs-layout'>(</span><span class='hs-conop'>:</span><span class='hs-layout'>)</span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span> </pre></body> </html>