libstb-hal/libeplayer3/output/output_subtitle.c

/*
 * Subtitle output to one registered client.
 *
 *
 * 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 <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <inttypes.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <memory.h>
#include <asm/types.h>
#include <errno.h>

#include "common.h"
#include "debug.h"
#include "output.h"

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

/* Error Constants */
#define cERR_SUBTITLE_NO_ERROR         0
#define cERR_SUBTITLE_ERROR            -1

/*
Number, Style, Name,, MarginL, MarginR, MarginV, Effect,, Text

1038,0,tdk,,0000,0000,0000,,That's not good.
1037,0,tdk,,0000,0000,0000,,{\i1}Rack them up, rack them up,{\i0}\N{\i1}rack them up.{\i0} [90]
1036,0,tdk,,0000,0000,0000,,Okay, rack them up.
*/

#define PUFFERSIZE 20

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


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

static pthread_mutex_t mutex;
static int isSubtitleOpened = 0;

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

/* ***************************** */
/* MISC Functions                */
/* ***************************** */

static void getMutex(int line __attribute__((unused)))
{
	subtitle_printf(100, "%d requesting mutex\n", line);

	pthread_mutex_lock(&mutex);

	subtitle_printf(100, "%d received mutex\n", line);
}

static void releaseMutex(int line __attribute__((unused)))
{
	pthread_mutex_unlock(&mutex);

	subtitle_printf(100, "%d released mutex\n", line);
}

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

static char *ass_get_text(char *str)
{
	// Events are stored in the Block in this order:
	// ReadOrder, Layer, Style, Name, MarginL, MarginR, MarginV, Effect, Text
	// 91,0,Default,,0,0,0,,maar hij smaakt vast tof.
	int i = 0;
	char *p_str = str;
	while (i < 8 && *p_str != '\0')
	{
		if (*p_str == ',')
			i++;
		p_str++;
	}
	// standardize hard break: '\N' -> '\n'
	// http://docs.aegisub.org/3.2/ASS_Tags/
	char *p_newline = NULL;
	while ((p_newline = strstr(p_str, "\\N")) != NULL)
		* (p_newline + 1) = 'n';
	return p_str;
}

static char *json_string_escape(char *str)
{
	static char tmp[2048];
	char *ptr1 = tmp;
	char *ptr2 = str;
	while (*ptr2 != '\0')
	{
		switch (*ptr2)
		{
			case '"':
				*ptr1++ = '\\';
				*ptr1++ = '\"';
				break;
			case '\\':
				*ptr1++ = '\\';
				*ptr1++ = '\\';
				break;
			case '\b':
				*ptr1++ = '\\';
				*ptr1++ = 'b';
				break;
			case '\f':
				*ptr1++ = '\\';
				*ptr1++ = 'f';
				break;
			case '\n':
				*ptr1++ = '\\';
				*ptr1++ = 'n';
				break;
			case '\r':
				*ptr1++ = '\\';
				*ptr1++ = 'r';
				break;
			case '\t':
				*ptr1++ = '\\';
				*ptr1++ = 't';
				break;
			default:
				*ptr1++ = *ptr2;
				break;
		}

		++ptr2;
	}
	*ptr1 = '\0';
	return tmp;
}

static int Flush()
{
	fprintf(stderr, "{\"s_f\":{\"r\":0}}\n");
	return cERR_SUBTITLE_NO_ERROR;
}

static int Write(Context_t *context, void *data)
{
	char *Encoding      = NULL;
	SubtitleOut_t *out  = NULL;
	int32_t curtrackid  = -1;

	subtitle_printf(10, "\n");

	if (data == NULL)
	{
		subtitle_err("null pointer passed\n");
		return cERR_SUBTITLE_ERROR;
	}

	out = (SubtitleOut_t *) data;

	context->manager->subtitle->Command(context, MANAGER_GET, &curtrackid);
	if (curtrackid != (int32_t)out->trackId)
	{
		Flush();
	}
	context->manager->subtitle->Command(context, MANAGER_GETENCODING, &Encoding);

	if (Encoding == NULL)
	{
		subtitle_err("encoding unknown\n");
		return cERR_SUBTITLE_ERROR;
	}

	subtitle_printf(20, "Encoding:%s Text:%s Len:%d\n", Encoding, (const char *) out->data, out->len);

	if (!strncmp("S_TEXT/SUBRIP", Encoding, 13))
	{
		fprintf(stderr, "{\"s_a\":{\"id\":%d,\"s\":%" PRId64 ",\"e\":%" PRId64 ",\"t\":\"%s\"}}\n", out->trackId, out->pts / 90, out->pts / 90 + out->durationMS, json_string_escape((char *)out->data));
	}
	else if (!strncmp("S_TEXT/ASS", Encoding, 10))
	{
		fprintf(stderr, "{\"s_a\":{\"id\":%d,\"s\":%" PRId64 ",\"e\":%" PRId64 ",\"t\":\"%s\"}}\n", out->trackId, out->pts / 90, out->pts / 90 + out->durationMS, ass_get_text((char *)out->data));
	}
	else if (!strncmp("D_WEBVTT/SUBTITLES", Encoding, 18))
	{
		fprintf(stderr, "{\"s_a\":{\"id\":%d,\"s\":%" PRId64 ",\"e\":%" PRId64 ",\"t\":\"%s\"}}\n", out->trackId, out->pts / 90, out->pts / 90 + out->durationMS, json_string_escape((char *)out->data));
	}
	else
	{
		subtitle_err("unknown encoding %s\n", Encoding);
		return cERR_SUBTITLE_ERROR;
	}

	subtitle_printf(10, "<\n");
	return cERR_SUBTITLE_NO_ERROR;
}

static int32_t subtitle_Open(Context_t *context __attribute__((unused)))
{
	//uint32_t i = 0 ;

	subtitle_printf(10, "\n");

	if (isSubtitleOpened == 1)
	{
		subtitle_err("already opened! ignoring\n");
		return cERR_SUBTITLE_ERROR;
	}

	getMutex(__LINE__);

	isSubtitleOpened = 1;

	releaseMutex(__LINE__);

	subtitle_printf(10, "<\n");
	return cERR_SUBTITLE_NO_ERROR;
}

static int32_t subtitle_Close(Context_t *context __attribute__((unused)))
{
	subtitle_printf(10, "\n");

	getMutex(__LINE__);

	isSubtitleOpened = 0;

	releaseMutex(__LINE__);

	subtitle_printf(10, "<\n");

	return cERR_SUBTITLE_NO_ERROR;
}

static int Command(Context_t *context, OutputCmd_t command, void *argument __attribute__((unused)))
{
	int ret = cERR_SUBTITLE_NO_ERROR;

	subtitle_printf(50, "%d\n", command);

	switch (command)
	{
		case OUTPUT_OPEN:
		{
			ret = subtitle_Open(context);
			break;
		}
		case OUTPUT_CLOSE:
		{
			ret = subtitle_Close(context);
			break;
		}
		case OUTPUT_PLAY:
		{
			break;
		}
		case OUTPUT_STOP:
		{
			break;
		}
		case OUTPUT_SWITCH:
		{
			ret = Flush();
			break;
		}
		case OUTPUT_FLUSH:
		{
			ret = Flush();
			break;
		}
		case OUTPUT_CLEAR:
		{
			ret = Flush();
			break;
		}
		case OUTPUT_PAUSE:
		{
			subtitle_err("Subtitle Pause not implemented\n");
			ret = cERR_SUBTITLE_ERROR;
			break;
		}
		case OUTPUT_CONTINUE:
		{
			subtitle_err("Subtitle Continue not implemented\n");
			ret = cERR_SUBTITLE_ERROR;
			break;
		}
		default:
			subtitle_err("OutputCmd %d not supported!\n", command);
			ret = cERR_SUBTITLE_ERROR;
			break;
	}

	subtitle_printf(50, "exiting with value %d\n", ret);

	return ret;
}


static char *SubtitleCapabilitis[] = { "subtitle", NULL };

Output_t SubtitleOutput =
{
	"Subtitle",
	&Command,
	&Write,
	SubtitleCapabilitis
};