ui/surface/compile_hlsl.py

   1 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import optparse
   6 import os.path
   7 import re
   8 import subprocess
   9 import sys
  10
  11
  12 def ConvertToCamelCase(input):
  13   """Converts the input string from 'unix_hacker' style to 'CamelCase' style."""
  14   return ''.join(x[:1].upper() + x[1:] for x in input.split('_'))
  15
  16
  17 def ExtractShaderTargetNamesFromSource(source_hlsl_file):
  18   """Parses '@gyp_compile' and '@gyp_namespace' metadata from an .hlsl file."""
  19   # matches strings like // @gyp_compile(arg_a, arg_b) ...
  20   gyp_compile = re.compile(
  21       '^//\s*@gyp_compile\(\s*(?P<profile>[a-zA-Z0-9_]+)\s*,'
  22       '\s*(?P<function_name>[a-zA-Z0-9_]+)\s*\).*')
  23   # matches strings like // @gyp_namespace(arg_a) ...
  24   gyp_namespace = re.compile(
  25       '^//\s*@gyp_namespace\(\s*(?P<namespace>[a-zA-Z0-9_]+)\s*\).*')
  26
  27   shader_targets = []  # tuples like ('vs_2_0', 'vertexMain')
  28   namespace = None
  29   with open(source_hlsl_file) as hlsl:
  30     for line_number, line in enumerate(hlsl.read().splitlines(), 1):
  31       m = gyp_compile.match(line)
  32       if m:
  33         shader_targets.append((m.group('profile'), m.group('function_name')))
  34         continue
  35       m = gyp_namespace.match(line)
  36       if m:
  37         namespace = m.group('namespace')
  38         continue
  39       if '@gyp' in line:
  40         print '%s(%d) : warning: ignoring malformed @gyp directive ' % (
  41             source_hlsl_file, line_number)
  42
  43   if not shader_targets:
  44     print (
  45 """%s(%d) : error: Reached end of file without finding @gyp_compile directive.
  46
  47     By convention, each HLSL source must contain one or more @gyp_compile
  48     directives in its comments, as metadata informing the Chrome build tool
  49     which entry points should be compiled. For example, to specify compilation
  50     of a function named 'vertexMain' as a shader model 2 vertex shader:
  51
  52         // @gyp_compile(vs_2_0, vertexMain)
  53
  54     Or to compile a pixel shader 2.0 function named 'someOtherShader':
  55
  56         // @gyp_compile(ps_2_0, someOtherShader)
  57
  58     To wrap everything in a C++ namespace 'foo_bar', add a line somewhere like:
  59
  60         // @gyp_namespace(foo_bar)
  61
  62     (Namespaces are optional)
  63 """ % (source_hlsl_file, line_number))
  64     sys.exit(1)
  65   return (shader_targets, namespace)
  66
  67
  68 def GetCppVariableName(function_name):
  69   return 'k%s' % ConvertToCamelCase(function_name)
  70
  71
  72 def CompileMultipleHLSLShadersToOneHeaderFile(fxc_compiler_path,
  73                                               source_hlsl_file,
  74                                               namespace,
  75                                               shader_targets,
  76                                               target_header_file,
  77                                               target_cc_file):
  78   """Compiles specified shaders from an .hlsl file into a single C++ header."""
  79   header_output = []
  80   # Invoke the compiler one at a time to write the c++ header file,
  81   # then read that header file into |header_output|.
  82   for (compiler_profile, hlsl_function_name) in shader_targets:
  83     file_name_only = os.path.basename(source_hlsl_file)
  84     base_filename, _ = os.path.splitext(file_name_only)
  85     cpp_global_var_name = GetCppVariableName(hlsl_function_name)
  86
  87     command = [fxc_compiler_path,
  88                source_hlsl_file,            # From this HLSL file
  89                '/E', hlsl_function_name,    # Compile one function
  90                '/T', compiler_profile,      # As a vertex or pixel shader
  91                '/Vn', cpp_global_var_name,  # Into a C++ constant thus named
  92                '/Fh', target_header_file,   # Declared in this C++ header file.
  93                '/O3']                       # Fast is better than slow.
  94     (out, err) = subprocess.Popen(command,
  95                                   stdout=subprocess.PIPE,
  96                                   stderr=subprocess.PIPE,
  97                                   shell=False).communicate()
  98     if err:
  99       print 'Error while compiling %s in file %s' % (
 100           hlsl_function_name, source_hlsl_file)
 101       print err
 102       sys.exit(1)
 103     with open(target_header_file, 'r') as header:
 104       header_output.append(header.read())
 105
 106   # Now, re-write the .h and .cc files with the concatenation of all
 107   # the individual passes.
 108   classname = '%sHLSL' % (ConvertToCamelCase(base_filename))
 109   preamble = '\n'.join([
 110     '/' * 77,
 111     '// This file is auto-generated from %s' % file_name_only,
 112     '//',
 113     "// To edit it directly would be a fool's errand.",
 114     '/' * 77,
 115     '',
 116     ''])
 117   with open(target_header_file, 'wb') as h:
 118     h.write(preamble)
 119     h.write('#pragma once\n')
 120     h.write('#include <windows.h>\n\n')
 121     if namespace:
 122       h.write('namespace %s {\n\n' % namespace)
 123     h.write('namespace %s {\n\n' % classname)
 124     for _, function_name in shader_targets:
 125       h.write('extern const BYTE %s[];\n' % GetCppVariableName(function_name))
 126     h.write('\n}  // namespace %s\n' % classname)
 127     if namespace:
 128       h.write('\n}  // namespace %s\n' % namespace)
 129
 130   with open(target_cc_file, 'wb') as cc:
 131     cc.write(preamble)
 132     cc.write('#include "%s"\n\n' % os.path.basename(target_header_file))
 133     if namespace:
 134       cc.write('namespace %s {\n\n' % namespace)
 135     cc.write('namespace %s {\n\n' % classname)
 136     cc.write(''.join(header_output))
 137     cc.write('\n}  // namespace %s\n' % classname)
 138     if namespace:
 139       cc.write('\n}  // namespace %s\n' % namespace)
 140
 141
 142 if __name__ == '__main__':
 143   parser = optparse.OptionParser()
 144   parser.add_option('--shader_compiler_tool', dest='compiler')
 145   parser.add_option('--output_h_file', dest='header_file')
 146   parser.add_option('--output_cc_file', dest='cc_file')
 147   parser.add_option('--input_hlsl_file', dest='hlsl_file')
 148   (options, args) = parser.parse_args()
 149
 150   hlsl_file = os.path.abspath(options.hlsl_file)
 151   shader_targets, namespace = ExtractShaderTargetNamesFromSource(hlsl_file)
 152
 153   header_file = os.path.normpath(options.header_file)
 154   cc_file = os.path.normpath(options.cc_file)
 155   CompileMultipleHLSLShadersToOneHeaderFile(options.compiler,
 156                                             hlsl_file,
 157                                             namespace,
 158                                             shader_targets,
 159                                             header_file,
 160                                             cc_file)