110967Sdavid.hashe@amd.com# Copyright (c) 2013 Advanced Micro Devices, Inc.
26657Snate@binkert.org# Copyright (c) 1999-2008 Mark D. Hill and David A. Wood
36657Snate@binkert.org# Copyright (c) 2009 The Hewlett-Packard Development Company
46657Snate@binkert.org# All rights reserved.
56657Snate@binkert.org#
66657Snate@binkert.org# Redistribution and use in source and binary forms, with or without
76657Snate@binkert.org# modification, are permitted provided that the following conditions are
86657Snate@binkert.org# met: redistributions of source code must retain the above copyright
96657Snate@binkert.org# notice, this list of conditions and the following disclaimer;
106657Snate@binkert.org# redistributions in binary form must reproduce the above copyright
116657Snate@binkert.org# notice, this list of conditions and the following disclaimer in the
126657Snate@binkert.org# documentation and/or other materials provided with the distribution;
136657Snate@binkert.org# neither the name of the copyright holders nor the names of its
146657Snate@binkert.org# contributors may be used to endorse or promote products derived from
156657Snate@binkert.org# this software without specific prior written permission.
166657Snate@binkert.org#
176657Snate@binkert.org# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
186657Snate@binkert.org# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
196657Snate@binkert.org# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
206657Snate@binkert.org# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
216657Snate@binkert.org# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
226657Snate@binkert.org# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
236657Snate@binkert.org# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
246657Snate@binkert.org# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
256657Snate@binkert.org# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
266657Snate@binkert.org# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
276657Snate@binkert.org# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
286657Snate@binkert.org
296657Snate@binkert.orgfrom slicc.ast.StatementAST import StatementAST
306657Snate@binkert.orgfrom slicc.symbols import Var
316657Snate@binkert.org
326657Snate@binkert.orgclass PeekStatementAST(StatementAST):
336863Sdrh5@cs.wisc.edu    def __init__(self, slicc, queue_name, type_ast, pairs, statements, method):
346863Sdrh5@cs.wisc.edu        super(PeekStatementAST, self).__init__(slicc, pairs)
356657Snate@binkert.org
366657Snate@binkert.org        self.queue_name = queue_name
376657Snate@binkert.org        self.type_ast = type_ast
386657Snate@binkert.org        self.statements = statements
396657Snate@binkert.org        self.method = method
406657Snate@binkert.org
416657Snate@binkert.org    def __repr__(self):
426657Snate@binkert.org        return "[PeekStatementAST: %r queue_name: %r type: %r %r]" % \
436657Snate@binkert.org               (self.method, self.queue_name, self.type_ast, self.statements)
446657Snate@binkert.org
456657Snate@binkert.org    def generate(self, code, return_type):
466657Snate@binkert.org        self.symtab.pushFrame()
476657Snate@binkert.org
486657Snate@binkert.org        msg_type = self.type_ast.type
496657Snate@binkert.org
506657Snate@binkert.org        # Add new local var to symbol table
516657Snate@binkert.org        var = Var(self.symtab, "in_msg", self.location, msg_type, "(*in_msg_ptr)",
526657Snate@binkert.org                  self.pairs)
536657Snate@binkert.org        self.symtab.newSymbol(var)
546657Snate@binkert.org
556657Snate@binkert.org        # Check the queue type
566657Snate@binkert.org        self.queue_name.assertType("InPort")
576657Snate@binkert.org
586657Snate@binkert.org        # Declare the new "in_msg_ptr" variable
5910967Sdavid.hashe@amd.com        mtid = msg_type.c_ident
606657Snate@binkert.org        qcode = self.queue_name.var.code
616657Snate@binkert.org        code('''
626657Snate@binkert.org{
637007Snate@binkert.org    // Declare message
648641Snate@binkert.org    const $mtid* in_msg_ptr M5_VAR_USED;
656657Snate@binkert.org    in_msg_ptr = dynamic_cast<const $mtid *>(($qcode).${{self.method}}());
6610963Sdavid.hashe@amd.com    if (in_msg_ptr == NULL) {
6710963Sdavid.hashe@amd.com        // If the cast fails, this is the wrong inport (wrong message type).
6810963Sdavid.hashe@amd.com        // Throw an exception, and the caller will decide to either try a
6910963Sdavid.hashe@amd.com        // different inport or punt.
7010963Sdavid.hashe@amd.com        throw RejectException();
7110963Sdavid.hashe@amd.com    }
726657Snate@binkert.org''')
737007Snate@binkert.org
7413675Sandreas.sandberg@arm.com        if "block_on" in self.pairs:
756863Sdrh5@cs.wisc.edu            address_field = self.pairs['block_on']
766863Sdrh5@cs.wisc.edu            code('''
7710231Ssteve.reinhardt@amd.com    if (m_is_blocking &&
7810231Ssteve.reinhardt@amd.com        (m_block_map.count(in_msg_ptr->m_$address_field) == 1) &&
7910231Ssteve.reinhardt@amd.com        (m_block_map[in_msg_ptr->m_$address_field] != &$qcode)) {
8011111Snilay@cs.wisc.edu            $qcode.delayHead(clockEdge(), cyclesToTicks(Cycles(1)));
816863Sdrh5@cs.wisc.edu            continue;
826863Sdrh5@cs.wisc.edu    }
836863Sdrh5@cs.wisc.edu            ''')
846657Snate@binkert.org
8513675Sandreas.sandberg@arm.com        if "wake_up" in self.pairs:
867567SBrad.Beckmann@amd.com            address_field = self.pairs['wake_up']
877567SBrad.Beckmann@amd.com            code('''
887567SBrad.Beckmann@amd.com    if (m_waiting_buffers.count(in_msg_ptr->m_$address_field) > 0) {
897567SBrad.Beckmann@amd.com        wakeUpBuffers(in_msg_ptr->m_$address_field);
907567SBrad.Beckmann@amd.com    }
917567SBrad.Beckmann@amd.com            ''')
927567SBrad.Beckmann@amd.com
936657Snate@binkert.org        # The other statements
946657Snate@binkert.org        self.statements.generate(code, return_type)
956657Snate@binkert.org        self.symtab.popFrame()
966657Snate@binkert.org        code("}")
976657Snate@binkert.org
986657Snate@binkert.org    def findResources(self, resources):
996657Snate@binkert.org        self.statements.findResources(resources)
100