krzosa/software_rasterizer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change file naming scheme

Browse Source
This commit is contained in:
Krzosa Karol
2022-08-05 23:39:40 +02:00
parent bb3723802a
commit fba01c6573
9 changed files with 3 additions and 1845 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
assets.cpp
View File

@@ -1,4 +1,4 @@
#include "obj_dump.cpp"
#include "sf_obj_dump.cpp"
function void
asset_log(Log_Kind kind, String string, char *file, int line){

2
build.bat
View File

@@ -1,5 +1,5 @@
@echo off
pushd %~dp0
clang main.cpp -O2 -mfma -mavx2 -Wall -Wno-unused-function -Wno-missing-braces -fno-exceptions -fdiagnostics-absolute-paths -Wno-deprecated-declarations -g -o main.exe -Wl,user32.lib
clang sf_main.cpp -O2 -mfma -mavx2 -Wall -Wno-unused-function -Wno-missing-braces -fno-exceptions -fdiagnostics-absolute-paths -Wno-deprecated-declarations -g -o main.exe -Wl,user32.lib
popd

927
main.cpp
View File

@@ -1,927 +0,0 @@
#include "obj_dump.cpp"
#include "vec.cpp"
#include "work_queue.cpp"
#define PROFILE_SCOPE(x)
#define MULTITHREADING 1
struct Vertex {
Vec3 pos;
Vec2 tex;
Vec3 norm;
};
struct Render_Command{
Bitmap *src;
Vec4 p0, p1, p2;
Vec2 tex0, tex1, tex2;
};
struct Render {
Mat4 camera;
Mat4 projection;
Mat4 transform;
Vec3 camera_pos;
Vec3 camera_direction;
Vec3 camera_forward_velocity;
Vec2 camera_yaw;
Vec3 camera_target;
Bitmap img;
B32 plot_ready;
Bitmap plot;
Bitmap screen320;
F32 *depth320;
WorkQueue work_queue;
Array_List<Render_Command> commands;
};
struct Render_Tile_Job_Data{
Render *r;
Rect2 region;
};
enum Scene {
Scene_F22,
Scene_Sponza,
Scene_Count,
};
global F32 light_rotation = 0;
global F32 zfar_value = 100000.f;
function
Vec4 srgb_to_almost_linear(Vec4 a) {
Vec4 result = {a.r*a.r, a.g*a.g, a.b*a.b, a.a};
return result; // @Note: Linear would be to power of 2.2
}
function
Vec4 almost_linear_to_srgb(Vec4 a) {
Vec4 result = { sqrtf(a.r), sqrtf(a.g), sqrtf(a.b), a.a };
return result;
}
function
Vec4 premultiplied_alpha(Vec4 dst, Vec4 src) {
Vec4 result;
result.r = src.r + ((1-src.a) * dst.r);
result.g = src.g + ((1-src.a) * dst.g);
result.b = src.b + ((1-src.a) * dst.b);
result.a = src.a + dst.a - src.a*dst.a;
return result;
}
function
void draw_rect(Bitmap* dst, F32 X, F32 Y, F32 w, F32 h, Vec4 color) {
int max_x = (int)(min(X + w, (F32)dst->x) + 0.5f);
int max_y = (int)(min(Y + h, (F32)dst->y) + 0.5f);
int min_x = (int)(max(0.f, X) + 0.5f);
int min_y = (int)(max(0.f, Y) + 0.5f);
color.rgb *= color.a;
color = srgb_to_almost_linear(color);
for (int y = min_y; y < max_y; y++) {
for (int x = min_x; x < max_x; x++) {
U32 *dst_pixel = dst->pixels + (x + y * dst->x);
Vec4 dstc = srgb_to_almost_linear(vec4abgr(*dst_pixel));
dstc = premultiplied_alpha(dstc, color);
U32 color32 = vec4_to_u32abgr(almost_linear_to_srgb(dstc));
*dst_pixel = color32;
}
}
}
function void
draw_bitmap(Bitmap *dst, Bitmap *src, Vec2 pos){
S64 minx = (S64)(pos.x + 0.5);
S64 miny = (S64)(pos.y + 0.5);
S64 maxx = minx + src->x;
S64 maxy = miny + src->y;
S64 offsetx = 0;
S64 offsety = 0;
if (maxx > dst->x) {
maxx = dst->x;
}
if (maxy > dst->y) {
maxy = dst->y;
}
if (minx < 0) {
offsetx = -minx;
minx = 0;
}
if (miny < 0) {
offsety = -miny;
miny = 0;
}
for (S64 y = miny; y < maxy; y++) {
for (S64 x = minx; x < maxx; x++) {
S64 tx = x - minx + offsetx;
S64 ty = y - miny + offsety;
U32 *dst_pixel = dst->pixels + (x + y * dst->x);
U32 *pixel = src->pixels + (tx + ty * src->x);
Vec4 result_color = srgb_to_almost_linear(vec4abgr(*pixel));
Vec4 dst_color = srgb_to_almost_linear(vec4abgr(*dst_pixel));
result_color = premultiplied_alpha(dst_color, result_color);
result_color = almost_linear_to_srgb(result_color);
U32 color32 = vec4_to_u32abgr(result_color);
*dst_pixel = color32;
}
}
}
function
void draw_bitmap(Bitmap* dst, Bitmap* src, Vec2 pos, Vec2 size) {
S64 minx = (S64)(pos.x + 0.5);
S64 miny = (S64)(pos.y + 0.5);
S64 maxx = minx + (S64)(size.x + 0.5f);
S64 maxy = miny + (S64)(size.y + 0.5f);
S64 offsetx = 0;
S64 offsety = 0;
maxx = clamp_top(maxx, (S64)dst->x);
maxy = clamp_top(maxy, (S64)dst->y);
if (minx < 0) {
offsetx = -minx;
minx = 0;
}
if (miny < 0) {
offsety = -miny;
miny = 0;
}
F32 distx = (F32)(maxx - minx);
F32 disty = (F32)(maxy - miny);
for (S64 y = miny; y < maxy; y++) {
for (S64 x = minx; x < maxx; x++) {
F32 u = (F32)(x - minx) / distx;
F32 v = (F32)(y - miny) / disty;
S64 tx = (S64)(u * src->x + 0.5f);
S64 ty = (S64)(v * src->y + 0.5f);
U32 *dst_pixel = dst->pixels + (x + y * dst->x);
U32 *pixel = src->pixels + (tx + ty * src->x);
Vec4 result_color = srgb_to_almost_linear(vec4abgr(*pixel));
Vec4 dst_color = srgb_to_almost_linear(vec4abgr(*dst_pixel));
result_color = premultiplied_alpha(dst_color, result_color);
result_color = almost_linear_to_srgb(result_color);
U32 color32 = vec4_to_u32abgr(result_color);
*dst_pixel = color32;
}
}
}
function
Vec4 base_string(Bitmap *dst, Font *font, String word, Vec2 pos, B32 draw) {
Vec2 og_position = pos;
F32 max_x = pos.x;
for (U64 i = 0; i < word.len; i++) {
if (word.str[i] == ' ') {
FontGlyph* g = &font->glyphs['_' - '!'];
pos.x += g->xadvance;
if (pos.x > max_x) max_x = pos.x;
}
else if (word.str[i] == '\n') {
pos.y -= font->line_advance;
pos.x = og_position.x;
}
else if((word.str[i] >= '!' && word.str[i] <= 127)){
FontGlyph* g = &font->glyphs[word.str[i] - '!'];
if(draw) draw_bitmap(dst, &g->bitmap, pos - g->bitmap.align);
pos.x += g->xadvance;
if (pos.x > max_x) max_x = pos.x;
}
}
Vec4 rect = vec4(og_position.x, pos.y, max_x - og_position.x, og_position.y - pos.y + font->line_advance);
return rect;
}
function
Vec4 draw_string(Bitmap *dst, Font *font, String word, Vec2 pos) {
return base_string(dst, font, word, pos, true);
}
function
Vec4 get_string_rect(Font *font, String word, Vec2 pos) {
return base_string(0, font, word, pos, false);
}
function
F32 edge_function(Vec4 vecp0, Vec4 vecp1, Vec4 p) {
F32 result = (vecp1.y - vecp0.y) * (p.x - vecp0.x) - (vecp1.x - vecp0.x) * (p.y - vecp0.y);
return result;
}
#define F32x8 __m256
#define S32x8 __m256i
S32 render_triangle_test_case_number = 5;
S32 render_triangle_test_case_angle = -1;
U64 filled_pixel_count;
U64 filled_pixel_cycles;
U64 triangle_count;
#include "optimization_log.cpp"
function
void draw_triangle_nearest(Bitmap* dst, F32 *depth_buffer, Bitmap *src,
Vec4 p0, Vec4 p1, Vec4 p2,
Vec2 tex0, Vec2 tex1, Vec2 tex2, Rect2 rect) {
if(src->pixels == 0) return;
// U64 fill_pixels_begin = __rdtsc();
F32 min_x1 = (F32)(min(p0.x, min(p1.x, p2.x)));
F32 min_y1 = (F32)(min(p0.y, min(p1.y, p2.y)));
F32 max_x1 = (F32)(max(p0.x, max(p1.x, p2.x)));
F32 max_y1 = (F32)(max(p0.y, max(p1.y, p2.y)));
S64 min_x = (S64)max(rect.min_x, floor(min_x1));
S64 min_y = (S64)max(rect.min_y, floor(min_y1));
S64 max_x = (S64)min(rect.max_x, ceil(max_x1));
S64 max_y = (S64)min(rect.max_y, ceil(max_y1));
if (min_y >= max_y) return;
if (min_x >= max_x) return;
F32 dy10 = (p1.y - p0.y);
F32 dy21 = (p2.y - p1.y);
F32 dy02 = (p0.y - p2.y);
F32 dx10 = (p1.x - p0.x);
F32 dx21 = (p2.x - p1.x);
F32 dx02 = (p0.x - p2.x);
F32x8 var255 = _mm256_set1_ps(255);
F32x8 var0 = _mm256_set1_ps(0);
F32x8 var1 = _mm256_set1_ps(1);
F32x8 var_max_x = _mm256_set1_ps(max_x);
F32x8 var07 = _mm256_set_ps(7,6,5,4,3,2,1,0);
F32x8 inv255 = _mm256_div_ps(var1, var255);
F32x8 var_src_x_minus_one = _mm256_set1_ps(src->x-1);
F32x8 var_src_y_minus_one = _mm256_set1_ps(src->y-1);
S32x8 var_src_y_minus_one_int = _mm256_set1_epi32(src->y-1);
S32x8 var_src_x_int = _mm256_set1_epi32(src->x);
S32x8 var_0xff000000 = _mm256_set1_epi32(0xff000000);
S32x8 var_0x00ff0000 = _mm256_set1_epi32(0x00ff0000);
S32x8 var_0x0000ff00 = _mm256_set1_epi32(0x0000ff00);
S32x8 var_0x000000ff = _mm256_set1_epi32(0x000000ff);
F32x8 var_tex0x = _mm256_set1_ps(tex0.x);
F32x8 var_tex1x = _mm256_set1_ps(tex1.x);
F32x8 var_tex2x = _mm256_set1_ps(tex2.x);
F32x8 var_tex0y = _mm256_set1_ps(tex0.y);
F32x8 var_tex1y = _mm256_set1_ps(tex1.y);
F32x8 var_tex2y = _mm256_set1_ps(tex2.y);
F32x8 inv_p0w = _mm256_div_ps(var1, _mm256_set1_ps(p0.w));
F32x8 inv_p1w = _mm256_div_ps(var1, _mm256_set1_ps(p1.w));
F32x8 inv_p2w = _mm256_div_ps(var1, _mm256_set1_ps(p2.w));
F32x8 one_over_p0w = _mm256_set1_ps(1.f / p0.w);
F32x8 one_over_p1w = _mm256_set1_ps(1.f / p1.w);
F32x8 one_over_p2w = _mm256_set1_ps(1.f / p2.w);
U32 *destination = dst->pixels + dst->x*min_y;
F32 area = (p1.y - p0.y) * (p2.x - p0.x) - (p1.x - p0.x) * (p2.y - p0.y);
F32x8 inv_area8 = _mm256_div_ps(var1, _mm256_set1_ps(area));
F32x8 _dy10 = _mm256_set1_ps(dy10);
F32x8 _dx10 = _mm256_set1_ps(dx10);
F32x8 _dy21 = _mm256_set1_ps(dy21);
F32x8 _dx21 = _mm256_set1_ps(dx21);
F32x8 _dy02 = _mm256_set1_ps(dy02);
F32x8 _dx02 = _mm256_set1_ps(dx02);
F32x8 p0_x = _mm256_set1_ps(p0.x);
F32x8 p0_y = _mm256_set1_ps(p0.y);
F32x8 p1_x = _mm256_set1_ps(p1.x);
F32x8 p1_y = _mm256_set1_ps(p1.y);
F32x8 p2_x = _mm256_set1_ps(p2.x);
F32x8 p2_y = _mm256_set1_ps(p2.y);
for (S64 y = min_y; y < max_y; y++) {
F32x8 Y = _mm256_set1_ps(y);
for (S64 x8 = min_x; x8 < max_x; x8+=8) {
F32x8 X = _mm256_add_ps(_mm256_set1_ps(x8), var07);
// Compute the edges
// F32x8 edge0 = (p1.y - p0.y) * (p.x - p0.x) - (p1.x - p0.x) * (p.y - p0.y);
F32x8 px_minus_0x = _mm256_sub_ps(X, p0_x);
F32x8 py_minus_0y = _mm256_sub_ps(Y, p0_y);
F32x8 right0 = _mm256_mul_ps(_dx10, py_minus_0y);
F32x8 edge0 = _mm256_fmsub_ps(_dy10, px_minus_0x, right0);
// F32 result = (p2.y - p1.y) * (p.x - p1.x) - (p2.x - p1.x) * (p.y - p1.y);
F32x8 px_minus_1x = _mm256_sub_ps(X, p1_x);
F32x8 py_minus_1y = _mm256_sub_ps(Y, p1_y);
F32x8 right1 = _mm256_mul_ps(_dx21, py_minus_1y);
F32x8 edge1 = _mm256_fmsub_ps(_dy21, px_minus_1x, right1);
// F32 result = (p0.y - p2.y) * (p.x - p2.x) - (p0.x - p2.x) * (p.y - p2.y);
F32x8 px_minus_2x = _mm256_sub_ps(X, p2_x);
F32x8 py_minus_2y = _mm256_sub_ps(Y, p2_y);
F32x8 right2 = _mm256_mul_ps(_dx02, py_minus_2y);
F32x8 edge2 = _mm256_fmsub_ps(_dy02, px_minus_2x, right2);
F32x8 should_fill;
F32x8 test_if_x_should_be_clipped = _mm256_cmp_ps(X, var_max_x, _CMP_LT_OQ);
F32x8 test_if_pixel_inside_edge_using_dot_result0 = _mm256_cmp_ps(edge0, var0, _CMP_GE_OQ);
F32x8 test_if_pixel_inside_edge_using_dot_result1 = _mm256_cmp_ps(edge1, var0, _CMP_GE_OQ);
F32x8 test_if_pixel_inside_edge_using_dot_result2 = _mm256_cmp_ps(edge2, var0, _CMP_GE_OQ);
F32x8 dot_result_combination0 = _mm256_and_ps(test_if_pixel_inside_edge_using_dot_result0, test_if_pixel_inside_edge_using_dot_result1);
F32x8 dot_result_combination1 = _mm256_and_ps(dot_result_combination0, test_if_pixel_inside_edge_using_dot_result2);
should_fill = _mm256_and_ps(test_if_x_should_be_clipped, dot_result_combination1);
F32x8 w0 = _mm256_mul_ps(edge1, inv_area8);
F32x8 w1 = _mm256_mul_ps(edge2, inv_area8);
F32x8 w2 = _mm256_mul_ps(edge0, inv_area8);
// @Todo: Turn this into 1 / interpolated_w, turns out in theory it should be
// more performant but couldn't make it work
// @Old_Note: We could do: interpolated_w = 1.f / interpolated_w to get proper depth
// but why waste an instruction, the smaller the depth value the farther the object
F32x8 interpolated_w = _mm256_mul_ps(one_over_p0w, w0);
interpolated_w = _mm256_fmadd_ps(one_over_p1w, w1, interpolated_w);
interpolated_w = _mm256_fmadd_ps(one_over_p2w, w2, interpolated_w);
F32 *depth_pointer = (depth_buffer + (x8 + y * dst->x));
F32x8 depth = _mm256_loadu_ps(depth_pointer);
F32x8 should_fill_term = _mm256_cmp_ps(depth, interpolated_w, _CMP_LT_OQ);
should_fill = _mm256_and_ps(should_fill, should_fill_term);
// If all pixels are not going to get drawn then opt out
F32x8 compare_with_zero = _mm256_cmpeq_epi32(should_fill, var0);
int mask = _mm256_movemask_epi8(compare_with_zero);
if(mask == 0xffffffff) {
continue;
}
F32x8 invw0 = _mm256_mul_ps(w0, inv_p0w);
F32x8 invw1 = _mm256_mul_ps(w1, inv_p1w);
F32x8 invw2 = _mm256_mul_ps(w2, inv_p2w);
F32x8 u0 = _mm256_mul_ps(var_tex0x, invw0);
u0 = _mm256_fmadd_ps(var_tex1x, invw1, u0);
u0 = _mm256_fmadd_ps(var_tex2x, invw2, u0);
F32x8 v0 = _mm256_mul_ps(var_tex0y, invw0);
v0 = _mm256_fmadd_ps(var_tex1y, invw1, v0);
v0 = _mm256_fmadd_ps(var_tex2y, invw2, v0);
F32x8 u1 = _mm256_div_ps(u0, interpolated_w);
F32x8 v1 = _mm256_div_ps(v0, interpolated_w);
F32x8 u_floored = _mm256_floor_ps(u1);
F32x8 v_floored = _mm256_floor_ps(v1);
F32x8 u2 = _mm256_sub_ps(u1, u_floored);
F32x8 v2 = _mm256_sub_ps(v1, v_floored);
F32x8 u3 = _mm256_mul_ps(u2, var_src_x_minus_one);
F32x8 v3 = _mm256_mul_ps(v2, var_src_y_minus_one);
F32x8 ui = _mm256_cvtps_epi32(u3);
F32x8 vi = _mm256_cvtps_epi32(v3);
// Origin UV (0,0) is in bottom left
_mm256_maskstore_epi32((int *)depth_pointer, should_fill, interpolated_w);
//
// Fetch and calculate texel values
//
S32x8 indices_to_fetch0 = _mm256_sub_epi32(var_src_y_minus_one_int, vi);
S32x8 indices_to_fetch1 = _mm256_mullo_epi32(var_src_x_int, indices_to_fetch0);
S32x8 indices_to_fetch2 = _mm256_add_epi32(indices_to_fetch1, ui);
S32x8 indices_to_fetch3 = _mm256_and_si256(indices_to_fetch2, should_fill);
S32x8 pixel = _mm256_set_epi32(
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 7)],
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 6)],
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 5)],
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 4)],
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 3)],
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 2)],
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 1)],
src->pixels[_mm256_extract_epi32(indices_to_fetch3, 0)]
);
S32x8 texel_i_a = _mm256_and_si256(pixel, var_0xff000000);
S32x8 texel_i_b = _mm256_and_si256(pixel, var_0x00ff0000);
S32x8 texel_i_g = _mm256_and_si256(pixel, var_0x0000ff00);
S32x8 texel_i_r = _mm256_and_si256(pixel, var_0x000000ff);
texel_i_a = _mm256_srli_epi32(texel_i_a, 24);
texel_i_b = _mm256_srli_epi32(texel_i_b, 16);
texel_i_g = _mm256_srli_epi32(texel_i_g, 8 );
F32x8 texel_a0 = _mm256_cvtepi32_ps(texel_i_a);
F32x8 texel_b0 = _mm256_cvtepi32_ps(texel_i_b);
F32x8 texel_g0 = _mm256_cvtepi32_ps(texel_i_g);
F32x8 texel_r0 = _mm256_cvtepi32_ps(texel_i_r);
F32x8 texel_b1 = _mm256_mul_ps(texel_b0, inv255);
F32x8 texel_g1 = _mm256_mul_ps(texel_g0, inv255);
F32x8 texel_r1 = _mm256_mul_ps(texel_r0, inv255);
F32x8 texel_a1 = _mm256_mul_ps(texel_a0, inv255);
texel_r1 = _mm256_mul_ps(texel_r1, texel_r1);
texel_g1 = _mm256_mul_ps(texel_g1, texel_g1);
texel_b1 = _mm256_mul_ps(texel_b1, texel_b1);
//
// Fetch and calculate dst pixels
//
U32 *dst_memory = destination + x8;
S32x8 dst_pixel = _mm256_maskload_epi32((const int *)dst_memory, should_fill);
S32x8 dst_i_a0 = _mm256_and_si256(dst_pixel, var_0xff000000);
S32x8 dst_i_b0 = _mm256_and_si256(dst_pixel, var_0x00ff0000);
S32x8 dst_i_g0 = _mm256_and_si256(dst_pixel, var_0x0000ff00);
S32x8 dst_i_r0 = _mm256_and_si256(dst_pixel, var_0x000000ff);
S32x8 dst_i_a1 = _mm256_srli_epi32(dst_i_a0, 24);
S32x8 dst_i_b1 = _mm256_srli_epi32(dst_i_b0, 16);
S32x8 dst_i_g1 = _mm256_srli_epi32(dst_i_g0, 8);
S32x8 dst_i_r1 = dst_i_r0;
F32x8 dst_a = _mm256_cvtepi32_ps(dst_i_a1);
F32x8 dst_b = _mm256_cvtepi32_ps(dst_i_b1);
F32x8 dst_g = _mm256_cvtepi32_ps(dst_i_g1);
F32x8 dst_r = _mm256_cvtepi32_ps(dst_i_r1);
dst_a = _mm256_mul_ps(dst_a, inv255);
dst_b = _mm256_mul_ps(dst_b, inv255);
dst_g = _mm256_mul_ps(dst_g, inv255);
dst_r = _mm256_mul_ps(dst_r, inv255);
dst_r = _mm256_mul_ps(dst_r, dst_r);
dst_g = _mm256_mul_ps(dst_g, dst_g);
dst_b = _mm256_mul_ps(dst_b, dst_b);
// Premultiplied alpha
{
F32x8 inv_texel_a = _mm256_sub_ps(var1,texel_a1);
dst_r = _mm256_fmadd_ps(inv_texel_a, dst_r, texel_r1);
dst_g = _mm256_fmadd_ps(inv_texel_a, dst_g, texel_g1);
dst_b = _mm256_fmadd_ps(inv_texel_a, dst_b, texel_b1);
dst_a = _mm256_sub_ps(_mm256_add_ps(texel_a1, dst_a), _mm256_mul_ps(texel_a1,dst_a));
}
// Almost linear to srgb
{
dst_r = _mm256_sqrt_ps(dst_r);
dst_g = _mm256_sqrt_ps(dst_g);
dst_b = _mm256_sqrt_ps(dst_b);
}
// Convert to integer format
dst_r = _mm256_mul_ps(dst_r, var255);
dst_g = _mm256_mul_ps(dst_g, var255);
dst_b = _mm256_mul_ps(dst_b, var255);
dst_a = _mm256_mul_ps(dst_a, var255);
S32x8 dst_r_int = _mm256_cvtps_epi32(dst_r);
S32x8 dst_g_int = _mm256_cvtps_epi32(dst_g);
S32x8 dst_b_int = _mm256_cvtps_epi32(dst_b);
S32x8 dst_a_int = _mm256_cvtps_epi32(dst_a);
S32x8 dst_int_a_shifted = _mm256_slli_epi32(dst_a_int, 24);
S32x8 dst_int_b_shifted = _mm256_slli_epi32(dst_b_int, 16);
S32x8 dst_int_g_shifted = _mm256_slli_epi32(dst_g_int, 8);
S32x8 dst_int_r_shifted = dst_r_int;
S32x8 packed_abgr0 = _mm256_or_si256(dst_int_a_shifted, dst_int_b_shifted);
S32x8 packed_abgr1 = _mm256_or_si256(dst_int_r_shifted, dst_int_g_shifted);
S32x8 packed_abgr2 = _mm256_or_si256(packed_abgr1, packed_abgr0);
_mm256_maskstore_epi32((int *)dst_memory, should_fill, packed_abgr2);
}
destination += dst->x;
}
// filled_pixel_cycles += __rdtsc() - fill_pixels_begin;
// filled_pixel_count += (max_x - min_x)*(max_y - min_y);
}
WORK_QUEUE_CALLBACK(draw_tile){
auto d = (Render_Tile_Job_Data *)data;
Render *r = d->r;
For_It(r->commands){
draw_triangle_nearest(&r->screen320, r->depth320, it.item->src, it.item->p0, it.item->p1, it.item->p2, it.item->tex0, it.item->tex1, it.item->tex2, d->region);
}
}
function
void draw_mesh(Render *r, String scene_name, Obj_Material *materials, Obj_Mesh *mesh, Vec3 *vertices, Vec2 *tex_coords, Vec3 *normals) {
for (int i = 0; i < mesh->indices.len; i++) {
Obj_Index *index = mesh->indices.data + i;
Bitmap *image = &r->img;
if(index->material_id != -1) {
Obj_Material *material = materials + index->material_id;
// @Todo: No size info from OBJ things, this stuff needs a bit of refactor
// Need to figure out how to accomodate multiple possible formats of input etc.
if(material->texture_ambient.pixels) {
image = &material->texture_ambient;
}
}
Vertex vert[] = {
{
vertices[index->vertex[0] - 1],
tex_coords[index->tex[0] - 1],
normals[index->normal[0] - 1],
},
{
vertices[index->vertex[1] - 1],
tex_coords[index->tex[1] - 1],
normals[index->normal[1] - 1],
},
{
vertices[index->vertex[2] - 1],
tex_coords[index->tex[2] - 1],
normals[index->normal[2] - 1],
},
};
//@Note: Transform
for (int j = 0; j < 3; j++) {
vert[j].pos = r->transform * vert[j].pos;
}
Vec3 p0_to_camera = r->camera_pos - vert[0].pos;
Vec3 p0_to_p1 = vert[1].pos - vert[0].pos;
Vec3 p0_to_p2 = vert[2].pos - vert[0].pos;
Vec3 normal = normalize(cross(p0_to_p1, p0_to_p2));
Vec3 light_direction = mat4_rotation_x(light_rotation) * vec3(0, 0, 1);
if (dot(normal, p0_to_camera) > 0) { //@Note: Backface culling
/// ## Clipping
///
/// There are 3 clipping stages, 2 clipping stages in 3D space against zfar and znear and 1 clipping
/// stage in 2D against left, bottom, right, top(2D image bounds).
///
/// First the triangles get clipped against the zfar plane,
/// if a triangle has even one vertex outside the clipping region, the entire triangle gets cut.
/// So far I didn't have problems with that. It simplifies the computations and splitting triangles
/// on zfar seems like a waste of power.
///
/// The second clipping stage is znear plane. Triangles get fully and nicely clipped against znear.
/// Every time a triangle gets partially outside the clipping region it gets cut to the znear and
/// either one or two new triangles get derived from the old one.
///
/// Last clipping stage is performed in the 2D image space. Every triangle has a corresponding AABB
/// box. In this box every pixel gets tested to see if it's in the triangle. In this clipping stage
/// the box is clipped to the image metrics - 0, 0, width, height.
///
///
// @Note: Zfar
B32 vertex_is_outside = false;
Vec3 zfar_normal = vec3(0, 0, -1);
Vec3 zfar_pos = vec3(0, 0, zfar_value);
for (S32 j = 0; j < 3; j++) {
// @Note: Camera
vert[j].pos = r->camera * vert[j].pos;
// @Note: Skip triangle if even one vertex gets outside the clipping plane
if ((dot(zfar_normal, vert[j].pos - zfar_pos) < 0)) {
vertex_is_outside = true;
break;
}
}
if (vertex_is_outside) {
continue;
}
// @Note: Znear, clip triangles to the near clipping plane
Vec3 znear_normal = vec3(0, 0, 1);
Vec3 znear_pos = vec3(0, 0, 1.f);
struct _Vertex {
Vec4 pos;
Vec2 tex;
Vec3 norm;
} in[4];
S32 in_count = 0;
Vertex *prev = vert + 2;
Vertex *curr = vert;
F32 prev_dot = dot(znear_normal, prev->pos - znear_pos);
F32 curr_dot = 0;
for (int j = 0; j < 3; j++) {
curr_dot = dot(znear_normal, curr->pos - znear_pos);
if (curr_dot * prev_dot < 0) {
F32 t = prev_dot / (prev_dot - curr_dot);
in[in_count].pos = vec4(lerp(prev->pos, curr->pos, t), 1);
in[in_count].tex = lerp(prev->tex, curr->tex, t);
in[in_count].norm = lerp(prev->norm, curr->norm, t);
in_count += 1;
}
if (curr_dot > 0) {
in[in_count].pos = vec4(vert[j].pos, 1);
in[in_count].tex = vert[j].tex;
in[in_count++].norm = vert[j].norm;
}
prev = curr++;
prev_dot = curr_dot;
}
if (in_count == 0) {
continue;
}
for(S64 j = 0; j < in_count; j++) {
//@Note: Perspective
in[j].pos = r->projection * in[j].pos;
in[j].pos.x = in[j].pos.x / in[j].pos.w;
in[j].pos.y = in[j].pos.y / in[j].pos.w;
// in[j].pos.z = in[j].pos.z / in[j].pos.w;
//@Note: To pixel space
in[j].pos.x *= r->screen320.x / 2;
in[j].pos.y *= r->screen320.y / 2;
in[j].pos.x += r->screen320.x / 2;
in[j].pos.y += r->screen320.y / 2;
}
triangle_count++;
if (in_count > 3) triangle_count++;
#if MULTITHREADING
Render_Command *command = array_alloc(os.perm_arena, &r->commands);
command->src = image;
command->p0 = in[0].pos;
command->p1 = in[1].pos;
command->p2 = in[2].pos;
command->tex0 = in[0].tex;
command->tex1 = in[1].tex;
command->tex2 = in[2].tex;
if(in_count > 3){
Render_Command *command = array_alloc(os.perm_arena, &r->commands);
command->src = image;
command->p0 = in[0].pos;
command->p1 = in[2].pos;
command->p2 = in[3].pos;
command->tex0 = in[0].tex;
command->tex1 = in[2].tex;
command->tex2 = in[3].tex;
}
#else
switch(render_triangle_test_case_number){
case 1:
draw_triangle_nearest_a(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[1].pos, in[2].pos, in[0].tex, in[1].tex, in[2].tex, in[0].norm, in[1].norm, in[2].norm);
if (in_count > 3) draw_triangle_nearest_a(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[2].pos, in[3].pos, in[0].tex, in[2].tex, in[3].tex, in[0].norm, in[2].norm, in[3].norm);
break;
case 2:
draw_triangle_nearest_b(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[1].pos, in[2].pos, in[0].tex, in[1].tex, in[2].tex, in[0].norm, in[1].norm, in[2].norm);
if (in_count > 3) draw_triangle_nearest_b(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[2].pos, in[3].pos, in[0].tex, in[2].tex, in[3].tex, in[0].norm, in[2].norm, in[3].norm);
break;
case 3:
draw_triangle_nearest_simd_with_overloads(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[1].pos, in[2].pos, in[0].tex, in[1].tex, in[2].tex, in[0].norm, in[1].norm, in[2].norm);
if (in_count > 3) draw_triangle_nearest_simd_with_overloads(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[2].pos, in[3].pos, in[0].tex, in[2].tex, in[3].tex, in[0].norm, in[2].norm, in[3].norm);
break;
case 4:
draw_triangle_nearest_simd_without_overloads(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[1].pos, in[2].pos, in[0].tex, in[1].tex, in[2].tex, in[0].norm, in[1].norm, in[2].norm);
if (in_count > 3) draw_triangle_nearest_simd_without_overloads(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[2].pos, in[3].pos, in[0].tex, in[2].tex, in[3].tex, in[0].norm, in[2].norm, in[3].norm);
break;
case 5:
draw_triangle_nearest_final(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[1].pos, in[2].pos, in[0].tex, in[1].tex, in[2].tex, in[0].norm, in[1].norm, in[2].norm);
if (in_count > 3) draw_triangle_nearest_final(&r->screen320, r->depth320, image, light_direction, in[0].pos, in[2].pos, in[3].pos, in[0].tex, in[2].tex, in[3].tex, in[0].norm, in[2].norm, in[3].norm);
break;
}
#endif
}
}
}
#include "ui.cpp"
global F32 speed = 100.f;
global F32 rotation = 0;
global Obj *f22;
global Obj *sponza;
global Obj *obj;
global Render r = {};
global Scene scene = Scene_Sponza;
function
UI_SIGNAL_CALLBACK(scene_callback) {
switch(scene) {
case Scene_F22: {
speed = 1;
r.camera_pos = vec3(0,0,-2);
obj = f22;
} break;
case Scene_Sponza: {
speed = 100;
r.camera_pos = vec3(-1020, 687, -85); r.camera_yaw = vec2(-1.3, -0.44);
obj = sponza;
} break;
case Scene_Count:
invalid_default_case;
}
scene = (Scene)(((int)scene + 1) % Scene_Count);
}
FILE *global_file;
function void
windows_log(Log_Kind kind, String string, char *file, int line){
fprintf(global_file, "%s", string.str);
// OutputDebugStringA((const char *)string.str);
}
function void
next_test_case(B32 first_time){
render_triangle_test_case_number += 1;
if(first_time || render_triangle_test_case_number == 6){
render_triangle_test_case_angle += 1;
render_triangle_test_case_number = 1;
try_again: switch(render_triangle_test_case_angle){
case 0: r.camera_pos = vec3(-1020, 687, -85); r.camera_yaw = vec2(-1.3, -0.44); break;
case 1: r.camera_pos = vec3(-356,89.5,168); r.camera_yaw = vec2(0.2, 0); break;
case 2: render_triangle_test_case_angle = 0; goto try_again; break;
}
}
}
int
main(int argc, char **argv) {
global_file = fopen("perfclocks.txt", "a");
thread_ctx.log_proc = windows_log;
os.window_size.x = 1920;
os.window_size.y = 1080;
os.window_resizable = 1;
assert(os_init());
Font font = os_load_font(os.perm_arena, 12*os.dpi_scale, "Arial", 0);
// test_array_list();
// f22 = load_obj_dump(os.perm_arena, "plane.bin"_s);
// sponza = load_obj_dump(os.perm_arena, "sponza.bin"_s);
Obj sponza_obj = load_obj(&os_process_heap, "assets/sponza/sponza.obj"_s);
sponza = &sponza_obj;
scene_callback();
next_test_case(true);
int screen_x = os.window_size.x;
int screen_y = os.window_size.y;
r.screen320 = {(U32 *)arena_push_size(os.perm_arena, screen_x*screen_y*sizeof(U32)), screen_x, screen_y};
r.depth320 = (F32 *)arena_push_size(os.perm_arena, sizeof(F32) * screen_x * screen_y);
r.commands.block_size = 1024*1024;
ThreadStartupInfo thread_infos[16] = {};
init_work_queue(&r.work_queue, buff_cap(thread_infos), thread_infos);
String frame_data = {};
String raster_details = {};
UISetup setup[] = {
UI_SIGNAL("Change scene"_s, scene_callback),
UI_LABEL(&frame_data),
UI_LABEL(&raster_details),
UI_LABEL(&os.text),
};
UI ui = ui_make(setup, buff_cap(setup));
B32 ui_mouse_lock = true;
while (os_game_loop()) {
PROFILE_SCOPE(main_loop);
if (ui_mouse_lock == false) {
r.camera_yaw.x += os.delta_mouse_pos.x * 0.01f;
r.camera_yaw.y -= os.delta_mouse_pos.y * 0.01f;
}
if (os.key[Key_Escape].pressed) os_quit();
if (os.key[Key_O].down) light_rotation += 0.05f;
if (os.key[Key_P].down) light_rotation -= 0.05f;
if (os.key[Key_F2].pressed) {
ui_mouse_lock = !ui_mouse_lock;
os_show_cursor(!os.cursor_visible);
}
if (os.key[Key_A].down) r.camera_pos.x -= speed * (F32)os.delta_time;
if (os.key[Key_D].down) r.camera_pos.x += speed * (F32)os.delta_time;
if (os.key[Key_W].down) {
r.camera_forward_velocity = r.camera_direction * speed * (F32)os.delta_time;
r.camera_pos = r.camera_pos + r.camera_forward_velocity;
}
if (os.key[Key_S].down) {
r.camera_forward_velocity = r.camera_direction * speed * (F32)os.delta_time;
r.camera_pos = r.camera_pos - r.camera_forward_velocity;
}
if (os.key[Key_R].down) r.camera_pos.y += speed * (F32)os.delta_time;
if (os.key[Key_F].down) r.camera_pos.y -= speed * (F32)os.delta_time;
// Clear screen and depth buffer
U32* p = r.screen320.pixels;
for (int y = 0; y < r.screen320.y; y++) {
for (int x = 0; x < r.screen320.x; x++) {
*p++ = 0x33333333;
}
}
F32* dp = r.depth320;
for (int y = 0; y < r.screen320.y; y++) {
for (int x = 0; x < r.screen320.x; x++) {
*dp++ = -F32MAX;
}
}
Mat4 camera_rotation = mat4_rotation_y(r.camera_yaw.x) * mat4_rotation_x(r.camera_yaw.y);
r.camera_direction = (camera_rotation * vec4(0,0,1,1)).xyz;
Vec3 target = r.camera_pos + r.camera_direction;
r.camera = mat4_look_at(r.camera_pos, target, vec3(0, 1, 0));
r.projection = mat4_perspective(60.f, (F32)os.screen->x, (F32)os.screen->y, 1.f, zfar_value);
r.transform = mat4_rotation_z(rotation);
r.transform = r.transform * mat4_rotation_y(rotation);
for (int i = 0; i < obj->mesh.len; i++) {
PROFILE_SCOPE(draw_all_meshes);
Vec2* tex_coords = (Vec2*)obj->texture_coordinates.data;
Vec3 *normals = (Vec3 *)obj->normals.data;
Obj_Mesh *mesh = obj->mesh.data;
Vec3* vertices = (Vec3 *)obj->vertices.data;
draw_mesh(&r, obj->name, obj->materials.data, mesh+i, vertices, tex_coords, normals);
}
#if MULTITHREADING
Render_Tile_Job_Data tile_job_data[32];
S32 x_tiles = 1;
S32 y_tiles = 16;
F32 block_size_x = r.screen320.x / x_tiles;
F32 block_size_y = r.screen320.y / y_tiles;
S32 i = 0;
for(S32 x = 0; x < x_tiles; x++){
for(S32 y = 0; y < y_tiles; y++){
Rect2 bounding_rect;
bounding_rect.min_x = block_size_x * x;
bounding_rect.min_y = block_size_y * y;
bounding_rect.max_x = bounding_rect.min_x + block_size_x;
bounding_rect.max_y = bounding_rect.min_y + block_size_y;
tile_job_data[i].region = bounding_rect;
tile_job_data[i].r = &r;
push_work(&r.work_queue, (void *)(tile_job_data + i), draw_tile);
i += 1;
}
}
wait_until_completion(&r.work_queue);
array_free_all_nodes(&r.commands);
#endif
// @Note: Draw 320screen to OS screen
U32* ptr = os.screen->pixels;
for (int y = 0; y < os.screen->y; y++) {
for (int x = 0; x < os.screen->x; x++) {
F32 u = (F32)x / (F32)os.screen->x;
F32 v = (F32)y / (F32)os.screen->y;
int tx = (int)(u * r.screen320.x );
int ty = (int)(v * r.screen320.y );
*ptr++ = r.screen320.pixels[tx + ty * (r.screen320.x)];
}
}
ui_end_frame(os.screen, &ui, &font);
frame_data = string_fmt(os.frame_arena, "FPS:%f dt:%f frame:%u camera_pos: %f %f %f camera_yaw: %f %f",
os.fps, os.delta_time*1000, os.frame, r.camera_pos.x, r.camera_pos.y, r.camera_pos.z, r.camera_yaw.x, r.camera_yaw.y);
#if MULTITHREADING
if(os.frame == 1) log_info("Angle;Frame_Time\n");
log_info("%d;%f\n", render_triangle_test_case_angle, os.delta_time*1000);
#else
if(os.frame == 1) log_info("Angle;Algorithm;Frame_Time;Cycles_Per_Pixel;Cycles_To_Process_Triangles;Pixels_Processed;Triangles\n");
log_info("%d;%d;%f;%llu;%llu;%llu;%llu\n", render_triangle_test_case_angle, render_triangle_test_case_number,
os.delta_time*1000, filled_pixel_cycles/filled_pixel_count, filled_pixel_cycles, filled_pixel_count, triangle_count);
#endif
filled_pixel_count = 0;
filled_pixel_cycles = 0;
triangle_count = 0;
// @Todo I think there is bug with test_case_number, after doing full round it
// skips a phase
if(os.frame % 15 == 0){
next_test_case(false);
}
}
}
/////////////////////////////////////////////////////////////////////////////////////
/// ### Resources that helped me build the rasterizer (Might be helpful to you too):
///
/// * Algorithm I used for triangle rasterization by Juan Pineda is described in paper called "A Parallel Algorithm for Polygon Rasterization"
/// * Casey Muratori's series on making a game from scratch(including a 2D software rasterizer(episode ~82) and 3d gpu renderer): https://hero.handmade.network/episode/code#
/// * Fabian Giessen's "Optimizing Software Occlusion Culling": https://fgiesen.wordpress.com/2013/02/17/optimizing-sw-occlusion-culling-index/
/// * Fabian Giessen's optimized software renderer: https://github.com/rygorous/intel_occlusion_cull/tree/blog/SoftwareOcclusionCulling
/// * Fabian Giessen's javascript triangle rasterizer: https://gist.github.com/rygorous/2486101
/// * Fabian Giessen's C++ triangle rasterizer: https://github.com/rygorous/trirast/blob/master/main.cpp
/// * Joy's Kenneth lectures about computer graphics: https://www.youtube.com/playlist?list=PL_w_qWAQZtAZhtzPI5pkAtcUVgmzdAP8g
/// * Joy's Kenneth article on clipping: https://import.cdn.thinkific.com/167815/JoyKennethClipping-200905-175314.pdf
/// * A bunch of helpful notes and links to resources: https://nlguillemot.wordpress.com/2016/07/10/rasterizer-notes/
/// * Very nice paid course on making a software rasterizer using a scanline method: https://pikuma.com/courses/learn-3d-computer-graphics-programming
/// * Reference for obj loader: https://github.com/tinyobjloader/tinyobjloader/blob/master/tiny_obj_loader.h
///
/// ### To read
///
/// * http://ce-publications.et.tudelft.nl/publications/1362_hardware_algorithms_for_tilebased_realtime_rendering.pdf

86
obj.cpp
View File

@@ -1,86 +0,0 @@
///
/// [?] - Cache bitmaps
/// [ ] - Fix potential portability issues due to compiler struct alignment differences etc.
///
struct Obj_Index {
int vertex[3];
int tex[3];
int normal[3];
S32 material_id;
S32 smoothing_group_id;
};
struct Obj_Mesh {
char name[64];
Array<Obj_Index> indices;
};
struct Obj_Material {
char name[64];
U32 name_len;
Bitmap texture_ambient; // map_Ka
Bitmap texture_diffuse; // map_Kd
Bitmap texture_dissolve; // map_d
Bitmap texture_displacement; // map_Disp
F32 non_transparency; // d
F32 transparency; // Tr
F32 optical_density; // Ni
F32 shininess; // Ns
S32 illumination_model; // illum
Vec3 ambient_color; // Ka
Vec3 diffuse_color; // Kd
Vec3 specular_color; // Ks
};
struct Obj {
String name;
Array<Vec3> vertices;
Array<Vec2> texture_coordinates;
Array<Vec3> normals;
Array<Obj_Mesh> mesh;
Array<Obj_Material> materials;
};
struct Stream{
U8 *cursor;
U8 *end;
};
#define stream_read_array(s,T,c) (T *)stream_read(s,sizeof(T)*(c))
#define stream_read_struct(s,T) stream_read_array(s,T,1)
function void *
stream_read(Stream *s, SizeU size){
U8 *result = s->cursor;
s->cursor += size;
assert(s->end >= s->cursor);
return result;
}
function Obj *
load_obj_dump(Allocator *allocator, String filename){
String string = os_read_file(allocator, filename);
Stream stream = {string.str, string.str + string.len};
Obj *obj = stream_read_struct(&stream, Obj);
obj->name.str = stream_read_array(&stream, U8, obj->name.len);
obj->texture_coordinates.data = stream_read_array(&stream, Vec2, obj->texture_coordinates.len);
obj->vertices.data = stream_read_array(&stream, Vec3, obj->vertices.len);
obj->normals.data = stream_read_array(&stream, Vec3, obj->normals.len);
obj->mesh.data = stream_read_array(&stream, Obj_Mesh, obj->mesh.len);
obj->materials.data = stream_read_array(&stream, Obj_Material, obj->materials.len);
For(obj->mesh){
it.indices.data = stream_read_array(&stream, Obj_Index, it.indices.len);
}
For(obj->materials){
it.texture_ambient.pixels = stream_read_array(&stream, U32, it.texture_ambient.x*it.texture_ambient.y);
it.texture_diffuse.pixels = stream_read_array(&stream, U32, it.texture_diffuse.x*it.texture_diffuse.y);
it.texture_dissolve.pixels = stream_read_array(&stream, U32, it.texture_dissolve.x*it.texture_dissolve.y);
it.texture_displacement.pixels = stream_read_array(&stream, U32, it.texture_displacement.x*it.texture_displacement.y);
}
return obj;
}

432
obj_dump.cpp
View File

@@ -1,432 +0,0 @@
#include "os_windows_multimedia.cpp"
#include "obj.cpp"
#define STBI_ASSERT assert
#define STB_IMAGE_IMPLEMENTATION
#include "dependencies/stb_image.h"
enum class Obj_Token_Type {
none, word, number, whitespace, end
};
struct Obj_Token {
Obj_Token_Type type;
double number;
union {
struct {
char* s;
int len;
};
String s8;
};
};
function Bitmap
load_image(String path) {
Scratch scratch;
String file = os_read_file(scratch, path);
int x, y, n;
unsigned char* data = stbi_load_from_memory(file.str, file.len, &x, &y, &n, 4);
Bitmap result = { (U32*)data, x, y };
if(data) {
U32 *p = result.pixels;
for (int Y = 0; Y < y; Y++) {
for (int X = 0; X < x; X++) {
Vec4 color = vec4abgr(*p);
color.r *= color.a;
color.g *= color.a;
color.b *= color.a;
*p++ = vec4_to_u32abgr(color);
}
}
}
return result;
}
function Obj_Token next_token_raw(char** data) {
Obj_Token result = {};
result.s = *data;
*data += 1;
if (is_alphabetic(*result.s)) {
result.type = Obj_Token_Type::word;
while (!is_whitespace(**data)) {
*data += 1;
}
result.len = (int)(*data - result.s);
}
else if (is_number(*result.s) || *result.s == '-') {
result.type = Obj_Token_Type::number;
while (is_number(**data) || **data == '.' || **data == 'e' || **data == '-') {
*data += 1;
}
result.number = atof(result.s);
result.len = (int)(*data - result.s);
}
else if (*result.s == '#') {
while (**data != '\n') *data += 1;
result = next_token_raw(data);
}
else if (is_whitespace(*result.s)) {
result.type = Obj_Token_Type::whitespace;
while (is_whitespace(**data)) *data += 1;
result.len = (int)(*data - result.s);
}
else if (*result.s == 0) {
result.type = Obj_Token_Type::end;
}
else if (*result.s >= '!') {
result.type = (Obj_Token_Type)*result.s;
}
return result;
}
function Obj_Token next_token(char** data) {
Obj_Token result;
do {
result = next_token_raw(data);
} while (result.type == Obj_Token_Type::whitespace);
return result;
}
function double expect_number(char** data) {
Obj_Token t = next_token(data);
assert(t.type == Obj_Token_Type::number); // @Todo: Error handling, error flag
return t.number;
}
function void expect_token(char** data, char token) {
Obj_Token t = next_token(data);
assert(t.type == (Obj_Token_Type)token); // @Todo: Error handling, error flag
}
function B32 match_token(char **data, char token){
char *save = *data;
Obj_Token t = next_token(&save);
if(t.type == (Obj_Token_Type)token){
*data = save;
return true;
}
return false;
}
function void debug_expect_raw(char** data, Obj_Token_Type type) {
char* data_temp = *data;
Obj_Token t = next_token_raw(&data_temp);
assert(t.type == type);
}
function void
parse_mtl(Obj* obj, String path_obj_folder, String mtl_file) {
Scratch scratch;
char *data = (char *)mtl_file.str;
Obj_Material *m = 0;
for (;;) {
Obj_Token token = next_token(&data);
if (token.type == Obj_Token_Type::end) break;
else if (token.type == Obj_Token_Type::word) {
if (string_compare(token.s8, "newmtl"_s)) {
token = next_token(&data);
m = obj->materials.push_empty_zero();
m->name_len = clamp_top(token.len, 64);
memory_copy(m->name, token.s8.str, m->name_len);
}
else if (string_compare(token.s8, "Ns"_s)) {
m->shininess = expect_number(&data);
}
else if (string_compare(token.s8, "Ka"_s)) {
m->ambient_color.x = expect_number(&data);
m->ambient_color.y = expect_number(&data);
m->ambient_color.z = expect_number(&data);
}
else if (string_compare(token.s8, "Kd"_s)) {
m->diffuse_color.x = expect_number(&data);
m->diffuse_color.y = expect_number(&data);
m->diffuse_color.z = expect_number(&data);
}
else if (string_compare(token.s8, "Ks"_s)) {
m->specular_color.x = expect_number(&data);
m->specular_color.y = expect_number(&data);
m->specular_color.z = expect_number(&data);
}
else if (string_compare(token.s8, "Ni"_s)) {
m->optical_density = expect_number(&data);
}
else if (string_compare(token.s8, "d"_s)) {
m->non_transparency = expect_number(&data);
}
else if (string_compare(token.s8, "illum"_s)) {
m->illumination_model = (S32)expect_number(&data);
}
else if (string_compare(token.s8, "map_Kd"_s)) {
Obj_Token t = next_token(&data);
String path = string_fmt(scratch, "%Q/%Q\0", path_obj_folder, t.s8);
m->texture_diffuse = load_image(path);
}
else if (string_compare(token.s8, "map_Ka"_s)) {
Obj_Token t = next_token(&data);
String path = string_fmt(scratch, "%Q/%Q\0", path_obj_folder, t.s8);
m->texture_ambient = load_image(path);
}
else if (string_compare(token.s8, "map_d"_s)) {
Obj_Token t = next_token(&data);
String path = string_fmt(scratch, "%Q/%Q\0", path_obj_folder, t.s8);
m->texture_dissolve = load_image(path);
}
else if (string_compare(token.s8, "map_Disp"_s)) {
Obj_Token t = next_token(&data);
String path = string_fmt(scratch, "%Q/%Q\0", path_obj_folder, t.s8);
m->texture_displacement = load_image(path);
}
}
}
}
function Obj
parse(Allocator *allocator, char* data, String path_obj_folder) {
Set_Allocator(allocator);
Scratch mtl_scratch;
Obj result = {};
//result.vertices.init(allocator, 160000);
//result.texture_coordinates.init(allocator, 160000);
//result.normals.init(allocator, 160000);
//result.mesh.init(allocator, 64);
//result.materials.init(allocator, 64);
int smoothing = 0;
Obj_Mesh *mesh = result.mesh.push_empty_zero();
//mesh->indices.init(allocator);
int material_id = -1;
S64 debug_i = 0;
for (;;debug_i++){
Obj_Token token = next_token(&data);
if (token.type == Obj_Token_Type::end) break;
else if (token.type == Obj_Token_Type::word) {
if (string_compare(token.s8, "v"_s)) {
Vec3 *vertex = result.vertices.push_empty_zero();
vertex->x = (float)expect_number(&data);
vertex->y = (float)expect_number(&data);
vertex->z = (float)expect_number(&data);
debug_expect_raw(&data, Obj_Token_Type::whitespace);
}
else if (string_compare(token.s8, "vt"_s)) {
Vec2 *tex = result.texture_coordinates.push_empty_zero();
tex->x = (float)expect_number(&data);
tex->y = (float)expect_number(&data);
debug_expect_raw(&data, Obj_Token_Type::whitespace);
}
else if (string_compare(token.s8, "vn"_s)) {
Vec3 *norm = result.normals.push_empty_zero();
norm->x = (float)expect_number(&data);
norm->y = (float)expect_number(&data);
norm->z = (float)expect_number(&data);
debug_expect_raw(&data, Obj_Token_Type::whitespace);
}
else if (string_compare(token.s8, "mtllib"_s)) {
Obj_Token t = next_token(&data);
String path = string_fmt(mtl_scratch, "%Q/%Q", path_obj_folder, t.s8);
String mtl_file = os_read_file(mtl_scratch, path);
if(mtl_file.str) {
parse_mtl(&result, path_obj_folder, mtl_file);
}
}
else if (string_compare(token.s8, "usemtl"_s)) {
Obj_Token t = next_token(&data);
assert(t.type == Obj_Token_Type::word);
for(U64 i = 0; i < result.materials.len; i++) {
Obj_Material *m = result.materials.data + i;
if(string_compare({(U8 *)m->name, m->name_len}, t.s8)) {
material_id = i;
break;
}
}
}
else if (string_compare(token.s8, "o"_s)) {
Obj_Token t = next_token(&data);
assert(t.type == Obj_Token_Type::word);
if (mesh->indices.len != 0) {
mesh = result.mesh.push_empty_zero();
}
else {
U64 len = clamp_top(t.len, 64);
memory_copy(mesh->name, t.s, len);
}
}
else if (string_compare(token.s8, "s"_s)) {
Obj_Token t = next_token(&data);
if (t.type == Obj_Token_Type::number) {
smoothing = (int)t.number;
}
else {
assert(t.type == Obj_Token_Type::word);
if (string_compare(t.s8, "on"_s)) {
smoothing = 1;
}
else if (string_compare(t.s8, "off"_s)) {
smoothing = 0;
}
else invalid_codepath;
}
}
else if (string_compare(token.s8, "g"_s)) {
Obj_Token t = next_token(&data);
assert(t.type == Obj_Token_Type::word);
}
else if (string_compare(token.s8, "f"_s)) {
Obj_Index *i = mesh->indices.push_empty_zero();
i->smoothing_group_id = smoothing;
i->material_id = material_id;
i->vertex[0] = (int)expect_number(&data);
expect_token(&data, '/');
i->tex[0] = (int)expect_number(&data);
if(match_token(&data, '/')) i->normal[0] = (int)expect_number(&data);
i->vertex[1] = (int)expect_number(&data);
expect_token(&data, '/');
i->tex[1] = (int)expect_number(&data);
if(match_token(&data, '/')) i->normal[1] = (int)expect_number(&data);
i->vertex[2] = (int)expect_number(&data);
expect_token(&data, '/');
i->tex[2] = (int)expect_number(&data);
if(match_token(&data, '/')) i->normal[2] = (int)expect_number(&data);
//debug_expect_raw(&data, Obj_Token_Type::whitespace);
}
}
}
return result;
}
function Obj
load_obj(Allocator *arena, String file) {
Scratch scratch;
String data = os_read_file(scratch, file);
assert(data.str);
String path = string_chop_last_slash(file);
Obj result = parse(arena, (char *)data.str, path);
result.name = file;
return result;
}
template<class T> void
dump_array(String_Builder *sb, Array<T> *arr){
sb->append_data(arr, sizeof(*arr));
sb->append_data(arr->data, sizeof(T)*arr->len);
}
function void
dump_bitmap_image(String_Builder *sb, Bitmap *bm){
sb->append_data(bm->pixels, sizeof(U32)*bm->x*bm->y);
}
function B32
_os_write_file(String file, String data, B32 append = false) {
B32 result = false;
DWORD access = GENERIC_WRITE;
DWORD creation_disposition = CREATE_ALWAYS;
if (append) {
access = FILE_APPEND_DATA;
creation_disposition = OPEN_ALWAYS;
}
// @Todo(Krzosa): Unicode
HANDLE handle = CreateFileA((const char *)file.str, access, 0, NULL, creation_disposition, FILE_ATTRIBUTE_NORMAL, NULL);
if (handle != INVALID_HANDLE_VALUE) {
DWORD bytes_written = 0;
// @Todo: can only read 32 byte size files?
assert_msg(data.len == (U32)data.len, "Max data size os_write can handle is 32 bytes, data to write is larger then 32 bytes!");
B32 error = WriteFile(handle, data.str, (U32)data.len, &bytes_written, NULL);
if (error == false) log_error("Failed to write to file: %Q", file);
else {
if (bytes_written != data.len) log_error("Failed to write to file: %Q, mismatch between length requested to write and length written", file);
else result = true;
}
CloseHandle(handle);
}
else {
log_error("File not found when trying to write: %Q", file);
}
return result;
}
function B32
os_write_file2(String file, String data) {
return _os_write_file(file, data, false);
}
function B32
os_append_file(String file, String data) {
return _os_write_file(file, data, true);
}
function void
dump_obj_to_file(Obj *obj, String out_name){
obj->vertices.allocator = 0;
obj->vertices.cap = obj->vertices.len;
obj->texture_coordinates.allocator = 0;
obj->texture_coordinates.cap = obj->texture_coordinates.len;
obj->normals.allocator = 0;
obj->normals.cap = obj->normals.len;
obj->mesh.allocator = 0;
obj->mesh.cap = obj->mesh.len;
obj->materials.allocator = 0;
obj->materials.cap = obj->materials.len;
For(obj->mesh){
it.indices.allocator = 0;
it.indices.cap = it.indices.len;
}
Scratch arena;
String_Builder sb = string_builder_make(arena, mib(4));
sb.append_data(obj, sizeof(Obj));
sb.append_data(obj->name.str, obj->name.len);
sb.append_data(obj->texture_coordinates.data, obj->texture_coordinates.len*sizeof(Vec2));
sb.append_data(obj->vertices.data, obj->vertices.len*sizeof(Vec3));
sb.append_data(obj->normals.data, obj->normals.len*sizeof(Vec3));
sb.append_data(obj->mesh.data, obj->mesh.len*sizeof(Obj_Mesh));
sb.append_data(obj->materials.data, obj->materials.len*sizeof(Obj_Material));
For(obj->mesh){
sb.append_data(it.indices.data, sizeof(Obj_Index)*it.indices.len);
}
For(obj->materials){
sb.append_data(&it, sizeof(Obj_Material));
dump_bitmap_image(&sb, &it.texture_ambient);
dump_bitmap_image(&sb, &it.texture_diffuse);
dump_bitmap_image(&sb, &it.texture_dissolve);
dump_bitmap_image(&sb, &it.texture_displacement);
}
String result = string_flatten(arena, &sb);
os_write_file2(out_name, result);
}

3
os_windows_multimedia.cpp
View File

@@ -525,8 +525,7 @@ api void os_set_window_size(S32 x, S32 y) {
Vec2I border = get_border_size(w32(os).window);
int actual_width = (int)(x + border.x);
int actual_height = (int)(y + border.y);
bool result =
SetWindowPos(w32(os).window, 0, 0, 0, actual_width, actual_height, SWP_NOMOVE | SWP_NOOWNERZORDER);
bool result = SetWindowPos(w32(os).window, 0, 0, 0, actual_width, actual_height, SWP_NOMOVE | SWP_NOOWNERZORDER);
assert_msg(result, "SetWindowPos returned invalid value");
os.window_size = get_window_size(w32(os).window);
}

242
ui.cpp
View File

@@ -1,242 +0,0 @@
enum UIWidgetKind {
UIWidgetKind_None,
UIWidgetKind_Boolean,
UIWidgetKind_Image,
UIWidgetKind_Label,
UIWidgetKind_Option,
UIWidgetKind_Signal,
UIWidgetKind_Group,
};
#define UI_SIGNAL_CALLBACK(name) void name()
typedef UI_SIGNAL_CALLBACK(UISignalCallback);
struct UISetup {
UIWidgetKind kind;
String text;
union {
void *v;
Bitmap *image;
B32 *b32;
String *label;
S32 *option;
UISignalCallback *signal_callback;
};
S32 option_max;
};
#define UI_BOOL(text, x) {UIWidgetKind_Boolean,text,(void*)(x)}
#define UI_IMAGE(x) {UIWidgetKind_Image,{},(void*)(x)}
#define UI_LABEL(x) {UIWidgetKind_Label,{},(void*)(x)}
#define UI_OPTION(text,x,option_max){UIWidgetKind_Option,text,(void*)(x),(option_max)}
#define UI_SIGNAL(text,x){UIWidgetKind_Signal,text,(void*)(x)}
struct UIWidget {
UIWidgetKind kind;
UIWidget *next;
UIWidget *prev;
UIWidget *first_child;
UIWidget *last_child;
String text;
Vec2 size;
S32 option_max;
union {
Bitmap *image;
B32 *b32;
String *label;
S32 *option;
UISignalCallback *signal_callback;
} ptr;
};
struct UI : UIWidget {
Arena arena;
UIWidget *hot;
UIWidget *active;
};
function UIWidget *ui_new_widget(Allocator *arena, UIWidgetKind kind) {
UIWidget *result = exp_alloc_type(arena, UIWidget, AF_ZeroMemory);
result->kind = kind;
return result;
}
function void ui_push_child(UIWidget *widget, UIWidget *child) {
DLLQueueAdd(widget->first_child, widget->last_child, child);
}
function UIWidget *ui_push_child(Arena *arena, UIWidget *widget, UIWidgetKind kind) {
UIWidget *result = ui_new_widget(arena, kind);
ui_push_child(widget, result);
return result;
}
function UIWidget *ui_push_image(Arena *arena, UIWidget *widget, Bitmap *img) {
UIWidget *result = ui_push_child(arena, widget, UIWidgetKind_Image);
result->ptr.image = img;
F32 ratio = (F32)result->ptr.image->x / (F32)result->ptr.image->y;
result->size.y = 64;
result->size.x = 64 * ratio;
return result;
}
function UIWidget *ui_push_bool(Arena *arena, UIWidget *widget, String string, B32 *b32) {
UIWidget *result = ui_push_child(arena, widget, UIWidgetKind_Boolean);
result->text = string;
result->ptr.b32 = b32;
return result;
}
function UIWidget *ui_push_string(Arena *arena, UIWidget *widget, String *string) {
UIWidget *result = ui_push_child(arena, widget, UIWidgetKind_Label);
result->ptr.label = string;
return result;
}
function UIWidget *ui_push_option(Arena *arena, UIWidget *widget, String string, S32 *option, S32 option_max) {
UIWidget *result = ui_push_child(arena, widget, UIWidgetKind_Option);
result->text = string;
result->ptr.option = option;
result->option_max = option_max;
return result;
}
function UIWidget *ui_push_signal(Arena *arena, UIWidget *widget, String string, UISignalCallback *callback) {
UIWidget *result = ui_push_child(arena, widget, UIWidgetKind_Signal);
result->text = string;
result->ptr.signal_callback = callback;
return result;
}
function UI ui_make(UISetup *setup, U64 len) {
UI result = {};
arena_init(&result.arena, "UI_Arena"_s);
UIWidget *parent = &result;
for (UISetup *s = setup; s != (setup+len); s++) {
switch (s->kind) {
case UIWidgetKind_Image: {
ui_push_image(&result.arena, parent, s->image);
} break;
case UIWidgetKind_Boolean: {
ui_push_bool(&result.arena, parent, s->text, s->b32);
} break;
case UIWidgetKind_Label: {
ui_push_string(&result.arena, parent, s->label);
} break;
case UIWidgetKind_Option: {
ui_push_option(&result.arena, parent, s->text, s->option, s->option_max);
} break;
case UIWidgetKind_Signal: {
ui_push_signal(&result.arena, parent, s->text, s->signal_callback);
} break;
invalid_default_case;
}
}
return result;
}
function B32 ui_mouse_test(UI *ui, UIWidget *w, Vec4 rect) {
B32 result = false;
if (os.mouse_pos.x > rect.x && os.mouse_pos.x < rect.x + rect.width &&
os.mouse_pos.y > rect.y && os.mouse_pos.y < rect.y + rect.height) {
ui->hot = w;
if (os.key[Key_MouseLeft].down) {
ui->active = w;
}
}
else if (w == ui->hot) {
ui->hot = 0;
}
if (os.key[Key_MouseLeft].released) {
if (ui->active == w) {
if (ui->hot == w)
result = true;
ui->active = 0;
}
}
return result;
}
function void ui_end_frame(Bitmap *dst, UI *ui, Font *font) {
Scratch scratch;
Vec2 pos = vec2(0, (F32)dst->y);
for (UIWidget *w = ui->first_child; w; w = w->next) {
Vec4 rect = {};
switch (w->kind) {
case UIWidgetKind_Image: {
pos.y -= w->size.y;
rect = vec4(pos, w->size);
ui_mouse_test(ui, w, rect);
String string = string_fmt(scratch, "%d %d", w->ptr.image->x, w->ptr.image->y);
draw_string(dst, font, string, pos);
draw_bitmap(dst, w->ptr.image, pos, w->size);
if (ui->active == w) {
F32 ratio = (F32)w->ptr.image->y / (F32)w->ptr.image->x;
w->size.x -= os.delta_mouse_pos.x;
w->size.y = w->size.x * ratio;
}
if (ui->hot == w) {
draw_rect(dst, rect.x, rect.y, rect.width, rect.height, vec4(1, 1, 1, 0.1f));
}
} break;
case UIWidgetKind_Boolean: {
pos.y -= font->height;
Vec4 color = vec4(0, 0, 0, 1);
String string = string_fmt(scratch, "%s %d", w->text, *w->ptr.b32);
rect = get_string_rect(font, string, pos);
B32 clicked = ui_mouse_test(ui, w, rect);
if (clicked) *w->ptr.b32 = !*w->ptr.b32;
if (ui->hot == w) {
color = vec4(0.4f, 0.4f, 0.4f, 1.f);
}
rect.y = rect.y-font->line_advance / 5;
draw_rect(dst, rect.x, rect.y, rect.width, rect.height, color);
rect = draw_string(dst, font, string, pos);
pos.y -= rect.height - font->height;
} break;
case UIWidgetKind_Label: {
pos.y -= font->height;
rect = draw_string(dst, font, *w->ptr.label, pos);
pos.y -= rect.height - font->height;
} break;
case UIWidgetKind_Option: {
pos.y -= font->height;
Vec4 color = vec4(0, 0, 0, 1);
String string = string_fmt(scratch, "%Q %Q", w->text, *w->ptr.option);
rect = get_string_rect(font, string, pos);
B32 clicked = ui_mouse_test(ui, w, rect);
if (clicked) {
*w->ptr.b32 = (*w->ptr.b32+1) % w->option_max;
}
if (ui->hot == w) {
color = vec4(0.4f, 0.4f, 0.4f, 1.f);
}
rect.y = rect.y-font->line_advance / 5;
draw_rect(dst, rect.x, rect.y, rect.width, rect.height, color);
rect = draw_string(dst, font, string, pos);
pos.y -= rect.height - font->height;
} break;
case UIWidgetKind_Signal: {
pos.y -= font->height;
Vec4 color = vec4(0, 0, 0, 1);
String string = string_fmt(scratch, "%Q", w->text);
rect = get_string_rect(font, string, pos);
B32 clicked = ui_mouse_test(ui, w, rect);
if (clicked) {
w->ptr.signal_callback();
}
if (ui->hot == w) {
color = vec4(0.4f, 0.4f, 0.4f, 1.f);
}
rect.y = rect.y-font->line_advance / 5;
draw_rect(dst, rect.x, rect.y, rect.width, rect.height, color);
rect = draw_string(dst, font, string, pos);
pos.y -= rect.height - font->height;
} break;
invalid_default_case;
}
}
}

54
vec.cpp
View File

@@ -1,54 +0,0 @@
#include <intrin.h>
union Vec8{
__m256 simd;
F32 e[8];
Vec4 v4[2];
force_inline F32 &operator[](S64 i){ return e[i]; }
};
force_inline Vec8 floor8(Vec8 v){ return {_mm256_floor_ps(v.simd)}; }
force_inline Vec8 loadu8(void *m){ return {_mm256_loadu_ps((const float *)m)}; }
force_inline Vec8 vec8(F32 x){return {_mm256_set1_ps(x)}; }
force_inline Vec8 vec8(F32 a, F32 b, F32 c, F32 d, F32 e, F32 f, F32 g, F32 h){ return {_mm256_set_ps(h, g, f, e, d, c, b, a)}; }
force_inline Vec8 operator+(Vec8 a, Vec8 b){ return {_mm256_add_ps(a.simd, b.simd)}; }
force_inline Vec8 operator-(Vec8 a, Vec8 b){ return {_mm256_sub_ps(a.simd, b.simd)}; }
force_inline Vec8 operator*(Vec8 a, Vec8 b){ return {_mm256_mul_ps(a.simd, b.simd)}; }
force_inline Vec8 operator/(Vec8 a, Vec8 b){ return {_mm256_div_ps(a.simd, b.simd)}; }
force_inline Vec8 operator>=(Vec8 a, Vec8 b){ return {_mm256_cmp_ps(a.simd, b.simd, _CMP_GE_OQ)}; }
force_inline Vec8 operator<=(Vec8 a, Vec8 b){ return {_mm256_cmp_ps(a.simd, b.simd, _CMP_LE_OQ)}; }
force_inline Vec8 operator<(Vec8 a, Vec8 b){ return {_mm256_cmp_ps(a.simd, b.simd, _CMP_LT_OQ)}; }
force_inline Vec8 operator>(Vec8 a, Vec8 b){ return {_mm256_cmp_ps(a.simd, b.simd, _CMP_GT_OQ)}; }
force_inline Vec8 operator&(Vec8 a, Vec8 b){ return {_mm256_and_ps(a.simd, b.simd)}; }
force_inline Vec8 operator+=(Vec8 &a, Vec8 b){ a = a + b; return a; }
force_inline Vec8 operator-=(Vec8 &a, Vec8 b){ a = a - b; return a; }
force_inline Vec8 operator*=(Vec8 &a, Vec8 b){ a = a * b; return a; }
force_inline Vec8 operator/=(Vec8 &a, Vec8 b){ a = a / b; return a; }
union Vec8I{
__m256i simd;
S32 e[8];
Vec4I v4[2];
force_inline S32 &operator[](S64 i){ return e[i]; }
};
force_inline Vec8I vec8i(S32 x){return {_mm256_set1_epi32(x)}; }
force_inline Vec8I vec8i(S32 a, S32 b, S32 c, S32 d, S32 e, S32 f, S32 g, S32 h){ return {_mm256_set_epi32(h, g, f, e, d, c, b, a)}; }
force_inline Vec8I operator>(Vec8I a, Vec8I b){
return {_mm256_cmpgt_epi32(a.simd, b.simd)};
}
force_inline Vec8I operator>>(Vec8I a, U8 v){ return {_mm256_srai_epi32(a.simd, v)}; }
force_inline Vec8I operator&(Vec8I a, Vec8I b){ return {_mm256_and_si256(a.simd, b.simd)}; }
force_inline Vec8I operator+(Vec8I a, Vec8I b){ return {_mm256_add_epi32(a.simd, b.simd)}; }
force_inline Vec8I operator-(Vec8I a, Vec8I b){ return {_mm256_sub_epi32(a.simd, b.simd)}; }
force_inline Vec8I operator*(Vec8I a, Vec8I b){
return {_mm256_mullo_epi32(a.simd, b.simd)}; //_mm256_mul_epi32
}
// Vec8I operator/(Vec8I a, Vec8I b){ return {_mm256_div_epi32(a.simd, b.simd)}; }
force_inline Vec8I operator+=(Vec8I &a, Vec8I b){ return a + b; }
force_inline Vec8I convert_vec8_to_vec8i(Vec8 v){ return Vec8I{_mm256_cvtps_epi32(v.simd)}; }
force_inline Vec8 convert_vec8i_to_vec8(Vec8I v){ return {_mm256_cvtepi32_ps(v.simd)}; }

100
work_queue.cpp
View File

@@ -1,100 +0,0 @@
// @Section: Work Queue
#define WORK_QUEUE_CALLBACK(name) void name(void *data)
typedef WORK_QUEUE_CALLBACK(WorkQueueCallback);
struct WorkQueueEntry {
WorkQueueCallback *callback;
void *data;
};
struct WorkQueue {
WorkQueueEntry entries[256];
S64 volatile index_to_write;
S64 volatile index_to_read;
S64 volatile completion_index;
S64 volatile completion_goal;
HANDLE semaphore;
};
struct ThreadStartupInfo {
DWORD thread_id;
S32 thread_index;
WorkQueue *queue;
};
S64 atomic_increment(volatile S64 *i){
return InterlockedIncrement64(i);
}
S64 atomic_compare_and_swap(volatile S64 *dst, S64 exchange, S64 comperand){
return InterlockedCompareExchange64(dst, exchange, comperand);
}
void push_work(WorkQueue *wq, void *data, WorkQueueCallback *callback) {
U32 new_index = (wq->index_to_write + 1) % buff_cap(wq->entries);
assert(new_index != wq->index_to_read);
WorkQueueEntry *entry = wq->entries + wq->index_to_write;
entry->data = data;
entry->callback = callback;
wq->completion_goal+=1;
_WriteBarrier();
wq->index_to_write = new_index;
ReleaseSemaphore(wq->semaphore, 1, 0);
}
bool try_doing_work(WorkQueue *wq) {
bool should_sleep = false;
S64 original_index_to_read = wq->index_to_read;
S64 new_index_to_read = (original_index_to_read + 1) % buff_cap(wq->entries);
if(original_index_to_read != wq->index_to_write) {
S64 index = atomic_compare_and_swap(&wq->index_to_read, new_index_to_read, original_index_to_read);
if(index == original_index_to_read) {
WorkQueueEntry *entry = wq->entries + index;
entry->callback(entry->data);
atomic_increment(&wq->completion_index);
}
}
else {
should_sleep = true;
}
return should_sleep;
}
DWORD WINAPI thread_proc(LPVOID param) {
auto ti = (ThreadStartupInfo *)param;
Thread_Ctx ctx = {};
ctx.thread_index = ti->thread_index;
for(;;) {
if(try_doing_work(ti->queue)) {
WaitForSingleObject(ti->queue->semaphore, INFINITE);
}
}
}
void init_work_queue(WorkQueue *queue, U32 thread_count, ThreadStartupInfo *info) {
queue->index_to_read = 0;
queue->index_to_write = 0;
queue->completion_index = 0;
queue->completion_goal = 0;
queue->semaphore = CreateSemaphoreExA(0, 0, thread_count, 0, 0, SEMAPHORE_ALL_ACCESS);
assert_msg(queue->semaphore != INVALID_HANDLE_VALUE, "Failed to create semaphore");
for(U32 i = 0; i < thread_count; i++) {
ThreadStartupInfo *ti = info + i;
ti->thread_index = i;
ti->queue = queue;
HANDLE thread_handle = CreateThread(0, 0, thread_proc, ti, 0, &ti->thread_id);
assert_msg(thread_handle != INVALID_HANDLE_VALUE, "Failed to create thread");
CloseHandle(thread_handle);
}
}
void wait_until_completion(WorkQueue *wq) {
while(wq->completion_goal != wq->completion_index) {
try_doing_work(wq);
}
}