FFmpeg
targaenc.c
Go to the documentation of this file.
1 /*
2  * Targa (.tga) image encoder
3  * Copyright (c) 2007 Bobby Bingham
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 <string.h>
23 
24 #include "libavutil/imgutils.h"
25 #include "libavutil/internal.h"
26 #include "libavutil/intreadwrite.h"
27 #include "libavutil/opt.h"
28 #include "libavutil/pixdesc.h"
29 #include "avcodec.h"
30 #include "internal.h"
31 #include "rle.h"
32 #include "targa.h"
33 
34 typedef struct TargaContext {
35  AVClass *class;
36 
37  int rle;
38 } TargaContext;
39 
40 /**
41  * RLE compress the image, with maximum size of out_size
42  * @param outbuf Output buffer
43  * @param out_size Maximum output size
44  * @param pic Image to compress
45  * @param bpp Bytes per pixel
46  * @param w Image width
47  * @param h Image height
48  * @return Size of output in bytes, or -1 if larger than out_size
49  */
50 static int targa_encode_rle(uint8_t *outbuf, int out_size, const AVFrame *pic,
51  int bpp, int w, int h)
52 {
53  int y,ret;
54  uint8_t *out;
55 
56  out = outbuf;
57 
58  for(y = 0; y < h; y ++) {
59  ret = ff_rle_encode(out, out_size, pic->data[0] + pic->linesize[0] * y, bpp, w, 0x7f, 0, -1, 0);
60  if(ret == -1){
61  return -1;
62  }
63  out+= ret;
64  out_size -= ret;
65  }
66 
67  return out - outbuf;
68 }
69 
70 static int targa_encode_normal(uint8_t *outbuf, const AVFrame *pic, int bpp, int w, int h)
71 {
72  int i, n = bpp * w;
73  uint8_t *out = outbuf;
74  uint8_t *ptr = pic->data[0];
75 
76  for(i=0; i < h; i++) {
77  memcpy(out, ptr, n);
78  out += n;
79  ptr += pic->linesize[0];
80  }
81 
82  return out - outbuf;
83 }
84 
86  const AVFrame *p, int *got_packet)
87 {
88  TargaContext *s = avctx->priv_data;
89  int bpp, picsize, datasize = -1, ret, i;
90  uint8_t *out;
91 
92  picsize = av_image_get_buffer_size(avctx->pix_fmt,
93  avctx->width, avctx->height, 1);
94  if ((ret = ff_alloc_packet2(avctx, pkt, picsize + 45, 0)) < 0)
95  return ret;
96 
97  /* zero out the header and only set applicable fields */
98  memset(pkt->data, 0, 12);
99  AV_WL16(pkt->data+12, avctx->width);
100  AV_WL16(pkt->data+14, avctx->height);
101  /* image descriptor byte: origin is always top-left, bits 0-3 specify alpha */
102  pkt->data[17] = 0x20 | (avctx->pix_fmt == AV_PIX_FMT_BGRA ? 8 : 0);
103 
104  out = pkt->data + 18; /* skip past the header we write */
105 
107  switch(avctx->pix_fmt) {
108  case AV_PIX_FMT_PAL8: {
109  int pal_bpp = 24; /* Only write 32bit palette if there is transparency information */
110  for (i = 0; i < 256; i++)
111  if (AV_RN32(p->data[1] + 4 * i) >> 24 != 0xFF) {
112  pal_bpp = 32;
113  break;
114  }
115  pkt->data[1] = 1; /* palette present */
116  pkt->data[2] = TGA_PAL; /* uncompressed palettised image */
117  pkt->data[6] = 1; /* palette contains 256 entries */
118  pkt->data[7] = pal_bpp; /* palette contains pal_bpp bit entries */
119  pkt->data[16] = 8; /* bpp */
120  for (i = 0; i < 256; i++)
121  if (pal_bpp == 32) {
122  AV_WL32(pkt->data + 18 + 4 * i, *(uint32_t *)(p->data[1] + i * 4));
123  } else {
124  AV_WL24(pkt->data + 18 + 3 * i, *(uint32_t *)(p->data[1] + i * 4));
125  }
126  out += 32 * pal_bpp; /* skip past the palette we just output */
127  break;
128  }
129  case AV_PIX_FMT_GRAY8:
130  pkt->data[2] = TGA_BW; /* uncompressed grayscale image */
131  avctx->bits_per_coded_sample = 0x28;
132  pkt->data[16] = 8; /* bpp */
133  break;
134  case AV_PIX_FMT_RGB555LE:
135  pkt->data[2] = TGA_RGB; /* uncompressed true-color image */
136  avctx->bits_per_coded_sample =
137  pkt->data[16] = 16; /* bpp */
138  break;
139  case AV_PIX_FMT_BGR24:
140  pkt->data[2] = TGA_RGB; /* uncompressed true-color image */
141  pkt->data[16] = 24; /* bpp */
142  break;
143  case AV_PIX_FMT_BGRA:
144  pkt->data[2] = TGA_RGB; /* uncompressed true-color image */
145  pkt->data[16] = 32; /* bpp */
146  break;
147  default:
148  av_log(avctx, AV_LOG_ERROR, "Pixel format '%s' not supported.\n",
149  av_get_pix_fmt_name(avctx->pix_fmt));
150  return AVERROR(EINVAL);
151  }
152  bpp = pkt->data[16] >> 3;
153 
154 
155  /* try RLE compression */
156  if (s->rle)
157  datasize = targa_encode_rle(out, picsize, p, bpp, avctx->width, avctx->height);
158 
159  /* if that worked well, mark the picture as RLE compressed */
160  if(datasize >= 0)
161  pkt->data[2] |= TGA_RLE;
162 
163  /* if RLE didn't make it smaller, go back to no compression */
164  else datasize = targa_encode_normal(out, p, bpp, avctx->width, avctx->height);
165 
166  out += datasize;
167 
168  /* The standard recommends including this section, even if we don't use
169  * any of the features it affords. TODO: take advantage of the pixel
170  * aspect ratio and encoder ID fields available? */
171  memcpy(out, "\0\0\0\0\0\0\0\0TRUEVISION-XFILE.", 26);
172 
173  pkt->size = out + 26 - pkt->data;
175  *got_packet = 1;
176 
177  return 0;
178 }
179 
181 {
182  if (avctx->width > 0xffff || avctx->height > 0xffff) {
183  av_log(avctx, AV_LOG_ERROR, "image dimensions too large\n");
184  return AVERROR(EINVAL);
185  }
186 
187  return 0;
188 }
189 
190 #define OFFSET(x) offsetof(TargaContext, x)
191 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
192 static const AVOption options[] = {
193  { "rle", "Use run-length compression", OFFSET(rle), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, VE },
194 
195  { NULL },
196 };
197 
198 static const AVClass targa_class = {
199  .class_name = "targa",
200  .item_name = av_default_item_name,
201  .option = options,
202  .version = LIBAVUTIL_VERSION_INT,
203 };
204 
206  .name = "targa",
207  .long_name = NULL_IF_CONFIG_SMALL("Truevision Targa image"),
208  .type = AVMEDIA_TYPE_VIDEO,
209  .id = AV_CODEC_ID_TARGA,
210  .priv_data_size = sizeof(TargaContext),
211  .priv_class = &targa_class,
213  .encode2 = targa_encode_frame,
214  .pix_fmts = (const enum AVPixelFormat[]){
217  },
218  .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
219 };
AVCodec
AVCodec.
Definition: codec.h:197
FF_CODEC_CAP_INIT_THREADSAFE
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
Definition: internal.h:41
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
TGA_RGB
@ TGA_RGB
Definition: targa.h:36
init
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:31
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
targa_class
static const AVClass targa_class
Definition: targaenc.c:198
AV_WL32
#define AV_WL32(p, v)
Definition: intreadwrite.h:426
rle.h
out
FILE * out
Definition: movenc.c:54
OFFSET
#define OFFSET(x)
Definition: targaenc.c:190
av_pix_fmt_desc_get
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:2541
out_size
int out_size
Definition: movenc.c:55
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:303
pixdesc.h
w
uint8_t w
Definition: llviddspenc.c:39
internal.h
AVPacket::data
uint8_t * data
Definition: packet.h:365
AVOption
AVOption.
Definition: opt.h:248
ff_targa_encoder
const AVCodec ff_targa_encoder
Definition: targaenc.c:205
AV_PIX_FMT_BGR24
@ AV_PIX_FMT_BGR24
packed RGB 8:8:8, 24bpp, BGRBGR...
Definition: pixfmt.h:69
AV_PIX_FMT_BGRA
@ AV_PIX_FMT_BGRA
packed BGRA 8:8:8:8, 32bpp, BGRABGRA...
Definition: pixfmt.h:95
av_get_bits_per_pixel
int av_get_bits_per_pixel(const AVPixFmtDescriptor *pixdesc)
Return the number of bits per pixel used by the pixel format described by pixdesc.
Definition: pixdesc.c:2493
targa.h
AV_PKT_FLAG_KEY
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:396
rle
static int rle(uint8_t *dst, const uint8_t *src, int compressed_size, int uncompressed_size)
Definition: exr.c:215
TGA_PAL
@ TGA_PAL
Definition: targa.h:35
AVFrame::data
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:317
VE
#define VE
Definition: targaenc.c:191
targa_encode_rle
static int targa_encode_rle(uint8_t *outbuf, int out_size, const AVFrame *pic, int bpp, int w, int h)
RLE compress the image, with maximum size of out_size.
Definition: targaenc.c:50
pkt
AVPacket * pkt
Definition: movenc.c:59
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:181
av_cold
#define av_cold
Definition: attributes.h:90
TGA_RLE
@ TGA_RLE
Definition: targa.h:38
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:257
TGA_BW
@ TGA_BW
Definition: targa.h:37
pix_fmts
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:289
TargaContext::rle
int rle
Definition: targaenc.c:37
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:67
NULL
#define NULL
Definition: coverity.c:32
targa_encode_init
static av_cold int targa_encode_init(AVCodecContext *avctx)
Definition: targaenc.c:180
TargaContext
Definition: targa.c:29
AV_WL24
#define AV_WL24(p, d)
Definition: intreadwrite.h:464
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:235
AV_RN32
#define AV_RN32(p)
Definition: intreadwrite.h:364
AV_PIX_FMT_GRAY8
@ AV_PIX_FMT_GRAY8
Y , 8bpp.
Definition: pixfmt.h:74
AVPacket::size
int size
Definition: packet.h:366
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
targa_encode_frame
static int targa_encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *p, int *got_packet)
Definition: targaenc.c:85
AV_WL16
#define AV_WL16(p, v)
Definition: intreadwrite.h:412
av_image_get_buffer_size
int av_image_get_buffer_size(enum AVPixelFormat pix_fmt, int width, int height, int align)
Return the size in bytes of the amount of data required to store an image with the given parameters.
Definition: imgutils.c:463
AVPacket::flags
int flags
A combination of AV_PKT_FLAG values.
Definition: packet.h:371
AVCodecContext::bits_per_coded_sample
int bits_per_coded_sample
bits per sample/pixel from the demuxer (needed for huffyuv).
Definition: avcodec.h:1524
AV_PIX_FMT_RGB555LE
@ AV_PIX_FMT_RGB555LE
packed RGB 5:5:5, 16bpp, (msb)1X 5R 5G 5B(lsb), little-endian, X=unused/undefined
Definition: pixfmt.h:108
i
int i
Definition: input.c:407
internal.h
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:204
AVCodecContext::height
int height
Definition: avcodec.h:674
AVCodecContext::pix_fmt
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:711
avcodec.h
AV_PIX_FMT_PAL8
@ AV_PIX_FMT_PAL8
8 bits with AV_PIX_FMT_RGB32 palette
Definition: pixfmt.h:77
ret
ret
Definition: filter_design.txt:187
AVClass::class_name
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:72
ff_rle_encode
int ff_rle_encode(uint8_t *outbuf, int out_size, const uint8_t *ptr, int bpp, int w, int add_rep, int xor_rep, int add_raw, int xor_raw)
RLE compress the row, with maximum size of out_size.
Definition: rle.c:52
AVCodecContext
main external API structure.
Definition: avcodec.h:501
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:225
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
targa_encode_normal
static int targa_encode_normal(uint8_t *outbuf, const AVFrame *pic, int bpp, int w, int h)
Definition: targaenc.c:70
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:528
AVPacket
This structure stores compressed data.
Definition: packet.h:342
AVCodecContext::width
int width
picture width / height.
Definition: avcodec.h:674
imgutils.h
AVFrame::linesize
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:334
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:28
h
h
Definition: vp9dsp_template.c:2038
AV_CODEC_ID_TARGA
@ AV_CODEC_ID_TARGA
Definition: codec_id.h:142
ff_alloc_packet2
int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size, int64_t min_size)
Check AVPacket size and/or allocate data.
Definition: encode.c:33
options
static const AVOption options[]
Definition: targaenc.c:192
av_get_pix_fmt_name
const char * av_get_pix_fmt_name(enum AVPixelFormat pix_fmt)
Return the short name for a pixel format, NULL in case pix_fmt is unknown.
Definition: pixdesc.c:2461