COPYRIGHT = """\ /* * Copyright 2017 Google * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associated documentation files (the * "Software"), to deal in the Software without restriction, including * without limitation the rights to use, copy, modify, merge, publish, * distribute, sub license, and/or sell copies of the Software, and to * permit persons to whom the Software is furnished to do so, subject to * the following conditions: * * The above copyright notice and this permission notice (including the * next paragraph) shall be included in all copies or substantial portions * of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ """ # stdlib import argparse from sys import stdout from mako.template import Template # local import format_parser def parse_args(): p = argparse.ArgumentParser() p.add_argument("csv") p.add_argument("out") return p.parse_args() def get_unorm_to_srgb_map(formats): names = set(fmt.name for fmt in formats) for fmt in formats: if fmt.colorspace != 'srgb': continue; replacements = [ ('SRGB', 'RGB'), ('SRGB', 'UNORM'), ('SRGB8_ALPHA8', 'RGBA'), ('SRGB8_ALPHA8', 'RGBA8'), ('SRGB_ALPHA_UNORM', 'RGBA_UNORM'), ] found_unorm_name = False for rep in replacements: if fmt.name.find(rep[0]) == -1: continue unorm_name = fmt.name.replace(rep[0], rep[1]) if unorm_name in names: yield unorm_name, fmt.name found_unorm_name = True break # Every sRGB format MUST have a UNORM equivalent assert found_unorm_name def get_rgbx_to_rgba_map(formats): names = set(fmt.name for fmt in formats) for fmt in formats: if not fmt.has_channel('r') or not fmt.has_channel('x'): continue # The condition above will still let MESA_FORMAT_R9G9B9E5_FLOAT # through. We need to ensure it actually has an X in the name. if not 'X' in fmt.name: continue rgbx_name = fmt.name rgba_name = rgbx_name.replace("X", "A") if rgba_name not in names: continue; yield rgbx_name, rgba_name def get_intensity_to_red_map(formats): names = set(fmt.name for fmt in formats) for fmt in formats: if str(fmt.swizzle) != 'xxxx': continue i_name = fmt.name r_name = i_name.replace("_I_", "_R_") assert r_name in names yield i_name, r_name TEMPLATE = Template(COPYRIGHT + """ #include "formats.h" #include "util/macros.h" /** * For an sRGB format, return the corresponding linear color space format. * For non-sRGB formats, return the format as-is. */ mesa_format _mesa_get_srgb_format_linear(mesa_format format) { switch (format) { %for unorm, srgb in unorm_to_srgb_map: case ${srgb}: return ${unorm}; %endfor default: return format; } } /** * For a linear format, return the corresponding sRGB color space format. * For an sRGB format, return the format as-is. * Assert-fails if the format is not sRGB and does not have an sRGB equivalent. */ mesa_format _mesa_get_linear_format_srgb(mesa_format format) { switch (format) { %for unorm, srgb in unorm_to_srgb_map: case ${unorm}: return ${srgb}; %endfor %for unorm, srgb in unorm_to_srgb_map: case ${srgb}: %endfor return format; default: unreachable("Given format does not have an sRGB equivalent"); } } /** * For an intensity format, return the corresponding red format. For other * formats, return the format as-is. */ mesa_format _mesa_get_intensity_format_red(mesa_format format) { switch (format) { %for i, r in intensity_to_red_map: case ${i}: return ${r}; %endfor default: return format; } } /** * If the format has an alpha channel, and there exists a non-alpha * variant of the format with an identical bit layout, then return * the non-alpha format. Otherwise return the original format. * * Examples: * Fallback exists: * MESA_FORMAT_R8G8B8X8_UNORM -> MESA_FORMAT_R8G8B8A8_UNORM * MESA_FORMAT_RGBX_UNORM16 -> MESA_FORMAT_RGBA_UNORM16 * * No fallback: * MESA_FORMAT_R8G8B8A8_UNORM -> MESA_FORMAT_R8G8B8A8_UNORM * MESA_FORMAT_Z_FLOAT32 -> MESA_FORMAT_Z_FLOAT32 */ mesa_format _mesa_format_fallback_rgbx_to_rgba(mesa_format format) { switch (format) { %for rgbx, rgba in rgbx_to_rgba_map: case ${rgbx}: return ${rgba}; %endfor default: return format; } } """); def main(): pargs = parse_args() formats = list(format_parser.parse(pargs.csv)) template_env = { 'unorm_to_srgb_map': list(get_unorm_to_srgb_map(formats)), 'rgbx_to_rgba_map': list(get_rgbx_to_rgba_map(formats)), 'intensity_to_red_map': list(get_intensity_to_red_map(formats)), } with open(pargs.out, 'w') as f: f.write(TEMPLATE.render(**template_env)) if __name__ == "__main__": main()