1f51a89d39
These were ignored by git accidentally. We want ALL OF THEM since they all came in the llvm/clang source distribution.
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
# This file provides common utility functions for the test suite.
|
|
|
|
from clang.cindex import Cursor
|
|
from clang.cindex import Index
|
|
|
|
def get_tu(source, lang='c', all_warnings=False):
|
|
"""Obtain a translation unit from source and language.
|
|
|
|
By default, the translation unit is created from source file "t.<ext>"
|
|
where <ext> is the default file extension for the specified language. By
|
|
default it is C, so "t.c" is the default file name.
|
|
|
|
Supported languages are {c, cpp, objc}.
|
|
|
|
all_warnings is a convenience argument to enable all compiler warnings.
|
|
"""
|
|
name = 't.c'
|
|
if lang == 'cpp':
|
|
name = 't.cpp'
|
|
elif lang == 'objc':
|
|
name = 't.m'
|
|
elif lang != 'c':
|
|
raise Exception('Unknown language: %s' % lang)
|
|
|
|
args = []
|
|
if all_warnings:
|
|
args = ['-Wall', '-Wextra']
|
|
|
|
index = Index.create()
|
|
tu = index.parse(name, args=args, unsaved_files=[(name, source)])
|
|
assert tu is not None
|
|
return tu
|
|
|
|
def get_cursor(source, spelling):
|
|
"""Obtain a cursor from a source object.
|
|
|
|
This provides a convenient search mechanism to find a cursor with specific
|
|
spelling within a source. The first argument can be either a
|
|
TranslationUnit or Cursor instance.
|
|
|
|
If the cursor is not found, None is returned.
|
|
"""
|
|
children = []
|
|
if isinstance(source, Cursor):
|
|
children = source.get_children()
|
|
else:
|
|
# Assume TU
|
|
children = source.cursor.get_children()
|
|
|
|
for cursor in children:
|
|
if cursor.spelling == spelling:
|
|
return cursor
|
|
|
|
# Recurse into children.
|
|
result = get_cursor(cursor, spelling)
|
|
if result is not None:
|
|
return result
|
|
|
|
return None
|
|
|
|
|
|
__all__ = [
|
|
'get_cursor',
|
|
'get_tu',
|
|
]
|