FFmpeg
assenc.c
Go to the documentation of this file.
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 <string.h>
23 
24 #include "avcodec.h"
25 #include "ass.h"
26 #include "internal.h"
27 #include "libavutil/avstring.h"
28 #include "libavutil/internal.h"
29 #include "libavutil/mem.h"
30 
32 {
33  avctx->extradata = av_malloc(avctx->subtitle_header_size + 1);
34  if (!avctx->extradata)
35  return AVERROR(ENOMEM);
36  memcpy(avctx->extradata, avctx->subtitle_header, avctx->subtitle_header_size);
37  avctx->extradata_size = avctx->subtitle_header_size;
38  avctx->extradata[avctx->extradata_size] = 0;
39  return 0;
40 }
41 
42 static int ass_encode_frame(AVCodecContext *avctx,
43  unsigned char *buf, int bufsize,
44  const AVSubtitle *sub)
45 {
46  int i, len, total_len = 0;
47 
48  for (i=0; i<sub->num_rects; i++) {
49  const char *ass = sub->rects[i]->ass;
50 
51  if (sub->rects[i]->type != SUBTITLE_ASS) {
52  av_log(avctx, AV_LOG_ERROR, "Only SUBTITLE_ASS type supported.\n");
53  return AVERROR(EINVAL);
54  }
55 
56  len = av_strlcpy(buf+total_len, ass, bufsize-total_len);
57 
58  if (len > bufsize-total_len-1) {
59  av_log(avctx, AV_LOG_ERROR, "Buffer too small for ASS event.\n");
61  }
62 
63  total_len += len;
64  }
65 
66  return total_len;
67 }
68 
69 #if CONFIG_SSA_ENCODER
70 const AVCodec ff_ssa_encoder = {
71  .name = "ssa",
72  .long_name = NULL_IF_CONFIG_SMALL("ASS (Advanced SubStation Alpha) subtitle"),
73  .type = AVMEDIA_TYPE_SUBTITLE,
74  .id = AV_CODEC_ID_ASS,
75  .init = ass_encode_init,
76  .encode_sub = ass_encode_frame,
77  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
78 };
79 #endif
80 
81 #if CONFIG_ASS_ENCODER
82 const AVCodec ff_ass_encoder = {
83  .name = "ass",
84  .long_name = NULL_IF_CONFIG_SMALL("ASS (Advanced SubStation Alpha) subtitle"),
85  .type = AVMEDIA_TYPE_SUBTITLE,
86  .id = AV_CODEC_ID_ASS,
87  .init = ass_encode_init,
88  .encode_sub = ass_encode_frame,
89  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
90 };
91 #endif
AVSubtitle
Definition: avcodec.h:2289
AVCodec
AVCodec.
Definition: codec.h:202
AVMEDIA_TYPE_SUBTITLE
@ AVMEDIA_TYPE_SUBTITLE
Definition: avutil.h:204
FF_CODEC_CAP_INIT_THREADSAFE
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
Definition: internal.h:42
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
sub
static float sub(float src0, float src1)
Definition: dnn_backend_native_layer_mathbinary.c:31
internal.h
AVCodecContext::subtitle_header
uint8_t * subtitle_header
Header containing style information for text subtitles.
Definition: avcodec.h:1683
ff_ass_encoder
const AVCodec ff_ass_encoder
ff_ssa_encoder
const AVCodec ff_ssa_encoder
av_malloc
#define av_malloc(s)
Definition: tableprint_vlc.h:31
SUBTITLE_ASS
@ SUBTITLE_ASS
Formatted text, the ass field must be set by the decoder and is authoritative.
Definition: avcodec.h:2256
AV_CODEC_ID_ASS
@ AV_CODEC_ID_ASS
Definition: codec_id.h:543
AVERROR_BUFFER_TOO_SMALL
#define AVERROR_BUFFER_TOO_SMALL
Buffer too small.
Definition: error.h:53
ass.h
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
av_cold
#define av_cold
Definition: attributes.h:90
AVCodecContext::extradata_size
int extradata_size
Definition: avcodec.h:485
AVCodecContext::subtitle_header_size
int subtitle_header_size
Definition: avcodec.h:1684
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
ass_encode_frame
static int ass_encode_frame(AVCodecContext *avctx, unsigned char *buf, int bufsize, const AVSubtitle *sub)
Definition: assenc.c:42
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:271
AVCodecContext::extradata
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:484
internal.h
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:209
len
int len
Definition: vorbis_enc_data.h:426
avcodec.h
AVCodecContext
main external API structure.
Definition: avcodec.h:383
ass_encode_init
static av_cold int ass_encode_init(AVCodecContext *avctx)
Definition: assenc.c:31
mem.h
av_strlcpy
size_t av_strlcpy(char *dst, const char *src, size_t size)
Copy the string src to dst, but no more than size - 1 bytes, and null-terminate dst.
Definition: avstring.c:83
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:28
avstring.h