Author: grosser Date: Sun Feb 5 05:42:25 2012 New Revision: 149830 URL: http://llvm.org/viewvc/llvm-project?rev=149830&view=rev Log: [clang.py] Implement Cursor.enum_type
Contributed by: Gregory Szorc <[email protected]> Modified: cfe/trunk/bindings/python/clang/cindex.py cfe/trunk/bindings/python/tests/cindex/test_cursor.py Modified: cfe/trunk/bindings/python/clang/cindex.py URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/bindings/python/clang/cindex.py?rev=149830&r1=149829&r2=149830&view=diff ============================================================================== --- cfe/trunk/bindings/python/clang/cindex.py (original) +++ cfe/trunk/bindings/python/clang/cindex.py Sun Feb 5 05:42:25 2012 @@ -995,6 +995,19 @@ return self._underlying_type @property + def enum_type(self): + """Return the integer type of an enum declaration. + + Returns a type corresponding to an integer. If the cursor is not for an + enum, this raises. + """ + if not hasattr(self, '_enum_type'): + assert self.kind == CursorKind.ENUM_DECL + self._enum_type = Cursor_enum_type(self) + + return self._enum_type + + @property def hash(self): """Returns a hash of the cursor as an int.""" if not hasattr(self, '_hash'): @@ -1818,6 +1831,11 @@ Cursor_underlying_type.restype = Type Cursor_underlying_type.errcheck = Type.from_result +Cursor_enum_type = lib.clang_getEnumDeclIntegerType +Cursor_enum_type.argtypes = [Cursor] +Cursor_enum_type.restype = Type +Cursor_enum_type.errcheck = Type.from_result + Cursor_visit_callback = CFUNCTYPE(c_int, Cursor, Cursor, py_object) Cursor_visit = lib.clang_visitChildren Cursor_visit.argtypes = [Cursor, Cursor_visit_callback, py_object] Modified: cfe/trunk/bindings/python/tests/cindex/test_cursor.py URL: http://llvm.org/viewvc/llvm-project/cfe/trunk/bindings/python/tests/cindex/test_cursor.py?rev=149830&r1=149829&r2=149830&view=diff ============================================================================== --- cfe/trunk/bindings/python/tests/cindex/test_cursor.py (original) +++ cfe/trunk/bindings/python/tests/cindex/test_cursor.py Sun Feb 5 05:42:25 2012 @@ -77,3 +77,18 @@ assert typedef.kind.is_declaration() underlying = typedef.underlying_typedef_type assert underlying.kind == TypeKind.INT + +def test_enum_type(): + source = 'enum TEST { FOO=1, BAR=2 };' + index = Index.create() + tu = index.parse('test.c', unsaved_files=[('test.c', source)]) + assert tu is not None + + for cursor in tu.cursor.get_children(): + if cursor.spelling == 'TEST': + enum = cursor + break + + assert enum.kind == CursorKind.ENUM_DECL + enum_type = enum.enum_type + assert enum_type.kind == TypeKind.UINT _______________________________________________ cfe-commits mailing list [email protected] http://lists.cs.uiuc.edu/mailman/listinfo/cfe-commits
