MandAzur/src/juliashader.cpp

167 lines
3.3 KiB
C++

#include <azur/gint/render.h>
#include "fractalshaders.h"
#include "iteration.h"
#include <cstdlib>
#include <cstdio>
#include <gint/rtc.h>
#include <gint/usb.h>
#include <gint/usb-ff-bulk.h>
#include <num/num.h>
uint8_t AZRP_SHADER_JULIA = -1;
__attribute__((constructor))
static void register_shader(void)
{
extern azrp_shader_t azrp_shader_julia;
AZRP_SHADER_JULIA = azrp_register_shader(azrp_shader_julia);
}
void azrp_shader_julia_configure(void)
{
azrp_set_uniforms(AZRP_SHADER_JULIA, (void *)azrp_width);
}
void azrp_julia_USBDEBUG( void )
{
char texttosend[1024];
sprintf( texttosend, "data" );
usb_fxlink_text(texttosend, 0);
}
void azrp_julia_init( )
{
}
void azrp_julia_close( void )
{
}
struct command {
uint8_t shader_id;
uint8_t current_frag;
libnum::num xmin;
libnum::num xmax;
libnum::num ymin;
libnum::num ymax;
libnum::num cr;
libnum::num ci;
libnum::num xinc;
libnum::num yinc;
};
void azrp_julia( float cr, float ci )
{
prof_enter(azrp_perf_cmdgen);
struct command cmd;
cmd.shader_id = AZRP_SHADER_JULIA;
cmd.current_frag = 0;
cmd.xmin = libnum::num(-1.98f);
cmd.xmax = libnum::num(1.98f);
cmd.ymin = libnum::num(-1.12f);
cmd.ymax = libnum::num(1.12f);
cmd.xinc = libnum::num((2*1.98f)/azrp_width);
cmd.yinc = libnum::num((2*1.12f)/azrp_height);
cmd.cr = libnum::num(cr);
cmd.ci = libnum::num(ci);
int fragmin = 0;
int fragcount = (azrp_height >> 4) + 1;
azrp_queue_command(&cmd, sizeof cmd, fragmin, fragcount);
prof_leave(azrp_perf_cmdgen);
}
void azrp_shader_julia( void *uniforms, void *comnd, void *fragment )
{
struct command *cmd = (struct command *) comnd;
uint16_t *frag = (uint16_t *) fragment;
int maxiter = 64;
libnum::num zr, zr2;
libnum::num zi, zi2;
libnum::num cr = cmd->cr;
libnum::num ci = cmd->ci;
libnum::num zrsave = cmd->xmin;
libnum::num zisave = cmd->ymax - libnum::num(cmd->current_frag*16)*cmd->yinc;
int u;
int offset = 0;
for( int j=0; j<16; j++)
{
for( int i=0; i<azrp_width; i++ )
{
zr = zrsave;
zi = zisave;
u = quadratic_iteration_32( cr.v, ci.v, zr.v, zi.v, (2*2)<<16, maxiter );
if (u==0) frag[offset + i] = C_RGB(0,0,0);
else
{
int CR, CG, CB;
if (u<maxiter/4)
{
CR = 0;
CG = u*4;
CB = 255;
}
else if (u<2*maxiter/4)
{
CR = 0;
CG = 255;
CB = 255-(u-maxiter)/4;
}
else if (u<3*maxiter/4)
{
CR = (u-2*maxiter)/4;
CG = 255;
CB = 0;
}
else if (u<4*maxiter/4)
{
CR = 255;
CG = 255-(u-3*maxiter)/4;
CB = 0;
}
frag[offset + i] = C_RGB(CR,CG,CB);
}
zrsave += cmd->xinc;
}
zrsave = cmd->xmin;
zisave -= cmd->yinc;
offset += azrp_width;
}
cmd->current_frag++;
}