Mercurial > libavcodec.hg
annotate mjpegbdec.c @ 11352:6e0af2cfdcfe libavcodec
Do MC and IDCT in coding (hilbert) order
This increases the slice size to 64 pixels, due to having to decode an
entire chroma superblock row per slice.
This can be up to 6% slower depending on clip and CPU, but is necessary
for future optimizations that gain significantly more than was lost.
author | conrad |
---|---|
date | Wed, 03 Mar 2010 23:27:40 +0000 |
parents | f9769330c214 |
children | 8a4984c5cacc |
rev | line source |
---|---|
5044 | 1 /* |
2 * Apple MJPEG-B decoder | |
3 * Copyright (c) 2002 Alex Beregszaszi | |
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 /** | |
8718
e9d9d946f213
Use full internal pathname in doxygen @file directives.
diego
parents:
7040
diff
changeset
|
23 * @file libavcodec/mjpegbdec.c |
5044 | 24 * Apple MJPEG-B decoder. |
25 */ | |
26 | |
27 #include "avcodec.h" | |
28 #include "mjpeg.h" | |
29 #include "mjpegdec.h" | |
30 | |
9737
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
31 static uint32_t read_offs(AVCodecContext *avctx, GetBitContext *gb, uint32_t size, const char *err_msg){ |
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
32 uint32_t offs= get_bits_long(gb, 32); |
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
33 if(offs >= size){ |
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
34 av_log(avctx, AV_LOG_WARNING, err_msg, offs, size); |
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
35 return 0; |
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
36 } |
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
37 return offs; |
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
38 } |
5044 | 39 |
40 static int mjpegb_decode_frame(AVCodecContext *avctx, | |
41 void *data, int *data_size, | |
9355
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
8718
diff
changeset
|
42 AVPacket *avpkt) |
5044 | 43 { |
9355
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
8718
diff
changeset
|
44 const uint8_t *buf = avpkt->data; |
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
8718
diff
changeset
|
45 int buf_size = avpkt->size; |
5044 | 46 MJpegDecodeContext *s = avctx->priv_data; |
6271 | 47 const uint8_t *buf_end, *buf_ptr; |
5044 | 48 AVFrame *picture = data; |
49 GetBitContext hgb; /* for the header */ | |
50 uint32_t dqt_offs, dht_offs, sof_offs, sos_offs, second_field_offs; | |
51 uint32_t field_size, sod_offs; | |
52 | |
53 buf_ptr = buf; | |
54 buf_end = buf + buf_size; | |
55 | |
56 read_header: | |
57 /* reset on every SOI */ | |
58 s->restart_interval = 0; | |
59 s->restart_count = 0; | |
60 s->mjpb_skiptosod = 0; | |
61 | |
62 init_get_bits(&hgb, buf_ptr, /*buf_size*/(buf_end - buf_ptr)*8); | |
63 | |
64 skip_bits(&hgb, 32); /* reserved zeros */ | |
65 | |
66 if (get_bits_long(&hgb, 32) != MKBETAG('m','j','p','g')) | |
67 { | |
68 av_log(avctx, AV_LOG_WARNING, "not mjpeg-b (bad fourcc)\n"); | |
69 return 0; | |
70 } | |
71 | |
72 field_size = get_bits_long(&hgb, 32); /* field size */ | |
73 av_log(avctx, AV_LOG_DEBUG, "field size: 0x%x\n", field_size); | |
74 skip_bits(&hgb, 32); /* padded field size */ | |
9737
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
75 second_field_offs = read_offs(avctx, &hgb, buf_end - buf_ptr, "second_field_offs is %d and size is %d\n"); |
5044 | 76 av_log(avctx, AV_LOG_DEBUG, "second field offs: 0x%x\n", second_field_offs); |
77 | |
9737
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
78 dqt_offs = read_offs(avctx, &hgb, buf_end - buf_ptr, "dqt is %d and size is %d\n"); |
5044 | 79 av_log(avctx, AV_LOG_DEBUG, "dqt offs: 0x%x\n", dqt_offs); |
80 if (dqt_offs) | |
81 { | |
5436 | 82 init_get_bits(&s->gb, buf_ptr+dqt_offs, (buf_end - (buf_ptr+dqt_offs))*8); |
5044 | 83 s->start_code = DQT; |
84 ff_mjpeg_decode_dqt(s); | |
85 } | |
86 | |
9737
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
87 dht_offs = read_offs(avctx, &hgb, buf_end - buf_ptr, "dht is %d and size is %d\n"); |
5044 | 88 av_log(avctx, AV_LOG_DEBUG, "dht offs: 0x%x\n", dht_offs); |
89 if (dht_offs) | |
90 { | |
5436 | 91 init_get_bits(&s->gb, buf_ptr+dht_offs, (buf_end - (buf_ptr+dht_offs))*8); |
5044 | 92 s->start_code = DHT; |
93 ff_mjpeg_decode_dht(s); | |
94 } | |
95 | |
9737
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
96 sof_offs = read_offs(avctx, &hgb, buf_end - buf_ptr, "sof is %d and size is %d\n"); |
5044 | 97 av_log(avctx, AV_LOG_DEBUG, "sof offs: 0x%x\n", sof_offs); |
98 if (sof_offs) | |
99 { | |
5436 | 100 init_get_bits(&s->gb, buf_ptr+sof_offs, (buf_end - (buf_ptr+sof_offs))*8); |
5044 | 101 s->start_code = SOF0; |
102 if (ff_mjpeg_decode_sof(s) < 0) | |
103 return -1; | |
104 } | |
105 | |
9737
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
106 sos_offs = read_offs(avctx, &hgb, buf_end - buf_ptr, "sos is %d and size is %d\n"); |
5044 | 107 av_log(avctx, AV_LOG_DEBUG, "sos offs: 0x%x\n", sos_offs); |
9737
f9769330c214
Make sure offsets of mjpeg b are within the buffer.
michael
parents:
9355
diff
changeset
|
108 sod_offs = read_offs(avctx, &hgb, buf_end - buf_ptr, "sof is %d and size is %d\n"); |
5044 | 109 av_log(avctx, AV_LOG_DEBUG, "sod offs: 0x%x\n", sod_offs); |
110 if (sos_offs) | |
111 { | |
112 // init_get_bits(&s->gb, buf+sos_offs, (buf_end - (buf+sos_offs))*8); | |
5436 | 113 init_get_bits(&s->gb, buf_ptr+sos_offs, field_size*8); |
5044 | 114 s->mjpb_skiptosod = (sod_offs - sos_offs - show_bits(&s->gb, 16)); |
115 s->start_code = SOS; | |
116 ff_mjpeg_decode_sos(s); | |
117 } | |
118 | |
119 if (s->interlaced) { | |
120 s->bottom_field ^= 1; | |
121 /* if not bottom field, do not output image yet */ | |
5985 | 122 if (s->bottom_field != s->interlace_polarity && second_field_offs) |
5044 | 123 { |
124 buf_ptr = buf + second_field_offs; | |
125 second_field_offs = 0; | |
126 goto read_header; | |
127 } | |
128 } | |
129 | |
130 //XXX FIXME factorize, this looks very similar to the EOI code | |
131 | |
132 *picture= s->picture; | |
133 *data_size = sizeof(AVFrame); | |
134 | |
135 if(!s->lossless){ | |
6655
22cca5d3173a
Implement FFMAX3(a,b,c) - maximum over three arguments.
voroshil
parents:
6271
diff
changeset
|
136 picture->quality= FFMAX3(s->qscale[0], s->qscale[1], s->qscale[2]); |
5044 | 137 picture->qstride= 0; |
138 picture->qscale_table= s->qscale_table; | |
139 memset(picture->qscale_table, picture->quality, (s->width+15)/16); | |
140 if(avctx->debug & FF_DEBUG_QP) | |
141 av_log(avctx, AV_LOG_DEBUG, "QP: %d\n", picture->quality); | |
142 picture->quality*= FF_QP2LAMBDA; | |
143 } | |
144 | |
145 return buf_ptr - buf; | |
146 } | |
147 | |
148 AVCodec mjpegb_decoder = { | |
149 "mjpegb", | |
150 CODEC_TYPE_VIDEO, | |
151 CODEC_ID_MJPEGB, | |
152 sizeof(MJpegDecodeContext), | |
153 ff_mjpeg_decode_init, | |
154 NULL, | |
155 ff_mjpeg_decode_end, | |
156 mjpegb_decode_frame, | |
157 CODEC_CAP_DR1, | |
6717 | 158 NULL, |
7040
e943e1409077
Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents:
6717
diff
changeset
|
159 .long_name = NULL_IF_CONFIG_SMALL("Apple MJPEG-B"), |
5044 | 160 }; |