<?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://code.haskell.org/~malcolm/hscolour/ --> <title>src/full/Agda/Compiler/Epic/CaseOpts.hs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> <pre><a name="line-1"></a><span class='hs-comment'>-- | Perform simple optimisations based on case-laws</span> <a name="line-2"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>Agda</span><span class='hs-varop'>.</span><span class='hs-conid'>Compiler</span><span class='hs-varop'>.</span><span class='hs-conid'>Epic</span><span class='hs-varop'>.</span><span class='hs-conid'>CaseOpts</span> <span class='hs-keyword'>where</span> <a name="line-3"></a> <a name="line-4"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Applicative</span> <a name="line-5"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span> <a name="line-6"></a> <a name="line-7"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Agda</span><span class='hs-varop'>.</span><span class='hs-conid'>TypeChecking</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span> <a name="line-8"></a> <a name="line-9"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Agda</span><span class='hs-varop'>.</span><span class='hs-conid'>Compiler</span><span class='hs-varop'>.</span><span class='hs-conid'>Epic</span><span class='hs-varop'>.</span><span class='hs-conid'>AuxAST</span> <a name="line-10"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Agda</span><span class='hs-varop'>.</span><span class='hs-conid'>Compiler</span><span class='hs-varop'>.</span><span class='hs-conid'>Epic</span><span class='hs-varop'>.</span><span class='hs-conid'>CompileState</span> <a name="line-11"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Agda</span><span class='hs-varop'>.</span><span class='hs-conid'>Compiler</span><span class='hs-varop'>.</span><span class='hs-conid'>Epic</span><span class='hs-varop'>.</span><span class='hs-conid'>Interface</span> <a name="line-12"></a> <a name="line-13"></a><a name="caseOpts"></a><span class='hs-definition'>caseOpts</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Fun</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Compile</span> <span class='hs-conid'>TCM</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Fun</span><span class='hs-keyglyph'>]</span> <a name="line-14"></a><span class='hs-definition'>caseOpts</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mapM</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span> <span class='hs-varid'>def</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>case</span> <span class='hs-varid'>def</span> <span class='hs-keyword'>of</span> <a name="line-15"></a> <span class='hs-conid'>Fun</span><span class='hs-layout'>{</span><span class='hs-layout'>}</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-16"></a> <span class='hs-varid'>e'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-layout'>(</span><span class='hs-varid'>funExpr</span> <span class='hs-varid'>def</span><span class='hs-layout'>)</span> <a name="line-17"></a> <span class='hs-varid'>return</span> <span class='hs-varid'>def</span> <span class='hs-layout'>{</span> <span class='hs-varid'>funExpr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>e'</span> <span class='hs-layout'>}</span> <a name="line-18"></a> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-varid'>def</span> <a name="line-19"></a> <a name="line-20"></a><a name="caseOptsExpr"></a><span class='hs-comment'>-- | Run the case-opts on an expression</span> <a name="line-21"></a><span class='hs-definition'>caseOptsExpr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Expr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Compile</span> <span class='hs-conid'>TCM</span> <span class='hs-conid'>Expr</span> <a name="line-22"></a><span class='hs-definition'>caseOptsExpr</span> <span class='hs-varid'>expr</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>expr</span> <span class='hs-keyword'>of</span> <a name="line-23"></a> <span class='hs-conid'>Var</span> <span class='hs-varid'>v</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>v</span> <a name="line-24"></a> <span class='hs-conid'>Lit</span> <span class='hs-varid'>l</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Lit</span> <span class='hs-varid'>l</span> <a name="line-25"></a> <span class='hs-conid'>Lam</span> <span class='hs-varid'>v</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Lam</span> <span class='hs-varid'>v</span> <span class='hs-varop'><$></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>e</span> <a name="line-26"></a> <span class='hs-conid'>Con</span> <span class='hs-varid'>c</span> <span class='hs-varid'>n</span> <span class='hs-varid'>es</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Con</span> <span class='hs-varid'>c</span> <span class='hs-varid'>n</span> <span class='hs-varop'><$></span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>es</span> <a name="line-27"></a> <span class='hs-conid'>App</span> <span class='hs-varid'>v</span> <span class='hs-varid'>es</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>apps</span> <span class='hs-varid'>v</span> <span class='hs-varop'><$></span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>es</span> <a name="line-28"></a> <span class='hs-comment'>-- If there is only one constructor branch, perform a projection to get the result</span> <a name="line-29"></a> <span class='hs-conid'>Case</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Branch</span> <span class='hs-layout'>{</span><span class='hs-varid'>brVars</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>vs</span><span class='hs-layout'>,</span> <span class='hs-varid'>brExpr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>eorg</span><span class='hs-layout'>}</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <a name="line-30"></a> <span class='hs-varid'>e'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>eorg</span> <a name="line-31"></a> <span class='hs-varid'>bindExpr</span> <span class='hs-varid'>e</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>var</span> <span class='hs-keyglyph'>-></span> <a name="line-32"></a> <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-varid'>foldr</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-layout'>(</span><span class='hs-varid'>v</span><span class='hs-layout'>,</span> <span class='hs-varid'>n</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>lett</span> <span class='hs-varid'>v</span> <span class='hs-varop'>$</span> <span class='hs-conid'>App</span> <span class='hs-str'>"proj"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Lit</span> <span class='hs-layout'>(</span><span class='hs-conid'>LInt</span> <span class='hs-varid'>n</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-conid'>Var</span> <span class='hs-varid'>var</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <span class='hs-varid'>e'</span> <a name="line-33"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>zip</span> <span class='hs-varid'>vs</span> <span class='hs-keyglyph'>[</span><span class='hs-num'>0</span><span class='hs-keyglyph'>..</span><span class='hs-keyglyph'>]</span> <a name="line-34"></a> <span class='hs-comment'>-- If there is only a default branch, the case is not necessary</span> <a name="line-35"></a> <span class='hs-conid'>Case</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Default</span><span class='hs-layout'>{</span><span class='hs-varid'>brExpr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>e'</span><span class='hs-layout'>}</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>e'</span> <a name="line-36"></a> <span class='hs-conid'>Case</span> <span class='hs-varid'>e</span> <span class='hs-varid'>brs</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Case</span> <span class='hs-varop'><$></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>e</span> <span class='hs-varop'><*></span> <span class='hs-layout'>(</span><span class='hs-varid'>mapM</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>br</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span> <span class='hs-varid'>e'</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-layout'>(</span><span class='hs-varid'>brExpr</span> <span class='hs-varid'>br</span><span class='hs-layout'>)</span> <a name="line-37"></a> <span class='hs-varid'>return</span> <span class='hs-varid'>br</span> <span class='hs-layout'>{</span><span class='hs-varid'>brExpr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>e'</span><span class='hs-layout'>}</span><span class='hs-layout'>)</span> <span class='hs-varid'>brs</span><span class='hs-layout'>)</span> <a name="line-38"></a> <span class='hs-conid'>If</span> <span class='hs-varid'>a</span> <span class='hs-varid'>b</span> <span class='hs-varid'>c</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>If</span> <span class='hs-varop'><$></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>a</span> <span class='hs-varop'><*></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>b</span> <span class='hs-varop'><*></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>c</span> <a name="line-39"></a> <span class='hs-conid'>Let</span> <span class='hs-varid'>v</span> <span class='hs-varid'>e1</span> <span class='hs-varid'>e2</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Let</span> <span class='hs-varid'>v</span> <span class='hs-varop'><$></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>e1</span> <span class='hs-varop'><*></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>e2</span> <a name="line-40"></a> <span class='hs-conid'>Lazy</span> <span class='hs-varid'>e</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Lazy</span> <span class='hs-varop'><$></span> <span class='hs-varid'>caseOptsExpr</span> <span class='hs-varid'>e</span> <a name="line-41"></a> <span class='hs-conid'>UNIT</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-conid'>UNIT</span> <a name="line-42"></a> <span class='hs-conid'>IMPOSSIBLE</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-conid'>IMPOSSIBLE</span> </pre></body> </html>