2009-08-30 12:02:36 +01:00
|
|
|
|
|
|
|
'''
|
|
|
|
/**************************************************************************
|
|
|
|
*
|
|
|
|
* Copyright 2009 VMware, Inc.
|
|
|
|
* All Rights Reserved.
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*
|
|
|
|
**************************************************************************/
|
|
|
|
'''
|
|
|
|
|
|
|
|
|
|
|
|
VOID, UNSIGNED, SIGNED, FIXED, FLOAT = range(5)
|
|
|
|
|
|
|
|
SWIZZLE_X, SWIZZLE_Y, SWIZZLE_Z, SWIZZLE_W, SWIZZLE_0, SWIZZLE_1, SWIZZLE_NONE, = range(7)
|
|
|
|
|
2010-02-24 16:09:44 +00:00
|
|
|
PLAIN = 'plain'
|
2009-08-30 12:02:36 +01:00
|
|
|
|
2010-02-26 15:45:48 +00:00
|
|
|
RGB = 'rgb'
|
|
|
|
SRGB = 'srgb'
|
|
|
|
YUV = 'yuv'
|
|
|
|
ZS = 'zs'
|
|
|
|
|
2009-08-30 12:02:36 +01:00
|
|
|
|
2010-02-26 10:11:36 +00:00
|
|
|
def is_pot(x):
|
2010-05-19 14:41:35 +01:00
|
|
|
return (x & (x - 1)) == 0
|
2010-02-26 10:11:36 +00:00
|
|
|
|
|
|
|
|
2010-02-26 15:45:48 +00:00
|
|
|
VERY_LARGE = 99999999999999999999999
|
|
|
|
|
|
|
|
|
2010-02-26 11:03:06 +00:00
|
|
|
class Channel:
|
|
|
|
'''Describe the channel of a color channel.'''
|
2009-08-30 12:02:36 +01:00
|
|
|
|
2011-09-19 15:04:48 +01:00
|
|
|
def __init__(self, type, norm, pure, size, name = ''):
|
2010-02-26 11:03:06 +00:00
|
|
|
self.type = type
|
2009-08-30 12:02:36 +01:00
|
|
|
self.norm = norm
|
2011-09-19 15:04:48 +01:00
|
|
|
self.pure = pure
|
2009-08-30 12:02:36 +01:00
|
|
|
self.size = size
|
2010-02-26 11:03:06 +00:00
|
|
|
self.sign = type in (SIGNED, FIXED, FLOAT)
|
2010-02-26 15:45:48 +00:00
|
|
|
self.name = name
|
2009-08-30 12:02:36 +01:00
|
|
|
|
|
|
|
def __str__(self):
|
2010-02-26 11:03:06 +00:00
|
|
|
s = str(self.type)
|
2009-08-30 12:02:36 +01:00
|
|
|
if self.norm:
|
|
|
|
s += 'n'
|
2011-09-19 15:04:48 +01:00
|
|
|
if self.pure:
|
|
|
|
s += 'p'
|
2009-08-30 12:02:36 +01:00
|
|
|
s += str(self.size)
|
|
|
|
return s
|
|
|
|
|
|
|
|
def __eq__(self, other):
|
2011-09-19 15:04:48 +01:00
|
|
|
return self.type == other.type and self.norm == other.norm and self.pure == other.pure and self.size == other.size
|
2009-08-30 12:02:36 +01:00
|
|
|
|
2010-02-26 15:45:48 +00:00
|
|
|
def max(self):
|
|
|
|
'''Maximum representable number.'''
|
|
|
|
if self.type == FLOAT:
|
|
|
|
return VERY_LARGE
|
2010-04-07 22:16:18 +01:00
|
|
|
if self.type == FIXED:
|
|
|
|
return (1 << (self.size/2)) - 1
|
2010-02-26 15:45:48 +00:00
|
|
|
if self.norm:
|
|
|
|
return 1
|
|
|
|
if self.type == UNSIGNED:
|
|
|
|
return (1 << self.size) - 1
|
|
|
|
if self.type == SIGNED:
|
2010-03-06 12:49:14 +00:00
|
|
|
return (1 << (self.size - 1)) - 1
|
2010-02-26 15:45:48 +00:00
|
|
|
assert False
|
|
|
|
|
|
|
|
def min(self):
|
|
|
|
'''Minimum representable number.'''
|
|
|
|
if self.type == FLOAT:
|
|
|
|
return -VERY_LARGE
|
2010-04-07 22:16:18 +01:00
|
|
|
if self.type == FIXED:
|
|
|
|
return -(1 << (self.size/2))
|
2010-02-26 15:45:48 +00:00
|
|
|
if self.type == UNSIGNED:
|
|
|
|
return 0
|
|
|
|
if self.norm:
|
|
|
|
return -1
|
|
|
|
if self.type == SIGNED:
|
|
|
|
return -(1 << (self.size - 1))
|
|
|
|
assert False
|
|
|
|
|
2009-08-30 12:02:36 +01:00
|
|
|
|
|
|
|
class Format:
|
|
|
|
'''Describe a pixel format.'''
|
|
|
|
|
2014-03-19 17:11:02 +00:00
|
|
|
def __init__(self, name, layout, block_width, block_height, le_channels, le_swizzles, be_channels, be_swizzles, colorspace):
|
2009-08-30 12:02:36 +01:00
|
|
|
self.name = name
|
|
|
|
self.layout = layout
|
|
|
|
self.block_width = block_width
|
|
|
|
self.block_height = block_height
|
2014-03-19 17:11:02 +00:00
|
|
|
self.le_channels = le_channels
|
|
|
|
self.le_swizzles = le_swizzles
|
|
|
|
self.be_channels = be_channels
|
|
|
|
self.be_swizzles = be_swizzles
|
2009-08-30 12:02:36 +01:00
|
|
|
self.name = name
|
|
|
|
self.colorspace = colorspace
|
|
|
|
|
|
|
|
def __str__(self):
|
|
|
|
return self.name
|
|
|
|
|
2010-02-24 15:10:46 +00:00
|
|
|
def short_name(self):
|
|
|
|
'''Make up a short norm for a format, suitable to be used as suffix in
|
|
|
|
function names.'''
|
|
|
|
|
|
|
|
name = self.name
|
|
|
|
if name.startswith('PIPE_FORMAT_'):
|
|
|
|
name = name[len('PIPE_FORMAT_'):]
|
|
|
|
name = name.lower()
|
|
|
|
return name
|
|
|
|
|
2009-08-30 12:02:36 +01:00
|
|
|
def block_size(self):
|
|
|
|
size = 0
|
2014-03-19 17:11:02 +00:00
|
|
|
for channel in self.le_channels:
|
2010-02-26 11:03:06 +00:00
|
|
|
size += channel.size
|
2009-08-30 12:02:36 +01:00
|
|
|
return size
|
|
|
|
|
2010-02-23 19:51:42 +00:00
|
|
|
def nr_channels(self):
|
|
|
|
nr_channels = 0
|
2014-03-19 17:11:02 +00:00
|
|
|
for channel in self.le_channels:
|
2010-02-26 11:03:06 +00:00
|
|
|
if channel.size:
|
2010-02-23 19:51:42 +00:00
|
|
|
nr_channels += 1
|
|
|
|
return nr_channels
|
|
|
|
|
2014-03-19 17:08:44 +00:00
|
|
|
def array_element(self):
|
2010-04-08 17:51:31 +01:00
|
|
|
if self.layout != PLAIN:
|
2014-03-19 17:08:44 +00:00
|
|
|
return None
|
2014-03-19 17:11:02 +00:00
|
|
|
ref_channel = self.le_channels[0]
|
2012-11-28 19:18:09 +00:00
|
|
|
if ref_channel.type == VOID:
|
2014-03-19 17:11:02 +00:00
|
|
|
ref_channel = self.le_channels[1]
|
|
|
|
for channel in self.le_channels:
|
2010-02-26 11:03:06 +00:00
|
|
|
if channel.size and (channel.size != ref_channel.size or channel.size % 8):
|
2014-03-19 17:08:44 +00:00
|
|
|
return None
|
2012-11-28 19:18:09 +00:00
|
|
|
if channel.type != VOID:
|
|
|
|
if channel.type != ref_channel.type:
|
2014-03-19 17:08:44 +00:00
|
|
|
return None
|
2012-11-28 19:18:09 +00:00
|
|
|
if channel.norm != ref_channel.norm:
|
2014-03-19 17:08:44 +00:00
|
|
|
return None
|
2012-11-28 19:18:09 +00:00
|
|
|
if channel.pure != ref_channel.pure:
|
2014-03-19 17:08:44 +00:00
|
|
|
return None
|
|
|
|
return ref_channel
|
|
|
|
|
|
|
|
def is_array(self):
|
|
|
|
return self.array_element() != None
|
2010-02-23 19:51:42 +00:00
|
|
|
|
|
|
|
def is_mixed(self):
|
2010-04-08 17:51:31 +01:00
|
|
|
if self.layout != PLAIN:
|
|
|
|
return False
|
2014-03-19 17:11:02 +00:00
|
|
|
ref_channel = self.le_channels[0]
|
2010-03-30 00:33:28 +01:00
|
|
|
if ref_channel.type == VOID:
|
2014-03-19 17:11:02 +00:00
|
|
|
ref_channel = self.le_channels[1]
|
|
|
|
for channel in self.le_channels[1:]:
|
2010-02-26 11:03:06 +00:00
|
|
|
if channel.type != VOID:
|
|
|
|
if channel.type != ref_channel.type:
|
2010-02-23 19:51:42 +00:00
|
|
|
return True
|
2010-02-26 11:03:06 +00:00
|
|
|
if channel.norm != ref_channel.norm:
|
2010-02-23 19:51:42 +00:00
|
|
|
return True
|
2011-09-19 15:04:48 +01:00
|
|
|
if channel.pure != ref_channel.pure:
|
|
|
|
return True
|
2010-02-23 19:51:42 +00:00
|
|
|
return False
|
|
|
|
|
2010-02-26 10:11:36 +00:00
|
|
|
def is_pot(self):
|
|
|
|
return is_pot(self.block_size())
|
|
|
|
|
2010-02-26 11:03:06 +00:00
|
|
|
def is_int(self):
|
2010-04-08 17:51:31 +01:00
|
|
|
if self.layout != PLAIN:
|
|
|
|
return False
|
2014-03-19 17:11:02 +00:00
|
|
|
for channel in self.le_channels:
|
2010-02-26 11:03:06 +00:00
|
|
|
if channel.type not in (VOID, UNSIGNED, SIGNED):
|
|
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
|
|
def is_float(self):
|
2010-04-08 17:51:31 +01:00
|
|
|
if self.layout != PLAIN:
|
|
|
|
return False
|
2014-03-19 17:11:02 +00:00
|
|
|
for channel in self.le_channels:
|
2010-02-26 11:03:06 +00:00
|
|
|
if channel.type not in (VOID, FLOAT):
|
|
|
|
return False
|
|
|
|
return True
|
|
|
|
|
2010-02-26 15:45:48 +00:00
|
|
|
def is_bitmask(self):
|
2010-04-08 17:51:31 +01:00
|
|
|
if self.layout != PLAIN:
|
|
|
|
return False
|
2010-03-06 12:48:39 +00:00
|
|
|
if self.block_size() not in (8, 16, 32):
|
2010-02-26 15:45:48 +00:00
|
|
|
return False
|
2014-03-19 17:11:02 +00:00
|
|
|
for channel in self.le_channels:
|
2010-02-26 15:45:48 +00:00
|
|
|
if channel.type not in (VOID, UNSIGNED, SIGNED):
|
|
|
|
return False
|
|
|
|
return True
|
|
|
|
|
2014-03-19 17:08:44 +00:00
|
|
|
def is_pure_color(self):
|
|
|
|
if self.layout != PLAIN or self.colorspace == ZS:
|
|
|
|
return False
|
|
|
|
pures = [channel.pure
|
2014-03-19 17:11:02 +00:00
|
|
|
for channel in self.le_channels
|
2014-03-19 17:08:44 +00:00
|
|
|
if channel.type != VOID]
|
|
|
|
for x in pures:
|
|
|
|
assert x == pures[0]
|
|
|
|
return pures[0]
|
|
|
|
|
|
|
|
def channel_type(self):
|
|
|
|
types = [channel.type
|
2014-03-19 17:11:02 +00:00
|
|
|
for channel in self.le_channels
|
2014-03-19 17:08:44 +00:00
|
|
|
if channel.type != VOID]
|
|
|
|
for x in types:
|
|
|
|
assert x == types[0]
|
|
|
|
return types[0]
|
|
|
|
|
|
|
|
def is_pure_signed(self):
|
|
|
|
return self.is_pure_color() and self.channel_type() == SIGNED
|
|
|
|
|
|
|
|
def is_pure_unsigned(self):
|
|
|
|
return self.is_pure_color() and self.channel_type() == UNSIGNED
|
|
|
|
|
|
|
|
def has_channel(self, id):
|
2014-03-19 17:11:02 +00:00
|
|
|
return self.le_swizzles[id] != SWIZZLE_NONE
|
2014-03-19 17:08:44 +00:00
|
|
|
|
|
|
|
def has_depth(self):
|
|
|
|
return self.colorspace == ZS and self.has_channel(0)
|
|
|
|
|
|
|
|
def has_stencil(self):
|
|
|
|
return self.colorspace == ZS and self.has_channel(1)
|
|
|
|
|
2009-08-30 12:02:36 +01:00
|
|
|
def stride(self):
|
|
|
|
return self.block_size()/8
|
|
|
|
|
|
|
|
|
2010-02-26 11:03:06 +00:00
|
|
|
_type_parse_map = {
|
2009-08-30 12:02:36 +01:00
|
|
|
'': VOID,
|
|
|
|
'x': VOID,
|
|
|
|
'u': UNSIGNED,
|
|
|
|
's': SIGNED,
|
|
|
|
'h': FIXED,
|
|
|
|
'f': FLOAT,
|
|
|
|
}
|
|
|
|
|
|
|
|
_swizzle_parse_map = {
|
|
|
|
'x': SWIZZLE_X,
|
|
|
|
'y': SWIZZLE_Y,
|
|
|
|
'z': SWIZZLE_Z,
|
|
|
|
'w': SWIZZLE_W,
|
|
|
|
'0': SWIZZLE_0,
|
|
|
|
'1': SWIZZLE_1,
|
|
|
|
'_': SWIZZLE_NONE,
|
|
|
|
}
|
|
|
|
|
2014-03-19 17:10:12 +00:00
|
|
|
def _parse_channels(fields, layout, colorspace, swizzles):
|
|
|
|
if layout == PLAIN:
|
|
|
|
names = ['']*4
|
|
|
|
if colorspace in (RGB, SRGB):
|
|
|
|
for i in range(4):
|
|
|
|
swizzle = swizzles[i]
|
|
|
|
if swizzle < 4:
|
|
|
|
names[swizzle] += 'rgba'[i]
|
|
|
|
elif colorspace == ZS:
|
|
|
|
for i in range(4):
|
|
|
|
swizzle = swizzles[i]
|
|
|
|
if swizzle < 4:
|
|
|
|
names[swizzle] += 'zs'[i]
|
|
|
|
else:
|
|
|
|
assert False
|
|
|
|
for i in range(4):
|
|
|
|
if names[i] == '':
|
|
|
|
names[i] = 'x'
|
|
|
|
else:
|
|
|
|
names = ['x', 'y', 'z', 'w']
|
|
|
|
|
|
|
|
channels = []
|
|
|
|
for i in range(0, 4):
|
|
|
|
field = fields[i]
|
|
|
|
if field:
|
|
|
|
type = _type_parse_map[field[0]]
|
|
|
|
if field[1] == 'n':
|
|
|
|
norm = True
|
|
|
|
pure = False
|
|
|
|
size = int(field[2:])
|
|
|
|
elif field[1] == 'p':
|
|
|
|
pure = True
|
|
|
|
norm = False
|
|
|
|
size = int(field[2:])
|
|
|
|
else:
|
|
|
|
norm = False
|
|
|
|
pure = False
|
|
|
|
size = int(field[1:])
|
|
|
|
else:
|
|
|
|
type = VOID
|
|
|
|
norm = False
|
|
|
|
pure = False
|
|
|
|
size = 0
|
|
|
|
channel = Channel(type, norm, pure, size, names[i])
|
|
|
|
channels.append(channel)
|
|
|
|
|
|
|
|
return channels
|
|
|
|
|
2009-08-30 12:02:36 +01:00
|
|
|
def parse(filename):
|
2016-01-29 01:58:29 +00:00
|
|
|
'''Parse the format description in CSV format in terms of the
|
2010-02-26 11:03:06 +00:00
|
|
|
Channel and Format classes above.'''
|
2009-08-30 12:02:36 +01:00
|
|
|
|
|
|
|
stream = open(filename)
|
|
|
|
formats = []
|
|
|
|
for line in stream:
|
2010-02-24 13:47:42 +00:00
|
|
|
try:
|
|
|
|
comment = line.index('#')
|
|
|
|
except ValueError:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
line = line[:comment]
|
|
|
|
line = line.strip()
|
|
|
|
if not line:
|
|
|
|
continue
|
2010-02-26 15:45:48 +00:00
|
|
|
|
2009-08-30 12:02:36 +01:00
|
|
|
fields = [field.strip() for field in line.split(',')]
|
2014-07-21 16:32:38 +01:00
|
|
|
if len (fields) == 10:
|
|
|
|
fields += fields[4:9]
|
|
|
|
assert len (fields) == 15
|
2010-02-26 15:45:48 +00:00
|
|
|
|
2009-08-30 12:02:36 +01:00
|
|
|
name = fields[0]
|
|
|
|
layout = fields[1]
|
|
|
|
block_width, block_height = map(int, fields[2:4])
|
2014-03-19 17:10:12 +00:00
|
|
|
colorspace = fields[9]
|
2010-02-26 15:45:48 +00:00
|
|
|
|
2014-03-19 17:11:02 +00:00
|
|
|
le_swizzles = [_swizzle_parse_map[swizzle] for swizzle in fields[8]]
|
|
|
|
le_channels = _parse_channels(fields[4:8], layout, colorspace, le_swizzles)
|
|
|
|
|
2014-07-21 16:32:38 +01:00
|
|
|
be_swizzles = [_swizzle_parse_map[swizzle] for swizzle in fields[14]]
|
|
|
|
be_channels = _parse_channels(fields[10:14], layout, colorspace, be_swizzles)
|
2014-03-19 17:11:02 +00:00
|
|
|
|
|
|
|
le_shift = 0
|
|
|
|
for channel in le_channels:
|
|
|
|
channel.shift = le_shift
|
|
|
|
le_shift += channel.size
|
|
|
|
|
|
|
|
be_shift = 0
|
|
|
|
for channel in be_channels[3::-1]:
|
|
|
|
channel.shift = be_shift
|
|
|
|
be_shift += channel.size
|
2010-02-26 15:45:48 +00:00
|
|
|
|
2014-03-19 17:11:02 +00:00
|
|
|
assert le_shift == be_shift
|
2014-07-21 16:32:38 +01:00
|
|
|
for i in range(4):
|
|
|
|
assert (le_swizzles[i] != SWIZZLE_NONE) == (be_swizzles[i] != SWIZZLE_NONE)
|
2013-06-24 14:48:56 +01:00
|
|
|
|
2014-03-19 17:11:02 +00:00
|
|
|
format = Format(name, layout, block_width, block_height, le_channels, le_swizzles, be_channels, be_swizzles, colorspace)
|
2010-02-26 15:45:48 +00:00
|
|
|
formats.append(format)
|
2009-08-30 12:02:36 +01:00
|
|
|
return formats
|
|
|
|
|