FFmpeg coverage

GCC Code Coverage Report

Directory:	../../../ffmpeg/		Exec	Total	Coverage
File:	src/libavcodec/targaenc.c	Lines:	50	90	55.6 %
Date:	2021-04-14 23:45:22	Branches:	13	34	38.2 %


/*
 * Targa (.tga) image encoder
 * Copyright (c) 2007 Bobby Bingham
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <string.h>

#include "libavutil/imgutils.h"
#include "libavutil/internal.h"
#include "libavutil/intreadwrite.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avcodec.h"
#include "internal.h"
#include "rle.h"
#include "targa.h"

typedef struct TargaContext {
    AVClass *class;

    int rle;
} TargaContext;

/**
 * RLE compress the image, with maximum size of out_size
 * @param outbuf Output buffer
 * @param out_size Maximum output size
 * @param pic Image to compress
 * @param bpp Bytes per pixel
 * @param w Image width
 * @param h Image height
 * @return Size of output in bytes, or -1 if larger than out_size
 */
static int targa_encode_rle(uint8_t *outbuf, int out_size, const AVFrame *pic,
                            int bpp, int w, int h)
{
    int y,ret;
    uint8_t *out;

    out = outbuf;

    for(y = 0; y < h; y ++) {
        ret = ff_rle_encode(out, out_size, pic->data[0] + pic->linesize[0] * y, bpp, w, 0x7f, 0, -1, 0);
        if(ret == -1){
            return -1;
        }
        out+= ret;
        out_size -= ret;
    }

    return out - outbuf;
}

static int targa_encode_normal(uint8_t *outbuf, const AVFrame *pic, int bpp, int w, int h)
{
    int i, n = bpp * w;
    uint8_t *out = outbuf;
    uint8_t *ptr = pic->data[0];

    for(i=0; i < h; i++) {
        memcpy(out, ptr, n);
        out += n;
        ptr += pic->linesize[0];
    }

    return out - outbuf;
}

static int targa_encode_frame(AVCodecContext *avctx, AVPacket *pkt,
                              const AVFrame *p, int *got_packet)
{
    TargaContext *s = avctx->priv_data;
    int bpp, picsize, datasize = -1, ret, i;
    uint8_t *out;

    picsize = av_image_get_buffer_size(avctx->pix_fmt,
                                       avctx->width, avctx->height, 1);
    if ((ret = ff_alloc_packet2(avctx, pkt, picsize + 45, 0)) < 0)
        return ret;

    /* zero out the header and only set applicable fields */
    memset(pkt->data, 0, 12);
    AV_WL16(pkt->data+12, avctx->width);
    AV_WL16(pkt->data+14, avctx->height);
    /* image descriptor byte: origin is always top-left, bits 0-3 specify alpha */
    pkt->data[17] = 0x20 | (avctx->pix_fmt == AV_PIX_FMT_BGRA ? 8 : 0);

    out = pkt->data + 18;  /* skip past the header we write */

    avctx->bits_per_coded_sample = av_get_bits_per_pixel(av_pix_fmt_desc_get(avctx->pix_fmt));
    switch(avctx->pix_fmt) {
    case AV_PIX_FMT_PAL8: {
        int pal_bpp = 24; /* Only write 32bit palette if there is transparency information */
        for (i = 0; i < 256; i++)
            if (AV_RN32(p->data[1] + 4 * i) >> 24 != 0xFF) {
                pal_bpp = 32;
                break;
            }
        pkt->data[1]  = 1;          /* palette present */
        pkt->data[2]  = TGA_PAL;    /* uncompressed palettised image */
        pkt->data[6]  = 1;          /* palette contains 256 entries */
        pkt->data[7]  = pal_bpp;    /* palette contains pal_bpp bit entries */
        pkt->data[16] = 8;          /* bpp */
        for (i = 0; i < 256; i++)
            if (pal_bpp == 32) {
                AV_WL32(pkt->data + 18 + 4 * i, *(uint32_t *)(p->data[1] + i * 4));
            } else {
            AV_WL24(pkt->data + 18 + 3 * i, *(uint32_t *)(p->data[1] + i * 4));
            }
        out += 32 * pal_bpp;        /* skip past the palette we just output */
        break;
        }
    case AV_PIX_FMT_GRAY8:
        pkt->data[2]  = TGA_BW;     /* uncompressed grayscale image */
        avctx->bits_per_coded_sample = 0x28;
        pkt->data[16] = 8;          /* bpp */
        break;
    case AV_PIX_FMT_RGB555LE:
        pkt->data[2]  = TGA_RGB;    /* uncompressed true-color image */
        avctx->bits_per_coded_sample =
        pkt->data[16] = 16;         /* bpp */
        break;
    case AV_PIX_FMT_BGR24:
        pkt->data[2]  = TGA_RGB;    /* uncompressed true-color image */
        pkt->data[16] = 24;         /* bpp */
        break;
    case AV_PIX_FMT_BGRA:
        pkt->data[2]  = TGA_RGB;    /* uncompressed true-color image */
        pkt->data[16] = 32;         /* bpp */
        break;
    default:
        av_log(avctx, AV_LOG_ERROR, "Pixel format '%s' not supported.\n",
               av_get_pix_fmt_name(avctx->pix_fmt));
        return AVERROR(EINVAL);
    }
    bpp = pkt->data[16] >> 3;


#if FF_API_CODER_TYPE
FF_DISABLE_DEPRECATION_WARNINGS
    if (avctx->coder_type == FF_CODER_TYPE_RAW)
        s->rle = 0;
FF_ENABLE_DEPRECATION_WARNINGS
#endif

    /* try RLE compression */
    if (s->rle)
        datasize = targa_encode_rle(out, picsize, p, bpp, avctx->width, avctx->height);

    /* if that worked well, mark the picture as RLE compressed */
    if(datasize >= 0)
        pkt->data[2] |= TGA_RLE;

    /* if RLE didn't make it smaller, go back to no compression */
    else datasize = targa_encode_normal(out, p, bpp, avctx->width, avctx->height);

    out += datasize;

    /* The standard recommends including this section, even if we don't use
     * any of the features it affords. TODO: take advantage of the pixel
     * aspect ratio and encoder ID fields available? */
    memcpy(out, "\0\0\0\0\0\0\0\0TRUEVISION-XFILE.", 26);

    pkt->size   = out + 26 - pkt->data;
    pkt->flags |= AV_PKT_FLAG_KEY;
    *got_packet = 1;

    return 0;
}

static av_cold int targa_encode_init(AVCodecContext *avctx)
{
    if (avctx->width > 0xffff || avctx->height > 0xffff) {
        av_log(avctx, AV_LOG_ERROR, "image dimensions too large\n");
        return AVERROR(EINVAL);
    }

#if FF_API_CODED_FRAME
FF_DISABLE_DEPRECATION_WARNINGS
    avctx->coded_frame->key_frame = 1;
    avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
FF_ENABLE_DEPRECATION_WARNINGS
#endif

    return 0;
}

#define OFFSET(x) offsetof(TargaContext, x)
#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
static const AVOption options[] = {
    { "rle", "Use run-length compression", OFFSET(rle), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, VE },

    { NULL },
};

static const AVClass targa_class = {
    .class_name = "targa",
    .item_name  = av_default_item_name,
    .option     = options,
    .version    = LIBAVUTIL_VERSION_INT,
};

AVCodec ff_targa_encoder = {
    .name           = "targa",
    .long_name      = NULL_IF_CONFIG_SMALL("Truevision Targa image"),
    .type           = AVMEDIA_TYPE_VIDEO,
    .id             = AV_CODEC_ID_TARGA,
    .priv_data_size = sizeof(TargaContext),
    .priv_class     = &targa_class,
    .init           = targa_encode_init,
    .encode2        = targa_encode_frame,
    .pix_fmts       = (const enum AVPixelFormat[]){
        AV_PIX_FMT_BGR24, AV_PIX_FMT_BGRA, AV_PIX_FMT_RGB555LE, AV_PIX_FMT_GRAY8, AV_PIX_FMT_PAL8,
        AV_PIX_FMT_NONE
    },
};


Generated by: GCOVR (Version 4.2)

Line	Branch	Exec	Source
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		13	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		13	out = outbuf;
57
58	✓✗	3744	for(y = 0; y < h; y ++) {
59		3744	ret = ff_rle_encode(out, out_size, pic->data[0] + pic->linesize[0] * y, bpp, w, 0x7f, 0, -1, 0);
60	✓✓	3744	if(ret == -1){
61		13	return -1;
62			}
63		3731	out+= ret;
64		3731	out_size -= ret;
65			}
66
67			return out - outbuf;
68			}
69
70		13	static int targa_encode_normal(uint8_t outbuf, const AVFrame pic, int bpp, int w, int h)
71			{
72		13	int i, n = bpp * w;
73		13	uint8_t *out = outbuf;
74		13	uint8_t *ptr = pic->data[0];
75
76	✓✓	3757	for(i=0; i < h; i++) {
77		3744	memcpy(out, ptr, n);
78		3744	out += n;
79		3744	ptr += pic->linesize[0];
80			}
81
82		13	return out - outbuf;
83			}
84
85		13	static int targa_encode_frame(AVCodecContext avctx, AVPacket pkt,
86			const AVFrame p, int got_packet)
87			{
88		13	TargaContext *s = avctx->priv_data;
89		13	int bpp, picsize, datasize = -1, ret, i;
90			uint8_t *out;
91
92		13	picsize = av_image_get_buffer_size(avctx->pix_fmt,
93			avctx->width, avctx->height, 1);
94	✗✓	13	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		13	memset(pkt->data, 0, 12);
99		13	AV_WL16(pkt->data+12, avctx->width);
100		13	AV_WL16(pkt->data+14, avctx->height);
101			/* image descriptor byte: origin is always top-left, bits 0-3 specify alpha */
102	✗✓	13	pkt->data[17] = 0x20 \| (avctx->pix_fmt == AV_PIX_FMT_BGRA ? 8 : 0);
103
104		13	out = pkt->data + 18; /* skip past the header we write */
105
106		13	avctx->bits_per_coded_sample = av_get_bits_per_pixel(av_pix_fmt_desc_get(avctx->pix_fmt));
107	✗✗✗✓ ✗✗	13	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		13	case AV_PIX_FMT_BGR24:
140		13	pkt->data[2] = TGA_RGB; /* uncompressed true-color image */
141		13	pkt->data[16] = 24; /* bpp */
142		13	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		13	bpp = pkt->data[16] >> 3;
153
154
155			#if FF_API_CODER_TYPE
156			FF_DISABLE_DEPRECATION_WARNINGS
157	✗✓	13	if (avctx->coder_type == FF_CODER_TYPE_RAW)
158			s->rle = 0;
159			FF_ENABLE_DEPRECATION_WARNINGS
160			#endif
161
162			/* try RLE compression */
163	✓✗	13	if (s->rle)
164		13	datasize = targa_encode_rle(out, picsize, p, bpp, avctx->width, avctx->height);
165
166			/* if that worked well, mark the picture as RLE compressed */
167	✗✓	13	if(datasize >= 0)
168			pkt->data[2] \|= TGA_RLE;
169
170			/* if RLE didn't make it smaller, go back to no compression */
171		13	else datasize = targa_encode_normal(out, p, bpp, avctx->width, avctx->height);
172
173		13	out += datasize;
174
175			/* The standard recommends including this section, even if we don't use
176			* any of the features it affords. TODO: take advantage of the pixel
177			* aspect ratio and encoder ID fields available? */
178		13	memcpy(out, "\0\0\0\0\0\0\0\0TRUEVISION-XFILE.", 26);
179
180		13	pkt->size = out + 26 - pkt->data;
181		13	pkt->flags \|= AV_PKT_FLAG_KEY;
182		13	*got_packet = 1;
183
184		13	return 0;
185			}
186
187		1	static av_cold int targa_encode_init(AVCodecContext *avctx)
188			{
189	✓✗✗✓	1	if (avctx->width > 0xffff \|\| avctx->height > 0xffff) {
190			av_log(avctx, AV_LOG_ERROR, "image dimensions too large\n");
191			return AVERROR(EINVAL);
192			}
193
194			#if FF_API_CODED_FRAME
195			FF_DISABLE_DEPRECATION_WARNINGS
196		1	avctx->coded_frame->key_frame = 1;
197		1	avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
198			FF_ENABLE_DEPRECATION_WARNINGS
199			#endif
200
201		1	return 0;
202			}
203
204			#define OFFSET(x) offsetof(TargaContext, x)
205			#define VE AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM
206			static const AVOption options[] = {
207			{ "rle", "Use run-length compression", OFFSET(rle), AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, VE },
208
209			{ NULL },
210			};
211
212			static const AVClass targa_class = {
213			.class_name = "targa",
214			.item_name = av_default_item_name,
215			.option = options,
216			.version = LIBAVUTIL_VERSION_INT,
217			};
218
219			AVCodec ff_targa_encoder = {
220			.name = "targa",
221			.long_name = NULL_IF_CONFIG_SMALL("Truevision Targa image"),
222			.type = AVMEDIA_TYPE_VIDEO,
223			.id = AV_CODEC_ID_TARGA,
224			.priv_data_size = sizeof(TargaContext),
225			.priv_class = &targa_class,
226			.init = targa_encode_init,
227			.encode2 = targa_encode_frame,
228			.pix_fmts = (const enum AVPixelFormat[]){
229			AV_PIX_FMT_BGR24, AV_PIX_FMT_BGRA, AV_PIX_FMT_RGB555LE, AV_PIX_FMT_GRAY8, AV_PIX_FMT_PAL8,
230			AV_PIX_FMT_NONE
231			},
232			};