Mercurial > libavcodec.hg
annotate 8bps.c @ 4074:861a55e86ef0 libavcodec
factorize update state out
author | michael |
---|---|
date | Tue, 24 Oct 2006 09:34:13 +0000 |
parents | c8c591fe26f8 |
children | f97a2081b5b1 |
rev | line source |
---|---|
1608 | 1 /* |
2 * Quicktime Planar RGB (8BPS) Video Decoder | |
3 * Copyright (C) 2003 Roberto Togni | |
4 * | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
5 * This file is part of FFmpeg. |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
6 * |
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
7 * FFmpeg is free software; you can redistribute it and/or |
1608 | 8 * modify it under the terms of the GNU Lesser General Public |
9 * License as published by the Free Software Foundation; either | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
10 * version 2.1 of the License, or (at your option) any later version. |
1608 | 11 * |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
12 * FFmpeg is distributed in the hope that it will be useful, |
1608 | 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 | |
3947
c8c591fe26f8
Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents:
3036
diff
changeset
|
18 * License along with FFmpeg; if not, write to the Free Software |
3036
0b546eab515d
Update licensing information: The FSF changed postal address.
diego
parents:
2979
diff
changeset
|
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
1608 | 20 * |
21 */ | |
22 | |
23 /** | |
24 * @file 8bps.c | |
25 * QT 8BPS Video Decoder by Roberto Togni <rtogni at bresciaonline dot it> | |
26 * For more information about the 8BPS format, visit: | |
27 * http://www.pcisys.net/~melanson/codecs/ | |
28 * | |
29 * Supports: PAL8 (RGB 8bpp, paletted) | |
30 * : BGR24 (RGB 24bpp) (can also output it as RGBA32) | |
31 * : RGBA32 (RGB 32bpp, 4th plane is probably alpha and it's ignored) | |
32 * | |
33 */ | |
34 | |
35 #include <stdio.h> | |
36 #include <stdlib.h> | |
37 | |
38 #include "common.h" | |
39 #include "avcodec.h" | |
40 | |
41 | |
42 const enum PixelFormat pixfmt_rgb24[] = {PIX_FMT_BGR24, PIX_FMT_RGBA32, -1}; | |
43 | |
44 /* | |
45 * Decoder context | |
46 */ | |
47 typedef struct EightBpsContext { | |
48 | |
2979 | 49 AVCodecContext *avctx; |
50 AVFrame pic; | |
1608 | 51 |
2979 | 52 unsigned char planes; |
53 unsigned char planemap[4]; | |
1608 | 54 } EightBpsContext; |
55 | |
56 | |
57 /* | |
58 * | |
59 * Decode a frame | |
60 * | |
61 */ | |
62 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, uint8_t *buf, int buf_size) | |
63 { | |
2979 | 64 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; |
65 unsigned char *encoded = (unsigned char *)buf; | |
66 unsigned char *pixptr, *pixptr_end; | |
67 unsigned int height = avctx->height; // Real image height | |
68 unsigned int dlen, p, row; | |
69 unsigned char *lp, *dp; | |
70 unsigned char count; | |
71 unsigned int px_inc; | |
72 unsigned int planes = c->planes; | |
73 unsigned char *planemap = c->planemap; | |
2967 | 74 |
2979 | 75 if(c->pic.data[0]) |
76 avctx->release_buffer(avctx, &c->pic); | |
1608 | 77 |
2979 | 78 c->pic.reference = 0; |
79 c->pic.buffer_hints = FF_BUFFER_HINTS_VALID; | |
80 if(avctx->get_buffer(avctx, &c->pic) < 0){ | |
81 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | |
82 return -1; | |
83 } | |
1608 | 84 |
2979 | 85 /* Set data pointer after line lengths */ |
86 dp = encoded + planes * (height << 1); | |
1608 | 87 |
2979 | 88 /* Ignore alpha plane, don't know what to do with it */ |
89 if (planes == 4) | |
90 planes--; | |
1608 | 91 |
2979 | 92 px_inc = planes + (avctx->pix_fmt == PIX_FMT_RGBA32); |
1608 | 93 |
2979 | 94 for (p = 0; p < planes; p++) { |
95 /* Lines length pointer for this plane */ | |
96 lp = encoded + p * (height << 1); | |
1608 | 97 |
2979 | 98 /* Decode a plane */ |
99 for(row = 0; row < height; row++) { | |
100 pixptr = c->pic.data[0] + row * c->pic.linesize[0] + planemap[p]; | |
101 pixptr_end = pixptr + c->pic.linesize[0]; | |
102 dlen = be2me_16(*(unsigned short *)(lp+row*2)); | |
103 /* Decode a row of this plane */ | |
104 while(dlen > 0) { | |
105 if(dp + 1 >= buf+buf_size) return -1; | |
106 if ((count = *dp++) <= 127) { | |
107 count++; | |
108 dlen -= count + 1; | |
109 if (pixptr + count * px_inc > pixptr_end) | |
110 break; | |
111 if(dp + count > buf+buf_size) return -1; | |
112 while(count--) { | |
113 *pixptr = *dp++; | |
114 pixptr += px_inc; | |
115 } | |
116 } else { | |
117 count = 257 - count; | |
118 if (pixptr + count * px_inc > pixptr_end) | |
119 break; | |
120 while(count--) { | |
121 *pixptr = *dp; | |
122 pixptr += px_inc; | |
123 } | |
124 dp++; | |
125 dlen -= 2; | |
126 } | |
127 } | |
128 } | |
129 } | |
1608 | 130 |
2979 | 131 if (avctx->palctrl) { |
132 memcpy (c->pic.data[1], avctx->palctrl->palette, AVPALETTE_SIZE); | |
133 if (avctx->palctrl->palette_changed) { | |
134 c->pic.palette_has_changed = 1; | |
135 avctx->palctrl->palette_changed = 0; | |
136 } else | |
137 c->pic.palette_has_changed = 0; | |
138 } | |
1608 | 139 |
2979 | 140 *data_size = sizeof(AVFrame); |
141 *(AVFrame*)data = c->pic; | |
1608 | 142 |
2979 | 143 /* always report that the buffer was completely consumed */ |
144 return buf_size; | |
1608 | 145 } |
146 | |
147 | |
148 /* | |
149 * | |
150 * Init 8BPS decoder | |
151 * | |
152 */ | |
153 static int decode_init(AVCodecContext *avctx) | |
154 { | |
2979 | 155 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; |
1608 | 156 |
2979 | 157 c->avctx = avctx; |
158 avctx->has_b_frames = 0; | |
1608 | 159 |
2979 | 160 c->pic.data[0] = NULL; |
1608 | 161 |
2429
4b350cc506a7
Use avcodec_check_dimensions instead of custom hack
rtognimp
parents:
2418
diff
changeset
|
162 if (avcodec_check_dimensions(avctx, avctx->width, avctx->height) < 0) { |
2418
82af834636c2
Check pointers before writing to memory, fix possible integer overflows
rtognimp
parents:
1608
diff
changeset
|
163 return 1; |
82af834636c2
Check pointers before writing to memory, fix possible integer overflows
rtognimp
parents:
1608
diff
changeset
|
164 } |
82af834636c2
Check pointers before writing to memory, fix possible integer overflows
rtognimp
parents:
1608
diff
changeset
|
165 |
2979 | 166 switch (avctx->bits_per_sample) { |
167 case 8: | |
168 avctx->pix_fmt = PIX_FMT_PAL8; | |
169 c->planes = 1; | |
170 c->planemap[0] = 0; // 1st plane is palette indexes | |
171 if (avctx->palctrl == NULL) { | |
172 av_log(avctx, AV_LOG_ERROR, "Error: PAL8 format but no palette from demuxer.\n"); | |
173 return -1; | |
174 } | |
175 break; | |
176 case 24: | |
177 avctx->pix_fmt = avctx->get_format(avctx, pixfmt_rgb24); | |
178 c->planes = 3; | |
179 c->planemap[0] = 2; // 1st plane is red | |
180 c->planemap[1] = 1; // 2nd plane is green | |
181 c->planemap[2] = 0; // 3rd plane is blue | |
182 break; | |
183 case 32: | |
184 avctx->pix_fmt = PIX_FMT_RGBA32; | |
185 c->planes = 4; | |
1608 | 186 #ifdef WORDS_BIGENDIAN |
2979 | 187 c->planemap[0] = 1; // 1st plane is red |
188 c->planemap[1] = 2; // 2nd plane is green | |
189 c->planemap[2] = 3; // 3rd plane is blue | |
190 c->planemap[3] = 0; // 4th plane is alpha??? | |
1608 | 191 #else |
2979 | 192 c->planemap[0] = 2; // 1st plane is red |
193 c->planemap[1] = 1; // 2nd plane is green | |
194 c->planemap[2] = 0; // 3rd plane is blue | |
195 c->planemap[3] = 3; // 4th plane is alpha??? | |
1608 | 196 #endif |
2979 | 197 break; |
198 default: | |
199 av_log(avctx, AV_LOG_ERROR, "Error: Unsupported color depth: %u.\n", avctx->bits_per_sample); | |
200 return -1; | |
201 } | |
1608 | 202 |
203 return 0; | |
204 } | |
205 | |
206 | |
207 | |
208 | |
209 /* | |
210 * | |
211 * Uninit 8BPS decoder | |
212 * | |
213 */ | |
214 static int decode_end(AVCodecContext *avctx) | |
215 { | |
2979 | 216 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; |
1608 | 217 |
2979 | 218 if (c->pic.data[0]) |
219 avctx->release_buffer(avctx, &c->pic); | |
1608 | 220 |
2979 | 221 return 0; |
1608 | 222 } |
223 | |
224 | |
225 | |
226 AVCodec eightbps_decoder = { | |
2979 | 227 "8bps", |
228 CODEC_TYPE_VIDEO, | |
229 CODEC_ID_8BPS, | |
230 sizeof(EightBpsContext), | |
231 decode_init, | |
232 NULL, | |
233 decode_end, | |
234 decode_frame, | |
235 CODEC_CAP_DR1, | |
1608 | 236 }; |