Func.py revision 7839
16657Snate@binkert.org# Copyright (c) 1999-2008 Mark D. Hill and David A. Wood
26657Snate@binkert.org# Copyright (c) 2009 The Hewlett-Packard Development Company
36657Snate@binkert.org# All rights reserved.
46657Snate@binkert.org#
56657Snate@binkert.org# Redistribution and use in source and binary forms, with or without
66657Snate@binkert.org# modification, are permitted provided that the following conditions are
76657Snate@binkert.org# met: redistributions of source code must retain the above copyright
86657Snate@binkert.org# notice, this list of conditions and the following disclaimer;
96657Snate@binkert.org# redistributions in binary form must reproduce the above copyright
106657Snate@binkert.org# notice, this list of conditions and the following disclaimer in the
116657Snate@binkert.org# documentation and/or other materials provided with the distribution;
126657Snate@binkert.org# neither the name of the copyright holders nor the names of its
136657Snate@binkert.org# contributors may be used to endorse or promote products derived from
146657Snate@binkert.org# this software without specific prior written permission.
156657Snate@binkert.org#
166657Snate@binkert.org# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
176657Snate@binkert.org# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
186657Snate@binkert.org# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
196657Snate@binkert.org# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
206657Snate@binkert.org# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
216657Snate@binkert.org# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
226657Snate@binkert.org# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
236657Snate@binkert.org# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
246657Snate@binkert.org# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
256657Snate@binkert.org# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
266657Snate@binkert.org# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
276657Snate@binkert.org
286657Snate@binkert.orgfrom slicc.symbols.Symbol import Symbol
296657Snate@binkert.orgfrom slicc.symbols.Type import Type
306657Snate@binkert.org
316657Snate@binkert.orgclass Func(Symbol):
326657Snate@binkert.org    def __init__(self, table, ident, location, return_type, param_types,
336657Snate@binkert.org                 param_strings, body, pairs, machine):
346657Snate@binkert.org        super(Func, self).__init__(table, ident, location, pairs)
356657Snate@binkert.org        self.return_type = return_type
366657Snate@binkert.org        self.param_types = param_types
376657Snate@binkert.org        self.param_strings = param_strings
386657Snate@binkert.org        self.body = body
396657Snate@binkert.org        self.isInternalMachineFunc = False
406657Snate@binkert.org
416657Snate@binkert.org        if machine is None:
426657Snate@binkert.org            self.c_ident = ident
436657Snate@binkert.org        elif "external" in self or "primitive" in self:
446657Snate@binkert.org            self.c_ident = ident
456657Snate@binkert.org        else:
466657Snate@binkert.org            self.machineStr = str(machine)
476657Snate@binkert.org            # Append with machine name
486657Snate@binkert.org            self.c_ident = "%s_%s" % (self.machineStr, ident)
496657Snate@binkert.org            self.isInternalMachineFunc = True
506657Snate@binkert.org
516657Snate@binkert.org    def __repr__(self):
526657Snate@binkert.org        return ""
536657Snate@binkert.org
546657Snate@binkert.org    @property
556657Snate@binkert.org    def prototype(self):
566657Snate@binkert.org        if "external" in self:
576657Snate@binkert.org            return ""
586657Snate@binkert.org
596657Snate@binkert.org        return_type = self.return_type.c_ident
606657Snate@binkert.org        void_type = self.symtab.find("void", Type)
616657Snate@binkert.org        if "return_by_ref" in self and self.return_type != void_type:
626657Snate@binkert.org            return_type += "&"
637839Snilay@cs.wisc.edu        elif "return_by_pointer" in self and self.return_type != void_type:
647839Snilay@cs.wisc.edu            return_type += "*"
656657Snate@binkert.org
666657Snate@binkert.org        return "%s %s(%s);" % (return_type, self.c_ident,
676657Snate@binkert.org                               ", ".join(self.param_strings))
686657Snate@binkert.org
696657Snate@binkert.org    def writeCodeFiles(self, path):
706657Snate@binkert.org        '''This write a function of object Chip'''
716657Snate@binkert.org        if "external" in self:
726657Snate@binkert.org            return
736657Snate@binkert.org
746999Snate@binkert.org        code = self.symtab.codeFormatter()
756657Snate@binkert.org
766657Snate@binkert.org        # Header
776657Snate@binkert.org        code('''
786657Snate@binkert.org/** Auto generated C++ code started by $__file__:$__line__ */
796657Snate@binkert.org
806657Snate@binkert.org#include "mem/protocol/Types.hh"
816657Snate@binkert.org''')
826657Snate@binkert.org
836657Snate@binkert.org        if self.isInternalMachineFunc:
846657Snate@binkert.org            code('#include "mem/protocol/${{self.machineStr}}_Controller.hh"')
856657Snate@binkert.org
867055Snate@binkert.org        code('using namespace std;')
876657Snate@binkert.org        # Generate function header
886657Snate@binkert.org        void_type = self.symtab.find("void", Type)
896657Snate@binkert.org        return_type = self.return_type.c_ident
906657Snate@binkert.org        if "return_by_ref" in self and self.return_type != void_type:
916657Snate@binkert.org            return_type += "&"
927839Snilay@cs.wisc.edu        if "return_by_pointer" in self and self.return_type != void_type:
937839Snilay@cs.wisc.edu            return_type += "*"
946657Snate@binkert.org
956657Snate@binkert.org        if self.isInternalMachineFunc:
966657Snate@binkert.org            klass = "%s_Controller" % self.machineStr
976657Snate@binkert.org        else:
986657Snate@binkert.org            klass = "Chip"
996657Snate@binkert.org
1006657Snate@binkert.org        params = ', '.join(self.param_strings)
1016657Snate@binkert.org
1026657Snate@binkert.org        code('''
1037007Snate@binkert.org$return_type
1047007Snate@binkert.org${klass}::${{self.c_ident}}($params)
1056657Snate@binkert.org{
1066657Snate@binkert.org${{self.body}}
1076657Snate@binkert.org}
1086657Snate@binkert.org''')
1096657Snate@binkert.org        code.write(path, "%s.cc" % self.c_ident)
1106657Snate@binkert.org
1116657Snate@binkert.org__all__ = [ "Func" ]
112