FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/ljpegenc.c
Date: 2024-11-20 23:03:26
Exec Total Coverage
Lines: 85 131 64.9%
Functions: 5 6 83.3%
Branches: 50 108 46.3%
Line Branch Exec Source
1 /*
2 * lossless JPEG encoder
3 * Copyright (c) 2000, 2001 Fabrice Bellard
4 * Copyright (c) 2003 Alex Beregszaszi
5 * Copyright (c) 2003-2004 Michael Niedermayer
6 *
7 * Support for external huffman table, various fixes (AVID workaround),
8 * aspecting, new decode_frame mechanism and apple mjpeg-b support
9 * by Alex Beregszaszi
10 *
11 * This file is part of FFmpeg.
12 *
13 * FFmpeg is free software; you can redistribute it and/or
14 * modify it under the terms of the GNU Lesser General Public
15 * License as published by the Free Software Foundation; either
16 * version 2.1 of the License, or (at your option) any later version.
17 *
18 * FFmpeg is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21 * Lesser General Public License for more details.
22 *
23 * You should have received a copy of the GNU Lesser General Public
24 * License along with FFmpeg; if not, write to the Free Software
25 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
26 */
27
28 /**
29 * @file
30 * lossless JPEG encoder.
31 */
32
33 #include "libavutil/frame.h"
34 #include "libavutil/mem.h"
35 #include "libavutil/opt.h"
36
37 #include "avcodec.h"
38 #include "codec_internal.h"
39 #include "encode.h"
40 #include "jpegtables.h"
41 #include "mjpegenc_common.h"
42 #include "mjpeg.h"
43
44 typedef struct LJpegEncContext {
45 AVClass *class;
46
47 int vsample[4];
48 int hsample[4];
49
50 uint16_t huff_code_dc_luminance[12];
51 uint16_t huff_code_dc_chrominance[12];
52 uint8_t huff_size_dc_luminance[12];
53 uint8_t huff_size_dc_chrominance[12];
54
55 uint16_t (*scratch)[4];
56 int pred;
57 } LJpegEncContext;
58
59 ✗ static int ljpeg_encode_bgr(AVCodecContext *avctx, PutBitContext *pb,
60 const AVFrame *frame)
61 {
62 ✗ LJpegEncContext *s = avctx->priv_data;
63 ✗ const int width = frame->width;
64 ✗ const int height = frame->height;
65 ✗ const int linesize = frame->linesize[0];
66 ✗ uint16_t (*buffer)[4] = s->scratch;
67 int left[4], top[4], topleft[4];
68 int x, y, i;
69
70 ✗ for (i = 0; i < 4; i++)
71 ✗ buffer[0][i] = 1 << (9 - 1);
72
73 ✗ for (y = 0; y < height; y++) {
74 ✗ const int modified_predictor = y ? s->pred : 1;
75 ✗ const uint8_t *ptr = frame->data[0] + (linesize * y);
76
77 ✗ if (put_bytes_left(pb, 0) < width * 4 * 4) {
78 ✗ av_log(avctx, AV_LOG_ERROR, "encoded frame too large\n");
79 ✗ return -1;
80 }
81
82 ✗ for (i = 0; i < 4; i++)
83 ✗ top[i]= left[i]= topleft[i]= buffer[0][i];
84
85 ✗ for (x = 0; x < width; x++) {
86 ✗ if(avctx->pix_fmt == AV_PIX_FMT_BGR24){
87 ✗ buffer[x][1] = ptr[3 * x + 0] - ptr[3 * x + 1] + 0x100;
88 ✗ buffer[x][2] = ptr[3 * x + 2] - ptr[3 * x + 1] + 0x100;
89 ✗ buffer[x][0] = (ptr[3 * x + 0] + 2 * ptr[3 * x + 1] + ptr[3 * x + 2]) >> 2;
90 }else{
91 ✗ buffer[x][1] = ptr[4 * x + 0] - ptr[4 * x + 1] + 0x100;
92 ✗ buffer[x][2] = ptr[4 * x + 2] - ptr[4 * x + 1] + 0x100;
93 ✗ buffer[x][0] = (ptr[4 * x + 0] + 2 * ptr[4 * x + 1] + ptr[4 * x + 2]) >> 2;
94 ✗ if (avctx->pix_fmt == AV_PIX_FMT_BGRA)
95 ✗ buffer[x][3] = ptr[4 * x + 3];
96 }
97
98 ✗ for (i = 0; i < 3 + (avctx->pix_fmt == AV_PIX_FMT_BGRA); i++) {
99 int pred, diff;
100
101 ✗ PREDICT(pred, topleft[i], top[i], left[i], modified_predictor);
102
103 ✗ topleft[i] = top[i];
104 ✗ top[i] = buffer[x+1][i];
105
106 ✗ left[i] = buffer[x][i];
107
108 ✗ diff = ((left[i] - pred + 0x100) & 0x1FF) - 0x100;
109
110 ✗ if (i == 0 || i == 3)
111 ✗ ff_mjpeg_encode_dc(pb, diff, s->huff_size_dc_luminance, s->huff_code_dc_luminance); //FIXME ugly
112 else
113 ✗ ff_mjpeg_encode_dc(pb, diff, s->huff_size_dc_chrominance, s->huff_code_dc_chrominance);
114 }
115 }
116 }
117
118 ✗ return 0;
119 }
120
121 3816050 static inline void ljpeg_encode_yuv_mb(LJpegEncContext *s, PutBitContext *pb,
122 const AVFrame *frame, int predictor,
123 int mb_x, int mb_y)
124 {
125 int i;
126
127
4/4
✓ Branch 0 taken 3793600 times.
✓ Branch 1 taken 22450 times.
✓ Branch 2 taken 27050 times.
✓ Branch 3 taken 3766550 times.
 3816050 if (mb_x == 0 || mb_y == 0) {
128
2/2
✓ Branch 0 taken 148500 times.
✓ Branch 1 taken 49500 times.
 198000 for (i = 0; i < 3; i++) {
129 const uint8_t *ptr;
130 int x, y, h, v, linesize;
131 148500 h = s->hsample[i];
132 148500 v = s->vsample[i];
133 148500 linesize = frame->linesize[i];
134
135
2/2
✓ Branch 0 taken 198000 times.
✓ Branch 1 taken 148500 times.
 346500 for (y = 0; y < v; y++) {
136
2/2
✓ Branch 0 taken 297000 times.
✓ Branch 1 taken 198000 times.
 495000 for (x = 0; x < h; x++) {
137 int pred;
138
139 297000 ptr = frame->data[i] + (linesize * (v * mb_y + y)) + (h * mb_x + x); //FIXME optimize this crap
140
4/4
✓ Branch 0 taken 198000 times.
✓ Branch 1 taken 99000 times.
✓ Branch 2 taken 109000 times.
✓ Branch 3 taken 89000 times.
 297000 if (y == 0 && mb_y == 0) {
141
4/4
✓ Branch 0 taken 81750 times.
✓ Branch 1 taken 27250 times.
✓ Branch 2 taken 600 times.
✓ Branch 3 taken 81150 times.
 109000 if (x == 0 && mb_x == 0)
142 600 pred = 128;
143 else
144 108400 pred = ptr[-1];
145 } else {
146
4/4
✓ Branch 0 taken 116250 times.
✓ Branch 1 taken 71750 times.
✓ Branch 2 taken 89200 times.
✓ Branch 3 taken 27050 times.
 188000 if (x == 0 && mb_x == 0) {
147 89200 pred = ptr[-linesize];
148 } else {
149
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 98800 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 98800 PREDICT(pred, ptr[-linesize - 1], ptr[-linesize],
150 ptr[-1], predictor);
151 }
152 }
153
154
2/2
✓ Branch 0 taken 198000 times.
✓ Branch 1 taken 99000 times.
 297000 if (i == 0)
155 198000 ff_mjpeg_encode_dc(pb, *ptr - pred, s->huff_size_dc_luminance, s->huff_code_dc_luminance); //FIXME ugly
156 else
157 99000 ff_mjpeg_encode_dc(pb, *ptr - pred, s->huff_size_dc_chrominance, s->huff_code_dc_chrominance);
158 }
159 }
160 }
161 } else {
162
2/2
✓ Branch 0 taken 11299650 times.
✓ Branch 1 taken 3766550 times.
 15066200 for (i = 0; i < 3; i++) {
163 const uint8_t *ptr;
164 int x, y, h, v, linesize;
165 11299650 h = s->hsample[i];
166 11299650 v = s->vsample[i];
167 11299650 linesize = frame->linesize[i];
168
169
2/2
✓ Branch 0 taken 15066200 times.
✓ Branch 1 taken 11299650 times.
 26365850 for (y = 0; y < v; y++) {
170
2/2
✓ Branch 0 taken 22599300 times.
✓ Branch 1 taken 15066200 times.
 37665500 for (x = 0; x < h; x++) {
171 int pred;
172
173 22599300 ptr = frame->data[i] + (linesize * (v * mb_y + y)) + (h * mb_x + x); //FIXME optimize this crap
174
1/8
✗ Branch 0 not taken.
✓ Branch 1 taken 22599300 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
 22599300 PREDICT(pred, ptr[-linesize - 1], ptr[-linesize], ptr[-1], predictor);
175
176
2/2
✓ Branch 0 taken 15066200 times.
✓ Branch 1 taken 7533100 times.
 22599300 if (i == 0)
177 15066200 ff_mjpeg_encode_dc(pb, *ptr - pred, s->huff_size_dc_luminance, s->huff_code_dc_luminance); //FIXME ugly
178 else
179 7533100 ff_mjpeg_encode_dc(pb, *ptr - pred, s->huff_size_dc_chrominance, s->huff_code_dc_chrominance);
180 }
181 }
182 }
183 }
184 3816050 }
185
186 200 static int ljpeg_encode_yuv(AVCodecContext *avctx, PutBitContext *pb,
187 const AVFrame *frame)
188 {
189 200 LJpegEncContext *s = avctx->priv_data;
190 200 const int mb_width = (avctx->width + s->hsample[0] - 1) / s->hsample[0];
191 200 const int mb_height = (avctx->height + s->vsample[0] - 1) / s->vsample[0];
192 int mb_x, mb_y;
193
194
2/2
✓ Branch 0 taken 22450 times.
✓ Branch 1 taken 200 times.
 22650 for (mb_y = 0; mb_y < mb_height; mb_y++) {
195 22450 if (put_bytes_left(pb, 0) <
196
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22450 times.
 22450 mb_width * 4 * 3 * s->hsample[0] * s->vsample[0]) {
197 ✗ av_log(avctx, AV_LOG_ERROR, "encoded frame too large\n");
198 ✗ return -1;
199 }
200
201
2/2
✓ Branch 0 taken 3816050 times.
✓ Branch 1 taken 22450 times.
 3838500 for (mb_x = 0; mb_x < mb_width; mb_x++)
202 3816050 ljpeg_encode_yuv_mb(s, pb, frame, s->pred, mb_x, mb_y);
203 }
204
205 200 return 0;
206 }
207
208 200 static int ljpeg_encode_frame(AVCodecContext *avctx, AVPacket *pkt,
209 const AVFrame *pict, int *got_packet)
210 {
211 200 LJpegEncContext *s = avctx->priv_data;
212 PutBitContext pb;
213 200 const int width = avctx->width;
214 200 const int height = avctx->height;
215 200 const int mb_width = (width + s->hsample[0] - 1) / s->hsample[0];
216 200 const int mb_height = (height + s->vsample[0] - 1) / s->vsample[0];
217 200 size_t max_pkt_size = FF_INPUT_BUFFER_MIN_SIZE;
218 int ret, header_bits;
219
220
1/2
✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
 200 if( avctx->pix_fmt == AV_PIX_FMT_BGR0
221
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
 200 || avctx->pix_fmt == AV_PIX_FMT_BGR24)
222 ✗ max_pkt_size += width * height * 3 * 4;
223
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
 200 else if(avctx->pix_fmt == AV_PIX_FMT_BGRA)
224 ✗ max_pkt_size += width * height * 4 * 4;
225 else {
226 200 max_pkt_size += mb_width * mb_height * 3 * 4
227 200 * s->hsample[0] * s->vsample[0];
228 }
229
230
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 200 times.
 200 if ((ret = ff_mjpeg_add_icc_profile_size(avctx, pict, &max_pkt_size)) < 0)
231 ✗ return ret;
232
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 200 times.
 200 if ((ret = ff_alloc_packet(avctx, pkt, max_pkt_size)) < 0)
233 ✗ return ret;
234
235 200 init_put_bits(&pb, pkt->data, pkt->size);
236
237 200 ff_mjpeg_encode_picture_header(avctx, &pb, pict, NULL, NULL,
238 s->pred, NULL, NULL, 0);
239
240 200 header_bits = put_bits_count(&pb);
241
242
1/2
✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
 200 if( avctx->pix_fmt == AV_PIX_FMT_BGR0
243
1/2
✓ Branch 0 taken 200 times.
✗ Branch 1 not taken.
 200 || avctx->pix_fmt == AV_PIX_FMT_BGRA
244
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
 200 || avctx->pix_fmt == AV_PIX_FMT_BGR24)
245 ✗ ret = ljpeg_encode_bgr(avctx, &pb, pict);
246 else
247 200 ret = ljpeg_encode_yuv(avctx, &pb, pict);
248
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 200 times.
 200 if (ret < 0)
249 ✗ return ret;
250
251 200 ff_mjpeg_escape_FF(&pb, header_bits >> 3);
252 200 ff_mjpeg_encode_picture_trailer(&pb, header_bits);
253
254 200 flush_put_bits(&pb);
255 200 pkt->size = put_bits_ptr(&pb) - pb.buf;
256 200 *got_packet = 1;
257
258 200 return 0;
259 }
260
261 4 static av_cold int ljpeg_encode_close(AVCodecContext *avctx)
262 {
263 4 LJpegEncContext *s = avctx->priv_data;
264
265 4 av_freep(&s->scratch);
266
267 4 return 0;
268 }
269
270 4 static av_cold int ljpeg_encode_init(AVCodecContext *avctx)
271 {
272 4 int ret = ff_mjpeg_encode_check_pix_fmt(avctx);
273 4 LJpegEncContext *s = avctx->priv_data;
274
275
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (ret < 0)
276 ✗ return ret;
277
278 4 s->scratch = av_malloc_array(avctx->width + 1, sizeof(*s->scratch));
279
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (!s->scratch)
280 ✗ return AVERROR(ENOMEM);
281
282 4 ff_mjpeg_init_hvsample(avctx, s->hsample, s->vsample);
283
284 4 ff_mjpeg_build_huffman_codes(s->huff_size_dc_luminance,
285 4 s->huff_code_dc_luminance,
286 ff_mjpeg_bits_dc_luminance,
287 ff_mjpeg_val_dc);
288 4 ff_mjpeg_build_huffman_codes(s->huff_size_dc_chrominance,
289 4 s->huff_code_dc_chrominance,
290 ff_mjpeg_bits_dc_chrominance,
291 ff_mjpeg_val_dc);
292
293 4 return 0;
294 }
295
296 #define OFFSET(x) offsetof(LJpegEncContext, x)
297 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
298 static const AVOption options[] = {
299 { "pred", "Prediction method", OFFSET(pred), AV_OPT_TYPE_INT, { .i64 = 1 }, 1, 3, VE, .unit = "pred" },
300 { "left", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = 1 }, INT_MIN, INT_MAX, VE, .unit = "pred" },
301 { "plane", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = 2 }, INT_MIN, INT_MAX, VE, .unit = "pred" },
302 { "median", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = 3 }, INT_MIN, INT_MAX, VE, .unit = "pred" },
303
304 { NULL},
305 };
306
307 static const AVClass ljpeg_class = {
308 .class_name = "ljpeg",
309 .item_name = av_default_item_name,
310 .option = options,
311 .version = LIBAVUTIL_VERSION_INT,
312 };
313
314 const FFCodec ff_ljpeg_encoder = {
315 .p.name = "ljpeg",
316 CODEC_LONG_NAME("Lossless JPEG"),
317 .p.type = AVMEDIA_TYPE_VIDEO,
318 .p.id = AV_CODEC_ID_LJPEG,
319 .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_FRAME_THREADS |
320 AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
321 .priv_data_size = sizeof(LJpegEncContext),
322 .p.priv_class = &ljpeg_class,
323 .init = ljpeg_encode_init,
324 FF_CODEC_ENCODE_CB(ljpeg_encode_frame),
325 .close = ljpeg_encode_close,
326 .p.pix_fmts = (const enum AVPixelFormat[]){
327 AV_PIX_FMT_BGR24 , AV_PIX_FMT_BGRA , AV_PIX_FMT_BGR0,
328 AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ422P,
329 AV_PIX_FMT_YUV420P , AV_PIX_FMT_YUV444P , AV_PIX_FMT_YUV422P,
330 AV_PIX_FMT_NONE},
331 .color_ranges = AVCOL_RANGE_MPEG | AVCOL_RANGE_JPEG,
332 };
333