FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavcodec/mpegvideo_enc.c
Date: 2022-07-07 01:21:54
Exec Total Coverage
Lines: 1928 2788 69.2%
Branches: 1120 1818 61.6%
Line Branch Exec Source
1 /*
2 * The simplest mpeg encoder (well, it was the simplest!)
3 * Copyright (c) 2000,2001 Fabrice Bellard
4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
5 *
6 * 4MV & hq & B-frame encoding stuff by Michael Niedermayer <michaelni@gmx.at>
7 *
8 * This file is part of FFmpeg.
9 *
10 * FFmpeg is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * FFmpeg is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with FFmpeg; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 */
24
25 /*
26 * non linear quantizers with large QPs and VBV with restrictive qmin fixes sponsored by NOA GmbH
27 */
28
29 /**
30 * @file
31 * The simplest mpeg encoder (well, it was the simplest!).
32 */
33
34 #include "config_components.h"
35
36 #include <stdint.h>
37
38 #include "libavutil/internal.h"
39 #include "libavutil/intmath.h"
40 #include "libavutil/mathematics.h"
41 #include "libavutil/mem_internal.h"
42 #include "libavutil/pixdesc.h"
43 #include "libavutil/opt.h"
44 #include "libavutil/thread.h"
45 #include "avcodec.h"
46 #include "dct.h"
47 #include "encode.h"
48 #include "idctdsp.h"
49 #include "mpeg12.h"
50 #include "mpeg12data.h"
51 #include "mpeg12enc.h"
52 #include "mpegvideo.h"
53 #include "mpegvideodata.h"
54 #include "mpegvideoenc.h"
55 #include "h261enc.h"
56 #include "h263.h"
57 #include "h263data.h"
58 #include "h263enc.h"
59 #include "mjpegenc_common.h"
60 #include "mathops.h"
61 #include "mpegutils.h"
62 #include "mjpegenc.h"
63 #include "speedhqenc.h"
64 #include "msmpeg4enc.h"
65 #include "pixblockdsp.h"
66 #include "qpeldsp.h"
67 #include "faandct.h"
68 #include "aandcttab.h"
69 #include "flvenc.h"
70 #include "mpeg4video.h"
71 #include "mpeg4videodata.h"
72 #include "mpeg4videoenc.h"
73 #include "internal.h"
74 #include "bytestream.h"
75 #include "wmv2enc.h"
76 #include "rv10enc.h"
77 #include "packet_internal.h"
78 #include <limits.h>
79 #include "sp5x.h"
80
81 #define QUANT_BIAS_SHIFT 8
82
83 #define QMAT_SHIFT_MMX 16
84 #define QMAT_SHIFT 21
85
86 static int encode_picture(MpegEncContext *s, int picture_number);
87 static int dct_quantize_refine(MpegEncContext *s, int16_t *block, int16_t *weight, int16_t *orig, int n, int qscale);
88 static int sse_mb(MpegEncContext *s);
89 static void denoise_dct_c(MpegEncContext *s, int16_t *block);
90 static int dct_quantize_trellis_c(MpegEncContext *s, int16_t *block, int n, int qscale, int *overflow);
91
92 static uint8_t default_mv_penalty[MAX_FCODE + 1][MAX_DMV * 2 + 1];
93 static uint8_t default_fcode_tab[MAX_MV * 2 + 1];
94
95 static const AVOption mpv_generic_options[] = {
96 FF_MPV_COMMON_OPTS
97 FF_MPV_COMMON_MOTION_EST_OPTS
98 { NULL },
99 };
100
101 const AVClass ff_mpv_enc_class = {
102 .class_name = "generic mpegvideo encoder",
103 .item_name = av_default_item_name,
104 .option = mpv_generic_options,
105 .version = LIBAVUTIL_VERSION_INT,
106 };
107
108 3722 void ff_convert_matrix(MpegEncContext *s, int (*qmat)[64],
109 uint16_t (*qmat16)[2][64],
110 const uint16_t *quant_matrix,
111 int bias, int qmin, int qmax, int intra)
112 {
113 3722 FDCTDSPContext *fdsp = &s->fdsp;
114 int qscale;
115 3722 int shift = 0;
116
117
2/2
✓ Branch 0 taken 97008 times.
✓ Branch 1 taken 3722 times.
 100730 for (qscale = qmin; qscale <= qmax; qscale++) {
118 int i;
119 int qscale2;
120
121
2/2
✓ Branch 0 taken 62 times.
✓ Branch 1 taken 96946 times.
 97008 if (s->q_scale_type) qscale2 = ff_mpeg2_non_linear_qscale[qscale];
122 96946 else qscale2 = qscale << 1;
123
124
2/2
✓ Branch 0 taken 96708 times.
✓ Branch 1 taken 300 times.
 97008 if (fdsp->fdct == ff_jpeg_fdct_islow_8 ||
125 #if CONFIG_FAANDCT
126
1/2
✓ Branch 0 taken 96708 times.
✗ Branch 1 not taken.
 96708 fdsp->fdct == ff_faandct ||
127 #endif /* CONFIG_FAANDCT */
128
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 96708 times.
 96708 fdsp->fdct == ff_jpeg_fdct_islow_10) {
129
2/2
✓ Branch 0 taken 19200 times.
✓ Branch 1 taken 300 times.
 19500 for (i = 0; i < 64; i++) {
130 19200 const int j = s->idsp.idct_permutation[i];
131 19200 int64_t den = (int64_t) qscale2 * quant_matrix[j];
132 /* 16 <= qscale * quant_matrix[i] <= 7905
133 * Assume x = ff_aanscales[i] * qscale * quant_matrix[i]
134 * 19952 <= x <= 249205026
135 * (1 << 36) / 19952 >= (1 << 36) / (x) >= (1 << 36) / 249205026
136 * 3444240 >= (1 << 36) / (x) >= 275 */
137
138 19200 qmat[qscale][i] = (int)((UINT64_C(2) << QMAT_SHIFT) / den);
139 }
140
1/2
✓ Branch 0 taken 96708 times.
✗ Branch 1 not taken.
 96708 } else if (fdsp->fdct == ff_fdct_ifast) {
141
2/2
✓ Branch 0 taken 6189312 times.
✓ Branch 1 taken 96708 times.
 6286020 for (i = 0; i < 64; i++) {
142 6189312 const int j = s->idsp.idct_permutation[i];
143 6189312 int64_t den = ff_aanscales[i] * (int64_t) qscale2 * quant_matrix[j];
144 /* 16 <= qscale * quant_matrix[i] <= 7905
145 * Assume x = ff_aanscales[i] * qscale * quant_matrix[i]
146 * 19952 <= x <= 249205026
147 * (1 << 36) / 19952 >= (1 << 36) / (x) >= (1 << 36) / 249205026
148 * 3444240 >= (1 << 36) / (x) >= 275 */
149
150 6189312 qmat[qscale][i] = (int)((UINT64_C(2) << (QMAT_SHIFT + 14)) / den);
151 }
152 } else {
153 ✗ for (i = 0; i < 64; i++) {
154 ✗ const int j = s->idsp.idct_permutation[i];
155 ✗ int64_t den = (int64_t) qscale2 * quant_matrix[j];
156 /* We can safely suppose that 16 <= quant_matrix[i] <= 255
157 * Assume x = qscale * quant_matrix[i]
158 * So 16 <= x <= 7905
159 * so (1 << 19) / 16 >= (1 << 19) / (x) >= (1 << 19) / 7905
160 * so 32768 >= (1 << 19) / (x) >= 67 */
161 ✗ qmat[qscale][i] = (int)((UINT64_C(2) << QMAT_SHIFT) / den);
162 //qmat [qscale][i] = (1 << QMAT_SHIFT_MMX) /
163 // (qscale * quant_matrix[i]);
164 ✗ qmat16[qscale][0][i] = (2 << QMAT_SHIFT_MMX) / den;
165
166 ✗ if (qmat16[qscale][0][i] == 0 ||
167 ✗ qmat16[qscale][0][i] == 128 * 256)
168 ✗ qmat16[qscale][0][i] = 128 * 256 - 1;
169 ✗ qmat16[qscale][1][i] =
170 ✗ ROUNDED_DIV(bias * (1<<(16 - QUANT_BIAS_SHIFT)),
171 qmat16[qscale][0][i]);
172 }
173 }
174
175
2/2
✓ Branch 0 taken 6116185 times.
✓ Branch 1 taken 97008 times.
 6213193 for (i = intra; i < 64; i++) {
176 6116185 int64_t max = 8191;
177
2/2
✓ Branch 0 taken 6097135 times.
✓ Branch 1 taken 19050 times.
 6116185 if (fdsp->fdct == ff_fdct_ifast) {
178 6097135 max = (8191LL * ff_aanscales[i]) >> 14;
179 }
180
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6116185 times.
 6116185 while (((max * qmat[qscale][i]) >> shift) > INT_MAX) {
181 ✗ shift++;
182 }
183 }
184 }
185
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3722 times.
 3722 if (shift) {
186 ✗ av_log(s->avctx, AV_LOG_INFO,
187 "Warning, QMAT_SHIFT is larger than %d, overflows possible\n",
188 QMAT_SHIFT - shift);
189 }
190 3722 }
191
192 1282204 static inline void update_qscale(MpegEncContext *s)
193 {
194 if (s->q_scale_type == 1 && 0) {
195 int i;
196 int bestdiff=INT_MAX;
197 int best = 1;
198
199 for (i = 0 ; i<FF_ARRAY_ELEMS(ff_mpeg2_non_linear_qscale); i++) {
200 int diff = FFABS((ff_mpeg2_non_linear_qscale[i]<<(FF_LAMBDA_SHIFT + 6)) - (int)s->lambda * 139);
201 if (ff_mpeg2_non_linear_qscale[i] < s->avctx->qmin ||
202 (ff_mpeg2_non_linear_qscale[i] > s->avctx->qmax && !s->vbv_ignore_qmax))
203 continue;
204 if (diff < bestdiff) {
205 bestdiff = diff;
206 best = i;
207 }
208 }
209 s->qscale = best;
210 } else {
211 1282204 s->qscale = (s->lambda * 139 + FF_LAMBDA_SCALE * 64) >>
212 (FF_LAMBDA_SHIFT + 7);
213
2/2
✓ Branch 0 taken 1282079 times.
✓ Branch 1 taken 125 times.
 1282204 s->qscale = av_clip(s->qscale, s->avctx->qmin, s->vbv_ignore_qmax ? 31 : s->avctx->qmax);
214 }
215
216 1282204 s->lambda2 = (s->lambda * s->lambda + FF_LAMBDA_SCALE / 2) >>
217 FF_LAMBDA_SHIFT;
218 1282204 }
219
220 784 void ff_write_quant_matrix(PutBitContext *pb, uint16_t *matrix)
221 {
222 int i;
223
224
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 784 times.
 784 if (matrix) {
225 ✗ put_bits(pb, 1, 1);
226 ✗ for (i = 0; i < 64; i++) {
227 ✗ put_bits(pb, 8, matrix[ff_zigzag_direct[i]]);
228 }
229 } else
230 784 put_bits(pb, 1, 0);
231 784 }
232
233 /**
234 * init s->current_picture.qscale_table from s->lambda_table
235 */
236 800 void ff_init_qscale_tab(MpegEncContext *s)
237 {
238 800 int8_t * const qscale_table = s->current_picture.qscale_table;
239 int i;
240
241
2/2
✓ Branch 0 taken 239550 times.
✓ Branch 1 taken 800 times.
 240350 for (i = 0; i < s->mb_num; i++) {
242 239550 unsigned int lam = s->lambda_table[s->mb_index2xy[i]];
243 239550 int qp = (lam * 139 + FF_LAMBDA_SCALE * 64) >> (FF_LAMBDA_SHIFT + 7);
244 239550 qscale_table[s->mb_index2xy[i]] = av_clip(qp, s->avctx->qmin,
245 239550 s->avctx->qmax);
246 }
247 800 }
248
249 600 static void update_duplicate_context_after_me(MpegEncContext *dst,
250 const MpegEncContext *src)
251 {
252 #define COPY(a) dst->a= src->a
253 600 COPY(pict_type);
254 600 COPY(current_picture);
255 600 COPY(f_code);
256 600 COPY(b_code);
257 600 COPY(qscale);
258 600 COPY(lambda);
259 600 COPY(lambda2);
260 600 COPY(frame_pred_frame_dct); // FIXME don't set in encode_header
261 600 COPY(progressive_frame); // FIXME don't set in encode_header
262 600 COPY(partitioned_frame); // FIXME don't set in encode_header
263 #undef COPY
264 600 }
265
266 187 static void mpv_encode_init_static(void)
267 {
268
2/2
✓ Branch 0 taken 5984 times.
✓ Branch 1 taken 187 times.
 6171 for (int i = -16; i < 16; i++)
269 5984 default_fcode_tab[i + MAX_MV] = 1;
270 187 }
271
272 /**
273 * Set the given MpegEncContext to defaults for encoding.
274 * the changed fields will not depend upon the prior state of the MpegEncContext.
275 */
276 187 static void mpv_encode_defaults(MpegEncContext *s)
277 {
278 static AVOnce init_static_once = AV_ONCE_INIT;
279
280 187 ff_mpv_common_defaults(s);
281
282 187 ff_thread_once(&init_static_once, mpv_encode_init_static);
283
284 187 s->me.mv_penalty = default_mv_penalty;
285 187 s->fcode_tab = default_fcode_tab;
286
287 187 s->input_picture_number = 0;
288 187 s->picture_in_gop_number = 0;
289 187 }
290
291 260 av_cold int ff_dct_encode_init(MpegEncContext *s)
292 {
293 #if ARCH_X86
294 260 ff_dct_encode_init_x86(s);
295 #endif
296
297 if (CONFIG_H263_ENCODER)
298 260 ff_h263dsp_init(&s->h263dsp);
299
1/2
✓ Branch 0 taken 260 times.
✗ Branch 1 not taken.
 260 if (!s->dct_quantize)
300 260 s->dct_quantize = ff_dct_quantize_c;
301
1/2
✓ Branch 0 taken 260 times.
✗ Branch 1 not taken.
 260 if (!s->denoise_dct)
302 260 s->denoise_dct = denoise_dct_c;
303 260 s->fast_dct_quantize = s->dct_quantize;
304
2/2
✓ Branch 0 taken 35 times.
✓ Branch 1 taken 225 times.
 260 if (s->avctx->trellis)
305 35 s->dct_quantize = dct_quantize_trellis_c;
306
307 260 return 0;
308 }
309
310 /* init video encoder */
311 187 av_cold int ff_mpv_encode_init(AVCodecContext *avctx)
312 {
313 187 MpegEncContext *s = avctx->priv_data;
314 AVCPBProperties *cpb_props;
315 int i, ret;
316
317 187 mpv_encode_defaults(s);
318
319
3/3
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 9 times.
✓ Branch 2 taken 173 times.
 187 switch (avctx->pix_fmt) {
320 5 case AV_PIX_FMT_YUVJ444P:
321 case AV_PIX_FMT_YUV444P:
322 5 s->chroma_format = CHROMA_444;
323 5 break;
324 9 case AV_PIX_FMT_YUVJ422P:
325 case AV_PIX_FMT_YUV422P:
326 9 s->chroma_format = CHROMA_422;
327 9 break;
328 173 case AV_PIX_FMT_YUVJ420P:
329 case AV_PIX_FMT_YUV420P:
330 default:
331 173 s->chroma_format = CHROMA_420;
332 173 break;
333 }
334
335 187 avctx->bits_per_raw_sample = av_clip(avctx->bits_per_raw_sample, 0, 8);
336
337 187 s->bit_rate = avctx->bit_rate;
338 187 s->width = avctx->width;
339 187 s->height = avctx->height;
340
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if (avctx->gop_size > 600 &&
341 ✗ avctx->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
342 ✗ av_log(avctx, AV_LOG_WARNING,
343 "keyframe interval too large!, reducing it from %d to %d\n",
344 avctx->gop_size, 600);
345 ✗ avctx->gop_size = 600;
346 }
347 187 s->gop_size = avctx->gop_size;
348 187 s->avctx = avctx;
349
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if (avctx->max_b_frames > MAX_B_FRAMES) {
350 ✗ av_log(avctx, AV_LOG_ERROR, "Too many B-frames requested, maximum "
351 "is %d.\n", MAX_B_FRAMES);
352 ✗ avctx->max_b_frames = MAX_B_FRAMES;
353
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 } else if (avctx->max_b_frames < 0) {
354 ✗ av_log(avctx, AV_LOG_ERROR,
355 "max b frames must be 0 or positive for mpegvideo based encoders\n");
356 ✗ return AVERROR(EINVAL);
357 }
358 187 s->max_b_frames = avctx->max_b_frames;
359 187 s->codec_id = avctx->codec->id;
360
3/4
✓ Branch 0 taken 43 times.
✓ Branch 1 taken 144 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 43 times.
 187 if (s->max_b_frames && !(avctx->codec->capabilities & AV_CODEC_CAP_DELAY)) {
361 ✗ av_log(avctx, AV_LOG_ERROR, "B-frames not supported by codec\n");
362 ✗ return AVERROR(EINVAL);
363 }
364
365 187 s->quarter_sample = (avctx->flags & AV_CODEC_FLAG_QPEL) != 0;
366 187 s->rtp_mode = !!s->rtp_payload_size;
367 187 s->intra_dc_precision = avctx->intra_dc_precision;
368
369 // workaround some differences between how applications specify dc precision
370
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if (s->intra_dc_precision < 0) {
371 ✗ s->intra_dc_precision += 8;
372
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 186 times.
 187 } else if (s->intra_dc_precision >= 8)
373 1 s->intra_dc_precision -= 8;
374
375
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if (s->intra_dc_precision < 0) {
376 ✗ av_log(avctx, AV_LOG_ERROR,
377 "intra dc precision must be positive, note some applications use"
378 " 0 and some 8 as base meaning 8bit, the value must not be smaller than that\n");
379 ✗ return AVERROR(EINVAL);
380 }
381
382
3/4
✓ Branch 0 taken 39 times.
✓ Branch 1 taken 148 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 187 times.
 187 if (s->intra_dc_precision > (avctx->codec_id == AV_CODEC_ID_MPEG2VIDEO ? 3 : 0)) {
383 ✗ av_log(avctx, AV_LOG_ERROR, "intra dc precision too large\n");
384 ✗ return AVERROR(EINVAL);
385 }
386 187 s->user_specified_pts = AV_NOPTS_VALUE;
387
388
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 186 times.
 187 if (s->gop_size <= 1) {
389 1 s->intra_only = 1;
390 1 s->gop_size = 12;
391 } else {
392 186 s->intra_only = 0;
393 }
394
395 /* Fixed QSCALE */
396 187 s->fixed_qscale = !!(avctx->flags & AV_CODEC_FLAG_QSCALE);
397
398 561 s->adaptive_quant = (avctx->lumi_masking ||
399
1/2
✓ Branch 0 taken 187 times.
✗ Branch 1 not taken.
 187 avctx->dark_masking ||
400
1/2
✓ Branch 0 taken 187 times.
✗ Branch 1 not taken.
 187 avctx->temporal_cplx_masking ||
401
2/2
✓ Branch 0 taken 183 times.
✓ Branch 1 taken 4 times.
 187 avctx->spatial_cplx_masking ||
402
1/2
✓ Branch 0 taken 183 times.
✗ Branch 1 not taken.
 183 avctx->p_masking ||
403
1/2
✓ Branch 0 taken 183 times.
✗ Branch 1 not taken.
 183 s->border_masking ||
404
3/4
✓ Branch 0 taken 187 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 12 times.
✓ Branch 3 taken 171 times.
 386 (s->mpv_flags & FF_MPV_FLAG_QP_RD)) &&
405
1/2
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
 16 !s->fixed_qscale;
406
407 187 s->loop_filter = !!(avctx->flags & AV_CODEC_FLAG_LOOP_FILTER);
408
409
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 185 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
 187 if (avctx->rc_max_rate && !avctx->rc_buffer_size) {
410 ✗ switch(avctx->codec_id) {
411 ✗ case AV_CODEC_ID_MPEG1VIDEO:
412 case AV_CODEC_ID_MPEG2VIDEO:
413 ✗ avctx->rc_buffer_size = FFMAX(avctx->rc_max_rate, 15000000) * 112LL / 15000000 * 16384;
414 ✗ break;
415 ✗ case AV_CODEC_ID_MPEG4:
416 case AV_CODEC_ID_MSMPEG4V1:
417 case AV_CODEC_ID_MSMPEG4V2:
418 case AV_CODEC_ID_MSMPEG4V3:
419 ✗ if (avctx->rc_max_rate >= 15000000) {
420 ✗ avctx->rc_buffer_size = 320 + (avctx->rc_max_rate - 15000000LL) * (760-320) / (38400000 - 15000000);
421 ✗ } else if(avctx->rc_max_rate >= 2000000) {
422 ✗ avctx->rc_buffer_size = 80 + (avctx->rc_max_rate - 2000000LL) * (320- 80) / (15000000 - 2000000);
423 ✗ } else if(avctx->rc_max_rate >= 384000) {
424 ✗ avctx->rc_buffer_size = 40 + (avctx->rc_max_rate - 384000LL) * ( 80- 40) / ( 2000000 - 384000);
425 } else
426 ✗ avctx->rc_buffer_size = 40;
427 ✗ avctx->rc_buffer_size *= 16384;
428 ✗ break;
429 }
430 ✗ if (avctx->rc_buffer_size) {
431 ✗ av_log(avctx, AV_LOG_INFO, "Automatically choosing VBV buffer size of %d kbyte\n", avctx->rc_buffer_size/8192);
432 }
433 }
434
435
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if ((!avctx->rc_max_rate) != (!avctx->rc_buffer_size)) {
436 ✗ av_log(avctx, AV_LOG_ERROR, "Either both buffer size and max rate or neither must be specified\n");
437 ✗ return AVERROR(EINVAL);
438 }
439
440
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 185 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
 187 if (avctx->rc_min_rate && avctx->rc_max_rate != avctx->rc_min_rate) {
441 ✗ av_log(avctx, AV_LOG_INFO,
442 "Warning min_rate > 0 but min_rate != max_rate isn't recommended!\n");
443 }
444
445
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 185 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
 187 if (avctx->rc_min_rate && avctx->rc_min_rate > avctx->bit_rate) {
446 ✗ av_log(avctx, AV_LOG_ERROR, "bitrate below min bitrate\n");
447 ✗ return AVERROR(EINVAL);
448 }
449
450
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 185 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
 187 if (avctx->rc_max_rate && avctx->rc_max_rate < avctx->bit_rate) {
451 ✗ av_log(avctx, AV_LOG_ERROR, "bitrate above max bitrate\n");
452 ✗ return AVERROR(EINVAL);
453 }
454
455
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 185 times.
 187 if (avctx->rc_max_rate &&
456
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 avctx->rc_max_rate == avctx->bit_rate &&
457
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 avctx->rc_max_rate != avctx->rc_min_rate) {
458 ✗ av_log(avctx, AV_LOG_INFO,
459 "impossible bitrate constraints, this will fail\n");
460 }
461
462
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 185 times.
 187 if (avctx->rc_buffer_size &&
463 2 avctx->bit_rate * (int64_t)avctx->time_base.num >
464
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 avctx->rc_buffer_size * (int64_t)avctx->time_base.den) {
465 ✗ av_log(avctx, AV_LOG_ERROR, "VBV buffer too small for bitrate\n");
466 ✗ return AVERROR(EINVAL);
467 }
468
469
2/2
✓ Branch 0 taken 35 times.
✓ Branch 1 taken 152 times.
 187 if (!s->fixed_qscale &&
470 35 avctx->bit_rate * av_q2d(avctx->time_base) >
471
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35 times.
 35 avctx->bit_rate_tolerance) {
472 ✗ double nbt = avctx->bit_rate * av_q2d(avctx->time_base) * 5;
473 ✗ av_log(avctx, AV_LOG_WARNING,
474 "bitrate tolerance %d too small for bitrate %"PRId64", overriding\n", avctx->bit_rate_tolerance, avctx->bit_rate);
475 ✗ if (nbt <= INT_MAX) {
476 ✗ avctx->bit_rate_tolerance = nbt;
477 } else
478 ✗ avctx->bit_rate_tolerance = INT_MAX;
479 }
480
481
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 185 times.
 187 if (avctx->rc_max_rate &&
482
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 avctx->rc_min_rate == avctx->rc_max_rate &&
483
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 (s->codec_id == AV_CODEC_ID_MPEG1VIDEO ||
484
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 s->codec_id == AV_CODEC_ID_MPEG2VIDEO) &&
485 2 90000LL * (avctx->rc_buffer_size - 1) >
486
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
 2 avctx->rc_max_rate * 0xFFFFLL) {
487 1 av_log(avctx, AV_LOG_INFO,
488 "Warning vbv_delay will be set to 0xFFFF (=VBR) as the "
489 "specified vbv buffer is too large for the given bitrate!\n");
490 }
491
492
3/4
✓ Branch 0 taken 32 times.
✓ Branch 1 taken 155 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 32 times.
 187 if ((avctx->flags & AV_CODEC_FLAG_4MV) && s->codec_id != AV_CODEC_ID_MPEG4 &&
493 ✗ s->codec_id != AV_CODEC_ID_H263 && s->codec_id != AV_CODEC_ID_H263P &&
494 ✗ s->codec_id != AV_CODEC_ID_FLV1) {
495 ✗ av_log(avctx, AV_LOG_ERROR, "4MV not supported by codec\n");
496 ✗ return AVERROR(EINVAL);
497 }
498
499
3/4
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 184 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3 times.
 187 if (s->obmc && avctx->mb_decision != FF_MB_DECISION_SIMPLE) {
500 ✗ av_log(avctx, AV_LOG_ERROR,
501 "OBMC is only supported with simple mb decision\n");
502 ✗ return AVERROR(EINVAL);
503 }
504
505
3/4
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 183 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 4 times.
 187 if (s->quarter_sample && s->codec_id != AV_CODEC_ID_MPEG4) {
506 ✗ av_log(avctx, AV_LOG_ERROR, "qpel not supported by codec\n");
507 ✗ return AVERROR(EINVAL);
508 }
509
510
2/2
✓ Branch 0 taken 127 times.
✓ Branch 1 taken 60 times.
 187 if ((s->codec_id == AV_CODEC_ID_MPEG4 ||
511
2/2
✓ Branch 0 taken 121 times.
✓ Branch 1 taken 6 times.
 127 s->codec_id == AV_CODEC_ID_H263 ||
512
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 118 times.
 121 s->codec_id == AV_CODEC_ID_H263P) &&
513
1/2
✓ Branch 0 taken 69 times.
✗ Branch 1 not taken.
 69 (avctx->sample_aspect_ratio.num > 255 ||
514
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 69 times.
 69 avctx->sample_aspect_ratio.den > 255)) {
515 ✗ av_log(avctx, AV_LOG_WARNING,
516 "Invalid pixel aspect ratio %i/%i, limit is 255/255 reducing\n",
517 avctx->sample_aspect_ratio.num, avctx->sample_aspect_ratio.den);
518 ✗ av_reduce(&avctx->sample_aspect_ratio.num, &avctx->sample_aspect_ratio.den,
519 ✗ avctx->sample_aspect_ratio.num, avctx->sample_aspect_ratio.den, 255);
520 }
521
522
2/2
✓ Branch 0 taken 181 times.
✓ Branch 1 taken 6 times.
 187 if ((s->codec_id == AV_CODEC_ID_H263 ||
523
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 178 times.
 181 s->codec_id == AV_CODEC_ID_H263P) &&
524
1/2
✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
 9 (avctx->width > 2048 ||
525
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
 9 avctx->height > 1152 )) {
526 ✗ av_log(avctx, AV_LOG_ERROR, "H.263 does not support resolutions above 2048x1152\n");
527 ✗ return AVERROR(EINVAL);
528 }
529
2/2
✓ Branch 0 taken 181 times.
✓ Branch 1 taken 6 times.
 187 if ((s->codec_id == AV_CODEC_ID_H263 ||
530
2/2
✓ Branch 0 taken 178 times.
✓ Branch 1 taken 3 times.
 181 s->codec_id == AV_CODEC_ID_H263P ||
531
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 175 times.
 178 s->codec_id == AV_CODEC_ID_RV20) &&
532
1/2
✓ Branch 0 taken 12 times.
✗ Branch 1 not taken.
 12 ((avctx->width &3) ||
533
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
 12 (avctx->height&3) )) {
534 ✗ av_log(avctx, AV_LOG_ERROR, "width and height must be a multiple of 4\n");
535 ✗ return AVERROR(EINVAL);
536 }
537
538
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 183 times.
 187 if (s->codec_id == AV_CODEC_ID_RV10 &&
539
1/2
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
 4 (avctx->width &15 ||
540
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 avctx->height&15 )) {
541 ✗ av_log(avctx, AV_LOG_ERROR, "width and height must be a multiple of 16\n");
542 ✗ return AVERROR(EINVAL);
543 }
544
545
2/2
✓ Branch 0 taken 183 times.
✓ Branch 1 taken 4 times.
 187 if ((s->codec_id == AV_CODEC_ID_WMV1 ||
546
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 179 times.
 183 s->codec_id == AV_CODEC_ID_WMV2) &&
547
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
 8 avctx->width & 1) {
548 ✗ av_log(avctx, AV_LOG_ERROR, "width must be multiple of 2\n");
549 ✗ return AVERROR(EINVAL);
550 }
551
552
2/2
✓ Branch 0 taken 17 times.
✓ Branch 1 taken 170 times.
 187 if ((avctx->flags & (AV_CODEC_FLAG_INTERLACED_DCT | AV_CODEC_FLAG_INTERLACED_ME)) &&
553
2/4
✓ Branch 0 taken 17 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 17 times.
 17 s->codec_id != AV_CODEC_ID_MPEG4 && s->codec_id != AV_CODEC_ID_MPEG2VIDEO) {
554 ✗ av_log(avctx, AV_LOG_ERROR, "interlacing not supported by codec\n");
555 ✗ return AVERROR(EINVAL);
556 }
557
558
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 187 if ((s->mpv_flags & FF_MPV_FLAG_CBP_RD) && !avctx->trellis) {
559 ✗ av_log(avctx, AV_LOG_ERROR, "CBP RD needs trellis quant\n");
560 ✗ return AVERROR(EINVAL);
561 }
562
563
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 175 times.
 187 if ((s->mpv_flags & FF_MPV_FLAG_QP_RD) &&
564
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
 12 avctx->mb_decision != FF_MB_DECISION_RD) {
565 ✗ av_log(avctx, AV_LOG_ERROR, "QP RD needs mbd=2\n");
566 ✗ return AVERROR(EINVAL);
567 }
568
569
1/2
✓ Branch 0 taken 187 times.
✗ Branch 1 not taken.
 187 if (s->scenechange_threshold < 1000000000 &&
570
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 (avctx->flags & AV_CODEC_FLAG_CLOSED_GOP)) {
571 ✗ av_log(avctx, AV_LOG_ERROR,
572 "closed gop with scene change detection are not supported yet, "
573 "set threshold to 1000000000\n");
574 ✗ return AVERROR_PATCHWELCOME;
575 }
576
577
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 186 times.
 187 if (avctx->flags & AV_CODEC_FLAG_LOW_DELAY) {
578
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (s->codec_id != AV_CODEC_ID_MPEG2VIDEO &&
579 ✗ avctx->strict_std_compliance >= FF_COMPLIANCE_NORMAL) {
580 ✗ av_log(avctx, AV_LOG_ERROR,
581 "low delay forcing is only available for mpeg2, "
582 "set strict_std_compliance to 'unofficial' or lower in order to allow it\n");
583 ✗ return AVERROR(EINVAL);
584 }
585
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (s->max_b_frames != 0) {
586 ✗ av_log(avctx, AV_LOG_ERROR,
587 "B-frames cannot be used with low delay\n");
588 ✗ return AVERROR(EINVAL);
589 }
590 }
591
592
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 186 times.
 187 if (s->q_scale_type == 1) {
593
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (avctx->qmax > 28) {
594 ✗ av_log(avctx, AV_LOG_ERROR,
595 "non linear quant only supports qmax <= 28 currently\n");
596 ✗ return AVERROR_PATCHWELCOME;
597 }
598 }
599
600
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 175 times.
 187 if (avctx->slices > 1 &&
601
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
 12 !(avctx->codec->capabilities & AV_CODEC_CAP_SLICE_THREADS)) {
602 ✗ av_log(avctx, AV_LOG_ERROR, "Multiple slices are not supported by this codec\n");
603 ✗ return AVERROR(EINVAL);
604 }
605
606
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 187 if (s->b_frame_strategy && (avctx->flags & AV_CODEC_FLAG_PASS2)) {
607 ✗ av_log(avctx, AV_LOG_INFO,
608 "notice: b_frame_strategy only affects the first pass\n");
609 ✗ s->b_frame_strategy = 0;
610 }
611
612 187 i = av_gcd(avctx->time_base.den, avctx->time_base.num);
613
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if (i > 1) {
614 ✗ av_log(avctx, AV_LOG_INFO, "removing common factors from framerate\n");
615 ✗ avctx->time_base.den /= i;
616 ✗ avctx->time_base.num /= i;
617 //return -1;
618 }
619
620
10/12
✓ Branch 0 taken 187 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 176 times.
✓ Branch 3 taken 11 times.
✓ Branch 4 taken 137 times.
✓ Branch 5 taken 39 times.
✓ Branch 6 taken 110 times.
✓ Branch 7 taken 27 times.
✓ Branch 8 taken 106 times.
✓ Branch 9 taken 4 times.
✗ Branch 10 not taken.
✓ Branch 11 taken 106 times.
 187 if (s->mpeg_quant || s->codec_id == AV_CODEC_ID_MPEG1VIDEO || s->codec_id == AV_CODEC_ID_MPEG2VIDEO || s->codec_id == AV_CODEC_ID_MJPEG || s->codec_id == AV_CODEC_ID_AMV || s->codec_id == AV_CODEC_ID_SPEEDHQ) {
621 // (a + x * 3 / 8) / x
622 81 s->intra_quant_bias = 3 << (QUANT_BIAS_SHIFT - 3);
623 81 s->inter_quant_bias = 0;
624 } else {
625 106 s->intra_quant_bias = 0;
626 // (a - x / 4) / x
627 106 s->inter_quant_bias = -(1 << (QUANT_BIAS_SHIFT - 2));
628 }
629
630
2/4
✓ Branch 0 taken 187 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 187 times.
 187 if (avctx->qmin > avctx->qmax || avctx->qmin <= 0) {
631 ✗ av_log(avctx, AV_LOG_ERROR, "qmin and or qmax are invalid, they must be 0 < min <= max\n");
632 ✗ return AVERROR(EINVAL);
633 }
634
635 187 av_log(avctx, AV_LOG_DEBUG, "intra_quant_bias = %d inter_quant_bias = %d\n",s->intra_quant_bias,s->inter_quant_bias);
636
637
2/2
✓ Branch 0 taken 60 times.
✓ Branch 1 taken 127 times.
 187 if (avctx->codec_id == AV_CODEC_ID_MPEG4 &&
638
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 avctx->time_base.den > (1 << 16) - 1) {
639 ✗ av_log(avctx, AV_LOG_ERROR,
640 "timebase %d/%d not supported by MPEG 4 standard, "
641 "the maximum admitted value for the timebase denominator "
642 "is %d\n", avctx->time_base.num, avctx->time_base.den,
643 (1 << 16) - 1);
644 ✗ return AVERROR(EINVAL);
645 }
646 187 s->time_increment_bits = av_log2(avctx->time_base.den - 1) + 1;
647
648
14/16
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 39 times.
✓ Branch 2 taken 31 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 6 times.
✓ Branch 5 taken 6 times.
✓ Branch 6 taken 3 times.
✓ Branch 7 taken 7 times.
✓ Branch 8 taken 4 times.
✓ Branch 9 taken 3 times.
✓ Branch 10 taken 60 times.
✓ Branch 11 taken 4 times.
✓ Branch 12 taken 5 times.
✓ Branch 13 taken 4 times.
✓ Branch 14 taken 4 times.
✗ Branch 15 not taken.
 187 switch (avctx->codec->id) {
649 11 case AV_CODEC_ID_MPEG1VIDEO:
650 11 s->out_format = FMT_MPEG1;
651 11 s->low_delay = !!(avctx->flags & AV_CODEC_FLAG_LOW_DELAY);
652
1/2
✓ Branch 0 taken 11 times.
✗ Branch 1 not taken.
 11 avctx->delay = s->low_delay ? 0 : (s->max_b_frames + 1);
653 11 break;
654 39 case AV_CODEC_ID_MPEG2VIDEO:
655 39 s->out_format = FMT_MPEG1;
656 39 s->low_delay = !!(avctx->flags & AV_CODEC_FLAG_LOW_DELAY);
657
2/2
✓ Branch 0 taken 38 times.
✓ Branch 1 taken 1 times.
 39 avctx->delay = s->low_delay ? 0 : (s->max_b_frames + 1);
658 39 s->rtp_mode = 1;
659 39 break;
660 #if CONFIG_MJPEG_ENCODER || CONFIG_AMV_ENCODER
661 31 case AV_CODEC_ID_MJPEG:
662 case AV_CODEC_ID_AMV:
663 31 s->out_format = FMT_MJPEG;
664 31 s->intra_only = 1; /* force intra only for jpeg */
665
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 31 times.
 31 if ((ret = ff_mjpeg_encode_init(s)) < 0)
666 ✗ return ret;
667 31 avctx->delay = 0;
668 31 s->low_delay = 1;
669 31 break;
670 #endif
671 ✗ case AV_CODEC_ID_SPEEDHQ:
672 ✗ s->out_format = FMT_SPEEDHQ;
673 ✗ s->intra_only = 1; /* force intra only for SHQ */
674 if (!CONFIG_SPEEDHQ_ENCODER)
675 return AVERROR_ENCODER_NOT_FOUND;
676 ✗ if ((ret = ff_speedhq_encode_init(s)) < 0)
677 ✗ return ret;
678 ✗ avctx->delay = 0;
679 ✗ s->low_delay = 1;
680 ✗ break;
681 6 case AV_CODEC_ID_H261:
682 if (!CONFIG_H261_ENCODER)
683 return AVERROR_ENCODER_NOT_FOUND;
684
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
 6 if (ff_h261_get_picture_format(s->width, s->height) < 0) {
685 ✗ av_log(avctx, AV_LOG_ERROR,
686 "The specified picture size of %dx%d is not valid for the "
687 "H.261 codec.\nValid sizes are 176x144, 352x288\n",
688 s->width, s->height);
689 ✗ return AVERROR(EINVAL);
690 }
691 6 s->out_format = FMT_H261;
692 6 avctx->delay = 0;
693 6 s->low_delay = 1;
694 6 s->rtp_mode = 0; /* Sliced encoding not supported */
695 6 break;
696 6 case AV_CODEC_ID_H263:
697 if (!CONFIG_H263_ENCODER)
698 return AVERROR_ENCODER_NOT_FOUND;
699
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 6 times.
 6 if (ff_match_2uint16(ff_h263_format, FF_ARRAY_ELEMS(ff_h263_format),
700 s->width, s->height) == 8) {
701 ✗ av_log(avctx, AV_LOG_ERROR,
702 "The specified picture size of %dx%d is not valid for "
703 "the H.263 codec.\nValid sizes are 128x96, 176x144, "
704 "352x288, 704x576, and 1408x1152. "
705 "Try H.263+.\n", s->width, s->height);
706 ✗ return AVERROR(EINVAL);
707 }
708 6 s->out_format = FMT_H263;
709 6 avctx->delay = 0;
710 6 s->low_delay = 1;
711 6 break;
712 3 case AV_CODEC_ID_H263P:
713 3 s->out_format = FMT_H263;
714 3 s->h263_plus = 1;
715 /* Fx */
716 3 s->h263_aic = (avctx->flags & AV_CODEC_FLAG_AC_PRED) ? 1 : 0;
717 3 s->modified_quant = s->h263_aic;
718 3 s->loop_filter = (avctx->flags & AV_CODEC_FLAG_LOOP_FILTER) ? 1 : 0;
719
3/6
✓ Branch 0 taken 3 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 3 times.
✗ Branch 5 not taken.
 3 s->unrestricted_mv = s->obmc || s->loop_filter || s->umvplus;
720
721 /* /Fx */
722 /* These are just to be sure */
723 3 avctx->delay = 0;
724 3 s->low_delay = 1;
725 3 break;
726 7 case AV_CODEC_ID_FLV1:
727 7 s->out_format = FMT_H263;
728 7 s->h263_flv = 2; /* format = 1; 11-bit codes */
729 7 s->unrestricted_mv = 1;
730 7 s->rtp_mode = 0; /* don't allow GOB */
731 7 avctx->delay = 0;
732 7 s->low_delay = 1;
733 7 break;
734 4 case AV_CODEC_ID_RV10:
735 4 s->out_format = FMT_H263;
736 4 avctx->delay = 0;
737 4 s->low_delay = 1;
738 4 break;
739 3 case AV_CODEC_ID_RV20:
740 3 s->out_format = FMT_H263;
741 3 avctx->delay = 0;
742 3 s->low_delay = 1;
743 3 s->modified_quant = 1;
744 3 s->h263_aic = 1;
745 3 s->h263_plus = 1;
746 3 s->loop_filter = 1;
747 3 s->unrestricted_mv = 0;
748 3 break;
749 60 case AV_CODEC_ID_MPEG4:
750 60 s->out_format = FMT_H263;
751 60 s->h263_pred = 1;
752 60 s->unrestricted_mv = 1;
753 60 s->low_delay = s->max_b_frames ? 0 : 1;
754
2/2
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 40 times.
 60 avctx->delay = s->low_delay ? 0 : (s->max_b_frames + 1);
755 60 break;
756 4 case AV_CODEC_ID_MSMPEG4V2:
757 4 s->out_format = FMT_H263;
758 4 s->h263_pred = 1;
759 4 s->unrestricted_mv = 1;
760 4 s->msmpeg4_version = 2;
761 4 avctx->delay = 0;
762 4 s->low_delay = 1;
763 4 break;
764 5 case AV_CODEC_ID_MSMPEG4V3:
765 5 s->out_format = FMT_H263;
766 5 s->h263_pred = 1;
767 5 s->unrestricted_mv = 1;
768 5 s->msmpeg4_version = 3;
769 5 s->flipflop_rounding = 1;
770 5 avctx->delay = 0;
771 5 s->low_delay = 1;
772 5 break;
773 4 case AV_CODEC_ID_WMV1:
774 4 s->out_format = FMT_H263;
775 4 s->h263_pred = 1;
776 4 s->unrestricted_mv = 1;
777 4 s->msmpeg4_version = 4;
778 4 s->flipflop_rounding = 1;
779 4 avctx->delay = 0;
780 4 s->low_delay = 1;
781 4 break;
782 4 case AV_CODEC_ID_WMV2:
783 4 s->out_format = FMT_H263;
784 4 s->h263_pred = 1;
785 4 s->unrestricted_mv = 1;
786 4 s->msmpeg4_version = 5;
787 4 s->flipflop_rounding = 1;
788 4 avctx->delay = 0;
789 4 s->low_delay = 1;
790 4 break;
791 ✗ default:
792 ✗ return AVERROR(EINVAL);
793 }
794
795 187 avctx->has_b_frames = !s->low_delay;
796
797 187 s->encoding = 1;
798
799 187 s->progressive_frame =
800
2/2
✓ Branch 0 taken 170 times.
✓ Branch 1 taken 17 times.
 357 s->progressive_sequence = !(avctx->flags & (AV_CODEC_FLAG_INTERLACED_DCT |
801 AV_CODEC_FLAG_INTERLACED_ME) ||
802
1/2
✓ Branch 0 taken 170 times.
✗ Branch 1 not taken.
 170 s->alternate_scan);
803
804 /* init */
805 187 ff_mpv_idct_init(s);
806
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 187 times.
 187 if ((ret = ff_mpv_common_init(s)) < 0)
807 ✗ return ret;
808
809 187 ff_fdctdsp_init(&s->fdsp, avctx);
810 187 ff_me_cmp_init(&s->mecc, avctx);
811 187 ff_mpegvideoencdsp_init(&s->mpvencdsp, avctx);
812 187 ff_pixblockdsp_init(&s->pdsp, avctx);
813 187 ff_qpeldsp_init(&s->qdsp);
814
815
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 if (!(avctx->stats_out = av_mallocz(256)) ||
816
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->q_intra_matrix, 32) ||
817
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->q_chroma_intra_matrix, 32) ||
818
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->q_inter_matrix, 32) ||
819
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->q_intra_matrix16, 32) ||
820
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->q_chroma_intra_matrix16, 32) ||
821
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->q_inter_matrix16, 32) ||
822
1/2
✓ Branch 1 taken 187 times.
✗ Branch 2 not taken.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->input_picture, MAX_PICTURE_COUNT) ||
823
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 187 times.
 187 !FF_ALLOCZ_TYPED_ARRAY(s->reordered_input_picture, MAX_PICTURE_COUNT))
824 ✗ return AVERROR(ENOMEM);
825
826
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 183 times.
 187 if (s->noise_reduction) {
827
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
 4 if (!FF_ALLOCZ_TYPED_ARRAY(s->dct_offset, 2))
828 ✗ return AVERROR(ENOMEM);
829 }
830
831 187 ff_dct_encode_init(s);
832
833
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 181 times.
 187 if ((CONFIG_H263P_ENCODER || CONFIG_RV20_ENCODER) && s->modified_quant)
834 6 s->chroma_qscale_table = ff_h263_chroma_qscale_table;
835
836
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 175 times.
 187 if (s->slice_context_count > 1) {
837 12 s->rtp_mode = 1;
838
839
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
 12 if (avctx->codec_id == AV_CODEC_ID_H263P)
840 ✗ s->h263_slice_structured = 1;
841 }
842
843 187 s->quant_precision = 5;
844
845 187 ff_set_cmp(&s->mecc, s->mecc.ildct_cmp, avctx->ildct_cmp);
846 187 ff_set_cmp(&s->mecc, s->mecc.frame_skip_cmp, s->frame_skip_cmp);
847
848
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 181 times.
 187 if (CONFIG_H261_ENCODER && s->out_format == FMT_H261) {
849 6 ff_h261_encode_init(s);
850 181 } else if ((CONFIG_MPEG1VIDEO_ENCODER || CONFIG_MPEG2VIDEO_ENCODER)
851
2/2
✓ Branch 0 taken 50 times.
✓ Branch 1 taken 131 times.
 181 && s->out_format == FMT_MPEG1) {
852 50 ff_mpeg1_encode_init(s);
853
2/2
✓ Branch 0 taken 100 times.
✓ Branch 1 taken 31 times.
 131 } else if (CONFIG_H263_ENCODER && s->out_format == FMT_H263) {
854 100 ff_h263_encode_init(s);
855
2/2
✓ Branch 0 taken 17 times.
✓ Branch 1 taken 83 times.
 100 if (CONFIG_MSMPEG4_ENCODER && s->msmpeg4_version)
856 17 ff_msmpeg4_encode_init(s);
857 }
858
859 /* init q matrix */
860
2/2
✓ Branch 0 taken 11968 times.
✓ Branch 1 taken 187 times.
 12155 for (i = 0; i < 64; i++) {
861 11968 int j = s->idsp.idct_permutation[i];
862
2/2
✓ Branch 0 taken 3840 times.
✓ Branch 1 taken 8128 times.
 11968 if (CONFIG_MPEG4_ENCODER && s->codec_id == AV_CODEC_ID_MPEG4 &&
863
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3840 times.
 3840 s->mpeg_quant) {
864 ✗ s->intra_matrix[j] = ff_mpeg4_default_intra_matrix[i];
865 ✗ s->inter_matrix[j] = ff_mpeg4_default_non_intra_matrix[i];
866
4/4
✓ Branch 0 taken 5568 times.
✓ Branch 1 taken 6400 times.
✓ Branch 2 taken 384 times.
✓ Branch 3 taken 5184 times.
 11968 } else if (s->out_format == FMT_H263 || s->out_format == FMT_H261) {
867 6784 s->intra_matrix[j] =
868 6784 s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
869
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5184 times.
 5184 } else if (CONFIG_SPEEDHQ_ENCODER && s->codec_id == AV_CODEC_ID_SPEEDHQ) {
870 ✗ s->intra_matrix[j] =
871 ✗ s->inter_matrix[j] = ff_mpeg1_default_intra_matrix[i];
872 } else {
873 /* MPEG-1/2 */
874 5184 s->chroma_intra_matrix[j] =
875 5184 s->intra_matrix[j] = ff_mpeg1_default_intra_matrix[i];
876 5184 s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
877 }
878
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11968 times.
 11968 if (avctx->intra_matrix)
879 ✗ s->intra_matrix[j] = avctx->intra_matrix[i];
880
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 11968 times.
 11968 if (avctx->inter_matrix)
881 ✗ s->inter_matrix[j] = avctx->inter_matrix[i];
882 }
883
884 /* precompute matrix */
885 /* for mjpeg, we do include qscale in the matrix */
886
2/2
✓ Branch 0 taken 156 times.
✓ Branch 1 taken 31 times.
 187 if (s->out_format != FMT_MJPEG) {
887 156 ff_convert_matrix(s, s->q_intra_matrix, s->q_intra_matrix16,
888 156 s->intra_matrix, s->intra_quant_bias, avctx->qmin,
889 31, 1);
890 156 ff_convert_matrix(s, s->q_inter_matrix, s->q_inter_matrix16,
891 156 s->inter_matrix, s->inter_quant_bias, avctx->qmin,
892 31, 0);
893 }
894
895
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 187 times.
 187 if ((ret = ff_rate_control_init(s)) < 0)
896 ✗ return ret;
897
898
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if (s->b_frame_strategy == 2) {
899 ✗ for (i = 0; i < s->max_b_frames + 2; i++) {
900 ✗ s->tmp_frames[i] = av_frame_alloc();
901 ✗ if (!s->tmp_frames[i])
902 ✗ return AVERROR(ENOMEM);
903
904 ✗ s->tmp_frames[i]->format = AV_PIX_FMT_YUV420P;
905 ✗ s->tmp_frames[i]->width = s->width >> s->brd_scale;
906 ✗ s->tmp_frames[i]->height = s->height >> s->brd_scale;
907
908 ✗ ret = av_frame_get_buffer(s->tmp_frames[i], 0);
909 ✗ if (ret < 0)
910 ✗ return ret;
911 }
912 }
913
914 187 cpb_props = ff_add_cpb_side_data(avctx);
915
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 187 times.
 187 if (!cpb_props)
916 ✗ return AVERROR(ENOMEM);
917 187 cpb_props->max_bitrate = avctx->rc_max_rate;
918 187 cpb_props->min_bitrate = avctx->rc_min_rate;
919 187 cpb_props->avg_bitrate = avctx->bit_rate;
920 187 cpb_props->buffer_size = avctx->rc_buffer_size;
921
922 187 return 0;
923 }
924
925 187 av_cold int ff_mpv_encode_end(AVCodecContext *avctx)
926 {
927 187 MpegEncContext *s = avctx->priv_data;
928 int i;
929
930 187 ff_rate_control_uninit(s);
931
932 187 ff_mpv_common_end(s);
933
934
2/2
✓ Branch 0 taken 3366 times.
✓ Branch 1 taken 187 times.
 3553 for (i = 0; i < FF_ARRAY_ELEMS(s->tmp_frames); i++)
935 3366 av_frame_free(&s->tmp_frames[i]);
936
937 187 av_frame_free(&s->new_picture);
938
939 187 av_freep(&avctx->stats_out);
940
941
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 156 times.
 187 if(s->q_chroma_intra_matrix != s->q_intra_matrix ) av_freep(&s->q_chroma_intra_matrix);
942
2/2
✓ Branch 0 taken 31 times.
✓ Branch 1 taken 156 times.
 187 if(s->q_chroma_intra_matrix16 != s->q_intra_matrix16) av_freep(&s->q_chroma_intra_matrix16);
943 187 s->q_chroma_intra_matrix= NULL;
944 187 s->q_chroma_intra_matrix16= NULL;
945 187 av_freep(&s->q_intra_matrix);
946 187 av_freep(&s->q_inter_matrix);
947 187 av_freep(&s->q_intra_matrix16);
948 187 av_freep(&s->q_inter_matrix16);
949 187 av_freep(&s->input_picture);
950 187 av_freep(&s->reordered_input_picture);
951 187 av_freep(&s->dct_offset);
952
953 187 return 0;
954 }
955
956 ✗ static int get_sae(uint8_t *src, int ref, int stride)
957 {
958 int x,y;
959 ✗ int acc = 0;
960
961 ✗ for (y = 0; y < 16; y++) {
962 ✗ for (x = 0; x < 16; x++) {
963 ✗ acc += FFABS(src[x + y * stride] - ref);
964 }
965 }
966
967 ✗ return acc;
968 }
969
970 ✗ static int get_intra_count(MpegEncContext *s, uint8_t *src,
971 uint8_t *ref, int stride)
972 {
973 int x, y, w, h;
974 ✗ int acc = 0;
975
976 ✗ w = s->width & ~15;
977 ✗ h = s->height & ~15;
978
979 ✗ for (y = 0; y < h; y += 16) {
980 ✗ for (x = 0; x < w; x += 16) {
981 ✗ int offset = x + y * stride;
982 ✗ int sad = s->mecc.sad[0](NULL, src + offset, ref + offset,
983 stride, 16);
984 ✗ int mean = (s->mpvencdsp.pix_sum(src + offset, stride) + 128) >> 8;
985 ✗ int sae = get_sae(src + offset, mean, stride);
986
987 ✗ acc += sae + 500 < sad;
988 }
989 }
990 ✗ return acc;
991 }
992
993 9263 static int alloc_picture(MpegEncContext *s, Picture *pic, int shared)
994 {
995 18526 return ff_alloc_picture(s->avctx, pic, &s->me, &s->sc, shared, 1,
996 9263 s->chroma_x_shift, s->chroma_y_shift, s->out_format,
997 s->mb_stride, s->mb_width, s->mb_height, s->b8_stride,
998 &s->linesize, &s->uvlinesize);
999 }
1000
1001 8790 static int load_input_picture(MpegEncContext *s, const AVFrame *pic_arg)
1002 {
1003 8790 Picture *pic = NULL;
1004 int64_t pts;
1005 8790 int i, display_picture_number = 0, ret;
1006 17580 int encoding_delay = s->max_b_frames ? s->max_b_frames
1007
2/2
✓ Branch 0 taken 2213 times.
✓ Branch 1 taken 6577 times.
 8790 : (s->low_delay ? 0 : 1);
1008 8790 int flush_offset = 1;
1009 8790 int direct = 1;
1010
1011
2/2
✓ Branch 0 taken 8564 times.
✓ Branch 1 taken 226 times.
 8790 if (pic_arg) {
1012 8564 pts = pic_arg->pts;
1013 8564 display_picture_number = s->input_picture_number++;
1014
1015
1/2
✓ Branch 0 taken 8564 times.
✗ Branch 1 not taken.
 8564 if (pts != AV_NOPTS_VALUE) {
1016
2/2
✓ Branch 0 taken 8377 times.
✓ Branch 1 taken 187 times.
 8564 if (s->user_specified_pts != AV_NOPTS_VALUE) {
1017 8377 int64_t last = s->user_specified_pts;
1018
1019
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8377 times.
 8377 if (pts <= last) {
1020 ✗ av_log(s->avctx, AV_LOG_ERROR,
1021 "Invalid pts (%"PRId64") <= last (%"PRId64")\n",
1022 pts, last);
1023 ✗ return AVERROR(EINVAL);
1024 }
1025
1026
4/4
✓ Branch 0 taken 3151 times.
✓ Branch 1 taken 5226 times.
✓ Branch 2 taken 69 times.
✓ Branch 3 taken 3082 times.
 8377 if (!s->low_delay && display_picture_number == 1)
1027 69 s->dts_delta = pts - last;
1028 }
1029 8564 s->user_specified_pts = pts;
1030 } else {
1031 ✗ if (s->user_specified_pts != AV_NOPTS_VALUE) {
1032 ✗ s->user_specified_pts =
1033 ✗ pts = s->user_specified_pts + 1;
1034 ✗ av_log(s->avctx, AV_LOG_INFO,
1035 "Warning: AVFrame.pts=? trying to guess (%"PRId64")\n",
1036 pts);
1037 } else {
1038 ✗ pts = display_picture_number;
1039 }
1040 }
1041
1042
1/2
✓ Branch 0 taken 8564 times.
✗ Branch 1 not taken.
 8564 if (!pic_arg->buf[0] ||
1043
2/2
✓ Branch 0 taken 721 times.
✓ Branch 1 taken 7843 times.
 8564 pic_arg->linesize[0] != s->linesize ||
1044
1/2
✓ Branch 0 taken 721 times.
✗ Branch 1 not taken.
 721 pic_arg->linesize[1] != s->uvlinesize ||
1045
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 721 times.
 721 pic_arg->linesize[2] != s->uvlinesize)
1046 7843 direct = 0;
1047
3/4
✓ Branch 0 taken 6939 times.
✓ Branch 1 taken 1625 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6939 times.
 8564 if ((s->width & 15) || (s->height & 15))
1048 1625 direct = 0;
1049
2/2
✓ Branch 0 taken 200 times.
✓ Branch 1 taken 8364 times.
 8564 if (((intptr_t)(pic_arg->data[0])) & (STRIDE_ALIGN-1))
1050 200 direct = 0;
1051
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (s->linesize & (STRIDE_ALIGN-1))
1052 ✗ direct = 0;
1053
1054 ff_dlog(s->avctx, "%d %d %"PTRDIFF_SPECIFIER" %"PTRDIFF_SPECIFIER"\n", pic_arg->linesize[0],
1055 pic_arg->linesize[1], s->linesize, s->uvlinesize);
1056
1057 8564 i = ff_find_unused_picture(s->avctx, s->picture, direct);
1058
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (i < 0)
1059 ✗ return i;
1060
1061 8564 pic = &s->picture[i];
1062 8564 pic->reference = 3;
1063
1064
2/2
✓ Branch 0 taken 649 times.
✓ Branch 1 taken 7915 times.
 8564 if (direct) {
1065
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 649 times.
 649 if ((ret = av_frame_ref(pic->f, pic_arg)) < 0)
1066 ✗ return ret;
1067 }
1068 8564 ret = alloc_picture(s, pic, direct);
1069
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (ret < 0)
1070 ✗ return ret;
1071
1072
2/2
✓ Branch 0 taken 7915 times.
✓ Branch 1 taken 649 times.
 8564 if (!direct) {
1073
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7915 times.
 7915 if (pic->f->data[0] + INPLACE_OFFSET == pic_arg->data[0] &&
1074 ✗ pic->f->data[1] + INPLACE_OFFSET == pic_arg->data[1] &&
1075 ✗ pic->f->data[2] + INPLACE_OFFSET == pic_arg->data[2]) {
1076 // empty
1077 } else {
1078 int h_chroma_shift, v_chroma_shift;
1079 7915 av_pix_fmt_get_chroma_sub_sample(s->avctx->pix_fmt,
1080 &h_chroma_shift,
1081 &v_chroma_shift);
1082
1083
2/2
✓ Branch 0 taken 23745 times.
✓ Branch 1 taken 7915 times.
 31660 for (i = 0; i < 3; i++) {
1084 23745 int src_stride = pic_arg->linesize[i];
1085
2/2
✓ Branch 0 taken 15830 times.
✓ Branch 1 taken 7915 times.
 23745 int dst_stride = i ? s->uvlinesize : s->linesize;
1086
2/2
✓ Branch 0 taken 15830 times.
✓ Branch 1 taken 7915 times.
 23745 int h_shift = i ? h_chroma_shift : 0;
1087
2/2
✓ Branch 0 taken 15830 times.
✓ Branch 1 taken 7915 times.
 23745 int v_shift = i ? v_chroma_shift : 0;
1088 23745 int w = s->width >> h_shift;
1089 23745 int h = s->height >> v_shift;
1090 23745 uint8_t *src = pic_arg->data[i];
1091 23745 uint8_t *dst = pic->f->data[i];
1092 23745 int vpad = 16;
1093
1094
2/2
✓ Branch 0 taken 4920 times.
✓ Branch 1 taken 18825 times.
 23745 if ( s->codec_id == AV_CODEC_ID_MPEG2VIDEO
1095
2/2
✓ Branch 0 taken 2475 times.
✓ Branch 1 taken 2445 times.
 4920 && !s->progressive_sequence
1096
2/2
✓ Branch 0 taken 600 times.
✓ Branch 1 taken 1875 times.
 2475 && FFALIGN(s->height, 32) - s->height > 16)
1097 600 vpad = 32;
1098
1099
2/2
✓ Branch 0 taken 23595 times.
✓ Branch 1 taken 150 times.
 23745 if (!s->avctx->rc_buffer_size)
1100 23595 dst += INPLACE_OFFSET;
1101
1102
2/2
✓ Branch 0 taken 303 times.
✓ Branch 1 taken 23442 times.
 23745 if (src_stride == dst_stride)
1103 303 memcpy(dst, src, src_stride * h);
1104 else {
1105 23442 int h2 = h;
1106 23442 uint8_t *dst2 = dst;
1107
2/2
✓ Branch 0 taken 3967972 times.
✓ Branch 1 taken 23442 times.
 3991414 while (h2--) {
1108 3967972 memcpy(dst2, src, w);
1109 3967972 dst2 += dst_stride;
1110 3967972 src += src_stride;
1111 }
1112 }
1113
3/4
✓ Branch 0 taken 18870 times.
✓ Branch 1 taken 4875 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 18870 times.
 23745 if ((s->width & 15) || (s->height & (vpad-1))) {
1114 4875 s->mpvencdsp.draw_edges(dst, dst_stride,
1115 w, h,
1116 16 >> h_shift,
1117 vpad >> v_shift,
1118 EDGE_BOTTOM);
1119 }
1120 }
1121 7915 emms_c();
1122 }
1123 }
1124 8564 ret = av_frame_copy_props(pic->f, pic_arg);
1125
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (ret < 0)
1126 ✗ return ret;
1127
1128 8564 pic->f->display_picture_number = display_picture_number;
1129 8564 pic->f->pts = pts; // we set this here to avoid modifying pic_arg
1130 } else {
1131 /* Flushing: When we have not received enough input frames,
1132 * ensure s->input_picture[0] contains the first picture */
1133
1/2
✓ Branch 0 taken 226 times.
✗ Branch 1 not taken.
 226 for (flush_offset = 0; flush_offset < encoding_delay + 1; flush_offset++)
1134
1/2
✓ Branch 0 taken 226 times.
✗ Branch 1 not taken.
 226 if (s->input_picture[flush_offset])
1135 226 break;
1136
1137
1/2
✓ Branch 0 taken 226 times.
✗ Branch 1 not taken.
 226 if (flush_offset <= 1)
1138 226 flush_offset = 1;
1139 else
1140 ✗ encoding_delay = encoding_delay - flush_offset + 1;
1141 }
1142
1143 /* shift buffer entries */
1144
2/2
✓ Branch 0 taken 307650 times.
✓ Branch 1 taken 8790 times.
 316440 for (i = flush_offset; i < MAX_PICTURE_COUNT /*s->encoding_delay + 1*/; i++)
1145 307650 s->input_picture[i - flush_offset] = s->input_picture[i];
1146
1147 8790 s->input_picture[encoding_delay] = (Picture*) pic;
1148
1149 8790 return 0;
1150 }
1151
1152 ✗ static int skip_check(MpegEncContext *s, Picture *p, Picture *ref)
1153 {
1154 int x, y, plane;
1155 ✗ int score = 0;
1156 ✗ int64_t score64 = 0;
1157
1158 ✗ for (plane = 0; plane < 3; plane++) {
1159 ✗ const int stride = p->f->linesize[plane];
1160 ✗ const int bw = plane ? 1 : 2;
1161 ✗ for (y = 0; y < s->mb_height * bw; y++) {
1162 ✗ for (x = 0; x < s->mb_width * bw; x++) {
1163 ✗ int off = p->shared ? 0 : 16;
1164 ✗ uint8_t *dptr = p->f->data[plane] + 8 * (x + y * stride) + off;
1165 ✗ uint8_t *rptr = ref->f->data[plane] + 8 * (x + y * stride);
1166 ✗ int v = s->mecc.frame_skip_cmp[1](s, dptr, rptr, stride, 8);
1167
1168 ✗ switch (FFABS(s->frame_skip_exp)) {
1169 ✗ case 0: score = FFMAX(score, v); break;
1170 ✗ case 1: score += FFABS(v); break;
1171 ✗ case 2: score64 += v * (int64_t)v; break;
1172 ✗ case 3: score64 += FFABS(v * (int64_t)v * v); break;
1173 ✗ case 4: score64 += (v * (int64_t)v) * (v * (int64_t)v); break;
1174 }
1175 }
1176 }
1177 }
1178 ✗ emms_c();
1179
1180 ✗ if (score)
1181 ✗ score64 = score;
1182 ✗ if (s->frame_skip_exp < 0)
1183 ✗ score64 = pow(score64 / (double)(s->mb_width * s->mb_height),
1184 ✗ -1.0/s->frame_skip_exp);
1185
1186 ✗ if (score64 < s->frame_skip_threshold)
1187 ✗ return 1;
1188 ✗ if (score64 < ((s->frame_skip_factor * (int64_t) s->lambda) >> 8))
1189 ✗ return 1;
1190 ✗ return 0;
1191 }
1192
1193 ✗ static int encode_frame(AVCodecContext *c, AVFrame *frame, AVPacket *pkt)
1194 {
1195 int ret;
1196 ✗ int size = 0;
1197
1198 ✗ ret = avcodec_send_frame(c, frame);
1199 ✗ if (ret < 0)
1200 ✗ return ret;
1201
1202 do {
1203 ✗ ret = avcodec_receive_packet(c, pkt);
1204 ✗ if (ret >= 0) {
1205 ✗ size += pkt->size;
1206 ✗ av_packet_unref(pkt);
1207 ✗ } else if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF)
1208 ✗ return ret;
1209 ✗ } while (ret >= 0);
1210
1211 ✗ return size;
1212 }
1213
1214 ✗ static int estimate_best_b_count(MpegEncContext *s)
1215 {
1216 AVPacket *pkt;
1217 ✗ const int scale = s->brd_scale;
1218 ✗ int width = s->width >> scale;
1219 ✗ int height = s->height >> scale;
1220 int i, j, out_size, p_lambda, b_lambda, lambda2;
1221 ✗ int64_t best_rd = INT64_MAX;
1222 ✗ int best_b_count = -1;
1223 ✗ int ret = 0;
1224
1225 ✗ av_assert0(scale >= 0 && scale <= 3);
1226
1227 ✗ pkt = av_packet_alloc();
1228 ✗ if (!pkt)
1229 ✗ return AVERROR(ENOMEM);
1230
1231 //emms_c();
1232 //s->next_picture_ptr->quality;
1233 ✗ p_lambda = s->last_lambda_for[AV_PICTURE_TYPE_P];
1234 //p_lambda * FFABS(s->avctx->b_quant_factor) + s->avctx->b_quant_offset;
1235 ✗ b_lambda = s->last_lambda_for[AV_PICTURE_TYPE_B];
1236 ✗ if (!b_lambda) // FIXME we should do this somewhere else
1237 ✗ b_lambda = p_lambda;
1238 ✗ lambda2 = (b_lambda * b_lambda + (1 << FF_LAMBDA_SHIFT) / 2) >>
1239 FF_LAMBDA_SHIFT;
1240
1241 ✗ for (i = 0; i < s->max_b_frames + 2; i++) {
1242 ✗ Picture pre_input, *pre_input_ptr = i ? s->input_picture[i - 1] :
1243 s->next_picture_ptr;
1244 uint8_t *data[4];
1245
1246 ✗ if (pre_input_ptr && (!i || s->input_picture[i - 1])) {
1247 ✗ pre_input = *pre_input_ptr;
1248 ✗ memcpy(data, pre_input_ptr->f->data, sizeof(data));
1249
1250 ✗ if (!pre_input.shared && i) {
1251 ✗ data[0] += INPLACE_OFFSET;
1252 ✗ data[1] += INPLACE_OFFSET;
1253 ✗ data[2] += INPLACE_OFFSET;
1254 }
1255
1256 ✗ s->mpvencdsp.shrink[scale](s->tmp_frames[i]->data[0],
1257 ✗ s->tmp_frames[i]->linesize[0],
1258 ✗ data[0],
1259 ✗ pre_input.f->linesize[0],
1260 width, height);
1261 ✗ s->mpvencdsp.shrink[scale](s->tmp_frames[i]->data[1],
1262 ✗ s->tmp_frames[i]->linesize[1],
1263 ✗ data[1],
1264 ✗ pre_input.f->linesize[1],
1265 width >> 1, height >> 1);
1266 ✗ s->mpvencdsp.shrink[scale](s->tmp_frames[i]->data[2],
1267 ✗ s->tmp_frames[i]->linesize[2],
1268 ✗ data[2],
1269 ✗ pre_input.f->linesize[2],
1270 width >> 1, height >> 1);
1271 }
1272 }
1273
1274 ✗ for (j = 0; j < s->max_b_frames + 1; j++) {
1275 AVCodecContext *c;
1276 ✗ int64_t rd = 0;
1277
1278 ✗ if (!s->input_picture[j])
1279 ✗ break;
1280
1281 ✗ c = avcodec_alloc_context3(NULL);
1282 ✗ if (!c) {
1283 ✗ ret = AVERROR(ENOMEM);
1284 ✗ goto fail;
1285 }
1286
1287 ✗ c->width = width;
1288 ✗ c->height = height;
1289 ✗ c->flags = AV_CODEC_FLAG_QSCALE | AV_CODEC_FLAG_PSNR;
1290 ✗ c->flags |= s->avctx->flags & AV_CODEC_FLAG_QPEL;
1291 ✗ c->mb_decision = s->avctx->mb_decision;
1292 ✗ c->me_cmp = s->avctx->me_cmp;
1293 ✗ c->mb_cmp = s->avctx->mb_cmp;
1294 ✗ c->me_sub_cmp = s->avctx->me_sub_cmp;
1295 ✗ c->pix_fmt = AV_PIX_FMT_YUV420P;
1296 ✗ c->time_base = s->avctx->time_base;
1297 ✗ c->max_b_frames = s->max_b_frames;
1298
1299 ✗ ret = avcodec_open2(c, s->avctx->codec, NULL);
1300 ✗ if (ret < 0)
1301 ✗ goto fail;
1302
1303
1304 ✗ s->tmp_frames[0]->pict_type = AV_PICTURE_TYPE_I;
1305 ✗ s->tmp_frames[0]->quality = 1 * FF_QP2LAMBDA;
1306
1307 ✗ out_size = encode_frame(c, s->tmp_frames[0], pkt);
1308 ✗ if (out_size < 0) {
1309 ✗ ret = out_size;
1310 ✗ goto fail;
1311 }
1312
1313 //rd += (out_size * lambda2) >> FF_LAMBDA_SHIFT;
1314
1315 ✗ for (i = 0; i < s->max_b_frames + 1; i++) {
1316 ✗ int is_p = i % (j + 1) == j || i == s->max_b_frames;
1317
1318 ✗ s->tmp_frames[i + 1]->pict_type = is_p ?
1319 ✗ AV_PICTURE_TYPE_P : AV_PICTURE_TYPE_B;
1320 ✗ s->tmp_frames[i + 1]->quality = is_p ? p_lambda : b_lambda;
1321
1322 ✗ out_size = encode_frame(c, s->tmp_frames[i + 1], pkt);
1323 ✗ if (out_size < 0) {
1324 ✗ ret = out_size;
1325 ✗ goto fail;
1326 }
1327
1328 ✗ rd += (out_size * lambda2) >> (FF_LAMBDA_SHIFT - 3);
1329 }
1330
1331 /* get the delayed frames */
1332 ✗ out_size = encode_frame(c, NULL, pkt);
1333 ✗ if (out_size < 0) {
1334 ✗ ret = out_size;
1335 ✗ goto fail;
1336 }
1337 ✗ rd += (out_size * lambda2) >> (FF_LAMBDA_SHIFT - 3);
1338
1339 ✗ rd += c->error[0] + c->error[1] + c->error[2];
1340
1341 ✗ if (rd < best_rd) {
1342 ✗ best_rd = rd;
1343 ✗ best_b_count = j;
1344 }
1345
1346 ✗ fail:
1347 ✗ avcodec_free_context(&c);
1348 ✗ av_packet_unref(pkt);
1349 ✗ if (ret < 0) {
1350 ✗ best_b_count = ret;
1351 ✗ break;
1352 }
1353 }
1354
1355 ✗ av_packet_free(&pkt);
1356
1357 ✗ return best_b_count;
1358 }
1359
1360 8790 static int select_input_picture(MpegEncContext *s)
1361 {
1362 int i, ret;
1363
1364
2/2
✓ Branch 0 taken 307650 times.
✓ Branch 1 taken 8790 times.
 316440 for (i = 1; i < MAX_PICTURE_COUNT; i++)
1365 307650 s->reordered_input_picture[i - 1] = s->reordered_input_picture[i];
1366 8790 s->reordered_input_picture[MAX_PICTURE_COUNT - 1] = NULL;
1367
1368 /* set next picture type & ordering */
1369
4/4
✓ Branch 0 taken 1347 times.
✓ Branch 1 taken 7443 times.
✓ Branch 2 taken 226 times.
✓ Branch 3 taken 7217 times.
 8790 if (!s->reordered_input_picture[0] && s->input_picture[0]) {
1370
2/4
✓ Branch 0 taken 7217 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 7217 times.
 7217 if (s->frame_skip_threshold || s->frame_skip_factor) {
1371 ✗ if (s->picture_in_gop_number < s->gop_size &&
1372 ✗ s->next_picture_ptr &&
1373 ✗ skip_check(s, s->input_picture[0], s->next_picture_ptr)) {
1374 // FIXME check that the gop check above is +-1 correct
1375 ✗ av_frame_unref(s->input_picture[0]->f);
1376
1377 ✗ ff_vbv_update(s, 0);
1378
1379 ✗ goto no_output_pic;
1380 }
1381 }
1382
1383 7217 if (/*s->picture_in_gop_number >= s->gop_size ||*/
1384
4/4
✓ Branch 0 taken 7030 times.
✓ Branch 1 taken 187 times.
✓ Branch 2 taken 1432 times.
✓ Branch 3 taken 5598 times.
 7217 !s->next_picture_ptr || s->intra_only) {
1385 1619 s->reordered_input_picture[0] = s->input_picture[0];
1386 1619 s->reordered_input_picture[0]->f->pict_type = AV_PICTURE_TYPE_I;
1387 1619 s->reordered_input_picture[0]->f->coded_picture_number =
1388 1619 s->coded_picture_number++;
1389 } else {
1390 5598 int b_frames = 0;
1391
1392
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5598 times.
 5598 if (s->avctx->flags & AV_CODEC_FLAG_PASS2) {
1393 ✗ for (i = 0; i < s->max_b_frames + 1; i++) {
1394 ✗ int pict_num = s->input_picture[0]->f->display_picture_number + i;
1395
1396 ✗ if (pict_num >= s->rc_context.num_entries)
1397 ✗ break;
1398 ✗ if (!s->input_picture[i]) {
1399 ✗ s->rc_context.entry[pict_num - 1].new_pict_type = AV_PICTURE_TYPE_P;
1400 ✗ break;
1401 }
1402
1403 ✗ s->input_picture[i]->f->pict_type =
1404 ✗ s->rc_context.entry[pict_num].new_pict_type;
1405 }
1406 }
1407
1408
1/2
✓ Branch 0 taken 5598 times.
✗ Branch 1 not taken.
 5598 if (s->b_frame_strategy == 0) {
1409 5598 b_frames = s->max_b_frames;
1410
4/4
✓ Branch 0 taken 735 times.
✓ Branch 1 taken 4948 times.
✓ Branch 2 taken 85 times.
✓ Branch 3 taken 650 times.
 5683 while (b_frames && !s->input_picture[b_frames])
1411 85 b_frames--;
1412 ✗ } else if (s->b_frame_strategy == 1) {
1413 ✗ for (i = 1; i < s->max_b_frames + 1; i++) {
1414 ✗ if (s->input_picture[i] &&
1415 ✗ s->input_picture[i]->b_frame_score == 0) {
1416 ✗ s->input_picture[i]->b_frame_score =
1417 ✗ get_intra_count(s,
1418 ✗ s->input_picture[i ]->f->data[0],
1419 ✗ s->input_picture[i - 1]->f->data[0],
1420 ✗ s->linesize) + 1;
1421 }
1422 }
1423 ✗ for (i = 0; i < s->max_b_frames + 1; i++) {
1424 ✗ if (!s->input_picture[i] ||
1425 ✗ s->input_picture[i]->b_frame_score - 1 >
1426 ✗ s->mb_num / s->b_sensitivity)
1427 break;
1428 }
1429
1430 ✗ b_frames = FFMAX(0, i - 1);
1431
1432 /* reset scores */
1433 ✗ for (i = 0; i < b_frames + 1; i++) {
1434 ✗ s->input_picture[i]->b_frame_score = 0;
1435 }
1436 ✗ } else if (s->b_frame_strategy == 2) {
1437 ✗ b_frames = estimate_best_b_count(s);
1438 ✗ if (b_frames < 0)
1439 ✗ return b_frames;
1440 }
1441
1442 5598 emms_c();
1443
1444
2/2
✓ Branch 0 taken 1347 times.
✓ Branch 1 taken 5598 times.
 6945 for (i = b_frames - 1; i >= 0; i--) {
1445 1347 int type = s->input_picture[i]->f->pict_type;
1446
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1347 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 1347 if (type && type != AV_PICTURE_TYPE_B)
1447 ✗ b_frames = i;
1448 }
1449
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5598 times.
 5598 if (s->input_picture[b_frames]->f->pict_type == AV_PICTURE_TYPE_B &&
1450 ✗ b_frames == s->max_b_frames) {
1451 ✗ av_log(s->avctx, AV_LOG_ERROR,
1452 "warning, too many B-frames in a row\n");
1453 }
1454
1455
2/2
✓ Branch 0 taken 555 times.
✓ Branch 1 taken 5043 times.
 5598 if (s->picture_in_gop_number + b_frames >= s->gop_size) {
1456
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 555 times.
 555 if ((s->mpv_flags & FF_MPV_FLAG_STRICT_GOP) &&
1457 ✗ s->gop_size > s->picture_in_gop_number) {
1458 ✗ b_frames = s->gop_size - s->picture_in_gop_number - 1;
1459 } else {
1460
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 555 times.
 555 if (s->avctx->flags & AV_CODEC_FLAG_CLOSED_GOP)
1461 ✗ b_frames = 0;
1462 555 s->input_picture[b_frames]->f->pict_type = AV_PICTURE_TYPE_I;
1463 }
1464 }
1465
1466
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 5598 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 5598 if ((s->avctx->flags & AV_CODEC_FLAG_CLOSED_GOP) && b_frames &&
1467 ✗ s->input_picture[b_frames]->f->pict_type == AV_PICTURE_TYPE_I)
1468 ✗ b_frames--;
1469
1470 5598 s->reordered_input_picture[0] = s->input_picture[b_frames];
1471
2/2
✓ Branch 0 taken 5041 times.
✓ Branch 1 taken 557 times.
 5598 if (s->reordered_input_picture[0]->f->pict_type != AV_PICTURE_TYPE_I)
1472 5041 s->reordered_input_picture[0]->f->pict_type = AV_PICTURE_TYPE_P;
1473 5598 s->reordered_input_picture[0]->f->coded_picture_number =
1474 5598 s->coded_picture_number++;
1475
2/2
✓ Branch 0 taken 1347 times.
✓ Branch 1 taken 5598 times.
 6945 for (i = 0; i < b_frames; i++) {
1476 1347 s->reordered_input_picture[i + 1] = s->input_picture[i];
1477 1347 s->reordered_input_picture[i + 1]->f->pict_type =
1478 AV_PICTURE_TYPE_B;
1479 1347 s->reordered_input_picture[i + 1]->f->coded_picture_number =
1480 1347 s->coded_picture_number++;
1481 }
1482 }
1483 }
1484 7171 no_output_pic:
1485 8790 av_frame_unref(s->new_picture);
1486
1487
2/2
✓ Branch 0 taken 8564 times.
✓ Branch 1 taken 226 times.
 8790 if (s->reordered_input_picture[0]) {
1488 8564 s->reordered_input_picture[0]->reference =
1489 8564 s->reordered_input_picture[0]->f->pict_type !=
1490
2/2
✓ Branch 0 taken 7217 times.
✓ Branch 1 taken 1347 times.
 8564 AV_PICTURE_TYPE_B ? 3 : 0;
1491
1492
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if ((ret = av_frame_ref(s->new_picture,
1493 8564 s->reordered_input_picture[0]->f)))
1494 ✗ return ret;
1495
1496
4/4
✓ Branch 0 taken 7915 times.
✓ Branch 1 taken 649 times.
✓ Branch 2 taken 50 times.
✓ Branch 3 taken 7865 times.
 8564 if (s->reordered_input_picture[0]->shared || s->avctx->rc_buffer_size) {
1497 // input is a shared pix, so we can't modify it -> allocate a new
1498 // one & ensure that the shared one is reuseable
1499
1500 Picture *pic;
1501 699 int i = ff_find_unused_picture(s->avctx, s->picture, 0);
1502
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 699 times.
 699 if (i < 0)
1503 ✗ return i;
1504 699 pic = &s->picture[i];
1505
1506 699 pic->reference = s->reordered_input_picture[0]->reference;
1507
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 699 times.
 699 if (alloc_picture(s, pic, 0) < 0) {
1508 ✗ return -1;
1509 }
1510
1511 699 ret = av_frame_copy_props(pic->f, s->reordered_input_picture[0]->f);
1512
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 699 times.
 699 if (ret < 0)
1513 ✗ return ret;
1514
1515 /* mark us unused / free shared pic */
1516 699 av_frame_unref(s->reordered_input_picture[0]->f);
1517 699 s->reordered_input_picture[0]->shared = 0;
1518
1519 699 s->current_picture_ptr = pic;
1520 } else {
1521 // input is not a shared pix -> reuse buffer for current_pix
1522 7865 s->current_picture_ptr = s->reordered_input_picture[0];
1523
2/2
✓ Branch 0 taken 31460 times.
✓ Branch 1 taken 7865 times.
 39325 for (i = 0; i < 4; i++) {
1524
2/2
✓ Branch 0 taken 23595 times.
✓ Branch 1 taken 7865 times.
 31460 if (s->new_picture->data[i])
1525 23595 s->new_picture->data[i] += INPLACE_OFFSET;
1526 }
1527 }
1528 8564 s->picture_number = s->new_picture->display_picture_number;
1529 }
1530 8790 return 0;
1531 }
1532
1533 8689 static void frame_end(MpegEncContext *s)
1534 {
1535
2/2
✓ Branch 0 taken 3955 times.
✓ Branch 1 taken 4734 times.
 8689 if (s->unrestricted_mv &&
1536
2/2
✓ Branch 0 taken 3315 times.
✓ Branch 1 taken 640 times.
 3955 s->current_picture.reference &&
1537
1/2
✓ Branch 0 taken 3315 times.
✗ Branch 1 not taken.
 3315 !s->intra_only) {
1538 3315 const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(s->avctx->pix_fmt);
1539 3315 int hshift = desc->log2_chroma_w;
1540 3315 int vshift = desc->log2_chroma_h;
1541 3315 s->mpvencdsp.draw_edges(s->current_picture.f->data[0],
1542 3315 s->current_picture.f->linesize[0],
1543 s->h_edge_pos, s->v_edge_pos,
1544 EDGE_WIDTH, EDGE_WIDTH,
1545 EDGE_TOP | EDGE_BOTTOM);
1546 3315 s->mpvencdsp.draw_edges(s->current_picture.f->data[1],
1547 3315 s->current_picture.f->linesize[1],
1548 3315 s->h_edge_pos >> hshift,
1549 3315 s->v_edge_pos >> vshift,
1550 EDGE_WIDTH >> hshift,
1551 EDGE_WIDTH >> vshift,
1552 EDGE_TOP | EDGE_BOTTOM);
1553 3315 s->mpvencdsp.draw_edges(s->current_picture.f->data[2],
1554 3315 s->current_picture.f->linesize[2],
1555 3315 s->h_edge_pos >> hshift,
1556 3315 s->v_edge_pos >> vshift,
1557 EDGE_WIDTH >> hshift,
1558 EDGE_WIDTH >> vshift,
1559 EDGE_TOP | EDGE_BOTTOM);
1560 }
1561
1562 8689 emms_c();
1563
1564 8689 s->last_pict_type = s->pict_type;
1565 8689 s->last_lambda_for [s->pict_type] = s->current_picture_ptr->f->quality;
1566
2/2
✓ Branch 0 taken 7342 times.
✓ Branch 1 taken 1347 times.
 8689 if (s->pict_type!= AV_PICTURE_TYPE_B)
1567 7342 s->last_non_b_pict_type = s->pict_type;
1568 8689 }
1569
1570 200 static void update_noise_reduction(MpegEncContext *s)
1571 {
1572 int intra, i;
1573
1574
2/2
✓ Branch 0 taken 400 times.
✓ Branch 1 taken 200 times.
 600 for (intra = 0; intra < 2; intra++) {
1575
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 392 times.
 400 if (s->dct_count[intra] > (1 << 16)) {
1576
2/2
✓ Branch 0 taken 512 times.
✓ Branch 1 taken 8 times.
 520 for (i = 0; i < 64; i++) {
1577 512 s->dct_error_sum[intra][i] >>= 1;
1578 }
1579 8 s->dct_count[intra] >>= 1;
1580 }
1581
1582
2/2
✓ Branch 0 taken 25600 times.
✓ Branch 1 taken 400 times.
 26000 for (i = 0; i < 64; i++) {
1583 25600 s->dct_offset[intra][i] = (s->noise_reduction *
1584 25600 s->dct_count[intra] +
1585 25600 s->dct_error_sum[intra][i] / 2) /
1586 25600 (s->dct_error_sum[intra][i] + 1);
1587 }
1588 }
1589 200 }
1590
1591 8564 static int frame_start(MpegEncContext *s)
1592 {
1593 int ret;
1594
1595 /* mark & release old frames */
1596
4/4
✓ Branch 0 taken 7217 times.
✓ Branch 1 taken 1347 times.
✓ Branch 2 taken 6844 times.
✓ Branch 3 taken 373 times.
 8564 if (s->pict_type != AV_PICTURE_TYPE_B && s->last_picture_ptr &&
1597
1/2
✓ Branch 0 taken 6844 times.
✗ Branch 1 not taken.
 6844 s->last_picture_ptr != s->next_picture_ptr &&
1598
1/2
✓ Branch 0 taken 6844 times.
✗ Branch 1 not taken.
 6844 s->last_picture_ptr->f->buf[0]) {
1599 6844 ff_mpeg_unref_picture(s->avctx, s->last_picture_ptr);
1600 }
1601
1602 8564 s->current_picture_ptr->f->pict_type = s->pict_type;
1603 8564 s->current_picture_ptr->f->key_frame = s->pict_type == AV_PICTURE_TYPE_I;
1604
1605 8564 ff_mpeg_unref_picture(s->avctx, &s->current_picture);
1606
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 8564 times.
 8564 if ((ret = ff_mpeg_ref_picture(s->avctx, &s->current_picture,
1607 s->current_picture_ptr)) < 0)
1608 ✗ return ret;
1609
1610
2/2
✓ Branch 0 taken 7217 times.
✓ Branch 1 taken 1347 times.
 8564 if (s->pict_type != AV_PICTURE_TYPE_B) {
1611 7217 s->last_picture_ptr = s->next_picture_ptr;
1612 7217 s->next_picture_ptr = s->current_picture_ptr;
1613 }
1614
1615
2/2
✓ Branch 0 taken 8377 times.
✓ Branch 1 taken 187 times.
 8564 if (s->last_picture_ptr) {
1616 8377 ff_mpeg_unref_picture(s->avctx, &s->last_picture);
1617
2/4
✓ Branch 0 taken 8377 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 8377 times.
 16754 if (s->last_picture_ptr->f->buf[0] &&
1618 8377 (ret = ff_mpeg_ref_picture(s->avctx, &s->last_picture,
1619 s->last_picture_ptr)) < 0)
1620 ✗ return ret;
1621 }
1622
1/2
✓ Branch 0 taken 8564 times.
✗ Branch 1 not taken.
 8564 if (s->next_picture_ptr) {
1623 8564 ff_mpeg_unref_picture(s->avctx, &s->next_picture);
1624
2/4
✓ Branch 0 taken 8564 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 8564 times.
 17128 if (s->next_picture_ptr->f->buf[0] &&
1625 8564 (ret = ff_mpeg_ref_picture(s->avctx, &s->next_picture,
1626 s->next_picture_ptr)) < 0)
1627 ✗ return ret;
1628 }
1629
1630
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (s->picture_structure!= PICT_FRAME) {
1631 int i;
1632 ✗ for (i = 0; i < 4; i++) {
1633 ✗ if (s->picture_structure == PICT_BOTTOM_FIELD) {
1634 ✗ s->current_picture.f->data[i] +=
1635 ✗ s->current_picture.f->linesize[i];
1636 }
1637 ✗ s->current_picture.f->linesize[i] *= 2;
1638 ✗ s->last_picture.f->linesize[i] *= 2;
1639 ✗ s->next_picture.f->linesize[i] *= 2;
1640 }
1641 }
1642
1643
3/4
✓ Branch 0 taken 6799 times.
✓ Branch 1 taken 1765 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6799 times.
 8564 if (s->mpeg_quant || s->codec_id == AV_CODEC_ID_MPEG2VIDEO) {
1644 1765 s->dct_unquantize_intra = s->dct_unquantize_mpeg2_intra;
1645 1765 s->dct_unquantize_inter = s->dct_unquantize_mpeg2_inter;
1646
4/4
✓ Branch 0 taken 2219 times.
✓ Branch 1 taken 4580 times.
✓ Branch 2 taken 300 times.
✓ Branch 3 taken 1919 times.
 6799 } else if (s->out_format == FMT_H263 || s->out_format == FMT_H261) {
1647 4880 s->dct_unquantize_intra = s->dct_unquantize_h263_intra;
1648 4880 s->dct_unquantize_inter = s->dct_unquantize_h263_inter;
1649 } else {
1650 1919 s->dct_unquantize_intra = s->dct_unquantize_mpeg1_intra;
1651 1919 s->dct_unquantize_inter = s->dct_unquantize_mpeg1_inter;
1652 }
1653
1654
2/2
✓ Branch 0 taken 200 times.
✓ Branch 1 taken 8364 times.
 8564 if (s->dct_error_sum) {
1655 av_assert2(s->noise_reduction && s->encoding);
1656 200 update_noise_reduction(s);
1657 }
1658
1659 8564 return 0;
1660 }
1661
1662 8790 int ff_mpv_encode_picture(AVCodecContext *avctx, AVPacket *pkt,
1663 const AVFrame *pic_arg, int *got_packet)
1664 {
1665 8790 MpegEncContext *s = avctx->priv_data;
1666 int i, stuffing_count, ret;
1667 8790 int context_count = s->slice_context_count;
1668
1669 8790 s->vbv_ignore_qmax = 0;
1670
1671 8790 s->picture_in_gop_number++;
1672
1673
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 8790 times.
 8790 if (load_input_picture(s, pic_arg) < 0)
1674 ✗ return -1;
1675
1676
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 8790 times.
 8790 if (select_input_picture(s) < 0) {
1677 ✗ return -1;
1678 }
1679
1680 /* output? */
1681
2/2
✓ Branch 0 taken 8564 times.
✓ Branch 1 taken 226 times.
 8790 if (s->new_picture->data[0]) {
1682
4/4
✓ Branch 0 taken 7964 times.
✓ Branch 1 taken 600 times.
✓ Branch 2 taken 7564 times.
✓ Branch 3 taken 400 times.
 8564 int growing_buffer = context_count == 1 && !s->data_partitioning;
1683 17128 size_t pkt_size = 10000 + s->mb_width * s->mb_height *
1684
2/2
✓ Branch 0 taken 7564 times.
✓ Branch 1 taken 1000 times.
 8564 (growing_buffer ? 64 : (MAX_MB_BYTES + 100));
1685
2/2
✓ Branch 0 taken 1239 times.
✓ Branch 1 taken 7325 times.
 8564 if (CONFIG_MJPEG_ENCODER && avctx->codec_id == AV_CODEC_ID_MJPEG) {
1686 1239 ret = ff_mjpeg_add_icc_profile_size(avctx, s->new_picture, &pkt_size);
1687
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1239 times.
 1239 if (ret < 0)
1688 ✗ return ret;
1689 }
1690
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 8564 times.
 8564 if ((ret = ff_alloc_packet(avctx, pkt, pkt_size)) < 0)
1691 ✗ return ret;
1692 8564 pkt->size = avctx->internal->byte_buffer_size - AV_INPUT_BUFFER_PADDING_SIZE;
1693
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (s->mb_info) {
1694 ✗ s->mb_info_ptr = av_packet_new_side_data(pkt,
1695 AV_PKT_DATA_H263_MB_INFO,
1696 ✗ s->mb_width*s->mb_height*12);
1697 ✗ s->prev_mb_info = s->last_mb_info = s->mb_info_size = 0;
1698 }
1699
1700
2/2
✓ Branch 0 taken 9164 times.
✓ Branch 1 taken 8564 times.
 17728 for (i = 0; i < context_count; i++) {
1701 9164 int start_y = s->thread_context[i]->start_mb_y;
1702 9164 int end_y = s->thread_context[i]-> end_mb_y;
1703 9164 int h = s->mb_height;
1704 9164 uint8_t *start = pkt->data + (size_t)(((int64_t) pkt->size) * start_y / h);
1705 9164 uint8_t *end = pkt->data + (size_t)(((int64_t) pkt->size) * end_y / h);
1706
1707 9164 init_put_bits(&s->thread_context[i]->pb, start, end - start);
1708 }
1709
1710 8564 s->pict_type = s->new_picture->pict_type;
1711 //emms_c();
1712 8564 ret = frame_start(s);
1713
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (ret < 0)
1714 ✗ return ret;
1715 8564 vbv_retry:
1716 8689 ret = encode_picture(s, s->picture_number);
1717
2/2
✓ Branch 0 taken 7689 times.
✓ Branch 1 taken 1000 times.
 8689 if (growing_buffer) {
1718
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7689 times.
 7689 av_assert0(s->pb.buf == avctx->internal->byte_buffer);
1719 7689 pkt->data = s->pb.buf;
1720 7689 pkt->size = avctx->internal->byte_buffer_size;
1721 }
1722
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8689 times.
 8689 if (ret < 0)
1723 ✗ return -1;
1724
1725 8689 frame_end(s);
1726
1727
2/2
✓ Branch 0 taken 1439 times.
✓ Branch 1 taken 7250 times.
 8689 if ((CONFIG_MJPEG_ENCODER || CONFIG_AMV_ENCODER) && s->out_format == FMT_MJPEG)
1728 1439 ff_mjpeg_encode_picture_trailer(&s->pb, s->header_bits);
1729
1730
2/2
✓ Branch 0 taken 175 times.
✓ Branch 1 taken 8514 times.
 8689 if (avctx->rc_buffer_size) {
1731 175 RateControlContext *rcc = &s->rc_context;
1732
2/2
✓ Branch 0 taken 150 times.
✓ Branch 1 taken 25 times.
 175 int max_size = FFMAX(rcc->buffer_index * avctx->rc_max_available_vbv_use, rcc->buffer_index - 500);
1733
2/4
✓ Branch 0 taken 175 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 175 times.
 175 int hq = (avctx->mb_decision == FF_MB_DECISION_RD || avctx->trellis);
1734
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 175 times.
 175 int min_step = hq ? 1 : (1<<(FF_LAMBDA_SHIFT + 7))/139;
1735
1736
2/2
✓ Branch 1 taken 125 times.
✓ Branch 2 taken 50 times.
 175 if (put_bits_count(&s->pb) > max_size &&
1737
1/2
✓ Branch 0 taken 125 times.
✗ Branch 1 not taken.
 125 s->lambda < s->lmax) {
1738 125 s->next_lambda = FFMAX(s->lambda + min_step, s->lambda *
1739 (s->qscale + 1) / s->qscale);
1740
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 125 times.
 125 if (s->adaptive_quant) {
1741 int i;
1742 ✗ for (i = 0; i < s->mb_height * s->mb_stride; i++)
1743 ✗ s->lambda_table[i] =
1744 ✗ FFMAX(s->lambda_table[i] + min_step,
1745 s->lambda_table[i] * (s->qscale + 1) /
1746 s->qscale);
1747 }
1748 125 s->mb_skipped = 0; // done in frame_start()
1749 // done in encode_picture() so we must undo it
1750
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 125 times.
 125 if (s->pict_type == AV_PICTURE_TYPE_P) {
1751 ✗ if (s->flipflop_rounding ||
1752 ✗ s->codec_id == AV_CODEC_ID_H263P ||
1753 ✗ s->codec_id == AV_CODEC_ID_MPEG4)
1754 ✗ s->no_rounding ^= 1;
1755 }
1756
1/2
✓ Branch 0 taken 125 times.
✗ Branch 1 not taken.
 125 if (s->pict_type != AV_PICTURE_TYPE_B) {
1757 125 s->time_base = s->last_time_base;
1758 125 s->last_non_b_time = s->time - s->pp_time;
1759 }
1760
2/2
✓ Branch 0 taken 125 times.
✓ Branch 1 taken 125 times.
 250 for (i = 0; i < context_count; i++) {
1761 125 PutBitContext *pb = &s->thread_context[i]->pb;
1762 125 init_put_bits(pb, pb->buf, pb->buf_end - pb->buf);
1763 }
1764 125 s->vbv_ignore_qmax = 1;
1765 125 av_log(avctx, AV_LOG_VERBOSE, "reencoding frame due to VBV\n");
1766 125 goto vbv_retry;
1767 }
1768
1769
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 50 times.
 50 av_assert0(avctx->rc_max_rate);
1770 }
1771
1772
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (avctx->flags & AV_CODEC_FLAG_PASS1)
1773 ✗ ff_write_pass1_stats(s);
1774
1775
2/2
✓ Branch 0 taken 34256 times.
✓ Branch 1 taken 8564 times.
 42820 for (i = 0; i < 4; i++) {
1776 34256 s->current_picture_ptr->encoding_error[i] = s->current_picture.encoding_error[i];
1777 34256 avctx->error[i] += s->current_picture_ptr->encoding_error[i];
1778 }
1779 8564 ff_side_data_set_encoder_stats(pkt, s->current_picture.f->quality,
1780 8564 s->current_picture_ptr->encoding_error,
1781 8564 (avctx->flags&AV_CODEC_FLAG_PSNR) ? MPEGVIDEO_MAX_PLANES : 0,
1782 s->pict_type);
1783
1784 8564 if (avctx->flags & AV_CODEC_FLAG_PASS1)
1785 assert(put_bits_count(&s->pb) == s->header_bits + s->mv_bits +
1786 s->misc_bits + s->i_tex_bits +
1787 s->p_tex_bits);
1788 8564 flush_put_bits(&s->pb);
1789 8564 s->frame_bits = put_bits_count(&s->pb);
1790
1791 8564 stuffing_count = ff_vbv_update(s, s->frame_bits);
1792 8564 s->stuffing_bits = 8*stuffing_count;
1793
2/2
✓ Branch 0 taken 50 times.
✓ Branch 1 taken 8514 times.
 8564 if (stuffing_count) {
1794
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 50 times.
 50 if (put_bytes_left(&s->pb, 0) < stuffing_count + 50) {
1795 ✗ av_log(avctx, AV_LOG_ERROR, "stuffing too large\n");
1796 ✗ return -1;
1797 }
1798
1799
1/3
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
 50 switch (s->codec_id) {
1800 50 case AV_CODEC_ID_MPEG1VIDEO:
1801 case AV_CODEC_ID_MPEG2VIDEO:
1802
2/2
✓ Branch 0 taken 3648955 times.
✓ Branch 1 taken 50 times.
 3649005 while (stuffing_count--) {
1803 3648955 put_bits(&s->pb, 8, 0);
1804 }
1805 50 break;
1806 ✗ case AV_CODEC_ID_MPEG4:
1807 ✗ put_bits(&s->pb, 16, 0);
1808 ✗ put_bits(&s->pb, 16, 0x1C3);
1809 ✗ stuffing_count -= 4;
1810 ✗ while (stuffing_count--) {
1811 ✗ put_bits(&s->pb, 8, 0xFF);
1812 }
1813 ✗ break;
1814 ✗ default:
1815 ✗ av_log(avctx, AV_LOG_ERROR, "vbv buffer overflow\n");
1816 ✗ s->stuffing_bits = 0;
1817 }
1818 50 flush_put_bits(&s->pb);
1819 50 s->frame_bits = put_bits_count(&s->pb);
1820 }
1821
1822 /* update MPEG-1/2 vbv_delay for CBR */
1823
2/2
✓ Branch 0 taken 50 times.
✓ Branch 1 taken 8514 times.
 8564 if (avctx->rc_max_rate &&
1824
1/2
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
 50 avctx->rc_min_rate == avctx->rc_max_rate &&
1825
1/2
✓ Branch 0 taken 50 times.
✗ Branch 1 not taken.
 50 s->out_format == FMT_MPEG1 &&
1826 50 90000LL * (avctx->rc_buffer_size - 1) <=
1827
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 25 times.
 50 avctx->rc_max_rate * 0xFFFFLL) {
1828 AVCPBProperties *props;
1829 size_t props_size;
1830
1831 int vbv_delay, min_delay;
1832 50 double inbits = avctx->rc_max_rate *
1833 25 av_q2d(avctx->time_base);
1834 25 int minbits = s->frame_bits - 8 *
1835 25 (s->vbv_delay_pos - 1);
1836 25 double bits = s->rc_context.buffer_index + minbits - inbits;
1837 25 uint8_t *const vbv_delay_ptr = s->pb.buf + s->vbv_delay_pos;
1838
1839
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 25 times.
 25 if (bits < 0)
1840 ✗ av_log(avctx, AV_LOG_ERROR,
1841 "Internal error, negative bits\n");
1842
1843 av_assert1(s->repeat_first_field == 0);
1844
1845 25 vbv_delay = bits * 90000 / avctx->rc_max_rate;
1846 25 min_delay = (minbits * 90000LL + avctx->rc_max_rate - 1) /
1847 25 avctx->rc_max_rate;
1848
1849 25 vbv_delay = FFMAX(vbv_delay, min_delay);
1850
1851
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 25 times.
 25 av_assert0(vbv_delay < 0xFFFF);
1852
1853 25 vbv_delay_ptr[0] &= 0xF8;
1854 25 vbv_delay_ptr[0] |= vbv_delay >> 13;
1855 25 vbv_delay_ptr[1] = vbv_delay >> 5;
1856 25 vbv_delay_ptr[2] &= 0x07;
1857 25 vbv_delay_ptr[2] |= vbv_delay << 3;
1858
1859 25 props = av_cpb_properties_alloc(&props_size);
1860
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 25 times.
 25 if (!props)
1861 ✗ return AVERROR(ENOMEM);
1862 25 props->vbv_delay = vbv_delay * 300;
1863
1864 25 ret = av_packet_add_side_data(pkt, AV_PKT_DATA_CPB_PROPERTIES,
1865 (uint8_t*)props, props_size);
1866
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 25 times.
 25 if (ret < 0) {
1867 ✗ av_freep(&props);
1868 ✗ return ret;
1869 }
1870 }
1871 8564 s->total_bits += s->frame_bits;
1872
1873 8564 pkt->pts = s->current_picture.f->pts;
1874
4/4
✓ Branch 0 taken 3220 times.
✓ Branch 1 taken 5344 times.
✓ Branch 2 taken 1873 times.
✓ Branch 3 taken 1347 times.
 8564 if (!s->low_delay && s->pict_type != AV_PICTURE_TYPE_B) {
1875
2/2
✓ Branch 0 taken 69 times.
✓ Branch 1 taken 1804 times.
 1873 if (!s->current_picture.f->coded_picture_number)
1876 69 pkt->dts = pkt->pts - s->dts_delta;
1877 else
1878 1804 pkt->dts = s->reordered_pts;
1879 1873 s->reordered_pts = pkt->pts;
1880 } else
1881 6691 pkt->dts = pkt->pts;
1882
2/2
✓ Branch 0 taken 2208 times.
✓ Branch 1 taken 6356 times.
 8564 if (s->current_picture.f->key_frame)
1883 2208 pkt->flags |= AV_PKT_FLAG_KEY;
1884
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8564 times.
 8564 if (s->mb_info)
1885 ✗ av_packet_shrink_side_data(pkt, AV_PKT_DATA_H263_MB_INFO, s->mb_info_size);
1886 } else {
1887 226 s->frame_bits = 0;
1888 }
1889
1890 /* release non-reference frames */
1891
2/2
✓ Branch 0 taken 316440 times.
✓ Branch 1 taken 8790 times.
 325230 for (i = 0; i < MAX_PICTURE_COUNT; i++) {
1892
2/2
✓ Branch 0 taken 293110 times.
✓ Branch 1 taken 23330 times.
 316440 if (!s->picture[i].reference)
1893 293110 ff_mpeg_unref_picture(avctx, &s->picture[i]);
1894 }
1895
1896 av_assert1((s->frame_bits & 7) == 0);
1897
1898 8790 pkt->size = s->frame_bits / 8;
1899 8790 *got_packet = !!pkt->size;
1900 8790 return 0;
1901 }
1902
1903 ✗ static inline void dct_single_coeff_elimination(MpegEncContext *s,
1904 int n, int threshold)
1905 {
1906 static const char tab[64] = {
1907 3, 2, 2, 1, 1, 1, 1, 1,
1908 1, 1, 1, 1, 1, 1, 1, 1,
1909 1, 1, 1, 1, 1, 1, 1, 1,
1910 0, 0, 0, 0, 0, 0, 0, 0,
1911 0, 0, 0, 0, 0, 0, 0, 0,
1912 0, 0, 0, 0, 0, 0, 0, 0,
1913 0, 0, 0, 0, 0, 0, 0, 0,
1914 0, 0, 0, 0, 0, 0, 0, 0
1915 };
1916 ✗ int score = 0;
1917 ✗ int run = 0;
1918 int i;
1919 ✗ int16_t *block = s->block[n];
1920 ✗ const int last_index = s->block_last_index[n];
1921 int skip_dc;
1922
1923 ✗ if (threshold < 0) {
1924 ✗ skip_dc = 0;
1925 ✗ threshold = -threshold;
1926 } else
1927 ✗ skip_dc = 1;
1928
1929 /* Are all we could set to zero already zero? */
1930 ✗ if (last_index <= skip_dc - 1)
1931 ✗ return;
1932
1933 ✗ for (i = 0; i <= last_index; i++) {
1934 ✗ const int j = s->intra_scantable.permutated[i];
1935 ✗ const int level = FFABS(block[j]);
1936 ✗ if (level == 1) {
1937 ✗ if (skip_dc && i == 0)
1938 ✗ continue;
1939 ✗ score += tab[run];
1940 ✗ run = 0;
1941 ✗ } else if (level > 1) {
1942 ✗ return;
1943 } else {
1944 ✗ run++;
1945 }
1946 }
1947 ✗ if (score >= threshold)
1948 ✗ return;
1949 ✗ for (i = skip_dc; i <= last_index; i++) {
1950 ✗ const int j = s->intra_scantable.permutated[i];
1951 ✗ block[j] = 0;
1952 }
1953 ✗ if (block[0])
1954 ✗ s->block_last_index[n] = 0;
1955 else
1956 ✗ s->block_last_index[n] = -1;
1957 }
1958
1959 ✗ static inline void clip_coeffs(MpegEncContext *s, int16_t *block,
1960 int last_index)
1961 {
1962 int i;
1963 ✗ const int maxlevel = s->max_qcoeff;
1964 ✗ const int minlevel = s->min_qcoeff;
1965 ✗ int overflow = 0;
1966
1967 ✗ if (s->mb_intra) {
1968 ✗ i = 1; // skip clipping of intra dc
1969 } else
1970 ✗ i = 0;
1971
1972 ✗ for (; i <= last_index; i++) {
1973 ✗ const int j = s->intra_scantable.permutated[i];
1974 ✗ int level = block[j];
1975
1976 ✗ if (level > maxlevel) {
1977 ✗ level = maxlevel;
1978 ✗ overflow++;
1979 ✗ } else if (level < minlevel) {
1980 ✗ level = minlevel;
1981 ✗ overflow++;
1982 }
1983
1984 ✗ block[j] = level;
1985 }
1986
1987 ✗ if (overflow && s->avctx->mb_decision == FF_MB_DECISION_SIMPLE)
1988 ✗ av_log(s->avctx, AV_LOG_INFO,
1989 "warning, clipping %d dct coefficients to %d..%d\n",
1990 overflow, minlevel, maxlevel);
1991 }
1992
1993 ✗ static void get_visual_weight(int16_t *weight, uint8_t *ptr, int stride)
1994 {
1995 int x, y;
1996 // FIXME optimize
1997 ✗ for (y = 0; y < 8; y++) {
1998 ✗ for (x = 0; x < 8; x++) {
1999 int x2, y2;
2000 ✗ int sum = 0;
2001 ✗ int sqr = 0;
2002 ✗ int count = 0;
2003
2004 ✗ for (y2 = FFMAX(y - 1, 0); y2 < FFMIN(8, y + 2); y2++) {
2005 ✗ for (x2= FFMAX(x - 1, 0); x2 < FFMIN(8, x + 2); x2++) {
2006 ✗ int v = ptr[x2 + y2 * stride];
2007 ✗ sum += v;
2008 ✗ sqr += v * v;
2009 ✗ count++;
2010 }
2011 }
2012 ✗ weight[x + 8 * y]= (36 * ff_sqrt(count * sqr - sum * sum)) / count;
2013 }
2014 }
2015 }
2016
2017 4747268 static av_always_inline void encode_mb_internal(MpegEncContext *s,
2018 int motion_x, int motion_y,
2019 int mb_block_height,
2020 int mb_block_width,
2021 int mb_block_count,
2022 int chroma_x_shift,
2023 int chroma_y_shift,
2024 int chroma_format)
2025 {
2026 /* Interlaced DCT is only possible with MPEG-2 and MPEG-4
2027 * and neither of these encoders currently supports 444. */
2028 #define INTERLACED_DCT(s) ((chroma_format == CHROMA_420 || chroma_format == CHROMA_422) && \
2029 (s)->avctx->flags & AV_CODEC_FLAG_INTERLACED_DCT)
2030 int16_t weight[12][64];
2031 int16_t orig[12][64];
2032 4747268 const int mb_x = s->mb_x;
2033 4747268 const int mb_y = s->mb_y;
2034 int i;
2035 int skip_dct[12];
2036 4747268 int dct_offset = s->linesize * 8; // default for progressive frames
2037 4747268 int uv_dct_offset = s->uvlinesize * 8;
2038 uint8_t *ptr_y, *ptr_cb, *ptr_cr;
2039 ptrdiff_t wrap_y, wrap_c;
2040
2041
2/2
✓ Branch 0 taken 30249022 times.
✓ Branch 1 taken 4747268 times.
 34996290 for (i = 0; i < mb_block_count; i++)
2042 30249022 skip_dct[i] = s->skipdct;
2043
2044
2/2
✓ Branch 0 taken 1271769 times.
✓ Branch 1 taken 3475499 times.
 4747268 if (s->adaptive_quant) {
2045 1271769 const int last_qp = s->qscale;
2046 1271769 const int mb_xy = mb_x + mb_y * s->mb_stride;
2047
2048 1271769 s->lambda = s->lambda_table[mb_xy];
2049 1271769 update_qscale(s);
2050
2051
2/2
✓ Branch 0 taken 232420 times.
✓ Branch 1 taken 1039349 times.
 1271769 if (!(s->mpv_flags & FF_MPV_FLAG_QP_RD)) {
2052 232420 s->qscale = s->current_picture_ptr->qscale_table[mb_xy];
2053 232420 s->dquant = s->qscale - last_qp;
2054
2055
1/2
✓ Branch 0 taken 232420 times.
✗ Branch 1 not taken.
 232420 if (s->out_format == FMT_H263) {
2056 232420 s->dquant = av_clip(s->dquant, -2, 2);
2057
2058
1/2
✓ Branch 0 taken 232420 times.
✗ Branch 1 not taken.
 232420 if (s->codec_id == AV_CODEC_ID_MPEG4) {
2059
2/2
✓ Branch 0 taken 221829 times.
✓ Branch 1 taken 10591 times.
 232420 if (!s->mb_intra) {
2060
2/2
✓ Branch 0 taken 177110 times.
✓ Branch 1 taken 44719 times.
 221829 if (s->pict_type == AV_PICTURE_TYPE_B) {
2061
4/4
✓ Branch 0 taken 170173 times.
✓ Branch 1 taken 6937 times.
✓ Branch 2 taken 62215 times.
✓ Branch 3 taken 107958 times.
 177110 if (s->dquant & 1 || s->mv_dir & MV_DIRECT)
2062 69152 s->dquant = 0;
2063 }
2064
2/2
✓ Branch 0 taken 31669 times.
✓ Branch 1 taken 190160 times.
 221829 if (s->mv_type == MV_TYPE_8X8)
2065 31669 s->dquant = 0;
2066 }
2067 }
2068 }
2069 }
2070 1271769 ff_set_qscale(s, last_qp + s->dquant);
2071
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3475499 times.
 3475499 } else if (s->mpv_flags & FF_MPV_FLAG_QP_RD)
2072 ✗ ff_set_qscale(s, s->qscale + s->dquant);
2073
2074 4747268 wrap_y = s->linesize;
2075 4747268 wrap_c = s->uvlinesize;
2076 4747268 ptr_y = s->new_picture->data[0] +
2077 4747268 (mb_y * 16 * wrap_y) + mb_x * 16;
2078 4747268 ptr_cb = s->new_picture->data[1] +
2079 4747268 (mb_y * mb_block_height * wrap_c) + mb_x * mb_block_width;
2080 4747268 ptr_cr = s->new_picture->data[2] +
2081 4747268 (mb_y * mb_block_height * wrap_c) + mb_x * mb_block_width;
2082
2083
6/6
✓ Branch 0 taken 4737074 times.
✓ Branch 1 taken 10194 times.
✓ Branch 2 taken 10379 times.
✓ Branch 3 taken 4726695 times.
✓ Branch 4 taken 20323 times.
✓ Branch 5 taken 250 times.
 4747268 if((mb_x * 16 + 16 > s->width || mb_y * 16 + 16 > s->height) && s->codec_id != AV_CODEC_ID_AMV){
2084 20323 uint8_t *ebuf = s->sc.edge_emu_buffer + 38 * wrap_y;
2085 20323 int cw = (s->width + chroma_x_shift) >> chroma_x_shift;
2086 20323 int ch = (s->height + chroma_y_shift) >> chroma_y_shift;
2087 20323 s->vdsp.emulated_edge_mc(ebuf, ptr_y,
2088 wrap_y, wrap_y,
2089 16, 16, mb_x * 16, mb_y * 16,
2090 s->width, s->height);
2091 20323 ptr_y = ebuf;
2092 20323 s->vdsp.emulated_edge_mc(ebuf + 16 * wrap_y, ptr_cb,
2093 wrap_c, wrap_c,
2094 mb_block_width, mb_block_height,
2095 mb_x * mb_block_width, mb_y * mb_block_height,
2096 cw, ch);
2097 20323 ptr_cb = ebuf + 16 * wrap_y;
2098 20323 s->vdsp.emulated_edge_mc(ebuf + 16 * wrap_y + 16, ptr_cr,
2099 wrap_c, wrap_c,
2100 mb_block_width, mb_block_height,
2101 mb_x * mb_block_width, mb_y * mb_block_height,
2102 cw, ch);
2103 20323 ptr_cr = ebuf + 16 * wrap_y + 16;
2104 }
2105
2106
2/2
✓ Branch 0 taken 1210424 times.
✓ Branch 1 taken 3536844 times.
 4747268 if (s->mb_intra) {
2107
6/6
✓ Branch 0 taken 415274 times.
✓ Branch 1 taken 795150 times.
✓ Branch 2 taken 355360 times.
✓ Branch 3 taken 59914 times.
✓ Branch 4 taken 317596 times.
✓ Branch 5 taken 832914 times.
 1210424 if (INTERLACED_DCT(s)) {
2108 int progressive_score, interlaced_score;
2109
2110 317596 s->interlaced_dct = 0;
2111 317596 progressive_score = s->mecc.ildct_cmp[4](s, ptr_y, NULL, wrap_y, 8) +
2112 317596 s->mecc.ildct_cmp[4](s, ptr_y + wrap_y * 8,
2113 NULL, wrap_y, 8) - 400;
2114
2115
2/2
✓ Branch 0 taken 298126 times.
✓ Branch 1 taken 19470 times.
 317596 if (progressive_score > 0) {
2116 298126 interlaced_score = s->mecc.ildct_cmp[4](s, ptr_y,
2117 NULL, wrap_y * 2, 8) +
2118 298126 s->mecc.ildct_cmp[4](s, ptr_y + wrap_y,
2119 NULL, wrap_y * 2, 8);
2120
2/2
✓ Branch 0 taken 364 times.
✓ Branch 1 taken 297762 times.
 298126 if (progressive_score > interlaced_score) {
2121 364 s->interlaced_dct = 1;
2122
2123 364 dct_offset = wrap_y;
2124 364 uv_dct_offset = wrap_c;
2125 364 wrap_y <<= 1;
2126
3/4
✓ Branch 0 taken 185 times.
✓ Branch 1 taken 179 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 185 times.
 364 if (chroma_format == CHROMA_422 ||
2127 chroma_format == CHROMA_444)
2128 179 wrap_c <<= 1;
2129 }
2130 }
2131 }
2132
2133 1210424 s->pdsp.get_pixels(s->block[0], ptr_y, wrap_y);
2134 1210424 s->pdsp.get_pixels(s->block[1], ptr_y + 8, wrap_y);
2135 1210424 s->pdsp.get_pixels(s->block[2], ptr_y + dct_offset, wrap_y);
2136 1210424 s->pdsp.get_pixels(s->block[3], ptr_y + dct_offset + 8, wrap_y);
2137
2138
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1210424 times.
 1210424 if (s->avctx->flags & AV_CODEC_FLAG_GRAY) {
2139 ✗ skip_dct[4] = 1;
2140 ✗ skip_dct[5] = 1;
2141 } else {
2142 1210424 s->pdsp.get_pixels(s->block[4], ptr_cb, wrap_c);
2143 1210424 s->pdsp.get_pixels(s->block[5], ptr_cr, wrap_c);
2144
2/2
✓ Branch 0 taken 355360 times.
✓ Branch 1 taken 855064 times.
 1210424 if (chroma_format == CHROMA_422) {
2145 355360 s->pdsp.get_pixels(s->block[6], ptr_cb + uv_dct_offset, wrap_c);
2146 355360 s->pdsp.get_pixels(s->block[7], ptr_cr + uv_dct_offset, wrap_c);
2147
2/2
✓ Branch 0 taken 59914 times.
✓ Branch 1 taken 795150 times.
 855064 } else if (chroma_format == CHROMA_444) {
2148 59914 s->pdsp.get_pixels(s->block[ 6], ptr_cb + 8, wrap_c);
2149 59914 s->pdsp.get_pixels(s->block[ 7], ptr_cr + 8, wrap_c);
2150 59914 s->pdsp.get_pixels(s->block[ 8], ptr_cb + uv_dct_offset, wrap_c);
2151 59914 s->pdsp.get_pixels(s->block[ 9], ptr_cr + uv_dct_offset, wrap_c);
2152 59914 s->pdsp.get_pixels(s->block[10], ptr_cb + uv_dct_offset + 8, wrap_c);
2153 59914 s->pdsp.get_pixels(s->block[11], ptr_cr + uv_dct_offset + 8, wrap_c);
2154 }
2155 }
2156 } else {
2157 op_pixels_func (*op_pix)[4];
2158 qpel_mc_func (*op_qpix)[16];
2159 uint8_t *dest_y, *dest_cb, *dest_cr;
2160
2161 3536844 dest_y = s->dest[0];
2162 3536844 dest_cb = s->dest[1];
2163 3536844 dest_cr = s->dest[2];
2164
2165
4/4
✓ Branch 0 taken 959117 times.
✓ Branch 1 taken 2577727 times.
✓ Branch 2 taken 379428 times.
✓ Branch 3 taken 579689 times.
 3536844 if ((!s->no_rounding) || s->pict_type == AV_PICTURE_TYPE_B) {
2166 2957155 op_pix = s->hdsp.put_pixels_tab;
2167 2957155 op_qpix = s->qdsp.put_qpel_pixels_tab;
2168 } else {
2169 579689 op_pix = s->hdsp.put_no_rnd_pixels_tab;
2170 579689 op_qpix = s->qdsp.put_no_rnd_qpel_pixels_tab;
2171 }
2172
2173
2/2
✓ Branch 0 taken 3186778 times.
✓ Branch 1 taken 350066 times.
 3536844 if (s->mv_dir & MV_DIR_FORWARD) {
2174 3186778 ff_mpv_motion(s, dest_y, dest_cb, dest_cr, 0,
2175 3186778 s->last_picture.f->data,
2176 op_pix, op_qpix);
2177 3186778 op_pix = s->hdsp.avg_pixels_tab;
2178 3186778 op_qpix = s->qdsp.avg_qpel_pixels_tab;
2179 }
2180
2/2
✓ Branch 0 taken 1023258 times.
✓ Branch 1 taken 2513586 times.
 3536844 if (s->mv_dir & MV_DIR_BACKWARD) {
2181 1023258 ff_mpv_motion(s, dest_y, dest_cb, dest_cr, 1,
2182 1023258 s->next_picture.f->data,
2183 op_pix, op_qpix);
2184 }
2185
2186
5/6
✓ Branch 0 taken 347605 times.
✓ Branch 1 taken 3189239 times.
✓ Branch 2 taken 347605 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 505519 times.
✓ Branch 5 taken 3031325 times.
 3536844 if (INTERLACED_DCT(s)) {
2187 int progressive_score, interlaced_score;
2188
2189 505519 s->interlaced_dct = 0;
2190 505519 progressive_score = s->mecc.ildct_cmp[0](s, dest_y, ptr_y, wrap_y, 8) +
2191 505519 s->mecc.ildct_cmp[0](s, dest_y + wrap_y * 8,
2192 505519 ptr_y + wrap_y * 8,
2193 wrap_y, 8) - 400;
2194
2195
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 505519 times.
 505519 if (s->avctx->ildct_cmp == FF_CMP_VSSE)
2196 ✗ progressive_score -= 400;
2197
2198
2/2
✓ Branch 0 taken 451992 times.
✓ Branch 1 taken 53527 times.
 505519 if (progressive_score > 0) {
2199 451992 interlaced_score = s->mecc.ildct_cmp[0](s, dest_y, ptr_y,
2200 wrap_y * 2, 8) +
2201 451992 s->mecc.ildct_cmp[0](s, dest_y + wrap_y,
2202 ptr_y + wrap_y,
2203 wrap_y * 2, 8);
2204
2205
2/2
✓ Branch 0 taken 11826 times.
✓ Branch 1 taken 440166 times.
 451992 if (progressive_score > interlaced_score) {
2206 11826 s->interlaced_dct = 1;
2207
2208 11826 dct_offset = wrap_y;
2209 11826 uv_dct_offset = wrap_c;
2210 11826 wrap_y <<= 1;
2211
2/2
✓ Branch 0 taken 8280 times.
✓ Branch 1 taken 3546 times.
 11826 if (chroma_format == CHROMA_422)
2212 8280 wrap_c <<= 1;
2213 }
2214 }
2215 }
2216
2217 3536844 s->pdsp.diff_pixels(s->block[0], ptr_y, dest_y, wrap_y);
2218 3536844 s->pdsp.diff_pixels(s->block[1], ptr_y + 8, dest_y + 8, wrap_y);
2219 3536844 s->pdsp.diff_pixels(s->block[2], ptr_y + dct_offset,
2220 3536844 dest_y + dct_offset, wrap_y);
2221 3536844 s->pdsp.diff_pixels(s->block[3], ptr_y + dct_offset + 8,
2222 3536844 dest_y + dct_offset + 8, wrap_y);
2223
2224
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3536844 times.
 3536844 if (s->avctx->flags & AV_CODEC_FLAG_GRAY) {
2225 ✗ skip_dct[4] = 1;
2226 ✗ skip_dct[5] = 1;
2227 } else {
2228 3536844 s->pdsp.diff_pixels(s->block[4], ptr_cb, dest_cb, wrap_c);
2229 3536844 s->pdsp.diff_pixels(s->block[5], ptr_cr, dest_cr, wrap_c);
2230
2/2
✓ Branch 0 taken 347605 times.
✓ Branch 1 taken 3189239 times.
 3536844 if (!chroma_y_shift) { /* 422 */
2231 347605 s->pdsp.diff_pixels(s->block[6], ptr_cb + uv_dct_offset,
2232 347605 dest_cb + uv_dct_offset, wrap_c);
2233 347605 s->pdsp.diff_pixels(s->block[7], ptr_cr + uv_dct_offset,
2234 347605 dest_cr + uv_dct_offset, wrap_c);
2235 }
2236 }
2237 /* pre quantization */
2238 3536844 if (s->current_picture.mc_mb_var[s->mb_stride * mb_y + mb_x] <
2239
2/2
✓ Branch 0 taken 2960929 times.
✓ Branch 1 taken 575915 times.
 3536844 2 * s->qscale * s->qscale) {
2240 // FIXME optimize
2241
2/2
✓ Branch 1 taken 956984 times.
✓ Branch 2 taken 2003945 times.
 2960929 if (s->mecc.sad[1](NULL, ptr_y, dest_y, wrap_y, 8) < 20 * s->qscale)
2242 956984 skip_dct[0] = 1;
2243
2/2
✓ Branch 1 taken 974455 times.
✓ Branch 2 taken 1986474 times.
 2960929 if (s->mecc.sad[1](NULL, ptr_y + 8, dest_y + 8, wrap_y, 8) < 20 * s->qscale)
2244 974455 skip_dct[1] = 1;
2245 2960929 if (s->mecc.sad[1](NULL, ptr_y + dct_offset, dest_y + dct_offset,
2246
2/2
✓ Branch 0 taken 975946 times.
✓ Branch 1 taken 1984983 times.
 2960929 wrap_y, 8) < 20 * s->qscale)
2247 975946 skip_dct[2] = 1;
2248 2960929 if (s->mecc.sad[1](NULL, ptr_y + dct_offset + 8, dest_y + dct_offset + 8,
2249
2/2
✓ Branch 0 taken 948997 times.
✓ Branch 1 taken 2011932 times.
 2960929 wrap_y, 8) < 20 * s->qscale)
2250 948997 skip_dct[3] = 1;
2251
2/2
✓ Branch 1 taken 1305959 times.
✓ Branch 2 taken 1654970 times.
 2960929 if (s->mecc.sad[1](NULL, ptr_cb, dest_cb, wrap_c, 8) < 20 * s->qscale)
2252 1305959 skip_dct[4] = 1;
2253
2/2
✓ Branch 1 taken 1274965 times.
✓ Branch 2 taken 1685964 times.
 2960929 if (s->mecc.sad[1](NULL, ptr_cr, dest_cr, wrap_c, 8) < 20 * s->qscale)
2254 1274965 skip_dct[5] = 1;
2255
2/2
✓ Branch 0 taken 294319 times.
✓ Branch 1 taken 2666610 times.
 2960929 if (!chroma_y_shift) { /* 422 */
2256 294319 if (s->mecc.sad[1](NULL, ptr_cb + uv_dct_offset,
2257 dest_cb + uv_dct_offset,
2258
2/2
✓ Branch 0 taken 153422 times.
✓ Branch 1 taken 140897 times.
 294319 wrap_c, 8) < 20 * s->qscale)
2259 153422 skip_dct[6] = 1;
2260 294319 if (s->mecc.sad[1](NULL, ptr_cr + uv_dct_offset,
2261 dest_cr + uv_dct_offset,
2262
2/2
✓ Branch 0 taken 147975 times.
✓ Branch 1 taken 146344 times.
 294319 wrap_c, 8) < 20 * s->qscale)
2263 147975 skip_dct[7] = 1;
2264 }
2265 }
2266 }
2267
2268
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4747268 times.
 4747268 if (s->quantizer_noise_shaping) {
2269 ✗ if (!skip_dct[0])
2270 ✗ get_visual_weight(weight[0], ptr_y , wrap_y);
2271 ✗ if (!skip_dct[1])
2272 ✗ get_visual_weight(weight[1], ptr_y + 8, wrap_y);
2273 ✗ if (!skip_dct[2])
2274 ✗ get_visual_weight(weight[2], ptr_y + dct_offset , wrap_y);
2275 ✗ if (!skip_dct[3])
2276 ✗ get_visual_weight(weight[3], ptr_y + dct_offset + 8, wrap_y);
2277 ✗ if (!skip_dct[4])
2278 ✗ get_visual_weight(weight[4], ptr_cb , wrap_c);
2279 ✗ if (!skip_dct[5])
2280 ✗ get_visual_weight(weight[5], ptr_cr , wrap_c);
2281 ✗ if (!chroma_y_shift) { /* 422 */
2282 ✗ if (!skip_dct[6])
2283 ✗ get_visual_weight(weight[6], ptr_cb + uv_dct_offset,
2284 wrap_c);
2285 ✗ if (!skip_dct[7])
2286 ✗ get_visual_weight(weight[7], ptr_cr + uv_dct_offset,
2287 wrap_c);
2288 }
2289 ✗ memcpy(orig[0], s->block[0], sizeof(int16_t) * 64 * mb_block_count);
2290 }
2291
2292 /* DCT & quantize */
2293 av_assert2(s->out_format != FMT_MJPEG || s->qscale == 8);
2294 {
2295
2/2
✓ Branch 0 taken 30249022 times.
✓ Branch 1 taken 4747268 times.
 34996290 for (i = 0; i < mb_block_count; i++) {
2296
2/2
✓ Branch 0 taken 23510319 times.
✓ Branch 1 taken 6738703 times.
 30249022 if (!skip_dct[i]) {
2297 int overflow;
2298 23510319 s->block_last_index[i] = s->dct_quantize(s, s->block[i], i, s->qscale, &overflow);
2299 // FIXME we could decide to change to quantizer instead of
2300 // clipping
2301 // JS: I don't think that would be a good idea it could lower
2302 // quality instead of improve it. Just INTRADC clipping
2303 // deserves changes in quantizer
2304
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 23510319 times.
 23510319 if (overflow)
2305 ✗ clip_coeffs(s, s->block[i], s->block_last_index[i]);
2306 } else
2307 6738703 s->block_last_index[i] = -1;
2308 }
2309
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4747268 times.
 4747268 if (s->quantizer_noise_shaping) {
2310 ✗ for (i = 0; i < mb_block_count; i++) {
2311 ✗ if (!skip_dct[i]) {
2312 ✗ s->block_last_index[i] =
2313 ✗ dct_quantize_refine(s, s->block[i], weight[i],
2314 ✗ orig[i], i, s->qscale);
2315 }
2316 }
2317 }
2318
2319
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4747268 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4747268 if (s->luma_elim_threshold && !s->mb_intra)
2320 ✗ for (i = 0; i < 4; i++)
2321 ✗ dct_single_coeff_elimination(s, i, s->luma_elim_threshold);
2322
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4747268 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4747268 if (s->chroma_elim_threshold && !s->mb_intra)
2323 ✗ for (i = 4; i < mb_block_count; i++)
2324 ✗ dct_single_coeff_elimination(s, i, s->chroma_elim_threshold);
2325
2326
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4747268 times.
 4747268 if (s->mpv_flags & FF_MPV_FLAG_CBP_RD) {
2327 ✗ for (i = 0; i < mb_block_count; i++) {
2328 ✗ if (s->block_last_index[i] == -1)
2329 ✗ s->coded_score[i] = INT_MAX / 256;
2330 }
2331 }
2332 }
2333
2334
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4747268 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4747268 if ((s->avctx->flags & AV_CODEC_FLAG_GRAY) && s->mb_intra) {
2335 ✗ s->block_last_index[4] =
2336 ✗ s->block_last_index[5] = 0;
2337 ✗ s->block[4][0] =
2338 ✗ s->block[5][0] = (1024 + s->c_dc_scale / 2) / s->c_dc_scale;
2339 ✗ if (!chroma_y_shift) { /* 422 / 444 */
2340 ✗ for (i=6; i<12; i++) {
2341 ✗ s->block_last_index[i] = 0;
2342 ✗ s->block[i][0] = s->block[4][0];
2343 }
2344 }
2345 }
2346
2347 // non c quantize code returns incorrect block_last_index FIXME
2348
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4747268 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4747268 if (s->alternate_scan && s->dct_quantize != ff_dct_quantize_c) {
2349 ✗ for (i = 0; i < mb_block_count; i++) {
2350 int j;
2351 ✗ if (s->block_last_index[i] > 0) {
2352 ✗ for (j = 63; j > 0; j--) {
2353 ✗ if (s->block[i][s->intra_scantable.permutated[j]])
2354 ✗ break;
2355 }
2356 ✗ s->block_last_index[i] = j;
2357 }
2358 }
2359 }
2360
2361 /* huffman encode */
2362
7/9
✓ Branch 0 taken 1711467 times.
✓ Branch 1 taken 1790211 times.
✓ Branch 2 taken 189450 times.
✓ Branch 3 taken 59850 times.
✓ Branch 4 taken 133678 times.
✓ Branch 5 taken 428550 times.
✓ Branch 6 taken 434062 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
 4747268 switch(s->codec_id){ //FIXME funct ptr could be slightly faster
2363 1711467 case AV_CODEC_ID_MPEG1VIDEO:
2364 case AV_CODEC_ID_MPEG2VIDEO:
2365 if (CONFIG_MPEG1VIDEO_ENCODER || CONFIG_MPEG2VIDEO_ENCODER)
2366 1711467 ff_mpeg1_encode_mb(s, s->block, motion_x, motion_y);
2367 1711467 break;
2368 1790211 case AV_CODEC_ID_MPEG4:
2369 if (CONFIG_MPEG4_ENCODER)
2370 1790211 ff_mpeg4_encode_mb(s, s->block, motion_x, motion_y);
2371 1790211 break;
2372 189450 case AV_CODEC_ID_MSMPEG4V2:
2373 case AV_CODEC_ID_MSMPEG4V3:
2374 case AV_CODEC_ID_WMV1:
2375 if (CONFIG_MSMPEG4_ENCODER)
2376 189450 ff_msmpeg4_encode_mb(s, s->block, motion_x, motion_y);
2377 189450 break;
2378 59850 case AV_CODEC_ID_WMV2:
2379 if (CONFIG_WMV2_ENCODER)
2380 59850 ff_wmv2_encode_mb(s, s->block, motion_x, motion_y);
2381 59850 break;
2382 133678 case AV_CODEC_ID_H261:
2383 if (CONFIG_H261_ENCODER)
2384 133678 ff_h261_encode_mb(s, s->block, motion_x, motion_y);
2385 133678 break;
2386 428550 case AV_CODEC_ID_H263:
2387 case AV_CODEC_ID_H263P:
2388 case AV_CODEC_ID_FLV1:
2389 case AV_CODEC_ID_RV10:
2390 case AV_CODEC_ID_RV20:
2391 if (CONFIG_H263_ENCODER)
2392 428550 ff_h263_encode_mb(s, s->block, motion_x, motion_y);
2393 428550 break;
2394 #if CONFIG_MJPEG_ENCODER || CONFIG_AMV_ENCODER
2395 434062 case AV_CODEC_ID_MJPEG:
2396 case AV_CODEC_ID_AMV:
2397 434062 ff_mjpeg_encode_mb(s, s->block);
2398 434062 break;
2399 #endif
2400 ✗ case AV_CODEC_ID_SPEEDHQ:
2401 if (CONFIG_SPEEDHQ_ENCODER)
2402 ✗ ff_speedhq_encode_mb(s, s->block);
2403 ✗ break;
2404 4747268 default:
2405 av_assert1(0);
2406 }
2407 4747268 }
2408
2409 4747268 static av_always_inline void encode_mb(MpegEncContext *s, int motion_x, int motion_y)
2410 {
2411
2/2
✓ Branch 0 taken 3984389 times.
✓ Branch 1 taken 762879 times.
 4747268 if (s->chroma_format == CHROMA_420)
2412 3984389 encode_mb_internal(s, motion_x, motion_y, 8, 8, 6, 1, 1, CHROMA_420);
2413
2/2
✓ Branch 0 taken 702965 times.
✓ Branch 1 taken 59914 times.
 762879 else if (s->chroma_format == CHROMA_422)
2414 702965 encode_mb_internal(s, motion_x, motion_y, 16, 8, 8, 1, 0, CHROMA_422);
2415 else
2416 59914 encode_mb_internal(s, motion_x, motion_y, 16, 16, 12, 0, 0, CHROMA_444);
2417 4747268 }
2418
2419 2583315 static inline void copy_context_before_encode(MpegEncContext *d,
2420 const MpegEncContext *s)
2421 {
2422 int i;
2423
2424 2583315 memcpy(d->last_mv, s->last_mv, 2*2*2*sizeof(int)); //FIXME is memcpy faster than a loop?
2425
2426 /* MPEG-1 */
2427 2583315 d->mb_skip_run= s->mb_skip_run;
2428
2/2
✓ Branch 0 taken 7749945 times.
✓ Branch 1 taken 2583315 times.
 10333260 for(i=0; i<3; i++)
2429 7749945 d->last_dc[i] = s->last_dc[i];
2430
2431 /* statistics */
2432 2583315 d->mv_bits= s->mv_bits;
2433 2583315 d->i_tex_bits= s->i_tex_bits;
2434 2583315 d->p_tex_bits= s->p_tex_bits;
2435 2583315 d->i_count= s->i_count;
2436 2583315 d->skip_count= s->skip_count;
2437 2583315 d->misc_bits= s->misc_bits;
2438 2583315 d->last_bits= 0;
2439
2440 2583315 d->mb_skipped= 0;
2441 2583315 d->qscale= s->qscale;
2442 2583315 d->dquant= s->dquant;
2443
2444 2583315 d->esc3_level_length= s->esc3_level_length;
2445 2583315 }
2446
2447 1525818 static inline void copy_context_after_encode(MpegEncContext *d,
2448 const MpegEncContext *s)
2449 {
2450 int i;
2451
2452 1525818 memcpy(d->mv, s->mv, 2*4*2*sizeof(int));
2453 1525818 memcpy(d->last_mv, s->last_mv, 2*2*2*sizeof(int)); //FIXME is memcpy faster than a loop?
2454
2455 /* MPEG-1 */
2456 1525818 d->mb_skip_run= s->mb_skip_run;
2457
2/2
✓ Branch 0 taken 4577454 times.
✓ Branch 1 taken 1525818 times.
 6103272 for(i=0; i<3; i++)
2458 4577454 d->last_dc[i] = s->last_dc[i];
2459
2460 /* statistics */
2461 1525818 d->mv_bits= s->mv_bits;
2462 1525818 d->i_tex_bits= s->i_tex_bits;
2463 1525818 d->p_tex_bits= s->p_tex_bits;
2464 1525818 d->i_count= s->i_count;
2465 1525818 d->skip_count= s->skip_count;
2466 1525818 d->misc_bits= s->misc_bits;
2467
2468 1525818 d->mb_intra= s->mb_intra;
2469 1525818 d->mb_skipped= s->mb_skipped;
2470 1525818 d->mv_type= s->mv_type;
2471 1525818 d->mv_dir= s->mv_dir;
2472 1525818 d->pb= s->pb;
2473
2/2
✓ Branch 0 taken 373903 times.
✓ Branch 1 taken 1151915 times.
 1525818 if(s->data_partitioning){
2474 373903 d->pb2= s->pb2;
2475 373903 d->tex_pb= s->tex_pb;
2476 }
2477 1525818 d->block= s->block;
2478
2/2
✓ Branch 0 taken 12206544 times.
✓ Branch 1 taken 1525818 times.
 13732362 for(i=0; i<8; i++)
2479 12206544 d->block_last_index[i]= s->block_last_index[i];
2480 1525818 d->interlaced_dct= s->interlaced_dct;
2481 1525818 d->qscale= s->qscale;
2482
2483 1525818 d->esc3_level_length= s->esc3_level_length;
2484 1525818 }
2485
2486 2051423 static inline void encode_mb_hq(MpegEncContext *s, MpegEncContext *backup, MpegEncContext *best,
2487 PutBitContext pb[2], PutBitContext pb2[2], PutBitContext tex_pb[2],
2488 int *dmin, int *next_block, int motion_x, int motion_y)
2489 {
2490 int score;
2491 uint8_t *dest_backup[3];
2492
2493 2051423 copy_context_before_encode(s, backup);
2494
2495 2051423 s->block= s->blocks[*next_block];
2496 2051423 s->pb= pb[*next_block];
2497
2/2
✓ Branch 0 taken 376394 times.
✓ Branch 1 taken 1675029 times.
 2051423 if(s->data_partitioning){
2498 376394 s->pb2 = pb2 [*next_block];
2499 376394 s->tex_pb= tex_pb[*next_block];
2500 }
2501
2502
2/2
✓ Branch 0 taken 1083400 times.
✓ Branch 1 taken 968023 times.
 2051423 if(*next_block){
2503 1083400 memcpy(dest_backup, s->dest, sizeof(s->dest));
2504 1083400 s->dest[0] = s->sc.rd_scratchpad;
2505 1083400 s->dest[1] = s->sc.rd_scratchpad + 16*s->linesize;
2506 1083400 s->dest[2] = s->sc.rd_scratchpad + 16*s->linesize + 8;
2507
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1083400 times.
 1083400 av_assert0(s->linesize >= 32); //FIXME
2508 }
2509
2510 2051423 encode_mb(s, motion_x, motion_y);
2511
2512 2051423 score= put_bits_count(&s->pb);
2513
2/2
✓ Branch 0 taken 376394 times.
✓ Branch 1 taken 1675029 times.
 2051423 if(s->data_partitioning){
2514 376394 score+= put_bits_count(&s->pb2);
2515 376394 score+= put_bits_count(&s->tex_pb);
2516 }
2517
2518
2/2
✓ Branch 0 taken 1674464 times.
✓ Branch 1 taken 376959 times.
 2051423 if(s->avctx->mb_decision == FF_MB_DECISION_RD){
2519 1674464 ff_mpv_reconstruct_mb(s, s->block);
2520
2521 1674464 score *= s->lambda2;
2522 1674464 score += sse_mb(s) << FF_LAMBDA_SHIFT;
2523 }
2524
2525
2/2
✓ Branch 0 taken 1083400 times.
✓ Branch 1 taken 968023 times.
 2051423 if(*next_block){
2526 1083400 memcpy(s->dest, dest_backup, sizeof(s->dest));
2527 }
2528
2529
2/2
✓ Branch 0 taken 993926 times.
✓ Branch 1 taken 1057497 times.
 2051423 if(score<*dmin){
2530 993926 *dmin= score;
2531 993926 *next_block^=1;
2532
2533 993926 copy_context_after_encode(best, s);
2534 }
2535 2051423 }
2536
2537 24804 static int sse(MpegEncContext *s, uint8_t *src1, uint8_t *src2, int w, int h, int stride){
2538 24804 const uint32_t *sq = ff_square_tab + 256;
2539 24804 int acc=0;
2540 int x,y;
2541
2542
3/4
✓ Branch 0 taken 3765 times.
✓ Branch 1 taken 21039 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 3765 times.
 24804 if(w==16 && h==16)
2543 ✗ return s->mecc.sse[0](NULL, src1, src2, stride, 16);
2544
3/4
✓ Branch 0 taken 7530 times.
✓ Branch 1 taken 17274 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 7530 times.
 24804 else if(w==8 && h==8)
2545 ✗ return s->mecc.sse[1](NULL, src1, src2, stride, 8);
2546
2547
2/2
✓ Branch 0 taken 107368 times.
✓ Branch 1 taken 24804 times.
 132172 for(y=0; y<h; y++){
2548
2/2
✓ Branch 0 taken 290118 times.
✓ Branch 1 taken 107368 times.
 397486 for(x=0; x<w; x++){
2549 290118 acc+= sq[src1[x + y*stride] - src2[x + y*stride]];
2550 }
2551 }
2552
2553 av_assert2(acc>=0);
2554
2555 24804 return acc;
2556 }
2557
2558 1674464 static int sse_mb(MpegEncContext *s){
2559 1674464 int w= 16;
2560 1674464 int h= 16;
2561
2562
2/2
✓ Branch 0 taken 4503 times.
✓ Branch 1 taken 1669961 times.
 1674464 if(s->mb_x*16 + 16 > s->width ) w= s->width - s->mb_x*16;
2563
2/2
✓ Branch 0 taken 5469 times.
✓ Branch 1 taken 1668995 times.
 1674464 if(s->mb_y*16 + 16 > s->height) h= s->height- s->mb_y*16;
2564
2565
4/4
✓ Branch 0 taken 1669961 times.
✓ Branch 1 taken 4503 times.
✓ Branch 2 taken 1666196 times.
✓ Branch 3 taken 3765 times.
 1674464 if(w==16 && h==16)
2566
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1666196 times.
 1666196 if(s->avctx->mb_cmp == FF_CMP_NSSE){
2567 ✗ return s->mecc.nsse[0](s, s->new_picture->data[0] + s->mb_x * 16 + s->mb_y * s->linesize * 16, s->dest[0], s->linesize, 16) +
2568 ✗ s->mecc.nsse[1](s, s->new_picture->data[1] + s->mb_x * 8 + s->mb_y * s->uvlinesize * 8, s->dest[1], s->uvlinesize, 8) +
2569 ✗ s->mecc.nsse[1](s, s->new_picture->data[2] + s->mb_x * 8 + s->mb_y * s->uvlinesize * 8, s->dest[2], s->uvlinesize, 8);
2570 }else{
2571 1666196 return s->mecc.sse[0](NULL, s->new_picture->data[0] + s->mb_x * 16 + s->mb_y * s->linesize * 16, s->dest[0], s->linesize, 16) +
2572 3332392 s->mecc.sse[1](NULL, s->new_picture->data[1] + s->mb_x * 8 + s->mb_y * s->uvlinesize * 8, s->dest[1], s->uvlinesize, 8) +
2573 1666196 s->mecc.sse[1](NULL, s->new_picture->data[2] + s->mb_x * 8 + s->mb_y * s->uvlinesize * 8, s->dest[2], s->uvlinesize, 8);
2574 }
2575 else
2576 8268 return sse(s, s->new_picture->data[0] + s->mb_x*16 + s->mb_y*s->linesize*16, s->dest[0], w, h, s->linesize)
2577 8268 +sse(s, s->new_picture->data[1] + s->mb_x*8 + s->mb_y*s->uvlinesize*8,s->dest[1], w>>1, h>>1, s->uvlinesize)
2578 8268 +sse(s, s->new_picture->data[2] + s->mb_x*8 + s->mb_y*s->uvlinesize*8,s->dest[2], w>>1, h>>1, s->uvlinesize);
2579 }
2580
2581 ✗ static int pre_estimate_motion_thread(AVCodecContext *c, void *arg){
2582 ✗ MpegEncContext *s= *(void**)arg;
2583
2584
2585 ✗ s->me.pre_pass=1;
2586 ✗ s->me.dia_size= s->avctx->pre_dia_size;
2587 ✗ s->first_slice_line=1;
2588 ✗ for(s->mb_y= s->end_mb_y-1; s->mb_y >= s->start_mb_y; s->mb_y--) {
2589 ✗ for(s->mb_x=s->mb_width-1; s->mb_x >=0 ;s->mb_x--) {
2590 ✗ ff_pre_estimate_p_frame_motion(s, s->mb_x, s->mb_y);
2591 }
2592 ✗ s->first_slice_line=0;
2593 }
2594
2595 ✗ s->me.pre_pass=0;
2596
2597 ✗ return 0;
2598 }
2599
2600 6930 static int estimate_motion_thread(AVCodecContext *c, void *arg){
2601 6930 MpegEncContext *s= *(void**)arg;
2602
2603 6930 s->me.dia_size= s->avctx->dia_size;
2604 6930 s->first_slice_line=1;
2605
2/2
✓ Branch 0 taken 101061 times.
✓ Branch 1 taken 6930 times.
 107991 for(s->mb_y= s->start_mb_y; s->mb_y < s->end_mb_y; s->mb_y++) {
2606 101061 s->mb_x=0; //for block init below
2607 101061 ff_init_block_index(s);
2608
2/2
✓ Branch 0 taken 2274759 times.
✓ Branch 1 taken 101061 times.
 2375820 for(s->mb_x=0; s->mb_x < s->mb_width; s->mb_x++) {
2609 2274759 s->block_index[0]+=2;
2610 2274759 s->block_index[1]+=2;
2611 2274759 s->block_index[2]+=2;
2612 2274759 s->block_index[3]+=2;
2613
2614 /* compute motion vector & mb_type and store in context */
2615
2/2
✓ Branch 0 taken 408312 times.
✓ Branch 1 taken 1866447 times.
 2274759 if(s->pict_type==AV_PICTURE_TYPE_B)
2616 408312 ff_estimate_b_frame_motion(s, s->mb_x, s->mb_y);
2617 else
2618 1866447 ff_estimate_p_frame_motion(s, s->mb_x, s->mb_y);
2619 }
2620 101061 s->first_slice_line=0;
2621 }
2622 6930 return 0;
2623 }
2624
2625 379 static int mb_var_thread(AVCodecContext *c, void *arg){
2626 379 MpegEncContext *s= *(void**)arg;
2627 int mb_x, mb_y;
2628
2629
2/2
✓ Branch 0 taken 5404 times.
✓ Branch 1 taken 379 times.
 5783 for(mb_y=s->start_mb_y; mb_y < s->end_mb_y; mb_y++) {
2630
2/2
✓ Branch 0 taken 123277 times.
✓ Branch 1 taken 5404 times.
 128681 for(mb_x=0; mb_x < s->mb_width; mb_x++) {
2631 123277 int xx = mb_x * 16;
2632 123277 int yy = mb_y * 16;
2633 123277 uint8_t *pix = s->new_picture->data[0] + (yy * s->linesize) + xx;
2634 int varc;
2635 123277 int sum = s->mpvencdsp.pix_sum(pix, s->linesize);
2636
2637 123277 varc = (s->mpvencdsp.pix_norm1(pix, s->linesize) -
2638 123277 (((unsigned) sum * sum) >> 8) + 500 + 128) >> 8;
2639
2640 123277 s->current_picture.mb_var [s->mb_stride * mb_y + mb_x] = varc;
2641 123277 s->current_picture.mb_mean[s->mb_stride * mb_y + mb_x] = (sum+128)>>8;
2642 123277 s->me.mb_var_sum_temp += varc;
2643 }
2644 }
2645 379 return 0;
2646 }
2647
2648 305424 static void write_slice_end(MpegEncContext *s){
2649
2/2
✓ Branch 0 taken 5913 times.
✓ Branch 1 taken 299511 times.
 305424 if(CONFIG_MPEG4_ENCODER && s->codec_id==AV_CODEC_ID_MPEG4){
2650
2/2
✓ Branch 0 taken 1760 times.
✓ Branch 1 taken 4153 times.
 5913 if(s->partitioned_frame){
2651 1760 ff_mpeg4_merge_partitions(s);
2652 }
2653
2654 5913 ff_mpeg4_stuffing(&s->pb);
2655 299511 } else if ((CONFIG_MJPEG_ENCODER || CONFIG_AMV_ENCODER) &&
2656
2/2
✓ Branch 0 taken 1439 times.
✓ Branch 1 taken 298072 times.
 299511 s->out_format == FMT_MJPEG) {
2657 1439 ff_mjpeg_encode_stuffing(s);
2658
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 298072 times.
 298072 } else if (CONFIG_SPEEDHQ_ENCODER && s->out_format == FMT_SPEEDHQ) {
2659 ✗ ff_speedhq_end_slice(s);
2660 }
2661
2662 305424 flush_put_bits(&s->pb);
2663
2664
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 305424 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 305424 if ((s->avctx->flags & AV_CODEC_FLAG_PASS1) && !s->partitioned_frame)
2665 ✗ s->misc_bits+= get_bits_diff(s);
2666 305424 }
2667
2668 ✗ static void write_mb_info(MpegEncContext *s)
2669 {
2670 ✗ uint8_t *ptr = s->mb_info_ptr + s->mb_info_size - 12;
2671 ✗ int offset = put_bits_count(&s->pb);
2672 ✗ int mba = s->mb_x + s->mb_width * (s->mb_y % s->gob_index);
2673 ✗ int gobn = s->mb_y / s->gob_index;
2674 int pred_x, pred_y;
2675 if (CONFIG_H263_ENCODER)
2676 ✗ ff_h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
2677 ✗ bytestream_put_le32(&ptr, offset);
2678 ✗ bytestream_put_byte(&ptr, s->qscale);
2679 ✗ bytestream_put_byte(&ptr, gobn);
2680 ✗ bytestream_put_le16(&ptr, mba);
2681 ✗ bytestream_put_byte(&ptr, pred_x); /* hmv1 */
2682 ✗ bytestream_put_byte(&ptr, pred_y); /* vmv1 */
2683 /* 4MV not implemented */
2684 ✗ bytestream_put_byte(&ptr, 0); /* hmv2 */
2685 ✗ bytestream_put_byte(&ptr, 0); /* vmv2 */
2686 }
2687
2688 3230281 static void update_mb_info(MpegEncContext *s, int startcode)
2689 {
2690
1/2
✓ Branch 0 taken 3230281 times.
✗ Branch 1 not taken.
 3230281 if (!s->mb_info)
2691 3230281 return;
2692 ✗ if (put_bytes_count(&s->pb, 0) - s->prev_mb_info >= s->mb_info) {
2693 ✗ s->mb_info_size += 12;
2694 ✗ s->prev_mb_info = s->last_mb_info;
2695 }
2696 ✗ if (startcode) {
2697 ✗ s->prev_mb_info = put_bytes_count(&s->pb, 0);
2698 /* This might have incremented mb_info_size above, and we return without
2699 * actually writing any info into that slot yet. But in that case,
2700 * this will be called again at the start of the after writing the
2701 * start code, actually writing the mb info. */
2702 ✗ return;
2703 }
2704
2705 ✗ s->last_mb_info = put_bytes_count(&s->pb, 0);
2706 ✗ if (!s->mb_info_size)
2707 ✗ s->mb_info_size += 12;
2708 ✗ write_mb_info(s);
2709 }
2710
2711 3230415 int ff_mpv_reallocate_putbitbuffer(MpegEncContext *s, size_t threshold, size_t size_increase)
2712 {
2713
2/2
✓ Branch 1 taken 24 times.
✓ Branch 2 taken 3230391 times.
 3230415 if (put_bytes_left(&s->pb, 0) < threshold
2714
1/2
✓ Branch 0 taken 24 times.
✗ Branch 1 not taken.
 24 && s->slice_context_count == 1
2715
1/2
✓ Branch 0 taken 24 times.
✗ Branch 1 not taken.
 24 && s->pb.buf == s->avctx->internal->byte_buffer) {
2716 24 int lastgob_pos = s->ptr_lastgob - s->pb.buf;
2717
2718 24 uint8_t *new_buffer = NULL;
2719 24 int new_buffer_size = 0;
2720
2721
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24 times.
 24 if ((s->avctx->internal->byte_buffer_size + size_increase) >= INT_MAX/8) {
2722 ✗ av_log(s->avctx, AV_LOG_ERROR, "Cannot reallocate putbit buffer\n");
2723 ✗ return AVERROR(ENOMEM);
2724 }
2725
2726 24 emms_c();
2727
2728 24 av_fast_padded_malloc(&new_buffer, &new_buffer_size,
2729 24 s->avctx->internal->byte_buffer_size + size_increase);
2730
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 24 times.
 24 if (!new_buffer)
2731 ✗ return AVERROR(ENOMEM);
2732
2733 24 memcpy(new_buffer, s->avctx->internal->byte_buffer, s->avctx->internal->byte_buffer_size);
2734 24 av_free(s->avctx->internal->byte_buffer);
2735 24 s->avctx->internal->byte_buffer = new_buffer;
2736 24 s->avctx->internal->byte_buffer_size = new_buffer_size;
2737 24 rebase_put_bits(&s->pb, new_buffer, new_buffer_size);
2738 24 s->ptr_lastgob = s->pb.buf + lastgob_pos;
2739 }
2740
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3230415 times.
 3230415 if (put_bytes_left(&s->pb, 0) < threshold)
2741 ✗ return AVERROR(EINVAL);
2742 3230415 return 0;
2743 }
2744
2745 9289 static int encode_thread(AVCodecContext *c, void *arg){
2746 9289 MpegEncContext *s= *(void**)arg;
2747 int mb_x, mb_y, mb_y_order;
2748 9289 int chr_h= 16>>s->chroma_y_shift;
2749 int i, j;
2750 9289 MpegEncContext best_s = { 0 }, backup_s;
2751 uint8_t bit_buf[2][MAX_MB_BYTES];
2752 uint8_t bit_buf2[2][MAX_MB_BYTES];
2753 uint8_t bit_buf_tex[2][MAX_MB_BYTES];
2754 PutBitContext pb[2], pb2[2], tex_pb[2];
2755
2756
2/2
✓ Branch 0 taken 18578 times.
✓ Branch 1 taken 9289 times.
 27867 for(i=0; i<2; i++){
2757 18578 init_put_bits(&pb [i], bit_buf [i], MAX_MB_BYTES);
2758 18578 init_put_bits(&pb2 [i], bit_buf2 [i], MAX_MB_BYTES);
2759 18578 init_put_bits(&tex_pb[i], bit_buf_tex[i], MAX_MB_BYTES);
2760 }
2761
2762 9289 s->last_bits= put_bits_count(&s->pb);
2763 9289 s->mv_bits=0;
2764 9289 s->misc_bits=0;
2765 9289 s->i_tex_bits=0;
2766 9289 s->p_tex_bits=0;
2767 9289 s->i_count=0;
2768 9289 s->skip_count=0;
2769
2770
2/2
✓ Branch 0 taken 27867 times.
✓ Branch 1 taken 9289 times.
 37156 for(i=0; i<3; i++){
2771 /* init last dc values */
2772 /* note: quant matrix value (8) is implied here */
2773 27867 s->last_dc[i] = 128 << s->intra_dc_precision;
2774
2775 27867 s->current_picture.encoding_error[i] = 0;
2776 }
2777
2/2
✓ Branch 0 taken 200 times.
✓ Branch 1 taken 9089 times.
 9289 if(s->codec_id==AV_CODEC_ID_AMV){
2778 200 s->last_dc[0] = 128*8/13;
2779 200 s->last_dc[1] = 128*8/14;
2780 200 s->last_dc[2] = 128*8/14;
2781 }
2782 9289 s->mb_skip_run = 0;
2783 9289 memset(s->last_mv, 0, sizeof(s->last_mv));
2784
2785 9289 s->last_mv_dir = 0;
2786
2787
3/3
✓ Branch 0 taken 840 times.
✓ Branch 1 taken 2790 times.
✓ Branch 2 taken 5659 times.
 9289 switch(s->codec_id){
2788 840 case AV_CODEC_ID_H263:
2789 case AV_CODEC_ID_H263P:
2790 case AV_CODEC_ID_FLV1:
2791 if (CONFIG_H263_ENCODER)
2792
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 840 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 840 s->gob_index = H263_GOB_HEIGHT(s->height);
2793 840 break;
2794 2790 case AV_CODEC_ID_MPEG4:
2795
2/2
✓ Branch 0 taken 544 times.
✓ Branch 1 taken 2246 times.
 2790 if(CONFIG_MPEG4_ENCODER && s->partitioned_frame)
2796 544 ff_mpeg4_init_partitions(s);
2797 2790 break;
2798 }
2799
2800 9289 s->resync_mb_x=0;
2801 9289 s->resync_mb_y=0;
2802 9289 s->first_slice_line = 1;
2803 9289 s->ptr_lastgob = s->pb.buf;
2804
2/2
✓ Branch 0 taken 138892 times.
✓ Branch 1 taken 9289 times.
 148181 for (mb_y_order = s->start_mb_y; mb_y_order < s->end_mb_y; mb_y_order++) {
2805
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 138892 times.
 138892 if (CONFIG_SPEEDHQ_ENCODER && s->codec_id == AV_CODEC_ID_SPEEDHQ) {
2806 int first_in_slice;
2807 ✗ mb_y = ff_speedhq_mb_y_order_to_mb(mb_y_order, s->mb_height, &first_in_slice);
2808 ✗ if (first_in_slice && mb_y_order != s->start_mb_y)
2809 ✗ ff_speedhq_end_slice(s);
2810 ✗ s->last_dc[0] = s->last_dc[1] = s->last_dc[2] = 1024 << s->intra_dc_precision;
2811 } else {
2812 138892 mb_y = mb_y_order;
2813 }
2814 138892 s->mb_x=0;
2815 138892 s->mb_y= mb_y;
2816
2817 138892 ff_set_qscale(s, s->qscale);
2818 138892 ff_init_block_index(s);
2819
2820
2/2
✓ Branch 0 taken 3227737 times.
✓ Branch 1 taken 138892 times.
 3366629 for(mb_x=0; mb_x < s->mb_width; mb_x++) {
2821 3227737 int xy= mb_y*s->mb_stride + mb_x; // removed const, H261 needs to adjust this
2822 3227737 int mb_type= s->mb_type[xy];
2823 // int d;
2824 3227737 int dmin= INT_MAX;
2825 int dir;
2826 3227737 int size_increase = s->avctx->internal->byte_buffer_size/4
2827 3227737 + s->mb_width*MAX_MB_BYTES;
2828
2829 3227737 ff_mpv_reallocate_putbitbuffer(s, MAX_MB_BYTES, size_increase);
2830
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 3227737 times.
 3227737 if (put_bytes_left(&s->pb, 0) < MAX_MB_BYTES){
2831 ✗ av_log(s->avctx, AV_LOG_ERROR, "encoded frame too large\n");
2832 ✗ return -1;
2833 }
2834
2/2
✓ Branch 0 taken 179550 times.
✓ Branch 1 taken 3048187 times.
 3227737 if(s->data_partitioning){
2835
2/4
✓ Branch 1 taken 179550 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 179550 times.
 359100 if (put_bytes_left(&s->pb2, 0) < MAX_MB_BYTES ||
2836 179550 put_bytes_left(&s->tex_pb, 0) < MAX_MB_BYTES) {
2837 ✗ av_log(s->avctx, AV_LOG_ERROR, "encoded partitioned frame too large\n");
2838 ✗ return -1;
2839 }
2840 }
2841
2842 3227737 s->mb_x = mb_x;
2843 3227737 s->mb_y = mb_y; // moved into loop, can get changed by H.261
2844 3227737 ff_update_block_index(s);
2845
2846
2/2
✓ Branch 0 taken 118800 times.
✓ Branch 1 taken 3108937 times.
 3227737 if(CONFIG_H261_ENCODER && s->codec_id == AV_CODEC_ID_H261){
2847 118800 ff_h261_reorder_mb_index(s);
2848 118800 xy= s->mb_y*s->mb_stride + s->mb_x;
2849 118800 mb_type= s->mb_type[xy];
2850 }
2851
2852 /* write gob / video packet header */
2853
2/2
✓ Branch 0 taken 1291380 times.
✓ Branch 1 taken 1936357 times.
 3227737 if(s->rtp_mode){
2854 int current_packet_size, is_gob_start;
2855
2856 1291380 current_packet_size = put_bytes_count(&s->pb, 1)
2857 1291380 - (s->ptr_lastgob - s->pb.buf);
2858
2859 3138060 is_gob_start = s->rtp_payload_size &&
2860
4/4
✓ Branch 0 taken 555300 times.
✓ Branch 1 taken 736080 times.
✓ Branch 2 taken 310021 times.
✓ Branch 3 taken 245279 times.
 1601401 current_packet_size >= s->rtp_payload_size &&
2861
2/2
✓ Branch 0 taken 309871 times.
✓ Branch 1 taken 150 times.
 310021 mb_y + mb_x > 0;
2862
2863
6/6
✓ Branch 0 taken 68595 times.
✓ Branch 1 taken 1222785 times.
✓ Branch 2 taken 10350 times.
✓ Branch 3 taken 58245 times.
✓ Branch 4 taken 600 times.
✓ Branch 5 taken 9750 times.
 1291380 if(s->start_mb_y == mb_y && mb_y > 0 && mb_x==0) is_gob_start=1;
2864
2865
4/5
✓ Branch 0 taken 59400 times.
✓ Branch 1 taken 992580 times.
✓ Branch 2 taken 59850 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 179550 times.
 1291380 switch(s->codec_id){
2866 59400 case AV_CODEC_ID_H263:
2867 case AV_CODEC_ID_H263P:
2868
1/2
✓ Branch 0 taken 59400 times.
✗ Branch 1 not taken.
 59400 if(!s->h263_slice_structured)
2869
3/4
✓ Branch 0 taken 2700 times.
✓ Branch 1 taken 56700 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2700 times.
 59400 if(s->mb_x || s->mb_y%s->gob_index) is_gob_start=0;
2870 59400 break;
2871 992580 case AV_CODEC_ID_MPEG2VIDEO:
2872
4/4
✓ Branch 0 taken 35165 times.
✓ Branch 1 taken 957415 times.
✓ Branch 2 taken 33275 times.
✓ Branch 3 taken 1890 times.
 992580 if(s->mb_x==0 && s->mb_y!=0) is_gob_start=1;
2873 case AV_CODEC_ID_MPEG1VIDEO:
2874
2/2
✓ Branch 0 taken 101523 times.
✓ Branch 1 taken 950907 times.
 1052430 if(s->mb_skip_run) is_gob_start=0;
2875 1052430 break;
2876 ✗ case AV_CODEC_ID_MJPEG:
2877 ✗ if(s->mb_x==0 && s->mb_y!=0) is_gob_start=1;
2878 ✗ break;
2879 }
2880
2881
2/2
✓ Branch 0 taken 296735 times.
✓ Branch 1 taken 994645 times.
 1291380 if(is_gob_start){
2882
4/4
✓ Branch 0 taken 7731 times.
✓ Branch 1 taken 289004 times.
✓ Branch 2 taken 7131 times.
✓ Branch 3 taken 600 times.
 296735 if(s->start_mb_y != mb_y || mb_x!=0){
2883 296135 write_slice_end(s);
2884
2885
4/4
✓ Branch 0 taken 3123 times.
✓ Branch 1 taken 293012 times.
✓ Branch 2 taken 1216 times.
✓ Branch 3 taken 1907 times.
 296135 if(CONFIG_MPEG4_ENCODER && s->codec_id==AV_CODEC_ID_MPEG4 && s->partitioned_frame){
2886 1216 ff_mpeg4_init_partitions(s);
2887 }
2888 }
2889
2890 av_assert2((put_bits_count(&s->pb)&7) == 0);
2891 296735 current_packet_size= put_bits_ptr(&s->pb) - s->ptr_lastgob;
2892
2893
4/4
✓ Branch 0 taken 437 times.
✓ Branch 1 taken 296298 times.
✓ Branch 2 taken 287 times.
✓ Branch 3 taken 150 times.
 296735 if (s->error_rate && s->resync_mb_x + s->resync_mb_y > 0) {
2894 287 int r = put_bytes_count(&s->pb, 0) + s->picture_number + 16 + s->mb_x + s->mb_y;
2895 287 int d = 100 / s->error_rate;
2896
2/2
✓ Branch 0 taken 32 times.
✓ Branch 1 taken 255 times.
 287 if(r % d == 0){
2897 32 current_packet_size=0;
2898 32 s->pb.buf_ptr= s->ptr_lastgob;
2899 av_assert1(put_bits_ptr(&s->pb) == s->ptr_lastgob);
2900 }
2901 }
2902
2903
3/4
✓ Branch 0 taken 3323 times.
✓ Branch 1 taken 290868 times.
✓ Branch 2 taken 2544 times.
✗ Branch 3 not taken.
 296735 switch(s->codec_id){
2904 3323 case AV_CODEC_ID_MPEG4:
2905 if (CONFIG_MPEG4_ENCODER) {
2906 3323 ff_mpeg4_encode_video_packet_header(s);
2907 3323 ff_mpeg4_clean_buffers(s);
2908 }
2909 3323 break;
2910 290868 case AV_CODEC_ID_MPEG1VIDEO:
2911 case AV_CODEC_ID_MPEG2VIDEO:
2912 if (CONFIG_MPEG1VIDEO_ENCODER || CONFIG_MPEG2VIDEO_ENCODER) {
2913 290868 ff_mpeg1_encode_slice_header(s);
2914 290868 ff_mpeg1_clean_buffers(s);
2915 }
2916 290868 break;
2917 2544 case AV_CODEC_ID_H263:
2918 case AV_CODEC_ID_H263P:
2919 if (CONFIG_H263_ENCODER) {
2920 2544 update_mb_info(s, 1);
2921 2544 ff_h263_encode_gob_header(s, mb_y);
2922 }
2923 2544 break;
2924 }
2925
2926
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 296735 times.
 296735 if (s->avctx->flags & AV_CODEC_FLAG_PASS1) {
2927 ✗ int bits= put_bits_count(&s->pb);
2928 ✗ s->misc_bits+= bits - s->last_bits;
2929 ✗ s->last_bits= bits;
2930 }
2931
2932 296735 s->ptr_lastgob += current_packet_size;
2933 296735 s->first_slice_line=1;
2934 296735 s->resync_mb_x=mb_x;
2935 296735 s->resync_mb_y=mb_y;
2936 }
2937 }
2938
2939
2/2
✓ Branch 0 taken 396635 times.
✓ Branch 1 taken 2831102 times.
 3227737 if( (s->resync_mb_x == s->mb_x)
2940
2/2
✓ Branch 0 taken 8504 times.
✓ Branch 1 taken 388131 times.
 396635 && s->resync_mb_y+1 == s->mb_y){
2941 8504 s->first_slice_line=0;
2942 }
2943
2944 3227737 s->mb_skipped=0;
2945 3227737 s->dquant=0; //only for QP_RD
2946
2947 3227737 update_mb_info(s, 0);
2948
2949
4/4
✓ Branch 0 taken 2714576 times.
✓ Branch 1 taken 513161 times.
✓ Branch 2 taken 18731 times.
✓ Branch 3 taken 2695845 times.
 3759629 if (mb_type & (mb_type-1) || (s->mpv_flags & FF_MPV_FLAG_QP_RD)) { // more than 1 MB type possible or FF_MPV_FLAG_QP_RD
2950 531892 int next_block=0;
2951 int pb_bits_count, pb2_bits_count, tex_pb_bits_count;
2952
2953 531892 copy_context_before_encode(&backup_s, s);
2954 531892 backup_s.pb= s->pb;
2955 531892 best_s.data_partitioning= s->data_partitioning;
2956 531892 best_s.partitioned_frame= s->partitioned_frame;
2957
2/2
✓ Branch 0 taken 141085 times.
✓ Branch 1 taken 390807 times.
 531892 if(s->data_partitioning){
2958 141085 backup_s.pb2= s->pb2;
2959 141085 backup_s.tex_pb= s->tex_pb;
2960 }
2961
2962
2/2
✓ Branch 0 taken 282073 times.
✓ Branch 1 taken 249819 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_INTER){
2963 282073 s->mv_dir = MV_DIR_FORWARD;
2964 282073 s->mv_type = MV_TYPE_16X16;
2965 282073 s->mb_intra= 0;
2966 282073 s->mv[0][0][0] = s->p_mv_table[xy][0];
2967 282073 s->mv[0][0][1] = s->p_mv_table[xy][1];
2968 282073 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
2969 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
2970 }
2971
2/2
✓ Branch 0 taken 13502 times.
✓ Branch 1 taken 518390 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_INTER_I){
2972 13502 s->mv_dir = MV_DIR_FORWARD;
2973 13502 s->mv_type = MV_TYPE_FIELD;
2974 13502 s->mb_intra= 0;
2975
2/2
✓ Branch 0 taken 27004 times.
✓ Branch 1 taken 13502 times.
 40506 for(i=0; i<2; i++){
2976 27004 j= s->field_select[0][i] = s->p_field_select_table[i][xy];
2977 27004 s->mv[0][i][0] = s->p_field_mv_table[i][j][xy][0];
2978 27004 s->mv[0][i][1] = s->p_field_mv_table[i][j][xy][1];
2979 }
2980 13502 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
2981 &dmin, &next_block, 0, 0);
2982 }
2983
2/2
✓ Branch 0 taken 60612 times.
✓ Branch 1 taken 471280 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_SKIPPED){
2984 60612 s->mv_dir = MV_DIR_FORWARD;
2985 60612 s->mv_type = MV_TYPE_16X16;
2986 60612 s->mb_intra= 0;
2987 60612 s->mv[0][0][0] = 0;
2988 60612 s->mv[0][0][1] = 0;
2989 60612 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
2990 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
2991 }
2992
2/2
✓ Branch 0 taken 209210 times.
✓ Branch 1 taken 322682 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_INTER4V){
2993 209210 s->mv_dir = MV_DIR_FORWARD;
2994 209210 s->mv_type = MV_TYPE_8X8;
2995 209210 s->mb_intra= 0;
2996
2/2
✓ Branch 0 taken 836840 times.
✓ Branch 1 taken 209210 times.
 1046050 for(i=0; i<4; i++){
2997 836840 s->mv[0][i][0] = s->current_picture.motion_val[0][s->block_index[i]][0];
2998 836840 s->mv[0][i][1] = s->current_picture.motion_val[0][s->block_index[i]][1];
2999 }
3000 209210 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3001 &dmin, &next_block, 0, 0);
3002 }
3003
2/2
✓ Branch 0 taken 229850 times.
✓ Branch 1 taken 302042 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_FORWARD){
3004 229850 s->mv_dir = MV_DIR_FORWARD;
3005 229850 s->mv_type = MV_TYPE_16X16;
3006 229850 s->mb_intra= 0;
3007 229850 s->mv[0][0][0] = s->b_forw_mv_table[xy][0];
3008 229850 s->mv[0][0][1] = s->b_forw_mv_table[xy][1];
3009 229850 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3010 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
3011 }
3012
2/2
✓ Branch 0 taken 229850 times.
✓ Branch 1 taken 302042 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_BACKWARD){
3013 229850 s->mv_dir = MV_DIR_BACKWARD;
3014 229850 s->mv_type = MV_TYPE_16X16;
3015 229850 s->mb_intra= 0;
3016 229850 s->mv[1][0][0] = s->b_back_mv_table[xy][0];
3017 229850 s->mv[1][0][1] = s->b_back_mv_table[xy][1];
3018 229850 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3019 &dmin, &next_block, s->mv[1][0][0], s->mv[1][0][1]);
3020 }
3021
2/2
✓ Branch 0 taken 229850 times.
✓ Branch 1 taken 302042 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_BIDIR){
3022 229850 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
3023 229850 s->mv_type = MV_TYPE_16X16;
3024 229850 s->mb_intra= 0;
3025 229850 s->mv[0][0][0] = s->b_bidir_forw_mv_table[xy][0];
3026 229850 s->mv[0][0][1] = s->b_bidir_forw_mv_table[xy][1];
3027 229850 s->mv[1][0][0] = s->b_bidir_back_mv_table[xy][0];
3028 229850 s->mv[1][0][1] = s->b_bidir_back_mv_table[xy][1];
3029 229850 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3030 &dmin, &next_block, 0, 0);
3031 }
3032
2/2
✓ Branch 0 taken 31700 times.
✓ Branch 1 taken 500192 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_FORWARD_I){
3033 31700 s->mv_dir = MV_DIR_FORWARD;
3034 31700 s->mv_type = MV_TYPE_FIELD;
3035 31700 s->mb_intra= 0;
3036
2/2
✓ Branch 0 taken 63400 times.
✓ Branch 1 taken 31700 times.
 95100 for(i=0; i<2; i++){
3037 63400 j= s->field_select[0][i] = s->b_field_select_table[0][i][xy];
3038 63400 s->mv[0][i][0] = s->b_field_mv_table[0][i][j][xy][0];
3039 63400 s->mv[0][i][1] = s->b_field_mv_table[0][i][j][xy][1];
3040 }
3041 31700 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3042 &dmin, &next_block, 0, 0);
3043 }
3044
2/2
✓ Branch 0 taken 31688 times.
✓ Branch 1 taken 500204 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_BACKWARD_I){
3045 31688 s->mv_dir = MV_DIR_BACKWARD;
3046 31688 s->mv_type = MV_TYPE_FIELD;
3047 31688 s->mb_intra= 0;
3048
2/2
✓ Branch 0 taken 63376 times.
✓ Branch 1 taken 31688 times.
 95064 for(i=0; i<2; i++){
3049 63376 j= s->field_select[1][i] = s->b_field_select_table[1][i][xy];
3050 63376 s->mv[1][i][0] = s->b_field_mv_table[1][i][j][xy][0];
3051 63376 s->mv[1][i][1] = s->b_field_mv_table[1][i][j][xy][1];
3052 }
3053 31688 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3054 &dmin, &next_block, 0, 0);
3055 }
3056
2/2
✓ Branch 0 taken 26999 times.
✓ Branch 1 taken 504893 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_BIDIR_I){
3057 26999 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
3058 26999 s->mv_type = MV_TYPE_FIELD;
3059 26999 s->mb_intra= 0;
3060
2/2
✓ Branch 0 taken 53998 times.
✓ Branch 1 taken 26999 times.
 80997 for(dir=0; dir<2; dir++){
3061
2/2
✓ Branch 0 taken 107996 times.
✓ Branch 1 taken 53998 times.
 161994 for(i=0; i<2; i++){
3062 107996 j= s->field_select[dir][i] = s->b_field_select_table[dir][i][xy];
3063 107996 s->mv[dir][i][0] = s->b_field_mv_table[dir][i][j][xy][0];
3064 107996 s->mv[dir][i][1] = s->b_field_mv_table[dir][i][j][xy][1];
3065 }
3066 }
3067 26999 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3068 &dmin, &next_block, 0, 0);
3069 }
3070
2/2
✓ Branch 0 taken 115405 times.
✓ Branch 1 taken 416487 times.
 531892 if(mb_type&CANDIDATE_MB_TYPE_INTRA){
3071 115405 s->mv_dir = 0;
3072 115405 s->mv_type = MV_TYPE_16X16;
3073 115405 s->mb_intra= 1;
3074 115405 s->mv[0][0][0] = 0;
3075 115405 s->mv[0][0][1] = 0;
3076 115405 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3077 &dmin, &next_block, 0, 0);
3078
3/4
✓ Branch 0 taken 38212 times.
✓ Branch 1 taken 77193 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 38212 times.
 115405 if(s->h263_pred || s->h263_aic){
3079
2/2
✓ Branch 0 taken 19644 times.
✓ Branch 1 taken 57549 times.
 77193 if(best_s.mb_intra)
3080 19644 s->mbintra_table[mb_x + mb_y*s->mb_stride]=1;
3081 else
3082 57549 ff_clean_intra_table_entries(s); //old mode?
3083 }
3084 }
3085
3086
4/4
✓ Branch 0 taken 179700 times.
✓ Branch 1 taken 352192 times.
✓ Branch 2 taken 179660 times.
✓ Branch 3 taken 40 times.
 531892 if ((s->mpv_flags & FF_MPV_FLAG_QP_RD) && dmin < INT_MAX) {
3087
2/2
✓ Branch 0 taken 161759 times.
✓ Branch 1 taken 17901 times.
 179660 if(best_s.mv_type==MV_TYPE_16X16){ //FIXME move 4mv after QPRD
3088 161759 const int last_qp= backup_s.qscale;
3089 int qpi, qp, dc[6];
3090 int16_t ac[6][16];
3091 161759 const int mvdir= (best_s.mv_dir&MV_DIR_BACKWARD) ? 1 : 0;
3092 static const int dquant_tab[4]={-1,1,-2,2};
3093
4/4
✓ Branch 0 taken 32407 times.
✓ Branch 1 taken 129352 times.
✓ Branch 2 taken 10117 times.
✓ Branch 3 taken 22290 times.
 161759 int storecoefs = s->mb_intra && s->dc_val[0];
3094
3095 av_assert2(backup_s.dquant == 0);
3096
3097 //FIXME intra
3098 161759 s->mv_dir= best_s.mv_dir;
3099 161759 s->mv_type = MV_TYPE_16X16;
3100 161759 s->mb_intra= best_s.mb_intra;
3101 161759 s->mv[0][0][0] = best_s.mv[0][0][0];
3102 161759 s->mv[0][0][1] = best_s.mv[0][0][1];
3103 161759 s->mv[1][0][0] = best_s.mv[1][0][0];
3104 161759 s->mv[1][0][1] = best_s.mv[1][0][1];
3105
3106
2/2
✓ Branch 0 taken 107866 times.
✓ Branch 1 taken 53893 times.
 161759 qpi = s->pict_type == AV_PICTURE_TYPE_B ? 2 : 0;
3107
2/2
✓ Branch 0 taken 431304 times.
✓ Branch 1 taken 161759 times.
 593063 for(; qpi<4; qpi++){
3108 431304 int dquant= dquant_tab[qpi];
3109 431304 qp= last_qp + dquant;
3110
4/4
✓ Branch 0 taken 395441 times.
✓ Branch 1 taken 35863 times.
✓ Branch 2 taken 5308 times.
✓ Branch 3 taken 390133 times.
 431304 if(qp < s->avctx->qmin || qp > s->avctx->qmax)
3111 41171 continue;
3112 390133 backup_s.dquant= dquant;
3113
2/2
✓ Branch 0 taken 37374 times.
✓ Branch 1 taken 352759 times.
 390133 if(storecoefs){
3114
2/2
✓ Branch 0 taken 224244 times.
✓ Branch 1 taken 37374 times.
 261618 for(i=0; i<6; i++){
3115 224244 dc[i]= s->dc_val[0][ s->block_index[i] ];
3116 224244 memcpy(ac[i], s->ac_val[0][s->block_index[i]], sizeof(int16_t)*16);
3117 }
3118 }
3119
3120 390133 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3121 &dmin, &next_block, s->mv[mvdir][0][0], s->mv[mvdir][0][1]);
3122
2/2
✓ Branch 0 taken 357862 times.
✓ Branch 1 taken 32271 times.
 390133 if(best_s.qscale != qp){
3123
2/2
✓ Branch 0 taken 33686 times.
✓ Branch 1 taken 324176 times.
 357862 if(storecoefs){
3124
2/2
✓ Branch 0 taken 202116 times.
✓ Branch 1 taken 33686 times.
 235802 for(i=0; i<6; i++){
3125 202116 s->dc_val[0][ s->block_index[i] ]= dc[i];
3126 202116 memcpy(s->ac_val[0][s->block_index[i]], ac[i], sizeof(int16_t)*16);
3127 }
3128 }
3129 }
3130 }
3131 }
3132 }
3133
2/2
✓ Branch 0 taken 153046 times.
✓ Branch 1 taken 378846 times.
 531892 if(CONFIG_MPEG4_ENCODER && mb_type&CANDIDATE_MB_TYPE_DIRECT){
3134 153046 int mx= s->b_direct_mv_table[xy][0];
3135 153046 int my= s->b_direct_mv_table[xy][1];
3136
3137 153046 backup_s.dquant = 0;
3138 153046 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
3139 153046 s->mb_intra= 0;
3140 153046 ff_mpeg4_set_direct_mv(s, mx, my);
3141 153046 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3142 &dmin, &next_block, mx, my);
3143 }
3144
2/2
✓ Branch 0 taken 47505 times.
✓ Branch 1 taken 484387 times.
 531892 if(CONFIG_MPEG4_ENCODER && mb_type&CANDIDATE_MB_TYPE_DIRECT0){
3145 47505 backup_s.dquant = 0;
3146 47505 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
3147 47505 s->mb_intra= 0;
3148 47505 ff_mpeg4_set_direct_mv(s, 0, 0);
3149 47505 encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3150 &dmin, &next_block, 0, 0);
3151 }
3152
3/4
✓ Branch 0 taken 493277 times.
✓ Branch 1 taken 38615 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 493277 times.
 531892 if (!best_s.mb_intra && s->mpv_flags & FF_MPV_FLAG_SKIP_RD) {
3153 ✗ int coded=0;
3154 ✗ for(i=0; i<6; i++)
3155 ✗ coded |= s->block_last_index[i];
3156 ✗ if(coded){
3157 int mx,my;
3158 ✗ memcpy(s->mv, best_s.mv, sizeof(s->mv));
3159 ✗ if(CONFIG_MPEG4_ENCODER && best_s.mv_dir & MV_DIRECT){
3160 ✗ mx=my=0; //FIXME find the one we actually used
3161 ✗ ff_mpeg4_set_direct_mv(s, mx, my);
3162 ✗ }else if(best_s.mv_dir&MV_DIR_BACKWARD){
3163 ✗ mx= s->mv[1][0][0];
3164 ✗ my= s->mv[1][0][1];
3165 }else{
3166 ✗ mx= s->mv[0][0][0];
3167 ✗ my= s->mv[0][0][1];
3168 }
3169
3170 ✗ s->mv_dir= best_s.mv_dir;
3171 ✗ s->mv_type = best_s.mv_type;
3172 ✗ s->mb_intra= 0;
3173 /* s->mv[0][0][0] = best_s.mv[0][0][0];
3174 s->mv[0][0][1] = best_s.mv[0][0][1];
3175 s->mv[1][0][0] = best_s.mv[1][0][0];
3176 s->mv[1][0][1] = best_s.mv[1][0][1];*/
3177 ✗ backup_s.dquant= 0;
3178 ✗ s->skipdct=1;
3179 ✗ encode_mb_hq(s, &backup_s, &best_s, pb, pb2, tex_pb,
3180 &dmin, &next_block, mx, my);
3181 ✗ s->skipdct=0;
3182 }
3183 }
3184
3185 531892 s->current_picture.qscale_table[xy] = best_s.qscale;
3186
3187 531892 copy_context_after_encode(s, &best_s);
3188
3189 531892 pb_bits_count= put_bits_count(&s->pb);
3190 531892 flush_put_bits(&s->pb);
3191 531892 ff_copy_bits(&backup_s.pb, bit_buf[next_block^1], pb_bits_count);
3192 531892 s->pb= backup_s.pb;
3193
3194
2/2
✓ Branch 0 taken 141085 times.
✓ Branch 1 taken 390807 times.
 531892 if(s->data_partitioning){
3195 141085 pb2_bits_count= put_bits_count(&s->pb2);
3196 141085 flush_put_bits(&s->pb2);
3197 141085 ff_copy_bits(&backup_s.pb2, bit_buf2[next_block^1], pb2_bits_count);
3198 141085 s->pb2= backup_s.pb2;
3199
3200 141085 tex_pb_bits_count= put_bits_count(&s->tex_pb);
3201 141085 flush_put_bits(&s->tex_pb);
3202 141085 ff_copy_bits(&backup_s.tex_pb, bit_buf_tex[next_block^1], tex_pb_bits_count);
3203 141085 s->tex_pb= backup_s.tex_pb;
3204 }
3205 531892 s->last_bits= put_bits_count(&s->pb);
3206
3207 531892 if (CONFIG_H263_ENCODER &&
3208
4/4
✓ Branch 0 taken 397164 times.
✓ Branch 1 taken 134728 times.
✓ Branch 2 taken 243978 times.
✓ Branch 3 taken 153186 times.
 531892 s->out_format == FMT_H263 && s->pict_type!=AV_PICTURE_TYPE_B)
3209 243978 ff_h263_update_motion_val(s);
3210
3211
2/2
✓ Branch 0 taken 225272 times.
✓ Branch 1 taken 306620 times.
 531892 if(next_block==0){ //FIXME 16 vs linesize16
3212 225272 s->hdsp.put_pixels_tab[0][0](s->dest[0], s->sc.rd_scratchpad , s->linesize ,16);
3213 225272 s->hdsp.put_pixels_tab[1][0](s->dest[1], s->sc.rd_scratchpad + 16*s->linesize , s->uvlinesize, 8);
3214 225272 s->hdsp.put_pixels_tab[1][0](s->dest[2], s->sc.rd_scratchpad + 16*s->linesize + 8, s->uvlinesize, 8);
3215 }
3216
3217
2/2
✓ Branch 0 taken 141006 times.
✓ Branch 1 taken 390886 times.
 531892 if(s->avctx->mb_decision == FF_MB_DECISION_BITS)
3218 141006 ff_mpv_reconstruct_mb(s, s->block);
3219 } else {
3220 2695845 int motion_x = 0, motion_y = 0;
3221 2695845 s->mv_type=MV_TYPE_16X16;
3222 // only one MB-Type possible
3223
3224
10/13
✓ Branch 0 taken 1012546 times.
✓ Branch 1 taken 1495426 times.
✓ Branch 2 taken 9451 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 15807 times.
✓ Branch 5 taken 78 times.
✓ Branch 6 taken 101729 times.
✓ Branch 7 taken 33452 times.
✓ Branch 8 taken 22694 times.
✓ Branch 9 taken 2174 times.
✓ Branch 10 taken 2488 times.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
 2695845 switch(mb_type){
3225 1012546 case CANDIDATE_MB_TYPE_INTRA:
3226 1012546 s->mv_dir = 0;
3227 1012546 s->mb_intra= 1;
3228 1012546 motion_x= s->mv[0][0][0] = 0;
3229 1012546 motion_y= s->mv[0][0][1] = 0;
3230 1012546 break;
3231 1495426 case CANDIDATE_MB_TYPE_INTER:
3232 1495426 s->mv_dir = MV_DIR_FORWARD;
3233 1495426 s->mb_intra= 0;
3234 1495426 motion_x= s->mv[0][0][0] = s->p_mv_table[xy][0];
3235 1495426 motion_y= s->mv[0][0][1] = s->p_mv_table[xy][1];
3236 1495426 break;
3237 9451 case CANDIDATE_MB_TYPE_INTER_I:
3238 9451 s->mv_dir = MV_DIR_FORWARD;
3239 9451 s->mv_type = MV_TYPE_FIELD;
3240 9451 s->mb_intra= 0;
3241
2/2
✓ Branch 0 taken 18902 times.
✓ Branch 1 taken 9451 times.
 28353 for(i=0; i<2; i++){
3242 18902 j= s->field_select[0][i] = s->p_field_select_table[i][xy];
3243 18902 s->mv[0][i][0] = s->p_field_mv_table[i][j][xy][0];
3244 18902 s->mv[0][i][1] = s->p_field_mv_table[i][j][xy][1];
3245 }
3246 9451 break;
3247 ✗ case CANDIDATE_MB_TYPE_INTER4V:
3248 ✗ s->mv_dir = MV_DIR_FORWARD;
3249 ✗ s->mv_type = MV_TYPE_8X8;
3250 ✗ s->mb_intra= 0;
3251 ✗ for(i=0; i<4; i++){
3252 ✗ s->mv[0][i][0] = s->current_picture.motion_val[0][s->block_index[i]][0];
3253 ✗ s->mv[0][i][1] = s->current_picture.motion_val[0][s->block_index[i]][1];
3254 }
3255 ✗ break;
3256 15807 case CANDIDATE_MB_TYPE_DIRECT:
3257 if (CONFIG_MPEG4_ENCODER) {
3258 15807 s->mv_dir = MV_DIR_FORWARD|MV_DIR_BACKWARD|MV_DIRECT;
3259 15807 s->mb_intra= 0;
3260 15807 motion_x=s->b_direct_mv_table[xy][0];
3261 15807 motion_y=s->b_direct_mv_table[xy][1];
3262 15807 ff_mpeg4_set_direct_mv(s, motion_x, motion_y);
3263 }
3264 15807 break;
3265 78 case CANDIDATE_MB_TYPE_DIRECT0:
3266 if (CONFIG_MPEG4_ENCODER) {
3267 78 s->mv_dir = MV_DIR_FORWARD|MV_DIR_BACKWARD|MV_DIRECT;
3268 78 s->mb_intra= 0;
3269 78 ff_mpeg4_set_direct_mv(s, 0, 0);
3270 }
3271 78 break;
3272 101729 case CANDIDATE_MB_TYPE_BIDIR:
3273 101729 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
3274 101729 s->mb_intra= 0;
3275 101729 s->mv[0][0][0] = s->b_bidir_forw_mv_table[xy][0];
3276 101729 s->mv[0][0][1] = s->b_bidir_forw_mv_table[xy][1];
3277 101729 s->mv[1][0][0] = s->b_bidir_back_mv_table[xy][0];
3278 101729 s->mv[1][0][1] = s->b_bidir_back_mv_table[xy][1];
3279 101729 break;
3280 33452 case CANDIDATE_MB_TYPE_BACKWARD:
3281 33452 s->mv_dir = MV_DIR_BACKWARD;
3282 33452 s->mb_intra= 0;
3283 33452 motion_x= s->mv[1][0][0] = s->b_back_mv_table[xy][0];
3284 33452 motion_y= s->mv[1][0][1] = s->b_back_mv_table[xy][1];
3285 33452 break;
3286 22694 case CANDIDATE_MB_TYPE_FORWARD:
3287 22694 s->mv_dir = MV_DIR_FORWARD;
3288 22694 s->mb_intra= 0;
3289 22694 motion_x= s->mv[0][0][0] = s->b_forw_mv_table[xy][0];
3290 22694 motion_y= s->mv[0][0][1] = s->b_forw_mv_table[xy][1];
3291 22694 break;
3292 2174 case CANDIDATE_MB_TYPE_FORWARD_I:
3293 2174 s->mv_dir = MV_DIR_FORWARD;
3294 2174 s->mv_type = MV_TYPE_FIELD;
3295 2174 s->mb_intra= 0;
3296
2/2
✓ Branch 0 taken 4348 times.
✓ Branch 1 taken 2174 times.
 6522 for(i=0; i<2; i++){
3297 4348 j= s->field_select[0][i] = s->b_field_select_table[0][i][xy];
3298 4348 s->mv[0][i][0] = s->b_field_mv_table[0][i][j][xy][0];
3299 4348 s->mv[0][i][1] = s->b_field_mv_table[0][i][j][xy][1];
3300 }
3301 2174 break;
3302 2488 case CANDIDATE_MB_TYPE_BACKWARD_I:
3303 2488 s->mv_dir = MV_DIR_BACKWARD;
3304 2488 s->mv_type = MV_TYPE_FIELD;
3305 2488 s->mb_intra= 0;
3306
2/2
✓ Branch 0 taken 4976 times.
✓ Branch 1 taken 2488 times.
 7464 for(i=0; i<2; i++){
3307 4976 j= s->field_select[1][i] = s->b_field_select_table[1][i][xy];
3308 4976 s->mv[1][i][0] = s->b_field_mv_table[1][i][j][xy][0];
3309 4976 s->mv[1][i][1] = s->b_field_mv_table[1][i][j][xy][1];
3310 }
3311 2488 break;
3312 ✗ case CANDIDATE_MB_TYPE_BIDIR_I:
3313 ✗ s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
3314