colorchord/attic/really_lowspec_basic_tests/test_using_work_selection_table.c
2023-01-20 03:13:34 -05:00

322 lines
8.2 KiB
C

/*
An experiment in very, very low-spec ColorChord. This technique foregoes
multiplies.
Approach 3: Based on Approach 2, but using a work selection table.
This won't work for the ch32v003, since the minimum practical table for 6
octaves at 24BPERO with 12kSPS and bottom freq of 22.5 is about 80kB.
*/
#include <stdio.h>
#include <math.h>
#define CNFG_IMPLEMENTATION
#define CNFGOGL
#include "rawdraw_sf.h"
#include "os_generic.h"
uint32_t EHSVtoHEX( uint8_t hue, uint8_t sat, uint8_t val );
int lastx = 200;
int lasty = 1000;
#define FSPS 16000
#define OCTAVES 6
#define BPERO 24
#define BASE_FREQ 22.5
#define QUADRATURE_STEP_DENOMINATOR 16384
// Careful: It makes a lot of sense to explore these relationships.
#define SAMPLE_Q 4
#define MAG_IIR 0
#define COMPLEX_IIR 2
#define RUNNING_IIR 31
#define TEST_SAMPLES 256
int sineshape;
// This will sort the head node back down into the heap, so the heap will
// remain a min-heap. This is done in log(n) time. But, with our data, it
// experimentally only needs to run for 6.47 iterations per call on average
// assuming 24 BPERO and 6 OCTAVES.
int main()
{
int16_t samples[TEST_SAMPLES];
int i;
CNFGSetup( "Example App", 1024, 768 );
// Not size of table (that's usually larger) but # of samples
// to record the work instructions for.
#define WORKLOOP 6144
// Make a running counter to count up by this amount every cycle.
// If the new number > 2048, then perform a quadrature step.
double spacings[BPERO*OCTAVES];
double runningspace[BPERO*OCTAVES];
for( i = 0; i < BPERO*OCTAVES; i++ )
{
double freq = pow( 2, (double)i / (double)BPERO ) * (BASE_FREQ/2.0);
double pfreq = freq;
double spacing = (FSPS / 2) / pfreq / 4;
// make spacings line up to a denominator of workloop, this makes it
// so you don't get a werid jump at the end of the work loop.
double wdt = WORKLOOP / spacing;
printf( "%f %f %f\n", wdt, spacing, WORKLOOP / (double)((int)wdt+0.5) );
wdt = (int)(wdt+0.5);
spacing = WORKLOOP / wdt;
spacings[i] = spacing;
}
uint8_t worktable[WORKLOOP*BPERO*OCTAVES];
int worktablelen = 0;
for( i = 0; i < WORKLOOP; i++ )
{
int j;
for( j = 0; j < BPERO*OCTAVES; j++ )
{
if( i >= runningspace[j] )
{
runningspace[j] += spacings[j];
worktable[worktablelen++] = j;
}
}
worktable[worktablelen++] = 255;
}
// This is for timing. Not accumulated data.
uint8_t quadrature_state[BPERO*OCTAVES] = { 0 };
uint32_t last_accumulated_value[BPERO*OCTAVES*2] = { 0 };
int32_t real_imaginary_running[BPERO*OCTAVES*2] = { 0 };
uint32_t sample_accumulator = 0;
int32_t time_accumulator = 0;
int32_t qcount[BPERO*OCTAVES] = { 0 };
int32_t magsum[BPERO*OCTAVES] = { 0 };
int frameno = 0;
double dLT = OGGetAbsoluteTime();
double ToneOmega = 0;
int worktableplace = 0;
while( CNFGHandleInput() )
{
CNFGClearFrame();
frameno++;
float freq =
//pow( 2, (frameno%600)/100.0 ) * 25;
pow( 2, (lastx)/100.0 ) * lastx;
//101;
for( i = 0; i < TEST_SAMPLES; i++ )
{
samples[i] = lasty/5 + sin( ToneOmega ) * 127;// + (rand()%128)-64;
ToneOmega += 1 / (double)FSPS * (double)freq * 3.14159 * 2.0;
}
char cts[1024];
sprintf( cts, "%f %d SINESHAPE: %d WT %d", freq, time_accumulator, sineshape , worktablelen);
CNFGColor( 0xffffffff );
CNFGPenX = 2;
CNFGPenY = 2;
CNFGDrawText( cts, 2 );
while( OGGetAbsoluteTime() < dLT + TEST_SAMPLES / (double)FSPS );
dLT += TEST_SAMPLES / (double)FSPS;
#define WATCHBIN -1
for( i = 0; i < TEST_SAMPLES; i++ )
{
sample_accumulator += samples[i];
time_accumulator += QUADRATURE_STEP_DENOMINATOR;
while( 1 )
{
int wtp = worktable[worktableplace];
worktableplace = worktableplace + 1;
if( worktableplace >= worktablelen ) worktableplace = 0;
if( wtp == 255 ) break;
// Event has occurred.
int binno = wtp;
//int j;
//for( j = 0; j < OCTAVES*BPERO; j++ ) printf( "[%d %d]", next_heap_events[j*2+0], next_heap_events[j*2+1] );
//printf( "\n" );
int qstate = quadrature_state[binno] = ( quadrature_state[binno] + 1 ) % 4;
int last_q_bin = (binno * 2) + ( qstate & 1 );
int delta;
if( !sineshape )
{
delta = sample_accumulator - last_accumulated_value[last_q_bin];
last_accumulated_value[last_q_bin] = sample_accumulator;
}
else
{
// TESTING: Sine Shape - this only integrates bumps for sin/cos
// instead of full triangle waves.
// Observation: For higher frequency bins, this seems to help
// a lot.
// side-benefit, this takes less RAM.
// BUT BUT! It messes with lower frequencies, making them uglier.
delta = sample_accumulator - last_accumulated_value[binno];
last_accumulated_value[binno] = sample_accumulator;
delta *= 1.4; // Just to normalize the results of the test (not for production)
}
if( binno == WATCHBIN )
printf( "Delta: %d\n", delta );
// Qstate =
// (0) = +Cos, (1) = +Sin, (2) = -Cos, (3) = -Sin
if( qstate & 2 ) delta *= -1;
// Update real and imaginary components with delta.
int running = real_imaginary_running[last_q_bin];
running = running - (running>>RUNNING_IIR) + delta;
real_imaginary_running[last_q_bin] = running;
int q = ++qcount[binno];
if( q == SAMPLE_Q ) // Effective Q factor.
{
qcount[binno] = 0;
int newmagR = real_imaginary_running[(binno * 2)];
int newmagI = real_imaginary_running[(binno * 2)+1];
real_imaginary_running[(binno * 2)] = newmagR - (newmagR>>COMPLEX_IIR);
real_imaginary_running[(binno * 2)+1] = newmagI - (newmagI>>COMPLEX_IIR);
// Super-cheap, non-multiply, approximate complex vector magnitude calculation.
newmagR = (newmagR<0)?-newmagR:newmagR;
newmagI = (newmagI<0)?-newmagI:newmagI;
int newmag =
//sqrt(newmagR*newmagR + newmagI*newmagI );
newmagR > newmagI ? newmagR + (newmagI>>1) : newmagI + (newmagR>>1);
int lastmag = magsum[binno];
magsum[binno] = lastmag - (lastmag>>MAG_IIR) + newmag;
}
}
}
int lx, ly;
for( i = 0; i < BPERO*OCTAVES; i++ )
{
CNFGColor( (EHSVtoHEX( (i * 256 / BPERO)&0xff, 255, 255 ) << 8) | 0xff );
float real = real_imaginary_running[i*2+0];
float imag = real_imaginary_running[i*2+1];
float mag = sqrt( real * real + imag * imag );
mag = (float)magsum[i] * pow( 2, i / (double)BPERO );
int y = 768 - ((int)(mag / FSPS * 10) >> MAG_IIR);
if( i ) CNFGTackSegment( i*4, y, lx*4, ly );
lx = i; ly= y;
//printf( "%d %d\n", real_imaginary_running[i*2+0], real_imaginary_running[i*2+1] );
}
CNFGSwapBuffers();
}
}
void HandleKey( int keycode, int bDown ) { if( keycode == 'a' && bDown ) sineshape = !sineshape; }
void HandleButton( int x, int y, int button, int bDown ) { }
void HandleMotion( int x, int y, int mask ) { lastx = x; lasty = y; }
void HandleDestroy() { }
uint32_t EHSVtoHEX( uint8_t hue, uint8_t sat, uint8_t val )
{
#define SIXTH1 43
#define SIXTH2 85
#define SIXTH3 128
#define SIXTH4 171
#define SIXTH5 213
uint16_t or = 0, og = 0, ob = 0;
hue -= SIXTH1; //Off by 60 degrees.
//TODO: There are colors that overlap here, consider
//tweaking this to make the best use of the colorspace.
if( hue < SIXTH1 ) //Ok: Yellow->Red.
{
or = 255;
og = 255 - ((uint16_t)hue * 255) / (SIXTH1);
}
else if( hue < SIXTH2 ) //Ok: Red->Purple
{
or = 255;
ob = (uint16_t)hue*255 / SIXTH1 - 255;
}
else if( hue < SIXTH3 ) //Ok: Purple->Blue
{
ob = 255;
or = ((SIXTH3-hue) * 255) / (SIXTH1);
}
else if( hue < SIXTH4 ) //Ok: Blue->Cyan
{
ob = 255;
og = (hue - SIXTH3)*255 / SIXTH1;
}
else if( hue < SIXTH5 ) //Ok: Cyan->Green.
{
og = 255;
ob = ((SIXTH5-hue)*255) / SIXTH1;
}
else //Green->Yellow
{
og = 255;
or = (hue - SIXTH5) * 255 / SIXTH1;
}
uint16_t rv = val;
if( rv > 128 ) rv++;
uint16_t rs = sat;
if( rs > 128 ) rs++;
//or, og, ob range from 0...255 now.
//Need to apply saturation and value.
or = (or * val)>>8;
og = (og * val)>>8;
ob = (ob * val)>>8;
//OR..OB == 0..65025
or = or * rs + 255 * (256-rs);
og = og * rs + 255 * (256-rs);
ob = ob * rs + 255 * (256-rs);
//printf( "__%d %d %d =-> %d\n", or, og, ob, rs );
or >>= 8;
og >>= 8;
ob >>= 8;
return or | (og<<8) | ((uint32_t)ob<<16);
}