src/util/format/u_format_table.py

   1 from __future__ import print_function
   2
   3 CopyRight = '''
   4 /**************************************************************************
   5  *
   6  * Copyright 2010 VMware, Inc.
   7  * All Rights Reserved.
   8  *
   9  * Permission is hereby granted, free of charge, to any person obtaining a
  10  * copy of this software and associated documentation files (the
  11  * "Software"), to deal in the Software without restriction, including
  12  * without limitation the rights to use, copy, modify, merge, publish,
  13  * distribute, sub license, and/or sell copies of the Software, and to
  14  * permit persons to whom the Software is furnished to do so, subject to
  15  * the following conditions:
  16  *
  17  * The above copyright notice and this permission notice (including the
  18  * next paragraph) shall be included in all copies or substantial portions
  19  * of the Software.
  20  *
  21  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  22  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  23  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  24  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  25  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  26  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  27  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  28  *
  29  **************************************************************************/
  30 '''
  31
  32
  33 import sys, os
  34
  35 from u_format_parse import *
  36 import u_format_pack
  37
  38
  39 def layout_map(layout):
  40     return 'UTIL_FORMAT_LAYOUT_' + str(layout).upper()
  41
  42
  43 def colorspace_map(colorspace):
  44     return 'UTIL_FORMAT_COLORSPACE_' + str(colorspace).upper()
  45
  46
  47 colorspace_channels_map = {
  48     'rgb': ['r', 'g', 'b', 'a'],
  49     'srgb': ['sr', 'sg', 'sb', 'a'],
  50     'zs': ['z', 's'],
  51     'yuv': ['y', 'u', 'v'],
  52 }
  53
  54
  55 type_map = {
  56     VOID:     "UTIL_FORMAT_TYPE_VOID",
  57     UNSIGNED: "UTIL_FORMAT_TYPE_UNSIGNED",
  58     SIGNED:   "UTIL_FORMAT_TYPE_SIGNED",
  59     FIXED:    "UTIL_FORMAT_TYPE_FIXED",
  60     FLOAT:    "UTIL_FORMAT_TYPE_FLOAT",
  61 }
  62
  63
  64 def bool_map(value):
  65     if value:
  66         return "TRUE"
  67     else:
  68         return "FALSE"
  69
  70
  71 swizzle_map = {
  72     SWIZZLE_X:    "PIPE_SWIZZLE_X",
  73     SWIZZLE_Y:    "PIPE_SWIZZLE_Y",
  74     SWIZZLE_Z:    "PIPE_SWIZZLE_Z",
  75     SWIZZLE_W:    "PIPE_SWIZZLE_W",
  76     SWIZZLE_0:    "PIPE_SWIZZLE_0",
  77     SWIZZLE_1:    "PIPE_SWIZZLE_1",
  78     SWIZZLE_NONE: "PIPE_SWIZZLE_NONE",
  79 }
  80
  81 def has_access(format):
  82     # We don't generate code for YUV formats, and many of the new ones lack
  83     # pack/unpack functions for softpipe/llvmpipe.
  84     noaccess_formats = [
  85         'r1_unorm',
  86         'yv12',
  87         'yv16',
  88         'iyuv',
  89         'nv12',
  90         'nv16',
  91         'nv21',
  92         'p010',
  93         'p012',
  94         'p016',
  95         'xyuv',
  96         'ayuv',
  97         'r8g8_r8b8_unorm',
  98         'g8r8_b8r8_unorm',
  99         'g8r8_g8b8_unorm',
 100         'y8_u8_v8_422_unorm',
 101         'y8_u8v8_422_unorm',
 102         'y8_u8_v8_444_unorm',
 103         'y16_u16_v16_420_unorm',
 104         'y16_u16_v16_422_unorm',
 105         'y16_u16v16_422_unorm',
 106         'y16_u16_v16_444_unorm',
 107     ]
 108     if format.short_name() in noaccess_formats:
 109         return False
 110     if format.layout in ('astc', 'atc', 'fxt1'):
 111         return False
 112     if format.layout == 'etc' and format.short_name() != 'etc1_rgb8':
 113         return False
 114     return True
 115
 116 def write_format_table_header(file):
 117     print('/* This file is autogenerated by u_format_table.py from u_format.csv. Do not edit directly. */', file=file)
 118     print(file=file)
 119     # This will print the copyright message on the top of this file
 120     print(CopyRight.strip(), file=file)
 121     print(file=file)
 122     print('#include "util/format/u_format.h"', file=file)
 123
 124 def write_format_table(formats):
 125     write_format_table_header(sys.stdout)
 126     print('#include "u_format_bptc.h"')
 127     print('#include "u_format_s3tc.h"')
 128     print('#include "u_format_rgtc.h"')
 129     print('#include "u_format_latc.h"')
 130     print('#include "u_format_etc.h"')
 131     print()
 132
 133     write_format_table_header(sys.stdout2)
 134
 135     u_format_pack.generate(formats)
 136
 137     def do_channel_array(channels, swizzles):
 138         print("   {")
 139         for i in range(4):
 140             channel = channels[i]
 141             if i < 3:
 142                 sep = ","
 143             else:
 144                 sep = ""
 145             if channel.size:
 146                 print("      {%s, %s, %s, %u, %u}%s\t/* %s = %s */" % (type_map[channel.type], bool_map(channel.norm), bool_map(channel.pure), channel.size, channel.shift, sep, "xyzw"[i], channel.name))
 147             else:
 148                 print("      {0, 0, 0, 0, 0}%s" % (sep,))
 149         print("   },")
 150
 151     def do_swizzle_array(channels, swizzles):
 152         print("   {")
 153         for i in range(4):
 154             swizzle = swizzles[i]
 155             if i < 3:
 156                 sep = ","
 157             else:
 158                 sep = ""
 159             try:
 160                 comment = colorspace_channels_map[format.colorspace][i]
 161             except (KeyError, IndexError):
 162                 comment = 'ignored'
 163             print("      %s%s\t/* %s */" % (swizzle_map[swizzle], sep, comment))
 164         print("   },")
 165
 166     def generate_table_getter(type):
 167         print("const struct util_format_%sdescription *" % type)
 168         print("util_format_%sdescription(enum pipe_format format)" % type)
 169         print("{")
 170         print("   if (format >= ARRAY_SIZE(util_format_%sdescriptions))" % (type))
 171         print("      return NULL;")
 172         print()
 173         print("   return &util_format_%sdescriptions[format];" % (type))
 174         print("}")
 175         print()
 176
 177     print('static const struct util_format_description')
 178     print('util_format_descriptions[] = {')
 179     for format in formats:
 180         sn = format.short_name()
 181
 182         print("   [%s] = {" % (format.name,))
 183         print("      %s," % (format.name,))
 184         print("      \"%s\"," % (format.name,))
 185         print("      \"%s\"," % (sn,))
 186         print("      {%u, %u, %u, %u},\t/* block */" % (format.block_width, format.block_height, format.block_depth, format.block_size()))
 187         print("      %s," % (layout_map(format.layout),))
 188         print("      %u,\t/* nr_channels */" % (format.nr_channels(),))
 189         print("      %s,\t/* is_array */" % (bool_map(format.is_array()),))
 190         print("      %s,\t/* is_bitmask */" % (bool_map(format.is_bitmask()),))
 191         print("      %s,\t/* is_mixed */" % (bool_map(format.is_mixed()),))
 192         print("      %s,\t/* is_unorm */" % (bool_map(format.is_unorm()),))
 193         print("      %s,\t/* is_snorm */" % (bool_map(format.is_snorm()),))
 194         u_format_pack.print_channels(format, do_channel_array)
 195         u_format_pack.print_channels(format, do_swizzle_array)
 196         print("      %s," % (colorspace_map(format.colorspace),))
 197         print("   },")
 198         print()
 199     print("};")
 200     print()
 201     generate_table_getter("")
 202
 203     print('static const struct util_format_pack_description')
 204     print('util_format_pack_descriptions[] = {')
 205     for format in formats:
 206         sn = format.short_name()
 207
 208         if not has_access(format):
 209             print("   [%s] = { 0 }," % (format.name,))
 210             continue
 211
 212         print("   [%s] = {" % (format.name,))
 213         if format.colorspace != ZS and not format.is_pure_color():
 214             print("      .pack_rgba_8unorm = &util_format_%s_pack_rgba_8unorm," % sn)
 215             print("      .pack_rgba_float = &util_format_%s_pack_rgba_float," % sn)
 216
 217         if format.has_depth():
 218             print("      .pack_z_32unorm = &util_format_%s_pack_z_32unorm," % sn)
 219             print("      .pack_z_float = &util_format_%s_pack_z_float," % sn)
 220
 221         if format.has_stencil():
 222             print("      .pack_s_8uint = &util_format_%s_pack_s_8uint," % sn)
 223
 224         if format.is_pure_unsigned() or format.is_pure_signed():
 225             print("      .pack_rgba_uint = &util_format_%s_pack_unsigned," % sn)
 226             print("      .pack_rgba_sint = &util_format_%s_pack_signed," % sn)
 227         print("   },")
 228         print()
 229     print("};")
 230     print()
 231     generate_table_getter("pack_")
 232
 233     print('static const struct util_format_unpack_description')
 234     print('util_format_unpack_descriptions[] = {')
 235     for format in formats:
 236         sn = format.short_name()
 237
 238         if not has_access(format):
 239             print("   [%s] = { 0 }," % (format.name,))
 240             continue
 241
 242         print("   [%s] = {" % (format.name,))
 243         if format.colorspace != ZS and not format.is_pure_color():
 244             print("      .unpack_rgba_8unorm = &util_format_%s_unpack_rgba_8unorm," % sn)
 245             if format.layout == 's3tc' or format.layout == 'rgtc':
 246                 print("      .fetch_rgba_8unorm = &util_format_%s_fetch_rgba_8unorm," % sn)
 247             print("      .unpack_rgba = &util_format_%s_unpack_rgba_float," % sn)
 248             print("      .fetch_rgba_float = &util_format_%s_fetch_rgba_float," % sn)
 249
 250         if format.has_depth():
 251             print("      .unpack_z_32unorm = &util_format_%s_unpack_z_32unorm," % sn)
 252             print("      .unpack_z_float = &util_format_%s_unpack_z_float," % sn)
 253
 254         if format.has_stencil():
 255             print("      .unpack_s_8uint = &util_format_%s_unpack_s_8uint," % sn)
 256
 257         if format.is_pure_unsigned():
 258             print("      .unpack_rgba = &util_format_%s_unpack_unsigned," % sn)
 259             print("      .fetch_rgba_uint = &util_format_%s_fetch_unsigned," % sn)
 260         elif format.is_pure_signed():
 261             print("      .unpack_rgba = &util_format_%s_unpack_signed," % sn)
 262             print("      .fetch_rgba_sint = &util_format_%s_fetch_signed," % sn)
 263         print("   },")
 264     print("};")
 265     print()
 266
 267     generate_table_getter("unpack_")
 268
 269 def main():
 270     formats = []
 271
 272     sys.stdout2 = open(os.devnull, "w")
 273
 274     for arg in sys.argv[1:]:
 275         if arg == '--header':
 276             sys.stdout2 = sys.stdout
 277             sys.stdout = open(os.devnull, "w")
 278             continue
 279
 280         formats.extend(parse(arg))
 281
 282     write_format_table(formats)
 283
 284 if __name__ == '__main__':
 285     main()