/******************************************************************************
*                            recordMyDesktop                                  *
*******************************************************************************
*                                                                             *
*            Copyright (C) 2006,2007,2008 John Varouhakis                     *
*                                                                             *
*                                                                             *
*   This program 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 Foundation; either version 2 of the License, or         *
*   (at your option) any later version.                                       *
*                                                                             *
*   This program 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 this program; if not, write to the Free Software               *
*   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA  *
*                                                                             *
*                                                                             *
*                                                                             *
*   For further information contact me at johnvarouhakis@gmail.com            *
******************************************************************************/

#include "config.h"
#include "rmd_rescue.h"

#include "rmd_yuv_utils.h"
#include "rmd_encode_cache.h"
#include "rmd_initialize_data.h"
#include "rmd_register_callbacks.h"
#include "rmd_specsfile.h"
#include "rmd_types.h"

#include <pthread.h>

#include <string.h>
#include <stdlib.h>


int rmdRescue(const char *path)
{
	unsigned short	width, height;
	ProgData	pdata = {};
	EncData		enc_data = {};
	CacheData	cache_data = {};

	rmdSetupDefaultArgs(&pdata.args);

	pdata.enc_data = &enc_data;
	pdata.cache_data = &cache_data;

	//projname
	cache_data.projname = malloc(strlen(path) + 2);
	strcpy(cache_data.projname, path);
	strcat(cache_data.projname, "/");//having two of these doesn't hurt...
	//image data
	cache_data.imgdata = malloc(strlen(cache_data.projname) + 11);
	strcpy(cache_data.imgdata, cache_data.projname);
	strcat(cache_data.imgdata, "img.out");
	//audio data
	cache_data.audiodata = malloc(strlen(cache_data.projname) + 10);
	strcpy(cache_data.audiodata, cache_data.projname);
	strcat(cache_data.audiodata, "audio.pcm");
	//specsfile
	cache_data.specsfile = malloc(strlen(cache_data.projname) + 10);
	strcpy(cache_data.specsfile, cache_data.projname);
	strcat(cache_data.specsfile, "specs.txt");

	if (rmdReadSpecsFile(&pdata))
		return 1;

	width = ((pdata.brwin.rrect.width + 15) >> 4) << 4;
	height = ((pdata.brwin.rrect.height + 15) >> 4) << 4;

	enc_data.yuv.y = malloc(height * width);
	enc_data.yuv.u = malloc(height * width / 4);
	enc_data.yuv.v = malloc(height * width / 4);
	enc_data.yuv.y_width = width;
	enc_data.yuv.y_height = height;
	enc_data.yuv.y_stride = width;

	enc_data.yuv.uv_width = width / 2;
	enc_data.yuv.uv_height = height / 2;
	enc_data.yuv.uv_stride = width / 2;

	{ /* FIXME TODO: move to a rmdYuvInit() */
		for (int i = 0; i < (enc_data.yuv.y_width * enc_data.yuv.y_height); i++)
			enc_data.yuv.y[i] = 0;

		for (int i = 0; i < (enc_data.yuv.uv_width * enc_data.yuv.uv_height); i++)
			enc_data.yuv.v[i] = enc_data.yuv.u[i] = 127;

		yblocks = malloc(sizeof(*yblocks) * (enc_data.yuv.y_width / Y_UNIT_WIDTH) *
					(enc_data.yuv.y_height / Y_UNIT_WIDTH));
		ublocks = malloc(sizeof(*ublocks) * (enc_data.yuv.y_width / Y_UNIT_WIDTH) *
					(enc_data.yuv.y_height / Y_UNIT_WIDTH));
		vblocks = malloc(sizeof(*vblocks) * (enc_data.yuv.y_width / Y_UNIT_WIDTH) *
					(enc_data.yuv.y_height / Y_UNIT_WIDTH));

		rmdYuvBlocksReset(enc_data.yuv.y_width / Y_UNIT_WIDTH,
				  enc_data.yuv.y_height / Y_UNIT_WIDTH);
	}

	pdata.frametime_us = 1000000 / pdata.args.fps;

	pthread_mutex_init(&pdata.theora_lib_mutex, NULL);
	pthread_mutex_init(&pdata.vorbis_lib_mutex, NULL);
	pthread_mutex_init(&pdata.libogg_mutex, NULL);
	pthread_cond_init(&pdata.theora_lib_clean, NULL);
	pthread_cond_init(&pdata.vorbis_lib_clean, NULL);
	pdata.th_encoding_clean = pdata.v_encoding_clean = 1;
	pdata.avd = 0;
	pdata.sound_buffer = NULL;
	pdata.running = TRUE;
	pdata.aborted = FALSE;

	rmdRegisterCallbacks(&pdata);
	fprintf(stderr, "Restoring %s!!!\n", path);

	if (rmdEncodeCache(&pdata))
		return 1;

	fprintf(stderr, "Done!!!\n");
	fprintf(stderr, "Goodbye!\n");
	rmdCleanUp();
	
	return 0;
}