/*
  Copyright (C) 1997-2016 Sam Lantinga <slouken@libsdl.org>

  This software is provided 'as-is', without any express or implied
  warranty.  In no event will the authors be held liable for any damages
  arising from the use of this software.

  Permission is granted to anyone to use this software for any purpose,
  including commercial applications, and to alter it and redistribute it
  freely.
*/
#include "SDL.h"

#ifdef __EMSCRIPTEN__
#include <emscripten/emscripten.h>
#endif

#define CAPTURE_SECONDS 5

#define DO_VIDEO defined(__ANDROID__) || defined(__IPHONEOS__) || defined(__EMSCRIPTEN__)
#if DO_VIDEO
static SDL_Window *window = NULL;
static SDL_Renderer *renderer = NULL;
#endif

static SDL_AudioSpec spec;
static Uint8 *sound = NULL;     /* Pointer to wave data */
static Uint32 soundlen = 0;     /* Length of wave data */
static Uint32 processed = 0;
static SDL_AudioDeviceID devid = 0;

void SDLCALL
capture_callback(void *arg, Uint8 * stream, int len)
{
    const int avail = (int) (soundlen - processed);
    if (len > avail) {
        len = avail;
    }

    /*SDL_Log("CAPTURE CALLBACK: %d more bytes\n", len);*/
    SDL_memcpy(sound + processed, stream, len);
    processed += len;
}

void SDLCALL
play_callback(void *arg, Uint8 * stream, int len)
{
    const Uint8 *waveptr = sound + processed;
    const int avail = soundlen - processed;
    int cpy = len;
    if (cpy > avail) {
        cpy = avail;
    }

    /*SDL_Log("PLAY CALLBACK: %d more bytes\n", cpy);*/
    SDL_memcpy(stream, waveptr, cpy);
    processed += cpy;

    len -= cpy;
    if (len > 0) {
        SDL_memset(stream + cpy, spec.silence, len);
    }
}

static void
loop()
{
    SDL_Event e;
    SDL_bool please_quit = SDL_FALSE;

    while (SDL_PollEvent(&e)) {
        if (e.type == SDL_QUIT) {
            please_quit = SDL_TRUE;
        }
    }

    #if DO_VIDEO
    if (spec.callback == capture_callback) {
        SDL_SetRenderDrawColor(renderer, 0, 255, 0, 255);
    } else {
        SDL_SetRenderDrawColor(renderer, 255, 0, 0, 255);
    }
    SDL_RenderClear(renderer);
    SDL_RenderPresent(renderer);
    #endif

    if ((!please_quit) && (processed >= soundlen)) {
        processed = 0;
        if (spec.callback == capture_callback) {
            SDL_Log("Done recording, playing back...\n");
            SDL_PauseAudioDevice(devid, 1);
            SDL_CloseAudioDevice(devid);

            spec.callback = play_callback;
            devid = SDL_OpenAudioDevice(NULL, 0, &spec, &spec, 0);
            if (!devid) {
                SDL_LogError(SDL_LOG_CATEGORY_APPLICATION, "Couldn't open an audio device for playback!\n");
                SDL_Quit();
                exit(1);
            }

            SDL_PauseAudioDevice(devid, 0);
        } else {
            SDL_Log("Done playing back.\n");
            please_quit = SDL_TRUE;
        }
    }

    if (please_quit) {
        /* stop playing back, quit. */
        SDL_Log("Shutting down.\n");
        SDL_PauseAudioDevice(devid, 1);
        SDL_CloseAudioDevice(devid);
        SDL_free(sound);
        sound = NULL;
        SDL_Quit();
        #ifdef __EMSCRIPTEN__
        emscripten_cancel_main_loop();
        #endif
        exit(0);
    }
}

int
main(int argc, char **argv)
{
    /* (argv[1] == NULL means "open default device.") */
    const char *devname = argv[1];
    int devcount;
    int i;

    /* Enable standard application logging */
    SDL_LogSetPriority(SDL_LOG_CATEGORY_APPLICATION, SDL_LOG_PRIORITY_INFO);

    /* Load the SDL library */
    if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO) < 0) {
        SDL_LogError(SDL_LOG_CATEGORY_APPLICATION, "Couldn't initialize SDL: %s\n", SDL_GetError());
        return (1);
    }

    /* Android apparently needs a window...? */
    #if DO_VIDEO
    window = SDL_CreateWindow("testaudiocapture", SDL_WINDOWPOS_CENTERED, SDL_WINDOWPOS_CENTERED, 320, 240, SDL_WINDOW_FULLSCREEN_DESKTOP);
    renderer = SDL_CreateRenderer(window, -1, 0);
    SDL_SetRenderDrawColor(renderer, 0, 0, 0, 255);
    SDL_RenderClear(renderer);
    SDL_RenderPresent(renderer);
    #endif

    SDL_Log("Using audio driver: %s\n", SDL_GetCurrentAudioDriver());

    devcount = SDL_GetNumAudioDevices(SDL_TRUE);
    for (i = 0; i < devcount; i++) {
        SDL_Log(" Capture device #%d: '%s'\n", i, SDL_GetAudioDeviceName(i, SDL_TRUE));
    }

    SDL_zero(spec);
    spec.freq = 44100;
    spec.format = AUDIO_F32SYS;
    spec.channels = 1;
    spec.samples = 1024;
    spec.callback = capture_callback;

    soundlen = spec.freq * (SDL_AUDIO_BITSIZE(spec.format) / 8) * spec.channels * CAPTURE_SECONDS;
    sound = (Uint8 *) SDL_malloc(soundlen);
    if (!sound) {
        SDL_LogError(SDL_LOG_CATEGORY_APPLICATION, "Out of memory\n");
        SDL_Quit();
        return 1;
    }

    SDL_Log("Opening device %s%s%s...\n",
            devname ? "'" : "",
            devname ? devname : "[[default]]",
            devname ? "'" : "");

    devid = SDL_OpenAudioDevice(argv[1], SDL_TRUE, &spec, &spec, 0);
    if (!devid) {
        SDL_LogError(SDL_LOG_CATEGORY_APPLICATION, "Couldn't open an audio device for capture: %s!\n", SDL_GetError());
        SDL_free(sound);
        SDL_Quit();
        exit(1);
    }

    SDL_Log("Recording for %d seconds...\n", CAPTURE_SECONDS);
    SDL_PauseAudioDevice(devid, 0);

#ifdef __EMSCRIPTEN__
    emscripten_set_main_loop(loop, 0, 1);
#else
    while (1) { loop(); SDL_Delay(16); }
#endif

    return 0;
}