Mercurial > libavcodec.hg
comparison sunrast.c @ 6084:d33520d1ca92 libavcodec
Sun Rasterfile decoder
author | ivo |
---|---|
date | Fri, 28 Dec 2007 13:07:43 +0000 |
parents | |
children | e458d5374a03 |
comparison
equal
deleted
inserted
replaced
6083:77d27412c35d | 6084:d33520d1ca92 |
---|---|
1 /* | |
2 * Sun Rasterfile (.sun/.ras/im{1,8,24}/.sunras) image decoder | |
3 * Copyright (c) 2007, 2008 Ivo van Poorten | |
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 #include "avcodec.h" | |
23 | |
24 #define RT_OLD 0 | |
25 #define RT_STANDARD 1 | |
26 #define RT_BYTE_ENCODED 2 | |
27 #define RT_FORMAT_RGB 3 | |
28 #define RT_FORMAT_TIFF 4 | |
29 #define RT_FORMAT_IFF 5 | |
30 | |
31 typedef struct SUNRASTContext { | |
32 AVFrame picture; | |
33 } SUNRASTContext; | |
34 | |
35 static int sunrast_init(AVCodecContext *avctx) { | |
36 SUNRASTContext *s = avctx->priv_data; | |
37 | |
38 avcodec_get_frame_defaults(&s->picture); | |
39 avctx->coded_frame= &s->picture; | |
40 | |
41 return 0; | |
42 } | |
43 | |
44 static int sunrast_decode_frame(AVCodecContext *avctx, void *data, | |
45 int *data_size, uint8_t *buf, int buf_size) { | |
46 SUNRASTContext * const s = avctx->priv_data; | |
47 AVFrame *picture = data; | |
48 AVFrame * const p = &s->picture; | |
49 unsigned int w, h, depth, type, maptype, maplength, stride, x, y, len, alen; | |
50 uint8_t *ptr, *bufstart = buf; | |
51 | |
52 if (AV_RB32(buf) != 0x59a66a95) { | |
53 av_log(avctx, AV_LOG_ERROR, "this is not sunras encoded data\n"); | |
54 return -1; | |
55 } | |
56 | |
57 w = AV_RB32(buf+4); | |
58 h = AV_RB32(buf+8); | |
59 depth = AV_RB32(buf+12); | |
60 type = AV_RB32(buf+20); | |
61 maptype = AV_RB32(buf+24); | |
62 maplength = AV_RB32(buf+28); | |
63 | |
64 if (type > RT_BYTE_ENCODED && type <= RT_FORMAT_IFF) { | |
65 av_log(avctx, AV_LOG_ERROR, "unsupported (compression) type\n"); | |
66 return -1; | |
67 } | |
68 if (type > RT_FORMAT_IFF) { | |
69 av_log(avctx, AV_LOG_ERROR, "invalid (compression) type\n"); | |
70 return -1; | |
71 } | |
72 if (maptype & ~1) { | |
73 av_log(avctx, AV_LOG_ERROR, "invalid colormap type\n"); | |
74 return -1; | |
75 } | |
76 | |
77 buf += 32; | |
78 | |
79 switch (depth) { | |
80 case 1: | |
81 avctx->pix_fmt = PIX_FMT_MONOWHITE; | |
82 break; | |
83 case 8: | |
84 avctx->pix_fmt = PIX_FMT_PAL8; | |
85 break; | |
86 case 24: | |
87 avctx->pix_fmt = PIX_FMT_BGR24; | |
88 break; | |
89 default: | |
90 av_log(avctx, AV_LOG_ERROR, "invalid depth\n"); | |
91 return -1; | |
92 } | |
93 | |
94 if (p->data[0]) | |
95 avctx->release_buffer(avctx, p); | |
96 | |
97 if (avcodec_check_dimensions(avctx, w, h)) | |
98 return -1; | |
99 if (w != avctx->width || h != avctx->height) | |
100 avcodec_set_dimensions(avctx, w, h); | |
101 if (avctx->get_buffer(avctx, p) < 0) { | |
102 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | |
103 return -1; | |
104 } | |
105 | |
106 p->pict_type = FF_I_TYPE; | |
107 | |
108 if (depth != 8 && maplength) { | |
109 av_log(avctx, AV_LOG_WARNING, "useless colormap found or file is corrupted, trying to recover\n"); | |
110 | |
111 } else if (depth == 8) { | |
112 unsigned int len = maplength / 3; | |
113 | |
114 if (!maplength) { | |
115 av_log(avctx, AV_LOG_ERROR, "colormap expected\n"); | |
116 return -1; | |
117 } | |
118 if (maplength % 3 || maplength > 768) { | |
119 av_log(avctx, AV_LOG_WARNING, "invalid colormap length\n"); | |
120 return -1; | |
121 } | |
122 | |
123 av_log(avctx, AV_LOG_DEBUG, "maplength %u\n", len); | |
124 | |
125 ptr = p->data[1]; | |
126 for (x=0; x<len; x++, ptr+=4) | |
127 *(uint32_t *)ptr = (buf[x]<<16) + (buf[len+x]<<8) + buf[len+len+x]; | |
128 } | |
129 | |
130 buf += maplength; | |
131 | |
132 ptr = p->data[0]; | |
133 stride = p->linesize[0]; | |
134 | |
135 /* scanlines are aligned on 16 bit boundaries */ | |
136 len = (depth * w + 7) >> 3; | |
137 alen = len + (len&1); | |
138 | |
139 if (type == RT_BYTE_ENCODED) { | |
140 int value, run; | |
141 uint8_t *end = ptr + h*stride; | |
142 | |
143 x = 0; | |
144 while (ptr != end) { | |
145 run = 1; | |
146 if ((value = *buf++) == 0x80) { | |
147 run = *buf++ + 1; | |
148 if (run != 1) | |
149 value = *buf++; | |
150 } | |
151 while (run--) { | |
152 if (x < len) | |
153 ptr[x] = value; | |
154 if (++x >= alen) { | |
155 x = 0; | |
156 ptr += stride; | |
157 if (ptr == end) | |
158 break; | |
159 } | |
160 } | |
161 } | |
162 } else { | |
163 for (y=0; y<h; y++) { | |
164 memcpy(ptr, buf, len); | |
165 ptr += stride; | |
166 buf += alen; | |
167 } | |
168 } | |
169 | |
170 *picture = s->picture; | |
171 *data_size = sizeof(AVFrame); | |
172 | |
173 return buf - bufstart; | |
174 } | |
175 | |
176 static int sunrast_end(AVCodecContext *avctx) { | |
177 SUNRASTContext *s = avctx->priv_data; | |
178 | |
179 if(s->picture.data[0]) | |
180 avctx->release_buffer(avctx, &s->picture); | |
181 | |
182 return 0; | |
183 } | |
184 | |
185 AVCodec sunrast_decoder = { | |
186 "sunrast", | |
187 CODEC_TYPE_VIDEO, | |
188 CODEC_ID_SUNRAST, | |
189 sizeof(SUNRASTContext), | |
190 sunrast_init, | |
191 NULL, | |
192 sunrast_end, | |
193 sunrast_decode_frame, | |
194 0, | |
195 NULL | |
196 }; |