ass.c 6.33 KB
Newer Older
1
/*
2
 * SSA/ASS common functions
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 * Copyright (c) 2010  Aurelien Jacobs <aurel@gnuage.org>
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include "avcodec.h"
#include "ass.h"
24
#include "libavutil/avassert.h"
25
#include "libavutil/avstring.h"
26
#include "libavutil/bprint.h"
27
#include "libavutil/common.h"
28

Aurelien Jacobs's avatar
Aurelien Jacobs committed
29 30 31 32
int ff_ass_subtitle_header(AVCodecContext *avctx,
                           const char *font, int font_size,
                           int color, int back_color,
                           int bold, int italic, int underline,
33
                           int border_style, int alignment)
Aurelien Jacobs's avatar
Aurelien Jacobs committed
34
{
35
    avctx->subtitle_header = av_asprintf(
Aurelien Jacobs's avatar
Aurelien Jacobs committed
36
             "[Script Info]\r\n"
37
             "; Script generated by FFmpeg/Lavc%s\r\n"
38
             "ScriptType: v4.00+\r\n"
39 40
             "PlayResX: %d\r\n"
             "PlayResY: %d\r\n"
Aurelien Jacobs's avatar
Aurelien Jacobs committed
41 42
             "\r\n"
             "[V4+ Styles]\r\n"
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61

             /* ASSv4 header */
             "Format: Name, "
             "Fontname, Fontsize, "
             "PrimaryColour, SecondaryColour, OutlineColour, BackColour, "
             "Bold, Italic, Underline, StrikeOut, "
             "ScaleX, ScaleY, "
             "Spacing, Angle, "
             "BorderStyle, Outline, Shadow, "
             "Alignment, MarginL, MarginR, MarginV, "
             "Encoding\r\n"

             "Style: "
             "Default,"             /* Name */
             "%s,%d,"               /* Font{name,size} */
             "&H%x,&H%x,&H%x,&H%x," /* {Primary,Secondary,Outline,Back}Colour */
             "%d,%d,%d,0,"          /* Bold, Italic, Underline, StrikeOut */
             "100,100,"             /* Scale{X,Y} */
             "0,0,"                 /* Spacing, Angle */
62
             "%d,1,0,"              /* BorderStyle, Outline, Shadow */
63 64 65
             "%d,10,10,10,"         /* Alignment, Margin[LRV] */
             "0\r\n"                /* Encoding */

Aurelien Jacobs's avatar
Aurelien Jacobs committed
66 67
             "\r\n"
             "[Events]\r\n"
68
             "Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text\r\n",
69
             !(avctx->flags & AV_CODEC_FLAG_BITEXACT) ? AV_STRINGIFY(LIBAVCODEC_VERSION) : "",
70
             ASS_DEFAULT_PLAYRESX, ASS_DEFAULT_PLAYRESY,
Aurelien Jacobs's avatar
Aurelien Jacobs committed
71
             font, font_size, color, color, back_color, back_color,
72
             -bold, -italic, -underline, border_style, alignment);
Aurelien Jacobs's avatar
Aurelien Jacobs committed
73 74 75 76 77 78 79 80 81 82

    if (!avctx->subtitle_header)
        return AVERROR(ENOMEM);
    avctx->subtitle_header_size = strlen(avctx->subtitle_header);
    return 0;
}

int ff_ass_subtitle_header_default(AVCodecContext *avctx)
{
    return ff_ass_subtitle_header(avctx, ASS_DEFAULT_FONT,
83 84 85 86 87 88
                               ASS_DEFAULT_FONT_SIZE,
                               ASS_DEFAULT_COLOR,
                               ASS_DEFAULT_BACK_COLOR,
                               ASS_DEFAULT_BOLD,
                               ASS_DEFAULT_ITALIC,
                               ASS_DEFAULT_UNDERLINE,
89
                               ASS_DEFAULT_BORDERSTYLE,
90
                               ASS_DEFAULT_ALIGNMENT);
Aurelien Jacobs's avatar
Aurelien Jacobs committed
91 92
}

93 94
char *ff_ass_get_dialog(int readorder, int layer, const char *style,
                        const char *speaker, const char *text)
95
{
96 97 98
    return av_asprintf("%d,%d,%s,%s,0,0,0,,%s",
                       readorder, layer, style ? style : "Default",
                       speaker ? speaker : "", text);
99 100 101
}

int ff_ass_add_rect(AVSubtitle *sub, const char *dialog,
102 103
                    int readorder, int layer, const char *style,
                    const char *speaker)
104
{
105
    char *ass_str;
106 107
    AVSubtitleRect **rects;

108
    rects = av_realloc_array(sub->rects, (sub->num_rects+1), sizeof(*sub->rects));
109
    if (!rects)
110
        return AVERROR(ENOMEM);
111 112
    sub->rects = rects;
    rects[sub->num_rects]       = av_mallocz(sizeof(*rects[0]));
113
    if (!rects[sub->num_rects])
114
        return AVERROR(ENOMEM);
115
    rects[sub->num_rects]->type = SUBTITLE_ASS;
116 117 118 119
    ass_str = ff_ass_get_dialog(readorder, layer, style, speaker, dialog);
    if (!ass_str)
        return AVERROR(ENOMEM);
    rects[sub->num_rects]->ass = ass_str;
120
    sub->num_rects++;
121
    return 0;
122
}
123

124
void ff_ass_decoder_flush(AVCodecContext *avctx)
125
{
126
    FFASSDecoderContext *s = avctx->priv_data;
127 128
    if (!(avctx->flags2 & AV_CODEC_FLAG2_RO_FLUSH_NOOP))
        s->readorder = 0;
129 130
}

131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167
void ff_ass_bprint_text_event(AVBPrint *buf, const char *p, int size,
                             const char *linebreaks, int keep_ass_markup)
{
    const char *p_end = p + size;

    for (; p < p_end && *p; p++) {

        /* forced custom line breaks, not accounted as "normal" EOL */
        if (linebreaks && strchr(linebreaks, *p)) {
            av_bprintf(buf, "\\N");

        /* standard ASS escaping so random characters don't get mis-interpreted
         * as ASS */
        } else if (!keep_ass_markup && strchr("{}\\", *p)) {
            av_bprintf(buf, "\\%c", *p);

        /* some packets might end abruptly (no \0 at the end, like for example
         * in some cases of demuxing from a classic video container), some
         * might be terminated with \n or \r\n which we have to remove (for
         * consistency with those who haven't), and we also have to deal with
         * evil cases such as \r at the end of the buffer (and no \0 terminated
         * character) */
        } else if (p[0] == '\n') {
            /* some stuff left so we can insert a line break */
            if (p < p_end - 1)
                av_bprintf(buf, "\\N");
        } else if (p[0] == '\r' && p < p_end - 1 && p[1] == '\n') {
            /* \r followed by a \n, we can skip it. We don't insert the \N yet
             * because we don't know if it is followed by more text */
            continue;

        /* finally, a sane character */
        } else {
            av_bprint_chars(buf, *p, 1);
        }
    }
}