Mercurial > libavcodec.hg
comparison 8bps.c @ 2979:bfabfdf9ce55 libavcodec
COSMETICS: tabs --> spaces, some prettyprinting
| author | diego |
|---|---|
| date | Thu, 22 Dec 2005 01:10:11 +0000 |
| parents | ef2149182f1c |
| children | 0b546eab515d |
comparison
equal
deleted
inserted
replaced
| 2978:403183bbb505 | 2979:bfabfdf9ce55 |
|---|---|
| 42 /* | 42 /* |
| 43 * Decoder context | 43 * Decoder context |
| 44 */ | 44 */ |
| 45 typedef struct EightBpsContext { | 45 typedef struct EightBpsContext { |
| 46 | 46 |
| 47 AVCodecContext *avctx; | 47 AVCodecContext *avctx; |
| 48 AVFrame pic; | 48 AVFrame pic; |
| 49 | 49 |
| 50 unsigned char planes; | 50 unsigned char planes; |
| 51 unsigned char planemap[4]; | 51 unsigned char planemap[4]; |
| 52 } EightBpsContext; | 52 } EightBpsContext; |
| 53 | 53 |
| 54 | 54 |
| 55 /* | 55 /* |
| 56 * | 56 * |
| 57 * Decode a frame | 57 * Decode a frame |
| 58 * | 58 * |
| 59 */ | 59 */ |
| 60 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, uint8_t *buf, int buf_size) | 60 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, uint8_t *buf, int buf_size) |
| 61 { | 61 { |
| 62 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; | 62 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; |
| 63 unsigned char *encoded = (unsigned char *)buf; | 63 unsigned char *encoded = (unsigned char *)buf; |
| 64 unsigned char *pixptr, *pixptr_end; | 64 unsigned char *pixptr, *pixptr_end; |
| 65 unsigned int height = avctx->height; // Real image height | 65 unsigned int height = avctx->height; // Real image height |
| 66 unsigned int dlen, p, row; | 66 unsigned int dlen, p, row; |
| 67 unsigned char *lp, *dp; | 67 unsigned char *lp, *dp; |
| 68 unsigned char count; | 68 unsigned char count; |
| 69 unsigned int px_inc; | 69 unsigned int px_inc; |
| 70 unsigned int planes = c->planes; | 70 unsigned int planes = c->planes; |
| 71 unsigned char *planemap = c->planemap; | 71 unsigned char *planemap = c->planemap; |
| 72 | 72 |
| 73 if(c->pic.data[0]) | 73 if(c->pic.data[0]) |
| 74 avctx->release_buffer(avctx, &c->pic); | 74 avctx->release_buffer(avctx, &c->pic); |
| 75 | 75 |
| 76 c->pic.reference = 0; | 76 c->pic.reference = 0; |
| 77 c->pic.buffer_hints = FF_BUFFER_HINTS_VALID; | 77 c->pic.buffer_hints = FF_BUFFER_HINTS_VALID; |
| 78 if(avctx->get_buffer(avctx, &c->pic) < 0){ | 78 if(avctx->get_buffer(avctx, &c->pic) < 0){ |
| 79 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | 79 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); |
| 80 return -1; | 80 return -1; |
| 81 } | 81 } |
| 82 | 82 |
| 83 /* Set data pointer after line lengths */ | 83 /* Set data pointer after line lengths */ |
| 84 dp = encoded + planes * (height << 1); | 84 dp = encoded + planes * (height << 1); |
| 85 | 85 |
| 86 /* Ignore alpha plane, don't know what to do with it */ | 86 /* Ignore alpha plane, don't know what to do with it */ |
| 87 if (planes == 4) | 87 if (planes == 4) |
| 88 planes--; | 88 planes--; |
| 89 | 89 |
| 90 px_inc = planes + (avctx->pix_fmt == PIX_FMT_RGBA32); | 90 px_inc = planes + (avctx->pix_fmt == PIX_FMT_RGBA32); |
| 91 | 91 |
| 92 for (p = 0; p < planes; p++) { | 92 for (p = 0; p < planes; p++) { |
| 93 /* Lines length pointer for this plane */ | 93 /* Lines length pointer for this plane */ |
| 94 lp = encoded + p * (height << 1); | 94 lp = encoded + p * (height << 1); |
| 95 | 95 |
| 96 /* Decode a plane */ | 96 /* Decode a plane */ |
| 97 for(row = 0; row < height; row++) { | 97 for(row = 0; row < height; row++) { |
| 98 pixptr = c->pic.data[0] + row * c->pic.linesize[0] + planemap[p]; | 98 pixptr = c->pic.data[0] + row * c->pic.linesize[0] + planemap[p]; |
| 99 pixptr_end = pixptr + c->pic.linesize[0]; | 99 pixptr_end = pixptr + c->pic.linesize[0]; |
| 100 dlen = be2me_16(*(unsigned short *)(lp+row*2)); | 100 dlen = be2me_16(*(unsigned short *)(lp+row*2)); |
| 101 /* Decode a row of this plane */ | 101 /* Decode a row of this plane */ |
| 102 while(dlen > 0) { | 102 while(dlen > 0) { |
| 103 if(dp + 1 >= buf+buf_size) return -1; | 103 if(dp + 1 >= buf+buf_size) return -1; |
| 104 if ((count = *dp++) <= 127) { | 104 if ((count = *dp++) <= 127) { |
| 105 count++; | 105 count++; |
| 106 dlen -= count + 1; | 106 dlen -= count + 1; |
| 107 if (pixptr + count * px_inc > pixptr_end) | 107 if (pixptr + count * px_inc > pixptr_end) |
| 108 break; | 108 break; |
| 109 if(dp + count > buf+buf_size) return -1; | 109 if(dp + count > buf+buf_size) return -1; |
| 110 while(count--) { | 110 while(count--) { |
| 111 *pixptr = *dp++; | 111 *pixptr = *dp++; |
| 112 pixptr += px_inc; | 112 pixptr += px_inc; |
| 113 } | 113 } |
| 114 } else { | 114 } else { |
| 115 count = 257 - count; | 115 count = 257 - count; |
| 116 if (pixptr + count * px_inc > pixptr_end) | 116 if (pixptr + count * px_inc > pixptr_end) |
| 117 break; | 117 break; |
| 118 while(count--) { | 118 while(count--) { |
| 119 *pixptr = *dp; | 119 *pixptr = *dp; |
| 120 pixptr += px_inc; | 120 pixptr += px_inc; |
| 121 } | 121 } |
| 122 dp++; | 122 dp++; |
| 123 dlen -= 2; | 123 dlen -= 2; |
| 124 } | 124 } |
| 125 } | 125 } |
| 126 } | 126 } |
| 127 } | 127 } |
| 128 | 128 |
| 129 if (avctx->palctrl) { | 129 if (avctx->palctrl) { |
| 130 memcpy (c->pic.data[1], avctx->palctrl->palette, AVPALETTE_SIZE); | 130 memcpy (c->pic.data[1], avctx->palctrl->palette, AVPALETTE_SIZE); |
| 131 if (avctx->palctrl->palette_changed) { | 131 if (avctx->palctrl->palette_changed) { |
| 132 c->pic.palette_has_changed = 1; | 132 c->pic.palette_has_changed = 1; |
| 133 avctx->palctrl->palette_changed = 0; | 133 avctx->palctrl->palette_changed = 0; |
| 134 } else | 134 } else |
| 135 c->pic.palette_has_changed = 0; | 135 c->pic.palette_has_changed = 0; |
| 136 } | 136 } |
| 137 | 137 |
| 138 *data_size = sizeof(AVFrame); | 138 *data_size = sizeof(AVFrame); |
| 139 *(AVFrame*)data = c->pic; | 139 *(AVFrame*)data = c->pic; |
| 140 | 140 |
| 141 /* always report that the buffer was completely consumed */ | 141 /* always report that the buffer was completely consumed */ |
| 142 return buf_size; | 142 return buf_size; |
| 143 } | 143 } |
| 144 | 144 |
| 145 | 145 |
| 146 /* | 146 /* |
| 147 * | 147 * |
| 148 * Init 8BPS decoder | 148 * Init 8BPS decoder |
| 149 * | 149 * |
| 150 */ | 150 */ |
| 151 static int decode_init(AVCodecContext *avctx) | 151 static int decode_init(AVCodecContext *avctx) |
| 152 { | 152 { |
| 153 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; | 153 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; |
| 154 | 154 |
| 155 c->avctx = avctx; | 155 c->avctx = avctx; |
| 156 avctx->has_b_frames = 0; | 156 avctx->has_b_frames = 0; |
| 157 | 157 |
| 158 c->pic.data[0] = NULL; | 158 c->pic.data[0] = NULL; |
| 159 | 159 |
| 160 if (avcodec_check_dimensions(avctx, avctx->width, avctx->height) < 0) { | 160 if (avcodec_check_dimensions(avctx, avctx->width, avctx->height) < 0) { |
| 161 return 1; | 161 return 1; |
| 162 } | 162 } |
| 163 | 163 |
| 164 switch (avctx->bits_per_sample) { | 164 switch (avctx->bits_per_sample) { |
| 165 case 8: | 165 case 8: |
| 166 avctx->pix_fmt = PIX_FMT_PAL8; | 166 avctx->pix_fmt = PIX_FMT_PAL8; |
| 167 c->planes = 1; | 167 c->planes = 1; |
| 168 c->planemap[0] = 0; // 1st plane is palette indexes | 168 c->planemap[0] = 0; // 1st plane is palette indexes |
| 169 if (avctx->palctrl == NULL) { | 169 if (avctx->palctrl == NULL) { |
| 170 av_log(avctx, AV_LOG_ERROR, "Error: PAL8 format but no palette from demuxer.\n"); | 170 av_log(avctx, AV_LOG_ERROR, "Error: PAL8 format but no palette from demuxer.\n"); |
| 171 return -1; | 171 return -1; |
| 172 } | 172 } |
| 173 break; | 173 break; |
| 174 case 24: | 174 case 24: |
| 175 avctx->pix_fmt = avctx->get_format(avctx, pixfmt_rgb24); | 175 avctx->pix_fmt = avctx->get_format(avctx, pixfmt_rgb24); |
| 176 c->planes = 3; | 176 c->planes = 3; |
| 177 c->planemap[0] = 2; // 1st plane is red | 177 c->planemap[0] = 2; // 1st plane is red |
| 178 c->planemap[1] = 1; // 2nd plane is green | 178 c->planemap[1] = 1; // 2nd plane is green |
| 179 c->planemap[2] = 0; // 3rd plane is blue | 179 c->planemap[2] = 0; // 3rd plane is blue |
| 180 break; | 180 break; |
| 181 case 32: | 181 case 32: |
| 182 avctx->pix_fmt = PIX_FMT_RGBA32; | 182 avctx->pix_fmt = PIX_FMT_RGBA32; |
| 183 c->planes = 4; | 183 c->planes = 4; |
| 184 #ifdef WORDS_BIGENDIAN | 184 #ifdef WORDS_BIGENDIAN |
| 185 c->planemap[0] = 1; // 1st plane is red | 185 c->planemap[0] = 1; // 1st plane is red |
| 186 c->planemap[1] = 2; // 2nd plane is green | 186 c->planemap[1] = 2; // 2nd plane is green |
| 187 c->planemap[2] = 3; // 3rd plane is blue | 187 c->planemap[2] = 3; // 3rd plane is blue |
| 188 c->planemap[3] = 0; // 4th plane is alpha??? | 188 c->planemap[3] = 0; // 4th plane is alpha??? |
| 189 #else | 189 #else |
| 190 c->planemap[0] = 2; // 1st plane is red | 190 c->planemap[0] = 2; // 1st plane is red |
| 191 c->planemap[1] = 1; // 2nd plane is green | 191 c->planemap[1] = 1; // 2nd plane is green |
| 192 c->planemap[2] = 0; // 3rd plane is blue | 192 c->planemap[2] = 0; // 3rd plane is blue |
| 193 c->planemap[3] = 3; // 4th plane is alpha??? | 193 c->planemap[3] = 3; // 4th plane is alpha??? |
| 194 #endif | 194 #endif |
| 195 break; | 195 break; |
| 196 default: | 196 default: |
| 197 av_log(avctx, AV_LOG_ERROR, "Error: Unsupported color depth: %u.\n", avctx->bits_per_sample); | 197 av_log(avctx, AV_LOG_ERROR, "Error: Unsupported color depth: %u.\n", avctx->bits_per_sample); |
| 198 return -1; | 198 return -1; |
| 199 } | 199 } |
| 200 | 200 |
| 201 return 0; | 201 return 0; |
| 202 } | 202 } |
| 203 | 203 |
| 204 | 204 |
| 209 * Uninit 8BPS decoder | 209 * Uninit 8BPS decoder |
| 210 * | 210 * |
| 211 */ | 211 */ |
| 212 static int decode_end(AVCodecContext *avctx) | 212 static int decode_end(AVCodecContext *avctx) |
| 213 { | 213 { |
| 214 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; | 214 EightBpsContext * const c = (EightBpsContext *)avctx->priv_data; |
| 215 | 215 |
| 216 if (c->pic.data[0]) | 216 if (c->pic.data[0]) |
| 217 avctx->release_buffer(avctx, &c->pic); | 217 avctx->release_buffer(avctx, &c->pic); |
| 218 | 218 |
| 219 return 0; | 219 return 0; |
| 220 } | 220 } |
| 221 | 221 |
| 222 | 222 |
| 223 | 223 |
| 224 AVCodec eightbps_decoder = { | 224 AVCodec eightbps_decoder = { |
| 225 "8bps", | 225 "8bps", |
| 226 CODEC_TYPE_VIDEO, | 226 CODEC_TYPE_VIDEO, |
| 227 CODEC_ID_8BPS, | 227 CODEC_ID_8BPS, |
| 228 sizeof(EightBpsContext), | 228 sizeof(EightBpsContext), |
| 229 decode_init, | 229 decode_init, |
| 230 NULL, | 230 NULL, |
| 231 decode_end, | 231 decode_end, |
| 232 decode_frame, | 232 decode_frame, |
| 233 CODEC_CAP_DR1, | 233 CODEC_CAP_DR1, |
| 234 }; | 234 }; |
