dmav_subtitles.c

/**
 * dmav_subtitles.c - parse subtitle data in DMAV sections of voice clips
 *
 * Copyright (c) 2011 Michael Lelli
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include "bnk_pc.h"
#include "char_tbl_latin.h"
#include "char_tbl_jp.h"
#include "char_tbl_sk.h"

//#define GENERATE_CHARTABLES

const char langs[14][3] = {
"us",
"es",
"it",
"jp",
"de",
"fr",
"nl",
"??",
"??",
"cz",
"pl",
"sk",
"ru",
"??"
};

const u16 *lang_tables[14] = {
char_tbl_latin,
char_tbl_latin,
char_tbl_latin,
char_tbl_jp,
char_tbl_latin,
char_tbl_latin,
char_tbl_latin,
0,
0,
char_tbl_latin,
char_tbl_latin,
char_tbl_sk,
char_tbl_latin,
0
};

#ifdef GENERATE_CHARTABLES

// all the latin/cyrillic character tables are identical
const char *charlists[14] = {
"us",
"us",
"us",
"jp",
"us",
"us",
"us",
0,
0,
"us",
"us",
"sk",
"us",
0
};

const u16 *charlists_lengths[14] = {
&char_tbl_latin_size,
&char_tbl_latin_size,
&char_tbl_latin_size,
&char_tbl_jp_size,
&char_tbl_latin_size,
&char_tbl_latin_size,
&char_tbl_latin_size,
0,
0,
&char_tbl_latin_size,
&char_tbl_latin_size,
&char_tbl_sk_size,
&char_tbl_latin_size,
0
};

u16 gen_table[0x10000];

int parse_table(int langnum)
{
	char name[1024];
	char buffer[1024];
	FILE *o;
	u16 len = 0;
	int i, offset = 256;

	memset(gen_table, 0, sizeof(gen_table));

	if (!charlists[langnum])
	{
		return 0;
	}

	sprintf(name, "charlist_%s.dat", charlists[langnum]);
	o = fopen(name, "r");

	if (o == NULL)
	{
		printf("cannot open %s for character data\n", name);
		return 1;
	}

	while (len == 0)
	{
		fgets(buffer, sizeof(buffer), o);

		if (feof(o))
		{
			printf("malformed charlist file (%s)\n", name);
			fclose(o);
			return 1;
		}

		sscanf(buffer, "count=%hu\r\n", &len);
	}

	for (i = 0; i < len; i++)
	{
		u16 c;
		fscanf(o, "%hu\r\n", &c);

		if (c <= 0xff)
		{
			gen_table[c] = c;
		}
		else
		{
			gen_table[offset++] = c;
		}
	}

	// check it

	if (len != *charlists_lengths[langnum])
	{
		printf("GENERROR ON %d (%s): Wrong length (expected %d, got %d)\n", langnum, name, *charlists_lengths[langnum], len);
	}

	// + 256 to account for unassigned characters in the 0 -> 255 range
	for (i = 0; i < *charlists_lengths[langnum] + 256; i++)
	{
		if (gen_table[i] != lang_tables[langnum][i])
		{
			printf("GENERROR ON %d (%s): Wrong character at offset %d (expected %d, got %d)\n", langnum, name, i, lang_tables[langnum][i], gen_table[i]);
		}
	}

	fclose(o);

	return 0;
}

#endif

void u16fprintf(FILE *f, char *format, ...)
{
	char buffer[1024];
	va_list args;
	u32 i;

	va_start(args, format);
	vsprintf(buffer, format, args);

	for (i = 0; i < strlen(buffer); i++)
	{
		u16 c = buffer[i];
		const u16 r = '\r';

		if (c == '\n')
		{
			fwrite(&r, 2, 1, f);
		}

		fwrite(&c, 2, 1, f);
	}

	va_end(args);
}

#define fprintf u16fprintf

int main(int argc, char *argv[])
{
	FILE *f = NULL;
	FILE *o = NULL;
	dmav_header head;
	dmav_subtitle_header sub_head[28];
	u32 i, t = 0, offset = sizeof(sub_head) + 4;
	const u16 bom = 0xFEFF;
	int r = 0;
	char name[1024];
	char oname[1024];

	if (argc != 2)
	{
		printf("dmav_subtitles " VERSION "\n"
		       "usage: %s voicedata.dmav\n", argv[0]);
		goto error;
	}

	memcpy(name, argv[1], strlen(argv[1]) + 1);
	name[strrchr(name, '.') - name] = 0;

	f = fopen(argv[1], "rb");

	if (f == NULL)
	{
		printf("cannot open file\n");
		goto error;
	}

	fread(&head, sizeof(dmav_header), 1, f);

	if (head.magic != DMAV_HEADER)
	{
		printf("not a dmav file\n");
		goto error;
	}

	if (head.offset1 != head.offset2)
	{
		printf("WARNING: offsets not equal, results might not be correct\n");
	}

	fseek(f, head.offset1 + sizeof(dmav_header), SEEK_SET);
	fread(&t, 4, 1, f);

	if (t == 0)
	{
		printf("file has no subtitle data\n");
		goto end;
	}

	sprintf(oname, "%s_subs.txt", name);
	o = fopen(oname, "wb");
	fwrite(&bom, 2, 1, o); // Byte-order mask for utf-16
	fprintf(o, "HEADER:\n");
	fprintf(o, "magic:        \"DMAV\"\n");
	fprintf(o, "version:      0x%08X\n", head.version);
	fprintf(o, "persona_id:   0x%08X\n", head.persona_id);
	fprintf(o, "voiceline_id: 0x%08X\n", head.voiceline_id);
	fprintf(o, "offset1:      0x%08X\n", head.offset1);
	fprintf(o, "u1:           0x%08X\n", head.u1);
	fprintf(o, "u2:           0x%08X\n", head.u2);
	fprintf(o, "offset2:      0x%08X\n", head.offset2);
	fprintf(o, "u3:           0x%08X\n\n", head.u3);
	fprintf(o, "SUBTITLE HEADERS:\n\n");

	if (t != 2)
	{
		fseek(f, head.offset1 + sizeof(dmav_header), SEEK_SET);
		offset = sizeof(dmav_subtitle_header) * 2;
	}

	// loop 1: get subtitle headers
	for (i = 0; (unsigned int) i < sizeof(sub_head) / sizeof(*sub_head); i++)
	{
		if (offset == sizeof(dmav_subtitle_header) * 2 && i > 1)
		{
			break;
		}

		fread(&sub_head[i], sizeof(dmav_subtitle_header), 1, f);
	}

	// loop 2: output headers + get subtitles
	for (i = 0; i < sizeof(sub_head) / sizeof(*sub_head); i++)
	{
		int l = i % 14;
		u32 count = 0;

		if (offset == sizeof(dmav_subtitle_header) * 2 && i > 1)
		{
			break;
		}

		#ifdef GENERATE_CHARTABLES
		if (parse_table(l))
		{
			printf("failed to parse chartable for %s\n", charlists[l]);
			goto error;
		}
		#endif

		fprintf(o, "size: 0x%08X\n", sub_head[i].size);
		fprintf(o, "offset: 0x%08X\n", sub_head[i].offset);
		fprintf(o, "%s: ", langs[l]);

		fseek(f, head.offset1 + sizeof(dmav_header) + offset + sub_head[i].offset, SEEK_SET);

		while (count + 2 < sub_head[i].size)
		{
			u16 charcode;

			fread(&charcode, 2, 1, f);
			count += 2;
			#ifdef GENERATE_CHARTABLES
			charcode = gen_table[charcode];
			#else
			charcode = lang_tables[l][charcode];
			#endif

			fwrite(&charcode, 2, 1, o);
		}

		fprintf(o, "\n\n");
	}
end:

	if (f != NULL)
	{
		fclose(f);
	}

	if (o != NULL)
	{
		fclose(o);
	}

	return r;

error:
	r = 1;
	goto end;
}