#include "output.h"
#include <libavcodec/avcodec.h>
#include <limits.h>
#include <string.h>
#include <stdint.h>
#include <glib.h>
#include "log.h"
#include "db.h"


static int output_codec_id;
static const char *output_file_format;

int mp3_bitrate;



static void output_shutdown(output_t *output);



static int output_got_packet(encoder_t *enc, void *u1, void *u2) {
	output_t *output = u1;

	dbg("{%s} output avpkt size is %i", output->file_name, (int) enc->avpkt.size);
	dbg("{%s} output pkt pts/dts is %li/%li", output->file_name, (long) enc->avpkt.pts,
			(long) enc->avpkt.dts);
	dbg("{%s} output dts %li", output->file_name, (long) output->encoder->mux_dts);

	av_write_frame(output->fmtctx, &enc->avpkt);

	output->fifo_pts += output->frame->nb_samples;

	return 0;
}


static int output_flush(output_t *output) {
	while (av_audio_fifo_size(output->fifo) >= output->frame->nb_samples) {

		if (av_audio_fifo_read(output->fifo, (void **) output->frame->data,
					output->frame->nb_samples) <= 0)
			abort();

		dbg("{%s} output fifo pts %lu", output->file_name, (unsigned long) output->fifo_pts);
		output->frame->pts = output->fifo_pts;

		encoder_input_data(output->encoder, output->frame, output_got_packet, output, NULL);
	}

	return 0;
}


int output_add(output_t *output, AVFrame *frame) {
	if (!output)
		return -1;
	if (!output->frame) // not ready - not configured
		return -1;

	dbg("{%s} output fifo size %u fifo_pts %lu", output->file_name, (unsigned int) av_audio_fifo_size(output->fifo),
			(unsigned long) output->fifo_pts);
	// fix up output pts
	if (av_audio_fifo_size(output->fifo) == 0)
		output->fifo_pts = frame->pts;

	if (av_audio_fifo_write(output->fifo, (void **) frame->extended_data, frame->nb_samples) < 0)
		return -1;

	return output_flush(output);
}


output_t *output_new(const char *path, const char *filename) {
	output_t *ret = g_slice_alloc0(sizeof(*ret));
	g_strlcpy(ret->file_path, path, sizeof(ret->file_path));
	g_strlcpy(ret->file_name, filename, sizeof(ret->file_name));
	snprintf(ret->full_filename, sizeof(ret->full_filename), "%s/%s", path, filename);
	ret->file_format = output_file_format;
	ret->encoder = encoder_new();
	return ret;
}


int output_config(output_t *output, const format_t *requested_format, format_t *actual_format) {
	const char *err;

//	// anything to do?
//	if (G_LIKELY(format_eq(requested_format, &output->requested_format)))
//		goto done;

	output_shutdown(output);

//	output->requested_format = *requested_format;

	err = "failed to alloc format context";
	output->fmtctx = avformat_alloc_context();
	if (!output->fmtctx)
		goto err;
	output->fmtctx->oformat = av_guess_format(output->file_format, NULL, NULL);
	err = "failed to determine output format";
	if (!output->fmtctx->oformat)
		goto err;

	if (encoder_config(output->encoder, output_codec_id, mp3_bitrate, requested_format, actual_format))
		goto err;

//	err = "output codec not found";
//	AVCodec *codec = avcodec_find_encoder(output_codec_id);
//	if (!codec)
//		goto err;
	err = "failed to alloc output stream";
	output->avst = avformat_new_stream(output->fmtctx, output->encoder->codec);
	if (!output->avst)
		goto err;
//#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 0, 0)
//	err = "failed to alloc codec context";
//	output->avcctx = avcodec_alloc_context3(codec);
//	if (!output->avcctx)
//		goto err;
//#else
//	output->avcctx = output->avst->codec;
//#endif

	// copy all format params
//	output->actual_format = output->requested_format;

	// determine sample format to use
//	output->actual_format.format = -1;
//	for (const enum AVSampleFormat *sfmt = codec->sample_fmts; sfmt && *sfmt != -1; sfmt++) {
//		dbg("supported sample format for output codec %s: %s", codec->name, av_get_sample_fmt_name(*sfmt));
//		if (*sfmt == requested_format->format)
//			output->actual_format.format = *sfmt;
//	}
//	if (output->actual_format.format == -1 && codec->sample_fmts)
//		output->actual_format.format = codec->sample_fmts[0];
//	dbg("using output sample format %s for codec %s", av_get_sample_fmt_name(output->actual_format.format), codec->name);
//
//	output->avcctx->channels = output->actual_format.channels;
//	output->avcctx->channel_layout = av_get_default_channel_layout(output->actual_format.channels);
//	output->avcctx->sample_rate = output->actual_format.clockrate;
//	output->avcctx->sample_fmt = output->actual_format.format;
//	output->avcctx->time_base = (AVRational){1,output->actual_format.clockrate};
//	output->avcctx->bit_rate = mp3_bitrate;
	output->avst->time_base = output->encoder->avcctx->time_base;

#if LIBAVFORMAT_VERSION_INT >= AV_VERSION_INT(57, 26, 0) // exact version? present in 57.56
	avcodec_parameters_from_context(output->avst->codecpar, output->encoder->avcctx);
#endif

	char full_fn[PATH_MAX*2];
	char suff[16] = "";
	for (int i = 1; i < 20; i++) {
		snprintf(full_fn, sizeof(full_fn), "%s%s.%s", output->full_filename, suff, output->file_format);
		if (!g_file_test(full_fn, G_FILE_TEST_EXISTS))
			goto got_fn;
		snprintf(suff, sizeof(suff), "-%i", i);
	}

	err = "failed to find unused output file number";
	goto err;

got_fn:
//	err = "failed to open output context";
//	int i = avcodec_open2(output->avcctx, codec, NULL);
//	if (i)
//		goto err;
	err = "failed to open avio";
	int i;
	i = avio_open(&output->fmtctx->pb, full_fn, AVIO_FLAG_WRITE);
	if (i < 0)
		goto err;
	err = "failed to write header";
	i = avformat_write_header(output->fmtctx, NULL);
	if (i)
		goto err;

//	av_init_packet(&output->avpkt);

	// output frame and fifo
	output->frame = av_frame_alloc();
	output->frame->nb_samples = output->encoder->avcctx->frame_size ? : 256;
	output->frame->format = output->encoder->avcctx->sample_fmt;
	output->frame->sample_rate = output->encoder->avcctx->sample_rate;
	output->frame->channel_layout = output->encoder->avcctx->channel_layout;
	if (!output->frame->channel_layout)
		output->frame->channel_layout = av_get_default_channel_layout(output->encoder->avcctx->channels);
	if (av_frame_get_buffer(output->frame, 0) < 0)
		abort();

	output->fifo = av_audio_fifo_alloc(output->encoder->avcctx->sample_fmt, output->encoder->avcctx->channels,
			output->frame->nb_samples);

	db_config_stream(output);
	return 0;

err:
	output_shutdown(output);
	ilog(LOG_ERR, "Error configuring media output: %s", err);
	return -1;
}


static void output_shutdown(output_t *output) {
	if (!output)
		return;
	if (!output->fmtctx)
		return;

	if (output->fmtctx->pb) {
		av_write_trailer(output->fmtctx);
		avio_closep(&output->fmtctx->pb);
	}
//	avcodec_close(output->avcctx);
//#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(57, 0, 0)
//	avcodec_free_context(&output->avcctx);
//#endif
	avformat_free_context(output->fmtctx);
	av_audio_fifo_free(output->fifo);
	av_frame_free(&output->frame);

	encoder_close(output->encoder);

//	output->avcctx = NULL;
	output->fmtctx = NULL;
	output->avst = NULL;
	output->fifo = NULL;

	output->fifo_pts = 0;

//	format_init(&output->requested_format);
//	format_init(&output->actual_format);
}


void output_close(output_t *output) {
	if (!output)
		return;
	output_shutdown(output);
	db_close_stream(output);
	encoder_free(output->encoder);
	g_slice_free1(sizeof(*output), output);
}


void output_init(const char *format) {
	if (!strcmp(format, "wav")) {
		output_codec_id = AV_CODEC_ID_PCM_S16LE;
		output_file_format = "wav";
	}
	else if (!strcmp(format, "mp3")) {
		output_codec_id = AV_CODEC_ID_MP3;
		output_file_format = "mp3";
	}
	else
		die("Unknown output format '%s'", format);
}