#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#define __STDC_CONSTANT_MACROS
#include <stdlib.h>
#include <stdio.h>
#include <stdint.h>

#include <SDL.h>
#include <SDL/SDL_ffmpeg.h>

extern "C" {
#include <libavcodec/avcodec.h>
}

#include "osc_graphics.h"
#include "osc_server.h"

#include "recorder.h"

/*
 * Macros
 */
#define SDL_FFMPEGFREE_SAFE(FILE) do {			\
	if (FILE) {					\
		SDL_ffmpegFree(FILE);			\
		FILE = NULL;				\
	}						\
} while (0)

#define SDL_FFMPEG_ERROR(FMT, ...) do {			\
	fprintf(stderr, "%s(%d): " FMT ": %s\n",	\
		__FILE__, __LINE__, ##__VA_ARGS__,	\
		SDL_ffmpegGetError());			\
} while (0)

extern "C" {

static int start_handler(const char *path, const char *types,
		       	 lo_arg **argv, int argc,
		       	 void *data, void *user_data);
static int stop_handler(const char *path, const char *types,
			lo_arg **argv, int argc,
		       	void *data, void *user_data);

}

void
Recorder::register_methods()
{
	osc_server.add_method("s", start_handler, this, "/recorder/start");
	osc_server.add_method("", stop_handler, this, "/recorder/stop");
}

static int
start_handler(const char *path, const char *types,
	      lo_arg **argv, int argc, void *data, void *user_data)
{
	Recorder *obj = (Recorder *)user_data;

	obj->start(&argv[0]->s);

	return 0;
}

void
Recorder::start(const char *filename)
{
	SDL_ffmpegCodec codec = {
		-1,			/* video codec based on file name */
		screen->w, screen->h,
		1, config_framerate,	/* framerate */
		6000000, -1, -1,
		-1, 0, -1, -1, -1, -1	/* no audio */
	};

	if (!filename || !*filename) {
		stop();
		return;
	}

	lock();

	SDL_FFMPEGFREE_SAFE(file);

	file = SDL_ffmpegCreate(filename);
	if (!file) {
		SDL_FFMPEG_ERROR("SDL_ffmpegCreate");
		exit(EXIT_FAILURE);
	}

	SDL_ffmpegAddVideoStream(file, codec);
	SDL_ffmpegSelectVideoStream(file, 0);

	start_time = SDL_GetTicks();

	unlock();
}

static int
stop_handler(const char *path, const char *types,
	     lo_arg **argv, int argc, void *data, void *user_data)
{
	Recorder *obj = (Recorder *)user_data;

	obj->stop();

	return 0;
}

void
Recorder::stop()
{
	lock();
	SDL_FFMPEGFREE_SAFE(file);
	unlock();
}

void
Recorder::record(SDL_Surface *surf)
{
	lock();

	if (file) {
		struct AVFrame *frame = file->videoStream->encodeFrame;
		int64_t pts = (SDL_GetTicks() - start_time)/FRAME_DELAY;

		if (pts > frame->pts) {
			frame->pts = pts;
			SDL_ffmpegAddVideoFrame(file, surf);
		}
	}

	unlock();
}

Recorder::~Recorder()
{
	osc_server.del_method("s", "/recorder/start");
	osc_server.del_method("",  "/recorder/stop");

	SDL_FFMPEGFREE_SAFE(file);
}