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.TypeFieldAST import TypeFieldAST
29from slicc.symbols import Event, State, RequestType
30
31class TypeFieldEnumAST(TypeFieldAST):
32    def __init__(self, slicc, field_id, pairs_ast):
33        super(TypeFieldEnumAST, self).__init__(slicc, pairs_ast)
34
35        self.field_id = field_id
36        self.pairs_ast = pairs_ast
37
38    def __repr__(self):
39        return "[TypeFieldEnum: %r]" % self.field_id
40
41    def generate(self, type):
42        if str(type) == "State":
43            self.error("States must in a State Declaration, not a normal enum.")
44
45        # Add enumeration
46        if not type.addEnum(self.field_id, self.pairs_ast.pairs):
47            self.error("Duplicate enumeration: %s:%s" % (type, self.field_id))
48
49        # Fill machine info
50        machine = self.symtab.state_machine
51
52        if str(type) == "Event":
53            if not machine:
54                self.error("Event declaration not part of a machine.")
55            e = Event(self.symtab, self.field_id, self.location, self.pairs)
56            machine.addEvent(e)
57
58        if str(type) == "RequestType":
59            if not machine:
60                self.error("RequestType declaration not part of a machine.")
61            s = RequestType(self.symtab, self.field_id, self.location,
62                           self.pairs)
63            machine.addRequestType(s)
64