Metal AST

metal.py

import sys
import os
import clang.cindex

shaders_file = sys.argv[1]
assert(shaders_file.endswith(“.metal”))
ast_file = shaders_file.rstrip(“.metal”) + “.ast”

os.system(f”xcrun metal -emit-ast {shaders_file}“)
xcode_path = os.popen(“xcode-select -p”).read().rstrip()

clang.cindex.Config.set_library_file(f”{xcode_path}/Toolchains/XcodeDefault.xctoolchain/usr/metal/macos/lib/metalfe/libmetal.dylib”)

index = clang.cindex.Index.create()
tu = index.read(ast_file)
print(‘Translation unit:’, tu.spelling)
# for i in tu.get_tokens(extent=tu.cursor.extent):
#     print(i.kind)