Mercurial > libavcodec.hg
annotate rawdec.c @ 12507:46191b10d663 libavcodec
rawdec: only allocate a full-frame size buffer if it actually will
be used, place palette buffer in the context to simplify this.
author | reimar |
---|---|
date | Thu, 23 Sep 2010 20:23:15 +0000 |
parents | f6a2af58f3e0 |
children | e6d711ba5760 |
rev | line source |
---|---|
5264 | 1 /* |
2 * Raw Video Decoder | |
8629
04423b2f6e0b
cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents:
8023
diff
changeset
|
3 * Copyright (c) 2001 Fabrice Bellard |
5264 | 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 /** | |
11644
7dd2a45249a9
Remove explicit filename from Doxygen @file commands.
diego
parents:
11560
diff
changeset
|
23 * @file |
5264 | 24 * Raw Video Decoder |
25 */ | |
26 | |
27 #include "avcodec.h" | |
28 #include "raw.h" | |
9335 | 29 #include "libavutil/intreadwrite.h" |
5264 | 30 |
31 typedef struct RawVideoContext { | |
12507
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
32 uint32_t palette[AVPALETTE_COUNT]; |
5264 | 33 unsigned char * buffer; /* block of memory for holding one frame */ |
34 int length; /* number of bytes in buffer */ | |
8993
80720c5427a0
Fix raw rgb/bgr vertical flip in avi based on info from http://www.fourcc.org/fccbihgt.php.
michael
parents:
8718
diff
changeset
|
35 int flip; |
5264 | 36 AVFrame pic; ///< AVCodecContext.coded_frame |
37 } RawVideoContext; | |
38 | |
11753
4c894b716948
Rename pixelFormatBpsAVI to pix_fmt_bps_avi and pixelFormatBpsMOV to
stefano
parents:
11752
diff
changeset
|
39 static const PixelFormatTag pix_fmt_bps_avi[] = { |
5419 | 40 { PIX_FMT_PAL8, 4 }, |
5264 | 41 { PIX_FMT_PAL8, 8 }, |
11839 | 42 { PIX_FMT_RGB444, 12 }, |
5264 | 43 { PIX_FMT_RGB555, 15 }, |
44 { PIX_FMT_RGB555, 16 }, | |
45 { PIX_FMT_BGR24, 24 }, | |
46 { PIX_FMT_RGB32, 32 }, | |
8023
76f6a08c9fe6
Fix some icc warnings by using enum PixelFormat instead of int where appropriate.
cehoyos
parents:
7823
diff
changeset
|
47 { PIX_FMT_NONE, 0 }, |
5264 | 48 }; |
49 | |
11753
4c894b716948
Rename pixelFormatBpsAVI to pix_fmt_bps_avi and pixelFormatBpsMOV to
stefano
parents:
11752
diff
changeset
|
50 static const PixelFormatTag pix_fmt_bps_mov[] = { |
10756
d2c1c2f835b7
Support raw black&white in mov, fixes issue 1654.
cehoyos
parents:
10736
diff
changeset
|
51 { PIX_FMT_MONOWHITE, 1 }, |
10772
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
52 { PIX_FMT_PAL8, 2 }, |
6423 | 53 { PIX_FMT_PAL8, 4 }, |
5264 | 54 { PIX_FMT_PAL8, 8 }, |
9620 | 55 // FIXME swscale does not support 16 bit in .mov, sample 16bit.mov |
56 // http://developer.apple.com/documentation/QuickTime/QTFF/QTFFChap3/qtff3.html | |
10757
9de8bd6b4ca4
Fix colourspace for 16 bit raw video in mov, fixes issue 1527 on big-endian.
cehoyos
parents:
10756
diff
changeset
|
57 { PIX_FMT_RGB555BE, 16 }, |
5264 | 58 { PIX_FMT_RGB24, 24 }, |
10736
8a9b0b693a30
Use correct pixfmt for 32 bit raw in mov also on big-endian, fixes issue 1630.
cehoyos
parents:
10390
diff
changeset
|
59 { PIX_FMT_ARGB, 32 }, |
8023
76f6a08c9fe6
Fix some icc warnings by using enum PixelFormat instead of int where appropriate.
cehoyos
parents:
7823
diff
changeset
|
60 { PIX_FMT_NONE, 0 }, |
5264 | 61 }; |
62 | |
11752 | 63 static enum PixelFormat find_pix_fmt(const PixelFormatTag *tags, unsigned int fourcc) |
5264 | 64 { |
65 while (tags->pix_fmt >= 0) { | |
66 if (tags->fourcc == fourcc) | |
67 return tags->pix_fmt; | |
68 tags++; | |
69 } | |
70 return PIX_FMT_YUV420P; | |
71 } | |
72 | |
6517
48759bfbd073
Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents:
6424
diff
changeset
|
73 static av_cold int raw_init_decoder(AVCodecContext *avctx) |
5264 | 74 { |
75 RawVideoContext *context = avctx->priv_data; | |
76 | |
77 if (avctx->codec_tag == MKTAG('r','a','w',' ')) | |
11753
4c894b716948
Rename pixelFormatBpsAVI to pix_fmt_bps_avi and pixelFormatBpsMOV to
stefano
parents:
11752
diff
changeset
|
78 avctx->pix_fmt = find_pix_fmt(pix_fmt_bps_mov, avctx->bits_per_coded_sample); |
5264 | 79 else if (avctx->codec_tag) |
11752 | 80 avctx->pix_fmt = find_pix_fmt(ff_raw_pix_fmt_tags, avctx->codec_tag); |
11861
f6a2af58f3e0
avcodec.h: Clarify AVCodecContext::pix_fmt documentation since its value may be set by the demuxer
thardin
parents:
11839
diff
changeset
|
81 else if (avctx->pix_fmt == PIX_FMT_NONE && avctx->bits_per_coded_sample) |
11753
4c894b716948
Rename pixelFormatBpsAVI to pix_fmt_bps_avi and pixelFormatBpsMOV to
stefano
parents:
11752
diff
changeset
|
82 avctx->pix_fmt = find_pix_fmt(pix_fmt_bps_avi, avctx->bits_per_coded_sample); |
5264 | 83 |
84 context->length = avpicture_get_size(avctx->pix_fmt, avctx->width, avctx->height); | |
12507
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
85 if((avctx->bits_per_coded_sample == 4 || avctx->bits_per_coded_sample == 2) && |
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
86 avctx->pix_fmt==PIX_FMT_PAL8 && |
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
87 (!avctx->codec_tag || avctx->codec_tag == MKTAG('r','a','w',' '))){ |
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
88 context->buffer = av_malloc(context->length); |
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
89 if (!context->buffer) |
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
90 return -1; |
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
91 } |
5264 | 92 context->pic.pict_type = FF_I_TYPE; |
93 context->pic.key_frame = 1; | |
94 | |
95 avctx->coded_frame= &context->pic; | |
96 | |
9845
45d331133468
Add new FOURCC (0x0003) for raw DIB video, set flip flag accordingly.
darkshikari
parents:
9712
diff
changeset
|
97 if((avctx->extradata_size >= 9 && !memcmp(avctx->extradata + avctx->extradata_size - 9, "BottomUp", 9)) || |
45d331133468
Add new FOURCC (0x0003) for raw DIB video, set flip flag accordingly.
darkshikari
parents:
9712
diff
changeset
|
98 avctx->codec_tag == MKTAG( 3 , 0 , 0 , 0 )) |
8993
80720c5427a0
Fix raw rgb/bgr vertical flip in avi based on info from http://www.fourcc.org/fccbihgt.php.
michael
parents:
8718
diff
changeset
|
99 context->flip=1; |
80720c5427a0
Fix raw rgb/bgr vertical flip in avi based on info from http://www.fourcc.org/fccbihgt.php.
michael
parents:
8718
diff
changeset
|
100 |
5264 | 101 return 0; |
102 } | |
103 | |
104 static void flip(AVCodecContext *avctx, AVPicture * picture){ | |
8995 | 105 picture->data[0] += picture->linesize[0] * (avctx->height-1); |
106 picture->linesize[0] *= -1; | |
5264 | 107 } |
108 | |
109 static int raw_decode(AVCodecContext *avctx, | |
110 void *data, int *data_size, | |
9355
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
9335
diff
changeset
|
111 AVPacket *avpkt) |
5264 | 112 { |
9355
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
9335
diff
changeset
|
113 const uint8_t *buf = avpkt->data; |
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
9335
diff
changeset
|
114 int buf_size = avpkt->size; |
5264 | 115 RawVideoContext *context = avctx->priv_data; |
116 | |
117 AVFrame * frame = (AVFrame *) data; | |
118 AVPicture * picture = (AVPicture *) data; | |
119 | |
120 frame->interlaced_frame = avctx->coded_frame->interlaced_frame; | |
121 frame->top_field_first = avctx->coded_frame->top_field_first; | |
122 | |
10773
8413193405c5
Cosmetics: Fix indentation and comment after r21027.
cehoyos
parents:
10772
diff
changeset
|
123 //2bpp and 4bpp raw in avi and mov (yes this is ugly ...) |
12507
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
124 if (context->buffer) { |
5419 | 125 int i; |
12507
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
126 uint8_t *dst = context->buffer; |
10771
cea7769a2b62
Improve readability of 4bpp raw decoder and prepare for supporting 2bpp.
cehoyos
parents:
10757
diff
changeset
|
127 buf_size = context->length - 256*4; |
10772
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
128 if (avctx->bits_per_coded_sample == 4){ |
10773
8413193405c5
Cosmetics: Fix indentation and comment after r21027.
cehoyos
parents:
10772
diff
changeset
|
129 for(i=0; 2*i+1 < buf_size; i++){ |
8413193405c5
Cosmetics: Fix indentation and comment after r21027.
cehoyos
parents:
10772
diff
changeset
|
130 dst[2*i+0]= buf[i]>>4; |
8413193405c5
Cosmetics: Fix indentation and comment after r21027.
cehoyos
parents:
10772
diff
changeset
|
131 dst[2*i+1]= buf[i]&15; |
8413193405c5
Cosmetics: Fix indentation and comment after r21027.
cehoyos
parents:
10772
diff
changeset
|
132 } |
10772
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
133 } else |
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
134 for(i=0; 4*i+3 < buf_size; i++){ |
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
135 dst[4*i+0]= buf[i]>>6; |
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
136 dst[4*i+1]= buf[i]>>4&3; |
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
137 dst[4*i+2]= buf[i]>>2&3; |
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
138 dst[4*i+3]= buf[i] &3; |
c76ee970d3bd
Support decoding raw 2bpp in mov, fixes issue 1528.
cehoyos
parents:
10771
diff
changeset
|
139 } |
10771
cea7769a2b62
Improve readability of 4bpp raw decoder and prepare for supporting 2bpp.
cehoyos
parents:
10757
diff
changeset
|
140 buf= dst; |
5419 | 141 } |
142 | |
11072
0f3a09795840
Fix r21223: AVup samples (issue 1685) need a buf offset like AV1x (issue 1684).
cehoyos
parents:
10895
diff
changeset
|
143 if(avctx->codec_tag == MKTAG('A', 'V', '1', 'x') || |
0f3a09795840
Fix r21223: AVup samples (issue 1685) need a buf offset like AV1x (issue 1684).
cehoyos
parents:
10895
diff
changeset
|
144 avctx->codec_tag == MKTAG('A', 'V', 'u', 'p')) |
10895
1cebb3ea364b
Correctly decode some Avid AV1x samples, fixes issue 1684.
cehoyos
parents:
10880
diff
changeset
|
145 buf += buf_size - context->length; |
1cebb3ea364b
Correctly decode some Avid AV1x samples, fixes issue 1684.
cehoyos
parents:
10880
diff
changeset
|
146 |
5264 | 147 if(buf_size < context->length - (avctx->pix_fmt==PIX_FMT_PAL8 ? 256*4 : 0)) |
148 return -1; | |
149 | |
150 avpicture_fill(picture, buf, avctx->pix_fmt, avctx->width, avctx->height); | |
151 if(avctx->pix_fmt==PIX_FMT_PAL8 && buf_size < context->length){ | |
12507
46191b10d663
rawdec: only allocate a full-frame size buffer if it actually will
reimar
parents:
11861
diff
changeset
|
152 frame->data[1]= context->palette; |
5264 | 153 } |
154 if (avctx->palctrl && avctx->palctrl->palette_changed) { | |
155 memcpy(frame->data[1], avctx->palctrl->palette, AVPALETTE_SIZE); | |
156 avctx->palctrl->palette_changed = 0; | |
157 } | |
11669 | 158 if(avctx->pix_fmt==PIX_FMT_BGR24 && ((frame->linesize[0]+3)&~3)*avctx->height <= buf_size) |
159 frame->linesize[0] = (frame->linesize[0]+3)&~3; | |
5264 | 160 |
8993
80720c5427a0
Fix raw rgb/bgr vertical flip in avi based on info from http://www.fourcc.org/fccbihgt.php.
michael
parents:
8718
diff
changeset
|
161 if(context->flip) |
8995 | 162 flip(avctx, picture); |
5264 | 163 |
9712 | 164 if ( avctx->codec_tag == MKTAG('Y', 'V', '1', '2') |
165 || avctx->codec_tag == MKTAG('Y', 'V', 'U', '9')) | |
10390 | 166 FFSWAP(uint8_t *, picture->data[1], picture->data[2]); |
5264 | 167 |
9335 | 168 if(avctx->codec_tag == AV_RL32("yuv2") && |
169 avctx->pix_fmt == PIX_FMT_YUYV422) { | |
170 int x, y; | |
171 uint8_t *line = picture->data[0]; | |
172 for(y = 0; y < avctx->height; y++) { | |
173 for(x = 0; x < avctx->width; x++) | |
174 line[2*x + 1] ^= 0x80; | |
175 line += picture->linesize[0]; | |
176 } | |
177 } | |
178 | |
5264 | 179 *data_size = sizeof(AVPicture); |
180 return buf_size; | |
181 } | |
182 | |
6517
48759bfbd073
Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents:
6424
diff
changeset
|
183 static av_cold int raw_close_decoder(AVCodecContext *avctx) |
5264 | 184 { |
185 RawVideoContext *context = avctx->priv_data; | |
186 | |
187 av_freep(&context->buffer); | |
188 return 0; | |
189 } | |
190 | |
191 AVCodec rawvideo_decoder = { | |
192 "rawvideo", | |
11560
8a4984c5cacc
Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents:
11072
diff
changeset
|
193 AVMEDIA_TYPE_VIDEO, |
5264 | 194 CODEC_ID_RAWVIDEO, |
195 sizeof(RawVideoContext), | |
196 raw_init_decoder, | |
197 NULL, | |
198 raw_close_decoder, | |
199 raw_decode, | |
7040
e943e1409077
Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents:
6712
diff
changeset
|
200 .long_name = NULL_IF_CONFIG_SMALL("raw video"), |
5264 | 201 }; |