13c827367444ee418f129b2c238299f49d3264554Jarkko Poyryimport sys 23c827367444ee418f129b2c238299f49d3264554Jarkko Poyryimport random 33c827367444ee418f129b2c238299f49d3264554Jarkko Poyryimport operator 43c827367444ee418f129b2c238299f49d3264554Jarkko Poyryimport itertools 53c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 63c827367444ee418f129b2c238299f49d3264554Jarkko Poyryfrom genutil import * 73c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 83c827367444ee418f129b2c238299f49d3264554Jarkko Poyryrandom.seed(1234567) 93c827367444ee418f129b2c238299f49d3264554Jarkko Poyryindices = xrange(sys.maxint) 103c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 113c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Constructors: 123c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# 133c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - scalars types 143c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * int <-> float <-> bool (also float(float) etc.) 153c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * to bool: zero means false, others true 163c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * from bool: false==0, true==1 173c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * \todo [petri] float<->int rounding rules? 183c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - scalar type from vector 193c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * choose the first component 203c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - vectors & matrices 213c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * vector from scalar: broadcast to all components 223c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * matrix from scalar: broadcast scalar to diagonal, other components zero 233c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * vector from vector: copy existing components 243c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# + illegal: vector from smaller vector 253c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * mat from mat: copy existing components, other components from identity matrix 263c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * from components: consumed by-component in column-major order, must have same 273c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# number of components, 283c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# + note: vec4(mat2) valid 293c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# \todo [petri] Implement! 303c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - notes: 313c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * type conversions are always allowed: mat3(ivec3, bvec3, bool, int, float) is valid! 323c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# 333c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Accessors: 343c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# 353c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - vector components 363c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * .xyzw, .rgba, .stpq 373c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * illegal to mix 383c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * now allowed for scalar types 393c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * legal to chain: vec4.rgba.xyzw.stpq 403c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * illegal to select more than 4 components 413c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * array indexing with [] operator 423c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * can also write! 433c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - matrix columns 443c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * [] accessor 453c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * note: mat4[0].x = 1.0; vs mat4[0][0] = 1.0; ?? 463c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# * out-of-bounds accesses 473c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# 483c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# \todo [petri] Accessors! 493c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# 503c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Spec issues: 513c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# 523c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - constructing larger vector from smaller: vec3(vec2) ? 533c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# - base type and size conversion at same time: vec4(bool), int(vec3) allowed? 543c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 553c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineVec(comps): 563c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res = [] 573c827367444ee418f129b2c238299f49d3264554Jarkko Poyry for ndx in range(len(comps[0])): 583c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# for x in comps: 593c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# print x[ndx].toFloat().getScalars() , 603c827367444ee418f129b2c238299f49d3264554Jarkko Poyry scalars = reduce(operator.add, [x[ndx].toFloat().getScalars() for x in comps]) 613c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# print "->", scalars 623c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res.append(Vec.fromScalarList(scalars)) 633c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return res 643c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 653c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineIVec(comps): 663c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res = [] 673c827367444ee418f129b2c238299f49d3264554Jarkko Poyry for ndx in range(len(comps[0])): 683c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res.append(Vec.fromScalarList(reduce(operator.add, [x[ndx].toInt().getScalars() for x in comps]))) 693c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return res 703c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 713c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineUVec(comps): 723c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return [x.toUint() for x in combineIVec(comps)] 733c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 743c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineBVec(comps): 753c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res = [] 763c827367444ee418f129b2c238299f49d3264554Jarkko Poyry for ndx in range(len(comps[0])): 773c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res.append(Vec.fromScalarList(reduce(operator.add, [x[ndx].toBool().getScalars() for x in comps]))) 783c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return res 793c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 803c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat(numCols, numRows, comps): 813c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res = [] 823c827367444ee418f129b2c238299f49d3264554Jarkko Poyry for ndx in range(len(comps[0])): 833c827367444ee418f129b2c238299f49d3264554Jarkko Poyry scalars = reduce(operator.add, [x[ndx].toFloat().getScalars() for x in comps]) 843c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res.append(Mat(numCols, numRows, scalars)) 853c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return res 863c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 873c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat2(comps): return combineMat(2, 2, comps) 883c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat2x3(comps): return combineMat(2, 3, comps) 893c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat2x4(comps): return combineMat(2, 4, comps) 903c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat3x2(comps): return combineMat(3, 2, comps) 913c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat3(comps): return combineMat(3, 3, comps) 923c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat3x4(comps): return combineMat(3, 4, comps) 933c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat4x2(comps): return combineMat(4, 2, comps) 943c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat4x3(comps): return combineMat(4, 3, comps) 953c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef combineMat4(comps): return combineMat(4, 4, comps) 963c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 973c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# 0 \+ [f*f for f in lst] 983c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# r = 0 \+ [f in lst -> f*f] 993c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# r = 0 \+ lst 1003c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1013c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Templates. 1023c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1033c827367444ee418f129b2c238299f49d3264554Jarkko Poyrys_simpleCaseTemplate = """ 1043c827367444ee418f129b2c238299f49d3264554Jarkko Poyrycase ${{NAME}} 1053c827367444ee418f129b2c238299f49d3264554Jarkko Poyry version 300 es 1063c827367444ee418f129b2c238299f49d3264554Jarkko Poyry values 1073c827367444ee418f129b2c238299f49d3264554Jarkko Poyry { 1083c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${{VALUES}} 1093c827367444ee418f129b2c238299f49d3264554Jarkko Poyry } 1103c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1113c827367444ee418f129b2c238299f49d3264554Jarkko Poyry both "" 1123c827367444ee418f129b2c238299f49d3264554Jarkko Poyry #version 300 es 1133c827367444ee418f129b2c238299f49d3264554Jarkko Poyry precision mediump float; 1143c827367444ee418f129b2c238299f49d3264554Jarkko Poyry precision mediump int; 1153c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1163c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${DECLARATIONS} 1173c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1183c827367444ee418f129b2c238299f49d3264554Jarkko Poyry void main() 1193c827367444ee418f129b2c238299f49d3264554Jarkko Poyry { 1203c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${SETUP} 1213c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${{OP}} 1223c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${OUTPUT} 1233c827367444ee418f129b2c238299f49d3264554Jarkko Poyry } 1243c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "" 1253c827367444ee418f129b2c238299f49d3264554Jarkko Poyryend 1263c827367444ee418f129b2c238299f49d3264554Jarkko Poyry"""[1:] 1273c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1283c827367444ee418f129b2c238299f49d3264554Jarkko Poyrys_simpleIllegalCaseTemplate = """ 1293c827367444ee418f129b2c238299f49d3264554Jarkko Poyrycase ${{NAME}} 1303c827367444ee418f129b2c238299f49d3264554Jarkko Poyry version 300 es 1313c827367444ee418f129b2c238299f49d3264554Jarkko Poyry expect compile_fail 1323c827367444ee418f129b2c238299f49d3264554Jarkko Poyry values {} 1333c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1343c827367444ee418f129b2c238299f49d3264554Jarkko Poyry both "" 1353c827367444ee418f129b2c238299f49d3264554Jarkko Poyry #version 300 es 1363c827367444ee418f129b2c238299f49d3264554Jarkko Poyry precision mediump float; 1373c827367444ee418f129b2c238299f49d3264554Jarkko Poyry precision mediump int; 1383c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1393c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${DECLARATIONS} 1403c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1413c827367444ee418f129b2c238299f49d3264554Jarkko Poyry void main() 1423c827367444ee418f129b2c238299f49d3264554Jarkko Poyry { 1433c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${SETUP} 1443c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${{OP}} 1453c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ${OUTPUT} 1463c827367444ee418f129b2c238299f49d3264554Jarkko Poyry } 1473c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "" 1483c827367444ee418f129b2c238299f49d3264554Jarkko Poyryend 1493c827367444ee418f129b2c238299f49d3264554Jarkko Poyry"""[1:] 1503c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1513c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass SimpleCase(ShaderCase): 1523c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __init__(self, name, inputs, outputs, op): 1533c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.name = name 1543c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.inputs = inputs 1553c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.outputs = outputs 1563c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.op = op 1573c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1583c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __str__(self): 1593c827367444ee418f129b2c238299f49d3264554Jarkko Poyry params = { 1603c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "NAME": self.name, 1613c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "VALUES": genValues(self.inputs, self.outputs), 1623c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "OP": self.op 1633c827367444ee418f129b2c238299f49d3264554Jarkko Poyry } 1643c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return fillTemplate(s_simpleCaseTemplate, params) 1653c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1663c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass ConversionCase(ShaderCase): 1673c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __init__(self, inValues, convFunc): 1683c827367444ee418f129b2c238299f49d3264554Jarkko Poyry outValues = convFunc(inValues) 1693c827367444ee418f129b2c238299f49d3264554Jarkko Poyry inType = inValues[0].typeString() 1703c827367444ee418f129b2c238299f49d3264554Jarkko Poyry outType = outValues[0].typeString() 1713c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.name = "%s_to_%s" % (inType, outType) 1723c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.op = "out0 = %s(in0);" % outType 1733c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.inputs = [("%s in0" % inType, inValues)] 1743c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.outputs = [("%s out0" % outType, outValues)] 1753c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1763c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __str__(self): 1773c827367444ee418f129b2c238299f49d3264554Jarkko Poyry params = { 1783c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "NAME": self.name, 1793c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "VALUES": genValues(self.inputs, self.outputs), 1803c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "OP": self.op 1813c827367444ee418f129b2c238299f49d3264554Jarkko Poyry } 1823c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return fillTemplate(s_simpleCaseTemplate, params) 1833c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1843c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass IllegalConversionCase(ShaderCase): 1853c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __init__(self, inValue, outValue): 1863c827367444ee418f129b2c238299f49d3264554Jarkko Poyry inType = inValue.typeString() 1873c827367444ee418f129b2c238299f49d3264554Jarkko Poyry outType = outValue.typeString() 1883c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.name = "%s_to_%s" % (inType, outType) 1893c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.op = "%s in0 = %s;\n%s out0 = %s(in0);" % (inType, str(inValue), outType, outType) 1903c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.inType = inType 1913c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.outType = outType 1923c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 1933c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __str__(self): 1943c827367444ee418f129b2c238299f49d3264554Jarkko Poyry params = { 1953c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "NAME": self.name, 1963c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "OP": self.op 1973c827367444ee418f129b2c238299f49d3264554Jarkko Poyry } 1983c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return fillTemplate(s_simpleIllegalCaseTemplate, params) 1993c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2003c827367444ee418f129b2c238299f49d3264554Jarkko Poyryclass CombineCase(ShaderCase): 2013c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __init__(self, inComps, combFunc): 2023c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.inComps = inComps 2033c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.outValues = combFunc(inComps) 2043c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.outType = self.outValues[0].typeString() 2053c827367444ee418f129b2c238299f49d3264554Jarkko Poyry inTypes = [values[0].typeString() for values in inComps] 2063c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.name = "%s_to_%s" % ("_".join(inTypes), self.outType) 2073c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.inputs = [("%s in%s" % (comp[0].typeString(), ndx), comp) for (comp, ndx) in zip(inComps, indices)] 2083c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.outputs = [("%s out0" % self.outType, self.outValues)] 2093c827367444ee418f129b2c238299f49d3264554Jarkko Poyry self.op = "out0 = %s(%s);" % (self.outType, ", ".join(["in%d" % x for x in range(len(inComps))])) 2103c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2113c827367444ee418f129b2c238299f49d3264554Jarkko Poyry def __str__(self): 2123c827367444ee418f129b2c238299f49d3264554Jarkko Poyry params = { 2133c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "NAME": self.name, 2143c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "VALUES": genValues(self.inputs, self.outputs), 2153c827367444ee418f129b2c238299f49d3264554Jarkko Poyry "OP": self.op 2163c827367444ee418f129b2c238299f49d3264554Jarkko Poyry } 2173c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return fillTemplate(s_simpleCaseTemplate, params) 2183c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2193c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# CASE DECLARATIONS 2203c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2213c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef toPos (value): 2223c827367444ee418f129b2c238299f49d3264554Jarkko Poyry if isinstance(value, list): 2233c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return [toPos(x) for x in value] 2243c827367444ee418f129b2c238299f49d3264554Jarkko Poyry else: 2253c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return GenMath.abs(value) 2263c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2273c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinFloat = [Scalar(x) for x in [0.0, 1.0, 2.0, 3.5, -0.5, -8.25, -20.125, 36.8125]] 2283c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinInt = [Scalar(x) for x in [0, 1, 2, 5, 8, 11, -12, -66, -192, 255]] 2293c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinUint = [Uint(x) for x in [0, 2, 3, 8, 9, 12, 10, 45, 193, 255]] 2303c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinBool = [Scalar(x) for x in [True, False]] 2313c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2323c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinVec4 = [Vec4(0.0, 0.5, 0.75, 0.825), Vec4(1.0, 1.25, 1.125, 1.75), 2333c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Vec4(-0.5, -2.25, -4.875, 9.0), Vec4(-32.0, 64.0, -51.0, 24.0), 2343c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Vec4(-0.75, -1.0/31.0, 1.0/19.0, 1.0/4.0)] 2353c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinVec3 = toVec3(inVec4) 2363c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinVec2 = toVec2(inVec4) 2373c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinIVec4 = toIVec4(inVec4) 2383c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinIVec3 = toIVec3(inVec4) 2393c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinIVec2 = toIVec2(inVec4) 2403c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinBVec4 = [Vec4(True, False, False, True), Vec4(False, False, False, True), Vec4(False, True, False, False), Vec4(True, True, True, True), Vec4(False, False, False, False)] 2413c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinBVec3 = toBVec3(inBVec4) 2423c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinBVec2 = toBVec2(inBVec4) 2433c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinUVec4 = toUVec4(toPos(inVec4)) 2443c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinUVec3 = toUVec3(toPos(inVec3)) 2453c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinUVec2 = toUVec2(toPos(inVec2)) 2463c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2473c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# \todo [petri] Enable large values when epsilon adapts to the values. 2483c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat4 = [Mat4(1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 1.0), 2493c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Mat4(6.5, 12.5, -0.75, 9.975, 32.0, 1.0/48.0, -8.425, -6.542, 1.0/8.0, 1.0/16.0, 1.0/32.0, 1.0/64.0, -6.725, -0.5, -0.0125, 9.975), 2503c827367444ee418f129b2c238299f49d3264554Jarkko Poyry #Mat4(128.0, 256.0, -512.0, -1024.0, 2048.0, -4096.0, 8192.0, -8192.0, 192.0, -384.0, 768.0, -1536.0, 8192.0, -8192.0, 6144.0, -6144.0) 2513c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ] 2523c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat2 = [Mat2(1.0, 0.0, 0.0, 1.0), 2533c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Mat2(6.5, 12.5, -0.75, 9.975), 2543c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Mat2(6.5, 12.5, -0.75, 9.975), 2553c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Mat2(8.0, 16.0, -24.0, -16.0), 2563c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Mat2(1.0/8.0, 1.0/16.0, 1.0/32.0, 1.0/64.0), 2573c827367444ee418f129b2c238299f49d3264554Jarkko Poyry Mat2(-18.725, -0.5, -0.0125, 19.975), 2583c827367444ee418f129b2c238299f49d3264554Jarkko Poyry #Mat2(128.0, -4096.0, 192.0, -1536.0), 2593c827367444ee418f129b2c238299f49d3264554Jarkko Poyry #Mat2(-1536.0, 8192.0, 6144.0, -6144.0) 2603c827367444ee418f129b2c238299f49d3264554Jarkko Poyry ] 2613c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2623c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat4x3 = toMat4x3(inMat4) 2633c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat4x2 = toMat4x2(inMat4) 2643c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat3x4 = toMat3x4(inMat4) 2653c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat3 = toMat3(inMat4) 2663c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat3x2 = toMat3x2(inMat4) 2673c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat2x4 = toMat2x4(inMat4) 2683c827367444ee418f129b2c238299f49d3264554Jarkko PoyryinMat2x3 = toMat2x3(inMat4) 2693c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2703c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef genConversionCases(inValueList, convFuncList): 2713c827367444ee418f129b2c238299f49d3264554Jarkko Poyry combinations = list(itertools.product(inValueList, convFuncList)) 2723c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return [ConversionCase(inValues, convFunc) for (inValues, convFunc) in combinations] 2733c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2743c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef genIllegalConversionCases(inValueList, outValueList): 2753c827367444ee418f129b2c238299f49d3264554Jarkko Poyry inValues = [x[0] for x in inValueList] 2763c827367444ee418f129b2c238299f49d3264554Jarkko Poyry outValues = [x[0] for x in outValueList] 2773c827367444ee418f129b2c238299f49d3264554Jarkko Poyry combinations = list(itertools.product(inValues, outValues)) 2783c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return [IllegalConversionCase(inVal, outVal) for (inVal, outVal) in combinations] 2793c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2803c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef shuffleSubLists(outer): 2813c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return [shuffled(inner) for inner in outer] 2823c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2833c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Generate all combinations of CombineCases. 2843c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# inTupleList a list of tuples of value-lists 2853c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# combFuncList a list of comb* functions to combine 2863c827367444ee418f129b2c238299f49d3264554Jarkko Poyrydef genComponentCases(inCompLists, combFuncList): 2873c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res = [] 2883c827367444ee418f129b2c238299f49d3264554Jarkko Poyry for comps in inCompLists: 2893c827367444ee418f129b2c238299f49d3264554Jarkko Poyry maxLen = reduce(max, [len(values) for values in comps]) 2903c827367444ee418f129b2c238299f49d3264554Jarkko Poyry comps = [repeatToLength(values, maxLen) for values in comps] 2913c827367444ee418f129b2c238299f49d3264554Jarkko Poyry comps = [shuffled(values) for values in comps] 2923c827367444ee418f129b2c238299f49d3264554Jarkko Poyry for combFunc in combFuncList: 2933c827367444ee418f129b2c238299f49d3264554Jarkko Poyry res += [CombineCase(comps, combFunc)] 2943c827367444ee418f129b2c238299f49d3264554Jarkko Poyry return res 2953c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2963c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases = [] 2973c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 2983c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Scalar-to-scalar conversions. 2993c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("scalar_to_scalar", "Scalar to Scalar Conversions", 3003c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inFloat, inInt, inUint, inBool], [toFloat, toInt, toBool]) +\ 3013c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([toPos(inFloat), toPos(inInt), inUint, inBool], [toUint]))) 3023c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3033c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Scalar-to-vector conversions. 3043c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("scalar_to_vector", "Scalar to Vector Conversions", 3053c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inFloat, inInt, inUint, inBool], [toVec2, toVec3, toVec4, toIVec2, toIVec3, toIVec4, toBVec2, toBVec3, toBVec4]) +\ 3063c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([toPos(inFloat), toPos(inInt), inUint, inBool], [toUVec2, toUVec3, toUVec4]))) 3073c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3083c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Vector-to-scalar conversions. 3093c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("vector_to_scalar", "Vector to Scalar Conversions", 3103c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inVec2, inVec3, inVec4, inIVec2, inIVec3, inIVec4, inUVec2, inUVec3, inUVec4, inBVec2, inBVec3, inBVec4], [toFloat, toInt, toBool]) +\ 3113c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([toPos(inVec2), toPos(inVec3), toPos(inVec4), toPos(inIVec2), toPos(inIVec3), toPos(inIVec4), inUVec2, inUVec3, inUVec4, inBVec2, inBVec3, inBVec4], [toUint]))) 3123c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3133c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Illegal vector-to-vector conversions (to longer vec). 3143c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("vector_illegal", "Illegal Vector Conversions", 3153c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genIllegalConversionCases([inVec2, inIVec2, inUVec2, inBVec2], [inVec3, inIVec3, inUVec3, inBVec3, inVec4, inIVec4, inUVec4, inBVec4]) +\ 3163c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genIllegalConversionCases([inVec3, inIVec3, inUVec3, inBVec3], [inVec4, inIVec4, inUVec4, inBVec4]))) 3173c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3183c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Vector-to-vector conversions (type conversions, downcasts). 3193c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("vector_to_vector", "Vector to Vector Conversions", 3203c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inVec4, inIVec4, inUVec4, inBVec4], [toVec4, toVec3, toVec2, toIVec4, toIVec3, toIVec2, toBVec4, toBVec3, toBVec2]) +\ 3213c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([toPos(inVec4), toPos(inIVec4), inUVec4, inBVec4], [toUVec4, toUVec3, toUVec2]) +\ 3223c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inVec3, inIVec3, inUVec3, inBVec3], [toVec3, toVec2, toIVec3, toIVec2, toBVec3, toBVec2]) +\ 3233c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([toPos(inVec3), toPos(inIVec3), inUVec3, inBVec3], [toUVec3, toUVec2]) +\ 3243c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inVec2, inIVec2, inUVec2, inBVec2], [toVec2, toIVec2, toBVec2]) +\ 3253c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([toPos(inVec2), toPos(inIVec2), inUVec2, inBVec2], [toUVec2]))) 3263c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3273c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Scalar-to-matrix. 3283c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("scalar_to_matrix", "Scalar to Matrix Conversions", 3293c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inFloat, inInt, inUint, inBool], [toMat4, toMat4x3, toMat4x2, toMat3x4, toMat3, toMat3x2, toMat2x4, toMat2x3, toMat2]))) 3303c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3313c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Vector-to-matrix. 3323c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#allConversionCases += genConversionCases([inVec4, inIVec4, inBVec4], [toMat4]) 3333c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#allConversionCases += genConversionCases([inVec3, inIVec3, inBVec3], [toMat3]) 3343c827367444ee418f129b2c238299f49d3264554Jarkko Poyry#allConversionCases += genConversionCases([inVec2, inIVec2, inBVec2], [toMat2]) 3353c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3363c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Matrix-to-matrix. 3373c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("matrix_to_matrix", "Matrix to Matrix Conversions", 3383c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genConversionCases([inMat4, inMat4x3, inMat4x2, inMat3x4, inMat3, inMat3x2, inMat2x4, inMat2x3, inMat2], [toMat4, toMat4x3, toMat4x2, toMat3x4, toMat3, toMat3x2, toMat2x4, toMat2x3, toMat2]))) 3393c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3403c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Vector-from-components, matrix-from-components. 3413c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin2Comp = [[inFloat, inFloat], [inInt, inInt], [inUint, inUint], [inBool, inBool], [inFloat, inInt], [inFloat, inBool], [inInt, inBool], [inInt, inUint], [inUint, inFloat]] 3423c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin3Comp = [[inFloat, inFloat, inFloat], [inInt, inInt, inInt], [inUint, inUint, inUint], [inBool, inBool, inBool], [inBool, inFloat, inInt], [inVec2, inBool], [inBVec2, inFloat], [inBVec2, inInt], [inBool, inIVec2], [inFloat, inUVec2]] 3433c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin4Comp = [[inVec2, inVec2], [inBVec2, inBVec2], [inFloat, inFloat, inFloat, inFloat], [inInt, inInt, inInt, inInt], [inUint, inUint, inUint, inUint], [inBool, inBool, inBool, inBool], [inBool, inFloat, inInt, inBool], [inVec2, inIVec2], [inVec2, inBVec2], [inBVec3, inFloat], [inVec3, inFloat], [inInt, inIVec2, inInt], [inBool, inFloat, inIVec2], [inFloat, inUVec3], [inInt, inUVec2, inBool]] 3443c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin6Comp = [[inVec3, inVec3], [inBVec3, inBVec3], [inFloat, inFloat, inFloat, inFloat, inFloat, inFloat], [inInt, inInt, inInt, inInt, inInt, inInt], [inBool, inBool, inBool, inBool, inBool, inBool], [inBool, inFloat, inInt, inBool, inFloat, inInt], [inVec3, inIVec3], [inVec2, inBVec4], [inBVec3, inFloat, inIVec2], [inVec3, inFloat, inBVec2]] 3453c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin8Comp = [[inVec3, inVec3, inVec2], [inIVec3, inIVec3, inIVec2], [inVec2, inIVec2, inFloat, inFloat, inInt, inBool], [inBool, inFloat, inInt, inVec2, inBool, inBVec2], [inBool, inBVec2, inInt, inVec4], [inFloat, inBVec4, inIVec2, inBool]] 3463c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin9Comp = [[inVec3, inVec3, inVec3], [inIVec3, inIVec3, inIVec3], [inVec2, inIVec2, inFloat, inFloat, inInt, inBool, inBool], [inBool, inFloat, inInt, inVec2, inBool, inBVec2, inFloat], [inBool, inBVec2, inInt, inVec4, inBool], [inFloat, inBVec4, inIVec2, inBool, inBool]] 3473c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin12Comp = [[inVec4, inVec4, inVec4], [inIVec4, inIVec4, inIVec4], [inVec2, inIVec2, inFloat, inFloat, inFloat, inInt, inInt, inBool, inBool, inBool], [inBool, inFloat, inInt, inVec3, inBool, inBVec3, inFloat, inBool], [inBool, inBVec4, inInt, inVec4, inBool, inFloat], [inFloat, inBVec4, inIVec4, inBool, inBool, inInt]] 3483c827367444ee418f129b2c238299f49d3264554Jarkko Poyryin16Comp = [[inVec4, inVec4, inVec4, inVec4], [inIVec4, inIVec4, inIVec4, inIVec4], [inBVec4, inBVec4, inBVec4, inBVec4], [inFloat, inIVec3, inBVec3, inVec4, inIVec2, inFloat, inVec2]] 3493c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3503c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("vector_combine", "Vector Combine Constructors", 3513c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in4Comp, [combineVec, combineIVec, combineBVec]) +\ 3523c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(toPos(in4Comp), [combineUVec]) +\ 3533c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in3Comp, [combineVec, combineIVec, combineBVec]) +\ 3543c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(toPos(in3Comp), [combineUVec]) +\ 3553c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in2Comp, [combineVec, combineIVec, combineBVec]) +\ 3563c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(toPos(in2Comp), [combineUVec]))) 3573c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3583c827367444ee418f129b2c238299f49d3264554Jarkko PoyryallConversionCases.append(CaseGroup("matrix_combine", "Matrix Combine Constructors", 3593c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in4Comp, [combineMat2]) +\ 3603c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in6Comp, [combineMat2x3]) +\ 3613c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in8Comp, [combineMat2x4]) +\ 3623c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in6Comp, [combineMat3x2]) +\ 3633c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in9Comp, [combineMat3]) +\ 3643c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in12Comp, [combineMat3x4]) +\ 3653c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in8Comp, [combineMat4x2]) +\ 3663c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in12Comp, [combineMat4x3]) +\ 3673c827367444ee418f129b2c238299f49d3264554Jarkko Poyry genComponentCases(in16Comp, [combineMat4]) 3683c827367444ee418f129b2c238299f49d3264554Jarkko Poyry )) 3693c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3703c827367444ee418f129b2c238299f49d3264554Jarkko Poyry# Main program. 3713c827367444ee418f129b2c238299f49d3264554Jarkko Poyry 3723c827367444ee418f129b2c238299f49d3264554Jarkko Poyryif __name__ == "__main__": 3733c827367444ee418f129b2c238299f49d3264554Jarkko Poyry print "Generating shader case files." 3743c827367444ee418f129b2c238299f49d3264554Jarkko Poyry writeAllCases("conversions.test", allConversionCases) 375