| @@ -15,6 +15,23 @@ libavutil: 2015-08-28 | |||
| API changes, most recent first: | |||
| 2016-xx-xx - xxxxxxx - lavc 57.26.100 - avcodec.h | |||
| Add a "sub_text_format" subtitles decoding option allowing the values "ass" | |||
| (recommended) and "ass_with_timings" (not recommended, deprecated, default). | |||
| The default value for this option will change to "ass" at the next major | |||
| libavcodec version bump. | |||
| The current default is "ass_with_timings" for compatibility. This means that | |||
| all subtitles text decoders currently still output ASS with timings printed | |||
| as strings in the AVSubtitles.rects[N]->ass fields. | |||
| Setting "sub_text_format" to "ass" allows a better timing accuracy (ASS | |||
| timing is limited to a 1/100 time base, so this is relevant for any subtitles | |||
| format needing a bigger one), ease timing adjustments, and prevents the need | |||
| of removing the timing from the decoded string yourself. This form is also | |||
| known as "the Matroska form". The timing information (start time, duration) | |||
| can be found in the AVSubtitles fields. | |||
| 2016-xx-xx - lavc 57.25.0 - avcodec.h | |||
| Add AVCodecContext.hw_frames_ctx. | |||
| @@ -90,101 +90,41 @@ int ff_ass_subtitle_header_default(AVCodecContext *avctx) | |||
| ASS_DEFAULT_ALIGNMENT); | |||
| } | |||
| static void insert_ts(AVBPrint *buf, int ts) | |||
| char *ff_ass_get_dialog(int readorder, int layer, const char *style, | |||
| const char *speaker, const char *text) | |||
| { | |||
| if (ts == -1) { | |||
| av_bprintf(buf, "9:59:59.99,"); | |||
| } else { | |||
| int h, m, s; | |||
| h = ts/360000; ts -= 360000*h; | |||
| m = ts/ 6000; ts -= 6000*m; | |||
| s = ts/ 100; ts -= 100*s; | |||
| av_bprintf(buf, "%d:%02d:%02d.%02d,", h, m, s, ts); | |||
| } | |||
| } | |||
| int ff_ass_bprint_dialog(AVBPrint *buf, const char *dialog, | |||
| int ts_start, int duration, int raw) | |||
| { | |||
| int dlen; | |||
| if (!raw || raw == 2) { | |||
| long int layer = 0; | |||
| if (raw == 2) { | |||
| /* skip ReadOrder */ | |||
| dialog = strchr(dialog, ','); | |||
| if (!dialog) | |||
| return AVERROR_INVALIDDATA; | |||
| dialog++; | |||
| /* extract Layer or Marked */ | |||
| layer = strtol(dialog, (char**)&dialog, 10); | |||
| if (*dialog != ',') | |||
| return AVERROR_INVALIDDATA; | |||
| dialog++; | |||
| } | |||
| av_bprintf(buf, "Dialogue: %ld,", layer); | |||
| insert_ts(buf, ts_start); | |||
| insert_ts(buf, duration == -1 ? -1 : ts_start + duration); | |||
| if (raw != 2) | |||
| av_bprintf(buf, "Default,,0,0,0,,"); | |||
| } | |||
| dlen = strcspn(dialog, "\n"); | |||
| dlen += dialog[dlen] == '\n'; | |||
| av_bprintf(buf, "%.*s", dlen, dialog); | |||
| if (raw == 2) | |||
| av_bprintf(buf, "\r\n"); | |||
| return dlen; | |||
| return av_asprintf("%d,%d,%s,%s,0,0,0,,%s", | |||
| readorder, layer, style ? style : "Default", | |||
| speaker ? speaker : "", text); | |||
| } | |||
| int ff_ass_add_rect(AVSubtitle *sub, const char *dialog, | |||
| int ts_start, int duration, int raw) | |||
| int readorder, int layer, const char *style, | |||
| const char *speaker) | |||
| { | |||
| AVBPrint buf; | |||
| int ret, dlen; | |||
| char *ass_str; | |||
| AVSubtitleRect **rects; | |||
| av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| if ((ret = ff_ass_bprint_dialog(&buf, dialog, ts_start, duration, raw)) < 0) | |||
| goto err; | |||
| dlen = ret; | |||
| if (!av_bprint_is_complete(&buf)) | |||
| goto errnomem; | |||
| rects = av_realloc_array(sub->rects, (sub->num_rects+1), sizeof(*sub->rects)); | |||
| if (!rects) | |||
| goto errnomem; | |||
| return AVERROR(ENOMEM); | |||
| sub->rects = rects; | |||
| sub->end_display_time = FFMAX(sub->end_display_time, 10 * duration); | |||
| rects[sub->num_rects] = av_mallocz(sizeof(*rects[0])); | |||
| if (!rects[sub->num_rects]) | |||
| goto errnomem; | |||
| return AVERROR(ENOMEM); | |||
| rects[sub->num_rects]->type = SUBTITLE_ASS; | |||
| ret = av_bprint_finalize(&buf, &rects[sub->num_rects]->ass); | |||
| if (ret < 0) | |||
| goto err; | |||
| ass_str = ff_ass_get_dialog(readorder, layer, style, speaker, dialog); | |||
| if (!ass_str) | |||
| return AVERROR(ENOMEM); | |||
| rects[sub->num_rects]->ass = ass_str; | |||
| sub->num_rects++; | |||
| return dlen; | |||
| errnomem: | |||
| ret = AVERROR(ENOMEM); | |||
| err: | |||
| av_bprint_finalize(&buf, NULL); | |||
| return ret; | |||
| return 0; | |||
| } | |||
| int ff_ass_add_rect_bprint(AVSubtitle *sub, AVBPrint *buf, | |||
| int ts_start, int duration) | |||
| void ff_ass_decoder_flush(AVCodecContext *avctx) | |||
| { | |||
| av_bprintf(buf, "\r\n"); | |||
| if (!av_bprint_is_complete(buf)) | |||
| return AVERROR(ENOMEM); | |||
| return ff_ass_add_rect(sub, buf->str, ts_start, duration, 0); | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| s->readorder = 0; | |||
| } | |||
| void ff_ass_bprint_text_event(AVBPrint *buf, const char *p, int size, | |||
| @@ -43,6 +43,10 @@ | |||
| #define ASS_DEFAULT_BORDERSTYLE 1 | |||
| /** @} */ | |||
| typedef struct FFASSDecoderContext { | |||
| int readorder; | |||
| } FFASSDecoderContext; | |||
| /** | |||
| * Generate a suitable AVCodecContext.subtitle_header for SUBTITLE_ASS. | |||
| * | |||
| @@ -74,55 +78,23 @@ int ff_ass_subtitle_header(AVCodecContext *avctx, | |||
| int ff_ass_subtitle_header_default(AVCodecContext *avctx); | |||
| /** | |||
| * Add an ASS dialog line to an AVSubtitle as a new AVSubtitleRect. | |||
| * | |||
| * @param sub pointer to the AVSubtitle | |||
| * @param dialog ASS dialog to add to sub | |||
| * @param ts_start start timestamp for this dialog (in 1/100 second unit) | |||
| * @param duration duration for this dialog (in 1/100 second unit), can be -1 | |||
| * to last until the end of the presentation | |||
| * @param raw when set to 2, it indicates that dialog contains an ASS | |||
| * dialog line as muxed in Matroska | |||
| * when set to 1, it indicates that dialog contains a whole SSA | |||
| * dialog line which should be copied as is. | |||
| * when set to 0, it indicates that dialog contains only the Text | |||
| * part of the ASS dialog line, the rest of the line | |||
| * will be generated. | |||
| * @return number of characters read from dialog. It can be less than the whole | |||
| * length of dialog, if dialog contains several lines of text. | |||
| * A negative value indicates an error. | |||
| * Craft an ASS dialog string. | |||
| */ | |||
| int ff_ass_add_rect(AVSubtitle *sub, const char *dialog, | |||
| int ts_start, int duration, int raw); | |||
| char *ff_ass_get_dialog(int readorder, int layer, const char *style, | |||
| const char *speaker, const char *text); | |||
| /** | |||
| * Same as ff_ass_add_rect, but taking an AVBPrint buffer instead of a | |||
| * string, and assuming raw=0. | |||
| * Add an ASS dialog to a subtitle. | |||
| */ | |||
| int ff_ass_add_rect_bprint(AVSubtitle *sub, AVBPrint *buf, | |||
| int ts_start, int duration); | |||
| int ff_ass_add_rect(AVSubtitle *sub, const char *dialog, | |||
| int readorder, int layer, const char *style, | |||
| const char *speaker); | |||
| /** | |||
| * Add an ASS dialog line to an AVBPrint buffer. | |||
| * | |||
| * @param buf pointer to an initialized AVBPrint buffer | |||
| * @param dialog ASS dialog to add to sub | |||
| * @param ts_start start timestamp for this dialog (in 1/100 second unit) | |||
| * @param duration duration for this dialog (in 1/100 second unit), can be -1 | |||
| * to last until the end of the presentation | |||
| * @param raw when set to 2, it indicates that dialog contains an ASS | |||
| * dialog line as muxed in Matroska | |||
| * when set to 1, it indicates that dialog contains a whole SSA | |||
| * dialog line which should be copied as is. | |||
| * when set to 0, it indicates that dialog contains only the Text | |||
| * part of the ASS dialog line, the rest of the line | |||
| * will be generated. | |||
| * @return number of characters read from dialog. It can be less than the whole | |||
| * length of dialog, if dialog contains several lines of text. | |||
| * A negative value indicates an error. | |||
| * Helper to flush a text subtitles decoder making use of the | |||
| * FFASSDecoderContext. | |||
| */ | |||
| int ff_ass_bprint_dialog(AVBPrint *buf, const char *dialog, | |||
| int ts_start, int duration, int raw); | |||
| void ff_ass_decoder_flush(AVCodecContext *avctx); | |||
| /** | |||
| * Escape a text subtitle using ASS syntax into an AVBPrint buffer. | |||
| @@ -409,6 +409,55 @@ ASSDialog *ff_ass_split_dialog(ASSSplitContext *ctx, const char *buf, | |||
| return dialog; | |||
| } | |||
| void ff_ass_free_dialog(ASSDialog **dialogp) | |||
| { | |||
| ASSDialog *dialog = *dialogp; | |||
| if (!dialog) | |||
| return; | |||
| av_freep(&dialog->style); | |||
| av_freep(&dialog->name); | |||
| av_freep(&dialog->effect); | |||
| av_freep(&dialog->text); | |||
| av_freep(dialogp); | |||
| } | |||
| ASSDialog *ff_ass_split_dialog2(ASSSplitContext *ctx, const char *buf) | |||
| { | |||
| int i; | |||
| static const ASSFields fields[] = { | |||
| {"ReadOrder", ASS_INT, offsetof(ASSDialog, readorder)}, | |||
| {"Layer", ASS_INT, offsetof(ASSDialog, layer) }, | |||
| {"Style", ASS_STR, offsetof(ASSDialog, style) }, | |||
| {"Name", ASS_STR, offsetof(ASSDialog, name) }, | |||
| {"MarginL", ASS_INT, offsetof(ASSDialog, margin_l) }, | |||
| {"MarginR", ASS_INT, offsetof(ASSDialog, margin_r) }, | |||
| {"MarginV", ASS_INT, offsetof(ASSDialog, margin_v) }, | |||
| {"Effect", ASS_STR, offsetof(ASSDialog, effect) }, | |||
| {"Text", ASS_STR, offsetof(ASSDialog, text) }, | |||
| }; | |||
| ASSDialog *dialog = av_mallocz(sizeof(*dialog)); | |||
| if (!dialog) | |||
| return NULL; | |||
| for (i = 0; i < FF_ARRAY_ELEMS(fields); i++) { | |||
| size_t len; | |||
| const int last = i == FF_ARRAY_ELEMS(fields) - 1; | |||
| const ASSFieldType type = fields[i].type; | |||
| uint8_t *ptr = (uint8_t *)dialog + fields[i].offset; | |||
| buf = skip_space(buf); | |||
| len = last ? strlen(buf) : strcspn(buf, ","); | |||
| if (len >= INT_MAX) { | |||
| ff_ass_free_dialog(&dialog); | |||
| return NULL; | |||
| } | |||
| convert_func[type](ptr, buf, len); | |||
| buf += len; | |||
| if (*buf) buf++; | |||
| } | |||
| return dialog; | |||
| } | |||
| void ff_ass_split_free(ASSSplitContext *ctx) | |||
| { | |||
| if (ctx) { | |||
| @@ -69,6 +69,7 @@ typedef struct { | |||
| * fields extracted from the [Events] section | |||
| */ | |||
| typedef struct { | |||
| int readorder; | |||
| int layer; /**< higher numbered layers are drawn over lower numbered */ | |||
| int start; /**< start time of the dialog in centiseconds */ | |||
| int end; /**< end time of the dialog in centiseconds */ | |||
| @@ -124,6 +125,20 @@ ASSSplitContext *ff_ass_split(const char *buf); | |||
| ASSDialog *ff_ass_split_dialog(ASSSplitContext *ctx, const char *buf, | |||
| int cache, int *number); | |||
| /** | |||
| * Free a dialogue obtained from ff_ass_split_dialog2(). | |||
| */ | |||
| void ff_ass_free_dialog(ASSDialog **dialogp); | |||
| /** | |||
| * Split one ASS Dialogue line from a string buffer. | |||
| * | |||
| * @param ctx Context previously initialized by ff_ass_split(). | |||
| * @param buf String containing the ASS "Dialogue" line. | |||
| * @return Pointer to the split ASSDialog. Must be freed with ff_ass_free_dialog() | |||
| */ | |||
| ASSDialog *ff_ass_split_dialog2(ASSSplitContext *ctx, const char *buf); | |||
| /** | |||
| * Free all the memory allocated for an ASSSplitContext. | |||
| * | |||
| @@ -40,24 +40,23 @@ static av_cold int ass_decode_init(AVCodecContext *avctx) | |||
| static int ass_decode_frame(AVCodecContext *avctx, void *data, int *got_sub_ptr, | |||
| AVPacket *avpkt) | |||
| { | |||
| int ret; | |||
| AVSubtitle *sub = data; | |||
| const char *ptr = avpkt->data; | |||
| static const AVRational ass_tb = {1, 100}; | |||
| const int ts_start = av_rescale_q(avpkt->pts, avctx->time_base, ass_tb); | |||
| const int ts_duration = av_rescale_q(avpkt->duration, avctx->time_base, ass_tb); | |||
| if (avpkt->size <= 0) | |||
| return avpkt->size; | |||
| ret = ff_ass_add_rect(sub, ptr, ts_start, ts_duration, 2); | |||
| if (ret < 0) { | |||
| if (ret == AVERROR_INVALIDDATA) | |||
| av_log(avctx, AV_LOG_ERROR, "Invalid ASS packet\n"); | |||
| return ret; | |||
| } | |||
| *got_sub_ptr = avpkt->size > 0; | |||
| sub->rects = av_malloc(sizeof(*sub->rects)); | |||
| if (!sub->rects) | |||
| return AVERROR(ENOMEM); | |||
| sub->rects[0] = av_mallocz(sizeof(*sub->rects[0])); | |||
| if (!sub->rects[0]) | |||
| return AVERROR(ENOMEM); | |||
| sub->num_rects = 1; | |||
| sub->rects[0]->type = SUBTITLE_ASS; | |||
| sub->rects[0]->ass = av_strdup(avpkt->data); | |||
| if (!sub->rects[0]->ass) | |||
| return AVERROR(ENOMEM); | |||
| *got_sub_ptr = 1; | |||
| return avpkt->size; | |||
| } | |||
| @@ -60,13 +60,7 @@ static int ass_encode_frame(AVCodecContext *avctx, | |||
| return -1; | |||
| } | |||
| if (strncmp(ass, "Dialogue: ", 10)) { | |||
| av_log(avctx, AV_LOG_ERROR, "AVSubtitle rectangle ass \"%s\"" | |||
| " does not look like a SSA markup\n", ass); | |||
| return AVERROR_INVALIDDATA; | |||
| } | |||
| // TODO: reindent | |||
| if (!strncmp(ass, "Dialogue: ", 10)) { | |||
| if (i > 0) { | |||
| av_log(avctx, AV_LOG_ERROR, "ASS encoder supports only one " | |||
| "ASS rectangle field.\n"); | |||
| @@ -91,6 +85,7 @@ static int ass_encode_frame(AVCodecContext *avctx, | |||
| snprintf(ass_line, sizeof(ass_line), "%d,%ld,%s", ++s->id, layer, p); | |||
| ass_line[strcspn(ass_line, "\r\n")] = 0; | |||
| ass = ass_line; | |||
| } | |||
| len = av_strlcpy(buf+total_len, ass, bufsize-total_len); | |||
| @@ -3285,6 +3285,10 @@ typedef struct AVCodecContext { | |||
| #define FF_SUB_CHARENC_MODE_AUTOMATIC 0 ///< libavcodec will select the mode itself | |||
| #define FF_SUB_CHARENC_MODE_PRE_DECODER 1 ///< the AVPacket data needs to be recoded to UTF-8 before being fed to the decoder, requires iconv | |||
| int sub_text_format; | |||
| #define FF_SUB_TEXT_FMT_ASS 0 | |||
| #define FF_SUB_TEXT_FMT_ASS_WITH_TIMINGS 1 | |||
| /** | |||
| * Skip processing alpha if supported by codec. | |||
| * Note that if the format uses pre-multiplied alpha (common with VP6, | |||
| @@ -248,6 +248,7 @@ typedef struct CCaptionSubContext { | |||
| char prev_cmd[2]; | |||
| /* buffer to store pkt data */ | |||
| AVBufferRef *pktbuf; | |||
| int readorder; | |||
| } CCaptionSubContext; | |||
| @@ -306,6 +307,7 @@ static void flush_decoder(AVCodecContext *avctx) | |||
| ctx->last_real_time = 0; | |||
| ctx->screen_touched = 0; | |||
| ctx->buffer_changed = 0; | |||
| ctx->readorder = 0; | |||
| av_bprint_clear(&ctx->buffer); | |||
| } | |||
| @@ -752,18 +754,16 @@ static int decode(AVCodecContext *avctx, void *data, int *got_sub, AVPacket *avp | |||
| if (*ctx->buffer.str || ctx->real_time) | |||
| { | |||
| int64_t sub_pts = ctx->real_time ? avpkt->pts : ctx->start_time; | |||
| int start_time = av_rescale_q(sub_pts, avctx->time_base, ass_tb); | |||
| int duration = -1; | |||
| if (!ctx->real_time) { | |||
| int end_time = av_rescale_q(ctx->end_time, avctx->time_base, ass_tb); | |||
| duration = end_time - start_time; | |||
| } | |||
| ff_dlog(ctx, "cdp writing data (%s)\n",ctx->buffer.str); | |||
| ret = ff_ass_add_rect_bprint(sub, &ctx->buffer, start_time, duration); | |||
| ret = ff_ass_add_rect(sub, ctx->buffer.str, ctx->readorder++, 0, NULL, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| sub->pts = av_rescale_q(sub_pts, avctx->time_base, AV_TIME_BASE_Q); | |||
| sub->pts = av_rescale_q(ctx->start_time, avctx->time_base, AV_TIME_BASE_Q); | |||
| if (!ctx->real_time) | |||
| sub->end_display_time = av_rescale_q(ctx->end_time - ctx->start_time, | |||
| avctx->time_base, av_make_q(1, 1000)); | |||
| else | |||
| sub->end_display_time = -1; | |||
| ctx->buffer_changed = 0; | |||
| ctx->last_real_time = avpkt->pts; | |||
| ctx->screen_touched = 0; | |||
| @@ -772,18 +772,16 @@ static int decode(AVCodecContext *avctx, void *data, int *got_sub, AVPacket *avp | |||
| if (ctx->real_time && ctx->screen_touched && | |||
| avpkt->pts > ctx->last_real_time + av_rescale_q(20, ass_tb, avctx->time_base)) { | |||
| int start_time; | |||
| ctx->last_real_time = avpkt->pts; | |||
| ctx->screen_touched = 0; | |||
| capture_screen(ctx); | |||
| ctx->buffer_changed = 0; | |||
| start_time = av_rescale_q(avpkt->pts, avctx->time_base, ass_tb); | |||
| ret = ff_ass_add_rect_bprint(sub, &ctx->buffer, start_time, -1); | |||
| ret = ff_ass_add_rect(sub, ctx->buffer.str, ctx->readorder++, 0, NULL, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| sub->pts = av_rescale_q(avpkt->pts, avctx->time_base, AV_TIME_BASE_Q); | |||
| sub->end_display_time = -1; | |||
| } | |||
| *got_sub = sub->num_rects > 0; | |||
| @@ -167,6 +167,7 @@ static int jacosub_decode_frame(AVCodecContext *avctx, | |||
| int ret; | |||
| AVSubtitle *sub = data; | |||
| const char *ptr = avpkt->data; | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| if (avpkt->size <= 0) | |||
| goto end; | |||
| @@ -181,7 +182,7 @@ static int jacosub_decode_frame(AVCodecContext *avctx, | |||
| av_bprint_init(&buffer, JSS_MAX_LINESIZE, JSS_MAX_LINESIZE); | |||
| jacosub_to_ass(avctx, &buffer, ptr); | |||
| ret = ff_ass_add_rect_bprint(sub, &buffer, avpkt->pts, avpkt->duration); | |||
| ret = ff_ass_add_rect(sub, buffer.str, s->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&buffer, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -199,4 +200,6 @@ AVCodec ff_jacosub_decoder = { | |||
| .id = AV_CODEC_ID_JACOSUB, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .decode = jacosub_decode_frame, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| @@ -74,6 +74,8 @@ typedef struct TeletextContext | |||
| vbi_export * ex; | |||
| #endif | |||
| vbi_sliced sliced[MAX_SLICES]; | |||
| int readorder; | |||
| } TeletextContext; | |||
| static int chop_spaces_utf8(const unsigned char* t, int len) | |||
| @@ -95,37 +97,21 @@ static void subtitle_rect_free(AVSubtitleRect **sub_rect) | |||
| av_freep(sub_rect); | |||
| } | |||
| static int create_ass_text(TeletextContext *ctx, const char *text, char **ass) | |||
| static char *create_ass_text(TeletextContext *ctx, const char *text) | |||
| { | |||
| int ret; | |||
| AVBPrint buf, buf2; | |||
| const int ts_start = av_rescale_q(ctx->pts, AV_TIME_BASE_Q, (AVRational){1, 100}); | |||
| const int ts_duration = av_rescale_q(ctx->sub_duration, (AVRational){1, 1000}, (AVRational){1, 100}); | |||
| char *dialog; | |||
| AVBPrint buf; | |||
| /* First we escape the plain text into buf. */ | |||
| av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| ff_ass_bprint_text_event(&buf, text, strlen(text), "", 0); | |||
| av_bprintf(&buf, "\r\n"); | |||
| if (!av_bprint_is_complete(&buf)) { | |||
| av_bprint_finalize(&buf, NULL); | |||
| return AVERROR(ENOMEM); | |||
| return NULL; | |||
| } | |||
| /* Then we create the ass dialog line in buf2 from the escaped text in buf. */ | |||
| av_bprint_init(&buf2, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| ff_ass_bprint_dialog(&buf2, buf.str, ts_start, ts_duration, 0); | |||
| dialog = ff_ass_get_dialog(ctx->readorder++, 0, NULL, NULL, buf.str); | |||
| av_bprint_finalize(&buf, NULL); | |||
| if (!av_bprint_is_complete(&buf2)) { | |||
| av_bprint_finalize(&buf2, NULL); | |||
| return AVERROR(ENOMEM); | |||
| } | |||
| if ((ret = av_bprint_finalize(&buf2, ass)) < 0) | |||
| return ret; | |||
| return 0; | |||
| return dialog; | |||
| } | |||
| /* Draw a page as text */ | |||
| @@ -181,11 +167,12 @@ static int gen_sub_text(TeletextContext *ctx, AVSubtitleRect *sub_rect, vbi_page | |||
| } | |||
| if (buf.len) { | |||
| int ret; | |||
| sub_rect->type = SUBTITLE_ASS; | |||
| if ((ret = create_ass_text(ctx, buf.str, &sub_rect->ass)) < 0) { | |||
| sub_rect->ass = create_ass_text(ctx, buf.str); | |||
| if (!sub_rect->ass) { | |||
| av_bprint_finalize(&buf, NULL); | |||
| return ret; | |||
| return AVERROR(ENOMEM); | |||
| } | |||
| av_log(ctx, AV_LOG_DEBUG, "subtext:%s:txetbus\n", sub_rect->ass); | |||
| } else { | |||
| @@ -541,6 +528,7 @@ static int teletext_close_decoder(AVCodecContext *avctx) | |||
| vbi_decoder_delete(ctx->vbi); | |||
| ctx->vbi = NULL; | |||
| ctx->pts = AV_NOPTS_VALUE; | |||
| ctx->readorder = 0; | |||
| return 0; | |||
| } | |||
| @@ -280,6 +280,7 @@ static int microdvd_decode_frame(AVCodecContext *avctx, | |||
| AVBPrint new_line; | |||
| char *line = avpkt->data; | |||
| char *end = avpkt->data + avpkt->size; | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| struct microdvd_tag tags[sizeof(MICRODVD_TAGS) - 1] = {{0}}; | |||
| if (avpkt->size <= 0) | |||
| @@ -308,14 +309,7 @@ static int microdvd_decode_frame(AVCodecContext *avctx, | |||
| } | |||
| } | |||
| if (new_line.len) { | |||
| int ret; | |||
| int64_t start = avpkt->pts; | |||
| int64_t duration = avpkt->duration; | |||
| int ts_start = av_rescale_q(start, avctx->time_base, (AVRational){1,100}); | |||
| int ts_duration = duration != -1 ? | |||
| av_rescale_q(duration, avctx->time_base, (AVRational){1,100}) : -1; | |||
| ret = ff_ass_add_rect_bprint(sub, &new_line, ts_start, ts_duration); | |||
| int ret = ff_ass_add_rect(sub, new_line.str, s->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&new_line, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -381,4 +375,6 @@ AVCodec ff_microdvd_decoder = { | |||
| .id = AV_CODEC_ID_MICRODVD, | |||
| .init = microdvd_init, | |||
| .decode = microdvd_decode_frame, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| @@ -99,6 +99,7 @@ typedef struct { | |||
| uint64_t tracksize; | |||
| int size_var; | |||
| int count_s, count_f; | |||
| int readorder; | |||
| } MovTextContext; | |||
| typedef struct { | |||
| @@ -424,7 +425,7 @@ static int mov_text_decode_frame(AVCodecContext *avctx, | |||
| { | |||
| AVSubtitle *sub = data; | |||
| MovTextContext *m = avctx->priv_data; | |||
| int ret, ts_start, ts_end; | |||
| int ret; | |||
| AVBPrint buf; | |||
| char *ptr = avpkt->data; | |||
| char *end; | |||
| @@ -454,13 +455,6 @@ static int mov_text_decode_frame(AVCodecContext *avctx, | |||
| end = ptr + FFMIN(2 + text_length, avpkt->size); | |||
| ptr += 2; | |||
| ts_start = av_rescale_q(avpkt->pts, | |||
| avctx->time_base, | |||
| (AVRational){1,100}); | |||
| ts_end = av_rescale_q(avpkt->pts + avpkt->duration, | |||
| avctx->time_base, | |||
| (AVRational){1,100}); | |||
| tsmb_size = 0; | |||
| m->tracksize = 2 + text_length; | |||
| m->style_entries = 0; | |||
| @@ -506,7 +500,7 @@ static int mov_text_decode_frame(AVCodecContext *avctx, | |||
| } else | |||
| text_to_ass(&buf, ptr, end, m); | |||
| ret = ff_ass_add_rect_bprint(sub, &buf, ts_start, ts_end - ts_start); | |||
| ret = ff_ass_add_rect(sub, buf.str, m->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&buf, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -521,6 +515,12 @@ static int mov_text_decode_close(AVCodecContext *avctx) | |||
| return 0; | |||
| } | |||
| static void mov_text_flush(AVCodecContext *avctx) | |||
| { | |||
| MovTextContext *m = avctx->priv_data; | |||
| m->readorder = 0; | |||
| } | |||
| AVCodec ff_movtext_decoder = { | |||
| .name = "mov_text", | |||
| .long_name = NULL_IF_CONFIG_SMALL("3GPP Timed Text subtitle"), | |||
| @@ -530,4 +530,5 @@ AVCodec ff_movtext_decoder = { | |||
| .init = mov_text_init, | |||
| .decode = mov_text_decode_frame, | |||
| .close = mov_text_decode_close, | |||
| .flush = mov_text_flush, | |||
| }; | |||
| @@ -332,16 +332,26 @@ static int mov_text_encode_frame(AVCodecContext *avctx, unsigned char *buf, | |||
| s->box_flags = 0; | |||
| s->style_entries = 0; | |||
| for (i = 0; i < sub->num_rects; i++) { | |||
| const char *ass = sub->rects[i]->ass; | |||
| if (sub->rects[i]->type != SUBTITLE_ASS) { | |||
| av_log(avctx, AV_LOG_ERROR, "Only SUBTITLE_ASS type supported.\n"); | |||
| return AVERROR(ENOSYS); | |||
| } | |||
| dialog = ff_ass_split_dialog(s->ass_ctx, sub->rects[i]->ass, 0, &num); | |||
| if (!strncmp(ass, "Dialogue: ", 10)) { | |||
| dialog = ff_ass_split_dialog(s->ass_ctx, ass, 0, &num); | |||
| // TODO reindent | |||
| for (; dialog && num--; dialog++) { | |||
| ff_ass_split_override_codes(&mov_text_callbacks, s, dialog->text); | |||
| } | |||
| } else { | |||
| dialog = ff_ass_split_dialog2(s->ass_ctx, ass); | |||
| if (!dialog) | |||
| return AVERROR(ENOMEM); | |||
| ff_ass_split_override_codes(&mov_text_callbacks, s, dialog->text); | |||
| ff_ass_free_dialog(&dialog); | |||
| } | |||
| for (j = 0; j < box_count; j++) { | |||
| box_types[j].encode(s, box_types[j].type); | |||
| @@ -69,13 +69,11 @@ static int mpl2_decode_frame(AVCodecContext *avctx, void *data, | |||
| AVBPrint buf; | |||
| AVSubtitle *sub = data; | |||
| const char *ptr = avpkt->data; | |||
| const int ts_start = av_rescale_q(avpkt->pts, avctx->time_base, (AVRational){1,100}); | |||
| const int ts_duration = avpkt->duration != -1 ? | |||
| av_rescale_q(avpkt->duration, avctx->time_base, (AVRational){1,100}) : -1; | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| if (ptr && avpkt->size > 0 && *ptr && !mpl2_event_to_ass(&buf, ptr)) | |||
| ret = ff_ass_add_rect_bprint(sub, &buf, ts_start, ts_duration); | |||
| ret = ff_ass_add_rect(sub, buf.str, s->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&buf, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -90,4 +88,6 @@ AVCodec ff_mpl2_decoder = { | |||
| .id = AV_CODEC_ID_MPL2, | |||
| .decode = mpl2_decode_frame, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| @@ -519,6 +519,9 @@ static const AVOption avcodec_options[] = { | |||
| {"do_nothing", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = FF_SUB_CHARENC_MODE_DO_NOTHING}, INT_MIN, INT_MAX, S|D, "sub_charenc_mode"}, | |||
| {"auto", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = FF_SUB_CHARENC_MODE_AUTOMATIC}, INT_MIN, INT_MAX, S|D, "sub_charenc_mode"}, | |||
| {"pre_decoder", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = FF_SUB_CHARENC_MODE_PRE_DECODER}, INT_MIN, INT_MAX, S|D, "sub_charenc_mode"}, | |||
| {"sub_text_format", "set decoded text subtitle format", OFFSET(sub_text_format), AV_OPT_TYPE_INT, {.i64 = FF_SUB_TEXT_FMT_ASS_WITH_TIMINGS}, 0, 1, S|D, "sub_text_format"}, | |||
| {"ass", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = FF_SUB_TEXT_FMT_ASS}, INT_MIN, INT_MAX, S|D, "sub_text_format"}, | |||
| {"ass_with_timings", NULL, 0, AV_OPT_TYPE_CONST, {.i64 = FF_SUB_TEXT_FMT_ASS_WITH_TIMINGS}, INT_MIN, INT_MAX, S|D, "sub_text_format"}, | |||
| {"refcounted_frames", NULL, OFFSET(refcounted_frames), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, A|V|D }, | |||
| #if FF_API_SIDEDATA_ONLY_PKT | |||
| {"side_data_only_packets", NULL, OFFSET(side_data_only_packets), AV_OPT_TYPE_BOOL, { .i64 = 1 }, 0, 1, A|V|E }, | |||
| @@ -61,13 +61,12 @@ static int realtext_decode_frame(AVCodecContext *avctx, | |||
| int ret = 0; | |||
| AVSubtitle *sub = data; | |||
| const char *ptr = avpkt->data; | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| AVBPrint buf; | |||
| av_bprint_init(&buf, 0, 4096); | |||
| // note: no need to rescale pts & duration since they are in the same | |||
| // timebase as ASS (1/100) | |||
| if (ptr && avpkt->size > 0 && !rt_event_to_ass(&buf, ptr)) | |||
| ret = ff_ass_add_rect_bprint(sub, &buf, avpkt->pts, avpkt->duration); | |||
| ret = ff_ass_add_rect(sub, buf.str, s->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&buf, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -82,4 +81,6 @@ AVCodec ff_realtext_decoder = { | |||
| .id = AV_CODEC_ID_REALTEXT, | |||
| .decode = realtext_decode_frame, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| @@ -35,6 +35,7 @@ typedef struct { | |||
| AVBPrint encoded_source; | |||
| AVBPrint encoded_content; | |||
| AVBPrint full; | |||
| int readorder; | |||
| } SAMIContext; | |||
| static int sami_paragraph_to_ass(AVCodecContext *avctx, const char *src) | |||
| @@ -131,10 +132,8 @@ static int sami_decode_frame(AVCodecContext *avctx, | |||
| SAMIContext *sami = avctx->priv_data; | |||
| if (ptr && avpkt->size > 0 && !sami_paragraph_to_ass(avctx, ptr)) { | |||
| int ts_start = av_rescale_q(avpkt->pts, avctx->time_base, (AVRational){1,100}); | |||
| int ts_duration = avpkt->duration != -1 ? | |||
| av_rescale_q(avpkt->duration, avctx->time_base, (AVRational){1,100}) : -1; | |||
| int ret = ff_ass_add_rect_bprint(sub, &sami->full, ts_start, ts_duration); | |||
| // TODO: pass escaped sami->encoded_source.str as source | |||
| int ret = ff_ass_add_rect(sub, sami->full.str, sami->readorder++, 0, NULL, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| } | |||
| @@ -164,6 +163,12 @@ static av_cold int sami_close(AVCodecContext *avctx) | |||
| return 0; | |||
| } | |||
| static void sami_flush(AVCodecContext *avctx) | |||
| { | |||
| SAMIContext *sami = avctx->priv_data; | |||
| sami->readorder = 0; | |||
| } | |||
| AVCodec ff_sami_decoder = { | |||
| .name = "sami", | |||
| .long_name = NULL_IF_CONFIG_SMALL("SAMI subtitle"), | |||
| @@ -173,4 +178,5 @@ AVCodec ff_sami_decoder = { | |||
| .init = sami_init, | |||
| .close = sami_close, | |||
| .decode = sami_decode_frame, | |||
| .flush = sami_flush, | |||
| }; | |||
| @@ -57,9 +57,10 @@ static int srt_decode_frame(AVCodecContext *avctx, | |||
| { | |||
| AVSubtitle *sub = data; | |||
| AVBPrint buffer; | |||
| int ts_start, ts_end, x1 = -1, y1 = -1, x2 = -1, y2 = -1; | |||
| int x1 = -1, y1 = -1, x2 = -1, y2 = -1; | |||
| int size, ret; | |||
| const uint8_t *p = av_packet_get_side_data(avpkt, AV_PKT_DATA_SUBTITLE_POSITION, &size); | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| if (p && size == 16) { | |||
| x1 = AV_RL32(p ); | |||
| @@ -73,16 +74,8 @@ static int srt_decode_frame(AVCodecContext *avctx, | |||
| av_bprint_init(&buffer, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| // Do final divide-by-10 outside rescale to force rounding down. | |||
| ts_start = av_rescale_q(avpkt->pts, | |||
| avctx->time_base, | |||
| (AVRational){1,100}); | |||
| ts_end = av_rescale_q(avpkt->pts + avpkt->duration, | |||
| avctx->time_base, | |||
| (AVRational){1,100}); | |||
| srt_to_ass(avctx, &buffer, avpkt->data, x1, y1, x2, y2); | |||
| ret = ff_ass_add_rect_bprint(sub, &buffer, ts_start, ts_end-ts_start); | |||
| ret = ff_ass_add_rect(sub, buffer.str, s->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&buffer, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -100,6 +93,8 @@ AVCodec ff_srt_decoder = { | |||
| .id = AV_CODEC_ID_SUBRIP, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .decode = srt_decode_frame, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| #endif | |||
| @@ -111,5 +106,7 @@ AVCodec ff_subrip_decoder = { | |||
| .id = AV_CODEC_ID_SUBRIP, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .decode = srt_decode_frame, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| #endif | |||
| @@ -237,18 +237,30 @@ static int encode_frame(AVCodecContext *avctx, | |||
| av_bprint_clear(&s->buffer); | |||
| for (i=0; i<sub->num_rects; i++) { | |||
| const char *ass = sub->rects[i]->ass; | |||
| if (sub->rects[i]->type != SUBTITLE_ASS) { | |||
| av_log(avctx, AV_LOG_ERROR, "Only SUBTITLE_ASS type supported.\n"); | |||
| return AVERROR(ENOSYS); | |||
| } | |||
| dialog = ff_ass_split_dialog(s->ass_ctx, sub->rects[i]->ass, 0, &num); | |||
| if (!strncmp(ass, "Dialogue: ", 10)) { | |||
| dialog = ff_ass_split_dialog(s->ass_ctx, ass, 0, &num); | |||
| // TODO reindent | |||
| for (; dialog && num--; dialog++) { | |||
| s->alignment_applied = 0; | |||
| srt_style_apply(s, dialog->style); | |||
| ff_ass_split_override_codes(cb, s, dialog->text); | |||
| } | |||
| } else { | |||
| dialog = ff_ass_split_dialog2(s->ass_ctx, ass); | |||
| if (!dialog) | |||
| return AVERROR(ENOMEM); | |||
| s->alignment_applied = 0; | |||
| srt_style_apply(s, dialog->style); | |||
| ff_ass_split_override_codes(cb, s, dialog->text); | |||
| ff_ass_free_dialog(&dialog); | |||
| } | |||
| } | |||
| if (!av_bprint_is_complete(&s->buffer)) | |||
| @@ -52,13 +52,12 @@ static int subviewer_decode_frame(AVCodecContext *avctx, | |||
| int ret = 0; | |||
| AVSubtitle *sub = data; | |||
| const char *ptr = avpkt->data; | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| AVBPrint buf; | |||
| av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| // note: no need to rescale pts & duration since they are in the same | |||
| // timebase as ASS (1/100) | |||
| if (ptr && avpkt->size > 0 && !subviewer_event_to_ass(&buf, ptr)) | |||
| ret = ff_ass_add_rect_bprint(sub, &buf, avpkt->pts, avpkt->duration); | |||
| ret = ff_ass_add_rect(sub, buf.str, s->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&buf, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -73,4 +72,6 @@ AVCodec ff_subviewer_decoder = { | |||
| .id = AV_CODEC_ID_SUBVIEWER, | |||
| .decode = subviewer_decode_frame, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| @@ -32,6 +32,7 @@ typedef struct { | |||
| AVClass *class; | |||
| const char *linebreaks; | |||
| int keep_ass_markup; | |||
| int readorder; | |||
| } TextContext; | |||
| #define OFFSET(x) offsetof(TextContext, x) | |||
| @@ -48,15 +49,12 @@ static int text_decode_frame(AVCodecContext *avctx, void *data, | |||
| AVBPrint buf; | |||
| AVSubtitle *sub = data; | |||
| const char *ptr = avpkt->data; | |||
| const TextContext *text = avctx->priv_data; | |||
| const int ts_start = av_rescale_q(avpkt->pts, avctx->time_base, (AVRational){1,100}); | |||
| const int ts_duration = avpkt->duration != -1 ? | |||
| av_rescale_q(avpkt->duration, avctx->time_base, (AVRational){1,100}) : -1; | |||
| TextContext *text = avctx->priv_data; | |||
| av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| if (ptr && avpkt->size > 0 && *ptr) { | |||
| ff_ass_bprint_text_event(&buf, ptr, avpkt->size, text->linebreaks, text->keep_ass_markup); | |||
| ret = ff_ass_add_rect_bprint(sub, &buf, ts_start, ts_duration); | |||
| ret = ff_ass_add_rect(sub, buf.str, text->readorder++, 0, NULL, NULL); | |||
| } | |||
| av_bprint_finalize(&buf, NULL); | |||
| if (ret < 0) | |||
| @@ -65,6 +63,12 @@ static int text_decode_frame(AVCodecContext *avctx, void *data, | |||
| return avpkt->size; | |||
| } | |||
| static void text_flush(AVCodecContext *avctx) | |||
| { | |||
| TextContext *text = avctx->priv_data; | |||
| text->readorder = 0; | |||
| } | |||
| #define DECLARE_CLASS(decname) static const AVClass decname ## _decoder_class = { \ | |||
| .class_name = #decname " decoder", \ | |||
| .item_name = av_default_item_name, \ | |||
| @@ -85,6 +89,7 @@ AVCodec ff_text_decoder = { | |||
| .decode = text_decode_frame, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .priv_class = &text_decoder_class, | |||
| .flush = text_flush, | |||
| }; | |||
| #endif | |||
| @@ -110,6 +115,7 @@ AVCodec ff_vplayer_decoder = { | |||
| .decode = text_decode_frame, | |||
| .init = linebreak_init, | |||
| .priv_class = &vplayer_decoder_class, | |||
| .flush = text_flush, | |||
| }; | |||
| #endif | |||
| @@ -126,6 +132,7 @@ AVCodec ff_stl_decoder = { | |||
| .decode = text_decode_frame, | |||
| .init = linebreak_init, | |||
| .priv_class = &stl_decoder_class, | |||
| .flush = text_flush, | |||
| }; | |||
| #endif | |||
| @@ -142,6 +149,7 @@ AVCodec ff_pjs_decoder = { | |||
| .decode = text_decode_frame, | |||
| .init = linebreak_init, | |||
| .priv_class = &pjs_decoder_class, | |||
| .flush = text_flush, | |||
| }; | |||
| #endif | |||
| @@ -158,6 +166,7 @@ AVCodec ff_subviewer1_decoder = { | |||
| .decode = text_decode_frame, | |||
| .init = linebreak_init, | |||
| .priv_class = &subviewer1_decoder_class, | |||
| .flush = text_flush, | |||
| }; | |||
| #endif | |||
| @@ -2426,6 +2426,76 @@ static int utf8_check(const uint8_t *str) | |||
| return 1; | |||
| } | |||
| static void insert_ts(AVBPrint *buf, int ts) | |||
| { | |||
| if (ts == -1) { | |||
| av_bprintf(buf, "9:59:59.99,"); | |||
| } else { | |||
| int h, m, s; | |||
| h = ts/360000; ts -= 360000*h; | |||
| m = ts/ 6000; ts -= 6000*m; | |||
| s = ts/ 100; ts -= 100*s; | |||
| av_bprintf(buf, "%d:%02d:%02d.%02d,", h, m, s, ts); | |||
| } | |||
| } | |||
| static int convert_sub_to_old_ass_form(AVSubtitle *sub, const AVPacket *pkt, AVRational tb) | |||
| { | |||
| int i; | |||
| AVBPrint buf; | |||
| av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| for (i = 0; i < sub->num_rects; i++) { | |||
| char *final_dialog; | |||
| const char *dialog; | |||
| AVSubtitleRect *rect = sub->rects[i]; | |||
| int ts_start, ts_duration = -1; | |||
| long int layer; | |||
| if (rect->type != SUBTITLE_ASS || !strncmp(rect->ass, "Dialogue ", 10)) | |||
| continue; | |||
| av_bprint_clear(&buf); | |||
| /* skip ReadOrder */ | |||
| dialog = strchr(rect->ass, ','); | |||
| if (!dialog) | |||
| continue; | |||
| dialog++; | |||
| /* extract Layer or Marked */ | |||
| layer = strtol(dialog, (char**)&dialog, 10); | |||
| if (*dialog != ',') | |||
| continue; | |||
| dialog++; | |||
| /* rescale timing to ASS time base (ms) */ | |||
| ts_start = av_rescale_q(pkt->pts, tb, av_make_q(1, 100)); | |||
| if (pkt->duration != -1) | |||
| ts_duration = av_rescale_q(pkt->duration, tb, av_make_q(1, 100)); | |||
| sub->end_display_time = FFMAX(sub->end_display_time, 10 * ts_duration); | |||
| /* construct ASS (standalone file form with timestamps) string */ | |||
| av_bprintf(&buf, "Dialogue: %ld,", layer); | |||
| insert_ts(&buf, ts_start); | |||
| insert_ts(&buf, ts_duration == -1 ? -1 : ts_start + ts_duration); | |||
| av_bprintf(&buf, "%s\r\n", dialog); | |||
| final_dialog = av_strdup(buf.str); | |||
| if (!av_bprint_is_complete(&buf) || !final_dialog) { | |||
| av_bprint_finalize(&buf, NULL); | |||
| return AVERROR(ENOMEM); | |||
| } | |||
| av_freep(&rect->ass); | |||
| rect->ass = final_dialog; | |||
| } | |||
| av_bprint_finalize(&buf, NULL); | |||
| return 0; | |||
| } | |||
| int avcodec_decode_subtitle2(AVCodecContext *avctx, AVSubtitle *sub, | |||
| int *got_sub_ptr, | |||
| AVPacket *avpkt) | |||
| @@ -2476,6 +2546,10 @@ int avcodec_decode_subtitle2(AVCodecContext *avctx, AVSubtitle *sub, | |||
| av_assert1((ret >= 0) >= !!*got_sub_ptr && | |||
| !!*got_sub_ptr >= !!sub->num_rects); | |||
| if (avctx->sub_text_format == FF_SUB_TEXT_FMT_ASS_WITH_TIMINGS | |||
| && *got_sub_ptr && sub->num_rects) | |||
| ret = convert_sub_to_old_ass_form(sub, avpkt, avctx->time_base); | |||
| if (sub->num_rects && !sub->end_display_time && avpkt->duration && | |||
| avctx->pkt_timebase.num) { | |||
| AVRational ms = { 1, 1000 }; | |||
| @@ -28,8 +28,8 @@ | |||
| #include "libavutil/version.h" | |||
| #define LIBAVCODEC_VERSION_MAJOR 57 | |||
| #define LIBAVCODEC_VERSION_MINOR 25 | |||
| #define LIBAVCODEC_VERSION_MICRO 101 | |||
| #define LIBAVCODEC_VERSION_MINOR 26 | |||
| #define LIBAVCODEC_VERSION_MICRO 100 | |||
| #define LIBAVCODEC_VERSION_INT AV_VERSION_INT(LIBAVCODEC_VERSION_MAJOR, \ | |||
| LIBAVCODEC_VERSION_MINOR, \ | |||
| @@ -85,15 +85,12 @@ static int webvtt_decode_frame(AVCodecContext *avctx, | |||
| int ret = 0; | |||
| AVSubtitle *sub = data; | |||
| const char *ptr = avpkt->data; | |||
| FFASSDecoderContext *s = avctx->priv_data; | |||
| AVBPrint buf; | |||
| av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |||
| if (ptr && avpkt->size > 0 && !webvtt_event_to_ass(&buf, ptr)) { | |||
| int ts_start = av_rescale_q(avpkt->pts, avctx->time_base, (AVRational){1,100}); | |||
| int ts_duration = avpkt->duration != -1 ? | |||
| av_rescale_q(avpkt->duration, avctx->time_base, (AVRational){1,100}) : -1; | |||
| ret = ff_ass_add_rect_bprint(sub, &buf, ts_start, ts_duration); | |||
| } | |||
| if (ptr && avpkt->size > 0 && !webvtt_event_to_ass(&buf, ptr)) | |||
| ret = ff_ass_add_rect(sub, buf.str, s->readorder++, 0, NULL, NULL); | |||
| av_bprint_finalize(&buf, NULL); | |||
| if (ret < 0) | |||
| return ret; | |||
| @@ -108,4 +105,6 @@ AVCodec ff_webvtt_decoder = { | |||
| .id = AV_CODEC_ID_WEBVTT, | |||
| .decode = webvtt_decode_frame, | |||
| .init = ff_ass_subtitle_header_default, | |||
| .flush = ff_ass_decoder_flush, | |||
| .priv_data_size = sizeof(FFASSDecoderContext), | |||
| }; | |||
| @@ -164,16 +164,28 @@ static int webvtt_encode_frame(AVCodecContext *avctx, | |||
| av_bprint_clear(&s->buffer); | |||
| for (i=0; i<sub->num_rects; i++) { | |||
| const char *ass = sub->rects[i]->ass; | |||
| if (sub->rects[i]->type != SUBTITLE_ASS) { | |||
| av_log(avctx, AV_LOG_ERROR, "Only SUBTITLE_ASS type supported.\n"); | |||
| return AVERROR(ENOSYS); | |||
| } | |||
| dialog = ff_ass_split_dialog(s->ass_ctx, sub->rects[i]->ass, 0, &num); | |||
| if (!strncmp(ass, "Dialogue: ", 10)) { | |||
| dialog = ff_ass_split_dialog(s->ass_ctx, ass, 0, &num); | |||
| // TODO reindent | |||
| for (; dialog && num--; dialog++) { | |||
| webvtt_style_apply(s, dialog->style); | |||
| ff_ass_split_override_codes(&webvtt_callbacks, s, dialog->text); | |||
| } | |||
| } else { | |||
| dialog = ff_ass_split_dialog2(s->ass_ctx, ass); | |||
| if (!dialog) | |||
| return AVERROR(ENOMEM); | |||
| webvtt_style_apply(s, dialog->style); | |||
| ff_ass_split_override_codes(&webvtt_callbacks, s, dialog->text); | |||
| ff_ass_free_dialog(&dialog); | |||
| } | |||
| } | |||
| if (!av_bprint_is_complete(&s->buffer)) | |||
| @@ -145,6 +145,7 @@ stream=0, decode=0 | |||
| pkt_timebase=1/25 | |||
| sub_charenc= | |||
| sub_charenc_mode=0x00000000 | |||
| sub_text_format=1 | |||
| refcounted_frames=false | |||
| side_data_only_packets=true | |||
| skip_alpha=false | |||
| @@ -300,6 +301,7 @@ stream=0, decode=1 | |||
| pkt_timebase=1/25 | |||
| sub_charenc= | |||
| sub_charenc_mode=0x00000000 | |||
| sub_text_format=1 | |||
| refcounted_frames=false | |||
| side_data_only_packets=true | |||
| skip_alpha=false | |||
| @@ -145,6 +145,7 @@ stream=0, decode=0 | |||
| pkt_timebase=1/25 | |||
| sub_charenc= | |||
| sub_charenc_mode=0x00000000 | |||
| sub_text_format=1 | |||
| refcounted_frames=false | |||
| side_data_only_packets=true | |||
| skip_alpha=false | |||
| @@ -300,6 +301,7 @@ stream=0, decode=1 | |||
| pkt_timebase=1/25 | |||
| sub_charenc= | |||
| sub_charenc_mode=0x00000000 | |||
| sub_text_format=1 | |||
| refcounted_frames=false | |||
| side_data_only_packets=true | |||
| skip_alpha=false | |||