<?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>codeGen/StgCmmEnv.hs</title>
<link type='text/css' rel='stylesheet' href='hscolour.css' />
</head>
<body>
<pre><a name="line-1"></a><span class='hs-comment'>-----------------------------------------------------------------------------</span>
<a name="line-2"></a><span class='hs-comment'>--</span>
<a name="line-3"></a><span class='hs-comment'>-- Stg to C-- code generation: the binding environment</span>
<a name="line-4"></a><span class='hs-comment'>--</span>
<a name="line-5"></a><span class='hs-comment'>-- (c) The University of Glasgow 2004-2006</span>
<a name="line-6"></a><span class='hs-comment'>--</span>
<a name="line-7"></a><span class='hs-comment'>-----------------------------------------------------------------------------</span>
<a name="line-8"></a>
<a name="line-9"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>StgCmmEnv</span> <span class='hs-layout'>(</span>
<a name="line-10"></a> <span class='hs-conid'>CgIdInfo</span><span class='hs-layout'>,</span>
<a name="line-11"></a>
<a name="line-12"></a> <span class='hs-varid'>cgIdInfoId</span><span class='hs-layout'>,</span> <span class='hs-varid'>cgIdInfoLF</span><span class='hs-layout'>,</span>
<a name="line-13"></a>
<a name="line-14"></a> <span class='hs-varid'>litIdInfo</span><span class='hs-layout'>,</span> <span class='hs-varid'>lneIdInfo</span><span class='hs-layout'>,</span> <span class='hs-varid'>regIdInfo</span><span class='hs-layout'>,</span>
<a name="line-15"></a> <span class='hs-varid'>idInfoToAmode</span><span class='hs-layout'>,</span>
<a name="line-16"></a>
<a name="line-17"></a> <span class='hs-conid'>NonVoid</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'>isVoidId</span><span class='hs-layout'>,</span> <span class='hs-varid'>nonVoidIds</span><span class='hs-layout'>,</span>
<a name="line-18"></a>
<a name="line-19"></a> <span class='hs-varid'>addBindC</span><span class='hs-layout'>,</span> <span class='hs-varid'>addBindsC</span><span class='hs-layout'>,</span>
<a name="line-20"></a>
<a name="line-21"></a> <span class='hs-varid'>bindArgsToRegs</span><span class='hs-layout'>,</span> <span class='hs-varid'>bindToReg</span><span class='hs-layout'>,</span> <span class='hs-varid'>rebindToReg</span><span class='hs-layout'>,</span>
<a name="line-22"></a> <span class='hs-varid'>bindArgToReg</span><span class='hs-layout'>,</span> <span class='hs-varid'>idToReg</span><span class='hs-layout'>,</span>
<a name="line-23"></a> <span class='hs-varid'>getArgAmode</span><span class='hs-layout'>,</span> <span class='hs-varid'>getNonVoidArgAmodes</span><span class='hs-layout'>,</span>
<a name="line-24"></a> <span class='hs-varid'>getCgIdInfo</span><span class='hs-layout'>,</span>
<a name="line-25"></a> <span class='hs-varid'>maybeLetNoEscape</span><span class='hs-layout'>,</span>
<a name="line-26"></a> <span class='hs-layout'>)</span> <span class='hs-keyword'>where</span>
<a name="line-27"></a>
<a name="line-28"></a><span class='hs-cpp'>#include "HsVersions.h"</span>
<a name="line-29"></a>
<a name="line-30"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>TyCon</span>
<a name="line-31"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>StgCmmMonad</span>
<a name="line-32"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>StgCmmUtils</span>
<a name="line-33"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>StgCmmClosure</span>
<a name="line-34"></a>
<a name="line-35"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CLabel</span>
<a name="line-36"></a>
<a name="line-37"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>BlockId</span>
<a name="line-38"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Cmm</span>
<a name="line-39"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>CmmUtils</span>
<a name="line-40"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>FastString</span>
<a name="line-41"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>PprCmm</span> <span class='hs-layout'>(</span> <span class='hs-comment'>{- instance Outputable -}</span> <span class='hs-layout'>)</span>
<a name="line-42"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Id</span>
<a name="line-43"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>VarEnv</span>
<a name="line-44"></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-45"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Name</span>
<a name="line-46"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>StgSyn</span>
<a name="line-47"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Outputable</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'>-- Non-void types</span>
<a name="line-51"></a><span class='hs-comment'>-------------------------------------</span>
<a name="line-52"></a><span class='hs-comment'>-- We frequently need the invariant that an Id or a an argument</span>
<a name="line-53"></a><span class='hs-comment'>-- is of a non-void type. This type is a witness to the invariant.</span>
<a name="line-54"></a>
<a name="line-55"></a><a name="NonVoid"></a><span class='hs-keyword'>newtype</span> <span class='hs-conid'>NonVoid</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>NonVoid</span> <span class='hs-varid'>a</span>
<a name="line-56"></a> <span class='hs-keyword'>deriving</span> <span class='hs-layout'>(</span><span class='hs-conid'>Eq</span><span class='hs-layout'>,</span> <span class='hs-conid'>Show</span><span class='hs-layout'>)</span>
<a name="line-57"></a>
<a name="line-58"></a><span class='hs-keyword'>instance</span> <span class='hs-layout'>(</span><span class='hs-conid'>Outputable</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>Outputable</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyword'>where</span>
<a name="line-59"></a> <span class='hs-varid'>ppr</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>ppr</span> <span class='hs-varid'>a</span>
<a name="line-60"></a>
<a name="line-61"></a><a name="isVoidId"></a><span class='hs-definition'>isVoidId</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Bool</span>
<a name="line-62"></a><span class='hs-definition'>isVoidId</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>isVoidRep</span> <span class='hs-varop'>.</span> <span class='hs-varid'>idPrimRep</span>
<a name="line-63"></a>
<a name="line-64"></a><a name="nonVoidIds"></a><span class='hs-definition'>nonVoidIds</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Id</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>NonVoid</span> <span class='hs-conid'>Id</span><span class='hs-keyglyph'>]</span>
<a name="line-65"></a><span class='hs-definition'>nonVoidIds</span> <span class='hs-varid'>ids</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>id</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>id</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>ids</span><span class='hs-layout'>,</span> <span class='hs-varid'>not</span> <span class='hs-layout'>(</span><span class='hs-varid'>isVoidRep</span> <span class='hs-layout'>(</span><span class='hs-varid'>idPrimRep</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span>
<a name="line-66"></a>
<a name="line-67"></a><span class='hs-comment'>-------------------------------------</span>
<a name="line-68"></a><span class='hs-comment'>-- Manipulating CgIdInfo</span>
<a name="line-69"></a><span class='hs-comment'>-------------------------------------</span>
<a name="line-70"></a>
<a name="line-71"></a><a name="mkCgIdInfo"></a><span class='hs-definition'>mkCgIdInfo</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>LambdaFormInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CmmExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CgIdInfo</span>
<a name="line-72"></a><span class='hs-definition'>mkCgIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-varid'>lf</span> <span class='hs-varid'>expr</span>
<a name="line-73"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>CgIdInfo</span> <span class='hs-layout'>{</span> <span class='hs-varid'>cg_id</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>id</span><span class='hs-layout'>,</span> <span class='hs-varid'>cg_loc</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>CmmLoc</span> <span class='hs-varid'>expr</span><span class='hs-layout'>,</span>
<a name="line-74"></a> <span class='hs-varid'>cg_lf</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>lf</span><span class='hs-layout'>,</span> <span class='hs-varid'>cg_rep</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>idPrimRep</span> <span class='hs-varid'>id</span><span class='hs-layout'>,</span>
<a name="line-75"></a> <span class='hs-varid'>cg_tag</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>lfDynTag</span> <span class='hs-varid'>lf</span> <span class='hs-layout'>}</span>
<a name="line-76"></a>
<a name="line-77"></a><a name="lneIdInfo"></a><span class='hs-definition'>lneIdInfo</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>LocalReg</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CgIdInfo</span>
<a name="line-78"></a><span class='hs-definition'>lneIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-varid'>regs</span>
<a name="line-79"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>CgIdInfo</span> <span class='hs-layout'>{</span> <span class='hs-varid'>cg_id</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>id</span><span class='hs-layout'>,</span> <span class='hs-varid'>cg_loc</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>LneLoc</span> <span class='hs-varid'>blk_id</span> <span class='hs-varid'>regs</span><span class='hs-layout'>,</span>
<a name="line-80"></a> <span class='hs-varid'>cg_lf</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>lf</span><span class='hs-layout'>,</span> <span class='hs-varid'>cg_rep</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>idPrimRep</span> <span class='hs-varid'>id</span><span class='hs-layout'>,</span>
<a name="line-81"></a> <span class='hs-varid'>cg_tag</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>lfDynTag</span> <span class='hs-varid'>lf</span> <span class='hs-layout'>}</span>
<a name="line-82"></a> <span class='hs-keyword'>where</span>
<a name="line-83"></a> <span class='hs-varid'>lf</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkLFLetNoEscape</span>
<a name="line-84"></a> <span class='hs-varid'>blk_id</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mkBlockId</span> <span class='hs-layout'>(</span><span class='hs-varid'>idUnique</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span>
<a name="line-85"></a>
<a name="line-86"></a><a name="litIdInfo"></a><span class='hs-definition'>litIdInfo</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>LambdaFormInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CmmLit</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CgIdInfo</span>
<a name="line-87"></a><span class='hs-definition'>litIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-varid'>lf_info</span> <span class='hs-varid'>lit</span> <span class='hs-keyglyph'>=</span> <span class='hs-comment'>--mkCgIdInfo id lf_info (CmmLit lit)</span>
<a name="line-88"></a> <span class='hs-varid'>mkCgIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-varid'>lf_info</span> <span class='hs-layout'>(</span><span class='hs-varid'>addDynTag</span> <span class='hs-layout'>(</span><span class='hs-conid'>CmmLit</span> <span class='hs-varid'>lit</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>lfDynTag</span> <span class='hs-varid'>lf_info</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-89"></a>
<a name="line-90"></a><a name="regIdInfo"></a><span class='hs-definition'>regIdInfo</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>LambdaFormInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>LocalReg</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CgIdInfo</span>
<a name="line-91"></a><span class='hs-definition'>regIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-varid'>lf_info</span> <span class='hs-varid'>reg</span> <span class='hs-keyglyph'>=</span>
<a name="line-92"></a> <span class='hs-varid'>mkCgIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-varid'>lf_info</span> <span class='hs-layout'>(</span><span class='hs-varid'>addDynTag</span> <span class='hs-layout'>(</span><span class='hs-conid'>CmmReg</span> <span class='hs-layout'>(</span><span class='hs-conid'>CmmLocal</span> <span class='hs-varid'>reg</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>lfDynTag</span> <span class='hs-varid'>lf_info</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-93"></a>
<a name="line-94"></a><a name="idInfoToAmode"></a><span class='hs-definition'>idInfoToAmode</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CgIdInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CmmExpr</span>
<a name="line-95"></a><span class='hs-comment'>-- Returns a CmmExpr for the *tagged* pointer</span>
<a name="line-96"></a><span class='hs-definition'>idInfoToAmode</span> <span class='hs-layout'>(</span><span class='hs-conid'>CgIdInfo</span> <span class='hs-layout'>{</span> <span class='hs-varid'>cg_loc</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>CmmLoc</span> <span class='hs-varid'>e</span> <span class='hs-layout'>}</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>e</span>
<a name="line-97"></a><span class='hs-definition'>idInfoToAmode</span> <span class='hs-varid'>cg_info</span>
<a name="line-98"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>pprPanic</span> <span class='hs-str'>"idInfoToAmode"</span> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-layout'>(</span><span class='hs-varid'>cg_id</span> <span class='hs-varid'>cg_info</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-comment'>-- LneLoc</span>
<a name="line-99"></a>
<a name="line-100"></a><a name="addDynTag"></a><span class='hs-definition'>addDynTag</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CmmExpr</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>DynTag</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CmmExpr</span>
<a name="line-101"></a><span class='hs-comment'>-- A tag adds a byte offset to the pointer</span>
<a name="line-102"></a><span class='hs-definition'>addDynTag</span> <span class='hs-varid'>expr</span> <span class='hs-varid'>tag</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>cmmOffsetB</span> <span class='hs-varid'>expr</span> <span class='hs-varid'>tag</span>
<a name="line-103"></a>
<a name="line-104"></a><a name="cgIdInfoId"></a><span class='hs-definition'>cgIdInfoId</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CgIdInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Id</span>
<a name="line-105"></a><span class='hs-definition'>cgIdInfoId</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>cg_id</span>
<a name="line-106"></a>
<a name="line-107"></a><a name="cgIdInfoLF"></a><span class='hs-definition'>cgIdInfoLF</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CgIdInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>LambdaFormInfo</span>
<a name="line-108"></a><span class='hs-definition'>cgIdInfoLF</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>cg_lf</span>
<a name="line-109"></a>
<a name="line-110"></a><a name="maybeLetNoEscape"></a><span class='hs-definition'>maybeLetNoEscape</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>CgIdInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Maybe</span> <span class='hs-layout'>(</span><span class='hs-conid'>BlockId</span><span class='hs-layout'>,</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>LocalReg</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-111"></a><span class='hs-definition'>maybeLetNoEscape</span> <span class='hs-layout'>(</span><span class='hs-conid'>CgIdInfo</span> <span class='hs-layout'>{</span> <span class='hs-varid'>cg_loc</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>LneLoc</span> <span class='hs-varid'>blk_id</span> <span class='hs-varid'>args</span><span class='hs-layout'>}</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Just</span> <span class='hs-layout'>(</span><span class='hs-varid'>blk_id</span><span class='hs-layout'>,</span> <span class='hs-varid'>args</span><span class='hs-layout'>)</span>
<a name="line-112"></a><span class='hs-definition'>maybeLetNoEscape</span> <span class='hs-sel'>_other</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Nothing</span>
<a name="line-113"></a>
<a name="line-114"></a>
<a name="line-115"></a>
<a name="line-116"></a><span class='hs-comment'>---------------------------------------------------------</span>
<a name="line-117"></a><span class='hs-comment'>-- The binding environment</span>
<a name="line-118"></a><span class='hs-comment'>-- </span>
<a name="line-119"></a><span class='hs-comment'>-- There are three basic routines, for adding (addBindC), </span>
<a name="line-120"></a><span class='hs-comment'>-- modifying(modifyBindC) and looking up (getCgIdInfo) bindings.</span>
<a name="line-121"></a><span class='hs-comment'>---------------------------------------------------------</span>
<a name="line-122"></a>
<a name="line-123"></a><a name="addBindC"></a><span class='hs-definition'>addBindC</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>CgIdInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-conid'>()</span>
<a name="line-124"></a><span class='hs-definition'>addBindC</span> <span class='hs-varid'>name</span> <span class='hs-varid'>stuff_to_bind</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span>
<a name="line-125"></a> <span class='hs-varid'>binds</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getBinds</span>
<a name="line-126"></a> <span class='hs-varid'>setBinds</span> <span class='hs-varop'>$</span> <span class='hs-varid'>extendVarEnv</span> <span class='hs-varid'>binds</span> <span class='hs-varid'>name</span> <span class='hs-varid'>stuff_to_bind</span>
<a name="line-127"></a>
<a name="line-128"></a><a name="addBindsC"></a><span class='hs-definition'>addBindsC</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>CgIdInfo</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-conid'>()</span>
<a name="line-129"></a><span class='hs-definition'>addBindsC</span> <span class='hs-varid'>new_bindings</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span>
<a name="line-130"></a> <span class='hs-varid'>binds</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getBinds</span>
<a name="line-131"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>new_binds</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>foldl</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span> <span class='hs-varid'>binds</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>extendVarEnv</span> <span class='hs-varid'>binds</span> <span class='hs-layout'>(</span><span class='hs-varid'>cg_id</span> <span class='hs-varid'>info</span><span class='hs-layout'>)</span> <span class='hs-varid'>info</span><span class='hs-layout'>)</span>
<a name="line-132"></a> <span class='hs-varid'>binds</span>
<a name="line-133"></a> <span class='hs-varid'>new_bindings</span>
<a name="line-134"></a> <span class='hs-varid'>setBinds</span> <span class='hs-varid'>new_binds</span>
<a name="line-135"></a>
<a name="line-136"></a><a name="getCgIdInfo"></a><span class='hs-definition'>getCgIdInfo</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-conid'>CgIdInfo</span>
<a name="line-137"></a><span class='hs-definition'>getCgIdInfo</span> <span class='hs-varid'>id</span>
<a name="line-138"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <span class='hs-layout'>{</span> <span class='hs-comment'>-- Try local bindings first</span>
<a name="line-139"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>local_binds</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getBinds</span>
<a name="line-140"></a> <span class='hs-layout'>;</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>lookupVarEnv</span> <span class='hs-varid'>local_binds</span> <span class='hs-varid'>id</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span>
<a name="line-141"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-varid'>info</span> <span class='hs-layout'>;</span>
<a name="line-142"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span>
<a name="line-143"></a>
<a name="line-144"></a> <span class='hs-layout'>{</span> <span class='hs-comment'>-- Try top-level bindings</span>
<a name="line-145"></a> <span class='hs-varid'>static_binds</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getStaticBinds</span>
<a name="line-146"></a> <span class='hs-layout'>;</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>lookupVarEnv</span> <span class='hs-varid'>static_binds</span> <span class='hs-varid'>id</span> <span class='hs-keyword'>of</span> <span class='hs-layout'>{</span>
<a name="line-147"></a> <span class='hs-conid'>Just</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>return</span> <span class='hs-varid'>info</span> <span class='hs-layout'>;</span>
<a name="line-148"></a> <span class='hs-conid'>Nothing</span> <span class='hs-keyglyph'>-></span>
<a name="line-149"></a>
<a name="line-150"></a> <span class='hs-comment'>-- Should be imported; make up a CgIdInfo for it</span>
<a name="line-151"></a> <span class='hs-keyword'>let</span>
<a name="line-152"></a> <span class='hs-varid'>name</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>idName</span> <span class='hs-varid'>id</span>
<a name="line-153"></a> <span class='hs-keyword'>in</span>
<a name="line-154"></a> <span class='hs-keyword'>if</span> <span class='hs-varid'>isExternalName</span> <span class='hs-varid'>name</span> <span class='hs-keyword'>then</span> <span class='hs-keyword'>do</span>
<a name="line-155"></a> <span class='hs-keyword'>let</span> <span class='hs-varid'>ext_lbl</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>CmmLabel</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkClosureLabel</span> <span class='hs-varid'>name</span> <span class='hs-varop'>$</span> <span class='hs-varid'>idCafInfo</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span>
<a name="line-156"></a> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>litIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkLFImported</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span> <span class='hs-varid'>ext_lbl</span><span class='hs-layout'>)</span>
<a name="line-157"></a> <span class='hs-keyword'>else</span>
<a name="line-158"></a> <span class='hs-comment'>-- Bug </span>
<a name="line-159"></a> <span class='hs-varid'>cgLookupPanic</span> <span class='hs-varid'>id</span>
<a name="line-160"></a> <span class='hs-layout'>}</span><span class='hs-layout'>}</span><span class='hs-layout'>}</span><span class='hs-layout'>}</span>
<a name="line-161"></a>
<a name="line-162"></a><a name="cgLookupPanic"></a><span class='hs-definition'>cgLookupPanic</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-varid'>a</span>
<a name="line-163"></a><span class='hs-definition'>cgLookupPanic</span> <span class='hs-varid'>id</span>
<a name="line-164"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <span class='hs-varid'>static_binds</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getStaticBinds</span>
<a name="line-165"></a> <span class='hs-varid'>local_binds</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getBinds</span>
<a name="line-166"></a> <span class='hs-varid'>srt</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getSRTLabel</span>
<a name="line-167"></a> <span class='hs-varid'>pprPanic</span> <span class='hs-str'>"StgCmmEnv: variable not found"</span>
<a name="line-168"></a> <span class='hs-layout'>(</span><span class='hs-varid'>vcat</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>ppr</span> <span class='hs-varid'>id</span><span class='hs-layout'>,</span>
<a name="line-169"></a> <span class='hs-varid'>ptext</span> <span class='hs-layout'>(</span><span class='hs-varid'>sLit</span> <span class='hs-str'>"static binds for:"</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span>
<a name="line-170"></a> <span class='hs-varid'>vcat</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>ppr</span> <span class='hs-layout'>(</span><span class='hs-varid'>cg_id</span> <span class='hs-varid'>info</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>varEnvElts</span> <span class='hs-varid'>static_binds</span> <span class='hs-keyglyph'>]</span><span class='hs-layout'>,</span>
<a name="line-171"></a> <span class='hs-varid'>ptext</span> <span class='hs-layout'>(</span><span class='hs-varid'>sLit</span> <span class='hs-str'>"local binds for:"</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span>
<a name="line-172"></a> <span class='hs-varid'>vcat</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>ppr</span> <span class='hs-layout'>(</span><span class='hs-varid'>cg_id</span> <span class='hs-varid'>info</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>varEnvElts</span> <span class='hs-varid'>local_binds</span> <span class='hs-keyglyph'>]</span><span class='hs-layout'>,</span>
<a name="line-173"></a> <span class='hs-varid'>ptext</span> <span class='hs-layout'>(</span><span class='hs-varid'>sLit</span> <span class='hs-str'>"SRT label"</span><span class='hs-layout'>)</span> <span class='hs-varop'><+></span> <span class='hs-varid'>pprCLabel</span> <span class='hs-varid'>srt</span>
<a name="line-174"></a> <span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-175"></a>
<a name="line-176"></a>
<a name="line-177"></a><a name="getArgAmode"></a><span class='hs-comment'>--------------------</span>
<a name="line-178"></a><span class='hs-definition'>getArgAmode</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>NonVoid</span> <span class='hs-conid'>StgArg</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-conid'>CmmExpr</span>
<a name="line-179"></a><span class='hs-definition'>getArgAmode</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-layout'>(</span><span class='hs-conid'>StgVarArg</span> <span class='hs-varid'>var</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span>
<a name="line-180"></a> <span class='hs-keyword'>do</span> <span class='hs-layout'>{</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getCgIdInfo</span> <span class='hs-varid'>var</span><span class='hs-layout'>;</span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span><span class='hs-varid'>idInfoToAmode</span> <span class='hs-varid'>info</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-181"></a><span class='hs-definition'>getArgAmode</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-layout'>(</span><span class='hs-conid'>StgLitArg</span> <span class='hs-varid'>lit</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>liftM</span> <span class='hs-conid'>CmmLit</span> <span class='hs-varop'>$</span> <span class='hs-varid'>cgLit</span> <span class='hs-varid'>lit</span>
<a name="line-182"></a><span class='hs-definition'>getArgAmode</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-layout'>(</span><span class='hs-conid'>StgTypeArg</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>panic</span> <span class='hs-str'>"getArgAmode: type arg"</span>
<a name="line-183"></a>
<a name="line-184"></a><a name="getNonVoidArgAmodes"></a><span class='hs-definition'>getNonVoidArgAmodes</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>StgArg</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>CmmExpr</span><span class='hs-keyglyph'>]</span>
<a name="line-185"></a><span class='hs-comment'>-- NB: Filters out void args, </span>
<a name="line-186"></a><span class='hs-comment'>-- so the result list may be shorter than the argument list</span>
<a name="line-187"></a><span class='hs-definition'>getNonVoidArgAmodes</span> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>return</span> <span class='hs-conid'>[]</span>
<a name="line-188"></a><span class='hs-definition'>getNonVoidArgAmodes</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg</span><span class='hs-conop'>:</span><span class='hs-varid'>args</span><span class='hs-layout'>)</span>
<a name="line-189"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>isVoidRep</span> <span class='hs-layout'>(</span><span class='hs-varid'>argPrimRep</span> <span class='hs-varid'>arg</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>getNonVoidArgAmodes</span> <span class='hs-varid'>args</span>
<a name="line-190"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <span class='hs-layout'>{</span> <span class='hs-varid'>amode</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getArgAmode</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>arg</span><span class='hs-layout'>)</span>
<a name="line-191"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>amodes</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getNonVoidArgAmodes</span> <span class='hs-varid'>args</span>
<a name="line-192"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>return</span> <span class='hs-layout'>(</span> <span class='hs-varid'>amode</span> <span class='hs-conop'>:</span> <span class='hs-varid'>amodes</span> <span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-193"></a>
<a name="line-194"></a>
<a name="line-195"></a><span class='hs-comment'>------------------------------------------------------------------------</span>
<a name="line-196"></a><span class='hs-comment'>-- Interface functions for binding and re-binding names</span>
<a name="line-197"></a><span class='hs-comment'>------------------------------------------------------------------------</span>
<a name="line-198"></a>
<a name="line-199"></a><a name="bindToReg"></a><span class='hs-definition'>bindToReg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>NonVoid</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>LambdaFormInfo</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-conid'>LocalReg</span>
<a name="line-200"></a><span class='hs-comment'>-- Bind an Id to a fresh LocalReg</span>
<a name="line-201"></a><span class='hs-definition'>bindToReg</span> <span class='hs-varid'>nvid</span><span class='hs-keyglyph'>@</span><span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span> <span class='hs-varid'>lf_info</span>
<a name="line-202"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <span class='hs-layout'>{</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>reg</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>idToReg</span> <span class='hs-varid'>nvid</span>
<a name="line-203"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>addBindC</span> <span class='hs-varid'>id</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkCgIdInfo</span> <span class='hs-varid'>id</span> <span class='hs-varid'>lf_info</span> <span class='hs-layout'>(</span><span class='hs-conid'>CmmReg</span> <span class='hs-layout'>(</span><span class='hs-conid'>CmmLocal</span> <span class='hs-varid'>reg</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-204"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>return</span> <span class='hs-varid'>reg</span> <span class='hs-layout'>}</span>
<a name="line-205"></a>
<a name="line-206"></a><a name="rebindToReg"></a><span class='hs-definition'>rebindToReg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>NonVoid</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-conid'>LocalReg</span>
<a name="line-207"></a><span class='hs-comment'>-- Like bindToReg, but the Id is already in scope, so </span>
<a name="line-208"></a><span class='hs-comment'>-- get its LF info from the envt</span>
<a name="line-209"></a><span class='hs-definition'>rebindToReg</span> <span class='hs-varid'>nvid</span><span class='hs-keyglyph'>@</span><span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span>
<a name="line-210"></a> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span> <span class='hs-layout'>{</span> <span class='hs-varid'>info</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>getCgIdInfo</span> <span class='hs-varid'>id</span>
<a name="line-211"></a> <span class='hs-layout'>;</span> <span class='hs-varid'>bindToReg</span> <span class='hs-varid'>nvid</span> <span class='hs-layout'>(</span><span class='hs-varid'>cgIdInfoLF</span> <span class='hs-varid'>info</span><span class='hs-layout'>)</span> <span class='hs-layout'>}</span>
<a name="line-212"></a>
<a name="line-213"></a><a name="bindArgToReg"></a><span class='hs-definition'>bindArgToReg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>NonVoid</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-conid'>LocalReg</span>
<a name="line-214"></a><span class='hs-definition'>bindArgToReg</span> <span class='hs-varid'>nvid</span><span class='hs-keyglyph'>@</span><span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>bindToReg</span> <span class='hs-varid'>nvid</span> <span class='hs-layout'>(</span><span class='hs-varid'>mkLFArgument</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span>
<a name="line-215"></a>
<a name="line-216"></a><a name="bindArgsToRegs"></a><span class='hs-definition'>bindArgsToRegs</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>NonVoid</span> <span class='hs-conid'>Id</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>FCode</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>LocalReg</span><span class='hs-keyglyph'>]</span>
<a name="line-217"></a><span class='hs-definition'>bindArgsToRegs</span> <span class='hs-varid'>args</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mapM</span> <span class='hs-varid'>bindArgToReg</span> <span class='hs-varid'>args</span>
<a name="line-218"></a>
<a name="line-219"></a><a name="idToReg"></a><span class='hs-definition'>idToReg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>NonVoid</span> <span class='hs-conid'>Id</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>LocalReg</span>
<a name="line-220"></a><span class='hs-comment'>-- Make a register from an Id, typically a function argument,</span>
<a name="line-221"></a><span class='hs-comment'>-- free variable, or case binder</span>
<a name="line-222"></a><span class='hs-comment'>--</span>
<a name="line-223"></a><span class='hs-comment'>-- We re-use the Unique from the Id to make it easier to see what is going on</span>
<a name="line-224"></a><span class='hs-comment'>--</span>
<a name="line-225"></a><span class='hs-comment'>-- By now the Ids should be uniquely named; else one would worry</span>
<a name="line-226"></a><span class='hs-comment'>-- about accidental collision </span>
<a name="line-227"></a><span class='hs-definition'>idToReg</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonVoid</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>LocalReg</span> <span class='hs-layout'>(</span><span class='hs-varid'>idUnique</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span>
<a name="line-228"></a> <span class='hs-layout'>(</span><span class='hs-keyword'>case</span> <span class='hs-varid'>idPrimRep</span> <span class='hs-varid'>id</span> <span class='hs-keyword'>of</span> <span class='hs-conid'>VoidRep</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>pprPanic</span> <span class='hs-str'>"idToReg"</span> <span class='hs-layout'>(</span><span class='hs-varid'>ppr</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span>
<a name="line-229"></a> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>primRepCmmType</span> <span class='hs-layout'>(</span><span class='hs-varid'>idPrimRep</span> <span class='hs-varid'>id</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-230"></a>
<a name="line-231"></a>
</pre></body>
</html>
