1#!/usr/bin/env python3 2# 3# Copyright (C) 2016 Google, Inc. 4# 5# Permission is hereby granted, free of charge, to any person obtaining a 6# copy of this software and associated documentation files (the "Software"), 7# to deal in the Software without restriction, including without limitation 8# the rights to use, copy, modify, merge, publish, distribute, sublicense, 9# and/or sell copies of the Software, and to permit persons to whom the 10# Software is furnished to do so, subject to the following conditions: 11# 12# The above copyright notice and this permission notice shall be included 13# in all copies or substantial portions of the Software. 14# 15# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 18# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 20# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER 21# DEALINGS IN THE SOFTWARE. 22 23"""Compile GLSL to SPIR-V. 24 25Depends on glslangValidator. 26""" 27 28import os 29import sys 30import subprocess 31import struct 32import re 33 34SPIRV_MAGIC = 0x07230203 35COLUMNS = 4 36INDENT = 4 37 38in_filename = sys.argv[1] 39out_filename = sys.argv[2] if len(sys.argv) > 2 else None 40validator = sys.argv[3] if len(sys.argv) > 3 else \ 41 "../../../glslang/build/install/bin/glslangValidator" 42 43def identifierize(s): 44 # translate invalid chars 45 s = re.sub("[^0-9a-zA-Z_]", "_", s) 46 # translate leading digits 47 return re.sub("^[^a-zA-Z_]+", "_", s) 48 49def compile_glsl(filename, tmpfile): 50 # invoke glslangValidator 51 try: 52 args = [validator, "-V", "-H", "-o", tmpfile, filename] 53 output = subprocess.check_output(args, universal_newlines=True) 54 except subprocess.CalledProcessError as e: 55 print(e.output, file=sys.stderr) 56 exit(1) 57 58 # read the temp file into a list of SPIR-V words 59 words = [] 60 with open(tmpfile, "rb") as f: 61 data = f.read() 62 assert(len(data) and len(data) % 4 == 0) 63 64 # determine endianness 65 fmt = ("<" if data[0] == (SPIRV_MAGIC & 0xff) else ">") + "I" 66 for i in range(0, len(data), 4): 67 words.append(struct.unpack(fmt, data[i:(i + 4)])[0]) 68 69 assert(words[0] == SPIRV_MAGIC) 70 71 72 # remove temp file 73 os.remove(tmpfile) 74 75 return (words, output.rstrip()) 76 77base = os.path.basename(in_filename) 78words, comments = compile_glsl(in_filename, base + ".tmp") 79 80literals = [] 81for i in range(0, len(words), COLUMNS): 82 columns = ["0x%08x" % word for word in words[i:(i + COLUMNS)]] 83 literals.append(" " * INDENT + ", ".join(columns) + ",") 84 85header = """#include <stdint.h> 86 87#if 0 88%s 89#endif 90 91static const uint32_t %s[%d] = { 92%s 93}; 94""" % (comments, identifierize(base), len(words), "\n".join(literals)) 95 96if out_filename: 97 with open(out_filename, "w") as f: 98 print(header, end="", file=f) 99else: 100 print(header, end="") 101