Add mat3 instances parameters.

This commit is contained in:
Roz K 2022-12-26 14:05:06 +01:00
parent 618e05bcaa
commit 6b33f8285c
Signed by: roz
GPG Key ID: 51FBF4E483E1C822
4 changed files with 239 additions and 59 deletions

View File

@ -47,6 +47,11 @@ def _float_addr(x):
class vec3(ctypes.Structure): class vec3(ctypes.Structure):
_fields_ = ('x', ctypes.c_float), ('y', ctypes.c_float), ('z', ctypes.c_float) _fields_ = ('x', ctypes.c_float), ('y', ctypes.c_float), ('z', ctypes.c_float)
def __iter__(self):
yield self.x
yield self.y
yield self.z
def set(self, x, y, z): def set(self, x, y, z):
self.x = x self.x = x
self.y = y self.y = y
@ -55,6 +60,12 @@ class vec3(ctypes.Structure):
class vec4(ctypes.Structure): class vec4(ctypes.Structure):
_fields_ = ('x', ctypes.c_float), ('y', ctypes.c_float), ('z', ctypes.c_float), ('w', ctypes.c_float) _fields_ = ('x', ctypes.c_float), ('y', ctypes.c_float), ('z', ctypes.c_float), ('w', ctypes.c_float)
def __iter__(self):
yield self.x
yield self.y
yield self.z
yield self.w
def set(self, x, y, z, w): def set(self, x, y, z, w):
self.x = x self.x = x
self.y = y self.y = y
@ -62,10 +73,59 @@ class vec4(ctypes.Structure):
self.w = w self.w = w
class mat3(ctypes.Structure): class mat3(ctypes.Structure):
_fields_ = ('x', vec3), ('y', vec3), ('z', vec3) _fields_ = ('right', vec3), ('forward', vec3), ('up', vec3)
def __iter__(self):
yield self.right
yield self.forward
yield self.up
def set(self, right, forward, up):
self.right = right
self.forward = forward
self.up = up
class mat4(ctypes.Structure): class mat4(ctypes.Structure):
_fields_ = ('x', vec4), ('y', vec4), ('z', vec4), ('w', vec4) _fields_ = ('right', vec4), ('forward', vec4), ('up', vec4), ('origin', vec4)
def __iter__(self):
yield self.right
yield self.forward
yield self.up
yield self.origin
def set(self, right, forward, up, origin):
self.right = right
self.forward = forward
self.up = up
self.origin = origin
def set_mat3_vec3(self, axes, origin):
self.right.set(*axes.right, 0.0)
self.forward.set(*axes.forward, 0.0)
self.up.set(*axes.up, 0.0)
self.origin.set(*origin, 1.0)
def set_axes(self, right, forward, up):
self.right = right
self.forward = forward
self.up = up
def set_axes_mat3(self, axes):
self.right.set(*axes.right, 0.0)
self.forward.set(*axes.forward, 0.0)
self.up.set(*axes.up, 0.0)
def set_origin(self, origin):
self.origin = origin
def set_origin_vec3(self, origin):
self.origin.set(*origin, 1.0)
assert ctypes.sizeof(vec3) == ctypes.sizeof(ctypes.c_float) * 3
assert ctypes.sizeof(vec4) == ctypes.sizeof(ctypes.c_float) * 4
assert ctypes.sizeof(mat3) == ctypes.sizeof(vec3) * 3
assert ctypes.sizeof(mat4) == ctypes.sizeof(vec4) * 4
_vec3_p = ctypes.POINTER(vec3) _vec3_p = ctypes.POINTER(vec3)
_vec4_p = ctypes.POINTER(vec4) _vec4_p = ctypes.POINTER(vec4)
@ -75,6 +135,23 @@ _mat4_p = ctypes.POINTER(mat4)
vec3_right = vec3(1.0, 0.0, 0.0) vec3_right = vec3(1.0, 0.0, 0.0)
vec3_forward = vec3(0.0, 1.0, 0.0) vec3_forward = vec3(0.0, 1.0, 0.0)
vec3_up = vec3(0.0, 0.0, 1.0) vec3_up = vec3(0.0, 0.0, 1.0)
vec3_origin = vec3(0.0, 0.0, 0.0)
vec4_right = vec4(1.0, 0.0, 0.0, 0.0)
vec4_forward = vec4(0.0, 1.0, 0.0, 0.0)
vec4_up = vec4(0.0, 0.0, 1.0, 0.0)
vec4_origin = vec4(0.0, 0.0, 0.0, 1.0)
mat3_identity = mat3(
vec3_right,
vec3_forward,
vec3_up)
mat4_identity = mat4(
vec4_right,
vec4_forward,
vec4_up,
vec4_origin)
vec3_rotate = _engine.rk_vec3_rotate vec3_rotate = _engine.rk_vec3_rotate
vec3_rotate.argtypes = ( vec3_rotate.argtypes = (
@ -185,6 +262,8 @@ def vertex_format(*format):
PARAM_FORMAT_VEC3_FLOAT = 1 PARAM_FORMAT_VEC3_FLOAT = 1
PARAM_FORMAT_VEC3_SHORT = 2 PARAM_FORMAT_VEC3_SHORT = 2
PARAM_FORMAT_VEC3_INT10 = 3 PARAM_FORMAT_VEC3_INT10 = 3
PARAM_FORMAT_MAT3_FLOAT = 4
PARAM_FORMAT_MAT3_INT10 = 5
PARAM_FORMAT_NORMALIZE = _flag(7) PARAM_FORMAT_NORMALIZE = _flag(7)
def params_format(*format): def params_format(*format):

View File

@ -55,7 +55,9 @@ enum : rk_ubyte { RK_VERTEX_FORMAT_MASK = RK_VERTEX_FORMAT_NORMALIZE - 1 };
enum rk_param_format : rk_ubyte { enum rk_param_format : rk_ubyte {
RK_PARAM_FORMAT_VEC3_FLOAT = 1, RK_PARAM_FORMAT_VEC3_FLOAT = 1,
RK_PARAM_FORMAT_VEC3_SHORT = 2, RK_PARAM_FORMAT_VEC3_SHORT = 2,
RK_PARAM_FORMAT_VEC3_INT10 = 3 RK_PARAM_FORMAT_VEC3_INT10 = 3,
RK_PARAM_FORMAT_MAT3_FLOAT = 4,
RK_PARAM_FORMAT_MAT3_INT10 = 5
}; };
enum : rk_ubyte { RK_PARAM_FORMAT_NORMALIZE = RK_FLAG(7) }; enum : rk_ubyte { RK_PARAM_FORMAT_NORMALIZE = RK_FLAG(7) };

View File

@ -290,18 +290,18 @@ rk_vertices_t rk_create_vertices(
unsigned offset = 0; unsigned offset = 0;
for (rk_vertex_format const * f = format; *f; ++f, ++vertices->layout) { for (rk_vertex_format const * f = format; *f; ++f, ++vertices->layout) {
glEnableVertexAttribArray(vertices->layout); glEnableVertexAttribArray(vertices->layout);
GLboolean const normalize = (*f & RK_VERTEX_FORMAT_NORMALIZE) != 0; GLboolean const norm = (*f & RK_VERTEX_FORMAT_NORMALIZE) != 0;
switch (*f & RK_VERTEX_FORMAT_MASK) { switch (*f & RK_VERTEX_FORMAT_MASK) {
case RK_VERTEX_FORMAT_VEC3_FLOAT: case RK_VERTEX_FORMAT_VEC3_FLOAT:
glVertexAttribFormat(vertices->layout, 3, GL_FLOAT, normalize, offset); glVertexAttribFormat(vertices->layout, 3, GL_FLOAT, norm, offset);
offset += sizeof(rk_vec3); offset += sizeof(rk_vec3);
break; break;
case RK_VERTEX_FORMAT_VEC3_INT10: case RK_VERTEX_FORMAT_VEC3_INT10:
glVertexAttribFormat(vertices->layout, 4, GL_INT_2_10_10_10_REV, normalize, offset); glVertexAttribFormat(vertices->layout, 4, GL_INT_2_10_10_10_REV, norm, offset);
offset += sizeof(rk_int); offset += sizeof(rk_int);
break; break;
case RK_VERTEX_FORMAT_VEC3_UINT10: case RK_VERTEX_FORMAT_VEC3_UINT10:
glVertexAttribFormat(vertices->layout, 4, GL_UNSIGNED_INT_2_10_10_10_REV, normalize, offset); glVertexAttribFormat(vertices->layout, 4, GL_UNSIGNED_INT_2_10_10_10_REV, norm, offset);
offset += sizeof(rk_uint); offset += sizeof(rk_uint);
break; break;
} }
@ -315,35 +315,77 @@ rk_vertices_t rk_create_vertices(
return vertices; return vertices;
} }
static void rk_pack_vec3( static void rk_pack_vec3_float(
rk_pack_dst dst, rk_pack_dst const dst,
rk_pack_src src) { rk_pack_src const src) {
*dst.vec3_ptr = *src.vec3_ptr; *dst.vec3_float = *src.vec3_float;
} }
static void rk_pack_vec3s( static void rk_pack_vec3_short(
rk_pack_dst dst, rk_pack_dst const dst,
rk_pack_src src) { rk_pack_src const src) {
dst.vec3s_ptr->x = static_cast<rk_short>(src.vec3_ptr->x); dst.vec3_short->x = static_cast<rk_short>(src.vec3_float->x);
dst.vec3s_ptr->y = static_cast<rk_short>(src.vec3_ptr->y); dst.vec3_short->y = static_cast<rk_short>(src.vec3_float->y);
dst.vec3s_ptr->z = static_cast<rk_short>(src.vec3_ptr->z); dst.vec3_short->z = static_cast<rk_short>(src.vec3_float->z);
} }
static void rk_pack_vec3s_norm( static void rk_pack_vec3_short_norm(
rk_pack_dst dst, rk_pack_dst const dst,
rk_pack_src src) { rk_pack_src const src) {
#define _convert(s) (static_cast<rk_short>((s) * ((s) < 0.f ? 32768.f : 32767.f))) #define _convert(s) (static_cast<rk_short>((s) * ((s) < 0.f ? 32768.f : 32767.f)))
dst.vec3s_ptr->x = _convert(src.vec3_ptr->x); dst.vec3_short->x = _convert(src.vec3_float->x);
dst.vec3s_ptr->y = _convert(src.vec3_ptr->y); dst.vec3_short->y = _convert(src.vec3_float->y);
dst.vec3s_ptr->z = _convert(src.vec3_ptr->z); dst.vec3_short->z = _convert(src.vec3_float->z);
#undef _convert #undef _convert
} }
static void rk_pack_vec3_int10( static void rk_pack_vec3_int10(
rk_pack_dst dst, rk_pack_dst const dst,
rk_pack_src src) { rk_pack_src const src) {
#define _convert(s) (static_cast<rk_int>((s)) & 1023)
*dst.vec3_int10 =
_convert(src.vec3_float->x) | (_convert(src.vec3_float->y) << 10) | (_convert(src.vec3_float->z) << 20);
#undef _convert
}
static void rk_pack_vec3_int10_norm(
rk_pack_dst const dst,
rk_pack_src const src) {
#define _convert(s) (static_cast<rk_int>((s) * ((s) < 0.f ? 512.f : 511.f)) & 1023) #define _convert(s) (static_cast<rk_int>((s) * ((s) < 0.f ? 512.f : 511.f)) & 1023)
*dst.int_ptr = _convert(src.vec3_ptr->x) | (_convert(src.vec3_ptr->y) << 10) | (_convert(src.vec3_ptr->z) << 20); *dst.vec3_int10 =
_convert(src.vec3_float->x) | (_convert(src.vec3_float->y) << 10) | (_convert(src.vec3_float->z) << 20);
#undef _convert
}
static void rk_pack_mat3_float(
rk_pack_dst const dst,
rk_pack_src const src) {
*dst.mat3_float = *src.mat3_float;
}
static void rk_pack_mat3_int10(
rk_pack_dst const dst,
rk_pack_src const src) {
#define _convert(s) (static_cast<rk_int>((s)) & 1023)
dst.mat3_int10->x =
_convert(src.mat3_float->x.x) | (_convert(src.mat3_float->x.y) << 10) | (_convert(src.mat3_float->x.z) << 20);
dst.mat3_int10->y =
_convert(src.mat3_float->y.x) | (_convert(src.mat3_float->y.y) << 10) | (_convert(src.mat3_float->y.z) << 20);
dst.mat3_int10->z =
_convert(src.mat3_float->z.x) | (_convert(src.mat3_float->z.y) << 10) | (_convert(src.mat3_float->z.z) << 20);
#undef _convert
}
static void rk_pack_mat3_int10_norm(
rk_pack_dst const dst,
rk_pack_src const src) {
#define _convert(s) (static_cast<rk_int>((s) * ((s) < 0.f ? 512.f : 511.f)) & 1023)
dst.mat3_int10->x =
_convert(src.mat3_float->x.x) | (_convert(src.mat3_float->x.y) << 10) | (_convert(src.mat3_float->x.z) << 20);
dst.mat3_int10->y =
_convert(src.mat3_float->y.x) | (_convert(src.mat3_float->y.y) << 10) | (_convert(src.mat3_float->y.z) << 20);
dst.mat3_int10->z =
_convert(src.mat3_float->z.x) | (_convert(src.mat3_float->z.y) << 10) | (_convert(src.mat3_float->z.z) << 20);
#undef _convert #undef _convert
} }
@ -363,16 +405,24 @@ rk_batch_t rk_create_batch(
for (rk_param_format const * f = params_format; *f; ++f, ++nparams) { for (rk_param_format const * f = params_format; *f; ++f, ++nparams) {
switch (*f & RK_PARAM_FORMAT_MASK) { switch (*f & RK_PARAM_FORMAT_MASK) {
case RK_PARAM_FORMAT_VEC3_FLOAT: case RK_PARAM_FORMAT_VEC3_FLOAT:
params_size += sizeof(rk_vec3); params_size += sizeof(rk_vec3_float);
packed_size += sizeof(rk_vec3); packed_size += sizeof(rk_vec3_float);
break; break;
case RK_PARAM_FORMAT_VEC3_SHORT: case RK_PARAM_FORMAT_VEC3_SHORT:
params_size += sizeof(rk_vec3); params_size += sizeof(rk_vec3_float);
packed_size += sizeof(rk_vec3s); packed_size += sizeof(rk_vec3_short);
break; break;
case RK_PARAM_FORMAT_VEC3_INT10: case RK_PARAM_FORMAT_VEC3_INT10:
params_size += sizeof(rk_vec3); params_size += sizeof(rk_vec3_float);
packed_size += sizeof(rk_int); packed_size += sizeof(rk_vec3_int10);
break;
case RK_PARAM_FORMAT_MAT3_FLOAT:
params_size += sizeof(rk_mat3_float);
packed_size += sizeof(rk_mat3_float);
break;
case RK_PARAM_FORMAT_MAT3_INT10:
params_size += sizeof(rk_mat3_float);
packed_size += sizeof(rk_mat3_int10);
break; break;
default: default:
rk_printf("rk_create_batch(): invalid param format."); rk_printf("rk_create_batch(): invalid param format.");
@ -391,39 +441,65 @@ rk_batch_t rk_create_batch(
batch->packers = new rk_packer[nparams]; batch->packers = new rk_packer[nparams];
batch->params = new rk_ubyte[max_size * packed_size]; batch->params = new rk_ubyte[max_size * packed_size];
glGenBuffers(1, &batch->params_buffer); glGenBuffers(1, &batch->params_buffer);
unsigned layout = vertices->layout;
rk_packer * packer = batch->packers; rk_packer * packer = batch->packers;
unsigned layout = vertices->layout;
unsigned offset = 0; unsigned offset = 0;
glBindVertexArray(vertices->array); glBindVertexArray(vertices->array);
for (rk_param_format const * f = params_format; *f; ++f, ++layout, ++packer) { for (rk_param_format const * f = params_format; *f; ++f, ++packer) {
GLboolean const normalize = (*f & RK_PARAM_FORMAT_NORMALIZE) != 0; GLboolean const norm = (*f & RK_PARAM_FORMAT_NORMALIZE) != 0;
glEnableVertexAttribArray(layout);
switch (*f & RK_PARAM_FORMAT_MASK) { switch (*f & RK_PARAM_FORMAT_MASK) {
case RK_PARAM_FORMAT_VEC3_FLOAT: case RK_PARAM_FORMAT_VEC3_FLOAT:
glVertexAttribFormat(layout, 3, GL_FLOAT, normalize, offset); glVertexAttribFormat(layout++, 3, GL_FLOAT, GL_FALSE, offset);
packer->pack = rk_pack_vec3; packer->pack = rk_pack_vec3_float;
packer->dst_incr = sizeof(rk_vec3); packer->src_incr = sizeof(rk_vec3_float);
packer->src_incr = sizeof(rk_vec3); packer->dst_incr = sizeof(rk_vec3_float);
break; break;
case RK_PARAM_FORMAT_VEC3_SHORT: case RK_PARAM_FORMAT_VEC3_SHORT:
glVertexAttribFormat(layout, 3, GL_SHORT, normalize, offset); glVertexAttribFormat(layout++, 3, GL_SHORT, norm, offset);
if (normalize) { if (norm) {
packer->pack = rk_pack_vec3s_norm; packer->pack = rk_pack_vec3_short_norm;
} else { } else {
packer->pack = rk_pack_vec3s; packer->pack = rk_pack_vec3_short;
} }
packer->dst_incr = sizeof(rk_vec3s); packer->src_incr = sizeof(rk_vec3_float);
packer->src_incr = sizeof(rk_vec3); packer->dst_incr = sizeof(rk_vec3_short);
break; break;
case RK_PARAM_FORMAT_VEC3_INT10: case RK_PARAM_FORMAT_VEC3_INT10:
glVertexAttribFormat(layout, 4, GL_INT_2_10_10_10_REV, normalize, offset); glVertexAttribFormat(layout++, 4, GL_INT_2_10_10_10_REV, norm, offset);
packer->pack = rk_pack_vec3_int10; if (norm) {
packer->dst_incr = sizeof(rk_int); packer->pack = rk_pack_vec3_int10_norm;
packer->src_incr = sizeof(rk_vec3); } else {
packer->pack = rk_pack_vec3_int10;
}
packer->src_incr = sizeof(rk_vec3_float);
packer->dst_incr = sizeof(rk_vec3_int10);
break;
case RK_PARAM_FORMAT_MAT3_FLOAT:
glVertexAttribFormat(layout++, 3, GL_FLOAT, GL_FALSE, offset + offsetof(rk_mat3_float, x));
glVertexAttribFormat(layout++, 3, GL_FLOAT, GL_FALSE, offset + offsetof(rk_mat3_float, y));
glVertexAttribFormat(layout++, 3, GL_FLOAT, GL_FALSE, offset + offsetof(rk_mat3_float, z));
packer->pack = rk_pack_mat3_float;
packer->src_incr = sizeof(rk_mat3_float);
packer->dst_incr = sizeof(rk_mat3_float);
break;
case RK_PARAM_FORMAT_MAT3_INT10:
glVertexAttribFormat(layout++, 4, GL_INT_2_10_10_10_REV, norm, offset + offsetof(rk_mat3_int10, x));
glVertexAttribFormat(layout++, 4, GL_INT_2_10_10_10_REV, norm, offset + offsetof(rk_mat3_int10, y));
glVertexAttribFormat(layout++, 4, GL_INT_2_10_10_10_REV, norm, offset + offsetof(rk_mat3_int10, z));
if (norm) {
packer->pack = rk_pack_mat3_int10_norm;
} else {
packer->pack = rk_pack_mat3_int10;
}
packer->src_incr = sizeof(rk_mat3_float);
packer->dst_incr = sizeof(rk_mat3_int10);
break; break;
} }
offset += packer->dst_incr; offset += packer->dst_incr;
glVertexAttribBinding(layout, RK_PARAMS_BINDING); }
for (unsigned attrib = vertices->layout; attrib < layout; ++attrib) {
glEnableVertexAttribArray(attrib);
glVertexAttribBinding(attrib, RK_PARAMS_BINDING);
} }
glVertexBindingDivisor(RK_PARAMS_BINDING, 1); glVertexBindingDivisor(RK_PARAMS_BINDING, 1);
glBindVertexBuffer(RK_PARAMS_BINDING, batch->params_buffer, 0, batch->packed_size); glBindVertexBuffer(RK_PARAMS_BINDING, batch->params_buffer, 0, batch->packed_size);
@ -580,8 +656,8 @@ static void rk_batch_pack(
rk_pack_src src(&params[batch.params_size * (*index)]); rk_pack_src src(&params[batch.params_size * (*index)]);
for (rk_packer const * packer = batch.packers; packer < last_packer; ++packer) { for (rk_packer const * packer = batch.packers; packer < last_packer; ++packer) {
packer->pack(dst, src); packer->pack(dst, src);
dst.ptr += packer->dst_incr;
src.ptr += packer->src_incr; src.ptr += packer->src_incr;
dst.ptr += packer->dst_incr;
} }
} }
} }

View File

@ -51,16 +51,37 @@ struct rk_vertices {
GLuint indices; GLuint indices;
}; };
struct rk_vec3s { struct rk_vec3_float {
float x;
float y;
float z;
};
struct rk_vec3_short {
rk_short x; rk_short x;
rk_short y; rk_short y;
rk_short z; rk_short z;
rk_short pad; rk_short pad;
}; };
typedef rk_int rk_vec3_int10;
struct rk_mat3_float {
rk_vec3_float x;
rk_vec3_float y;
rk_vec3_float z;
};
struct rk_mat3_int10 {
rk_vec3_int10 x;
rk_vec3_int10 y;
rk_vec3_int10 z;
};
union rk_pack_src { union rk_pack_src {
rk_ubyte const * __restrict ptr; rk_ubyte const * __restrict ptr;
rk_vec3 const * __restrict vec3_ptr; rk_vec3_float const * __restrict vec3_float;
rk_mat3_float const * __restrict mat3_float;
inline rk_pack_src() {} inline rk_pack_src() {}
inline rk_pack_src(rk_ubyte const * const __restrict src) : ptr(src) {} inline rk_pack_src(rk_ubyte const * const __restrict src) : ptr(src) {}
@ -68,20 +89,22 @@ union rk_pack_src {
union rk_pack_dst { union rk_pack_dst {
rk_ubyte * __restrict ptr; rk_ubyte * __restrict ptr;
rk_vec3 * __restrict vec3_ptr; rk_vec3_float * __restrict vec3_float;
rk_vec3s * __restrict vec3s_ptr; rk_vec3_short * __restrict vec3_short;
rk_int * __restrict int_ptr; rk_vec3_int10 * __restrict vec3_int10;
rk_mat3_float * __restrict mat3_float;
rk_mat3_int10 * __restrict mat3_int10;
inline rk_pack_dst() {} inline rk_pack_dst() {}
inline rk_pack_dst(rk_ubyte * const __restrict dst) : ptr(dst) {} inline rk_pack_dst(rk_ubyte * const __restrict dst) : ptr(dst) {}
}; };
typedef void (*rk_packer_fn)(rk_pack_dst, rk_pack_src); typedef void (*rk_packer_fn)(rk_pack_dst const, rk_pack_src const);
struct rk_packer { struct rk_packer {
rk_packer_fn pack; rk_packer_fn pack;
unsigned dst_incr;
unsigned src_incr; unsigned src_incr;
unsigned dst_incr;
}; };
struct rk_command { struct rk_command {