/*  SSNES - A Super Nintendo Entertainment System (SNES) Emulator frontend for libsnes.
 *  Copyright (C) 2010-2011 - Hans-Kristian Arntzen
 *
 *  Some code herein may be based on code found in BSNES.
 * 
 *  SSNES is free software: you can redistribute it and/or modify it under the terms
 *  of the GNU General Public License as published by the Free Software Found-
 *  ation, either version 3 of the License, or (at your option) any later version.
 *
 *  SSNES is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 *  without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 *  PURPOSE.  See the GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along with SSNES.
 *  If not, see <http://www.gnu.org/licenses/>.
 */


#include "driver.h"
#include <stdlib.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <string.h>

#include "SDL.h"
#include "SDL_audio.h"
#include "SDL_thread.h"
#include "general.h"
#include "buffer.h"

typedef struct sdl_audio
{
   bool nonblock;

   SDL_mutex *lock;
   SDL_cond *cond;
   rsound_fifo_buffer_t *buffer;
} sdl_audio_t;

static void sdl_audio_cb(void *data, Uint8 *stream, int len)
{
   sdl_audio_t *sdl = data;

   size_t avail = rsnd_fifo_read_avail(sdl->buffer);
   size_t write_size = len > avail ? avail : len;
   rsnd_fifo_read(sdl->buffer, stream, write_size);
   SDL_CondSignal(sdl->cond);

   // If underrun, fill rest with silence.
   memset(stream + write_size, 0, len - write_size);
}

// Interesting hack from http://www-graphics.stanford.edu/~seander/bithacks.html#RoundUpPowerOf2
static inline uint32_t next_pow2(uint32_t v)
{
   v--;
   v |= v >> 1;
   v |= v >> 2;
   v |= v >> 4;
   v |= v >> 8;
   v |= v >> 16;
   v++;
   return v;
}

static inline int find_num_frames(int rate, int latency)
{
   int frames = (rate * latency) / 1000;
   // SDL only likes 2^n sized buffers.
   return next_pow2(frames);
}

static void* sdl_audio_init(const char* device, int rate, int latency)
{
   (void)device;
   if (SDL_InitSubSystem(SDL_INIT_AUDIO) < 0)
      return NULL;

   sdl_audio_t *sdl = calloc(1, sizeof(*sdl));
   if (!sdl)
      return NULL;

   // We have to buffer up some data ourselves, so we let SDL carry approx half of the latency. SDL double buffers audio and we do as well.
   int frames = find_num_frames(rate, latency / 4);

   SDL_AudioSpec spec = {
      .freq = rate,
      .format = AUDIO_S16SYS,
      .channels = 2,
      .samples = frames, // This is in audio frames, not samples ... :(
      .callback = sdl_audio_cb,
      .userdata = sdl
   };

   SDL_AudioSpec out;

   if (SDL_OpenAudio(&spec, &out) < 0)
   {
      SSNES_ERR("Failed to open SDL audio: %s\n", SDL_GetError());
      free(sdl);
      return 0;
   }
   g_settings.audio.out_rate = out.freq;

   sdl->lock = SDL_CreateMutex();
   sdl->cond = SDL_CreateCond();

   SSNES_LOG("SDL audio: Requested %d ms latency, got %d ms\n", latency, (int)(out.samples * 4 * 1000 / g_settings.audio.out_rate));

   // Create a buffer twice as big as needed and prefill the buffer.
   size_t bufsize = out.samples * 4 * sizeof(int16_t);
   void *tmp = calloc(1, bufsize);
   sdl->buffer = rsnd_fifo_new(bufsize);
   if (tmp)
   {
      rsnd_fifo_write(sdl->buffer, tmp, bufsize);
      free(tmp);
   }

   SDL_PauseAudio(0);
   return sdl;
}

static ssize_t sdl_audio_write(void* data, const void* buf, size_t size)
{
   sdl_audio_t *sdl = data;

   ssize_t ret = 0;
   if (sdl->nonblock)
   {
      SDL_LockAudio();
      size_t avail = rsnd_fifo_write_avail(sdl->buffer);
      size_t write_amt = avail > size ? size : avail;
      rsnd_fifo_write(sdl->buffer, buf, write_amt);
      SDL_UnlockAudio();
      ret = write_amt;
   }
   else
   {
      size_t written = 0;
      while (written < size)
      {
         SDL_LockAudio();
         size_t avail = rsnd_fifo_write_avail(sdl->buffer);

         if (avail == 0)
         {
            SDL_UnlockAudio();
            SDL_mutexP(sdl->lock);
            SDL_CondWait(sdl->cond, sdl->lock);
            SDL_mutexV(sdl->lock);
         }
         else
         {
            size_t write_amt = size - written > avail ? avail : size - written;
            rsnd_fifo_write(sdl->buffer, (const char*)buf + written, write_amt);
            SDL_UnlockAudio();
            written += write_amt;
         }
      }
      ret = written;
   }

   return ret;
}

static bool sdl_audio_stop(void *data)
{
   (void)data;
   SDL_PauseAudio(1);
   return true;
}

static bool sdl_audio_start(void *data)
{
   (void)data;
   SDL_PauseAudio(0);
   return true;
}

static void sdl_audio_set_nonblock_state(void *data, bool state)
{
   sdl_audio_t *sdl = data;
   sdl->nonblock = state;
}

static void sdl_audio_free(void *data)
{
   SDL_CloseAudio();
   SDL_QuitSubSystem(SDL_INIT_AUDIO);

   sdl_audio_t *sdl = data;
   if (sdl)
   {
      rsnd_fifo_free(sdl->buffer);
      SDL_DestroyMutex(sdl->lock);
      SDL_DestroyCond(sdl->cond);
   }
   free(sdl);
}

const audio_driver_t audio_sdl = {
   .init = sdl_audio_init,
   .write = sdl_audio_write,
   .stop = sdl_audio_stop,
   .start = sdl_audio_start,
   .set_nonblock_state = sdl_audio_set_nonblock_state,
   .free = sdl_audio_free,
   .ident = "sdl"
};