[fishladder.git] / vg / vg_audio.h

// Copyright (C) 2021 Harry Godden (hgn) - All Rights Reserved

#define MINIAUDIO_IMPLEMENTATION
#include "dr_soft/miniaudio.h"

#define STB_VORBIS_MAX_CHANNELS 2
#include "stb/stb_vorbis.h"

#define SFX_MAX_SYSTEMS 16
#define SFX_FLAG_ONESHOT 0x1
#define SFX_FLAG_STEREO  0x2
#define FADEOUT_LENGTH 441
#define FADEOUT_DIVISOR (1.f/(float)FADEOUT_LENGTH)

typedef struct sfx_vol sfx_vol_t;
typedef struct sfx_system sfx_system_t;

struct sfx_vol
{
        float val;
        float cmp;
};

struct sfx_system
{
        // Source buffer start
        float *source;
        float   *replacement;
        
        // Modifiers
        sfx_vol_t *vol_src;
        float   vol;
        
        float   spd;
        
        // Info
        int ch;  // channels
        u32 end; // buffer end
        u32 cur; // cursor position
        u32 flags;
        
        // Effects
        u32 snh;
        u32 fadeout;
        
        // The 'Opposite' pointer
        sfx_system_t *optr;
        
        // Diagnostic
        float   cvol; // Current signal volume
        const char *name;
};

// 0
int sfx_save( sfx_system_t *sys );      // Mark change to be uploaded to queue system
void sfx_sys_init(void);                                // Miniaudio.h init
void sfx_sys_free(void);                                // Shutdown audio device
sfx_system_t *sfx_alloc(void);          // Create and return slot for a sound

// 1
void sfx_localize(void);                                // Copy in data from all queued
void sfx_redist(void);                                  // Send out updates to sources
void audio_mixer_callback( ma_device *pDevice, void *pOutBuf, const void *pInput, ma_uint32 frameCount );       // miniaudio.h interface

ma_device g_aud_device;
ma_device_config g_aud_dconfig;

// Thread 2 - background loader ( to be moved )
// ======================================================

// Thread 1 - audio engine ( spawned from miniaudio.h )
// ======================================================
sfx_system_t    sfx_sys[SFX_MAX_SYSTEMS];
int                             sfx_sys_len = 0;

// Thread 0 - Critical transfer section
// ======================================================
MUTEX_TYPE               sfx_mux_t01; // Resources share: 0 & 1

sfx_system_t    *sfx_q[SFX_MAX_SYSTEMS]; // Stuff changed
int                              sfx_q_len = 0;                  // How much

// x / 2
// ======================================================

// g_vol_master is never directly acessed by users
float g_master_volume = 1.f;

sfx_vol_t g_vol_music;
sfx_vol_t g_vol_sfx;

#define SFX_NUM_VOLUMES 2
sfx_vol_t *g_volumes[] = { &g_vol_music, &g_vol_sfx };

// Decompress entire vorbis stream into buffer
float *sfx_vorbis_stream( const unsigned char *data, int len, int channels, uint32_t *samples )
{
        int err;
        stb_vorbis *pv = stb_vorbis_open_memory( data, len, &err, NULL );
        
        if( !pv )
        {
                vg_error( "stb_vorbis_open_memory() failed with error code: %i\n", err );
                return NULL;
        }
                
        u32 length_samples = stb_vorbis_stream_length_in_samples( pv );
        float *buffer = (float *)malloc( length_samples * channels * sizeof( float ));
        
        if( !buffer )
        {
                stb_vorbis_close( pv );
                vg_error( "out of memory while allocating sound resource\n" );
                return NULL;
        }
        
        int read_samples = stb_vorbis_get_samples_float_interleaved( pv, channels, buffer, length_samples * channels );
        if( read_samples != length_samples )
        {
                vg_warn( "| warning: sample count mismatch. Expected %u got %i\n", length_samples, read_samples );
                length_samples = read_samples;
        }
        
        stb_vorbis_close( pv );
        *samples = length_samples;
        return buffer;
}

float *sfx_vorbis( const char *strFileName, int channels, u32 *samples )
{
        i64 len;
        void *filedata = vg_asset_read_s( strFileName, &len );
        
        if( filedata )
        {
                float *wav = sfx_vorbis_stream( filedata, len, channels, samples );
                free( filedata );
                return wav;
        }
        else
        {
                vg_error( "OGG load failed\n" );
                return NULL;
        }
}

typedef struct sfx_bgload sfx_bgload_t;
struct sfx_bgload
{
        char    *path;
        u32     channels;
        
        float *buffer;
        u32      samples;
        
        void    *user;
        
        void(*OnComplete)(sfx_bgload_t *inf);
};

// Thread worker for background load job
void *sfx_vorbis_a_t( void *_inf )
{
        sfx_bgload_t *info = _inf;
        
        // Load the ogg clip
        info->buffer = sfx_vorbis( info->path, info->channels, &info->samples );
        info->OnComplete( info );
        
        return NULL;
}

// Asynchronous resource load
int sfx_vorbis_a( const char *path, int channels, void(*OnComplete)(sfx_bgload_t *inf), void *user )
{
        vg_info( "background job started for: %s\n", path );
        
        sfx_bgload_t *params = malloc( sizeof( sfx_bgload_t ) );
        params->path = malloc( strlen( path ) + 1 );
        strcpy( params->path, path );
        params->OnComplete = OnComplete;
        params->user = user;
        params->channels = channels;

        return vg_thread_run( sfx_vorbis_a_t, params );
}

// Asynchronous load-to-system callback
struct sfx_vorbis_a_to_inf
{
        sfx_system_t  *sys;
        u32                             flags;
};

#define SFX_A_FLAG_AUTOSTART 0x1
#define SFX_A_FLAG_AUTOFREE  0x2

// Asynchronous load-to-system callback
void sfx_vorbis_a_to_c( sfx_bgload_t *loadinf )
{
        struct sfx_vorbis_a_to_inf *inf = loadinf->user;
        
        // Mark buffer for deallocation if autofree is set
        if( inf->flags & SFX_A_FLAG_AUTOFREE )
        {
                inf->sys->replacement = loadinf->buffer;
        }
        else
        {
                inf->sys->source = loadinf->buffer;
        }
        
        inf->sys->end = loadinf->samples;
        
        if( inf->flags & SFX_A_FLAG_AUTOSTART )
        {
                sfx_save( inf->sys );
        }
        
        free( loadinf->path );
        free( loadinf );
        free( inf );
}

// Asynchronous vorbis load into audio system
void sfx_vorbis_a_to( sfx_system_t *sys, const char *strFileName, int channels, uint32_t flags )
{
        struct sfx_vorbis_a_to_inf *inf = malloc( sizeof( struct sfx_vorbis_a_to_inf ) );
        inf->flags = flags;
        inf->sys = sys;
        
        sys->ch = channels;
        
        if( !sfx_vorbis_a( strFileName, channels, sfx_vorbis_a_to_c, inf ) )
        {
                free( inf );
        }
}

// 0
// ======================================================

// Mark change to be uploaded to queue system
int sfx_save( sfx_system_t *sys )
{
        MUTEX_LOCK( sfx_mux_t01 );

        if( sfx_q_len >= SFX_MAX_SYSTEMS )
        {
                vg_error( "Warning: No free space in sound queue\n" );
                
                MUTEX_UNLOCK( sfx_mux_t01 );
                return 0;
        }
        
        // Mark change in queue
        sfx_q[ sfx_q_len ++ ] = sys;
        
        MUTEX_UNLOCK( sfx_mux_t01 );
        
        return 1;
}

// Edit a volume float, has to be function round-tripped
// because of mutex
void sfx_vol_fset( sfx_vol_t *src, float to )
{
        MUTEX_LOCK( sfx_mux_t01 );
        
        src->val = to;
        src->cmp = g_master_volume * to;
        
        MUTEX_UNLOCK( sfx_mux_t01 );
}

// thread-safe get volume value
float sfx_vol_fget( sfx_vol_t *src )
{
        float val;
        
        MUTEX_LOCK( sfx_mux_t01 );
        
        val = src->val;
        
        MUTEX_UNLOCK( sfx_mux_t01 );
        
        return val;
}

// thread-safe set master volume
void sfx_set_master( float to )
{
        MUTEX_LOCK( sfx_mux_t01 );
        
        g_master_volume = to;
        
        for( int i = 0; i < SFX_NUM_VOLUMES; i ++ )
        {
                g_volumes[ i ]->cmp = g_volumes[ i ]->val * g_master_volume;
        }
        
        MUTEX_UNLOCK( sfx_mux_t01 );
}

// thread-safe get master volume
float sfx_get_master(void)
{
        float val;

        MUTEX_LOCK( sfx_mux_t01 );
        
        val = g_master_volume;
        
        MUTEX_UNLOCK( sfx_mux_t01 );
        
        return val;
}

// Miniaudio.h init
void vg_audio_init(void)
{
        // Setup volume values
        // Todo: load these from config
        g_vol_sfx.val = 1.f;
        g_vol_music.val = 1.f;
        sfx_set_master( 1.f );

        g_aud_dconfig = ma_device_config_init( ma_device_type_playback );
        g_aud_dconfig.playback.format   = ma_format_f32;
        g_aud_dconfig.playback.channels         = 2;
        g_aud_dconfig.sampleRate                        = 44100;
        g_aud_dconfig.dataCallback              = audio_mixer_callback;
        
        g_aud_dconfig.pUserData = NULL;
        
        vg_info( "Starting audio engine\n" );
        
        if( ma_device_init( NULL, &g_aud_dconfig, &g_aud_device ) != MA_SUCCESS )
        {
                vg_exiterr( "ma_device failed to initialize" );
        } 
        else 
        {       
                if( ma_device_start( &g_aud_device ) != MA_SUCCESS )
                {
                        ma_device_uninit( &g_aud_device );
                        vg_exiterr( "ma_device failed to start" );
                }
        }
}

#ifndef VYGER_RELEASE
uint32_t num_sfx_sets = 0;
#endif

// Shutdown audio device
void vg_audio_free(void)
{
        ma_device_uninit( &g_aud_device );
}

// (debug) make sure we are shutting down safely
void sfx_sys_chkerr(void)
{
#ifndef VYGER_RELEASE
        if( num_sfx_sets )
        {
                vg_error( "Leaked %u sfx sets\n", num_sfx_sets );
        }
#endif
}

// 1
// ======================================================

// Create and return slot for a sound
sfx_system_t *sfx_alloc(void)
{
        if( sfx_sys_len >= SFX_MAX_SYSTEMS )
        {
                vg_error( "Warning: No free space in sound system\n" );
                
                return NULL;
        }
        
        // A conditional is done against this in localization step,
        // Needs to be initialized.
        sfx_sys[ sfx_sys_len ].source = NULL;
        
        return sfx_sys + (sfx_sys_len++);
}

// Copy in data from all queued
void sfx_localize(void)
{
        MUTEX_LOCK( sfx_mux_t01 );
        
        while( sfx_q_len --> 0 )
        {
                sfx_system_t *src = sfx_q[sfx_q_len];
                
                // This is a 'new' sound if optr not set.
                if( !src->optr || src->flags & SFX_FLAG_ONESHOT )
                {
                        src->optr = sfx_alloc();
                }
                
                // run replacement routine if one is waiting
                if( src->replacement )
                {
                        if( src->source )
                        {
                                printf( "Deallocating previous source buffer\n" );
                        }
                        
                        free( src->source );
                        
                        src->source = src->replacement;
                        src->replacement = NULL;
                }

                src->optr->source = src->source;
                
                // Localize data to thread 1's memory pool
                // memcpy( src->optr, src, sizeof( sfx_system_t ) );
                
                src->optr->spd = src->spd;
                src->optr->ch = src->ch;
                src->optr->end = src->end;
                src->optr->cur = src->cur;
                src->optr->flags = src->flags;
                // src->optr->sng = src->snh;
                src->optr->fadeout = src->fadeout;
                // src->optr->optr = src->optr;
                // src->optr->cvol = src->cvol;
                src->optr->vol_src = src->vol_src;
                src->optr->name = src->name;

                // loopback pointer on system system
                src->optr->optr = src;
        }
        sfx_q_len = 0;
        
        // Pull in volume sliders
        for( int i = 0; i < sfx_sys_len; i ++ )
        {
                sfx_system_t *sys = sfx_sys + i;
                sys->vol = sys->optr->vol;
                if( sys->vol_src ) { sys->vol *= sys->vol_src->cmp; };
        }
        
        MUTEX_UNLOCK( sfx_mux_t01 );
}

// Send out updates to sources
void sfx_redist(void)
{
        MUTEX_LOCK( sfx_mux_t01 );
        
        unsigned int idx = 0, wr = 0;
        while( idx != sfx_sys_len )
        {
                sfx_system_t *src = sfx_sys + idx;
        
                // Keep only if cursor is before end
                if( src->cur < src->end ) 
                {
                        if( !(src->flags & SFX_FLAG_ONESHOT) )
                        {
                                // Correct source pointer
                                src->optr->optr = sfx_sys + wr;
                        }
                        
                        sfx_sys[ wr ++ ] = sfx_sys[ idx ];
                }
                else
                {
                        if( !(src->flags & SFX_FLAG_ONESHOT) )
                        {
                                // Clear link on source 
                                src->optr->optr = NULL;
                        }
                }
                
                idx ++ ;
        }
        sfx_sys_len = wr;
        
        MUTEX_UNLOCK( sfx_mux_t01 );
}

// Fetch samples into pcf
void audio_mixer_getsamples( float *pcf, float *source, uint32_t cur, uint32_t ch )
{
        if( ch == 2 )
        {
                pcf[0] = source[ cur*2+0 ];
                pcf[1] = source[ cur*2+1 ];
        }
        else
        {
                pcf[0] = source[ cur ];
                pcf[1] = source[ cur ];
        }
}

// miniaudio.h interface
void audio_mixer_callback( ma_device *pDevice, void *pOutBuf, const void *pInput, ma_uint32 frameCount )
{
        sfx_localize();

        // Clear buffer ( is necessary ? )
        float *pOut32F = (float *)pOutBuf;
        for( int i = 0; i < frameCount * 2; i ++ ){
                pOut32F[i] = 0.f;
        }
        
        // Do something with local..
        for( int i = 0; i < sfx_sys_len; i ++ )
        {
                sfx_system_t *sys = sfx_sys + i;
                
                uint32_t cursor = sys->cur;
                uint32_t bpos = 0;
                
                float avgvol = 0.f;
                
                float pcf[2] = { 0.f };
                
                while( cursor < vg_min( sys->cur + frameCount, sys->end ) )
                {
                        audio_mixer_getsamples( pcf, sys->source, cursor, sys->ch );
                        
                        avgvol += fabs( pcf[0] * sys->vol );
                        avgvol += fabs( pcf[1] * sys->vol );
                        
                        pOut32F[ bpos*2+0 ] += pcf[0] * sys->vol;
                        pOut32F[ bpos*2+1 ] += pcf[1] * sys->vol;
                        
                        // Blend the fadeout cursor in to prevent popping
                        // This lasts 441 samples
                        if( sys->fadeout )
                        {
                                if( sys->snh < sys->end )
                                {
                                        audio_mixer_getsamples( pcf, sys->source, sys->snh, sys->ch );
                                        
                                        float mul = (float)sys->fadeout * FADEOUT_DIVISOR;
                                        
                                        pOut32F[ bpos*2+0 ] += pcf[0] * sys->vol * mul;
                                        pOut32F[ bpos*2+1 ] += pcf[1] * sys->vol * mul;
                                        
                                        sys->snh ++;
                                        sys->fadeout --;
                                }
                                else
                                {
                                        sys->fadeout = 0;
                                }
                        }
                        
                        cursor ++;
                        bpos ++;
                }
                
                if( !sys->fadeout )
                {
                        sys->snh = cursor;
                }
                
                sys->cvol = avgvol / (float)(bpos*2);
                sys->cur += frameCount;
        }

        sfx_redist();
        
        (void)pInput;
}

// Set of up to 8 sound effects packed into one
typedef struct sfx_set sfx_set_t;
struct sfx_set
{
        float *main;
        char *sources;
        
        uint32_t segments[16];  //from->to,from->to ...
        uint32_t numsegments;
        uint32_t ch;
        uint32_t flags;
};

// Load strings into sfx_set's memory
// String layout: "sounda.ogg\0soundb.ogg\0soundc.ogg\0\0"
void sfx_set_strings( sfx_set_t *dest, char *strSources, uint32_t flags, int bAsync )
{
        printf( "Init sfx set\n|   start   |    end    |  length   | name \n" );

        dest->ch = (flags & SFX_FLAG_STEREO)? 2: 1;
        
        dest->main = NULL;
        dest->numsegments = 0;
        char *source = strSources;
        
        uint32_t total = 0;
        int len;
        while( (len = strlen( source )) )
        {
                uint32_t samples;
                float *sound = sfx_vorbis( source, dest->ch, &samples );
                
                if( !sound )
                {
                        free( dest->main );
                        dest->numsegments = 0;
                        return;
                }
                
                total += samples;
                
                float *nbuf = realloc( dest->main, total * dest->ch * sizeof(float) );
                
                if( nbuf )
                {
                        dest->main = nbuf;
                        memcpy( dest->main + (total-samples)*dest->ch, sound, samples*dest->ch*sizeof(float) );
                        free( sound );
                        
                        dest->segments[ dest->numsegments*2+0 ] = total-samples;
                        dest->segments[ dest->numsegments*2+1 ] = total;
                        
                        printf( "| %09u | %09u | %09u | %s\n", total-samples, total, samples, source );
                }
                else
                {
                        vg_error( "realloc() failed\n" );
                        free( sound );
                        return;
                }
                
                source += len +1;
                dest->numsegments ++;
        }
        
        vg_info( "finished, numsegments: %u\n", dest->numsegments );
}


// If sources is non-null then it will try to pull from
// internally set string
//
// internal set string should be literal, otherwise leak if not
// handled correctly
void sfx_set_init( sfx_set_t *dest, char *sources )
{
#ifndef VYGER_RELEASE
        num_sfx_sets ++;
#endif

        if( !sources )
        {
                sfx_set_strings( dest, dest->sources, dest->flags, 0 );
        }
        else
        {
                sfx_set_strings( dest, sources, dest->flags, 0 );
        }
}

// Pick a random sound from the buffer and play it into system
void sfx_set_playrnd( sfx_set_t *source, sfx_system_t *sys )
{
        if( !source->numsegments )
        {
                return;
        }

        int pick = rand() % source->numsegments;
        
        sys->source = source->main;
        sys->cur        = source->segments[ pick*2 + 0 ];
        sys->end        = source->segments[ pick*2 + 1 ];
        sys->ch                 = source->ch;
        
        sfx_save( sys );
}

// Free set resources
void sfx_set_free( sfx_set_t *set )
{
#ifndef VYGER_RELEASE
        num_sfx_sets --;
#endif
        free( set->main );
}