2 * PC Paintbrush PCX (.pcx) image encoder
3 * Copyright (c) 2009 Daniel Verkamp <daniel at drv.nu>
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
25 * @author Daniel Verkamp
26 * @see http://www.qzx.com/pc-gpe/pcx.txt
30 #include "bytestream.h"
31 #include "libavutil/imgutils.h"
33 typedef struct PCXContext {
37 static const uint32_t monoblack_pal[16] = { 0x000000, 0xFFFFFF };
39 static av_cold int pcx_encode_init(AVCodecContext *avctx)
41 PCXContext *s = avctx->priv_data;
43 avcodec_get_frame_defaults(&s->picture);
44 avctx->coded_frame = &s->picture;
50 * PCX run-length encoder
51 * @param dst output buffer
52 * @param dst_size size of output buffer
53 * @param src input buffer
54 * @param src_plane_size size of one plane of input buffer in bytes
55 * @param nplanes number of planes in input buffer
56 * @return number of bytes written to dst or -1 on error
57 * @bug will not work for nplanes != 1 && bpp != 8
59 static int pcx_rle_encode( uint8_t *dst, int dst_size,
60 const uint8_t *src, int src_plane_size, int nplanes)
63 const uint8_t *dst_start = dst;
65 // check worst-case upper bound on dst_size
66 if (dst_size < 2LL * src_plane_size * nplanes || src_plane_size <= 0)
69 for (p = 0; p < nplanes; p++) {
71 const uint8_t *src_plane = src + p;
72 const uint8_t *src_plane_end = src_plane + src_plane_size * nplanes;
73 uint8_t prev = *src_plane;
76 for (; ; src_plane += nplanes) {
77 if (src_plane < src_plane_end && *src_plane == prev && count < 0x3F) {
78 // current byte is same as prev
81 // output prev * count
82 if (count != 1 || prev >= 0xC0)
83 *dst++ = 0xC0 | count;
86 if (src_plane == src_plane_end)
96 return dst - dst_start;
99 static int pcx_encode_frame(AVCodecContext *avctx,
100 unsigned char *buf, int buf_size, void *data)
102 PCXContext *s = avctx->priv_data;
103 AVFrame *const pict = &s->picture;
104 const uint8_t *buf_start = buf;
105 const uint8_t *buf_end = buf + buf_size;
107 int bpp, nplanes, i, y, line_bytes, written;
108 const uint32_t *pal = NULL;
109 uint32_t palette256[256];
112 *pict = *(AVFrame *)data;
113 pict->pict_type = AV_PICTURE_TYPE_I;
116 if (avctx->width > 65535 || avctx->height > 65535) {
117 av_log(avctx, AV_LOG_ERROR, "image dimensions do not fit in 16 bits\n");
121 switch (avctx->pix_fmt) {
128 case PIX_FMT_RGB4_BYTE:
129 case PIX_FMT_BGR4_BYTE:
133 ff_set_systematic_pal2(palette256, avctx->pix_fmt);
139 pal = (uint32_t *)pict->data[1];
141 case PIX_FMT_MONOBLACK:
147 av_log(avctx, AV_LOG_ERROR, "unsupported pixfmt\n");
151 line_bytes = (avctx->width * bpp + 7) >> 3;
152 line_bytes = (line_bytes + 1) & ~1;
154 bytestream_put_byte(&buf, 10); // manufacturer
155 bytestream_put_byte(&buf, 5); // version
156 bytestream_put_byte(&buf, 1); // encoding
157 bytestream_put_byte(&buf, bpp); // bits per pixel per plane
158 bytestream_put_le16(&buf, 0); // x min
159 bytestream_put_le16(&buf, 0); // y min
160 bytestream_put_le16(&buf, avctx->width - 1); // x max
161 bytestream_put_le16(&buf, avctx->height - 1); // y max
162 bytestream_put_le16(&buf, 0); // horizontal DPI
163 bytestream_put_le16(&buf, 0); // vertical DPI
164 for (i = 0; i < 16; i++)
165 bytestream_put_be24(&buf, pal ? pal[i] : 0);// palette (<= 16 color only)
166 bytestream_put_byte(&buf, 0); // reserved
167 bytestream_put_byte(&buf, nplanes); // number of planes
168 bytestream_put_le16(&buf, line_bytes); // scanline plane size in bytes
170 while (buf - buf_start < 128)
175 for (y = 0; y < avctx->height; y++) {
176 if ((written = pcx_rle_encode(buf, buf_end - buf,
177 src, line_bytes, nplanes)) < 0) {
178 av_log(avctx, AV_LOG_ERROR, "buffer too small\n");
182 src += pict->linesize[0];
185 if (nplanes == 1 && bpp == 8) {
186 if (buf_end - buf < 257) {
187 av_log(avctx, AV_LOG_ERROR, "buffer too small\n");
190 bytestream_put_byte(&buf, 12);
191 for (i = 0; i < 256; i++) {
192 bytestream_put_be24(&buf, pal[i]);
196 return buf - buf_start;
199 AVCodec ff_pcx_encoder = {
201 .type = AVMEDIA_TYPE_VIDEO,
203 .priv_data_size = sizeof(PCXContext),
204 .init = pcx_encode_init,
205 .encode = pcx_encode_frame,
206 .pix_fmts = (const enum PixelFormat[]){
208 PIX_FMT_RGB8, PIX_FMT_BGR8, PIX_FMT_RGB4_BYTE, PIX_FMT_BGR4_BYTE, PIX_FMT_GRAY8, PIX_FMT_PAL8,
211 .long_name = NULL_IF_CONFIG_SMALL("PC Paintbrush PCX image"),