EnqueueStatementAST.py revision 9508:dde110931867
1# Copyright (c) 1999-2008 Mark D. Hill and David A. Wood 2# Copyright (c) 2009 The Hewlett-Packard Development Company 3# All rights reserved. 4# 5# Redistribution and use in source and binary forms, with or without 6# modification, are permitted provided that the following conditions are 7# met: redistributions of source code must retain the above copyright 8# notice, this list of conditions and the following disclaimer; 9# redistributions in binary form must reproduce the above copyright 10# notice, this list of conditions and the following disclaimer in the 11# documentation and/or other materials provided with the distribution; 12# neither the name of the copyright holders nor the names of its 13# contributors may be used to endorse or promote products derived from 14# this software without specific prior written permission. 15# 16# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 17# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 18# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 19# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 20# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 21# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 22# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 26# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 28from slicc.ast.StatementAST import StatementAST 29from slicc.symbols import Var 30 31class EnqueueStatementAST(StatementAST): 32 def __init__(self, slicc, queue_name, type_ast, pairs, statements): 33 super(EnqueueStatementAST, self).__init__(slicc, pairs) 34 35 self.queue_name = queue_name 36 self.type_ast = type_ast 37 self.statements = statements 38 39 def __repr__(self): 40 return "[EnqueueStatementAst: %s %s %s]" % \ 41 (self.queue_name, self.type_ast.ident, self.statements) 42 43 def generate(self, code, return_type): 44 code("{") 45 code.indent() 46 self.symtab.pushFrame() 47 48 msg_type = self.type_ast.type 49 50 # Add new local var to symbol table 51 v = Var(self.symtab, "out_msg", self.location, msg_type, "*out_msg", 52 self.pairs) 53 self.symtab.newSymbol(v) 54 55 # Declare message 56 code("${{msg_type.ident}} *out_msg = "\ 57 "new ${{msg_type.ident}}(clockEdge());") 58 59 # The other statements 60 t = self.statements.generate(code, None) 61 62 self.queue_name.assertType("OutPort") 63 64 args = [ "out_msg" ] 65 if "latency" in self: 66 latency = self["latency"] 67 try: 68 # see if this is an integer 69 latency = int(latency) 70 args.append("Cycles(%s)" % latency) 71 except ValueError: 72 # if not, it should be a member 73 args.append("m_%s" % latency) 74 75 args = ", ".join(args) 76 code('(${{self.queue_name.var.code}}).enqueue($args);') 77 78 79 # End scope 80 self.symtab.popFrame() 81 code.dedent() 82 code("}") 83 84 def findResources(self, resources): 85 var = self.queue_name.var 86 res_count = int(resources.get(var, 0)) 87 resources[var] = str(res_count + 1) 88