diff options
Diffstat (limited to 'giscanner/transformer.py')
-rw-r--r-- | giscanner/transformer.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/giscanner/transformer.py b/giscanner/transformer.py index 8e335ff6..b7e8c5df 100644 --- a/giscanner/transformer.py +++ b/giscanner/transformer.py @@ -20,7 +20,7 @@ import os -from .ast import (Callback, Enum, Function, Namespace, Member, +from .ast import (Bitfield, Callback, Enum, Function, Namespace, Member, Parameter, Return, Struct, Field, Type, Array, Alias, Interface, Class, Node, Union, Varargs, Constant, type_name_from_ctype, @@ -210,9 +210,13 @@ class Transformer(object): child.ident)) enum_name = self.remove_prefix(symbol.ident) - enum = Enum(enum_name, symbol.ident, members) - self._names.type_names[symbol.ident] = (None, enum) - return enum + if symbol.base_type.is_bitfield: + klass = Bitfield + else: + klass = Enum + node = klass(enum_name, symbol.ident, members) + self._names.type_names[symbol.ident] = (None, node) + return node def _create_object(self, symbol): return Member(symbol.ident, symbol.base_type.name, |