1 |
|
|
/* |
2 |
|
|
* SGI image encoder |
3 |
|
|
* Todd Kirby <doubleshot@pacbell.net> |
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 "libavutil/opt.h" |
23 |
|
|
|
24 |
|
|
#include "avcodec.h" |
25 |
|
|
#include "bytestream.h" |
26 |
|
|
#include "internal.h" |
27 |
|
|
#include "sgi.h" |
28 |
|
|
#include "rle.h" |
29 |
|
|
|
30 |
|
|
#define SGI_SINGLE_CHAN 2 |
31 |
|
|
#define SGI_MULTI_CHAN 3 |
32 |
|
|
|
33 |
|
|
typedef struct SgiContext { |
34 |
|
|
AVClass *class; |
35 |
|
|
|
36 |
|
|
int rle; |
37 |
|
|
} SgiContext; |
38 |
|
|
|
39 |
|
1 |
static av_cold int encode_init(AVCodecContext *avctx) |
40 |
|
|
{ |
41 |
✓✗✗✓
|
1 |
if (avctx->width > 65535 || avctx->height > 65535) { |
42 |
|
|
av_log(avctx, AV_LOG_ERROR, "Unsupported resolution %dx%d. " |
43 |
|
|
"SGI does not support resolutions above 65535x65535\n", |
44 |
|
|
avctx->width, avctx->height); |
45 |
|
|
return AVERROR_INVALIDDATA; |
46 |
|
|
} |
47 |
|
|
|
48 |
|
1 |
return 0; |
49 |
|
|
} |
50 |
|
|
|
51 |
|
11232 |
static int sgi_rle_encode(PutByteContext *pbc, const uint8_t *src, |
52 |
|
|
int w, int bpp) |
53 |
|
|
{ |
54 |
|
11232 |
int val, count, x, start = bytestream2_tell_p(pbc); |
55 |
|
|
void (*bytestream2_put)(PutByteContext *, unsigned int); |
56 |
|
|
|
57 |
✓✗ |
11232 |
if (bpp == 1) |
58 |
|
11232 |
bytestream2_put = bytestream2_put_byte; |
59 |
|
|
else |
60 |
|
|
bytestream2_put = bytestream2_put_be16; |
61 |
|
|
|
62 |
✓✓ |
97313 |
for (x = 0; x < w; x += count) { |
63 |
|
|
/* see if we can encode the next set of pixels with RLE */ |
64 |
|
86081 |
count = ff_rle_count_pixels(src, w - x, bpp, 1); |
65 |
✓✓ |
86081 |
if (count > 1) { |
66 |
✗✓ |
34170 |
if (bytestream2_get_bytes_left_p(pbc) < bpp * 2) |
67 |
|
|
return AVERROR_INVALIDDATA; |
68 |
|
|
|
69 |
✓✗ |
34170 |
val = bpp == 1 ? *src : AV_RB16(src); |
70 |
|
34170 |
bytestream2_put(pbc, count); |
71 |
|
34170 |
bytestream2_put(pbc, val); |
72 |
|
|
} else { |
73 |
|
|
int i; |
74 |
|
|
/* fall back on uncompressed */ |
75 |
|
51911 |
count = ff_rle_count_pixels(src, w - x, bpp, 0); |
76 |
✗✓ |
51911 |
if (bytestream2_get_bytes_left_p(pbc) < bpp * (count + 1)) |
77 |
|
|
return AVERROR_INVALIDDATA; |
78 |
|
|
|
79 |
|
51911 |
bytestream2_put(pbc, count + 0x80); |
80 |
✓✓ |
3834780 |
for (i = 0; i < count; i++) { |
81 |
✓✗ |
3782869 |
val = bpp == 1 ? src[i] : AV_RB16(src + i * bpp); |
82 |
|
3782869 |
bytestream2_put(pbc, val); |
83 |
|
|
} |
84 |
|
|
} |
85 |
|
|
|
86 |
|
86081 |
src += count * bpp; |
87 |
|
|
} |
88 |
|
|
|
89 |
|
11232 |
return bytestream2_tell_p(pbc) - start; |
90 |
|
|
} |
91 |
|
|
|
92 |
|
13 |
static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, |
93 |
|
|
const AVFrame *frame, int *got_packet) |
94 |
|
|
{ |
95 |
|
13 |
SgiContext *s = avctx->priv_data; |
96 |
|
13 |
const AVFrame * const p = frame; |
97 |
|
|
PutByteContext pbc; |
98 |
|
|
uint8_t *in_buf, *encode_buf; |
99 |
|
|
int x, y, z, length, tablesize, ret, i; |
100 |
|
|
unsigned int width, height, depth, dimension; |
101 |
|
|
unsigned int bytes_per_channel, pixmax, put_be; |
102 |
|
|
|
103 |
|
|
#if FF_API_CODED_FRAME |
104 |
|
|
FF_DISABLE_DEPRECATION_WARNINGS |
105 |
|
13 |
avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I; |
106 |
|
13 |
avctx->coded_frame->key_frame = 1; |
107 |
|
|
FF_ENABLE_DEPRECATION_WARNINGS |
108 |
|
|
#endif |
109 |
|
|
|
110 |
|
|
#if FF_API_CODER_TYPE |
111 |
|
|
FF_DISABLE_DEPRECATION_WARNINGS |
112 |
✗✓ |
13 |
if (avctx->coder_type == FF_CODER_TYPE_RAW) |
113 |
|
|
s->rle = 0; |
114 |
|
|
FF_ENABLE_DEPRECATION_WARNINGS |
115 |
|
|
#endif |
116 |
|
|
|
117 |
|
13 |
width = avctx->width; |
118 |
|
13 |
height = avctx->height; |
119 |
|
13 |
bytes_per_channel = 1; |
120 |
|
13 |
pixmax = 0xFF; |
121 |
|
13 |
put_be = HAVE_BIGENDIAN; |
122 |
|
|
|
123 |
✗✓✗✗ ✗✗✗✗ ✗✗ |
13 |
switch (avctx->pix_fmt) { |
124 |
|
|
case AV_PIX_FMT_GRAY8: |
125 |
|
|
dimension = SGI_SINGLE_CHAN; |
126 |
|
|
depth = SGI_GRAYSCALE; |
127 |
|
|
break; |
128 |
|
13 |
case AV_PIX_FMT_RGB24: |
129 |
|
13 |
dimension = SGI_MULTI_CHAN; |
130 |
|
13 |
depth = SGI_RGB; |
131 |
|
13 |
break; |
132 |
|
|
case AV_PIX_FMT_RGBA: |
133 |
|
|
dimension = SGI_MULTI_CHAN; |
134 |
|
|
depth = SGI_RGBA; |
135 |
|
|
break; |
136 |
|
|
case AV_PIX_FMT_GRAY16LE: |
137 |
|
|
put_be = !HAVE_BIGENDIAN; |
138 |
|
|
case AV_PIX_FMT_GRAY16BE: |
139 |
|
|
bytes_per_channel = 2; |
140 |
|
|
pixmax = 0xFFFF; |
141 |
|
|
dimension = SGI_SINGLE_CHAN; |
142 |
|
|
depth = SGI_GRAYSCALE; |
143 |
|
|
break; |
144 |
|
|
case AV_PIX_FMT_RGB48LE: |
145 |
|
|
put_be = !HAVE_BIGENDIAN; |
146 |
|
|
case AV_PIX_FMT_RGB48BE: |
147 |
|
|
bytes_per_channel = 2; |
148 |
|
|
pixmax = 0xFFFF; |
149 |
|
|
dimension = SGI_MULTI_CHAN; |
150 |
|
|
depth = SGI_RGB; |
151 |
|
|
break; |
152 |
|
|
case AV_PIX_FMT_RGBA64LE: |
153 |
|
|
put_be = !HAVE_BIGENDIAN; |
154 |
|
|
case AV_PIX_FMT_RGBA64BE: |
155 |
|
|
bytes_per_channel = 2; |
156 |
|
|
pixmax = 0xFFFF; |
157 |
|
|
dimension = SGI_MULTI_CHAN; |
158 |
|
|
depth = SGI_RGBA; |
159 |
|
|
break; |
160 |
|
|
default: |
161 |
|
|
return AVERROR_INVALIDDATA; |
162 |
|
|
} |
163 |
|
|
|
164 |
|
13 |
tablesize = depth * height * 4; |
165 |
|
13 |
length = SGI_HEADER_SIZE; |
166 |
✗✓ |
13 |
if (!s->rle) |
167 |
|
|
length += depth * height * width; |
168 |
|
|
else // assume sgi_rle_encode() produces at most 2x size of input |
169 |
|
13 |
length += tablesize * 2 + depth * height * (2 * width + 1); |
170 |
|
|
|
171 |
✗✓ |
13 |
if ((ret = ff_alloc_packet2(avctx, pkt, bytes_per_channel * length, 0)) < 0) |
172 |
|
|
return ret; |
173 |
|
|
|
174 |
|
13 |
bytestream2_init_writer(&pbc, pkt->data, pkt->size); |
175 |
|
|
|
176 |
|
|
/* Encode header. */ |
177 |
|
13 |
bytestream2_put_be16(&pbc, SGI_MAGIC); |
178 |
|
13 |
bytestream2_put_byte(&pbc, s->rle); /* RLE 1 - VERBATIM 0 */ |
179 |
|
13 |
bytestream2_put_byte(&pbc, bytes_per_channel); |
180 |
|
13 |
bytestream2_put_be16(&pbc, dimension); |
181 |
|
13 |
bytestream2_put_be16(&pbc, width); |
182 |
|
13 |
bytestream2_put_be16(&pbc, height); |
183 |
|
13 |
bytestream2_put_be16(&pbc, depth); |
184 |
|
|
|
185 |
|
13 |
bytestream2_put_be32(&pbc, 0L); /* pixmin */ |
186 |
|
13 |
bytestream2_put_be32(&pbc, pixmax); |
187 |
|
13 |
bytestream2_put_be32(&pbc, 0L); /* dummy */ |
188 |
|
|
|
189 |
|
|
/* name */ |
190 |
✓✓ |
1053 |
for (i = 0; i < 80; i++) |
191 |
|
1040 |
bytestream2_put_byte(&pbc, 0L); |
192 |
|
|
|
193 |
|
|
/* colormap */ |
194 |
|
13 |
bytestream2_put_be32(&pbc, 0L); |
195 |
|
|
|
196 |
|
|
/* The rest of the 512 byte header is unused. */ |
197 |
✓✓ |
5265 |
for (i = 0; i < 404; i++) |
198 |
|
5252 |
bytestream2_put_byte(&pbc, 0L); |
199 |
|
|
|
200 |
✓✗ |
13 |
if (s->rle) { |
201 |
|
|
PutByteContext taboff_pcb, tablen_pcb; |
202 |
|
|
|
203 |
|
|
/* Skip RLE offset table. */ |
204 |
|
13 |
bytestream2_init_writer(&taboff_pcb, pbc.buffer, tablesize); |
205 |
|
13 |
bytestream2_skip_p(&pbc, tablesize); |
206 |
|
|
|
207 |
|
|
/* Skip RLE length table. */ |
208 |
|
13 |
bytestream2_init_writer(&tablen_pcb, pbc.buffer, tablesize); |
209 |
|
13 |
bytestream2_skip_p(&pbc, tablesize); |
210 |
|
|
|
211 |
|
|
/* Make an intermediate consecutive buffer. */ |
212 |
✗✓ |
13 |
if (!(encode_buf = av_malloc(width * bytes_per_channel))) |
213 |
|
|
return AVERROR(ENOMEM); |
214 |
|
|
|
215 |
✓✓ |
52 |
for (z = 0; z < depth; z++) { |
216 |
|
39 |
in_buf = p->data[0] + p->linesize[0] * (height - 1) + z * bytes_per_channel; |
217 |
|
|
|
218 |
✓✓ |
11271 |
for (y = 0; y < height; y++) { |
219 |
|
11232 |
bytestream2_put_be32(&taboff_pcb, bytestream2_tell_p(&pbc)); |
220 |
|
|
|
221 |
✓✓ |
3964896 |
for (x = 0; x < width * bytes_per_channel; x += bytes_per_channel) |
222 |
✓✗ |
3953664 |
if (bytes_per_channel == 1) { |
223 |
|
3953664 |
encode_buf[x] = in_buf[depth * x]; |
224 |
|
|
} else if (HAVE_BIGENDIAN ^ put_be) { |
225 |
|
|
encode_buf[x + 1] = in_buf[depth * x]; |
226 |
|
|
encode_buf[x] = in_buf[depth * x + 1]; |
227 |
|
|
} else { |
228 |
|
|
encode_buf[x] = in_buf[depth * x]; |
229 |
|
|
encode_buf[x + 1] = in_buf[depth * x + 1]; |
230 |
|
|
} |
231 |
|
|
|
232 |
|
11232 |
length = sgi_rle_encode(&pbc, encode_buf, width, |
233 |
|
|
bytes_per_channel); |
234 |
✗✓ |
11232 |
if (length < 1) { |
235 |
|
|
av_free(encode_buf); |
236 |
|
|
return AVERROR_INVALIDDATA; |
237 |
|
|
} |
238 |
|
|
|
239 |
|
11232 |
bytestream2_put_be32(&tablen_pcb, length); |
240 |
|
11232 |
in_buf -= p->linesize[0]; |
241 |
|
|
} |
242 |
|
|
} |
243 |
|
|
|
244 |
|
13 |
av_free(encode_buf); |
245 |
|
|
} else { |
246 |
|
|
for (z = 0; z < depth; z++) { |
247 |
|
|
in_buf = p->data[0] + p->linesize[0] * (height - 1) + z * bytes_per_channel; |
248 |
|
|
|
249 |
|
|
for (y = 0; y < height; y++) { |
250 |
|
|
for (x = 0; x < width * depth; x += depth) |
251 |
|
|
if (bytes_per_channel == 1) |
252 |
|
|
bytestream2_put_byte(&pbc, in_buf[x]); |
253 |
|
|
else |
254 |
|
|
if (put_be) |
255 |
|
|
bytestream2_put_be16(&pbc, ((uint16_t *)in_buf)[x]); |
256 |
|
|
else |
257 |
|
|
bytestream2_put_le16(&pbc, ((uint16_t *)in_buf)[x]); |
258 |
|
|
|
259 |
|
|
in_buf -= p->linesize[0]; |
260 |
|
|
} |
261 |
|
|
} |
262 |
|
|
} |
263 |
|
|
|
264 |
|
|
/* total length */ |
265 |
|
13 |
pkt->size = bytestream2_tell_p(&pbc); |
266 |
|
13 |
pkt->flags |= AV_PKT_FLAG_KEY; |
267 |
|
13 |
*got_packet = 1; |
268 |
|
|
|
269 |
|
13 |
return 0; |
270 |
|
|
} |
271 |
|
|
|
272 |
|
|
#define OFFSET(x) offsetof(SgiContext, x) |
273 |
|
|
#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM |
274 |
|
|
static const AVOption options[] = { |
275 |
|
|
{ "rle", "Use run-length compression", OFFSET(rle), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, VE }, |
276 |
|
|
|
277 |
|
|
{ NULL }, |
278 |
|
|
}; |
279 |
|
|
|
280 |
|
|
static const AVClass sgi_class = { |
281 |
|
|
.class_name = "sgi", |
282 |
|
|
.item_name = av_default_item_name, |
283 |
|
|
.option = options, |
284 |
|
|
.version = LIBAVUTIL_VERSION_INT, |
285 |
|
|
}; |
286 |
|
|
|
287 |
|
|
AVCodec ff_sgi_encoder = { |
288 |
|
|
.name = "sgi", |
289 |
|
|
.long_name = NULL_IF_CONFIG_SMALL("SGI image"), |
290 |
|
|
.type = AVMEDIA_TYPE_VIDEO, |
291 |
|
|
.id = AV_CODEC_ID_SGI, |
292 |
|
|
.priv_data_size = sizeof(SgiContext), |
293 |
|
|
.priv_class = &sgi_class, |
294 |
|
|
.init = encode_init, |
295 |
|
|
.encode2 = encode_frame, |
296 |
|
|
.pix_fmts = (const enum AVPixelFormat[]) { |
297 |
|
|
AV_PIX_FMT_RGB24, AV_PIX_FMT_RGBA, |
298 |
|
|
AV_PIX_FMT_RGB48LE, AV_PIX_FMT_RGB48BE, |
299 |
|
|
AV_PIX_FMT_RGBA64LE, AV_PIX_FMT_RGBA64BE, |
300 |
|
|
AV_PIX_FMT_GRAY16LE, AV_PIX_FMT_GRAY16BE, AV_PIX_FMT_GRAY8, |
301 |
|
|
AV_PIX_FMT_NONE |
302 |
|
|
}, |
303 |
|
|
.caps_internal = FF_CODEC_CAP_INIT_THREADSAFE, |
304 |
|
|
}; |