[cfe-commits] r149830 - in /cfe/trunk/bindings/python: clang/cindex.py tests/cindex/test_cursor.py

[Tobias Grosser]

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 <gregory.szorc at gmail.com>

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