IDL: traverse enums and implements

third_party/WebKit/Source/bindings/scripts/idl_definitions.py

 # Copyright (C) 2013 Google Inc. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions are
 # met:
 #
 #     * Redistributions of source code must retain the above copyright
 # notice, this list of conditions and the following disclaimer.
 #     * Redistributions in binary form must reproduce the above
 # copyright notice, this list of conditions and the following disclaimer
@@ skipping 119 matching lines @@
                 raise ValueError('Unrecognized node class: %s' % child_class)
 
     def accept(self, visitor):
         visitor.visit_definitions(self)
         for interface in self.interfaces.itervalues():
             interface.accept(visitor)
         for callback_function in self.callback_functions.itervalues():
             callback_function.accept(visitor)
         for dictionary in self.dictionaries.itervalues():
             dictionary.accept(visitor)
+        for enumeration in self.enumerations.itervalues():
+            enumeration.accept(visitor)
+        for implement in self.implements:
+            implement.accept(visitor)
         for typedef in self.typedefs.itervalues():
             typedef.accept(visitor)
 
     def update(self, other):
         """Update with additional IdlDefinitions."""
         for interface_name, new_interface in other.interfaces.iteritems():
             if not new_interface.is_partial:
                 # Add as new interface
                 self.interfaces[interface_name] = new_interface
                 continue
@@ skipping 97 matching lines @@
 
 class IdlEnum(object):
     # FIXME: remove, just treat enums as a dictionary
     def __init__(self, idl_name, node):
         self.idl_name = idl_name
         self.name = node.GetName()
         self.values = []
         for child in node.GetChildren():
             self.values.append(child.GetName())
 
+    def accept(self, visitor):
+        visitor.visit_enumeration(self)
+
 
 ################################################################################
 # Typedefs
 ################################################################################
 
 class IdlTypedef(object):
     idl_type_attributes = ('idl_type',)
 
     def __init__(self, node):
         self.name = node.GetName()
@@ skipping 563 matching lines @@
 
 ################################################################################
 # Implement statements
 ################################################################################
 
 class IdlImplement(object):
     def __init__(self, node):
         self.left_interface = node.GetName()
         self.right_interface = node.GetProperty('REFERENCE')
 
+    def accept(self, visitor):
+        visitor.visit_implement(self)
+
 
 ################################################################################
 # Extended attributes
 ################################################################################
 
 class Exposure:
     """An Exposure holds one Exposed or RuntimeEnabled condition.
     Each exposure has two properties: exposed and runtime_enabled.
     Exposure(e, r) corresponds to [Exposed(e r)]. Exposure(e) corresponds to
     [Exposed=e].
@@ skipping 235 matching lines @@
 
     def visit_callback_function(self, callback_function):
         self.visit_typed_object(callback_function)
 
     def visit_dictionary(self, dictionary):
         pass
 
     def visit_dictionary_member(self, member):
         self.visit_typed_object(member)
 
+    def visit_enumeration(self, enumeration):
+        pass
+
+    def visit_implement(self, implement):
+        pass
+
     def visit_interface(self, interface):
         pass
 
     def visit_typedef(self, typedef):
         self.visit_typed_object(typedef)
 
     def visit_attribute(self, attribute):
         self.visit_typed_object(attribute)
 
     def visit_constant(self, constant):
         self.visit_typed_object(constant)
 
     def visit_operation(self, operation):
         self.visit_typed_object(operation)
 
     def visit_argument(self, argument):
         self.visit_typed_object(argument)
 
     def visit_iterable(self, iterable):
         self.visit_typed_object(iterable)
 
     def visit_maplike(self, maplike):
         self.visit_typed_object(maplike)
 
     def visit_setlike(self, setlike):
         self.visit_typed_object(setlike)