v4k-git-backup/demos/99-temperature.c

92 lines
3.2 KiB
C
Raw Normal View History

2023-09-09 09:24:54 +00:00
#define COOK_ON_DEMAND 1
2023-09-09 00:36:52 +00:00
#include "v4k.h"
2023-09-09 13:16:05 +00:00
int TEMP_GPU = 1;
2023-09-09 00:36:52 +00:00
2023-09-09 13:16:05 +00:00
#define TEX_WIDTH 1024
2023-09-09 00:36:52 +00:00
void temp_calc(vec4 *pixels){
// flood it
for (int y = 1; y < TEX_WIDTH-1; y++){
for (int x = 1; x < TEX_WIDTH-1; x++){
int idx = (y * TEX_WIDTH) + x;
vec4 *m = &pixels[idx];
vec4 sum = vec4(0,0,0,0);
for (int cy = -1; cy <= 1; ++cy) {
for (int cx = -1; cx <= 1; ++cx) {
sum = add4(sum, pixels[((y+cy)*TEX_WIDTH)+(x+cx)]);
}
}
*m = scale4(sum,1/9.0f);
}
}
}
int main() {
2023-09-09 13:16:05 +00:00
window_create(50, WINDOW_SQUARE|WINDOW_VSYNC_DISABLED);
2023-09-09 00:36:52 +00:00
window_title(__FILE__);
2023-09-10 13:03:01 +00:00
window_fps_unlock();
2023-09-09 00:36:52 +00:00
texture_t tex;
vec4 *img = REALLOC(0, TEX_WIDTH*TEX_WIDTH*sizeof(vec4));
for (int i=0; i <TEX_WIDTH*TEX_WIDTH; i++){
img[i] = vec4(0.3,0.3,0.3,1);
}
2023-09-09 13:16:05 +00:00
2023-09-09 00:36:52 +00:00
tex = texture_create(TEX_WIDTH, TEX_WIDTH, 4, img, TEXTURE_LINEAR|TEXTURE_FLOAT);
unsigned comp = compute(vfs_read("shaders/temperature.glsl"));
shader_bind(comp);
shader_image(tex, 0, 0, 0, READ_WRITE);
while ( window_swap() && !input_down(KEY_ESC) ){
2023-09-09 13:16:05 +00:00
if (input_down(KEY_F5)) window_reload();
if (input_down(KEY_F8)) TEMP_GPU ^= 1;
if (TEMP_GPU) {
glActiveTexture(GL_TEXTURE0);
glBindTexture(GL_TEXTURE_2D, tex.id);
glGetTexImage(GL_TEXTURE_2D, 0, GL_RGBA, GL_FLOAT, img);
}
for (int i=0; i <TEX_WIDTH*64; i++){
2023-09-09 00:36:52 +00:00
img[i] = vec4(0,0,1,1);
2023-09-09 13:16:05 +00:00
img[(TEX_WIDTH - 64) * TEX_WIDTH+i] = vec4(1,0,0,1);
2023-09-09 00:36:52 +00:00
}
int enabled = !ui_active() && !ui_hover();
vec4 mouse = enabled ? vec4(input(MOUSE_X),input(MOUSE_Y),input(MOUSE_L),input(MOUSE_R)) : vec4(0,0,0,0); // x,y,l,r
2023-09-09 13:16:05 +00:00
int strong = input(KEY_LSHIFT)?5:1;
int x = (int)clampf(floorf(TEX_WIDTH/(float)window_width() * mouse.x), 1, TEX_WIDTH-2);
int y = (int)clampf(floorf(TEX_WIDTH/(float)window_height() * mouse.y), 1, TEX_WIDTH-2);
if (mouse.z || mouse.w){
for (int cy = -5*strong; cy <= 5*strong; ++cy) {
for (int cx = -5*strong; cx <= 5*strong; ++cx) {
int px = (int)clampf(cx+x, 1, TEX_WIDTH-2);
int py = (int)clampf(cy+y, 1, TEX_WIDTH-2);
img[(py*TEX_WIDTH)+px] = vec4(mouse.z,0,mouse.w,1);
}
}
2023-09-09 00:36:52 +00:00
}
2023-09-09 13:16:05 +00:00
if (TEMP_GPU) {
glBindTexture(GL_TEXTURE_2D, tex.id);
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA32F, TEX_WIDTH, TEX_WIDTH, 0, GL_RGBA, GL_FLOAT, img);
shader_bind(comp);
compute_dispatch(TEX_WIDTH/16, TEX_WIDTH/16, 1);
image_write_barrier();
} else {
temp_calc(img);
texture_update(&tex, TEX_WIDTH, TEX_WIDTH, 4, img, TEXTURE_LINEAR|TEXTURE_FLOAT);
2023-09-09 00:36:52 +00:00
}
fullscreen_quad_rgb(tex, 2.2);
2023-09-09 13:16:05 +00:00
ddraw_text2d(vec2(0,0), va("mode: %s\nimage res: %d\ninputs: %.01f %.01f %.01f %.01f %s", TEMP_GPU?"GPU compute":"CPU", TEX_WIDTH, mouse.x, mouse.y, mouse.z, mouse.w, strong>1?"lshift":""));
ddraw_text2d(vec2(0,window_height() - 20), va("delta: %.2f ms", window_delta()*1000));
2023-09-09 00:36:52 +00:00
}
return 0;
}