3 * Todd Kirby <doubleshot@pacbell.net>
5 * This file is part of FFmpeg.
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.
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.
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
23 #include "bytestream.h"
27 #define SGI_SINGLE_CHAN 2
28 #define SGI_MULTI_CHAN 3
30 typedef struct SgiContext
{
34 static av_cold
int encode_init(AVCodecContext
*avctx
)
36 SgiContext
*s
= avctx
->priv_data
;
38 avcodec_get_frame_defaults(&s
->picture
);
39 avctx
->coded_frame
= &s
->picture
;
44 static int encode_frame(AVCodecContext
*avctx
, unsigned char *buf
,
45 int buf_size
, void *data
)
47 SgiContext
*s
= avctx
->priv_data
;
48 AVFrame
* const p
= &s
->picture
;
49 uint8_t *offsettab
, *lengthtab
, *in_buf
, *encode_buf
;
50 int x
, y
, z
, length
, tablesize
;
51 unsigned int width
, height
, depth
, dimension
;
52 unsigned char *orig_buf
= buf
, *end_buf
= buf
+ buf_size
;
55 p
->pict_type
= FF_I_TYPE
;
59 height
= avctx
->height
;
61 switch (avctx
->pix_fmt
) {
63 dimension
= SGI_SINGLE_CHAN
;
64 depth
= SGI_GRAYSCALE
;
67 dimension
= SGI_MULTI_CHAN
;
71 dimension
= SGI_MULTI_CHAN
;
75 return AVERROR_INVALIDDATA
;
78 tablesize
= depth
* height
* 4;
79 length
= tablesize
* 2 + SGI_HEADER_SIZE
;
81 if (buf_size
< length
) {
82 av_log(avctx
, AV_LOG_ERROR
, "buf_size too small(need %d, got %d)\n", length
, buf_size
);
87 bytestream_put_be16(&buf
, SGI_MAGIC
);
88 bytestream_put_byte(&buf
, avctx
->coder_type
!= FF_CODER_TYPE_RAW
); /* RLE 1 - VERBATIM 0*/
89 bytestream_put_byte(&buf
, 1); /* bytes_per_channel */
90 bytestream_put_be16(&buf
, dimension
);
91 bytestream_put_be16(&buf
, width
);
92 bytestream_put_be16(&buf
, height
);
93 bytestream_put_be16(&buf
, depth
);
95 /* The rest are constant in this implementation. */
96 bytestream_put_be32(&buf
, 0L); /* pixmin */
97 bytestream_put_be32(&buf
, 255L); /* pixmax */
98 bytestream_put_be32(&buf
, 0L); /* dummy */
101 memset(buf
, 0, SGI_HEADER_SIZE
);
105 bytestream_put_be32(&buf
, 0L);
107 /* The rest of the 512 byte header is unused. */
111 if (avctx
->coder_type
!= FF_CODER_TYPE_RAW
) {
112 /* Skip RLE offset table. */
116 /* Skip RLE length table. */
119 /* Make an intermediate consecutive buffer. */
120 if (!(encode_buf
= av_malloc(width
)))
123 for (z
= 0; z
< depth
; z
++) {
124 in_buf
= p
->data
[0] + p
->linesize
[0] * (height
- 1) + z
;
126 for (y
= 0; y
< height
; y
++) {
127 bytestream_put_be32(&offsettab
, buf
- orig_buf
);
129 for (x
= 0; x
< width
; x
++)
130 encode_buf
[x
] = in_buf
[depth
* x
];
132 if ((length
= ff_rle_encode(buf
, end_buf
- buf
- 1, encode_buf
, 1, width
, 0, 0, 0x80, 0)) < 1) {
138 bytestream_put_byte(&buf
, 0);
139 bytestream_put_be32(&lengthtab
, length
+ 1);
140 in_buf
-= p
->linesize
[0];
146 for (z
= 0; z
< depth
; z
++) {
147 in_buf
= p
->data
[0] + p
->linesize
[0] * (height
- 1) + z
;
149 for (y
= 0; y
< height
; y
++) {
150 for (x
= 0; x
< width
* depth
; x
+= depth
)
151 bytestream_put_byte(&buf
, in_buf
[x
]);
153 in_buf
-= p
->linesize
[0];
159 return buf
- orig_buf
;
162 AVCodec sgi_encoder
= {
170 .pix_fmts
= (const enum PixelFormat
[]){PIX_FMT_RGB24
, PIX_FMT_RGBA
, PIX_FMT_GRAY8
, PIX_FMT_NONE
},
171 .long_name
= NULL_IF_CONFIG_SMALL("SGI image"),