/*

  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

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 ((!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...? */

    #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;

}