/*
 * linuxdvb output/writer handling.
 *
 * konfetti 2010
 *
 * 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
 *
 */

/* ***************************** */
/* Includes                      */
/* ***************************** */

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

#include "misc.h"
#include "writer.h"
#include "common.h"
#include "debug.h"

/* ***************************** */
/* Makros/Constants              */
/* ***************************** */

#define getDVBMutex(pmtx) do { if (pmtx) pthread_mutex_lock(pmtx);} while(false);
#define releaseDVBMutex(pmtx) do { if (pmtx) pthread_mutex_unlock(pmtx);} while(false);

/* ***************************** */
/* Types                         */
/* ***************************** */

typedef enum
{
	DVB_STS_UNKNOWN,
	DVB_STS_SEEK,
	DVB_STS_PAUSE,
	DVB_STS_EXIT
} DVBState_t;

/* ***************************** */
/* Variables                     */
/* ***************************** */

static Writer_t *AvailableWriter[] =
{
	&WriterAudioAAC,
	&WriterAudioAACLATM,
	&WriterAudioAACPLUS,
	&WriterAudioAC3,
	&WriterAudioEAC3,
	&WriterAudioMP3,
	&WriterAudioMPEGL3,
	&WriterAudioPCM,
	&WriterAudioIPCM,
	&WriterAudioLPCM,
	&WriterAudioDTS,
	&WriterAudioWMA,
	&WriterAudioWMAPRO,
	&WriterAudioOPUS,
	&WriterAudioVORBIS,

	&WriterVideoH264,
	&WriterVideoH265,
	&WriterVideoH263,
	&WriterVideoMPEG4,
	&WriterVideoMPEG2,
	&WriterVideoMPEG1,
	&WriterVideoVC1,
	&WriterVideoDIVX3,
	&WriterVideoVP6,
	&WriterVideoVP8,
	&WriterVideoVP9,
	&WriterVideoFLV,
	&WriterVideoWMV,
	&WriterVideoMJPEG,
	&WriterVideoRV40,
	&WriterVideoRV30,
	&WriterVideoAVS2,
	NULL
};

/* ***************************** */
/* Prototypes                    */
/* ***************************** */

/* ***************************** */
/*  Functions                    */
/* ***************************** */

ssize_t WriteWithRetry(Context_t *context, int pipefd, int fd, void *pDVBMtx __attribute__((unused)), const void *buf, int size)
{
	fd_set rfds;
	fd_set wfds;

	ssize_t ret;
	int retval = -1;
	int maxFd = pipefd > fd ? pipefd : fd;
	struct timeval tv;
	while (size > 0 && PlaybackDieNow(0) == 0 && !context->playback->isSeeking)
	{
		FD_ZERO(&rfds);
		FD_ZERO(&wfds);

		FD_SET(pipefd, &rfds);
		FD_SET(fd, &wfds);

		/* When we PAUSE LINUX DVB outputs buffers, then audio/video buffers
		 * will continue to be filled. Unfortunately, in such case after resume
		 * select() will never return with fd set - bug in DVB drivers?
		 * There are to workarounds possible:
		 *   1. write to pipe at resume to return from select() immediately
		 *   2. make timeout select(), limit max time spend in the select()
		 *      to for example 0,1s
		 *   (at now first workaround is used)
		 */
		//tv.tv_sec = 0;
		//tv.tv_usec = 100000; // 100ms

		retval = select(maxFd + 1, &rfds, &wfds, NULL, NULL); //&tv);
		if (retval < 0)
		{
			break;
		}

		//if (retval == 0)
		//{
		//	//printf("RETURN FROM SELECT DUE TO TIMEOUT\n");
		//	continue;
		//}

		if (FD_ISSET(pipefd, &rfds))
		{
			FlushPipe(pipefd);
			//printf("RETURN FROM SELECT DUE TO pipefd SET\n");
			continue;
		}

		if (FD_ISSET(fd, &wfds))
		{
			ret = write(fd, buf, size);
			if (ret < 0)
			{
				switch (errno)
				{
					case EINTR:
					case EAGAIN:
						continue;
					default:
						retval = -3;
						break;
				}
				if (retval < 0)
				{
					break;
				}

				return ret;
			}
			else if (ret == 0)
			{
				// printf("This should not happen. Select return fd ready to write, but write return 0, errno [%d]\n", errno);
				// wait 10ms before next try
				tv.tv_sec = 0;
				tv.tv_usec = 10000; // 10ms
				retval = select(pipefd + 1, &rfds, NULL, NULL, &tv);
				if (retval)
					FlushPipe(pipefd);
				continue;
			}

			size -= ret;
			buf += ret;
		}
	}
	return 0;
}

ssize_t write_with_retry(int fd, const void *buf, int size)
{
	ssize_t ret;
	int retval = 0;
	while (size > 0 && PlaybackDieNow(0) == 0)
	{
		ret = write(fd, buf, size);
		if (ret < 0)
		{
			switch (errno)
			{
				case EINTR:
				case EAGAIN:
					usleep(1000);
					continue;
				default:
					retval = -3;
					break;
			}
			if (retval < 0)
			{
				break;
			}
		}

		if (ret < 0)
		{
			return ret;
		}

		size -= ret;
		buf += ret;

		if (size > 0)
		{
			if (usleep(1000))
			{
				writer_err("usleep error \n");
			}
		}
	}
	return 0;
}

ssize_t writev_with_retry(int fd, const struct iovec *iov, int ic)
{
	ssize_t len = 0;
	int i = 0;
	for (i = 0; i < ic; ++i)
	{
		write_with_retry(fd, iov[i].iov_base, iov[i].iov_len);
		len += iov[i].iov_len;
		if (PlaybackDieNow(0))
		{
			return -1;
		}
	}
	return len;
}

Writer_t *getWriter(char *encoding)
{
	int i;

	for (i = 0; AvailableWriter[i] != NULL; i++)
	{
		if (strcmp(AvailableWriter[i]->caps->textEncoding, encoding) == 0)
		{
			writer_printf(50, "%s: found writer \"%s\" for \"%s\"\n", __func__, AvailableWriter[i]->caps->name, encoding);
			return AvailableWriter[i];
		}
	}

	writer_printf(1, "%s: no writer found for \"%s\"\n", __func__, encoding);

	return NULL;
}

Writer_t *getDefaultVideoWriter()
{
	int i;

	for (i = 0; AvailableWriter[i] != NULL; i++)
	{
		if (strcmp(AvailableWriter[i]->caps->textEncoding, "V_MPEG2") == 0)
		{
			writer_printf(50, "%s: found writer \"%s\"\n", __func__, AvailableWriter[i]->caps->name);
			return AvailableWriter[i];
		}
	}

	writer_printf(1, "%s: no writer found\n", __func__);

	return NULL;
}

Writer_t *getDefaultAudioWriter()
{
	int i;

	for (i = 0; AvailableWriter[i] != NULL; i++)
	{
		if (strcmp(AvailableWriter[i]->caps->textEncoding, "A_MP3") == 0)
		{
			writer_printf(50, "%s: found writer \"%s\"\n", __func__, AvailableWriter[i]->caps->name);
			return AvailableWriter[i];
		}
	}

	writer_printf(1, "%s: no writer found\n", __func__);

	return NULL;
}