annotate xsubdec.c @ 10550:eb415f52f9f9 libavcodec

Factorize av_get_bit_rate (for future use outside of libavcodec). Patch by Robert Krger, krueger signal7 de
author cehoyos
date Sat, 21 Nov 2009 09:58:15 +0000
parents 8ac9bc10b485
children 8a4984c5cacc
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
5493
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
1 /*
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
2 * XSUB subtitle decoder
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
3 * Copyright (c) 2007 Reimar Döffinger
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
4 *
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
5 * This file is part of FFmpeg.
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
6 *
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
7 * FFmpeg is free software; you can redistribute it and/or
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
9 * License as published by the Free Software Foundation; either
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
10 * version 2.1 of the License, or (at your option) any later version.
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
11 *
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
12 * FFmpeg is distributed in the hope that it will be useful,
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
15 * Lesser General Public License for more details.
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
16 *
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
17 * You should have received a copy of the GNU Lesser General Public
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
18 * License along with FFmpeg; if not, write to the Free Software
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
80b481ec72a6 Add proper license header
reimar
parents: 5492
diff changeset
20 */
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
21 #include "avcodec.h"
9428
0dce4fe6e6f3 Rename bitstream.h to get_bits.h.
stefano
parents: 9355
diff changeset
22 #include "get_bits.h"
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
23 #include "bytestream.h"
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
24
6517
48759bfbd073 Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents: 6218
diff changeset
25 static av_cold int decode_init(AVCodecContext *avctx) {
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
26 avctx->pix_fmt = PIX_FMT_PAL8;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
27 return 0;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
28 }
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
29
5484
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
30 static const uint8_t tc_offsets[9] = { 0, 1, 3, 4, 6, 7, 9, 10, 11 };
8769
6fc3497dc81f Fix timestamp parsing in XSUB decoder.
cehoyos
parents: 8516
diff changeset
31 static const uint8_t tc_muls[9] = { 10, 6, 10, 6, 10, 10, 10, 10, 1 };
5484
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
32
10054
8ab9fbad11b2 Fix start_display_time/end_display_time to be relative to packet pts in xsub decoder.
reimar
parents: 10053
diff changeset
33 static int64_t parse_timecode(const uint8_t *buf, int64_t packet_time) {
5484
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
34 int i;
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
35 int64_t ms = 0;
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
36 if (buf[2] != ':' || buf[5] != ':' || buf[8] != '.')
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
37 return AV_NOPTS_VALUE;
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
38 for (i = 0; i < sizeof(tc_offsets); i++) {
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
39 uint8_t c = buf[tc_offsets[i]] - '0';
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
40 if (c > 9) return AV_NOPTS_VALUE;
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
41 ms = (ms + c) * tc_muls[i];
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
42 }
10054
8ab9fbad11b2 Fix start_display_time/end_display_time to be relative to packet pts in xsub decoder.
reimar
parents: 10053
diff changeset
43 return ms - packet_time;
5484
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
44 }
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
45
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
46 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size,
9355
54bc8a2727b0 Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents: 8810
diff changeset
47 AVPacket *avpkt) {
54bc8a2727b0 Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents: 8810
diff changeset
48 const uint8_t *buf = avpkt->data;
54bc8a2727b0 Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents: 8810
diff changeset
49 int buf_size = avpkt->size;
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
50 AVSubtitle *sub = data;
6218
michael
parents: 5982
diff changeset
51 const uint8_t *buf_end = buf + buf_size;
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
52 uint8_t *bitmap;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
53 int w, h, x, y, rlelen, i;
10054
8ab9fbad11b2 Fix start_display_time/end_display_time to be relative to packet pts in xsub decoder.
reimar
parents: 10053
diff changeset
54 int64_t packet_time = 0;
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
55 GetBitContext gb;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
56
10050
b8313eb3dca8 Make sure AVSubtitle is initialized, memset it to 0.
reimar
parents: 9988
diff changeset
57 memset(sub, 0, sizeof(*sub));
8810
367e4d405cd9 Set AVSubtitle format to 0. Neither dvdsubdec nor xsubdec intializes format
diego
parents: 8769
diff changeset
58
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
59 // check that at least header fits
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
60 if (buf_size < 27 + 7 * 2 + 4 * 3) {
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
61 av_log(avctx, AV_LOG_ERROR, "coded frame too small\n");
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
62 return -1;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
63 }
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
64
5484
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
65 // read start and end time
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
66 if (buf[0] != '[' || buf[13] != '-' || buf[26] != ']') {
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
67 av_log(avctx, AV_LOG_ERROR, "invalid time code\n");
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
68 return -1;
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
69 }
10054
8ab9fbad11b2 Fix start_display_time/end_display_time to be relative to packet pts in xsub decoder.
reimar
parents: 10053
diff changeset
70 if (avpkt->pts != AV_NOPTS_VALUE)
8ab9fbad11b2 Fix start_display_time/end_display_time to be relative to packet pts in xsub decoder.
reimar
parents: 10053
diff changeset
71 packet_time = av_rescale_q(avpkt->pts, AV_TIME_BASE_Q, (AVRational){1, 1000});
8ab9fbad11b2 Fix start_display_time/end_display_time to be relative to packet pts in xsub decoder.
reimar
parents: 10053
diff changeset
72 sub->start_display_time = parse_timecode(buf + 1, packet_time);
8ab9fbad11b2 Fix start_display_time/end_display_time to be relative to packet pts in xsub decoder.
reimar
parents: 10053
diff changeset
73 sub->end_display_time = parse_timecode(buf + 14, packet_time);
5484
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
74 buf += 27;
ad3e24850727 Add forgotten xsub timecode parsing
reimar
parents: 5483
diff changeset
75
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
76 // read header
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
77 w = bytestream_get_le16(&buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
78 h = bytestream_get_le16(&buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
79 if (avcodec_check_dimensions(avctx, w, h) < 0)
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
80 return -1;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
81 x = bytestream_get_le16(&buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
82 y = bytestream_get_le16(&buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
83 // skip bottom right position, it gives no new information
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
84 bytestream_get_le16(&buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
85 bytestream_get_le16(&buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
86 rlelen = bytestream_get_le16(&buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
87
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
88 // allocate sub and set values
10053
089dd845c8c0 Reindent.
reimar
parents: 10052
diff changeset
89 sub->rects = av_mallocz(sizeof(*sub->rects));
089dd845c8c0 Reindent.
reimar
parents: 10052
diff changeset
90 sub->rects[0] = av_mallocz(sizeof(*sub->rects[0]));
089dd845c8c0 Reindent.
reimar
parents: 10052
diff changeset
91 sub->num_rects = 1;
8512
aa45029f5cd7 Change AVSubtitle.rects to an array of pointers so ABI does not break
michael
parents: 7040
diff changeset
92 sub->rects[0]->x = x; sub->rects[0]->y = y;
aa45029f5cd7 Change AVSubtitle.rects to an array of pointers so ABI does not break
michael
parents: 7040
diff changeset
93 sub->rects[0]->w = w; sub->rects[0]->h = h;
9988
b8a9cfe64488 Set subtitle type in DVD and XSUB subtitle decoders.
stefano
parents: 9428
diff changeset
94 sub->rects[0]->type = SUBTITLE_BITMAP;
8516
315b302fcd1d Replace AVSubtitleRect.rgba_palette and bitmap by AVPicture.
michael
parents: 8512
diff changeset
95 sub->rects[0]->pict.linesize[0] = w;
315b302fcd1d Replace AVSubtitleRect.rgba_palette and bitmap by AVPicture.
michael
parents: 8512
diff changeset
96 sub->rects[0]->pict.data[0] = av_malloc(w * h);
8512
aa45029f5cd7 Change AVSubtitle.rects to an array of pointers so ABI does not break
michael
parents: 7040
diff changeset
97 sub->rects[0]->nb_colors = 4;
10069
8ac9bc10b485 Always allocate a buffer of AVPALETTE_SIZE for palette in the subtitle
reimar
parents: 10054
diff changeset
98 sub->rects[0]->pict.data[1] = av_mallocz(AVPALETTE_SIZE);
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
99
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
100 // read palette
8512
aa45029f5cd7 Change AVSubtitle.rects to an array of pointers so ABI does not break
michael
parents: 7040
diff changeset
101 for (i = 0; i < sub->rects[0]->nb_colors; i++)
8516
315b302fcd1d Replace AVSubtitleRect.rgba_palette and bitmap by AVPicture.
michael
parents: 8512
diff changeset
102 ((uint32_t*)sub->rects[0]->pict.data[1])[i] = bytestream_get_be24(&buf);
5489
dc54869af30b Colours except background should not be transparent
reimar
parents: 5488
diff changeset
103 // make all except background (first entry) non-transparent
8512
aa45029f5cd7 Change AVSubtitle.rects to an array of pointers so ABI does not break
michael
parents: 7040
diff changeset
104 for (i = 1; i < sub->rects[0]->nb_colors; i++)
8516
315b302fcd1d Replace AVSubtitleRect.rgba_palette and bitmap by AVPicture.
michael
parents: 8512
diff changeset
105 ((uint32_t*)sub->rects[0]->pict.data[1])[i] |= 0xff000000;
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
106
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
107 // process RLE-compressed data
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
108 rlelen = FFMIN(rlelen, buf_end - buf);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
109 init_get_bits(&gb, buf, rlelen * 8);
8516
315b302fcd1d Replace AVSubtitleRect.rgba_palette and bitmap by AVPicture.
michael
parents: 8512
diff changeset
110 bitmap = sub->rects[0]->pict.data[0];
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
111 for (y = 0; y < h; y++) {
5490
12d77ed34985 Deinterlace xsub subtitles
reimar
parents: 5489
diff changeset
112 // interlaced: do odd lines
8516
315b302fcd1d Replace AVSubtitleRect.rgba_palette and bitmap by AVPicture.
michael
parents: 8512
diff changeset
113 if (y == (h + 1) / 2) bitmap = sub->rects[0]->pict.data[0] + w;
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
114 for (x = 0; x < w; ) {
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
115 int log2 = ff_log2_tab[show_bits(&gb, 8)];
5487
3039b660bf35 get rid of xsubdec array and calculate value instead
reimar
parents: 5486
diff changeset
116 int run = get_bits(&gb, 14 - 4 * (log2 >> 1));
5982
1900b70712ab colour --> color in variable names
diego
parents: 5493
diff changeset
117 int color = get_bits(&gb, 2);
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
118 run = FFMIN(run, w - x);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
119 // run length 0 means till end of row
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
120 if (!run) run = w - x;
5982
1900b70712ab colour --> color in variable names
diego
parents: 5493
diff changeset
121 memset(bitmap, color, run);
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
122 bitmap += run;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
123 x += run;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
124 }
5490
12d77ed34985 Deinterlace xsub subtitles
reimar
parents: 5489
diff changeset
125 // interlaced, skip every second line
12d77ed34985 Deinterlace xsub subtitles
reimar
parents: 5489
diff changeset
126 bitmap += w;
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
127 align_get_bits(&gb);
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
128 }
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
129 *data_size = 1;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
130 return buf_size;
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
131 }
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
132
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
133 AVCodec xsub_decoder = {
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
134 "xsub",
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
135 CODEC_TYPE_SUBTITLE,
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
136 CODEC_ID_XSUB,
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
137 0,
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
138 decode_init,
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
139 NULL,
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
140 NULL,
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
141 decode_frame,
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 6710
diff changeset
142 .long_name = NULL_IF_CONFIG_SMALL("XSUB"),
5483
447409d8770f First version of xsub decoder, not yet tested
reimar
parents:
diff changeset
143 };