31 uint32_t global_palette[16];
37 #define PUTNIBBLE(val)\
40 *q++ = bitbuf | ((val) & 0x0f);\
46 const uint8_t *bitmap,
int linesize,
51 unsigned int bitbuf = 0;
57 for (y = 0; y <
h; ++y) {
59 for(x = 0; x <
w; x +=
len) {
68 }
else if (
len < 0x10) {
71 }
else if (
len < 0x40) {
75 }
else if (x+
len ==
w) {
101 int alpha_a = 8, alpha_b = 8;
103 for (
i = 24;
i >= 0;
i -= 8) {
104 d = alpha_a * (
int)((
a >>
i) & 0xFF) -
105 alpha_b * (
int)((
b >>
i) & 0xFF);
121 unsigned count[256] = { 0 };
122 uint32_t *palette = (uint32_t *)
r->data[1];
124 int x, y,
i, j, match, d, best_d,
av_uninit(best_j);
127 for (y = 0; y <
r->h; y++) {
128 for (x = 0; x <
r->w; x++)
130 p +=
r->linesize[0] -
r->w;
132 for (
i = 0;
i < 256;
i++) {
137 match =
color < 0x33000000 ? 0 :
color < 0xCC000000 ? 1 : 17;
140 for (j = 0; j < 16; j++) {
150 hits[match] += count[
i];
155 int out_alpha[4],
unsigned hits[33])
158 int i, j, bright,
mult;
160 int selected[4] = { 0 };
161 uint32_t pseudopal[33] = { 0 };
162 uint32_t refcolor[3] = { 0x00000000, 0xFFFFFFFF, 0xFF000000 };
168 for (
i = 0;
i < 16;
i++) {
169 if (!(hits[1 +
i] + hits[17 +
i]))
173 for (j = 0; j < 3; j++, color >>= 8)
174 bright += (
color & 0xFF) < 0x40 || (
color & 0xFF) >= 0xC0;
176 hits[ 1 +
i] *=
mult;
177 hits[17 +
i] *=
mult;
181 for (
i = 0;
i < 4;
i++) {
182 for (j = 0; j < 33; j++)
183 if (hits[j] > hits[selected[
i]])
185 hits[selected[
i]] = 0;
190 for (
i = 0;
i < 16;
i++) {
194 for (
i = 0;
i < 3;
i++) {
196 for (j =
i + 1; j < 4; j++) {
199 FFSWAP(
int, selected[
i], selected[j]);
206 for (
i = 0;
i < 4;
i++) {
207 out_palette[
i] = selected[
i] ? (selected[
i] - 1) & 0xF : 0;
208 out_alpha [
i] = !selected[
i] ? 0 : selected[
i] < 17 ? 0x80 : 0xFF;
213 const uint32_t palette[],
214 const int out_palette[],
unsigned int const out_alpha[])
218 uint32_t pseudopal[4];
220 for (
i = 0;
i < 4;
i++)
221 pseudopal[
i] = (out_alpha[
i] << 24) |
223 for (
i = 0;
i < 256;
i++) {
225 for (j = 0; j < 4; j++) {
241 q = dst->
data[0] + (
src->x - dst->
x) +
243 for (y = 0; y <
src->h; y++) {
244 for (x = 0; x <
src->w; x++)
245 *(q++) = cmap[*(p++)];
246 p +=
src->linesize[0] -
src->w;
252 uint8_t *outbuf,
int outbuf_size,
257 int offset1, offset2;
258 int i, rects =
h->num_rects, ret;
259 unsigned global_palette_hits[33] = { 0 };
268 if (rects == 0 || !
h->rects)
270 for (
i = 0;
i < rects;
i++)
276 for (
i = 0;
i < rects;
i++)
284 for (
i = 0;
i < rects;
i++)
285 if (!
h->rects[
i]->data[0]) {
288 for (j = 0; j < 4; j++) {
290 rect->linesize[j] =
rect->pict.linesize[j];
296 vrect = *
h->rects[0];
303 int xmin =
h->rects[0]->x, xmax = xmin +
h->rects[0]->w;
304 int ymin =
h->rects[0]->y, ymax = ymin +
h->rects[0]->h;
305 for (
i = 1;
i < rects;
i++) {
306 xmin =
FFMIN(xmin,
h->rects[
i]->x);
307 ymin =
FFMIN(ymin,
h->rects[
i]->y);
308 xmax =
FFMAX(xmax,
h->rects[
i]->x +
h->rects[
i]->w);
309 ymax =
FFMAX(ymax,
h->rects[
i]->y +
h->rects[
i]->h);
313 vrect.
w = xmax - xmin;
314 vrect.
h = ymax - ymin;
319 global_palette_hits[0] = vrect.
w * vrect.
h;
320 for (
i = 0;
i < rects;
i++)
321 global_palette_hits[0] -=
h->rects[
i]->w *
h->rects[
i]->h;
324 for (
i = 0;
i < rects;
i++)
326 select_palette(avctx, out_palette, out_alpha, global_palette_hits);
331 vrect.
data [0] = vrect_data;
333 for (
i = 0;
i < rects;
i++) {
335 out_palette, out_alpha);
338 for (
i = 0;
i < 4;
i++)
342 out_palette, out_alpha);
346 for (
i = 0;
i < 4;
i++)
349 out_palette[
i], out_alpha[
i] >> 4);
354 offset1 = q - outbuf;
356 if ((q - outbuf) + vrect.
w * vrect.
h / 2 + 17 + 21 > outbuf_size) {
362 vrect.
w, (vrect.
h + 1) >> 1, cmap);
363 offset2 = q - outbuf;
365 vrect.
w, vrect.
h >> 1, cmap);
376 bytestream_put_be16(&qq, q - outbuf);
379 bytestream_put_be16(&q, (
h->start_display_time*90) >> 10);
380 bytestream_put_be16(&q, (q - outbuf) + 8 + 12 + 2);
382 *q++ = (out_palette[3] << 4) | out_palette[2];
383 *q++ = (out_palette[1] << 4) | out_palette[0];
385 *q++ = (out_alpha[3] & 0xF0) | (out_alpha[2] >> 4);
386 *q++ = (out_alpha[1] & 0xF0) | (out_alpha[0] >> 4);
389 x2 = vrect.
x + vrect.
w - 1;
390 y2 = vrect.
y + vrect.
h - 1;
395 *q++ = (vrect.
x << 4) | ((x2 >> 8) & 0xf);
399 *q++ = (vrect.
y << 4) | ((y2 >> 8) & 0xf);
404 bytestream_put_be16(&q, offset1);
405 bytestream_put_be16(&q, offset2);
407 *q++ = forced ? 0x00 : 0x01;
411 bytestream_put_be16(&q, (
h->end_display_time*90) >> 10);
412 bytestream_put_be16(&q, (q - outbuf) - 2 );
417 bytestream_put_be16(&qq, q - outbuf);
453 static const uint32_t default_palette[16] = {
454 0x000000, 0x0000FF, 0x00FF00, 0xFF0000,
455 0xFFFF00, 0xFF00FF, 0x00FFFF, 0xFFFFFF,
456 0x808000, 0x8080FF, 0x800080, 0x80FF80,
457 0x008080, 0xFF8080, 0x555555, 0xAAAAAA,
473 for (
i = 0;
i < 16;
i++)
485 unsigned char *buf,
int buf_size,
495 #define OFFSET(x) offsetof(DVDSubtitleContext, x)
496 #define SE AV_OPT_FLAG_SUBTITLE_PARAM | AV_OPT_FLAG_ENCODING_PARAM
499 {
"even_rows_fix",
"Make number of rows even (workaround for some players)",
OFFSET(even_rows_fix),
AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1,
SE},
simple assert() macros that are a bit more flexible than ISO C assert().
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Libavcodec external API header.
void av_bprintf(AVBPrint *buf, const char *fmt,...)
void av_bprint_init(AVBPrint *buf, unsigned size_init, unsigned size_max)
int av_bprint_finalize(AVBPrint *buf, char **ret_str)
Finalize a print buffer.
static int av_bprint_is_complete(const AVBPrint *buf)
Test if the print buffer is complete (not truncated).
#define AV_BPRINT_SIZE_AUTOMATIC
#define FFSWAP(type, a, b)
static float sub(float src0, float src1)
void ff_dvdsub_parse_palette(uint32_t *palette, const char *p)
static const AVOption options[]
static int dvdsub_init(AVCodecContext *avctx)
static int encode_dvd_subtitles(AVCodecContext *avctx, uint8_t *outbuf, int outbuf_size, const AVSubtitle *h)
static int dvdsub_encode(AVCodecContext *avctx, unsigned char *buf, int buf_size, const AVSubtitle *sub)
static int bprint_to_extradata(AVCodecContext *avctx, struct AVBPrint *buf)
static const AVClass dvdsubenc_class
static int color_distance(uint32_t a, uint32_t b)
static void select_palette(AVCodecContext *avctx, int out_palette[4], int out_alpha[4], unsigned hits[33])
static void count_colors(AVCodecContext *avctx, unsigned hits[33], const AVSubtitleRect *r)
Count colors used in a rectangle, quantizing alpha and grouping by nearest global palette entry.
static void dvd_encode_rle(uint8_t **pq, const uint8_t *bitmap, int linesize, int w, int h, const int cmap[256])
static void build_color_map(AVCodecContext *avctx, int cmap[], const uint32_t palette[], const int out_palette[], unsigned int const out_alpha[])
static void copy_rectangle(AVSubtitleRect *dst, AVSubtitleRect *src, int cmap[])
AVCodec ff_dvdsub_encoder
#define AV_SUBTITLE_FLAG_FORCED
@ SUBTITLE_BITMAP
A bitmap, pict will be set.
@ AV_CODEC_ID_DVD_SUBTITLE
#define AVERROR_BUFFER_TOO_SMALL
Buffer too small.
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
const char * av_default_item_name(void *ptr)
Return the context name.
void * av_calloc(size_t nmemb, size_t size)
Non-inlined equivalent of av_mallocz_array().
int av_image_check_size(unsigned int w, unsigned int h, int log_offset, void *log_ctx)
Check if the given dimension of an image is valid, meaning that all bytes of the image can be address...
#define LIBAVUTIL_VERSION_INT
static int16_t mult(Float11 *f1, Float11 *f2)
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
#define FF_DISABLE_DEPRECATION_WARNINGS
#define PTRDIFF_SPECIFIER
#define FF_ENABLE_DEPRECATION_WARNINGS
static const uint32_t color[16+AV_CLASS_CATEGORY_NB]
Describe the class of an AVClass context structure.
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
main external API structure.
int width
picture width / height.
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
const char * name
Name of the codec implementation.
int x
top left corner of pict, undefined when pict is not set
int w
width of pict, undefined when pict is not set
uint8_t * data[4]
data+linesize for the bitmap of this subtitle.
int y
top left corner of pict, undefined when pict is not set
int h
height of pict, undefined when pict is not set
uint32_t global_palette[16]