]> git.sesse.net Git - ffmpeg/blob - libavcodec/sgienc.c
lavc: Add coded bitstream read/write support for AV1
[ffmpeg] / libavcodec / sgienc.c
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 static av_cold int encode_init(AVCodecContext *avctx)
40 {
41     if (avctx->width > 65535 || avctx->height > 65535) {
42         av_log(avctx, AV_LOG_ERROR,
43                "Unsupported resolution %dx%d.\n", avctx->width, avctx->height);
44         av_log(avctx, AV_LOG_ERROR, "SGI does not support resolutions above 65535x65535\n");
45         return AVERROR_INVALIDDATA;
46     }
47
48     return 0;
49 }
50
51 static int sgi_rle_encode(PutByteContext *pbc, const uint8_t *src,
52                           int w, int bpp)
53 {
54     int val, count, x, start = bytestream2_tell_p(pbc);
55     void (*bytestream2_put)(PutByteContext *, unsigned int);
56
57     if (bpp == 1)
58         bytestream2_put = bytestream2_put_byte;
59     else
60         bytestream2_put = bytestream2_put_be16;
61
62     for (x = 0; x < w; x += count) {
63         /* see if we can encode the next set of pixels with RLE */
64         count = ff_rle_count_pixels(src, w - x, bpp, 1);
65         if (count > 1) {
66             if (bytestream2_get_bytes_left_p(pbc) < bpp * 2)
67                 return AVERROR_INVALIDDATA;
68
69             val = bpp == 1 ? *src : AV_RB16(src);
70             bytestream2_put(pbc, count);
71             bytestream2_put(pbc, val);
72         } else {
73             int i;
74             /* fall back on uncompressed */
75             count = ff_rle_count_pixels(src, w - x, bpp, 0);
76             if (bytestream2_get_bytes_left_p(pbc) < bpp * (count + 1))
77                 return AVERROR_INVALIDDATA;
78
79             bytestream2_put(pbc, count + 0x80);
80             for (i = 0; i < count; i++) {
81                 val = bpp == 1 ? src[i] : AV_RB16(src + i * bpp);
82                 bytestream2_put(pbc, val);
83             }
84         }
85
86         src += count * bpp;
87     }
88
89     return bytestream2_tell_p(pbc) - start;
90 }
91
92 static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
93                         const AVFrame *frame, int *got_packet)
94 {
95     SgiContext *s = avctx->priv_data;
96     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     avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
106     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     if (avctx->coder_type == FF_CODER_TYPE_RAW)
113         s->rle = 0;
114 FF_ENABLE_DEPRECATION_WARNINGS
115 #endif
116
117     width  = avctx->width;
118     height = avctx->height;
119     bytes_per_channel = 1;
120     pixmax = 0xFF;
121     put_be = HAVE_BIGENDIAN;
122
123     switch (avctx->pix_fmt) {
124     case AV_PIX_FMT_GRAY8:
125         dimension = SGI_SINGLE_CHAN;
126         depth     = SGI_GRAYSCALE;
127         break;
128     case AV_PIX_FMT_RGB24:
129         dimension = SGI_MULTI_CHAN;
130         depth     = SGI_RGB;
131         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     tablesize = depth * height * 4;
165     length = SGI_HEADER_SIZE;
166     if (!s->rle)
167         length += depth * height * width;
168     else // assume sgi_rle_encode() produces at most 2x size of input
169         length += tablesize * 2 + depth * height * (2 * width + 1);
170
171     if ((ret = ff_alloc_packet2(avctx, pkt, bytes_per_channel * length, 0)) < 0)
172         return ret;
173
174     bytestream2_init_writer(&pbc, pkt->data, pkt->size);
175
176     /* Encode header. */
177     bytestream2_put_be16(&pbc, SGI_MAGIC);
178     bytestream2_put_byte(&pbc, s->rle); /* RLE 1 - VERBATIM 0 */
179     bytestream2_put_byte(&pbc, bytes_per_channel);
180     bytestream2_put_be16(&pbc, dimension);
181     bytestream2_put_be16(&pbc, width);
182     bytestream2_put_be16(&pbc, height);
183     bytestream2_put_be16(&pbc, depth);
184
185     bytestream2_put_be32(&pbc, 0L); /* pixmin */
186     bytestream2_put_be32(&pbc, pixmax);
187     bytestream2_put_be32(&pbc, 0L); /* dummy */
188
189     /* name */
190     for (i = 0; i < 80; i++)
191         bytestream2_put_byte(&pbc, 0L);
192
193     /* colormap */
194     bytestream2_put_be32(&pbc, 0L);
195
196     /* The rest of the 512 byte header is unused. */
197     for (i = 0; i < 404; i++)
198         bytestream2_put_byte(&pbc, 0L);
199
200     if (s->rle) {
201         PutByteContext taboff_pcb, tablen_pcb;
202
203         /* Skip RLE offset table. */
204         bytestream2_init_writer(&taboff_pcb, pbc.buffer, tablesize);
205         bytestream2_skip_p(&pbc, tablesize);
206
207         /* Skip RLE length table. */
208         bytestream2_init_writer(&tablen_pcb, pbc.buffer, tablesize);
209         bytestream2_skip_p(&pbc, tablesize);
210
211         /* Make an intermediate consecutive buffer. */
212         if (!(encode_buf = av_malloc(width * bytes_per_channel)))
213             return AVERROR(ENOMEM);
214
215         for (z = 0; z < depth; z++) {
216             in_buf = p->data[0] + p->linesize[0] * (height - 1) + z * bytes_per_channel;
217
218             for (y = 0; y < height; y++) {
219                 bytestream2_put_be32(&taboff_pcb, bytestream2_tell_p(&pbc));
220
221                 for (x = 0; x < width * bytes_per_channel; x += bytes_per_channel)
222                     encode_buf[x] = in_buf[depth * x];
223
224                 length = sgi_rle_encode(&pbc, encode_buf, width,
225                                         bytes_per_channel);
226                 if (length < 1) {
227                     av_free(encode_buf);
228                     return AVERROR_INVALIDDATA;
229                 }
230
231                 bytestream2_put_be32(&tablen_pcb, length);
232                 in_buf -= p->linesize[0];
233             }
234         }
235
236         av_free(encode_buf);
237     } else {
238         for (z = 0; z < depth; z++) {
239             in_buf = p->data[0] + p->linesize[0] * (height - 1) + z * bytes_per_channel;
240
241             for (y = 0; y < height; y++) {
242                 for (x = 0; x < width * depth; x += depth)
243                     if (bytes_per_channel == 1)
244                         bytestream2_put_byte(&pbc, in_buf[x]);
245                     else
246                         if (put_be)
247                             bytestream2_put_be16(&pbc, ((uint16_t *)in_buf)[x]);
248                         else
249                             bytestream2_put_le16(&pbc, ((uint16_t *)in_buf)[x]);
250
251                 in_buf -= p->linesize[0];
252             }
253         }
254     }
255
256     /* total length */
257     pkt->size   = bytestream2_tell_p(&pbc);
258     pkt->flags |= AV_PKT_FLAG_KEY;
259     *got_packet = 1;
260
261     return 0;
262 }
263
264 #define OFFSET(x) offsetof(SgiContext, x)
265 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
266 static const AVOption options[] = {
267     { "rle", "Use run-length compression", OFFSET(rle), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, VE },
268
269     { NULL },
270 };
271
272 static const AVClass sgi_class = {
273     .class_name = "sgi",
274     .item_name  = av_default_item_name,
275     .option     = options,
276     .version    = LIBAVUTIL_VERSION_INT,
277 };
278
279 AVCodec ff_sgi_encoder = {
280     .name      = "sgi",
281     .long_name = NULL_IF_CONFIG_SMALL("SGI image"),
282     .type      = AVMEDIA_TYPE_VIDEO,
283     .id        = AV_CODEC_ID_SGI,
284     .priv_data_size = sizeof(SgiContext),
285     .priv_class = &sgi_class,
286     .init      = encode_init,
287     .encode2   = encode_frame,
288     .pix_fmts  = (const enum AVPixelFormat[]) {
289         AV_PIX_FMT_RGB24, AV_PIX_FMT_RGBA,
290         AV_PIX_FMT_RGB48LE, AV_PIX_FMT_RGB48BE,
291         AV_PIX_FMT_RGBA64LE, AV_PIX_FMT_RGBA64BE,
292         AV_PIX_FMT_GRAY16LE, AV_PIX_FMT_GRAY16BE, AV_PIX_FMT_GRAY8,
293         AV_PIX_FMT_NONE
294     },
295 };