/src/ffmpeg/libavcodec/assenc.c
Line | Count | Source (jump to first uncovered line) |
1 | | /* |
2 | | * SSA/ASS encoder |
3 | | * Copyright (c) 2010 Aurelien Jacobs <aurel@gnuage.org> |
4 | | * |
5 | | * This file is part of FFmpeg. |
6 | | * |
7 | | * FFmpeg is free software; you can redistribute it and/or |
8 | | * modify it under the terms of the GNU Lesser General Public |
9 | | * License as published by the Free Software Foundation; either |
10 | | * version 2.1 of the License, or (at your option) any later version. |
11 | | * |
12 | | * FFmpeg is distributed in the hope that it will be useful, |
13 | | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
14 | | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
15 | | * Lesser General Public License for more details. |
16 | | * |
17 | | * You should have received a copy of the GNU Lesser General Public |
18 | | * License along with FFmpeg; if not, write to the Free Software |
19 | | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
20 | | */ |
21 | | |
22 | | #include "config_components.h" |
23 | | |
24 | | #include <string.h> |
25 | | |
26 | | #include "avcodec.h" |
27 | | #include "ass.h" |
28 | | #include "codec_internal.h" |
29 | | #include "libavutil/avstring.h" |
30 | | #include "libavutil/internal.h" |
31 | | #include "libavutil/mem.h" |
32 | | |
33 | | static av_cold int ass_encode_init(AVCodecContext *avctx) |
34 | 0 | { |
35 | 0 | avctx->extradata = av_malloc(avctx->subtitle_header_size + 1); |
36 | 0 | if (!avctx->extradata) |
37 | 0 | return AVERROR(ENOMEM); |
38 | 0 | memcpy(avctx->extradata, avctx->subtitle_header, avctx->subtitle_header_size); |
39 | 0 | avctx->extradata_size = avctx->subtitle_header_size; |
40 | 0 | avctx->extradata[avctx->extradata_size] = 0; |
41 | 0 | return 0; |
42 | 0 | } |
43 | | |
44 | | static int ass_encode_frame(AVCodecContext *avctx, |
45 | | unsigned char *buf, int bufsize, |
46 | | const AVSubtitle *sub) |
47 | 0 | { |
48 | 0 | size_t len; |
49 | |
|
50 | 0 | if (sub->num_rects != 1) { |
51 | 0 | av_log(avctx, AV_LOG_ERROR, "Only one rect per AVSubtitle is supported in ASS.\n"); |
52 | 0 | return AVERROR_INVALIDDATA; |
53 | 0 | } |
54 | | |
55 | 0 | if (sub->rects[0]->type != SUBTITLE_ASS) { |
56 | 0 | av_log(avctx, AV_LOG_ERROR, "Only SUBTITLE_ASS type supported.\n"); |
57 | 0 | return AVERROR(EINVAL); |
58 | 0 | } |
59 | | |
60 | 0 | len = av_strlcpy(buf, sub->rects[0]->ass, bufsize); |
61 | |
|
62 | 0 | if (len >= bufsize) { |
63 | 0 | av_log(avctx, AV_LOG_ERROR, "Buffer too small for ASS event.\n"); |
64 | 0 | return AVERROR_BUFFER_TOO_SMALL; |
65 | 0 | } |
66 | | |
67 | 0 | return len; |
68 | 0 | } |
69 | | |
70 | | #if CONFIG_SSA_ENCODER |
71 | | const FFCodec ff_ssa_encoder = { |
72 | | .p.name = "ssa", |
73 | | CODEC_LONG_NAME("ASS (Advanced SubStation Alpha) subtitle"), |
74 | | .p.type = AVMEDIA_TYPE_SUBTITLE, |
75 | | .p.id = AV_CODEC_ID_ASS, |
76 | | .init = ass_encode_init, |
77 | | FF_CODEC_ENCODE_SUB_CB(ass_encode_frame), |
78 | | }; |
79 | | #endif |
80 | | |
81 | | #if CONFIG_ASS_ENCODER |
82 | | const FFCodec ff_ass_encoder = { |
83 | | .p.name = "ass", |
84 | | CODEC_LONG_NAME("ASS (Advanced SubStation Alpha) subtitle"), |
85 | | .p.type = AVMEDIA_TYPE_SUBTITLE, |
86 | | .p.id = AV_CODEC_ID_ASS, |
87 | | .init = ass_encode_init, |
88 | | FF_CODEC_ENCODE_SUB_CB(ass_encode_frame), |
89 | | }; |
90 | | #endif |