//
//  AudioVisualizer.m
//  CaseLights
//
//  Based on the ideas in:
//  http://archive.gamedev.net/archive/reference/programming/features/beatdetection/
//
//  The detected sound frequency of beats will be mapped to the hue of the resulting color,
//  the variance of the beat is mapped to the brightness of the color. The colors
//  of all detected beats will be added together to form the final displayed color.
//
//  Created by Thomas Buck on 01.01.16.
//  Copyright © 2016 xythobuz. All rights reserved.
//

#ifdef DEBUG
#define DEBUG_LOG_BEATS
#endif

#import "AudioVisualizer.h"
#import "AppDelegate.h"

#import "EZAudioFFT.h"
#import "EZAudioPlot.h"

// Parameters for fine-tuning beat detection
#define FFT_BUCKET_COUNT 64
#define FFT_BUCKET_HISTORY 45
#define FFT_C_FACTOR 3.3
#define FFT_V0_FACTOR 0.00001
#define FFT_MAX_V0_COLOR 0.00025
#define FFT_COLOR_DECAY 0.98

// Use this to skip specific frequencies
// Only check bass frequencies
//#define FFT_BUCKET_SKIP_CONDITION (i > (FFT_BUCKET_COUNT / 4))
// Only check mid frequencies
//#define FFT_BUCKET_SKIP_CONDITION ((i < (FFT_BUCKET_COUNT / 4)) || (i > (FFT_BUCKET_COUNT * 3 / 4)))
// Only check high frequencies
//#define FFT_BUCKET_SKIP_CONDITION (i < (FFT_BUCKET_COUNT * 3 / 4))

// Factors for nicer debug display
#define FFT_DEBUG_RAW_FACTOR 42.0
#define FFT_DEBUG_FACTOR 230.0

static AppDelegate *appDelegate = nil;
static EZAudioFFT *fft = nil;
static int maxBufferSize = 0;
static float sensitivity = 1.0f;
static float history[FFT_BUCKET_COUNT][FFT_BUCKET_HISTORY];
static int nextHistory =  0;
static int samplesPerBucket = 0;
static unsigned char lastRed = 0, lastGreen = 0, lastBlue = 0;

static BOOL shouldShowWindow = NO;
static NSWindow *window = nil;
static EZAudioPlot *plot = nil;
static NSTextField *label = nil;

@implementation AudioVisualizer

+ (void)setDelegate:(AppDelegate *)delegate {
    appDelegate = delegate;
    
    // Initialize static history variables
    for (int i = 0; i < FFT_BUCKET_COUNT; i++) {
        for (int j = 0; j < FFT_BUCKET_HISTORY; j++) {
            history[i][j] = 0.5f;
        }
    }
}

+ (void)setSensitivity:(float)sens {
    sensitivity = sens / 100.0;
}

+ (void)updateBuffer:(float *)buffer withBufferSize:(UInt32)bufferSize {
    // Create Fast Fourier Transformation object
    if (fft == nil) {
        maxBufferSize = bufferSize;
        samplesPerBucket = bufferSize / FFT_BUCKET_COUNT;
        fft = [EZAudioFFT fftWithMaximumBufferSize:maxBufferSize sampleRate:appDelegate.microphone.audioStreamBasicDescription.mSampleRate];
        
#ifdef DEBUG
        NSLog(@"Created FFT with max. freq.: %.2f\n", appDelegate.microphone.audioStreamBasicDescription.mSampleRate / 2);
#endif
    }
    
    // Check for changing buffer sizes
    if (bufferSize > maxBufferSize) {
        NSLog(@"Buffer Size changed?! %d != %d\n", maxBufferSize, bufferSize);
        maxBufferSize = bufferSize;
        samplesPerBucket = bufferSize / FFT_BUCKET_COUNT;
        fft = [EZAudioFFT fftWithMaximumBufferSize:maxBufferSize sampleRate:appDelegate.microphone.audioStreamBasicDescription.mSampleRate];
    }
    
    // Scale input if required
    if (sensitivity != 1.0f) {
        for (int i = 0; i < bufferSize; i++) {
            buffer[i] *= sensitivity;
        }
    }
    
    // Perform fast fourier transformation
    [fft computeFFTWithBuffer:buffer withBufferSize:bufferSize];
    
    // Split FFT output into a small number of 'buckets' or 'bins' and add to circular history buffer
    for (int i = 0; i < FFT_BUCKET_COUNT; i++) {
        float sum = 0.0f;
        for (int j = 0; j < samplesPerBucket; j++) {
            sum += fft.fftData[(i + samplesPerBucket) + j];
        }
        history[i][nextHistory] = sum / samplesPerBucket;
    }
    
    // Slowly fade old colors to black
    lastRed = lastRed * FFT_COLOR_DECAY;
    lastGreen = lastGreen * FFT_COLOR_DECAY;
    lastBlue = lastBlue * FFT_COLOR_DECAY;
    
    // Check for any beats
    int beatCount = 0;
    for (int i = 0; i < FFT_BUCKET_COUNT; i++) {
        // Skip frequency bands, if required
#ifdef FFT_BUCKET_SKIP_CONDITION
        if (FFT_BUCKET_SKIP_CONDITION) continue;
#endif
        
        // Calculate average of history of this frequency
        float average = 0.0f;
        for (int j = 0; j < FFT_BUCKET_HISTORY; j++) {
            average += history[i][j];
        }
        average /= FFT_BUCKET_HISTORY;
        
        // Calculate variance of current bucket in history
        float v = 0.0f;
        for (int j = 0; j < FFT_BUCKET_HISTORY; j++) {
            float tmp = history[i][j] - average;
            tmp *= tmp;
            v += tmp;
        }
        v /= FFT_BUCKET_HISTORY;
        
        // Check for beat conditions
        if ((history[i][nextHistory] > (FFT_C_FACTOR * average)) && (v > FFT_V0_FACTOR)) {
            // Found a beat on this frequency band, map to a single color
            if (v < FFT_V0_FACTOR) v = FFT_V0_FACTOR;
            if (v > FFT_MAX_V0_COLOR) v = FFT_MAX_V0_COLOR;
            float bright = [AppDelegate map:v FromMin:FFT_V0_FACTOR FromMax:FFT_MAX_V0_COLOR ToMin:0.0 ToMax:100.0];
            float hue = [AppDelegate map:i FromMin:0.0 FromMax:FFT_BUCKET_COUNT ToMin:0.0 ToMax:360.0];
            unsigned char r, g, b;
            [AppDelegate convertH:hue S:1.0 V:bright toR:&r G:&g B:&b];
            
            // Blend with last color using averaging
            int tmpR = (lastRed + r) / 2;
            int tmpG = (lastGreen + g) / 2;
            int tmpB = (lastBlue + b) / 2;
            lastRed = tmpR;
            lastGreen = tmpG;
            lastBlue = tmpB;
            
#ifdef DEBUG_LOG_BEATS
            NSLog(@"Beat in %d with c: %f v: %f", i, (history[i][nextHistory] / average), v);
#endif
            
            beatCount++;
        }
    }
    
    // Send new RGB value to lights, if it has changed
    static unsigned char lastSentRed = 42, lastSentGreen = 23, lastSentBlue = 99;
    if ((lastSentRed != lastRed) || (lastSentGreen != lastGreen) || (lastSentBlue != lastBlue)) {
        [appDelegate setLightsR:lastRed G:lastGreen B:lastBlue];
        lastSentRed = lastRed;
        lastSentGreen = lastGreen;
        lastSentBlue = lastBlue;
    }

    // Update debug FFT plot, if required
    if (shouldShowWindow && (window != nil) && (plot != nil) && (label != nil)) {
        for (UInt32 i = 0; i < FFT_BUCKET_COUNT; i++) {
            // Copy output to input buffer (a bit ugly, but is always big enough)
            buffer[i] = history[i][nextHistory];
            
            // Scale so user can see something
            buffer[i] *= FFT_DEBUG_FACTOR;
            if (buffer[i] > 1.0f) buffer[i] = 1.0f;
            if (buffer[i] < -1.0f) buffer[i] = -1.0f;
        }
        [plot updateBuffer:buffer withBufferSize:bufferSize];
        
        // Change background color to match color output and show beat counter
        [window setBackgroundColor:[NSColor colorWithCalibratedRed:lastRed / 255.0 green:lastGreen / 255.0 blue:lastBlue / 255.0 alpha:1.0]];
        [label setStringValue:[NSString stringWithFormat:@"Beats: %d", beatCount]];
    }
        
    // Point to next history buffer
    nextHistory++;
    if (nextHistory >= FFT_BUCKET_HISTORY) {
        nextHistory = 0;
    }
}

+ (void)setShowWindow:(BOOL)showWindow {
    shouldShowWindow = showWindow;
    
    // Close window if it was visible and should no longer be
    if (showWindow == YES) {
        if ((window == nil) || (plot == nil) || (label == nil)) {
            // Create window
            NSRect frame = NSMakeRect(450, 300, 600, 400);
            window = [[NSWindow alloc] initWithContentRect:frame
                                                 styleMask:NSClosableWindowMask | NSTitledWindowMask | NSBorderlessWindowMask
                                                   backing:NSBackingStoreBuffered
                                                     defer:NO];
            [window setTitle:@"CaseLights FFT"];
            [window setReleasedWhenClosed:NO];
            
            // Create FFT Plot and add to window
            plot = [[EZAudioPlot alloc] initWithFrame:window.contentView.frame];
            plot.color = [NSColor whiteColor];
            plot.shouldOptimizeForRealtimePlot = NO; // Not working with 'YES' here?!
            plot.shouldFill = YES;
            plot.shouldCenterYAxis = NO;
            plot.shouldMirror = NO;
            plot.plotType = EZPlotTypeBuffer;
            [window.contentView addSubview:plot];
            
            // Create beat count label
            label = [[NSTextField alloc] initWithFrame:NSMakeRect(0, 380, 600, 20)];
            [label setTextColor:[NSColor whiteColor]];
            [label setEditable:NO];
            [label setBezeled:NO];
            [label setDrawsBackground:NO];
            [label setSelectable:NO];
            [label setStringValue:@"-"];
            [window.contentView addSubview:label];
            
#ifdef DEBUG
            NSLog(@"Created debugging FFT Plot window...\n");
#endif
        }
        
        if ([window isVisible] == NO) {
            // Make window visible
            [window makeKeyAndOrderFront:appDelegate.application];
            
#ifdef DEBUG
            NSLog(@"Made debugging FFT Plot window visible...\n");
#endif
        }
    } else {
        if (window != nil) {
            if ([window isVisible] == YES) {
                [window close];
                
#ifdef DEBUG
                NSLog(@"Closed debugging FFT Plot window...\n");
#endif
            }
        }
    }
}

@end