SPIRV-Tools/utils/generate_vim_syntax.py
Ryan Harrison 65f49dfc39
Remove unneeded future imports (#2739)
Also, adds explicitly setting python executable in the NDK build script, rewrites 
some Python2-isms to 3isms, and formats some code.

Fixes #2738
2019-07-24 15:29:38 -04:00

206 lines
7.1 KiB
Python
Executable File

#!/usr/bin/env python
# Copyright (c) 2016 Google Inc.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Generates Vim syntax rules for SPIR-V assembly (.spvasm) files"""
import json
PREAMBLE="""" Vim syntax file
" Language: spvasm
" Generated by SPIRV-Tools
if version < 600
syntax clear
elseif exists("b:current_syntax")
finish
endif
syn case match
"""
POSTAMBLE="""
syntax keyword spvasmTodo TODO FIXME contained
syn match spvasmIdNumber /%\d\+\>/
" The assembler treats the leading minus sign as part of the number token.
" This applies to integers, and to floats below.
syn match spvasmNumber /-\?\<\d\+\>/
" Floating point literals.
" In general, C++ requires at least digit in the mantissa, and the
" floating point is optional. This applies to both the regular decimal float
" case and the hex float case.
" First case: digits before the optional decimal, no trailing digits.
syn match spvasmFloat /-\?\d\+\.\?\(e[+-]\d\+\)\?/
" Second case: optional digits before decimal, trailing digits
syn match spvasmFloat /-\?\d*\.\d\+\(e[+-]\d\+\)\?/
" First case: hex digits before the optional decimal, no trailing hex digits.
syn match spvasmFloat /-\?0[xX]\\x\+\.\?p[-+]\d\+/
" Second case: optional hex digits before decimal, trailing hex digits
syn match spvasmFloat /-\?0[xX]\\x*\.\\x\+p[-+]\d\+/
syn match spvasmComment /;.*$/ contains=spvasmTodo
syn region spvasmString start=/"/ skip=/\\\\"/ end=/"/
syn match spvasmId /%[a-zA-Z_][a-zA-Z_0-9]*/
" Highlight unknown constants and statements as errors
syn match spvasmError /[a-zA-Z][a-zA-Z_0-9]*/
if version >= 508 || !exists("did_c_syn_inits")
if version < 508
let did_c_syn_inits = 1
command -nargs=+ HiLink hi link <args>
else
command -nargs=+ HiLink hi def link <args>
endif
HiLink spvasmStatement Statement
HiLink spvasmNumber Number
HiLink spvasmComment Comment
HiLink spvasmString String
HiLink spvasmFloat Float
HiLink spvasmConstant Constant
HiLink spvasmIdNumber Identifier
HiLink spvasmId Identifier
HiLink spvasmTodo Todo
delcommand HiLink
endif
let b:current_syntax = "spvasm"
"""
# This list is taken from the description of OpSpecConstantOp in SPIR-V 1.1.
# TODO(dneto): Propose that this information be embedded in the grammar file.
SPEC_CONSTANT_OP_OPCODES = """
OpSConvert, OpFConvert
OpSNegate, OpNot
OpIAdd, OpISub
OpIMul, OpUDiv, OpSDiv, OpUMod, OpSRem, OpSMod
OpShiftRightLogical, OpShiftRightArithmetic, OpShiftLeftLogical
OpBitwiseOr, OpBitwiseXor, OpBitwiseAnd
OpVectorShuffle, OpCompositeExtract, OpCompositeInsert
OpLogicalOr, OpLogicalAnd, OpLogicalNot,
OpLogicalEqual, OpLogicalNotEqual
OpSelect
OpIEqual, OpINotEqual
OpULessThan, OpSLessThan
OpUGreaterThan, OpSGreaterThan
OpULessThanEqual, OpSLessThanEqual
OpUGreaterThanEqual, OpSGreaterThanEqual
OpQuantizeToF16
OpConvertFToS, OpConvertSToF
OpConvertFToU, OpConvertUToF
OpUConvert
OpConvertPtrToU, OpConvertUToPtr
OpGenericCastToPtr, OpPtrCastToGeneric
OpBitcast
OpFNegate
OpFAdd, OpFSub
OpFMul, OpFDiv
OpFRem, OpFMod
OpAccessChain, OpInBoundsAccessChain
OpPtrAccessChain, OpInBoundsPtrAccessChain"""
def EmitAsStatement(name):
"""Emits the given name as a statement token"""
print('syn keyword spvasmStatement', name)
def EmitAsEnumerant(name):
"""Emits the given name as an named operand token"""
print('syn keyword spvasmConstant', name)
def main():
"""Parses arguments, then generates the Vim syntax rules for SPIR-V assembly
on stdout."""
import argparse
parser = argparse.ArgumentParser(description='Generate SPIR-V info tables')
parser.add_argument('--spirv-core-grammar', metavar='<path>',
type=str, required=True,
help='input JSON grammar file for core SPIR-V '
'instructions')
parser.add_argument('--extinst-glsl-grammar', metavar='<path>',
type=str, required=False, default=None,
help='input JSON grammar file for GLSL extended '
'instruction set')
parser.add_argument('--extinst-opencl-grammar', metavar='<path>',
type=str, required=False, default=None,
help='input JSON grammar file for OpenGL extended '
'instruction set')
parser.add_argument('--extinst-debuginfo-grammar', metavar='<path>',
type=str, required=False, default=None,
help='input JSON grammar file for DebugInfo extended '
'instruction set')
args = parser.parse_args()
# Generate the syntax rules.
print(PREAMBLE)
core = json.loads(open(args.spirv_core_grammar).read())
print('\n" Core instructions')
for inst in core["instructions"]:
EmitAsStatement(inst['opname'])
print('\n" Core operand enums')
for operand_kind in core["operand_kinds"]:
if 'enumerants' in operand_kind:
for e in operand_kind['enumerants']:
EmitAsEnumerant(e['enumerant'])
if args.extinst_glsl_grammar is not None:
print('\n" GLSL.std.450 extended instructions')
glsl = json.loads(open(args.extinst_glsl_grammar).read())
# These opcodes are really enumerant operands for the OpExtInst
# instruction.
for inst in glsl["instructions"]:
EmitAsEnumerant(inst['opname'])
if args.extinst_opencl_grammar is not None:
print('\n" OpenCL.std extended instructions')
opencl = json.loads(open(args.extinst_opencl_grammar).read())
for inst in opencl["instructions"]:
EmitAsEnumerant(inst['opname'])
if args.extinst_debuginfo_grammar is not None:
print('\n" DebugInfo extended instructions')
debuginfo = json.loads(open(args.extinst_debuginfo_grammar).read())
for inst in debuginfo["instructions"]:
EmitAsEnumerant(inst['opname'])
print('\n" DebugInfo operand enums')
for operand_kind in debuginfo["operand_kinds"]:
if 'enumerants' in operand_kind:
for e in operand_kind['enumerants']:
EmitAsEnumerant(e['enumerant'])
print('\n" OpSpecConstantOp opcodes')
for word in SPEC_CONSTANT_OP_OPCODES.split(' '):
stripped = word.strip('\n,')
if stripped != "":
# Treat as an enumerant, but without the leading "Op"
EmitAsEnumerant(stripped[2:])
print(POSTAMBLE)
if __name__ == '__main__':
main()