yading@10: /* yading@10: * SSA/ASS spliting functions yading@10: * Copyright (c) 2010 Aurelien Jacobs yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #include "avcodec.h" yading@10: #include "ass_split.h" yading@10: yading@10: typedef enum { yading@10: ASS_STR, yading@10: ASS_INT, yading@10: ASS_FLT, yading@10: ASS_COLOR, yading@10: ASS_TIMESTAMP, yading@10: ASS_ALGN, yading@10: } ASSFieldType; yading@10: yading@10: typedef struct { yading@10: const char *name; yading@10: int type; yading@10: int offset; yading@10: } ASSFields; yading@10: yading@10: typedef struct { yading@10: const char *section; yading@10: const char *format_header; yading@10: const char *fields_header; yading@10: int size; yading@10: int offset; yading@10: int offset_count; yading@10: ASSFields fields[10]; yading@10: } ASSSection; yading@10: yading@10: static const ASSSection ass_sections[] = { yading@10: { .section = "Script Info", yading@10: .offset = offsetof(ASS, script_info), yading@10: .fields = {{"ScriptType", ASS_STR, offsetof(ASSScriptInfo, script_type)}, yading@10: {"Collisions", ASS_STR, offsetof(ASSScriptInfo, collisions) }, yading@10: {"PlayResX", ASS_INT, offsetof(ASSScriptInfo, play_res_x) }, yading@10: {"PlayResY", ASS_INT, offsetof(ASSScriptInfo, play_res_y) }, yading@10: {"Timer", ASS_FLT, offsetof(ASSScriptInfo, timer) }, yading@10: {0}, yading@10: } yading@10: }, yading@10: { .section = "V4+ Styles", yading@10: .format_header = "Format", yading@10: .fields_header = "Style", yading@10: .size = sizeof(ASSStyle), yading@10: .offset = offsetof(ASS, styles), yading@10: .offset_count = offsetof(ASS, styles_count), yading@10: .fields = {{"Name", ASS_STR, offsetof(ASSStyle, name) }, yading@10: {"Fontname", ASS_STR, offsetof(ASSStyle, font_name) }, yading@10: {"Fontsize", ASS_INT, offsetof(ASSStyle, font_size) }, yading@10: {"PrimaryColour",ASS_COLOR,offsetof(ASSStyle, primary_color)}, yading@10: {"BackColour", ASS_COLOR,offsetof(ASSStyle, back_color) }, yading@10: {"Bold", ASS_INT, offsetof(ASSStyle, bold) }, yading@10: {"Italic", ASS_INT, offsetof(ASSStyle, italic) }, yading@10: {"Underline", ASS_INT, offsetof(ASSStyle, underline) }, yading@10: {"Alignment", ASS_INT, offsetof(ASSStyle, alignment) }, yading@10: {0}, yading@10: } yading@10: }, yading@10: { .section = "V4 Styles", yading@10: .format_header = "Format", yading@10: .fields_header = "Style", yading@10: .size = sizeof(ASSStyle), yading@10: .offset = offsetof(ASS, styles), yading@10: .offset_count = offsetof(ASS, styles_count), yading@10: .fields = {{"Name", ASS_STR, offsetof(ASSStyle, name) }, yading@10: {"Fontname", ASS_STR, offsetof(ASSStyle, font_name) }, yading@10: {"Fontsize", ASS_INT, offsetof(ASSStyle, font_size) }, yading@10: {"PrimaryColour",ASS_COLOR,offsetof(ASSStyle, primary_color)}, yading@10: {"BackColour", ASS_COLOR,offsetof(ASSStyle, back_color) }, yading@10: {"Bold", ASS_INT, offsetof(ASSStyle, bold) }, yading@10: {"Italic", ASS_INT, offsetof(ASSStyle, italic) }, yading@10: {"Alignment", ASS_ALGN, offsetof(ASSStyle, alignment) }, yading@10: {0}, yading@10: } yading@10: }, yading@10: { .section = "Events", yading@10: .format_header = "Format", yading@10: .fields_header = "Dialogue", yading@10: .size = sizeof(ASSDialog), yading@10: .offset = offsetof(ASS, dialogs), yading@10: .offset_count = offsetof(ASS, dialogs_count), yading@10: .fields = {{"Layer", ASS_INT, offsetof(ASSDialog, layer) }, yading@10: {"Start", ASS_TIMESTAMP, offsetof(ASSDialog, start) }, yading@10: {"End", ASS_TIMESTAMP, offsetof(ASSDialog, end) }, yading@10: {"Style", ASS_STR, offsetof(ASSDialog, style) }, yading@10: {"Text", ASS_STR, offsetof(ASSDialog, text) }, yading@10: {0}, yading@10: } yading@10: }, yading@10: }; yading@10: yading@10: yading@10: typedef int (*ASSConvertFunc)(void *dest, const char *buf, int len); yading@10: yading@10: static int convert_str(void *dest, const char *buf, int len) yading@10: { yading@10: char *str = av_malloc(len + 1); yading@10: if (str) { yading@10: memcpy(str, buf, len); yading@10: str[len] = 0; yading@10: if (*(void **)dest) yading@10: av_free(*(void **)dest); yading@10: *(char **)dest = str; yading@10: } yading@10: return !str; yading@10: } yading@10: static int convert_int(void *dest, const char *buf, int len) yading@10: { yading@10: return sscanf(buf, "%d", (int *)dest) == 1; yading@10: } yading@10: static int convert_flt(void *dest, const char *buf, int len) yading@10: { yading@10: return sscanf(buf, "%f", (float *)dest) == 1; yading@10: } yading@10: static int convert_color(void *dest, const char *buf, int len) yading@10: { yading@10: return sscanf(buf, "&H%8x", (int *)dest) == 1 || yading@10: sscanf(buf, "%d", (int *)dest) == 1; yading@10: } yading@10: static int convert_timestamp(void *dest, const char *buf, int len) yading@10: { yading@10: int c, h, m, s, cs; yading@10: if ((c = sscanf(buf, "%d:%02d:%02d.%02d", &h, &m, &s, &cs)) == 4) yading@10: *(int *)dest = 360000*h + 6000*m + 100*s + cs; yading@10: return c == 4; yading@10: } yading@10: static int convert_alignment(void *dest, const char *buf, int len) yading@10: { yading@10: int a; yading@10: if (sscanf(buf, "%d", &a) == 1) { yading@10: /* convert V4 Style alignment to V4+ Style */ yading@10: *(int *)dest = a + ((a&4) >> 1) - 5*!!(a&8); yading@10: return 1; yading@10: } yading@10: return 0; yading@10: } yading@10: yading@10: static const ASSConvertFunc convert_func[] = { yading@10: [ASS_STR] = convert_str, yading@10: [ASS_INT] = convert_int, yading@10: [ASS_FLT] = convert_flt, yading@10: [ASS_COLOR] = convert_color, yading@10: [ASS_TIMESTAMP] = convert_timestamp, yading@10: [ASS_ALGN] = convert_alignment, yading@10: }; yading@10: yading@10: yading@10: struct ASSSplitContext { yading@10: ASS ass; yading@10: int current_section; yading@10: int field_number[FF_ARRAY_ELEMS(ass_sections)]; yading@10: int *field_order[FF_ARRAY_ELEMS(ass_sections)]; yading@10: }; yading@10: yading@10: yading@10: static uint8_t *realloc_section_array(ASSSplitContext *ctx) yading@10: { yading@10: const ASSSection *section = &ass_sections[ctx->current_section]; yading@10: int *count = (int *)((uint8_t *)&ctx->ass + section->offset_count); yading@10: void **section_ptr = (void **)((uint8_t *)&ctx->ass + section->offset); yading@10: uint8_t *tmp = av_realloc(*section_ptr, (*count+1)*section->size); yading@10: if (!tmp) yading@10: return NULL; yading@10: *section_ptr = tmp; yading@10: tmp += *count * section->size; yading@10: memset(tmp, 0, section->size); yading@10: (*count)++; yading@10: return tmp; yading@10: } yading@10: yading@10: static inline int is_eol(char buf) yading@10: { yading@10: return buf == '\r' || buf == '\n' || buf == 0; yading@10: } yading@10: yading@10: static inline const char *skip_space(const char *buf) yading@10: { yading@10: while (*buf == ' ') yading@10: buf++; yading@10: return buf; yading@10: } yading@10: yading@10: static const char *ass_split_section(ASSSplitContext *ctx, const char *buf) yading@10: { yading@10: const ASSSection *section = &ass_sections[ctx->current_section]; yading@10: int *number = &ctx->field_number[ctx->current_section]; yading@10: int *order = ctx->field_order[ctx->current_section]; yading@10: int *tmp, i, len; yading@10: yading@10: while (buf && *buf) { yading@10: if (buf[0] == '[') { yading@10: ctx->current_section = -1; yading@10: break; yading@10: } yading@10: if (buf[0] == ';' || (buf[0] == '!' && buf[1] == ':')) { yading@10: /* skip comments */ yading@10: } else if (section->format_header && !order) { yading@10: len = strlen(section->format_header); yading@10: if (strncmp(buf, section->format_header, len) || buf[len] != ':') yading@10: return NULL; yading@10: buf += len + 1; yading@10: while (!is_eol(*buf)) { yading@10: buf = skip_space(buf); yading@10: len = strcspn(buf, ", \r\n"); yading@10: if (!(tmp = av_realloc(order, (*number + 1) * sizeof(*order)))) yading@10: return NULL; yading@10: order = tmp; yading@10: order[*number] = -1; yading@10: for (i=0; section->fields[i].name; i++) yading@10: if (!strncmp(buf, section->fields[i].name, len)) { yading@10: order[*number] = i; yading@10: break; yading@10: } yading@10: (*number)++; yading@10: buf = skip_space(buf + len + (buf[len] == ',')); yading@10: } yading@10: ctx->field_order[ctx->current_section] = order; yading@10: } else if (section->fields_header) { yading@10: len = strlen(section->fields_header); yading@10: if (!strncmp(buf, section->fields_header, len) && buf[len] == ':') { yading@10: uint8_t *ptr, *struct_ptr = realloc_section_array(ctx); yading@10: if (!struct_ptr) return NULL; yading@10: buf += len + 1; yading@10: for (i=0; !is_eol(*buf) && i < *number; i++) { yading@10: int last = i == *number - 1; yading@10: buf = skip_space(buf); yading@10: len = strcspn(buf, last ? "\r\n" : ",\r\n"); yading@10: if (order[i] >= 0) { yading@10: ASSFieldType type = section->fields[order[i]].type; yading@10: ptr = struct_ptr + section->fields[order[i]].offset; yading@10: convert_func[type](ptr, buf, len); yading@10: } yading@10: buf += len; yading@10: if (!last && *buf) buf++; yading@10: buf = skip_space(buf); yading@10: } yading@10: } yading@10: } else { yading@10: len = strcspn(buf, ":\r\n"); yading@10: if (buf[len] == ':') { yading@10: for (i=0; section->fields[i].name; i++) yading@10: if (!strncmp(buf, section->fields[i].name, len)) { yading@10: ASSFieldType type = section->fields[i].type; yading@10: uint8_t *ptr = (uint8_t *)&ctx->ass + section->offset; yading@10: ptr += section->fields[i].offset; yading@10: buf = skip_space(buf + len + 1); yading@10: convert_func[type](ptr, buf, strcspn(buf, "\r\n")); yading@10: break; yading@10: } yading@10: } yading@10: } yading@10: buf += strcspn(buf, "\n"); yading@10: buf += !!*buf; yading@10: } yading@10: return buf; yading@10: } yading@10: yading@10: static int ass_split(ASSSplitContext *ctx, const char *buf) yading@10: { yading@10: char c, section[16]; yading@10: int i; yading@10: yading@10: if (ctx->current_section >= 0) yading@10: buf = ass_split_section(ctx, buf); yading@10: yading@10: while (buf && *buf) { yading@10: if (sscanf(buf, "[%15[0-9A-Za-z+ ]]%c", section, &c) == 2) { yading@10: buf += strcspn(buf, "\n"); yading@10: buf += !!*buf; yading@10: for (i=0; icurrent_section = i; yading@10: buf = ass_split_section(ctx, buf); yading@10: } yading@10: } else { yading@10: buf += strcspn(buf, "\n"); yading@10: buf += !!*buf; yading@10: } yading@10: } yading@10: return buf ? 0 : AVERROR_INVALIDDATA; yading@10: } yading@10: yading@10: ASSSplitContext *ff_ass_split(const char *buf) yading@10: { yading@10: ASSSplitContext *ctx = av_mallocz(sizeof(*ctx)); yading@10: ctx->current_section = -1; yading@10: if (ass_split(ctx, buf) < 0) { yading@10: ff_ass_split_free(ctx); yading@10: return NULL; yading@10: } yading@10: return ctx; yading@10: } yading@10: yading@10: static void free_section(ASSSplitContext *ctx, const ASSSection *section) yading@10: { yading@10: uint8_t *ptr = (uint8_t *)&ctx->ass + section->offset; yading@10: int i, j, *count, c = 1; yading@10: yading@10: if (section->format_header) { yading@10: ptr = *(void **)ptr; yading@10: count = (int *)((uint8_t *)&ctx->ass + section->offset_count); yading@10: } else yading@10: count = &c; yading@10: yading@10: if (ptr) yading@10: for (i=0; i<*count; i++, ptr += section->size) yading@10: for (j=0; section->fields[j].name; j++) { yading@10: const ASSFields *field = §ion->fields[j]; yading@10: if (field->type == ASS_STR) yading@10: av_freep(ptr + field->offset); yading@10: } yading@10: *count = 0; yading@10: yading@10: if (section->format_header) yading@10: av_freep((uint8_t *)&ctx->ass + section->offset); yading@10: } yading@10: yading@10: ASSDialog *ff_ass_split_dialog(ASSSplitContext *ctx, const char *buf, yading@10: int cache, int *number) yading@10: { yading@10: ASSDialog *dialog = NULL; yading@10: int i, count; yading@10: if (!cache) yading@10: for (i=0; iass.dialogs_count; yading@10: if (ass_split(ctx, buf) == 0) yading@10: dialog = ctx->ass.dialogs + count; yading@10: if (number) yading@10: *number = ctx->ass.dialogs_count - count; yading@10: return dialog; yading@10: } yading@10: yading@10: void ff_ass_split_free(ASSSplitContext *ctx) yading@10: { yading@10: if (ctx) { yading@10: int i; yading@10: for (i=0; ifield_order[i])); yading@10: } yading@10: av_free(ctx); yading@10: } yading@10: } yading@10: yading@10: yading@10: int ff_ass_split_override_codes(const ASSCodesCallbacks *callbacks, void *priv, yading@10: const char *buf) yading@10: { yading@10: const char *text = NULL; yading@10: char new_line[2]; yading@10: int text_len = 0; yading@10: yading@10: while (buf && *buf) { yading@10: if (text && callbacks->text && yading@10: (sscanf(buf, "\\%1[nN]", new_line) == 1 || yading@10: !strncmp(buf, "{\\", 2))) { yading@10: callbacks->text(priv, text, text_len); yading@10: text = NULL; yading@10: } yading@10: if (sscanf(buf, "\\%1[nN]", new_line) == 1) { yading@10: if (callbacks->new_line) yading@10: callbacks->new_line(priv, new_line[0] == 'N'); yading@10: buf += 2; yading@10: } else if (!strncmp(buf, "{\\", 2)) { yading@10: buf++; yading@10: while (*buf == '\\') { yading@10: char style[2], c[2], sep[2], c_num[2] = "0", tmp[128] = {0}; yading@10: unsigned int color = 0xFFFFFFFF; yading@10: int len, size = -1, an = -1, alpha = -1; yading@10: int x1, y1, x2, y2, t1 = -1, t2 = -1; yading@10: if (sscanf(buf, "\\%1[bisu]%1[01\\}]%n", style, c, &len) > 1) { yading@10: int close = c[0] == '0' ? 1 : c[0] == '1' ? 0 : -1; yading@10: len += close != -1; yading@10: if (callbacks->style) yading@10: callbacks->style(priv, style[0], close); yading@10: } else if (sscanf(buf, "\\c%1[\\}]%n", sep, &len) > 0 || yading@10: sscanf(buf, "\\c&H%X&%1[\\}]%n", &color, sep, &len) > 1 || yading@10: sscanf(buf, "\\%1[1234]c%1[\\}]%n", c_num, sep, &len) > 1 || yading@10: sscanf(buf, "\\%1[1234]c&H%X&%1[\\}]%n", c_num, &color, sep, &len) > 2) { yading@10: if (callbacks->color) yading@10: callbacks->color(priv, color, c_num[0] - '0'); yading@10: } else if (sscanf(buf, "\\alpha%1[\\}]%n", sep, &len) > 0 || yading@10: sscanf(buf, "\\alpha&H%2X&%1[\\}]%n", &alpha, sep, &len) > 1 || yading@10: sscanf(buf, "\\%1[1234]a%1[\\}]%n", c_num, sep, &len) > 1 || yading@10: sscanf(buf, "\\%1[1234]a&H%2X&%1[\\}]%n", c_num, &alpha, sep, &len) > 2) { yading@10: if (callbacks->alpha) yading@10: callbacks->alpha(priv, alpha, c_num[0] - '0'); yading@10: } else if (sscanf(buf, "\\fn%1[\\}]%n", sep, &len) > 0 || yading@10: sscanf(buf, "\\fn%127[^\\}]%1[\\}]%n", tmp, sep, &len) > 1) { yading@10: if (callbacks->font_name) yading@10: callbacks->font_name(priv, tmp[0] ? tmp : NULL); yading@10: } else if (sscanf(buf, "\\fs%1[\\}]%n", sep, &len) > 0 || yading@10: sscanf(buf, "\\fs%u%1[\\}]%n", &size, sep, &len) > 1) { yading@10: if (callbacks->font_size) yading@10: callbacks->font_size(priv, size); yading@10: } else if (sscanf(buf, "\\a%1[\\}]%n", sep, &len) > 0 || yading@10: sscanf(buf, "\\a%2u%1[\\}]%n", &an, sep, &len) > 1 || yading@10: sscanf(buf, "\\an%1[\\}]%n", sep, &len) > 0 || yading@10: sscanf(buf, "\\an%1u%1[\\}]%n", &an, sep, &len) > 1) { yading@10: if (an != -1 && buf[2] != 'n') yading@10: an = (an&3) + (an&4 ? 6 : an&8 ? 3 : 0); yading@10: if (callbacks->alignment) yading@10: callbacks->alignment(priv, an); yading@10: } else if (sscanf(buf, "\\r%1[\\}]%n", sep, &len) > 0 || yading@10: sscanf(buf, "\\r%127[^\\}]%1[\\}]%n", tmp, sep, &len) > 1) { yading@10: if (callbacks->cancel_overrides) yading@10: callbacks->cancel_overrides(priv, tmp); yading@10: } else if (sscanf(buf, "\\move(%d,%d,%d,%d)%1[\\}]%n", &x1, &y1, &x2, &y2, sep, &len) > 4 || yading@10: sscanf(buf, "\\move(%d,%d,%d,%d,%d,%d)%1[\\}]%n", &x1, &y1, &x2, &y2, &t1, &t2, sep, &len) > 6) { yading@10: if (callbacks->move) yading@10: callbacks->move(priv, x1, y1, x2, y2, t1, t2); yading@10: } else if (sscanf(buf, "\\pos(%d,%d)%1[\\}]%n", &x1, &y1, sep, &len) > 2) { yading@10: if (callbacks->move) yading@10: callbacks->move(priv, x1, y1, x1, y1, -1, -1); yading@10: } else if (sscanf(buf, "\\org(%d,%d)%1[\\}]%n", &x1, &y1, sep, &len) > 2) { yading@10: if (callbacks->origin) yading@10: callbacks->origin(priv, x1, y1); yading@10: } else { yading@10: len = strcspn(buf+1, "\\}") + 2; /* skip unknown code */ yading@10: } yading@10: buf += len - 1; yading@10: } yading@10: if (*buf++ != '}') yading@10: return AVERROR_INVALIDDATA; yading@10: } else { yading@10: if (!text) { yading@10: text = buf; yading@10: text_len = 1; yading@10: } else yading@10: text_len++; yading@10: buf++; yading@10: } yading@10: } yading@10: if (text && callbacks->text) yading@10: callbacks->text(priv, text, text_len); yading@10: if (callbacks->end) yading@10: callbacks->end(priv); yading@10: return 0; yading@10: } yading@10: yading@10: ASSStyle *ff_ass_style_get(ASSSplitContext *ctx, const char *style) yading@10: { yading@10: ASS *ass = &ctx->ass; yading@10: int i; yading@10: yading@10: if (!style || !*style) yading@10: style = "Default"; yading@10: for (i=0; istyles_count; i++) yading@10: if (!strcmp(ass->styles[i].name, style)) yading@10: return ass->styles + i; yading@10: return NULL; yading@10: }