annotate mpegvideo_enc.c @ 12197:fbf4d5b1b664 libavcodec

Remove FF_MM_SSE2/3 flags for CPUs where this is generally not faster than regular MMX code. Examples of this are the Core1 CPU. Instead, set a new flag, FF_MM_SSE2/3SLOW, which can be checked for particular SSE2/3 functions that have been checked specifically on such CPUs and are actually faster than their MMX counterparts. In addition, use this flag to enable particular VP8 and LPC SSE2 functions that are faster than their MMX counterparts. Based on a patch by Loren Merritt <lorenm AT u washington edu>.
author rbultje
date Mon, 19 Jul 2010 22:38:23 +0000
parents b6cf19580e47
children 04e68eb1aab7
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1 /*
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2 * The simplest mpeg encoder (well, it was the simplest!)
8629
04423b2f6e0b cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents: 8611
diff changeset
3 * Copyright (c) 2000,2001 Fabrice Bellard
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
4 * Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at>
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
5 *
5214
470601203f44 Group all copyright and author notices together.
diego
parents: 5211
diff changeset
6 * 4MV & hq & B-frame encoding stuff by Michael Niedermayer <michaelni@gmx.at>
470601203f44 Group all copyright and author notices together.
diego
parents: 5211
diff changeset
7 *
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
8 * This file is part of FFmpeg.
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
9 *
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
10 * FFmpeg is free software; you can redistribute it and/or
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
11 * modify it under the terms of the GNU Lesser General Public
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
12 * License as published by the Free Software Foundation; either
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
13 * version 2.1 of the License, or (at your option) any later version.
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
14 *
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
15 * FFmpeg is distributed in the hope that it will be useful,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
18 * Lesser General Public License for more details.
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
19 *
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
20 * You should have received a copy of the GNU Lesser General Public
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
21 * License along with FFmpeg; if not, write to the Free Software
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
23 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
24
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
25 /**
11644
7dd2a45249a9 Remove explicit filename from Doxygen @file commands.
diego
parents: 11560
diff changeset
26 * @file
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
27 * The simplest mpeg encoder (well, it was the simplest!).
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
28 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
29
11417
421c9441c11e Move ff_sqrt() to libavutil/intmath.h
mru
parents: 11195
diff changeset
30 #include "libavutil/intmath.h"
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
31 #include "avcodec.h"
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
32 #include "dsputil.h"
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
33 #include "mpegvideo.h"
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
34 #include "mpegvideo_common.h"
10832
f20726a6d538 Add a function to match a 2 element vector of uint16_t and use it in h263 and svq1
michael
parents: 10803
diff changeset
35 #include "h263.h"
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
36 #include "mjpegenc.h"
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
37 #include "msmpeg4.h"
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
38 #include "faandct.h"
8218
03054192daac Move aanscales tables to their own file; fixes compilation without encoders.
diego
parents: 8217
diff changeset
39 #include "aandcttab.h"
10787
3f8e40fe25c2 Split out flv encoding.
michael
parents: 10749
diff changeset
40 #include "flv.h"
10803
4605bd2fdb7f Split the mpeg4 encoder and decoder off h263.c
michael
parents: 10789
diff changeset
41 #include "mpeg4video.h"
10843
f6afc7837f83 Add missing internal.h to files calling ff_match_2uint16().
astrange
parents: 10832
diff changeset
42 #include "internal.h"
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
43 #include <limits.h>
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
44
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
45 //#undef NDEBUG
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
46 //#include <assert.h>
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
47
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
48 static int encode_picture(MpegEncContext *s, int picture_number);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
49 static int dct_quantize_refine(MpegEncContext *s, DCTELEM *block, int16_t *weight, DCTELEM *orig, int n, int qscale);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
50 static int sse_mb(MpegEncContext *s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
51
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
52 /* enable all paranoid tests for rounding, overflows, etc... */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
53 //#define PARANOID
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
54
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
55 //#define DEBUG
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
56
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
57 static uint8_t default_mv_penalty[MAX_FCODE+1][MAX_MV*2+1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
58 static uint8_t default_fcode_tab[MAX_MV*2+1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
59
5789
a40edc95456d export convert_matrix and add ff_ prefix
bcoudurier
parents: 5420
diff changeset
60 void ff_convert_matrix(DSPContext *dsp, int (*qmat)[64], uint16_t (*qmat16)[2][64],
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
61 const uint16_t *quant_matrix, int bias, int qmin, int qmax, int intra)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
62 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
63 int qscale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
64 int shift=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
65
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
66 for(qscale=qmin; qscale<=qmax; qscale++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
67 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
68 if (dsp->fdct == ff_jpeg_fdct_islow
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
69 #ifdef FAAN_POSTSCALE
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
70 || dsp->fdct == ff_faandct
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
71 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
72 ) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
73 for(i=0;i<64;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
74 const int j= dsp->idct_permutation[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
75 /* 16 <= qscale * quant_matrix[i] <= 7905 */
8217
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
76 /* 19952 <= ff_aanscales[i] * qscale * quant_matrix[i] <= 249205026 */
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
77 /* (1 << 36) / 19952 >= (1 << 36) / (ff_aanscales[i] * qscale * quant_matrix[i]) >= (1 << 36) / 249205026 */
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
78 /* 3444240 >= (1 << 36) / (ff_aanscales[i] * qscale * quant_matrix[i]) >= 275 */
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
79
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
80 qmat[qscale][i] = (int)((UINT64_C(1) << QMAT_SHIFT) /
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
81 (qscale * quant_matrix[j]));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
82 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
83 } else if (dsp->fdct == fdct_ifast
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
84 #ifndef FAAN_POSTSCALE
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
85 || dsp->fdct == ff_faandct
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
86 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
87 ) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
88 for(i=0;i<64;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
89 const int j= dsp->idct_permutation[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
90 /* 16 <= qscale * quant_matrix[i] <= 7905 */
8217
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
91 /* 19952 <= ff_aanscales[i] * qscale * quant_matrix[i] <= 249205026 */
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
92 /* (1 << 36) / 19952 >= (1 << 36) / (ff_aanscales[i] * qscale * quant_matrix[i]) >= (1<<36)/249205026 */
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
93 /* 3444240 >= (1 << 36) / (ff_aanscales[i] * qscale * quant_matrix[i]) >= 275 */
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
94
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
95 qmat[qscale][i] = (int)((UINT64_C(1) << (QMAT_SHIFT + 14)) /
8217
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
96 (ff_aanscales[i] * qscale * quant_matrix[j]));
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
97 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
98 } else {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
99 for(i=0;i<64;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
100 const int j= dsp->idct_permutation[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
101 /* We can safely suppose that 16 <= quant_matrix[i] <= 255
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
102 So 16 <= qscale * quant_matrix[i] <= 7905
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
103 so (1<<19) / 16 >= (1<<19) / (qscale * quant_matrix[i]) >= (1<<19) / 7905
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
104 so 32768 >= (1<<19) / (qscale * quant_matrix[i]) >= 67
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
105 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
106 qmat[qscale][i] = (int)((UINT64_C(1) << QMAT_SHIFT) / (qscale * quant_matrix[j]));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
107 // qmat [qscale][i] = (1 << QMAT_SHIFT_MMX) / (qscale * quant_matrix[i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
108 qmat16[qscale][0][i] = (1 << QMAT_SHIFT_MMX) / (qscale * quant_matrix[j]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
109
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
110 if(qmat16[qscale][0][i]==0 || qmat16[qscale][0][i]==128*256) qmat16[qscale][0][i]=128*256-1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
111 qmat16[qscale][1][i]= ROUNDED_DIV(bias<<(16-QUANT_BIAS_SHIFT), qmat16[qscale][0][i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
112 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
113 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
114
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
115 for(i=intra; i<64; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
116 int64_t max= 8191;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
117 if (dsp->fdct == fdct_ifast
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
118 #ifndef FAAN_POSTSCALE
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
119 || dsp->fdct == ff_faandct
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
120 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
121 ) {
8217
f88ab1c3e333 cosmetics: Rename aanscales to ff_aanscales, it will soon be externally visible.
diego
parents: 8129
diff changeset
122 max = (8191LL*ff_aanscales[i]) >> 14;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
123 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
124 while(((max * qmat[qscale][i]) >> shift) > INT_MAX){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
125 shift++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
126 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
127 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
128 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
129 if(shift){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
130 av_log(NULL, AV_LOG_INFO, "Warning, QMAT_SHIFT is larger than %d, overflows possible\n", QMAT_SHIFT - shift);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
131 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
132 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
133
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
134 static inline void update_qscale(MpegEncContext *s){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
135 s->qscale= (s->lambda*139 + FF_LAMBDA_SCALE*64) >> (FF_LAMBDA_SHIFT + 7);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
136 s->qscale= av_clip(s->qscale, s->avctx->qmin, s->avctx->qmax);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
137
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
138 s->lambda2= (s->lambda*s->lambda + FF_LAMBDA_SCALE/2) >> FF_LAMBDA_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
139 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
140
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
141 void ff_write_quant_matrix(PutBitContext *pb, uint16_t *matrix){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
142 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
143
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
144 if(matrix){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
145 put_bits(pb, 1, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
146 for(i=0;i<64;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
147 put_bits(pb, 8, matrix[ ff_zigzag_direct[i] ]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
148 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
149 }else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
150 put_bits(pb, 1, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
151 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
152
10636
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
153 /**
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
154 * init s->current_picture.qscale_table from s->lambda_table
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
155 */
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
156 void ff_init_qscale_tab(MpegEncContext *s){
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
157 int8_t * const qscale_table= s->current_picture.qscale_table;
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
158 int i;
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
159
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
160 for(i=0; i<s->mb_num; i++){
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
161 unsigned int lam= s->lambda_table[ s->mb_index2xy[i] ];
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
162 int qp= (lam*139 + FF_LAMBDA_SCALE*64) >> (FF_LAMBDA_SHIFT + 7);
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
163 qscale_table[ s->mb_index2xy[i] ]= av_clip(qp, s->avctx->qmin, s->avctx->qmax);
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
164 }
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
165 }
703cfed31320 Move ff_init_qscale_tab() from h263.c to mpegvideo, the function is not h263 specific.
michael
parents: 10443
diff changeset
166
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
167 static void copy_picture_attributes(MpegEncContext *s, AVFrame *dst, AVFrame *src){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
168 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
169
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
170 dst->pict_type = src->pict_type;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
171 dst->quality = src->quality;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
172 dst->coded_picture_number = src->coded_picture_number;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
173 dst->display_picture_number = src->display_picture_number;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
174 // dst->reference = src->reference;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
175 dst->pts = src->pts;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
176 dst->interlaced_frame = src->interlaced_frame;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
177 dst->top_field_first = src->top_field_first;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
178
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
179 if(s->avctx->me_threshold){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
180 if(!src->motion_val[0])
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
181 av_log(s->avctx, AV_LOG_ERROR, "AVFrame.motion_val not set!\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
182 if(!src->mb_type)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
183 av_log(s->avctx, AV_LOG_ERROR, "AVFrame.mb_type not set!\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
184 if(!src->ref_index[0])
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
185 av_log(s->avctx, AV_LOG_ERROR, "AVFrame.ref_index not set!\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
186 if(src->motion_subsample_log2 != dst->motion_subsample_log2)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
187 av_log(s->avctx, AV_LOG_ERROR, "AVFrame.motion_subsample_log2 doesn't match! (%d!=%d)\n",
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
188 src->motion_subsample_log2, dst->motion_subsample_log2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
189
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
190 memcpy(dst->mb_type, src->mb_type, s->mb_stride * s->mb_height * sizeof(dst->mb_type[0]));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
191
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
192 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
193 int stride= ((16*s->mb_width )>>src->motion_subsample_log2) + 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
194 int height= ((16*s->mb_height)>>src->motion_subsample_log2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
195
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
196 if(src->motion_val[i] && src->motion_val[i] != dst->motion_val[i]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
197 memcpy(dst->motion_val[i], src->motion_val[i], 2*stride*height*sizeof(int16_t));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
198 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
199 if(src->ref_index[i] && src->ref_index[i] != dst->ref_index[i]){
11531
31033caa5344 Change ref_index structure so it matches how its organized in h264.
michael
parents: 11495
diff changeset
200 memcpy(dst->ref_index[i], src->ref_index[i], s->mb_stride*4*s->mb_height*sizeof(int8_t));
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
201 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
202 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
203 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
204 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
205
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
206 static void update_duplicate_context_after_me(MpegEncContext *dst, MpegEncContext *src){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
207 #define COPY(a) dst->a= src->a
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
208 COPY(pict_type);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
209 COPY(current_picture);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
210 COPY(f_code);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
211 COPY(b_code);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
212 COPY(qscale);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
213 COPY(lambda);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
214 COPY(lambda2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
215 COPY(picture_in_gop_number);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
216 COPY(gop_picture_number);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
217 COPY(frame_pred_frame_dct); //FIXME don't set in encode_header
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
218 COPY(progressive_frame); //FIXME don't set in encode_header
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
219 COPY(partitioned_frame); //FIXME don't set in encode_header
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
220 #undef COPY
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
221 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
222
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
223 /**
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
224 * sets the given MpegEncContext to defaults for encoding.
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
225 * the changed fields will not depend upon the prior state of the MpegEncContext.
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
226 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
227 static void MPV_encode_defaults(MpegEncContext *s){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
228 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
229 MPV_common_defaults(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
230
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
231 for(i=-16; i<16; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
232 default_fcode_tab[i + MAX_MV]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
233 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
234 s->me.mv_penalty= default_mv_penalty;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
235 s->fcode_tab= default_fcode_tab;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
236 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
237
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
238 /* init video encoder */
6517
48759bfbd073 Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents: 6497
diff changeset
239 av_cold int MPV_encode_init(AVCodecContext *avctx)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
240 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
241 MpegEncContext *s = avctx->priv_data;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
242 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
243 int chroma_h_shift, chroma_v_shift;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
244
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
245 MPV_encode_defaults(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
246
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
247 switch (avctx->codec_id) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
248 case CODEC_ID_MPEG2VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
249 if(avctx->pix_fmt != PIX_FMT_YUV420P && avctx->pix_fmt != PIX_FMT_YUV422P){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
250 av_log(avctx, AV_LOG_ERROR, "only YUV420 and YUV422 are supported\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
251 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
252 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
253 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
254 case CODEC_ID_LJPEG:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
255 case CODEC_ID_MJPEG:
8807
fa99e152760b Let rgb32 pass for ljpeg.
michael
parents: 8718
diff changeset
256 if(avctx->pix_fmt != PIX_FMT_YUVJ420P && avctx->pix_fmt != PIX_FMT_YUVJ422P && avctx->pix_fmt != PIX_FMT_RGB32 &&
12084
b6cf19580e47 Change all occurences of "inofficial" to "unofficial" in code, comments
cehoyos
parents: 11644
diff changeset
257 ((avctx->pix_fmt != PIX_FMT_YUV420P && avctx->pix_fmt != PIX_FMT_YUV422P) || avctx->strict_std_compliance>FF_COMPLIANCE_UNOFFICIAL)){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
258 av_log(avctx, AV_LOG_ERROR, "colorspace not supported in jpeg\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
259 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
260 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
261 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
262 default:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
263 if(avctx->pix_fmt != PIX_FMT_YUV420P){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
264 av_log(avctx, AV_LOG_ERROR, "only YUV420 is supported\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
265 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
266 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
267 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
268
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
269 switch (avctx->pix_fmt) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
270 case PIX_FMT_YUVJ422P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
271 case PIX_FMT_YUV422P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
272 s->chroma_format = CHROMA_422;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
273 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
274 case PIX_FMT_YUVJ420P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
275 case PIX_FMT_YUV420P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
276 default:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
277 s->chroma_format = CHROMA_420;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
278 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
279 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
280
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
281 s->bit_rate = avctx->bit_rate;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
282 s->width = avctx->width;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
283 s->height = avctx->height;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
284 if(avctx->gop_size > 600 && avctx->strict_std_compliance>FF_COMPLIANCE_EXPERIMENTAL){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
285 av_log(avctx, AV_LOG_ERROR, "Warning keyframe interval too large! reducing it ...\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
286 avctx->gop_size=600;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
287 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
288 s->gop_size = avctx->gop_size;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
289 s->avctx = avctx;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
290 s->flags= avctx->flags;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
291 s->flags2= avctx->flags2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
292 s->max_b_frames= avctx->max_b_frames;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
293 s->codec_id= avctx->codec->id;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
294 s->luma_elim_threshold = avctx->luma_elim_threshold;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
295 s->chroma_elim_threshold= avctx->chroma_elim_threshold;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
296 s->strict_std_compliance= avctx->strict_std_compliance;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
297 s->data_partitioning= avctx->flags & CODEC_FLAG_PART;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
298 s->quarter_sample= (avctx->flags & CODEC_FLAG_QPEL)!=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
299 s->mpeg_quant= avctx->mpeg_quant;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
300 s->rtp_mode= !!avctx->rtp_payload_size;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
301 s->intra_dc_precision= avctx->intra_dc_precision;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
302 s->user_specified_pts = AV_NOPTS_VALUE;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
303
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
304 if (s->gop_size <= 1) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
305 s->intra_only = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
306 s->gop_size = 12;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
307 } else {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
308 s->intra_only = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
309 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
310
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
311 s->me_method = avctx->me_method;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
312
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
313 /* Fixed QSCALE */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
314 s->fixed_qscale = !!(avctx->flags & CODEC_FLAG_QSCALE);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
315
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
316 s->adaptive_quant= ( s->avctx->lumi_masking
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
317 || s->avctx->dark_masking
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
318 || s->avctx->temporal_cplx_masking
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
319 || s->avctx->spatial_cplx_masking
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
320 || s->avctx->p_masking
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
321 || s->avctx->border_masking
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
322 || (s->flags&CODEC_FLAG_QP_RD))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
323 && !s->fixed_qscale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
324
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
325 s->obmc= !!(s->flags & CODEC_FLAG_OBMC);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
326 s->loop_filter= !!(s->flags & CODEC_FLAG_LOOP_FILTER);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
327 s->alternate_scan= !!(s->flags & CODEC_FLAG_ALT_SCAN);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
328 s->intra_vlc_format= !!(s->flags2 & CODEC_FLAG2_INTRA_VLC);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
329 s->q_scale_type= !!(s->flags2 & CODEC_FLAG2_NON_LINEAR_QUANT);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
330
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
331 if(avctx->rc_max_rate && !avctx->rc_buffer_size){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
332 av_log(avctx, AV_LOG_ERROR, "a vbv buffer size is needed, for encoding with a maximum bitrate\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
333 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
334 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
335
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
336 if(avctx->rc_min_rate && avctx->rc_max_rate != avctx->rc_min_rate){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
337 av_log(avctx, AV_LOG_INFO, "Warning min_rate > 0 but min_rate != max_rate isn't recommended!\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
338 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
339
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
340 if(avctx->rc_min_rate && avctx->rc_min_rate > avctx->bit_rate){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
341 av_log(avctx, AV_LOG_ERROR, "bitrate below min bitrate\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
342 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
343 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
344
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
345 if(avctx->rc_max_rate && avctx->rc_max_rate < avctx->bit_rate){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
346 av_log(avctx, AV_LOG_INFO, "bitrate above max bitrate\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
347 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
348 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
349
5420
e5b6e68deadc warn user if max=bitrate && max > min
michael
parents: 5278
diff changeset
350 if(avctx->rc_max_rate && avctx->rc_max_rate == avctx->bit_rate && avctx->rc_max_rate != avctx->rc_min_rate){
e5b6e68deadc warn user if max=bitrate && max > min
michael
parents: 5278
diff changeset
351 av_log(avctx, AV_LOG_INFO, "impossible bitrate constraints, this will fail\n");
e5b6e68deadc warn user if max=bitrate && max > min
michael
parents: 5278
diff changeset
352 }
e5b6e68deadc warn user if max=bitrate && max > min
michael
parents: 5278
diff changeset
353
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
354 if(avctx->rc_buffer_size && avctx->bit_rate*av_q2d(avctx->time_base) > avctx->rc_buffer_size){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
355 av_log(avctx, AV_LOG_ERROR, "VBV buffer too small for bitrate\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
356 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
357 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
358
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
359 if(avctx->bit_rate*av_q2d(avctx->time_base) > avctx->bit_rate_tolerance){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
360 av_log(avctx, AV_LOG_ERROR, "bitrate tolerance too small for bitrate\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
361 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
362 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
363
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
364 if( s->avctx->rc_max_rate && s->avctx->rc_min_rate == s->avctx->rc_max_rate
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
365 && (s->codec_id == CODEC_ID_MPEG1VIDEO || s->codec_id == CODEC_ID_MPEG2VIDEO)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
366 && 90000LL * (avctx->rc_buffer_size-1) > s->avctx->rc_max_rate*0xFFFFLL){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
367
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
368 av_log(avctx, AV_LOG_INFO, "Warning vbv_delay will be set to 0xFFFF (=VBR) as the specified vbv buffer is too large for the given bitrate!\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
369 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
370
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
371 if((s->flags & CODEC_FLAG_4MV) && s->codec_id != CODEC_ID_MPEG4
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
372 && s->codec_id != CODEC_ID_H263 && s->codec_id != CODEC_ID_H263P && s->codec_id != CODEC_ID_FLV1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
373 av_log(avctx, AV_LOG_ERROR, "4MV not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
374 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
375 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
376
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
377 if(s->obmc && s->avctx->mb_decision != FF_MB_DECISION_SIMPLE){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
378 av_log(avctx, AV_LOG_ERROR, "OBMC is only supported with simple mb decision\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
379 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
380 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
381
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
382 if(s->obmc && s->codec_id != CODEC_ID_H263 && s->codec_id != CODEC_ID_H263P){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
383 av_log(avctx, AV_LOG_ERROR, "OBMC is only supported with H263(+)\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
384 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
385 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
386
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
387 if(s->quarter_sample && s->codec_id != CODEC_ID_MPEG4){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
388 av_log(avctx, AV_LOG_ERROR, "qpel not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
389 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
390 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
391
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
392 if(s->data_partitioning && s->codec_id != CODEC_ID_MPEG4){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
393 av_log(avctx, AV_LOG_ERROR, "data partitioning not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
394 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
395 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
396
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
397 if(s->max_b_frames && s->codec_id != CODEC_ID_MPEG4 && s->codec_id != CODEC_ID_MPEG1VIDEO && s->codec_id != CODEC_ID_MPEG2VIDEO){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
398 av_log(avctx, AV_LOG_ERROR, "b frames not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
399 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
400 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
401
10683
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
402 if ((s->codec_id == CODEC_ID_MPEG4 || s->codec_id == CODEC_ID_H263 ||
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
403 s->codec_id == CODEC_ID_H263P) &&
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
404 (avctx->sample_aspect_ratio.num > 255 || avctx->sample_aspect_ratio.den > 255)) {
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
405 av_log(avctx, AV_LOG_ERROR, "Invalid pixel aspect ratio %i/%i, limit is 255/255\n",
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
406 avctx->sample_aspect_ratio.num, avctx->sample_aspect_ratio.den);
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
407 return -1;
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
408 }
e5fac8ac71f6 Check sample_aspect_ratio validity for mpeg4/h263/h263p encoding.
michael
parents: 10670
diff changeset
409
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
410 if((s->flags & (CODEC_FLAG_INTERLACED_DCT|CODEC_FLAG_INTERLACED_ME|CODEC_FLAG_ALT_SCAN))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
411 && s->codec_id != CODEC_ID_MPEG4 && s->codec_id != CODEC_ID_MPEG2VIDEO){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
412 av_log(avctx, AV_LOG_ERROR, "interlacing not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
413 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
414 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
415
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
416 if(s->mpeg_quant && s->codec_id != CODEC_ID_MPEG4){ //FIXME mpeg2 uses that too
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
417 av_log(avctx, AV_LOG_ERROR, "mpeg2 style quantization not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
418 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
419 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
420
7034
130784da5595 uniformize trellis quant option
bcoudurier
parents: 6788
diff changeset
421 if((s->flags & CODEC_FLAG_CBP_RD) && !avctx->trellis){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
422 av_log(avctx, AV_LOG_ERROR, "CBP RD needs trellis quant\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
423 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
424 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
425
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
426 if((s->flags & CODEC_FLAG_QP_RD) && s->avctx->mb_decision != FF_MB_DECISION_RD){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
427 av_log(avctx, AV_LOG_ERROR, "QP RD needs mbd=2\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
428 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
429 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
430
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
431 if(s->avctx->scenechange_threshold < 1000000000 && (s->flags & CODEC_FLAG_CLOSED_GOP)){
6486
3f96dc62c4b9 misc spelling fixes
diego
parents: 6481
diff changeset
432 av_log(avctx, AV_LOG_ERROR, "closed gop with scene change detection are not supported yet, set threshold to 1000000000\n");
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
433 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
434 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
435
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
436 if((s->flags2 & CODEC_FLAG2_INTRA_VLC) && s->codec_id != CODEC_ID_MPEG2VIDEO){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
437 av_log(avctx, AV_LOG_ERROR, "intra vlc table not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
438 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
439 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
440
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
441 if(s->flags & CODEC_FLAG_LOW_DELAY){
7502
d5c528384f13 Prevent low delay from being used with mpeg1.
michael
parents: 7040
diff changeset
442 if (s->codec_id != CODEC_ID_MPEG2VIDEO){
d5c528384f13 Prevent low delay from being used with mpeg1.
michael
parents: 7040
diff changeset
443 av_log(avctx, AV_LOG_ERROR, "low delay forcing is only available for mpeg2\n");
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
444 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
445 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
446 if (s->max_b_frames != 0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
447 av_log(avctx, AV_LOG_ERROR, "b frames cannot be used with low delay\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
448 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
449 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
450 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
451
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
452 if(s->q_scale_type == 1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
453 if(s->codec_id != CODEC_ID_MPEG2VIDEO){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
454 av_log(avctx, AV_LOG_ERROR, "non linear quant is only available for mpeg2\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
455 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
456 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
457 if(avctx->qmax > 12){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
458 av_log(avctx, AV_LOG_ERROR, "non linear quant only supports qmax <= 12 currently\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
459 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
460 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
461 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
462
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
463 if(s->avctx->thread_count > 1 && s->codec_id != CODEC_ID_MPEG4
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
464 && s->codec_id != CODEC_ID_MPEG1VIDEO && s->codec_id != CODEC_ID_MPEG2VIDEO
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
465 && (s->codec_id != CODEC_ID_H263P || !(s->flags & CODEC_FLAG_H263P_SLICE_STRUCT))){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
466 av_log(avctx, AV_LOG_ERROR, "multi threaded encoding not supported by codec\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
467 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
468 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
469
10155
30e57cd5d5f5 Check for thread_count==0.
michael
parents: 10146
diff changeset
470 if(s->avctx->thread_count < 1){
30e57cd5d5f5 Check for thread_count==0.
michael
parents: 10146
diff changeset
471 av_log(avctx, AV_LOG_ERROR, "automatic thread number detection not supported by codec, patch welcome\n");
30e57cd5d5f5 Check for thread_count==0.
michael
parents: 10146
diff changeset
472 return -1;
30e57cd5d5f5 Check for thread_count==0.
michael
parents: 10146
diff changeset
473 }
30e57cd5d5f5 Check for thread_count==0.
michael
parents: 10146
diff changeset
474
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
475 if(s->avctx->thread_count > 1)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
476 s->rtp_mode= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
477
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
478 if(!avctx->time_base.den || !avctx->time_base.num){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
479 av_log(avctx, AV_LOG_ERROR, "framerate not set\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
480 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
481 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
482
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
483 i= (INT_MAX/2+128)>>8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
484 if(avctx->me_threshold >= i){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
485 av_log(avctx, AV_LOG_ERROR, "me_threshold too large, max is %d\n", i - 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
486 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
487 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
488 if(avctx->mb_threshold >= i){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
489 av_log(avctx, AV_LOG_ERROR, "mb_threshold too large, max is %d\n", i - 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
490 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
491 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
492
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
493 if(avctx->b_frame_strategy && (avctx->flags&CODEC_FLAG_PASS2)){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
494 av_log(avctx, AV_LOG_INFO, "notice: b_frame_strategy only affects the first pass\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
495 avctx->b_frame_strategy = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
496 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
497
8611
a637db9f59b1 export gcd function as av_gcd()
aurel
parents: 8596
diff changeset
498 i= av_gcd(avctx->time_base.den, avctx->time_base.num);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
499 if(i > 1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
500 av_log(avctx, AV_LOG_INFO, "removing common factors from framerate\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
501 avctx->time_base.den /= i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
502 avctx->time_base.num /= i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
503 // return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
504 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
505
11493
495095fb615b use mpeg2 quantization bias for mjpeg.
michael
parents: 11417
diff changeset
506 if(s->mpeg_quant || s->codec_id==CODEC_ID_MPEG1VIDEO || s->codec_id==CODEC_ID_MPEG2VIDEO || s->codec_id==CODEC_ID_MJPEG){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
507 s->intra_quant_bias= 3<<(QUANT_BIAS_SHIFT-3); //(a + x*3/8)/x
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
508 s->inter_quant_bias= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
509 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
510 s->intra_quant_bias=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
511 s->inter_quant_bias=-(1<<(QUANT_BIAS_SHIFT-2)); //(a - x/4)/x
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
512 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
513
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
514 if(avctx->intra_quant_bias != FF_DEFAULT_QUANT_BIAS)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
515 s->intra_quant_bias= avctx->intra_quant_bias;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
516 if(avctx->inter_quant_bias != FF_DEFAULT_QUANT_BIAS)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
517 s->inter_quant_bias= avctx->inter_quant_bias;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
518
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
519 avcodec_get_chroma_sub_sample(avctx->pix_fmt, &chroma_h_shift, &chroma_v_shift);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
520
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
521 if(avctx->codec_id == CODEC_ID_MPEG4 && s->avctx->time_base.den > (1<<16)-1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
522 av_log(avctx, AV_LOG_ERROR, "timebase not supported by mpeg 4 standard\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
523 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
524 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
525 s->time_increment_bits = av_log2(s->avctx->time_base.den - 1) + 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
526
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
527 switch(avctx->codec->id) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
528 case CODEC_ID_MPEG1VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
529 s->out_format = FMT_MPEG1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
530 s->low_delay= !!(s->flags & CODEC_FLAG_LOW_DELAY);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
531 avctx->delay= s->low_delay ? 0 : (s->max_b_frames + 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
532 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
533 case CODEC_ID_MPEG2VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
534 s->out_format = FMT_MPEG1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
535 s->low_delay= !!(s->flags & CODEC_FLAG_LOW_DELAY);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
536 avctx->delay= s->low_delay ? 0 : (s->max_b_frames + 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
537 s->rtp_mode= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
538 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
539 case CODEC_ID_LJPEG:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
540 case CODEC_ID_MJPEG:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
541 s->out_format = FMT_MJPEG;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
542 s->intra_only = 1; /* force intra only for jpeg */
10670
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
543 if(avctx->codec->id == CODEC_ID_LJPEG && avctx->pix_fmt == PIX_FMT_BGRA){
10665
6cebfebeb18f Fix RGB LJPEG encoding.
michael
parents: 10637
diff changeset
544 s->mjpeg_vsample[0] = s->mjpeg_hsample[0] =
6cebfebeb18f Fix RGB LJPEG encoding.
michael
parents: 10637
diff changeset
545 s->mjpeg_vsample[1] = s->mjpeg_hsample[1] =
6cebfebeb18f Fix RGB LJPEG encoding.
michael
parents: 10637
diff changeset
546 s->mjpeg_vsample[2] = s->mjpeg_hsample[2] = 1;
10670
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
547 }else{
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
548 s->mjpeg_vsample[0] = 2;
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
549 s->mjpeg_vsample[1] = 2>>chroma_v_shift;
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
550 s->mjpeg_vsample[2] = 2>>chroma_v_shift;
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
551 s->mjpeg_hsample[0] = 2;
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
552 s->mjpeg_hsample[1] = 2>>chroma_h_shift;
7986ecc952e1 Cosmetic (restructure RGB ljpeg check to be more readable)
michael
parents: 10669
diff changeset
553 s->mjpeg_hsample[2] = 2>>chroma_h_shift;
10665
6cebfebeb18f Fix RGB LJPEG encoding.
michael
parents: 10637
diff changeset
554 }
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
555 if (!(CONFIG_MJPEG_ENCODER || CONFIG_LJPEG_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
556 || ff_mjpeg_encode_init(s) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
557 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
558 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
559 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
560 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
561 case CODEC_ID_H261:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
562 if (!CONFIG_H261_ENCODER) return -1;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
563 if (ff_h261_get_picture_format(s->width, s->height) < 0) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
564 av_log(avctx, AV_LOG_ERROR, "The specified picture size of %dx%d is not valid for the H.261 codec.\nValid sizes are 176x144, 352x288\n", s->width, s->height);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
565 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
566 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
567 s->out_format = FMT_H261;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
568 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
569 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
570 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
571 case CODEC_ID_H263:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
572 if (!CONFIG_H263_ENCODER) return -1;
10832
f20726a6d538 Add a function to match a 2 element vector of uint16_t and use it in h263 and svq1
michael
parents: 10803
diff changeset
573 if (ff_match_2uint16(h263_format, FF_ARRAY_ELEMS(h263_format), s->width, s->height) == 7) {
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
574 av_log(avctx, AV_LOG_INFO, "The specified picture size of %dx%d is not valid for the H.263 codec.\nValid sizes are 128x96, 176x144, 352x288, 704x576, and 1408x1152. Try H.263+.\n", s->width, s->height);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
575 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
576 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
577 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
578 s->obmc= (avctx->flags & CODEC_FLAG_OBMC) ? 1:0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
579 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
580 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
581 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
582 case CODEC_ID_H263P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
583 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
584 s->h263_plus = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
585 /* Fx */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
586 s->umvplus = (avctx->flags & CODEC_FLAG_H263P_UMV) ? 1:0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
587 s->h263_aic= (avctx->flags & CODEC_FLAG_AC_PRED) ? 1:0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
588 s->modified_quant= s->h263_aic;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
589 s->alt_inter_vlc= (avctx->flags & CODEC_FLAG_H263P_AIV) ? 1:0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
590 s->obmc= (avctx->flags & CODEC_FLAG_OBMC) ? 1:0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
591 s->loop_filter= (avctx->flags & CODEC_FLAG_LOOP_FILTER) ? 1:0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
592 s->unrestricted_mv= s->obmc || s->loop_filter || s->umvplus;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
593 s->h263_slice_structured= (s->flags & CODEC_FLAG_H263P_SLICE_STRUCT) ? 1:0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
594
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
595 /* /Fx */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
596 /* These are just to be sure */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
597 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
598 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
599 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
600 case CODEC_ID_FLV1:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
601 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
602 s->h263_flv = 2; /* format = 1; 11-bit codes */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
603 s->unrestricted_mv = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
604 s->rtp_mode=0; /* don't allow GOB */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
605 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
606 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
607 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
608 case CODEC_ID_RV10:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
609 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
610 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
611 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
612 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
613 case CODEC_ID_RV20:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
614 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
615 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
616 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
617 s->modified_quant=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
618 s->h263_aic=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
619 s->h263_plus=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
620 s->loop_filter=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
621 s->unrestricted_mv= s->obmc || s->loop_filter || s->umvplus;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
622 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
623 case CODEC_ID_MPEG4:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
624 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
625 s->h263_pred = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
626 s->unrestricted_mv = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
627 s->low_delay= s->max_b_frames ? 0 : 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
628 avctx->delay= s->low_delay ? 0 : (s->max_b_frames + 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
629 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
630 case CODEC_ID_MSMPEG4V1:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
631 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
632 s->h263_msmpeg4 = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
633 s->h263_pred = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
634 s->unrestricted_mv = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
635 s->msmpeg4_version= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
636 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
637 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
638 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
639 case CODEC_ID_MSMPEG4V2:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
640 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
641 s->h263_msmpeg4 = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
642 s->h263_pred = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
643 s->unrestricted_mv = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
644 s->msmpeg4_version= 2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
645 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
646 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
647 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
648 case CODEC_ID_MSMPEG4V3:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
649 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
650 s->h263_msmpeg4 = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
651 s->h263_pred = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
652 s->unrestricted_mv = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
653 s->msmpeg4_version= 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
654 s->flipflop_rounding=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
655 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
656 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
657 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
658 case CODEC_ID_WMV1:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
659 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
660 s->h263_msmpeg4 = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
661 s->h263_pred = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
662 s->unrestricted_mv = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
663 s->msmpeg4_version= 4;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
664 s->flipflop_rounding=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
665 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
666 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
667 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
668 case CODEC_ID_WMV2:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
669 s->out_format = FMT_H263;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
670 s->h263_msmpeg4 = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
671 s->h263_pred = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
672 s->unrestricted_mv = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
673 s->msmpeg4_version= 5;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
674 s->flipflop_rounding=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
675 avctx->delay=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
676 s->low_delay=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
677 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
678 default:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
679 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
680 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
681
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
682 avctx->has_b_frames= !s->low_delay;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
683
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
684 s->encoding = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
685
9690
cc0380d2bbc3 Set progressive_sequence before MPV_common_init which cares about it when
bcoudurier
parents: 9638
diff changeset
686 s->progressive_frame=
cc0380d2bbc3 Set progressive_sequence before MPV_common_init which cares about it when
bcoudurier
parents: 9638
diff changeset
687 s->progressive_sequence= !(avctx->flags & (CODEC_FLAG_INTERLACED_DCT|CODEC_FLAG_INTERLACED_ME|CODEC_FLAG_ALT_SCAN));
cc0380d2bbc3 Set progressive_sequence before MPV_common_init which cares about it when
bcoudurier
parents: 9638
diff changeset
688
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
689 /* init */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
690 if (MPV_common_init(s) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
691 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
692
5211
413c5e2eff68 move mpeg encoder specific initialization in the encoder specific file
aurel
parents: 5209
diff changeset
693 if(!s->dct_quantize)
413c5e2eff68 move mpeg encoder specific initialization in the encoder specific file
aurel
parents: 5209
diff changeset
694 s->dct_quantize = dct_quantize_c;
413c5e2eff68 move mpeg encoder specific initialization in the encoder specific file
aurel
parents: 5209
diff changeset
695 if(!s->denoise_dct)
413c5e2eff68 move mpeg encoder specific initialization in the encoder specific file
aurel
parents: 5209
diff changeset
696 s->denoise_dct = denoise_dct_c;
413c5e2eff68 move mpeg encoder specific initialization in the encoder specific file
aurel
parents: 5209
diff changeset
697 s->fast_dct_quantize = s->dct_quantize;
7034
130784da5595 uniformize trellis quant option
bcoudurier
parents: 6788
diff changeset
698 if(avctx->trellis)
5211
413c5e2eff68 move mpeg encoder specific initialization in the encoder specific file
aurel
parents: 5209
diff changeset
699 s->dct_quantize = dct_quantize_trellis_c;
413c5e2eff68 move mpeg encoder specific initialization in the encoder specific file
aurel
parents: 5209
diff changeset
700
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
701 if((CONFIG_H263P_ENCODER || CONFIG_RV20_ENCODER) && s->modified_quant)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
702 s->chroma_qscale_table= ff_h263_chroma_qscale_table;
9690
cc0380d2bbc3 Set progressive_sequence before MPV_common_init which cares about it when
bcoudurier
parents: 9638
diff changeset
703
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
704 s->quant_precision=5;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
705
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
706 ff_set_cmp(&s->dsp, s->dsp.ildct_cmp, s->avctx->ildct_cmp);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
707 ff_set_cmp(&s->dsp, s->dsp.frame_skip_cmp, s->avctx->frame_skip_cmp);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
708
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
709 if (CONFIG_H261_ENCODER && s->out_format == FMT_H261)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
710 ff_h261_encode_init(s);
10727
47e1e474e1f1 Replace CONFIG_ANY_H263_ENCODER definition by CONFIG_H263_ENCODER.
diego
parents: 10683
diff changeset
711 if (CONFIG_H263_ENCODER && s->out_format == FMT_H263)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
712 h263_encode_init(s);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
713 if (CONFIG_MSMPEG4_ENCODER && s->msmpeg4_version)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
714 ff_msmpeg4_encode_init(s);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
715 if ((CONFIG_MPEG1VIDEO_ENCODER || CONFIG_MPEG2VIDEO_ENCODER)
5208
430f60820b21 split mpeg12 encoder out of mpeg12.c
aurel
parents: 5204
diff changeset
716 && s->out_format == FMT_MPEG1)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
717 ff_mpeg1_encode_init(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
718
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
719 /* init q matrix */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
720 for(i=0;i<64;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
721 int j= s->dsp.idct_permutation[i];
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
722 if(CONFIG_MPEG4_ENCODER && s->codec_id==CODEC_ID_MPEG4 && s->mpeg_quant){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
723 s->intra_matrix[j] = ff_mpeg4_default_intra_matrix[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
724 s->inter_matrix[j] = ff_mpeg4_default_non_intra_matrix[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
725 }else if(s->out_format == FMT_H263 || s->out_format == FMT_H261){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
726 s->intra_matrix[j] =
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
727 s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
728 }else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
729 { /* mpeg1/2 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
730 s->intra_matrix[j] = ff_mpeg1_default_intra_matrix[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
731 s->inter_matrix[j] = ff_mpeg1_default_non_intra_matrix[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
732 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
733 if(s->avctx->intra_matrix)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
734 s->intra_matrix[j] = s->avctx->intra_matrix[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
735 if(s->avctx->inter_matrix)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
736 s->inter_matrix[j] = s->avctx->inter_matrix[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
737 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
738
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
739 /* precompute matrix */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
740 /* for mjpeg, we do include qscale in the matrix */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
741 if (s->out_format != FMT_MJPEG) {
5789
a40edc95456d export convert_matrix and add ff_ prefix
bcoudurier
parents: 5420
diff changeset
742 ff_convert_matrix(&s->dsp, s->q_intra_matrix, s->q_intra_matrix16,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
743 s->intra_matrix, s->intra_quant_bias, avctx->qmin, 31, 1);
5789
a40edc95456d export convert_matrix and add ff_ prefix
bcoudurier
parents: 5420
diff changeset
744 ff_convert_matrix(&s->dsp, s->q_inter_matrix, s->q_inter_matrix16,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
745 s->inter_matrix, s->inter_quant_bias, avctx->qmin, 31, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
746 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
747
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
748 if(ff_rate_control_init(s) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
749 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
750
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
751 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
752 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
753
6517
48759bfbd073 Apply 'cold' attribute to init/uninit functions in libavcodec
zuxy
parents: 6497
diff changeset
754 av_cold int MPV_encode_end(AVCodecContext *avctx)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
755 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
756 MpegEncContext *s = avctx->priv_data;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
757
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
758 ff_rate_control_uninit(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
759
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
760 MPV_common_end(s);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
761 if ((CONFIG_MJPEG_ENCODER || CONFIG_LJPEG_ENCODER) && s->out_format == FMT_MJPEG)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
762 ff_mjpeg_encode_close(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
763
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
764 av_freep(&avctx->extradata);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
765
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
766 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
767 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
768
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
769 static int get_sae(uint8_t *src, int ref, int stride){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
770 int x,y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
771 int acc=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
772
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
773 for(y=0; y<16; y++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
774 for(x=0; x<16; x++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
775 acc+= FFABS(src[x+y*stride] - ref);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
776 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
777 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
778
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
779 return acc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
780 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
781
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
782 static int get_intra_count(MpegEncContext *s, uint8_t *src, uint8_t *ref, int stride){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
783 int x, y, w, h;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
784 int acc=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
785
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
786 w= s->width &~15;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
787 h= s->height&~15;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
788
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
789 for(y=0; y<h; y+=16){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
790 for(x=0; x<w; x+=16){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
791 int offset= x + y*stride;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
792 int sad = s->dsp.sad[0](NULL, src + offset, ref + offset, stride, 16);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
793 int mean= (s->dsp.pix_sum(src + offset, stride) + 128)>>8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
794 int sae = get_sae(src + offset, mean, stride);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
795
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
796 acc+= sae + 500 < sad;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
797 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
798 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
799 return acc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
800 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
801
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
802
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
803 static int load_input_picture(MpegEncContext *s, AVFrame *pic_arg){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
804 AVFrame *pic=NULL;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
805 int64_t pts;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
806 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
807 const int encoding_delay= s->max_b_frames;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
808 int direct=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
809
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
810 if(pic_arg){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
811 pts= pic_arg->pts;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
812 pic_arg->display_picture_number= s->input_picture_number++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
813
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
814 if(pts != AV_NOPTS_VALUE){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
815 if(s->user_specified_pts != AV_NOPTS_VALUE){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
816 int64_t time= pts;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
817 int64_t last= s->user_specified_pts;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
818
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
819 if(time <= last){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
820 av_log(s->avctx, AV_LOG_ERROR, "Error, Invalid timestamp=%"PRId64", last=%"PRId64"\n", pts, s->user_specified_pts);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
821 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
822 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
823 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
824 s->user_specified_pts= pts;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
825 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
826 if(s->user_specified_pts != AV_NOPTS_VALUE){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
827 s->user_specified_pts=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
828 pts= s->user_specified_pts + 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
829 av_log(s->avctx, AV_LOG_INFO, "Warning: AVFrame.pts=? trying to guess (%"PRId64")\n", pts);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
830 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
831 pts= pic_arg->display_picture_number;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
832 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
833 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
834 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
835
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
836 if(pic_arg){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
837 if(encoding_delay && !(s->flags&CODEC_FLAG_INPUT_PRESERVED)) direct=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
838 if(pic_arg->linesize[0] != s->linesize) direct=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
839 if(pic_arg->linesize[1] != s->uvlinesize) direct=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
840 if(pic_arg->linesize[2] != s->uvlinesize) direct=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
841
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
842 // av_log(AV_LOG_DEBUG, "%d %d %d %d\n",pic_arg->linesize[0], pic_arg->linesize[1], s->linesize, s->uvlinesize);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
843
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
844 if(direct){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
845 i= ff_find_unused_picture(s, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
846
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
847 pic= (AVFrame*)&s->picture[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
848 pic->reference= 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
849
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
850 for(i=0; i<4; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
851 pic->data[i]= pic_arg->data[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
852 pic->linesize[i]= pic_arg->linesize[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
853 }
11534
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
854 if(ff_alloc_picture(s, (Picture*)pic, 1) < 0){
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
855 return -1;
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
856 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
857 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
858 i= ff_find_unused_picture(s, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
859
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
860 pic= (AVFrame*)&s->picture[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
861 pic->reference= 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
862
11534
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
863 if(ff_alloc_picture(s, (Picture*)pic, 0) < 0){
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
864 return -1;
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
865 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
866
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
867 if( pic->data[0] + INPLACE_OFFSET == pic_arg->data[0]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
868 && pic->data[1] + INPLACE_OFFSET == pic_arg->data[1]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
869 && pic->data[2] + INPLACE_OFFSET == pic_arg->data[2]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
870 // empty
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
871 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
872 int h_chroma_shift, v_chroma_shift;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
873 avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &h_chroma_shift, &v_chroma_shift);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
874
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
875 for(i=0; i<3; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
876 int src_stride= pic_arg->linesize[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
877 int dst_stride= i ? s->uvlinesize : s->linesize;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
878 int h_shift= i ? h_chroma_shift : 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
879 int v_shift= i ? v_chroma_shift : 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
880 int w= s->width >>h_shift;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
881 int h= s->height>>v_shift;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
882 uint8_t *src= pic_arg->data[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
883 uint8_t *dst= pic->data[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
884
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
885 if(!s->avctx->rc_buffer_size)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
886 dst +=INPLACE_OFFSET;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
887
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
888 if(src_stride==dst_stride)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
889 memcpy(dst, src, src_stride*h);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
890 else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
891 while(h--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
892 memcpy(dst, src, w);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
893 dst += dst_stride;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
894 src += src_stride;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
895 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
896 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
897 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
898 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
899 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
900 copy_picture_attributes(s, pic, pic_arg);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
901 pic->pts= pts; //we set this here to avoid modifiying pic_arg
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
902 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
903
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
904 /* shift buffer entries */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
905 for(i=1; i<MAX_PICTURE_COUNT /*s->encoding_delay+1*/; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
906 s->input_picture[i-1]= s->input_picture[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
907
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
908 s->input_picture[encoding_delay]= (Picture*)pic;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
909
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
910 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
911 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
912
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
913 static int skip_check(MpegEncContext *s, Picture *p, Picture *ref){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
914 int x, y, plane;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
915 int score=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
916 int64_t score64=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
917
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
918 for(plane=0; plane<3; plane++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
919 const int stride= p->linesize[plane];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
920 const int bw= plane ? 1 : 2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
921 for(y=0; y<s->mb_height*bw; y++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
922 for(x=0; x<s->mb_width*bw; x++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
923 int off= p->type == FF_BUFFER_TYPE_SHARED ? 0: 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
924 int v= s->dsp.frame_skip_cmp[1](s, p->data[plane] + 8*(x + y*stride)+off, ref->data[plane] + 8*(x + y*stride), stride, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
925
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
926 switch(s->avctx->frame_skip_exp){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
927 case 0: score= FFMAX(score, v); break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
928 case 1: score+= FFABS(v);break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
929 case 2: score+= v*v;break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
930 case 3: score64+= FFABS(v*v*(int64_t)v);break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
931 case 4: score64+= v*v*(int64_t)(v*v);break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
932 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
933 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
934 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
935 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
936
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
937 if(score) score64= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
938
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
939 if(score64 < s->avctx->frame_skip_threshold)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
940 return 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
941 if(score64 < ((s->avctx->frame_skip_factor * (int64_t)s->lambda)>>8))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
942 return 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
943 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
944 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
945
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
946 static int estimate_best_b_count(MpegEncContext *s){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
947 AVCodec *codec= avcodec_find_encoder(s->avctx->codec_id);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
948 AVCodecContext *c= avcodec_alloc_context();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
949 AVFrame input[FF_MAX_B_FRAMES+2];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
950 const int scale= s->avctx->brd_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
951 int i, j, out_size, p_lambda, b_lambda, lambda2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
952 int outbuf_size= s->width * s->height; //FIXME
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
953 uint8_t *outbuf= av_malloc(outbuf_size);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
954 int64_t best_rd= INT64_MAX;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
955 int best_b_count= -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
956
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
957 assert(scale>=0 && scale <=3);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
958
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
959 // emms_c();
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
960 p_lambda= s->last_lambda_for[FF_P_TYPE]; //s->next_picture_ptr->quality;
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
961 b_lambda= s->last_lambda_for[FF_B_TYPE]; //p_lambda *FFABS(s->avctx->b_quant_factor) + s->avctx->b_quant_offset;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
962 if(!b_lambda) b_lambda= p_lambda; //FIXME we should do this somewhere else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
963 lambda2= (b_lambda*b_lambda + (1<<FF_LAMBDA_SHIFT)/2 ) >> FF_LAMBDA_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
964
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
965 c->width = s->width >> scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
966 c->height= s->height>> scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
967 c->flags= CODEC_FLAG_QSCALE | CODEC_FLAG_PSNR | CODEC_FLAG_INPUT_PRESERVED /*| CODEC_FLAG_EMU_EDGE*/;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
968 c->flags|= s->avctx->flags & CODEC_FLAG_QPEL;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
969 c->mb_decision= s->avctx->mb_decision;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
970 c->me_cmp= s->avctx->me_cmp;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
971 c->mb_cmp= s->avctx->mb_cmp;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
972 c->me_sub_cmp= s->avctx->me_sub_cmp;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
973 c->pix_fmt = PIX_FMT_YUV420P;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
974 c->time_base= s->avctx->time_base;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
975 c->max_b_frames= s->max_b_frames;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
976
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
977 if (avcodec_open(c, codec) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
978 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
979
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
980 for(i=0; i<s->max_b_frames+2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
981 int ysize= c->width*c->height;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
982 int csize= (c->width/2)*(c->height/2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
983 Picture pre_input, *pre_input_ptr= i ? s->input_picture[i-1] : s->next_picture_ptr;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
984
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
985 avcodec_get_frame_defaults(&input[i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
986 input[i].data[0]= av_malloc(ysize + 2*csize);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
987 input[i].data[1]= input[i].data[0] + ysize;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
988 input[i].data[2]= input[i].data[1] + csize;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
989 input[i].linesize[0]= c->width;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
990 input[i].linesize[1]=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
991 input[i].linesize[2]= c->width/2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
992
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
993 if(pre_input_ptr && (!i || s->input_picture[i-1])) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
994 pre_input= *pre_input_ptr;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
995
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
996 if(pre_input.type != FF_BUFFER_TYPE_SHARED && i) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
997 pre_input.data[0]+=INPLACE_OFFSET;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
998 pre_input.data[1]+=INPLACE_OFFSET;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
999 pre_input.data[2]+=INPLACE_OFFSET;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1000 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1001
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1002 s->dsp.shrink[scale](input[i].data[0], input[i].linesize[0], pre_input.data[0], pre_input.linesize[0], c->width, c->height);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1003 s->dsp.shrink[scale](input[i].data[1], input[i].linesize[1], pre_input.data[1], pre_input.linesize[1], c->width>>1, c->height>>1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1004 s->dsp.shrink[scale](input[i].data[2], input[i].linesize[2], pre_input.data[2], pre_input.linesize[2], c->width>>1, c->height>>1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1005 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1006 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1007
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1008 for(j=0; j<s->max_b_frames+1; j++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1009 int64_t rd=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1010
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1011 if(!s->input_picture[j])
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1012 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1013
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1014 c->error[0]= c->error[1]= c->error[2]= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1015
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1016 input[0].pict_type= FF_I_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1017 input[0].quality= 1 * FF_QP2LAMBDA;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1018 out_size = avcodec_encode_video(c, outbuf, outbuf_size, &input[0]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1019 // rd += (out_size * lambda2) >> FF_LAMBDA_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1020
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1021 for(i=0; i<s->max_b_frames+1; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1022 int is_p= i % (j+1) == j || i==s->max_b_frames;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1023
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1024 input[i+1].pict_type= is_p ? FF_P_TYPE : FF_B_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1025 input[i+1].quality= is_p ? p_lambda : b_lambda;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1026 out_size = avcodec_encode_video(c, outbuf, outbuf_size, &input[i+1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1027 rd += (out_size * lambda2) >> (FF_LAMBDA_SHIFT - 3);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1028 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1029
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1030 /* get the delayed frames */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1031 while(out_size){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1032 out_size = avcodec_encode_video(c, outbuf, outbuf_size, NULL);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1033 rd += (out_size * lambda2) >> (FF_LAMBDA_SHIFT - 3);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1034 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1035
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1036 rd += c->error[0] + c->error[1] + c->error[2];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1037
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1038 if(rd < best_rd){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1039 best_rd= rd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1040 best_b_count= j;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1041 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1042 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1043
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1044 av_freep(&outbuf);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1045 avcodec_close(c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1046 av_freep(&c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1047
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1048 for(i=0; i<s->max_b_frames+2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1049 av_freep(&input[i].data[0]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1050 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1051
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1052 return best_b_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1053 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1054
11534
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1055 static int select_input_picture(MpegEncContext *s){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1056 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1057
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1058 for(i=1; i<MAX_PICTURE_COUNT; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1059 s->reordered_input_picture[i-1]= s->reordered_input_picture[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1060 s->reordered_input_picture[MAX_PICTURE_COUNT-1]= NULL;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1061
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1062 /* set next picture type & ordering */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1063 if(s->reordered_input_picture[0]==NULL && s->input_picture[0]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1064 if(/*s->picture_in_gop_number >= s->gop_size ||*/ s->next_picture_ptr==NULL || s->intra_only){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1065 s->reordered_input_picture[0]= s->input_picture[0];
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1066 s->reordered_input_picture[0]->pict_type= FF_I_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1067 s->reordered_input_picture[0]->coded_picture_number= s->coded_picture_number++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1068 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1069 int b_frames;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1070
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1071 if(s->avctx->frame_skip_threshold || s->avctx->frame_skip_factor){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1072 if(s->picture_in_gop_number < s->gop_size && skip_check(s, s->input_picture[0], s->next_picture_ptr)){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1073 //FIXME check that te gop check above is +-1 correct
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1074 //av_log(NULL, AV_LOG_DEBUG, "skip %p %"PRId64"\n", s->input_picture[0]->data[0], s->input_picture[0]->pts);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1075
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1076 if(s->input_picture[0]->type == FF_BUFFER_TYPE_SHARED){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1077 for(i=0; i<4; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1078 s->input_picture[0]->data[i]= NULL;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1079 s->input_picture[0]->type= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1080 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1081 assert( s->input_picture[0]->type==FF_BUFFER_TYPE_USER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1082 || s->input_picture[0]->type==FF_BUFFER_TYPE_INTERNAL);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1083
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1084 s->avctx->release_buffer(s->avctx, (AVFrame*)s->input_picture[0]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1085 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1086
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1087 emms_c();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1088 ff_vbv_update(s, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1089
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1090 goto no_output_pic;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1091 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1092 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1093
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1094 if(s->flags&CODEC_FLAG_PASS2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1095 for(i=0; i<s->max_b_frames+1; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1096 int pict_num= s->input_picture[0]->display_picture_number + i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1097
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1098 if(pict_num >= s->rc_context.num_entries)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1099 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1100 if(!s->input_picture[i]){
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1101 s->rc_context.entry[pict_num-1].new_pict_type = FF_P_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1102 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1103 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1104
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1105 s->input_picture[i]->pict_type=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1106 s->rc_context.entry[pict_num].new_pict_type;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1107 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1108 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1109
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1110 if(s->avctx->b_frame_strategy==0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1111 b_frames= s->max_b_frames;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1112 while(b_frames && !s->input_picture[b_frames]) b_frames--;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1113 }else if(s->avctx->b_frame_strategy==1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1114 for(i=1; i<s->max_b_frames+1; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1115 if(s->input_picture[i] && s->input_picture[i]->b_frame_score==0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1116 s->input_picture[i]->b_frame_score=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1117 get_intra_count(s, s->input_picture[i ]->data[0],
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1118 s->input_picture[i-1]->data[0], s->linesize) + 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1119 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1120 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1121 for(i=0; i<s->max_b_frames+1; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1122 if(s->input_picture[i]==NULL || s->input_picture[i]->b_frame_score - 1 > s->mb_num/s->avctx->b_sensitivity) break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1123 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1124
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1125 b_frames= FFMAX(0, i-1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1126
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1127 /* reset scores */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1128 for(i=0; i<b_frames+1; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1129 s->input_picture[i]->b_frame_score=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1130 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1131 }else if(s->avctx->b_frame_strategy==2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1132 b_frames= estimate_best_b_count(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1133 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1134 av_log(s->avctx, AV_LOG_ERROR, "illegal b frame strategy\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1135 b_frames=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1136 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1137
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1138 emms_c();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1139 //static int b_count=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1140 //b_count+= b_frames;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1141 //av_log(s->avctx, AV_LOG_DEBUG, "b_frames: %d\n", b_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1142
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1143 for(i= b_frames - 1; i>=0; i--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1144 int type= s->input_picture[i]->pict_type;
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1145 if(type && type != FF_B_TYPE)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1146 b_frames= i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1147 }
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1148 if(s->input_picture[b_frames]->pict_type == FF_B_TYPE && b_frames == s->max_b_frames){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1149 av_log(s->avctx, AV_LOG_ERROR, "warning, too many b frames in a row\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1150 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1151
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1152 if(s->picture_in_gop_number + b_frames >= s->gop_size){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1153 if((s->flags2 & CODEC_FLAG2_STRICT_GOP) && s->gop_size > s->picture_in_gop_number){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1154 b_frames= s->gop_size - s->picture_in_gop_number - 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1155 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1156 if(s->flags & CODEC_FLAG_CLOSED_GOP)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1157 b_frames=0;
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1158 s->input_picture[b_frames]->pict_type= FF_I_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1159 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1160 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1161
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1162 if( (s->flags & CODEC_FLAG_CLOSED_GOP)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1163 && b_frames
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1164 && s->input_picture[b_frames]->pict_type== FF_I_TYPE)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1165 b_frames--;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1166
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1167 s->reordered_input_picture[0]= s->input_picture[b_frames];
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1168 if(s->reordered_input_picture[0]->pict_type != FF_I_TYPE)
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1169 s->reordered_input_picture[0]->pict_type= FF_P_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1170 s->reordered_input_picture[0]->coded_picture_number= s->coded_picture_number++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1171 for(i=0; i<b_frames; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1172 s->reordered_input_picture[i+1]= s->input_picture[i];
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1173 s->reordered_input_picture[i+1]->pict_type= FF_B_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1174 s->reordered_input_picture[i+1]->coded_picture_number= s->coded_picture_number++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1175 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1176 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1177 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1178 no_output_pic:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1179 if(s->reordered_input_picture[0]){
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1180 s->reordered_input_picture[0]->reference= s->reordered_input_picture[0]->pict_type!=FF_B_TYPE ? 3 : 0;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1181
7974
2b20ad98e424 Rename copy_picture to ff_copy_picture.
cehoyos
parents: 7824
diff changeset
1182 ff_copy_picture(&s->new_picture, s->reordered_input_picture[0]);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1183
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1184 if(s->reordered_input_picture[0]->type == FF_BUFFER_TYPE_SHARED || s->avctx->rc_buffer_size){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1185 // input is a shared pix, so we can't modifiy it -> alloc a new one & ensure that the shared one is reuseable
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1186
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1187 int i= ff_find_unused_picture(s, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1188 Picture *pic= &s->picture[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1189
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1190 pic->reference = s->reordered_input_picture[0]->reference;
11534
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1191 if(ff_alloc_picture(s, pic, 0) < 0){
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1192 return -1;
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1193 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1194
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1195 /* mark us unused / free shared pic */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1196 if(s->reordered_input_picture[0]->type == FF_BUFFER_TYPE_INTERNAL)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1197 s->avctx->release_buffer(s->avctx, (AVFrame*)s->reordered_input_picture[0]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1198 for(i=0; i<4; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1199 s->reordered_input_picture[0]->data[i]= NULL;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1200 s->reordered_input_picture[0]->type= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1201
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1202 copy_picture_attributes(s, (AVFrame*)pic, (AVFrame*)s->reordered_input_picture[0]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1203
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1204 s->current_picture_ptr= pic;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1205 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1206 // input is not a shared pix -> reuse buffer for current_pix
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1207
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1208 assert( s->reordered_input_picture[0]->type==FF_BUFFER_TYPE_USER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1209 || s->reordered_input_picture[0]->type==FF_BUFFER_TYPE_INTERNAL);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1210
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1211 s->current_picture_ptr= s->reordered_input_picture[0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1212 for(i=0; i<4; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1213 s->new_picture.data[i]+= INPLACE_OFFSET;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1214 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1215 }
7974
2b20ad98e424 Rename copy_picture to ff_copy_picture.
cehoyos
parents: 7824
diff changeset
1216 ff_copy_picture(&s->current_picture, s->current_picture_ptr);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1217
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1218 s->picture_number= s->new_picture.display_picture_number;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1219 //printf("dpn:%d\n", s->picture_number);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1220 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1221 memset(&s->new_picture, 0, sizeof(Picture));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1222 }
11534
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1223 return 0;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1224 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1225
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1226 int MPV_encode_picture(AVCodecContext *avctx,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1227 unsigned char *buf, int buf_size, void *data)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1228 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1229 MpegEncContext *s = avctx->priv_data;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1230 AVFrame *pic_arg = data;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1231 int i, stuffing_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1232
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1233 for(i=0; i<avctx->thread_count; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1234 int start_y= s->thread_context[i]->start_mb_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1235 int end_y= s->thread_context[i]-> end_mb_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1236 int h= s->mb_height;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1237 uint8_t *start= buf + (size_t)(((int64_t) buf_size)*start_y/h);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1238 uint8_t *end = buf + (size_t)(((int64_t) buf_size)* end_y/h);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1239
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1240 init_put_bits(&s->thread_context[i]->pb, start, end - start);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1241 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1242
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1243 s->picture_in_gop_number++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1244
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1245 if(load_input_picture(s, pic_arg) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1246 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1247
11534
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1248 if(select_input_picture(s) < 0){
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1249 return -1;
baece61a55cf Always check if ff_alloc_picture() succeeds.
cehoyos
parents: 11531
diff changeset
1250 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1251
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1252 /* output? */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1253 if(s->new_picture.data[0]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1254 s->pict_type= s->new_picture.pict_type;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1255 //emms_c();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1256 //printf("qs:%f %f %d\n", s->new_picture.quality, s->current_picture.quality, s->qscale);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1257 MPV_frame_start(s, avctx);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1258 vbv_retry:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1259 if (encode_picture(s, s->picture_number) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1260 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1261
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1262 avctx->header_bits = s->header_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1263 avctx->mv_bits = s->mv_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1264 avctx->misc_bits = s->misc_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1265 avctx->i_tex_bits = s->i_tex_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1266 avctx->p_tex_bits = s->p_tex_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1267 avctx->i_count = s->i_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1268 avctx->p_count = s->mb_num - s->i_count - s->skip_count; //FIXME f/b_count in avctx
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1269 avctx->skip_count = s->skip_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1270
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1271 MPV_frame_end(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1272
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
1273 if (CONFIG_MJPEG_ENCODER && s->out_format == FMT_MJPEG)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1274 ff_mjpeg_encode_picture_trailer(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1275
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1276 if(avctx->rc_buffer_size){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1277 RateControlContext *rcc= &s->rc_context;
8227
596677256482 Implement the fields rc_max_available_vbv_use and
stefano
parents: 8218
diff changeset
1278 int max_size= rcc->buffer_index * avctx->rc_max_available_vbv_use;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1279
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1280 if(put_bits_count(&s->pb) > max_size && s->lambda < s->avctx->lmax){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1281 s->next_lambda= FFMAX(s->lambda+1, s->lambda*(s->qscale+1) / s->qscale);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1282 if(s->adaptive_quant){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1283 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1284 for(i=0; i<s->mb_height*s->mb_stride; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1285 s->lambda_table[i]= FFMAX(s->lambda_table[i]+1, s->lambda_table[i]*(s->qscale+1) / s->qscale);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1286 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1287 s->mb_skipped = 0; //done in MPV_frame_start()
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1288 if(s->pict_type==FF_P_TYPE){ //done in encode_picture() so we must undo it
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1289 if(s->flipflop_rounding || s->codec_id == CODEC_ID_H263P || s->codec_id == CODEC_ID_MPEG4)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1290 s->no_rounding ^= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1291 }
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1292 if(s->pict_type!=FF_B_TYPE){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1293 s->time_base= s->last_time_base;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1294 s->last_non_b_time= s->time - s->pp_time;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1295 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1296 // av_log(NULL, AV_LOG_ERROR, "R:%d ", s->next_lambda);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1297 for(i=0; i<avctx->thread_count; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1298 PutBitContext *pb= &s->thread_context[i]->pb;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1299 init_put_bits(pb, pb->buf, pb->buf_end - pb->buf);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1300 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1301 goto vbv_retry;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1302 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1303
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1304 assert(s->avctx->rc_max_rate);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1305 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1306
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1307 if(s->flags&CODEC_FLAG_PASS1)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1308 ff_write_pass1_stats(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1309
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1310 for(i=0; i<4; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1311 s->current_picture_ptr->error[i]= s->current_picture.error[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1312 avctx->error[i] += s->current_picture_ptr->error[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1313 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1314
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1315 if(s->flags&CODEC_FLAG_PASS1)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1316 assert(avctx->header_bits + avctx->mv_bits + avctx->misc_bits + avctx->i_tex_bits + avctx->p_tex_bits == put_bits_count(&s->pb));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1317 flush_put_bits(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1318 s->frame_bits = put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1319
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1320 stuffing_count= ff_vbv_update(s, s->frame_bits);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1321 if(stuffing_count){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1322 if(s->pb.buf_end - s->pb.buf - (put_bits_count(&s->pb)>>3) < stuffing_count + 50){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1323 av_log(s->avctx, AV_LOG_ERROR, "stuffing too large\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1324 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1325 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1326
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1327 switch(s->codec_id){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1328 case CODEC_ID_MPEG1VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1329 case CODEC_ID_MPEG2VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1330 while(stuffing_count--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1331 put_bits(&s->pb, 8, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1332 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1333 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1334 case CODEC_ID_MPEG4:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1335 put_bits(&s->pb, 16, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1336 put_bits(&s->pb, 16, 0x1C3);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1337 stuffing_count -= 4;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1338 while(stuffing_count--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1339 put_bits(&s->pb, 8, 0xFF);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1340 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1341 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1342 default:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1343 av_log(s->avctx, AV_LOG_ERROR, "vbv buffer overflow\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1344 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1345 flush_put_bits(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1346 s->frame_bits = put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1347 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1348
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1349 /* update mpeg1/2 vbv_delay for CBR */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1350 if(s->avctx->rc_max_rate && s->avctx->rc_min_rate == s->avctx->rc_max_rate && s->out_format == FMT_MPEG1
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1351 && 90000LL * (avctx->rc_buffer_size-1) <= s->avctx->rc_max_rate*0xFFFFLL){
9638
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1352 int vbv_delay, min_delay;
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1353 double inbits = s->avctx->rc_max_rate*av_q2d(s->avctx->time_base);
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1354 int minbits= s->frame_bits - 8*(s->vbv_delay_ptr - s->pb.buf - 1);
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1355 double bits = s->rc_context.buffer_index + minbits - inbits;
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1356
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1357 if(bits<0)
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1358 av_log(s->avctx, AV_LOG_ERROR, "Internal error, negative bits\n");
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1359
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1360 assert(s->repeat_first_field==0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1361
9638
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1362 vbv_delay= bits * 90000 / s->avctx->rc_max_rate;
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1363 min_delay= (minbits * 90000LL + s->avctx->rc_max_rate - 1)/ s->avctx->rc_max_rate;
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1364
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1365 vbv_delay= FFMAX(vbv_delay, min_delay);
cbd09311872d Fix VBV delay for mpeg1/2 CBR.
michael
parents: 9591
diff changeset
1366
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1367 assert(vbv_delay < 0xFFFF);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1368
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1369 s->vbv_delay_ptr[0] &= 0xF8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1370 s->vbv_delay_ptr[0] |= vbv_delay>>13;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1371 s->vbv_delay_ptr[1] = vbv_delay>>5;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1372 s->vbv_delay_ptr[2] &= 0x07;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1373 s->vbv_delay_ptr[2] |= vbv_delay<<3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1374 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1375 s->total_bits += s->frame_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1376 avctx->frame_bits = s->frame_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1377 }else{
9431
932543edc1d2 Rename pbBufPtr() to put_bits_ptr().
stefano
parents: 9083
diff changeset
1378 assert((put_bits_ptr(&s->pb) == s->pb.buf));
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1379 s->frame_bits=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1380 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1381 assert((s->frame_bits&7)==0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1382
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1383 return s->frame_bits/8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1384 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1385
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1386 static inline void dct_single_coeff_elimination(MpegEncContext *s, int n, int threshold)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1387 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1388 static const char tab[64]=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1389 {3,2,2,1,1,1,1,1,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1390 1,1,1,1,1,1,1,1,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1391 1,1,1,1,1,1,1,1,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1392 0,0,0,0,0,0,0,0,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1393 0,0,0,0,0,0,0,0,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1394 0,0,0,0,0,0,0,0,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1395 0,0,0,0,0,0,0,0,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1396 0,0,0,0,0,0,0,0};
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1397 int score=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1398 int run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1399 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1400 DCTELEM *block= s->block[n];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1401 const int last_index= s->block_last_index[n];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1402 int skip_dc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1403
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1404 if(threshold<0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1405 skip_dc=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1406 threshold= -threshold;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1407 }else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1408 skip_dc=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1409
6497
e44443c512b1 typo fixes
diego
parents: 6486
diff changeset
1410 /* Are all we could set to zero already zero? */
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1411 if(last_index<=skip_dc - 1) return;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1412
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1413 for(i=0; i<=last_index; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1414 const int j = s->intra_scantable.permutated[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1415 const int level = FFABS(block[j]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1416 if(level==1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1417 if(skip_dc && i==0) continue;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1418 score+= tab[run];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1419 run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1420 }else if(level>1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1421 return;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1422 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1423 run++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1424 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1425 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1426 if(score >= threshold) return;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1427 for(i=skip_dc; i<=last_index; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1428 const int j = s->intra_scantable.permutated[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1429 block[j]=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1430 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1431 if(block[0]) s->block_last_index[n]= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1432 else s->block_last_index[n]= -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1433 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1434
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1435 static inline void clip_coeffs(MpegEncContext *s, DCTELEM *block, int last_index)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1436 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1437 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1438 const int maxlevel= s->max_qcoeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1439 const int minlevel= s->min_qcoeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1440 int overflow=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1441
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1442 if(s->mb_intra){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1443 i=1; //skip clipping of intra dc
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1444 }else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1445 i=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1446
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1447 for(;i<=last_index; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1448 const int j= s->intra_scantable.permutated[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1449 int level = block[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1450
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1451 if (level>maxlevel){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1452 level=maxlevel;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1453 overflow++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1454 }else if(level<minlevel){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1455 level=minlevel;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1456 overflow++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1457 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1458
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1459 block[j]= level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1460 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1461
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1462 if(overflow && s->avctx->mb_decision == FF_MB_DECISION_SIMPLE)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1463 av_log(s->avctx, AV_LOG_INFO, "warning, clipping %d dct coefficients to %d..%d\n", overflow, minlevel, maxlevel);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1464 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1465
5909
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1466 static void get_visual_weight(int16_t *weight, uint8_t *ptr, int stride){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1467 int x, y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1468 //FIXME optimize
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1469 for(y=0; y<8; y++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1470 for(x=0; x<8; x++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1471 int x2, y2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1472 int sum=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1473 int sqr=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1474 int count=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1475
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1476 for(y2= FFMAX(y-1, 0); y2 < FFMIN(8, y+2); y2++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1477 for(x2= FFMAX(x-1, 0); x2 < FFMIN(8, x+2); x2++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1478 int v= ptr[x2 + y2*stride];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1479 sum += v;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1480 sqr += v*v;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1481 count++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1482 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1483 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1484 weight[x + 8*y]= (36*ff_sqrt(count*sqr - sum*sum)) / count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1485 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1486 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1487 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1488
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1489 static av_always_inline void encode_mb_internal(MpegEncContext *s, int motion_x, int motion_y, int mb_block_height, int mb_block_count)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1490 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1491 int16_t weight[8][64];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1492 DCTELEM orig[8][64];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1493 const int mb_x= s->mb_x;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1494 const int mb_y= s->mb_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1495 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1496 int skip_dct[8];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1497 int dct_offset = s->linesize*8; //default for progressive frames
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1498 uint8_t *ptr_y, *ptr_cb, *ptr_cr;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1499 int wrap_y, wrap_c;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1500
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1501 for(i=0; i<mb_block_count; i++) skip_dct[i]=s->skipdct;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1502
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1503 if(s->adaptive_quant){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1504 const int last_qp= s->qscale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1505 const int mb_xy= mb_x + mb_y*s->mb_stride;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1506
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1507 s->lambda= s->lambda_table[mb_xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1508 update_qscale(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1509
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1510 if(!(s->flags&CODEC_FLAG_QP_RD)){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1511 s->qscale= s->current_picture_ptr->qscale_table[mb_xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1512 s->dquant= s->qscale - last_qp;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1513
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1514 if(s->out_format==FMT_H263){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1515 s->dquant= av_clip(s->dquant, -2, 2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1516
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1517 if(s->codec_id==CODEC_ID_MPEG4){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1518 if(!s->mb_intra){
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1519 if(s->pict_type == FF_B_TYPE){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1520 if(s->dquant&1 || s->mv_dir&MV_DIRECT)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1521 s->dquant= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1522 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1523 if(s->mv_type==MV_TYPE_8X8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1524 s->dquant=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1525 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1526 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1527 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1528 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1529 ff_set_qscale(s, last_qp + s->dquant);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1530 }else if(s->flags&CODEC_FLAG_QP_RD)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1531 ff_set_qscale(s, s->qscale + s->dquant);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1532
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1533 wrap_y = s->linesize;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1534 wrap_c = s->uvlinesize;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1535 ptr_y = s->new_picture.data[0] + (mb_y * 16 * wrap_y) + mb_x * 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1536 ptr_cb = s->new_picture.data[1] + (mb_y * mb_block_height * wrap_c) + mb_x * 8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1537 ptr_cr = s->new_picture.data[2] + (mb_y * mb_block_height * wrap_c) + mb_x * 8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1538
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1539 if(mb_x*16+16 > s->width || mb_y*16+16 > s->height){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1540 uint8_t *ebuf= s->edge_emu_buffer + 32;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1541 ff_emulated_edge_mc(ebuf , ptr_y , wrap_y,16,16,mb_x*16,mb_y*16, s->width , s->height);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1542 ptr_y= ebuf;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1543 ff_emulated_edge_mc(ebuf+18*wrap_y , ptr_cb, wrap_c, 8, mb_block_height, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1544 ptr_cb= ebuf+18*wrap_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1545 ff_emulated_edge_mc(ebuf+18*wrap_y+8, ptr_cr, wrap_c, 8, mb_block_height, mb_x*8, mb_y*8, s->width>>1, s->height>>1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1546 ptr_cr= ebuf+18*wrap_y+8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1547 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1548
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1549 if (s->mb_intra) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1550 if(s->flags&CODEC_FLAG_INTERLACED_DCT){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1551 int progressive_score, interlaced_score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1552
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1553 s->interlaced_dct=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1554 progressive_score= s->dsp.ildct_cmp[4](s, ptr_y , NULL, wrap_y, 8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1555 +s->dsp.ildct_cmp[4](s, ptr_y + wrap_y*8, NULL, wrap_y, 8) - 400;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1556
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1557 if(progressive_score > 0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1558 interlaced_score = s->dsp.ildct_cmp[4](s, ptr_y , NULL, wrap_y*2, 8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1559 +s->dsp.ildct_cmp[4](s, ptr_y + wrap_y , NULL, wrap_y*2, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1560 if(progressive_score > interlaced_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1561 s->interlaced_dct=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1562
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1563 dct_offset= wrap_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1564 wrap_y<<=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1565 if (s->chroma_format == CHROMA_422)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1566 wrap_c<<=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1567 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1568 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1569 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1570
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1571 s->dsp.get_pixels(s->block[0], ptr_y , wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1572 s->dsp.get_pixels(s->block[1], ptr_y + 8, wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1573 s->dsp.get_pixels(s->block[2], ptr_y + dct_offset , wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1574 s->dsp.get_pixels(s->block[3], ptr_y + dct_offset + 8, wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1575
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1576 if(s->flags&CODEC_FLAG_GRAY){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1577 skip_dct[4]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1578 skip_dct[5]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1579 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1580 s->dsp.get_pixels(s->block[4], ptr_cb, wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1581 s->dsp.get_pixels(s->block[5], ptr_cr, wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1582 if(!s->chroma_y_shift){ /* 422 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1583 s->dsp.get_pixels(s->block[6], ptr_cb + (dct_offset>>1), wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1584 s->dsp.get_pixels(s->block[7], ptr_cr + (dct_offset>>1), wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1585 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1586 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1587 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1588 op_pixels_func (*op_pix)[4];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1589 qpel_mc_func (*op_qpix)[16];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1590 uint8_t *dest_y, *dest_cb, *dest_cr;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1591
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1592 dest_y = s->dest[0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1593 dest_cb = s->dest[1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1594 dest_cr = s->dest[2];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1595
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1596 if ((!s->no_rounding) || s->pict_type==FF_B_TYPE){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1597 op_pix = s->dsp.put_pixels_tab;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1598 op_qpix= s->dsp.put_qpel_pixels_tab;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1599 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1600 op_pix = s->dsp.put_no_rnd_pixels_tab;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1601 op_qpix= s->dsp.put_no_rnd_qpel_pixels_tab;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1602 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1603
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1604 if (s->mv_dir & MV_DIR_FORWARD) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1605 MPV_motion(s, dest_y, dest_cb, dest_cr, 0, s->last_picture.data, op_pix, op_qpix);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1606 op_pix = s->dsp.avg_pixels_tab;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1607 op_qpix= s->dsp.avg_qpel_pixels_tab;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1608 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1609 if (s->mv_dir & MV_DIR_BACKWARD) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1610 MPV_motion(s, dest_y, dest_cb, dest_cr, 1, s->next_picture.data, op_pix, op_qpix);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1611 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1612
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1613 if(s->flags&CODEC_FLAG_INTERLACED_DCT){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1614 int progressive_score, interlaced_score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1615
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1616 s->interlaced_dct=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1617 progressive_score= s->dsp.ildct_cmp[0](s, dest_y , ptr_y , wrap_y, 8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1618 +s->dsp.ildct_cmp[0](s, dest_y + wrap_y*8, ptr_y + wrap_y*8, wrap_y, 8) - 400;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1619
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1620 if(s->avctx->ildct_cmp == FF_CMP_VSSE) progressive_score -= 400;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1621
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1622 if(progressive_score>0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1623 interlaced_score = s->dsp.ildct_cmp[0](s, dest_y , ptr_y , wrap_y*2, 8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1624 +s->dsp.ildct_cmp[0](s, dest_y + wrap_y , ptr_y + wrap_y , wrap_y*2, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1625
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1626 if(progressive_score > interlaced_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1627 s->interlaced_dct=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1628
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1629 dct_offset= wrap_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1630 wrap_y<<=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1631 if (s->chroma_format == CHROMA_422)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1632 wrap_c<<=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1633 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1634 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1635 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1636
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1637 s->dsp.diff_pixels(s->block[0], ptr_y , dest_y , wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1638 s->dsp.diff_pixels(s->block[1], ptr_y + 8, dest_y + 8, wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1639 s->dsp.diff_pixels(s->block[2], ptr_y + dct_offset , dest_y + dct_offset , wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1640 s->dsp.diff_pixels(s->block[3], ptr_y + dct_offset + 8, dest_y + dct_offset + 8, wrap_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1641
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1642 if(s->flags&CODEC_FLAG_GRAY){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1643 skip_dct[4]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1644 skip_dct[5]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1645 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1646 s->dsp.diff_pixels(s->block[4], ptr_cb, dest_cb, wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1647 s->dsp.diff_pixels(s->block[5], ptr_cr, dest_cr, wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1648 if(!s->chroma_y_shift){ /* 422 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1649 s->dsp.diff_pixels(s->block[6], ptr_cb + (dct_offset>>1), dest_cb + (dct_offset>>1), wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1650 s->dsp.diff_pixels(s->block[7], ptr_cr + (dct_offset>>1), dest_cr + (dct_offset>>1), wrap_c);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1651 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1652 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1653 /* pre quantization */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1654 if(s->current_picture.mc_mb_var[s->mb_stride*mb_y+ mb_x]<2*s->qscale*s->qscale){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1655 //FIXME optimize
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1656 if(s->dsp.sad[1](NULL, ptr_y , dest_y , wrap_y, 8) < 20*s->qscale) skip_dct[0]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1657 if(s->dsp.sad[1](NULL, ptr_y + 8, dest_y + 8, wrap_y, 8) < 20*s->qscale) skip_dct[1]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1658 if(s->dsp.sad[1](NULL, ptr_y +dct_offset , dest_y +dct_offset , wrap_y, 8) < 20*s->qscale) skip_dct[2]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1659 if(s->dsp.sad[1](NULL, ptr_y +dct_offset+ 8, dest_y +dct_offset+ 8, wrap_y, 8) < 20*s->qscale) skip_dct[3]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1660 if(s->dsp.sad[1](NULL, ptr_cb , dest_cb , wrap_c, 8) < 20*s->qscale) skip_dct[4]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1661 if(s->dsp.sad[1](NULL, ptr_cr , dest_cr , wrap_c, 8) < 20*s->qscale) skip_dct[5]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1662 if(!s->chroma_y_shift){ /* 422 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1663 if(s->dsp.sad[1](NULL, ptr_cb +(dct_offset>>1), dest_cb +(dct_offset>>1), wrap_c, 8) < 20*s->qscale) skip_dct[6]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1664 if(s->dsp.sad[1](NULL, ptr_cr +(dct_offset>>1), dest_cr +(dct_offset>>1), wrap_c, 8) < 20*s->qscale) skip_dct[7]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1665 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1666 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1667 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1668
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1669 if(s->avctx->quantizer_noise_shaping){
5909
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1670 if(!skip_dct[0]) get_visual_weight(weight[0], ptr_y , wrap_y);
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1671 if(!skip_dct[1]) get_visual_weight(weight[1], ptr_y + 8, wrap_y);
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1672 if(!skip_dct[2]) get_visual_weight(weight[2], ptr_y + dct_offset , wrap_y);
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1673 if(!skip_dct[3]) get_visual_weight(weight[3], ptr_y + dct_offset + 8, wrap_y);
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1674 if(!skip_dct[4]) get_visual_weight(weight[4], ptr_cb , wrap_c);
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1675 if(!skip_dct[5]) get_visual_weight(weight[5], ptr_cr , wrap_c);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1676 if(!s->chroma_y_shift){ /* 422 */
5909
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1677 if(!skip_dct[6]) get_visual_weight(weight[6], ptr_cb + (dct_offset>>1), wrap_c);
4e3c522ca492 cosmetics: Fix vissual --> visual typo.
diego
parents: 5789
diff changeset
1678 if(!skip_dct[7]) get_visual_weight(weight[7], ptr_cr + (dct_offset>>1), wrap_c);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1679 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1680 memcpy(orig[0], s->block[0], sizeof(DCTELEM)*64*mb_block_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1681 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1682
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1683 /* DCT & quantize */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1684 assert(s->out_format!=FMT_MJPEG || s->qscale==8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1685 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1686 for(i=0;i<mb_block_count;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1687 if(!skip_dct[i]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1688 int overflow;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1689 s->block_last_index[i] = s->dct_quantize(s, s->block[i], i, s->qscale, &overflow);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1690 // FIXME we could decide to change to quantizer instead of clipping
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1691 // JS: I don't think that would be a good idea it could lower quality instead
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1692 // of improve it. Just INTRADC clipping deserves changes in quantizer
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1693 if (overflow) clip_coeffs(s, s->block[i], s->block_last_index[i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1694 }else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1695 s->block_last_index[i]= -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1696 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1697 if(s->avctx->quantizer_noise_shaping){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1698 for(i=0;i<mb_block_count;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1699 if(!skip_dct[i]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1700 s->block_last_index[i] = dct_quantize_refine(s, s->block[i], weight[i], orig[i], i, s->qscale);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1701 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1702 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1703 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1704
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1705 if(s->luma_elim_threshold && !s->mb_intra)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1706 for(i=0; i<4; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1707 dct_single_coeff_elimination(s, i, s->luma_elim_threshold);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1708 if(s->chroma_elim_threshold && !s->mb_intra)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1709 for(i=4; i<mb_block_count; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1710 dct_single_coeff_elimination(s, i, s->chroma_elim_threshold);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1711
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1712 if(s->flags & CODEC_FLAG_CBP_RD){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1713 for(i=0;i<mb_block_count;i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1714 if(s->block_last_index[i] == -1)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1715 s->coded_score[i]= INT_MAX/256;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1716 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1717 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1718 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1719
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1720 if((s->flags&CODEC_FLAG_GRAY) && s->mb_intra){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1721 s->block_last_index[4]=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1722 s->block_last_index[5]= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1723 s->block[4][0]=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1724 s->block[5][0]= (1024 + s->c_dc_scale/2)/ s->c_dc_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1725 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1726
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1727 //non c quantize code returns incorrect block_last_index FIXME
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1728 if(s->alternate_scan && s->dct_quantize != dct_quantize_c){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1729 for(i=0; i<mb_block_count; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1730 int j;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1731 if(s->block_last_index[i]>0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1732 for(j=63; j>0; j--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1733 if(s->block[i][ s->intra_scantable.permutated[j] ]) break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1734 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1735 s->block_last_index[i]= j;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1736 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1737 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1738 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1739
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1740 /* huffman encode */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1741 switch(s->codec_id){ //FIXME funct ptr could be slightly faster
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1742 case CODEC_ID_MPEG1VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1743 case CODEC_ID_MPEG2VIDEO:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
1744 if (CONFIG_MPEG1VIDEO_ENCODER || CONFIG_MPEG2VIDEO_ENCODER)
5209
2ac10c5e5885 cosmetics: indentation
aurel
parents: 5208
diff changeset
1745 mpeg1_encode_mb(s, s->block, motion_x, motion_y);
2ac10c5e5885 cosmetics: indentation
aurel
parents: 5208
diff changeset
1746 break;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1747 case CODEC_ID_MPEG4:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
1748 if (CONFIG_MPEG4_ENCODER)
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
1749 mpeg4_encode_mb(s, s->block, motion_x, motion_y);
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
1750 break;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1751 case CODEC_ID_MSMPEG4V2:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1752 case CODEC_ID_MSMPEG4V3:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1753 case CODEC_ID_WMV1:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
1754 if (CONFIG_MSMPEG4_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1755 msmpeg4_encode_mb(s, s->block, motion_x, motion_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1756 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1757 case CODEC_ID_WMV2:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
1758 if (CONFIG_WMV2_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1759 ff_wmv2_encode_mb(s, s->block, motion_x, motion_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1760 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1761 case CODEC_ID_H261:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
1762 if (CONFIG_H261_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1763 ff_h261_encode_mb(s, s->block, motion_x, motion_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1764 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1765 case CODEC_ID_H263:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1766 case CODEC_ID_H263P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1767 case CODEC_ID_FLV1:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1768 case CODEC_ID_RV10:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1769 case CODEC_ID_RV20:
10034
ea2422aa4635 Do not check for both CONFIG_H263_ENCODER and CONFIG_FLV_ENCODER.
diego
parents: 10032
diff changeset
1770 if (CONFIG_H263_ENCODER)
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
1771 h263_encode_mb(s, s->block, motion_x, motion_y);
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
1772 break;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1773 case CODEC_ID_MJPEG:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
1774 if (CONFIG_MJPEG_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1775 ff_mjpeg_encode_mb(s, s->block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1776 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1777 default:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1778 assert(0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1779 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1780 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1781
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1782 static av_always_inline void encode_mb(MpegEncContext *s, int motion_x, int motion_y)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1783 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1784 if (s->chroma_format == CHROMA_420) encode_mb_internal(s, motion_x, motion_y, 8, 6);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1785 else encode_mb_internal(s, motion_x, motion_y, 16, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1786 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1787
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1788 static inline void copy_context_before_encode(MpegEncContext *d, MpegEncContext *s, int type){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1789 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1790
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1791 memcpy(d->last_mv, s->last_mv, 2*2*2*sizeof(int)); //FIXME is memcpy faster then a loop?
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1792
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1793 /* mpeg1 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1794 d->mb_skip_run= s->mb_skip_run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1795 for(i=0; i<3; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1796 d->last_dc[i]= s->last_dc[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1797
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1798 /* statistics */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1799 d->mv_bits= s->mv_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1800 d->i_tex_bits= s->i_tex_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1801 d->p_tex_bits= s->p_tex_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1802 d->i_count= s->i_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1803 d->f_count= s->f_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1804 d->b_count= s->b_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1805 d->skip_count= s->skip_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1806 d->misc_bits= s->misc_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1807 d->last_bits= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1808
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1809 d->mb_skipped= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1810 d->qscale= s->qscale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1811 d->dquant= s->dquant;
6074
036d93ae3095 Fix encoding wmv1 with -mbd 1.
michael
parents: 5909
diff changeset
1812
036d93ae3095 Fix encoding wmv1 with -mbd 1.
michael
parents: 5909
diff changeset
1813 d->esc3_level_length= s->esc3_level_length;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1814 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1815
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1816 static inline void copy_context_after_encode(MpegEncContext *d, MpegEncContext *s, int type){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1817 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1818
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1819 memcpy(d->mv, s->mv, 2*4*2*sizeof(int));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1820 memcpy(d->last_mv, s->last_mv, 2*2*2*sizeof(int)); //FIXME is memcpy faster then a loop?
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1821
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1822 /* mpeg1 */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1823 d->mb_skip_run= s->mb_skip_run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1824 for(i=0; i<3; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1825 d->last_dc[i]= s->last_dc[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1826
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1827 /* statistics */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1828 d->mv_bits= s->mv_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1829 d->i_tex_bits= s->i_tex_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1830 d->p_tex_bits= s->p_tex_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1831 d->i_count= s->i_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1832 d->f_count= s->f_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1833 d->b_count= s->b_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1834 d->skip_count= s->skip_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1835 d->misc_bits= s->misc_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1836
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1837 d->mb_intra= s->mb_intra;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1838 d->mb_skipped= s->mb_skipped;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1839 d->mv_type= s->mv_type;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1840 d->mv_dir= s->mv_dir;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1841 d->pb= s->pb;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1842 if(s->data_partitioning){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1843 d->pb2= s->pb2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1844 d->tex_pb= s->tex_pb;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1845 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1846 d->block= s->block;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1847 for(i=0; i<8; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1848 d->block_last_index[i]= s->block_last_index[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1849 d->interlaced_dct= s->interlaced_dct;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1850 d->qscale= s->qscale;
6074
036d93ae3095 Fix encoding wmv1 with -mbd 1.
michael
parents: 5909
diff changeset
1851
036d93ae3095 Fix encoding wmv1 with -mbd 1.
michael
parents: 5909
diff changeset
1852 d->esc3_level_length= s->esc3_level_length;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1853 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1854
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1855 static inline void encode_mb_hq(MpegEncContext *s, MpegEncContext *backup, MpegEncContext *best, int type,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1856 PutBitContext pb[2], PutBitContext pb2[2], PutBitContext tex_pb[2],
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1857 int *dmin, int *next_block, int motion_x, int motion_y)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1858 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1859 int score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1860 uint8_t *dest_backup[3];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1861
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1862 copy_context_before_encode(s, backup, type);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1863
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1864 s->block= s->blocks[*next_block];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1865 s->pb= pb[*next_block];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1866 if(s->data_partitioning){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1867 s->pb2 = pb2 [*next_block];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1868 s->tex_pb= tex_pb[*next_block];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1869 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1870
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1871 if(*next_block){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1872 memcpy(dest_backup, s->dest, sizeof(s->dest));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1873 s->dest[0] = s->rd_scratchpad;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1874 s->dest[1] = s->rd_scratchpad + 16*s->linesize;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1875 s->dest[2] = s->rd_scratchpad + 16*s->linesize + 8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1876 assert(s->linesize >= 32); //FIXME
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1877 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1878
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1879 encode_mb(s, motion_x, motion_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1880
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1881 score= put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1882 if(s->data_partitioning){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1883 score+= put_bits_count(&s->pb2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1884 score+= put_bits_count(&s->tex_pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1885 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1886
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1887 if(s->avctx->mb_decision == FF_MB_DECISION_RD){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1888 MPV_decode_mb(s, s->block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1889
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1890 score *= s->lambda2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1891 score += sse_mb(s) << FF_LAMBDA_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1892 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1893
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1894 if(*next_block){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1895 memcpy(s->dest, dest_backup, sizeof(s->dest));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1896 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1897
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1898 if(score<*dmin){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1899 *dmin= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1900 *next_block^=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1901
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1902 copy_context_after_encode(best, s, type);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1903 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1904 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1905
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1906 static int sse(MpegEncContext *s, uint8_t *src1, uint8_t *src2, int w, int h, int stride){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1907 uint32_t *sq = ff_squareTbl + 256;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1908 int acc=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1909 int x,y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1910
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1911 if(w==16 && h==16)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1912 return s->dsp.sse[0](NULL, src1, src2, stride, 16);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1913 else if(w==8 && h==8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1914 return s->dsp.sse[1](NULL, src1, src2, stride, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1915
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1916 for(y=0; y<h; y++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1917 for(x=0; x<w; x++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1918 acc+= sq[src1[x + y*stride] - src2[x + y*stride]];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1919 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1920 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1921
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1922 assert(acc>=0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1923
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1924 return acc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1925 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1926
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1927 static int sse_mb(MpegEncContext *s){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1928 int w= 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1929 int h= 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1930
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1931 if(s->mb_x*16 + 16 > s->width ) w= s->width - s->mb_x*16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1932 if(s->mb_y*16 + 16 > s->height) h= s->height- s->mb_y*16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1933
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1934 if(w==16 && h==16)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1935 if(s->avctx->mb_cmp == FF_CMP_NSSE){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1936 return s->dsp.nsse[0](s, s->new_picture.data[0] + s->mb_x*16 + s->mb_y*s->linesize*16, s->dest[0], s->linesize, 16)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1937 +s->dsp.nsse[1](s, s->new_picture.data[1] + s->mb_x*8 + s->mb_y*s->uvlinesize*8,s->dest[1], s->uvlinesize, 8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1938 +s->dsp.nsse[1](s, s->new_picture.data[2] + s->mb_x*8 + s->mb_y*s->uvlinesize*8,s->dest[2], s->uvlinesize, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1939 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1940 return s->dsp.sse[0](NULL, s->new_picture.data[0] + s->mb_x*16 + s->mb_y*s->linesize*16, s->dest[0], s->linesize, 16)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1941 +s->dsp.sse[1](NULL, s->new_picture.data[1] + s->mb_x*8 + s->mb_y*s->uvlinesize*8,s->dest[1], s->uvlinesize, 8)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1942 +s->dsp.sse[1](NULL, s->new_picture.data[2] + s->mb_x*8 + s->mb_y*s->uvlinesize*8,s->dest[2], s->uvlinesize, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1943 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1944 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1945 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)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1946 +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)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1947 +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);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1948 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1949
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1950 static int pre_estimate_motion_thread(AVCodecContext *c, void *arg){
8129
a9734fe0811e Making it easier to send arbitrary structures as work orders to MT workers
romansh
parents: 8117
diff changeset
1951 MpegEncContext *s= *(void**)arg;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1952
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1953
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1954 s->me.pre_pass=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1955 s->me.dia_size= s->avctx->pre_dia_size;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1956 s->first_slice_line=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1957 for(s->mb_y= s->end_mb_y-1; s->mb_y >= s->start_mb_y; s->mb_y--) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1958 for(s->mb_x=s->mb_width-1; s->mb_x >=0 ;s->mb_x--) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1959 ff_pre_estimate_p_frame_motion(s, s->mb_x, s->mb_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1960 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1961 s->first_slice_line=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1962 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1963
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1964 s->me.pre_pass=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1965
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1966 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1967 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1968
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1969 static int estimate_motion_thread(AVCodecContext *c, void *arg){
8129
a9734fe0811e Making it easier to send arbitrary structures as work orders to MT workers
romansh
parents: 8117
diff changeset
1970 MpegEncContext *s= *(void**)arg;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1971
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1972 ff_check_alignment();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1973
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1974 s->me.dia_size= s->avctx->dia_size;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1975 s->first_slice_line=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1976 for(s->mb_y= s->start_mb_y; s->mb_y < s->end_mb_y; s->mb_y++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1977 s->mb_x=0; //for block init below
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1978 ff_init_block_index(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1979 for(s->mb_x=0; s->mb_x < s->mb_width; s->mb_x++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1980 s->block_index[0]+=2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1981 s->block_index[1]+=2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1982 s->block_index[2]+=2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1983 s->block_index[3]+=2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1984
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1985 /* compute motion vector & mb_type and store in context */
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
1986 if(s->pict_type==FF_B_TYPE)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1987 ff_estimate_b_frame_motion(s, s->mb_x, s->mb_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1988 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1989 ff_estimate_p_frame_motion(s, s->mb_x, s->mb_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1990 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1991 s->first_slice_line=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1992 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1993 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1994 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1995
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1996 static int mb_var_thread(AVCodecContext *c, void *arg){
8129
a9734fe0811e Making it easier to send arbitrary structures as work orders to MT workers
romansh
parents: 8117
diff changeset
1997 MpegEncContext *s= *(void**)arg;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1998 int mb_x, mb_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
1999
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2000 ff_check_alignment();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2001
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2002 for(mb_y=s->start_mb_y; mb_y < s->end_mb_y; mb_y++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2003 for(mb_x=0; mb_x < s->mb_width; mb_x++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2004 int xx = mb_x * 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2005 int yy = mb_y * 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2006 uint8_t *pix = s->new_picture.data[0] + (yy * s->linesize) + xx;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2007 int varc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2008 int sum = s->dsp.pix_sum(pix, s->linesize);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2009
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2010 varc = (s->dsp.pix_norm1(pix, s->linesize) - (((unsigned)(sum*sum))>>8) + 500 + 128)>>8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2011
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2012 s->current_picture.mb_var [s->mb_stride * mb_y + mb_x] = varc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2013 s->current_picture.mb_mean[s->mb_stride * mb_y + mb_x] = (sum+128)>>8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2014 s->me.mb_var_sum_temp += varc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2015 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2016 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2017 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2018 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2019
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2020 static void write_slice_end(MpegEncContext *s){
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2021 if(CONFIG_MPEG4_ENCODER && s->codec_id==CODEC_ID_MPEG4){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2022 if(s->partitioned_frame){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2023 ff_mpeg4_merge_partitions(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2024 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2025
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2026 ff_mpeg4_stuffing(&s->pb);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2027 }else if(CONFIG_MJPEG_ENCODER && s->out_format == FMT_MJPEG){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2028 ff_mjpeg_encode_stuffing(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2029 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2030
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2031 align_put_bits(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2032 flush_put_bits(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2033
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2034 if((s->flags&CODEC_FLAG_PASS1) && !s->partitioned_frame)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2035 s->misc_bits+= get_bits_diff(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2036 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2037
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2038 static int encode_thread(AVCodecContext *c, void *arg){
8129
a9734fe0811e Making it easier to send arbitrary structures as work orders to MT workers
romansh
parents: 8117
diff changeset
2039 MpegEncContext *s= *(void**)arg;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2040 int mb_x, mb_y, pdif = 0;
7520
b8ad314749ff fix 4:2:2 psnr
bcoudurier
parents: 7502
diff changeset
2041 int chr_h= 16>>s->chroma_y_shift;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2042 int i, j;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2043 MpegEncContext best_s, backup_s;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2044 uint8_t bit_buf[2][MAX_MB_BYTES];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2045 uint8_t bit_buf2[2][MAX_MB_BYTES];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2046 uint8_t bit_buf_tex[2][MAX_MB_BYTES];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2047 PutBitContext pb[2], pb2[2], tex_pb[2];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2048 //printf("%d->%d\n", s->resync_mb_y, s->end_mb_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2049
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2050 ff_check_alignment();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2051
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2052 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2053 init_put_bits(&pb [i], bit_buf [i], MAX_MB_BYTES);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2054 init_put_bits(&pb2 [i], bit_buf2 [i], MAX_MB_BYTES);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2055 init_put_bits(&tex_pb[i], bit_buf_tex[i], MAX_MB_BYTES);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2056 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2057
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2058 s->last_bits= put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2059 s->mv_bits=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2060 s->misc_bits=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2061 s->i_tex_bits=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2062 s->p_tex_bits=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2063 s->i_count=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2064 s->f_count=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2065 s->b_count=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2066 s->skip_count=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2067
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2068 for(i=0; i<3; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2069 /* init last dc values */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2070 /* note: quant matrix value (8) is implied here */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2071 s->last_dc[i] = 128 << s->intra_dc_precision;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2072
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2073 s->current_picture.error[i] = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2074 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2075 s->mb_skip_run = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2076 memset(s->last_mv, 0, sizeof(s->last_mv));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2077
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2078 s->last_mv_dir = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2079
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2080 switch(s->codec_id){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2081 case CODEC_ID_H263:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2082 case CODEC_ID_H263P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2083 case CODEC_ID_FLV1:
10034
ea2422aa4635 Do not check for both CONFIG_H263_ENCODER and CONFIG_FLV_ENCODER.
diego
parents: 10032
diff changeset
2084 if (CONFIG_H263_ENCODER)
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2085 s->gob_index = ff_h263_get_gob_height(s);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2086 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2087 case CODEC_ID_MPEG4:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2088 if(CONFIG_MPEG4_ENCODER && s->partitioned_frame)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2089 ff_mpeg4_init_partitions(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2090 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2091 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2092
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2093 s->resync_mb_x=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2094 s->resync_mb_y=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2095 s->first_slice_line = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2096 s->ptr_lastgob = s->pb.buf;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2097 for(mb_y= s->start_mb_y; mb_y < s->end_mb_y; mb_y++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2098 // printf("row %d at %X\n", s->mb_y, (int)s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2099 s->mb_x=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2100 s->mb_y= mb_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2101
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2102 ff_set_qscale(s, s->qscale);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2103 ff_init_block_index(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2104
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2105 for(mb_x=0; mb_x < s->mb_width; mb_x++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2106 int xy= mb_y*s->mb_stride + mb_x; // removed const, H261 needs to adjust this
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2107 int mb_type= s->mb_type[xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2108 // int d;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2109 int dmin= INT_MAX;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2110 int dir;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2111
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2112 if(s->pb.buf_end - s->pb.buf - (put_bits_count(&s->pb)>>3) < MAX_MB_BYTES){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2113 av_log(s->avctx, AV_LOG_ERROR, "encoded frame too large\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2114 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2115 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2116 if(s->data_partitioning){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2117 if( s->pb2 .buf_end - s->pb2 .buf - (put_bits_count(&s-> pb2)>>3) < MAX_MB_BYTES
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2118 || s->tex_pb.buf_end - s->tex_pb.buf - (put_bits_count(&s->tex_pb )>>3) < MAX_MB_BYTES){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2119 av_log(s->avctx, AV_LOG_ERROR, "encoded frame too large\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2120 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2121 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2122 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2123
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2124 s->mb_x = mb_x;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2125 s->mb_y = mb_y; // moved into loop, can get changed by H.261
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2126 ff_update_block_index(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2127
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2128 if(CONFIG_H261_ENCODER && s->codec_id == CODEC_ID_H261){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2129 ff_h261_reorder_mb_index(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2130 xy= s->mb_y*s->mb_stride + s->mb_x;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2131 mb_type= s->mb_type[xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2132 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2133
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2134 /* write gob / video packet header */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2135 if(s->rtp_mode){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2136 int current_packet_size, is_gob_start;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2137
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2138 current_packet_size= ((put_bits_count(&s->pb)+7)>>3) - (s->ptr_lastgob - s->pb.buf);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2139
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2140 is_gob_start= s->avctx->rtp_payload_size && current_packet_size >= s->avctx->rtp_payload_size && mb_y + mb_x>0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2141
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2142 if(s->start_mb_y == mb_y && mb_y > 0 && mb_x==0) is_gob_start=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2143
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2144 switch(s->codec_id){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2145 case CODEC_ID_H263:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2146 case CODEC_ID_H263P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2147 if(!s->h263_slice_structured)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2148 if(s->mb_x || s->mb_y%s->gob_index) is_gob_start=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2149 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2150 case CODEC_ID_MPEG2VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2151 if(s->mb_x==0 && s->mb_y!=0) is_gob_start=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2152 case CODEC_ID_MPEG1VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2153 if(s->mb_skip_run) is_gob_start=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2154 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2155 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2156
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2157 if(is_gob_start){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2158 if(s->start_mb_y != mb_y || mb_x!=0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2159 write_slice_end(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2160
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2161 if(CONFIG_MPEG4_ENCODER && s->codec_id==CODEC_ID_MPEG4 && s->partitioned_frame){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2162 ff_mpeg4_init_partitions(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2163 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2164 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2165
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2166 assert((put_bits_count(&s->pb)&7) == 0);
9431
932543edc1d2 Rename pbBufPtr() to put_bits_ptr().
stefano
parents: 9083
diff changeset
2167 current_packet_size= put_bits_ptr(&s->pb) - s->ptr_lastgob;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2168
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2169 if(s->avctx->error_rate && s->resync_mb_x + s->resync_mb_y > 0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2170 int r= put_bits_count(&s->pb)/8 + s->picture_number + 16 + s->mb_x + s->mb_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2171 int d= 100 / s->avctx->error_rate;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2172 if(r % d == 0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2173 current_packet_size=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2174 #ifndef ALT_BITSTREAM_WRITER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2175 s->pb.buf_ptr= s->ptr_lastgob;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2176 #endif
9431
932543edc1d2 Rename pbBufPtr() to put_bits_ptr().
stefano
parents: 9083
diff changeset
2177 assert(put_bits_ptr(&s->pb) == s->ptr_lastgob);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2178 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2179 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2180
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2181 if (s->avctx->rtp_callback){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2182 int number_mb = (mb_y - s->resync_mb_y)*s->mb_width + mb_x - s->resync_mb_x;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2183 s->avctx->rtp_callback(s->avctx, s->ptr_lastgob, current_packet_size, number_mb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2184 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2185
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2186 switch(s->codec_id){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2187 case CODEC_ID_MPEG4:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2188 if (CONFIG_MPEG4_ENCODER) {
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2189 ff_mpeg4_encode_video_packet_header(s);
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2190 ff_mpeg4_clean_buffers(s);
5277
7b3fcb7c61ce Avoid linking with h263.c functions when the relevant codecs
aurel
parents: 5273
diff changeset
2191 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2192 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2193 case CODEC_ID_MPEG1VIDEO:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2194 case CODEC_ID_MPEG2VIDEO:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2195 if (CONFIG_MPEG1VIDEO_ENCODER || CONFIG_MPEG2VIDEO_ENCODER) {
5209
2ac10c5e5885 cosmetics: indentation
aurel
parents: 5208
diff changeset
2196 ff_mpeg1_encode_slice_header(s);
2ac10c5e5885 cosmetics: indentation
aurel
parents: 5208
diff changeset
2197 ff_mpeg1_clean_buffers(s);
5208
430f60820b21 split mpeg12 encoder out of mpeg12.c
aurel
parents: 5204
diff changeset
2198 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2199 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2200 case CODEC_ID_H263:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2201 case CODEC_ID_H263P:
10032
29c9829a9684 Do not check for both CONFIG_H263_ENCODER and CONFIG_H263P_ENCODER.
diego
parents: 10029
diff changeset
2202 if (CONFIG_H263_ENCODER)
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2203 h263_encode_gob_header(s, mb_y);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2204 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2205 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2206
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2207 if(s->flags&CODEC_FLAG_PASS1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2208 int bits= put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2209 s->misc_bits+= bits - s->last_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2210 s->last_bits= bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2211 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2212
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2213 s->ptr_lastgob += current_packet_size;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2214 s->first_slice_line=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2215 s->resync_mb_x=mb_x;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2216 s->resync_mb_y=mb_y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2217 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2218 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2219
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2220 if( (s->resync_mb_x == s->mb_x)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2221 && s->resync_mb_y+1 == s->mb_y){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2222 s->first_slice_line=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2223 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2224
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2225 s->mb_skipped=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2226 s->dquant=0; //only for QP_RD
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2227
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2228 if(mb_type & (mb_type-1) || (s->flags & CODEC_FLAG_QP_RD)){ // more than 1 MB type possible or CODEC_FLAG_QP_RD
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2229 int next_block=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2230 int pb_bits_count, pb2_bits_count, tex_pb_bits_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2231
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2232 copy_context_before_encode(&backup_s, s, -1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2233 backup_s.pb= s->pb;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2234 best_s.data_partitioning= s->data_partitioning;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2235 best_s.partitioned_frame= s->partitioned_frame;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2236 if(s->data_partitioning){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2237 backup_s.pb2= s->pb2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2238 backup_s.tex_pb= s->tex_pb;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2239 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2240
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2241 if(mb_type&CANDIDATE_MB_TYPE_INTER){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2242 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2243 s->mv_type = MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2244 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2245 s->mv[0][0][0] = s->p_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2246 s->mv[0][0][1] = s->p_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2247 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_INTER, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2248 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2249 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2250 if(mb_type&CANDIDATE_MB_TYPE_INTER_I){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2251 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2252 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2253 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2254 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2255 j= s->field_select[0][i] = s->p_field_select_table[i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2256 s->mv[0][i][0] = s->p_field_mv_table[i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2257 s->mv[0][i][1] = s->p_field_mv_table[i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2258 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2259 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_INTER_I, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2260 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2261 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2262 if(mb_type&CANDIDATE_MB_TYPE_SKIPPED){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2263 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2264 s->mv_type = MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2265 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2266 s->mv[0][0][0] = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2267 s->mv[0][0][1] = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2268 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_SKIPPED, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2269 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2270 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2271 if(mb_type&CANDIDATE_MB_TYPE_INTER4V){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2272 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2273 s->mv_type = MV_TYPE_8X8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2274 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2275 for(i=0; i<4; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2276 s->mv[0][i][0] = s->current_picture.motion_val[0][s->block_index[i]][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2277 s->mv[0][i][1] = s->current_picture.motion_val[0][s->block_index[i]][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2278 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2279 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_INTER4V, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2280 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2281 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2282 if(mb_type&CANDIDATE_MB_TYPE_FORWARD){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2283 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2284 s->mv_type = MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2285 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2286 s->mv[0][0][0] = s->b_forw_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2287 s->mv[0][0][1] = s->b_forw_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2288 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_FORWARD, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2289 &dmin, &next_block, s->mv[0][0][0], s->mv[0][0][1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2290 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2291 if(mb_type&CANDIDATE_MB_TYPE_BACKWARD){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2292 s->mv_dir = MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2293 s->mv_type = MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2294 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2295 s->mv[1][0][0] = s->b_back_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2296 s->mv[1][0][1] = s->b_back_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2297 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_BACKWARD, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2298 &dmin, &next_block, s->mv[1][0][0], s->mv[1][0][1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2299 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2300 if(mb_type&CANDIDATE_MB_TYPE_BIDIR){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2301 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2302 s->mv_type = MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2303 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2304 s->mv[0][0][0] = s->b_bidir_forw_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2305 s->mv[0][0][1] = s->b_bidir_forw_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2306 s->mv[1][0][0] = s->b_bidir_back_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2307 s->mv[1][0][1] = s->b_bidir_back_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2308 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_BIDIR, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2309 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2310 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2311 if(mb_type&CANDIDATE_MB_TYPE_FORWARD_I){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2312 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2313 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2314 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2315 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2316 j= s->field_select[0][i] = s->b_field_select_table[0][i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2317 s->mv[0][i][0] = s->b_field_mv_table[0][i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2318 s->mv[0][i][1] = s->b_field_mv_table[0][i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2319 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2320 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_FORWARD_I, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2321 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2322 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2323 if(mb_type&CANDIDATE_MB_TYPE_BACKWARD_I){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2324 s->mv_dir = MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2325 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2326 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2327 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2328 j= s->field_select[1][i] = s->b_field_select_table[1][i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2329 s->mv[1][i][0] = s->b_field_mv_table[1][i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2330 s->mv[1][i][1] = s->b_field_mv_table[1][i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2331 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2332 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_BACKWARD_I, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2333 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2334 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2335 if(mb_type&CANDIDATE_MB_TYPE_BIDIR_I){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2336 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2337 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2338 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2339 for(dir=0; dir<2; dir++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2340 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2341 j= s->field_select[dir][i] = s->b_field_select_table[dir][i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2342 s->mv[dir][i][0] = s->b_field_mv_table[dir][i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2343 s->mv[dir][i][1] = s->b_field_mv_table[dir][i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2344 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2345 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2346 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_BIDIR_I, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2347 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2348 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2349 if(mb_type&CANDIDATE_MB_TYPE_INTRA){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2350 s->mv_dir = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2351 s->mv_type = MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2352 s->mb_intra= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2353 s->mv[0][0][0] = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2354 s->mv[0][0][1] = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2355 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_INTRA, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2356 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2357 if(s->h263_pred || s->h263_aic){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2358 if(best_s.mb_intra)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2359 s->mbintra_table[mb_x + mb_y*s->mb_stride]=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2360 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2361 ff_clean_intra_table_entries(s); //old mode?
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2362 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2363 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2364
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2365 if((s->flags & CODEC_FLAG_QP_RD) && dmin < INT_MAX){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2366 if(best_s.mv_type==MV_TYPE_16X16){ //FIXME move 4mv after QPRD
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2367 const int last_qp= backup_s.qscale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2368 int qpi, qp, dc[6];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2369 DCTELEM ac[6][16];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2370 const int mvdir= (best_s.mv_dir&MV_DIR_BACKWARD) ? 1 : 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2371 static const int dquant_tab[4]={-1,1,-2,2};
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2372
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2373 assert(backup_s.dquant == 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2374
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2375 //FIXME intra
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2376 s->mv_dir= best_s.mv_dir;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2377 s->mv_type = MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2378 s->mb_intra= best_s.mb_intra;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2379 s->mv[0][0][0] = best_s.mv[0][0][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2380 s->mv[0][0][1] = best_s.mv[0][0][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2381 s->mv[1][0][0] = best_s.mv[1][0][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2382 s->mv[1][0][1] = best_s.mv[1][0][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2383
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2384 qpi = s->pict_type == FF_B_TYPE ? 2 : 0;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2385 for(; qpi<4; qpi++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2386 int dquant= dquant_tab[qpi];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2387 qp= last_qp + dquant;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2388 if(qp < s->avctx->qmin || qp > s->avctx->qmax)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2389 continue;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2390 backup_s.dquant= dquant;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2391 if(s->mb_intra && s->dc_val[0]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2392 for(i=0; i<6; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2393 dc[i]= s->dc_val[0][ s->block_index[i] ];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2394 memcpy(ac[i], s->ac_val[0][s->block_index[i]], sizeof(DCTELEM)*16);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2395 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2396 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2397
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2398 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_INTER /* wrong but unused */, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2399 &dmin, &next_block, s->mv[mvdir][0][0], s->mv[mvdir][0][1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2400 if(best_s.qscale != qp){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2401 if(s->mb_intra && s->dc_val[0]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2402 for(i=0; i<6; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2403 s->dc_val[0][ s->block_index[i] ]= dc[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2404 memcpy(s->ac_val[0][s->block_index[i]], ac[i], sizeof(DCTELEM)*16);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2405 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2406 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2407 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2408 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2409 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2410 }
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2411 if(CONFIG_MPEG4_ENCODER && mb_type&CANDIDATE_MB_TYPE_DIRECT){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2412 int mx= s->b_direct_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2413 int my= s->b_direct_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2414
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2415 backup_s.dquant = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2416 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2417 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2418 ff_mpeg4_set_direct_mv(s, mx, my);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2419 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_DIRECT, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2420 &dmin, &next_block, mx, my);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2421 }
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2422 if(CONFIG_MPEG4_ENCODER && mb_type&CANDIDATE_MB_TYPE_DIRECT0){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2423 backup_s.dquant = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2424 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD | MV_DIRECT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2425 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2426 ff_mpeg4_set_direct_mv(s, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2427 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_DIRECT, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2428 &dmin, &next_block, 0, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2429 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2430 if(!best_s.mb_intra && s->flags2&CODEC_FLAG2_SKIP_RD){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2431 int coded=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2432 for(i=0; i<6; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2433 coded |= s->block_last_index[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2434 if(coded){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2435 int mx,my;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2436 memcpy(s->mv, best_s.mv, sizeof(s->mv));
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2437 if(CONFIG_MPEG4_ENCODER && best_s.mv_dir & MV_DIRECT){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2438 mx=my=0; //FIXME find the one we actually used
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2439 ff_mpeg4_set_direct_mv(s, mx, my);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2440 }else if(best_s.mv_dir&MV_DIR_BACKWARD){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2441 mx= s->mv[1][0][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2442 my= s->mv[1][0][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2443 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2444 mx= s->mv[0][0][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2445 my= s->mv[0][0][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2446 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2447
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2448 s->mv_dir= best_s.mv_dir;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2449 s->mv_type = best_s.mv_type;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2450 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2451 /* s->mv[0][0][0] = best_s.mv[0][0][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2452 s->mv[0][0][1] = best_s.mv[0][0][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2453 s->mv[1][0][0] = best_s.mv[1][0][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2454 s->mv[1][0][1] = best_s.mv[1][0][1];*/
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2455 backup_s.dquant= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2456 s->skipdct=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2457 encode_mb_hq(s, &backup_s, &best_s, CANDIDATE_MB_TYPE_INTER /* wrong but unused */, pb, pb2, tex_pb,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2458 &dmin, &next_block, mx, my);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2459 s->skipdct=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2460 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2461 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2462
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2463 s->current_picture.qscale_table[xy]= best_s.qscale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2464
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2465 copy_context_after_encode(s, &best_s, -1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2466
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2467 pb_bits_count= put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2468 flush_put_bits(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2469 ff_copy_bits(&backup_s.pb, bit_buf[next_block^1], pb_bits_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2470 s->pb= backup_s.pb;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2471
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2472 if(s->data_partitioning){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2473 pb2_bits_count= put_bits_count(&s->pb2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2474 flush_put_bits(&s->pb2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2475 ff_copy_bits(&backup_s.pb2, bit_buf2[next_block^1], pb2_bits_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2476 s->pb2= backup_s.pb2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2477
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2478 tex_pb_bits_count= put_bits_count(&s->tex_pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2479 flush_put_bits(&s->tex_pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2480 ff_copy_bits(&backup_s.tex_pb, bit_buf_tex[next_block^1], tex_pb_bits_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2481 s->tex_pb= backup_s.tex_pb;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2482 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2483 s->last_bits= put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2484
10727
47e1e474e1f1 Replace CONFIG_ANY_H263_ENCODER definition by CONFIG_H263_ENCODER.
diego
parents: 10683
diff changeset
2485 if (CONFIG_H263_ENCODER &&
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2486 s->out_format == FMT_H263 && s->pict_type!=FF_B_TYPE)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2487 ff_h263_update_motion_val(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2488
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2489 if(next_block==0){ //FIXME 16 vs linesize16
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2490 s->dsp.put_pixels_tab[0][0](s->dest[0], s->rd_scratchpad , s->linesize ,16);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2491 s->dsp.put_pixels_tab[1][0](s->dest[1], s->rd_scratchpad + 16*s->linesize , s->uvlinesize, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2492 s->dsp.put_pixels_tab[1][0](s->dest[2], s->rd_scratchpad + 16*s->linesize + 8, s->uvlinesize, 8);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2493 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2494
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2495 if(s->avctx->mb_decision == FF_MB_DECISION_BITS)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2496 MPV_decode_mb(s, s->block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2497 } else {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2498 int motion_x = 0, motion_y = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2499 s->mv_type=MV_TYPE_16X16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2500 // only one MB-Type possible
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2501
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2502 switch(mb_type){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2503 case CANDIDATE_MB_TYPE_INTRA:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2504 s->mv_dir = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2505 s->mb_intra= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2506 motion_x= s->mv[0][0][0] = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2507 motion_y= s->mv[0][0][1] = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2508 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2509 case CANDIDATE_MB_TYPE_INTER:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2510 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2511 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2512 motion_x= s->mv[0][0][0] = s->p_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2513 motion_y= s->mv[0][0][1] = s->p_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2514 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2515 case CANDIDATE_MB_TYPE_INTER_I:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2516 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2517 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2518 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2519 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2520 j= s->field_select[0][i] = s->p_field_select_table[i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2521 s->mv[0][i][0] = s->p_field_mv_table[i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2522 s->mv[0][i][1] = s->p_field_mv_table[i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2523 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2524 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2525 case CANDIDATE_MB_TYPE_INTER4V:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2526 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2527 s->mv_type = MV_TYPE_8X8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2528 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2529 for(i=0; i<4; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2530 s->mv[0][i][0] = s->current_picture.motion_val[0][s->block_index[i]][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2531 s->mv[0][i][1] = s->current_picture.motion_val[0][s->block_index[i]][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2532 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2533 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2534 case CANDIDATE_MB_TYPE_DIRECT:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2535 if (CONFIG_MPEG4_ENCODER) {
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2536 s->mv_dir = MV_DIR_FORWARD|MV_DIR_BACKWARD|MV_DIRECT;
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2537 s->mb_intra= 0;
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2538 motion_x=s->b_direct_mv_table[xy][0];
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2539 motion_y=s->b_direct_mv_table[xy][1];
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2540 ff_mpeg4_set_direct_mv(s, motion_x, motion_y);
5277
7b3fcb7c61ce Avoid linking with h263.c functions when the relevant codecs
aurel
parents: 5273
diff changeset
2541 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2542 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2543 case CANDIDATE_MB_TYPE_DIRECT0:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2544 if (CONFIG_MPEG4_ENCODER) {
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2545 s->mv_dir = MV_DIR_FORWARD|MV_DIR_BACKWARD|MV_DIRECT;
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2546 s->mb_intra= 0;
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2547 ff_mpeg4_set_direct_mv(s, 0, 0);
5277
7b3fcb7c61ce Avoid linking with h263.c functions when the relevant codecs
aurel
parents: 5273
diff changeset
2548 }
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2549 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2550 case CANDIDATE_MB_TYPE_BIDIR:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2551 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2552 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2553 s->mv[0][0][0] = s->b_bidir_forw_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2554 s->mv[0][0][1] = s->b_bidir_forw_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2555 s->mv[1][0][0] = s->b_bidir_back_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2556 s->mv[1][0][1] = s->b_bidir_back_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2557 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2558 case CANDIDATE_MB_TYPE_BACKWARD:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2559 s->mv_dir = MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2560 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2561 motion_x= s->mv[1][0][0] = s->b_back_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2562 motion_y= s->mv[1][0][1] = s->b_back_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2563 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2564 case CANDIDATE_MB_TYPE_FORWARD:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2565 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2566 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2567 motion_x= s->mv[0][0][0] = s->b_forw_mv_table[xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2568 motion_y= s->mv[0][0][1] = s->b_forw_mv_table[xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2569 // printf(" %d %d ", motion_x, motion_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2570 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2571 case CANDIDATE_MB_TYPE_FORWARD_I:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2572 s->mv_dir = MV_DIR_FORWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2573 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2574 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2575 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2576 j= s->field_select[0][i] = s->b_field_select_table[0][i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2577 s->mv[0][i][0] = s->b_field_mv_table[0][i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2578 s->mv[0][i][1] = s->b_field_mv_table[0][i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2579 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2580 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2581 case CANDIDATE_MB_TYPE_BACKWARD_I:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2582 s->mv_dir = MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2583 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2584 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2585 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2586 j= s->field_select[1][i] = s->b_field_select_table[1][i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2587 s->mv[1][i][0] = s->b_field_mv_table[1][i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2588 s->mv[1][i][1] = s->b_field_mv_table[1][i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2589 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2590 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2591 case CANDIDATE_MB_TYPE_BIDIR_I:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2592 s->mv_dir = MV_DIR_FORWARD | MV_DIR_BACKWARD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2593 s->mv_type = MV_TYPE_FIELD;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2594 s->mb_intra= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2595 for(dir=0; dir<2; dir++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2596 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2597 j= s->field_select[dir][i] = s->b_field_select_table[dir][i][xy];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2598 s->mv[dir][i][0] = s->b_field_mv_table[dir][i][j][xy][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2599 s->mv[dir][i][1] = s->b_field_mv_table[dir][i][j][xy][1];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2600 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2601 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2602 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2603 default:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2604 av_log(s->avctx, AV_LOG_ERROR, "illegal MB type\n");
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2605 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2606
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2607 encode_mb(s, motion_x, motion_y);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2608
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2609 // RAL: Update last macroblock type
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2610 s->last_mv_dir = s->mv_dir;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2611
10727
47e1e474e1f1 Replace CONFIG_ANY_H263_ENCODER definition by CONFIG_H263_ENCODER.
diego
parents: 10683
diff changeset
2612 if (CONFIG_H263_ENCODER &&
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2613 s->out_format == FMT_H263 && s->pict_type!=FF_B_TYPE)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2614 ff_h263_update_motion_val(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2615
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2616 MPV_decode_mb(s, s->block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2617 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2618
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2619 /* clean the MV table in IPS frames for direct mode in B frames */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2620 if(s->mb_intra /* && I,P,S_TYPE */){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2621 s->p_mv_table[xy][0]=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2622 s->p_mv_table[xy][1]=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2623 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2624
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2625 if(s->flags&CODEC_FLAG_PSNR){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2626 int w= 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2627 int h= 16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2628
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2629 if(s->mb_x*16 + 16 > s->width ) w= s->width - s->mb_x*16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2630 if(s->mb_y*16 + 16 > s->height) h= s->height- s->mb_y*16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2631
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2632 s->current_picture.error[0] += sse(
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2633 s, s->new_picture.data[0] + s->mb_x*16 + s->mb_y*s->linesize*16,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2634 s->dest[0], w, h, s->linesize);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2635 s->current_picture.error[1] += sse(
7520
b8ad314749ff fix 4:2:2 psnr
bcoudurier
parents: 7502
diff changeset
2636 s, s->new_picture.data[1] + s->mb_x*8 + s->mb_y*s->uvlinesize*chr_h,
b8ad314749ff fix 4:2:2 psnr
bcoudurier
parents: 7502
diff changeset
2637 s->dest[1], w>>1, h>>s->chroma_y_shift, s->uvlinesize);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2638 s->current_picture.error[2] += sse(
7520
b8ad314749ff fix 4:2:2 psnr
bcoudurier
parents: 7502
diff changeset
2639 s, s->new_picture.data[2] + s->mb_x*8 + s->mb_y*s->uvlinesize*chr_h,
b8ad314749ff fix 4:2:2 psnr
bcoudurier
parents: 7502
diff changeset
2640 s->dest[2], w>>1, h>>s->chroma_y_shift, s->uvlinesize);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2641 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2642 if(s->loop_filter){
10727
47e1e474e1f1 Replace CONFIG_ANY_H263_ENCODER definition by CONFIG_H263_ENCODER.
diego
parents: 10683
diff changeset
2643 if(CONFIG_H263_ENCODER && s->out_format == FMT_H263)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2644 ff_h263_loop_filter(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2645 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2646 //printf("MB %d %d bits\n", s->mb_x+s->mb_y*s->mb_stride, put_bits_count(&s->pb));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2647 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2648 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2649
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2650 //not beautiful here but we must write it before flushing so it has to be here
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2651 if (CONFIG_MSMPEG4_ENCODER && s->msmpeg4_version && s->msmpeg4_version<4 && s->pict_type == FF_I_TYPE)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2652 msmpeg4_encode_ext_header(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2653
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2654 write_slice_end(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2655
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2656 /* Send the last GOB if RTP */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2657 if (s->avctx->rtp_callback) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2658 int number_mb = (mb_y - s->resync_mb_y)*s->mb_width - s->resync_mb_x;
9431
932543edc1d2 Rename pbBufPtr() to put_bits_ptr().
stefano
parents: 9083
diff changeset
2659 pdif = put_bits_ptr(&s->pb) - s->ptr_lastgob;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2660 /* Call the RTP callback to send the last GOB */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2661 emms_c();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2662 s->avctx->rtp_callback(s->avctx, s->ptr_lastgob, pdif, number_mb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2663 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2664
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2665 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2666 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2667
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2668 #define MERGE(field) dst->field += src->field; src->field=0
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2669 static void merge_context_after_me(MpegEncContext *dst, MpegEncContext *src){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2670 MERGE(me.scene_change_score);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2671 MERGE(me.mc_mb_var_sum_temp);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2672 MERGE(me.mb_var_sum_temp);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2673 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2674
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2675 static void merge_context_after_encode(MpegEncContext *dst, MpegEncContext *src){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2676 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2677
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2678 MERGE(dct_count[0]); //note, the other dct vars are not part of the context
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2679 MERGE(dct_count[1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2680 MERGE(mv_bits);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2681 MERGE(i_tex_bits);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2682 MERGE(p_tex_bits);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2683 MERGE(i_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2684 MERGE(f_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2685 MERGE(b_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2686 MERGE(skip_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2687 MERGE(misc_bits);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2688 MERGE(error_count);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2689 MERGE(padding_bug_score);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2690 MERGE(current_picture.error[0]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2691 MERGE(current_picture.error[1]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2692 MERGE(current_picture.error[2]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2693
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2694 if(dst->avctx->noise_reduction){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2695 for(i=0; i<64; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2696 MERGE(dct_error_sum[0][i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2697 MERGE(dct_error_sum[1][i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2698 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2699 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2700
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2701 assert(put_bits_count(&src->pb) % 8 ==0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2702 assert(put_bits_count(&dst->pb) % 8 ==0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2703 ff_copy_bits(&dst->pb, src->pb.buf, put_bits_count(&src->pb));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2704 flush_put_bits(&dst->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2705 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2706
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2707 static int estimate_qp(MpegEncContext *s, int dry_run){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2708 if (s->next_lambda){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2709 s->current_picture_ptr->quality=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2710 s->current_picture.quality = s->next_lambda;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2711 if(!dry_run) s->next_lambda= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2712 } else if (!s->fixed_qscale) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2713 s->current_picture_ptr->quality=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2714 s->current_picture.quality = ff_rate_estimate_qscale(s, dry_run);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2715 if (s->current_picture.quality < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2716 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2717 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2718
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2719 if(s->adaptive_quant){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2720 switch(s->codec_id){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2721 case CODEC_ID_MPEG4:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2722 if (CONFIG_MPEG4_ENCODER)
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2723 ff_clean_mpeg4_qscales(s);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2724 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2725 case CODEC_ID_H263:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2726 case CODEC_ID_H263P:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2727 case CODEC_ID_FLV1:
10034
ea2422aa4635 Do not check for both CONFIG_H263_ENCODER and CONFIG_FLV_ENCODER.
diego
parents: 10032
diff changeset
2728 if (CONFIG_H263_ENCODER)
5278
ef85411bb7e8 cosmetics: indentation
aurel
parents: 5277
diff changeset
2729 ff_clean_h263_qscales(s);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2730 break;
10637
afd44c3040f5 Make sure ff_init_qscale_tab() is called one way or another when
michael
parents: 10636
diff changeset
2731 default:
afd44c3040f5 Make sure ff_init_qscale_tab() is called one way or another when
michael
parents: 10636
diff changeset
2732 ff_init_qscale_tab(s);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2733 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2734
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2735 s->lambda= s->lambda_table[0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2736 //FIXME broken
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2737 }else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2738 s->lambda= s->current_picture.quality;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2739 //printf("%d %d\n", s->avctx->global_quality, s->current_picture.quality);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2740 update_qscale(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2741 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2742 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2743
5273
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2744 /* must be called before writing the header */
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2745 static void set_frame_distances(MpegEncContext * s){
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2746 assert(s->current_picture_ptr->pts != AV_NOPTS_VALUE);
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2747 s->time= s->current_picture_ptr->pts*s->avctx->time_base.num;
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2748
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2749 if(s->pict_type==FF_B_TYPE){
5273
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2750 s->pb_time= s->pp_time - (s->last_non_b_time - s->time);
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2751 assert(s->pb_time > 0 && s->pb_time < s->pp_time);
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2752 }else{
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2753 s->pp_time= s->time - s->last_non_b_time;
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2754 s->last_non_b_time= s->time;
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2755 assert(s->picture_number==0 || s->pp_time > 0);
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2756 }
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2757 }
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2758
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2759 static int encode_picture(MpegEncContext *s, int picture_number)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2760 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2761 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2762 int bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2763
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2764 s->picture_number = picture_number;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2765
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2766 /* Reset the average MB variance */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2767 s->me.mb_var_sum_temp =
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2768 s->me.mc_mb_var_sum_temp = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2769
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2770 /* we need to initialize some time vars before we can encode b-frames */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2771 // RAL: Condition added for MPEG1VIDEO
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2772 if (s->codec_id == CODEC_ID_MPEG1VIDEO || s->codec_id == CODEC_ID_MPEG2VIDEO || (s->h263_pred && !s->h263_msmpeg4))
5273
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2773 set_frame_distances(s);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2774 if(CONFIG_MPEG4_ENCODER && s->codec_id == CODEC_ID_MPEG4)
5273
101f20612a94 Split ff_set_mpeg4_time() and move the non mpeg4 specific part
aurel
parents: 5229
diff changeset
2775 ff_set_mpeg4_time(s);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2776
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2777 s->me.scene_change_score=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2778
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
2779 // s->lambda= s->current_picture_ptr->quality; //FIXME qscale / ... stuff for ME rate distortion
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2780
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2781 if(s->pict_type==FF_I_TYPE){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2782 if(s->msmpeg4_version >= 3) s->no_rounding=1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2783 else s->no_rounding=0;
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2784 }else if(s->pict_type!=FF_B_TYPE){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2785 if(s->flipflop_rounding || s->codec_id == CODEC_ID_H263P || s->codec_id == CODEC_ID_MPEG4)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2786 s->no_rounding ^= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2787 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2788
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2789 if(s->flags & CODEC_FLAG_PASS2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2790 if (estimate_qp(s,1) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2791 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2792 ff_get_2pass_fcode(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2793 }else if(!(s->flags & CODEC_FLAG_QSCALE)){
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2794 if(s->pict_type==FF_B_TYPE)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2795 s->lambda= s->last_lambda_for[s->pict_type];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2796 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2797 s->lambda= s->last_lambda_for[s->last_non_b_pict_type];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2798 update_qscale(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2799 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2800
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2801 s->mb_intra=0; //for the rate distortion & bit compare functions
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2802 for(i=1; i<s->avctx->thread_count; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2803 ff_update_duplicate_context(s->thread_context[i], s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2804 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2805
7767
f692318902dc Check ff_init_me()s return value.
michael
parents: 7520
diff changeset
2806 if(ff_init_me(s)<0)
f692318902dc Check ff_init_me()s return value.
michael
parents: 7520
diff changeset
2807 return -1;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2808
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2809 /* Estimate motion for every MB */
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2810 if(s->pict_type != FF_I_TYPE){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2811 s->lambda = (s->lambda * s->avctx->me_penalty_compensation + 128)>>8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2812 s->lambda2= (s->lambda2* (int64_t)s->avctx->me_penalty_compensation + 128)>>8;
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2813 if(s->pict_type != FF_B_TYPE && s->avctx->me_threshold==0){
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2814 if((s->avctx->pre_me && s->last_non_b_pict_type==FF_I_TYPE) || s->avctx->pre_me==2){
10368
59ec306245a4 Remove casts that are useless since the argument is void *.
reimar
parents: 10155
diff changeset
2815 s->avctx->execute(s->avctx, pre_estimate_motion_thread, &s->thread_context[0], NULL, s->avctx->thread_count, sizeof(void*));
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2816 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2817 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2818
10368
59ec306245a4 Remove casts that are useless since the argument is void *.
reimar
parents: 10155
diff changeset
2819 s->avctx->execute(s->avctx, estimate_motion_thread, &s->thread_context[0], NULL, s->avctx->thread_count, sizeof(void*));
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2820 }else /* if(s->pict_type == FF_I_TYPE) */{
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2821 /* I-Frame */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2822 for(i=0; i<s->mb_stride*s->mb_height; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2823 s->mb_type[i]= CANDIDATE_MB_TYPE_INTRA;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2824
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2825 if(!s->fixed_qscale){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2826 /* finding spatial complexity for I-frame rate control */
10368
59ec306245a4 Remove casts that are useless since the argument is void *.
reimar
parents: 10155
diff changeset
2827 s->avctx->execute(s->avctx, mb_var_thread, &s->thread_context[0], NULL, s->avctx->thread_count, sizeof(void*));
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2828 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2829 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2830 for(i=1; i<s->avctx->thread_count; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2831 merge_context_after_me(s, s->thread_context[i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2832 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2833 s->current_picture.mc_mb_var_sum= s->current_picture_ptr->mc_mb_var_sum= s->me.mc_mb_var_sum_temp;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2834 s->current_picture. mb_var_sum= s->current_picture_ptr-> mb_var_sum= s->me. mb_var_sum_temp;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2835 emms_c();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2836
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2837 if(s->me.scene_change_score > s->avctx->scenechange_threshold && s->pict_type == FF_P_TYPE){
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2838 s->pict_type= FF_I_TYPE;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2839 for(i=0; i<s->mb_stride*s->mb_height; i++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2840 s->mb_type[i]= CANDIDATE_MB_TYPE_INTRA;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2841 //printf("Scene change detected, encoding as I Frame %d %d\n", s->current_picture.mb_var_sum, s->current_picture.mc_mb_var_sum);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2842 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2843
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2844 if(!s->umvplus){
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2845 if(s->pict_type==FF_P_TYPE || s->pict_type==FF_S_TYPE) {
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2846 s->f_code= ff_get_best_fcode(s, s->p_mv_table, CANDIDATE_MB_TYPE_INTER);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2847
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2848 if(s->flags & CODEC_FLAG_INTERLACED_ME){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2849 int a,b;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2850 a= ff_get_best_fcode(s, s->p_field_mv_table[0][0], CANDIDATE_MB_TYPE_INTER_I); //FIXME field_select
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2851 b= ff_get_best_fcode(s, s->p_field_mv_table[1][1], CANDIDATE_MB_TYPE_INTER_I);
6655
22cca5d3173a Implement FFMAX3(a,b,c) - maximum over three arguments.
voroshil
parents: 6517
diff changeset
2852 s->f_code= FFMAX3(s->f_code, a, b);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2853 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2854
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2855 ff_fix_long_p_mvs(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2856 ff_fix_long_mvs(s, NULL, 0, s->p_mv_table, s->f_code, CANDIDATE_MB_TYPE_INTER, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2857 if(s->flags & CODEC_FLAG_INTERLACED_ME){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2858 int j;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2859 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2860 for(j=0; j<2; j++)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2861 ff_fix_long_mvs(s, s->p_field_select_table[i], j,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2862 s->p_field_mv_table[i][j], s->f_code, CANDIDATE_MB_TYPE_INTER_I, 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2863 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2864 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2865 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2866
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2867 if(s->pict_type==FF_B_TYPE){
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2868 int a, b;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2869
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2870 a = ff_get_best_fcode(s, s->b_forw_mv_table, CANDIDATE_MB_TYPE_FORWARD);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2871 b = ff_get_best_fcode(s, s->b_bidir_forw_mv_table, CANDIDATE_MB_TYPE_BIDIR);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2872 s->f_code = FFMAX(a, b);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2873
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2874 a = ff_get_best_fcode(s, s->b_back_mv_table, CANDIDATE_MB_TYPE_BACKWARD);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2875 b = ff_get_best_fcode(s, s->b_bidir_back_mv_table, CANDIDATE_MB_TYPE_BIDIR);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2876 s->b_code = FFMAX(a, b);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2877
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2878 ff_fix_long_mvs(s, NULL, 0, s->b_forw_mv_table, s->f_code, CANDIDATE_MB_TYPE_FORWARD, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2879 ff_fix_long_mvs(s, NULL, 0, s->b_back_mv_table, s->b_code, CANDIDATE_MB_TYPE_BACKWARD, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2880 ff_fix_long_mvs(s, NULL, 0, s->b_bidir_forw_mv_table, s->f_code, CANDIDATE_MB_TYPE_BIDIR, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2881 ff_fix_long_mvs(s, NULL, 0, s->b_bidir_back_mv_table, s->b_code, CANDIDATE_MB_TYPE_BIDIR, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2882 if(s->flags & CODEC_FLAG_INTERLACED_ME){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2883 int dir, j;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2884 for(dir=0; dir<2; dir++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2885 for(i=0; i<2; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2886 for(j=0; j<2; j++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2887 int type= dir ? (CANDIDATE_MB_TYPE_BACKWARD_I|CANDIDATE_MB_TYPE_BIDIR_I)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2888 : (CANDIDATE_MB_TYPE_FORWARD_I |CANDIDATE_MB_TYPE_BIDIR_I);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2889 ff_fix_long_mvs(s, s->b_field_select_table[dir][i], j,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2890 s->b_field_mv_table[dir][i][j], dir ? s->b_code : s->f_code, type, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2891 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2892 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2893 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2894 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2895 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2896 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2897
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2898 if (estimate_qp(s, 0) < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2899 return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2900
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2901 if(s->qscale < 3 && s->max_qcoeff<=128 && s->pict_type==FF_I_TYPE && !(s->flags & CODEC_FLAG_QSCALE))
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2902 s->qscale= 3; //reduce clipping problems
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2903
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2904 if (s->out_format == FMT_MJPEG) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2905 /* for mjpeg, we do include qscale in the matrix */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2906 for(i=1;i<64;i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2907 int j= s->dsp.idct_permutation[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2908
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2909 s->intra_matrix[j] = av_clip_uint8((ff_mpeg1_default_intra_matrix[i] * s->qscale) >> 3);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2910 }
11494
ed6cd3544622 Support intra_dc_precision>8 in jpeg
michael
parents: 11493
diff changeset
2911 s->y_dc_scale_table=
11495
2029a714b794 Add ff_ prefix for mpeg2_dc_scale_table.
michael
parents: 11494
diff changeset
2912 s->c_dc_scale_table= ff_mpeg2_dc_scale_table[s->intra_dc_precision];
2029a714b794 Add ff_ prefix for mpeg2_dc_scale_table.
michael
parents: 11494
diff changeset
2913 s->intra_matrix[0] = ff_mpeg2_dc_scale_table[s->intra_dc_precision][8];
5789
a40edc95456d export convert_matrix and add ff_ prefix
bcoudurier
parents: 5420
diff changeset
2914 ff_convert_matrix(&s->dsp, s->q_intra_matrix, s->q_intra_matrix16,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2915 s->intra_matrix, s->intra_quant_bias, 8, 8, 1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2916 s->qscale= 8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2917 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2918
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2919 //FIXME var duplication
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2920 s->current_picture_ptr->key_frame=
6481
493dc59d469a add FF_ prefix to all (frame)_TYPE usage
aurel
parents: 6401
diff changeset
2921 s->current_picture.key_frame= s->pict_type == FF_I_TYPE; //FIXME pic_ptr
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2922 s->current_picture_ptr->pict_type=
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2923 s->current_picture.pict_type= s->pict_type;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2924
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2925 if(s->current_picture.key_frame)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2926 s->picture_in_gop_number=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2927
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2928 s->last_bits= put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2929 switch(s->out_format) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2930 case FMT_MJPEG:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2931 if (CONFIG_MJPEG_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2932 ff_mjpeg_encode_picture_header(s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2933 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2934 case FMT_H261:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2935 if (CONFIG_H261_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2936 ff_h261_encode_picture_header(s, picture_number);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2937 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2938 case FMT_H263:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2939 if (CONFIG_WMV2_ENCODER && s->codec_id == CODEC_ID_WMV2)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2940 ff_wmv2_encode_picture_header(s, picture_number);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2941 else if (CONFIG_MSMPEG4_ENCODER && s->h263_msmpeg4)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2942 msmpeg4_encode_picture_header(s, picture_number);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2943 else if (CONFIG_MPEG4_ENCODER && s->h263_pred)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2944 mpeg4_encode_picture_header(s, picture_number);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2945 else if (CONFIG_RV10_ENCODER && s->codec_id == CODEC_ID_RV10)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2946 rv10_encode_picture_header(s, picture_number);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2947 else if (CONFIG_RV20_ENCODER && s->codec_id == CODEC_ID_RV20)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2948 rv20_encode_picture_header(s, picture_number);
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2949 else if (CONFIG_FLV_ENCODER && s->codec_id == CODEC_ID_FLV1)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2950 ff_flv_encode_picture_header(s, picture_number);
10727
47e1e474e1f1 Replace CONFIG_ANY_H263_ENCODER definition by CONFIG_H263_ENCODER.
diego
parents: 10683
diff changeset
2951 else if (CONFIG_H263_ENCODER)
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2952 h263_encode_picture_header(s, picture_number);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2953 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2954 case FMT_MPEG1:
8596
68e959302527 replace all occurrence of ENABLE_ by the corresponding CONFIG_, HAVE_ or ARCH_
aurel
parents: 8227
diff changeset
2955 if (CONFIG_MPEG1VIDEO_ENCODER || CONFIG_MPEG2VIDEO_ENCODER)
5209
2ac10c5e5885 cosmetics: indentation
aurel
parents: 5208
diff changeset
2956 mpeg1_encode_picture_header(s, picture_number);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2957 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2958 case FMT_H264:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2959 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2960 default:
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2961 assert(0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2962 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2963 bits= put_bits_count(&s->pb);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2964 s->header_bits= bits - s->last_bits;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2965
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2966 for(i=1; i<s->avctx->thread_count; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2967 update_duplicate_context_after_me(s->thread_context[i], s);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2968 }
10368
59ec306245a4 Remove casts that are useless since the argument is void *.
reimar
parents: 10155
diff changeset
2969 s->avctx->execute(s->avctx, encode_thread, &s->thread_context[0], NULL, s->avctx->thread_count, sizeof(void*));
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2970 for(i=1; i<s->avctx->thread_count; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2971 merge_context_after_encode(s, s->thread_context[i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2972 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2973 emms_c();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2974 return 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2975 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2976
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2977 void denoise_dct_c(MpegEncContext *s, DCTELEM *block){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2978 const int intra= s->mb_intra;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2979 int i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2980
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2981 s->dct_count[intra]++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2982
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2983 for(i=0; i<64; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2984 int level= block[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2985
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2986 if(level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2987 if(level>0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2988 s->dct_error_sum[intra][i] += level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2989 level -= s->dct_offset[intra][i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2990 if(level<0) level=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2991 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2992 s->dct_error_sum[intra][i] -= level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2993 level += s->dct_offset[intra][i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2994 if(level>0) level=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2995 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2996 block[i]= level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2997 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2998 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
2999 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3000
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3001 int dct_quantize_trellis_c(MpegEncContext *s,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3002 DCTELEM *block, int n,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3003 int qscale, int *overflow){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3004 const int *qmat;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3005 const uint8_t *scantable= s->intra_scantable.scantable;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3006 const uint8_t *perm_scantable= s->intra_scantable.permutated;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3007 int max=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3008 unsigned int threshold1, threshold2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3009 int bias=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3010 int run_tab[65];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3011 int level_tab[65];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3012 int score_tab[65];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3013 int survivor[65];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3014 int survivor_count;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3015 int last_run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3016 int last_level=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3017 int last_score= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3018 int last_i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3019 int coeff[2][64];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3020 int coeff_count[64];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3021 int qmul, qadd, start_i, last_non_zero, i, dc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3022 const int esc_length= s->ac_esc_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3023 uint8_t * length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3024 uint8_t * last_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3025 const int lambda= s->lambda2 >> (FF_LAMBDA_SHIFT - 6);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3026
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3027 s->dsp.fdct (block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3028
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3029 if(s->dct_error_sum)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3030 s->denoise_dct(s, block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3031 qmul= qscale*16;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3032 qadd= ((qscale-1)|1)*8;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3033
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3034 if (s->mb_intra) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3035 int q;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3036 if (!s->h263_aic) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3037 if (n < 4)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3038 q = s->y_dc_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3039 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3040 q = s->c_dc_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3041 q = q << 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3042 } else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3043 /* For AIC we skip quant/dequant of INTRADC */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3044 q = 1 << 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3045 qadd=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3046 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3047
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3048 /* note: block[0] is assumed to be positive */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3049 block[0] = (block[0] + (q >> 1)) / q;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3050 start_i = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3051 last_non_zero = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3052 qmat = s->q_intra_matrix[qscale];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3053 if(s->mpeg_quant || s->out_format == FMT_MPEG1)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3054 bias= 1<<(QMAT_SHIFT-1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3055 length = s->intra_ac_vlc_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3056 last_length= s->intra_ac_vlc_last_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3057 } else {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3058 start_i = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3059 last_non_zero = -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3060 qmat = s->q_inter_matrix[qscale];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3061 length = s->inter_ac_vlc_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3062 last_length= s->inter_ac_vlc_last_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3063 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3064 last_i= start_i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3065
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3066 threshold1= (1<<QMAT_SHIFT) - bias - 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3067 threshold2= (threshold1<<1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3068
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3069 for(i=63; i>=start_i; i--) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3070 const int j = scantable[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3071 int level = block[j] * qmat[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3072
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3073 if(((unsigned)(level+threshold1))>threshold2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3074 last_non_zero = i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3075 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3076 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3077 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3078
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3079 for(i=start_i; i<=last_non_zero; i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3080 const int j = scantable[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3081 int level = block[j] * qmat[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3082
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3083 // if( bias+level >= (1<<(QMAT_SHIFT - 3))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3084 // || bias-level >= (1<<(QMAT_SHIFT - 3))){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3085 if(((unsigned)(level+threshold1))>threshold2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3086 if(level>0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3087 level= (bias + level)>>QMAT_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3088 coeff[0][i]= level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3089 coeff[1][i]= level-1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3090 // coeff[2][k]= level-2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3091 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3092 level= (bias - level)>>QMAT_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3093 coeff[0][i]= -level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3094 coeff[1][i]= -level+1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3095 // coeff[2][k]= -level+2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3096 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3097 coeff_count[i]= FFMIN(level, 2);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3098 assert(coeff_count[i]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3099 max |=level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3100 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3101 coeff[0][i]= (level>>31)|1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3102 coeff_count[i]= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3103 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3104 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3105
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3106 *overflow= s->max_qcoeff < max; //overflow might have happened
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3107
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3108 if(last_non_zero < start_i){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3109 memset(block + start_i, 0, (64-start_i)*sizeof(DCTELEM));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3110 return last_non_zero;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3111 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3112
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3113 score_tab[start_i]= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3114 survivor[0]= start_i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3115 survivor_count= 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3116
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3117 for(i=start_i; i<=last_non_zero; i++){
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3118 int level_index, j, zero_distortion;
6401
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3119 int dct_coeff= FFABS(block[ scantable[i] ]);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3120 int best_score=256*256*256*120;
6401
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3121
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3122 if ( s->dsp.fdct == fdct_ifast
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3123 #ifndef FAAN_POSTSCALE
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3124 || s->dsp.fdct == ff_faandct
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3125 #endif
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3126 )
8117
a0f9045e0a82 Promote inv_aanscales array to global scope (ff_inv_aanscales)
pross
parents: 7974
diff changeset
3127 dct_coeff= (dct_coeff*ff_inv_aanscales[ scantable[i] ]) >> 12;
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3128 zero_distortion= dct_coeff*dct_coeff;
6401
345cffb2c613 Fix trellis quant + AAN DCT.
michael
parents: 6127
diff changeset
3129
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3130 for(level_index=0; level_index < coeff_count[i]; level_index++){
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3131 int distortion;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3132 int level= coeff[level_index][i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3133 const int alevel= FFABS(level);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3134 int unquant_coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3135
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3136 assert(level);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3137
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3138 if(s->out_format == FMT_H263){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3139 unquant_coeff= alevel*qmul + qadd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3140 }else{ //MPEG1
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3141 j= s->dsp.idct_permutation[ scantable[i] ]; //FIXME optimize
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3142 if(s->mb_intra){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3143 unquant_coeff = (int)( alevel * qscale * s->intra_matrix[j]) >> 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3144 unquant_coeff = (unquant_coeff - 1) | 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3145 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3146 unquant_coeff = ((( alevel << 1) + 1) * qscale * ((int) s->inter_matrix[j])) >> 4;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3147 unquant_coeff = (unquant_coeff - 1) | 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3148 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3149 unquant_coeff<<= 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3150 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3151
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3152 distortion= (unquant_coeff - dct_coeff) * (unquant_coeff - dct_coeff) - zero_distortion;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3153 level+=64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3154 if((level&(~127)) == 0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3155 for(j=survivor_count-1; j>=0; j--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3156 int run= i - survivor[j];
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3157 int score= distortion + length[UNI_AC_ENC_INDEX(run, level)]*lambda;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3158 score += score_tab[i-run];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3159
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3160 if(score < best_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3161 best_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3162 run_tab[i+1]= run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3163 level_tab[i+1]= level-64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3164 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3165 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3166
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3167 if(s->out_format == FMT_H263){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3168 for(j=survivor_count-1; j>=0; j--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3169 int run= i - survivor[j];
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3170 int score= distortion + last_length[UNI_AC_ENC_INDEX(run, level)]*lambda;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3171 score += score_tab[i-run];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3172 if(score < last_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3173 last_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3174 last_run= run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3175 last_level= level-64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3176 last_i= i+1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3177 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3178 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3179 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3180 }else{
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3181 distortion += esc_length*lambda;
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3182 for(j=survivor_count-1; j>=0; j--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3183 int run= i - survivor[j];
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3184 int score= distortion + score_tab[i-run];
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3185
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3186 if(score < best_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3187 best_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3188 run_tab[i+1]= run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3189 level_tab[i+1]= level-64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3190 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3191 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3192
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3193 if(s->out_format == FMT_H263){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3194 for(j=survivor_count-1; j>=0; j--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3195 int run= i - survivor[j];
6719
e88e719b5e77 Typo: distoration -> distortion.
ramiro
parents: 6717
diff changeset
3196 int score= distortion + score_tab[i-run];
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3197 if(score < last_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3198 last_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3199 last_run= run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3200 last_level= level-64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3201 last_i= i+1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3202 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3203 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3204 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3205 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3206 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3207
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3208 score_tab[i+1]= best_score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3209
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3210 //Note: there is a vlc code in mpeg4 which is 1 bit shorter then another one with a shorter run and the same level
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3211 if(last_non_zero <= 27){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3212 for(; survivor_count; survivor_count--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3213 if(score_tab[ survivor[survivor_count-1] ] <= best_score)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3214 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3215 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3216 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3217 for(; survivor_count; survivor_count--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3218 if(score_tab[ survivor[survivor_count-1] ] <= best_score + lambda)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3219 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3220 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3221 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3222
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3223 survivor[ survivor_count++ ]= i+1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3224 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3225
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3226 if(s->out_format != FMT_H263){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3227 last_score= 256*256*256*120;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3228 for(i= survivor[0]; i<=last_non_zero + 1; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3229 int score= score_tab[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3230 if(i) score += lambda*2; //FIXME exacter?
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3231
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3232 if(score < last_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3233 last_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3234 last_i= i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3235 last_level= level_tab[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3236 last_run= run_tab[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3237 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3238 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3239 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3240
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3241 s->coded_score[n] = last_score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3242
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3243 dc= FFABS(block[0]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3244 last_non_zero= last_i - 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3245 memset(block + start_i, 0, (64-start_i)*sizeof(DCTELEM));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3246
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3247 if(last_non_zero < start_i)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3248 return last_non_zero;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3249
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3250 if(last_non_zero == 0 && start_i == 0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3251 int best_level= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3252 int best_score= dc * dc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3253
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3254 for(i=0; i<coeff_count[0]; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3255 int level= coeff[i][0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3256 int alevel= FFABS(level);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3257 int unquant_coeff, score, distortion;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3258
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3259 if(s->out_format == FMT_H263){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3260 unquant_coeff= (alevel*qmul + qadd)>>3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3261 }else{ //MPEG1
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3262 unquant_coeff = ((( alevel << 1) + 1) * qscale * ((int) s->inter_matrix[0])) >> 4;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3263 unquant_coeff = (unquant_coeff - 1) | 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3264 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3265 unquant_coeff = (unquant_coeff + 4) >> 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3266 unquant_coeff<<= 3 + 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3267
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3268 distortion= (unquant_coeff - dc) * (unquant_coeff - dc);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3269 level+=64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3270 if((level&(~127)) == 0) score= distortion + last_length[UNI_AC_ENC_INDEX(0, level)]*lambda;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3271 else score= distortion + esc_length*lambda;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3272
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3273 if(score < best_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3274 best_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3275 best_level= level - 64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3276 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3277 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3278 block[0]= best_level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3279 s->coded_score[n] = best_score - dc*dc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3280 if(best_level == 0) return -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3281 else return last_non_zero;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3282 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3283
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3284 i= last_i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3285 assert(last_level);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3286
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3287 block[ perm_scantable[last_non_zero] ]= last_level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3288 i -= last_run + 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3289
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3290 for(; i>start_i; i -= run_tab[i] + 1){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3291 block[ perm_scantable[i-1] ]= level_tab[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3292 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3293
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3294 return last_non_zero;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3295 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3296
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3297 //#define REFINE_STATS 1
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3298 static int16_t basis[64][64];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3299
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3300 static void build_basis(uint8_t *perm){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3301 int i, j, x, y;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3302 emms_c();
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3303 for(i=0; i<8; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3304 for(j=0; j<8; j++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3305 for(y=0; y<8; y++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3306 for(x=0; x<8; x++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3307 double s= 0.25*(1<<BASIS_SHIFT);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3308 int index= 8*i + j;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3309 int perm_index= perm[index];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3310 if(i==0) s*= sqrt(0.5);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3311 if(j==0) s*= sqrt(0.5);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3312 basis[perm_index][8*x + y]= lrintf(s * cos((M_PI/8.0)*i*(x+0.5)) * cos((M_PI/8.0)*j*(y+0.5)));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3313 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3314 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3315 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3316 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3317 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3318
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3319 static int dct_quantize_refine(MpegEncContext *s, //FIXME breaks denoise?
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3320 DCTELEM *block, int16_t *weight, DCTELEM *orig,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3321 int n, int qscale){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3322 int16_t rem[64];
11195
d464f498e19f Use LOCAL_ALIGNED macro for local arrays
mru
parents: 10961
diff changeset
3323 LOCAL_ALIGNED_16(DCTELEM, d1, [64]);
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3324 const uint8_t *scantable= s->intra_scantable.scantable;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3325 const uint8_t *perm_scantable= s->intra_scantable.permutated;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3326 // unsigned int threshold1, threshold2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3327 // int bias=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3328 int run_tab[65];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3329 int prev_run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3330 int prev_level=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3331 int qmul, qadd, start_i, last_non_zero, i, dc;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3332 uint8_t * length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3333 uint8_t * last_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3334 int lambda;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3335 int rle_index, run, q = 1, sum; //q is only used when s->mb_intra is true
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3336 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3337 static int count=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3338 static int after_last=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3339 static int to_zero=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3340 static int from_zero=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3341 static int raise=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3342 static int lower=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3343 static int messed_sign=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3344 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3345
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3346 if(basis[0][0] == 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3347 build_basis(s->dsp.idct_permutation);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3348
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3349 qmul= qscale*2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3350 qadd= (qscale-1)|1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3351 if (s->mb_intra) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3352 if (!s->h263_aic) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3353 if (n < 4)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3354 q = s->y_dc_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3355 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3356 q = s->c_dc_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3357 } else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3358 /* For AIC we skip quant/dequant of INTRADC */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3359 q = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3360 qadd=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3361 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3362 q <<= RECON_SHIFT-3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3363 /* note: block[0] is assumed to be positive */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3364 dc= block[0]*q;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3365 // block[0] = (block[0] + (q >> 1)) / q;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3366 start_i = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3367 // if(s->mpeg_quant || s->out_format == FMT_MPEG1)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3368 // bias= 1<<(QMAT_SHIFT-1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3369 length = s->intra_ac_vlc_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3370 last_length= s->intra_ac_vlc_last_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3371 } else {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3372 dc= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3373 start_i = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3374 length = s->inter_ac_vlc_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3375 last_length= s->inter_ac_vlc_last_length;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3376 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3377 last_non_zero = s->block_last_index[n];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3378
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3379 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3380 {START_TIMER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3381 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3382 dc += (1<<(RECON_SHIFT-1));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3383 for(i=0; i<64; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3384 rem[i]= dc - (orig[i]<<RECON_SHIFT); //FIXME use orig dirrectly instead of copying to rem[]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3385 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3386 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3387 STOP_TIMER("memset rem[]")}
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3388 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3389 sum=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3390 for(i=0; i<64; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3391 int one= 36;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3392 int qns=4;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3393 int w;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3394
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3395 w= FFABS(weight[i]) + qns*one;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3396 w= 15 + (48*qns*one + w/2)/w; // 16 .. 63
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3397
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3398 weight[i] = w;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3399 // w=weight[i] = (63*qns + (w/2)) / w;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3400
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3401 assert(w>0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3402 assert(w<(1<<6));
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3403 sum += w*w;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3404 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3405 lambda= sum*(uint64_t)s->lambda2 >> (FF_LAMBDA_SHIFT - 6 + 6 + 6 + 6);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3406 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3407 {START_TIMER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3408 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3409 run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3410 rle_index=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3411 for(i=start_i; i<=last_non_zero; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3412 int j= perm_scantable[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3413 const int level= block[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3414 int coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3415
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3416 if(level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3417 if(level<0) coeff= qmul*level - qadd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3418 else coeff= qmul*level + qadd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3419 run_tab[rle_index++]=run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3420 run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3421
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3422 s->dsp.add_8x8basis(rem, basis[j], coeff);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3423 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3424 run++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3425 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3426 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3427 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3428 if(last_non_zero>0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3429 STOP_TIMER("init rem[]")
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3430 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3431 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3432
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3433 {START_TIMER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3434 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3435 for(;;){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3436 int best_score=s->dsp.try_8x8basis(rem, weight, basis[0], 0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3437 int best_coeff=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3438 int best_change=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3439 int run2, best_unquant_change=0, analyze_gradient;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3440 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3441 {START_TIMER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3442 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3443 analyze_gradient = last_non_zero > 2 || s->avctx->quantizer_noise_shaping >= 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3444
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3445 if(analyze_gradient){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3446 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3447 {START_TIMER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3448 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3449 for(i=0; i<64; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3450 int w= weight[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3451
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3452 d1[i] = (rem[i]*w*w + (1<<(RECON_SHIFT+12-1)))>>(RECON_SHIFT+12);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3453 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3454 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3455 STOP_TIMER("rem*w*w")}
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3456 {START_TIMER
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3457 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3458 s->dsp.fdct(d1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3459 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3460 STOP_TIMER("dct")}
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3461 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3462 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3463
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3464 if(start_i){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3465 const int level= block[0];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3466 int change, old_coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3467
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3468 assert(s->mb_intra);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3469
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3470 old_coeff= q*level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3471
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3472 for(change=-1; change<=1; change+=2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3473 int new_level= level + change;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3474 int score, new_coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3475
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3476 new_coeff= q*new_level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3477 if(new_coeff >= 2048 || new_coeff < 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3478 continue;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3479
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3480 score= s->dsp.try_8x8basis(rem, weight, basis[0], new_coeff - old_coeff);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3481 if(score<best_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3482 best_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3483 best_coeff= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3484 best_change= change;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3485 best_unquant_change= new_coeff - old_coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3486 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3487 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3488 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3489
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3490 run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3491 rle_index=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3492 run2= run_tab[rle_index++];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3493 prev_level=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3494 prev_run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3495
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3496 for(i=start_i; i<64; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3497 int j= perm_scantable[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3498 const int level= block[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3499 int change, old_coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3500
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3501 if(s->avctx->quantizer_noise_shaping < 3 && i > last_non_zero + 1)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3502 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3503
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3504 if(level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3505 if(level<0) old_coeff= qmul*level - qadd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3506 else old_coeff= qmul*level + qadd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3507 run2= run_tab[rle_index++]; //FIXME ! maybe after last
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3508 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3509 old_coeff=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3510 run2--;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3511 assert(run2>=0 || i >= last_non_zero );
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3512 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3513
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3514 for(change=-1; change<=1; change+=2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3515 int new_level= level + change;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3516 int score, new_coeff, unquant_change;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3517
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3518 score=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3519 if(s->avctx->quantizer_noise_shaping < 2 && FFABS(new_level) > FFABS(level))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3520 continue;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3521
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3522 if(new_level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3523 if(new_level<0) new_coeff= qmul*new_level - qadd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3524 else new_coeff= qmul*new_level + qadd;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3525 if(new_coeff >= 2048 || new_coeff <= -2048)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3526 continue;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3527 //FIXME check for overflow
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3528
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3529 if(level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3530 if(level < 63 && level > -63){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3531 if(i < last_non_zero)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3532 score += length[UNI_AC_ENC_INDEX(run, new_level+64)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3533 - length[UNI_AC_ENC_INDEX(run, level+64)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3534 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3535 score += last_length[UNI_AC_ENC_INDEX(run, new_level+64)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3536 - last_length[UNI_AC_ENC_INDEX(run, level+64)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3537 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3538 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3539 assert(FFABS(new_level)==1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3540
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3541 if(analyze_gradient){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3542 int g= d1[ scantable[i] ];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3543 if(g && (g^new_level) >= 0)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3544 continue;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3545 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3546
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3547 if(i < last_non_zero){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3548 int next_i= i + run2 + 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3549 int next_level= block[ perm_scantable[next_i] ] + 64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3550
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3551 if(next_level&(~127))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3552 next_level= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3553
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3554 if(next_i < last_non_zero)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3555 score += length[UNI_AC_ENC_INDEX(run, 65)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3556 + length[UNI_AC_ENC_INDEX(run2, next_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3557 - length[UNI_AC_ENC_INDEX(run + run2 + 1, next_level)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3558 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3559 score += length[UNI_AC_ENC_INDEX(run, 65)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3560 + last_length[UNI_AC_ENC_INDEX(run2, next_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3561 - last_length[UNI_AC_ENC_INDEX(run + run2 + 1, next_level)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3562 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3563 score += last_length[UNI_AC_ENC_INDEX(run, 65)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3564 if(prev_level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3565 score += length[UNI_AC_ENC_INDEX(prev_run, prev_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3566 - last_length[UNI_AC_ENC_INDEX(prev_run, prev_level)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3567 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3568 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3569 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3570 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3571 new_coeff=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3572 assert(FFABS(level)==1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3573
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3574 if(i < last_non_zero){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3575 int next_i= i + run2 + 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3576 int next_level= block[ perm_scantable[next_i] ] + 64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3577
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3578 if(next_level&(~127))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3579 next_level= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3580
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3581 if(next_i < last_non_zero)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3582 score += length[UNI_AC_ENC_INDEX(run + run2 + 1, next_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3583 - length[UNI_AC_ENC_INDEX(run2, next_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3584 - length[UNI_AC_ENC_INDEX(run, 65)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3585 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3586 score += last_length[UNI_AC_ENC_INDEX(run + run2 + 1, next_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3587 - last_length[UNI_AC_ENC_INDEX(run2, next_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3588 - length[UNI_AC_ENC_INDEX(run, 65)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3589 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3590 score += -last_length[UNI_AC_ENC_INDEX(run, 65)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3591 if(prev_level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3592 score += last_length[UNI_AC_ENC_INDEX(prev_run, prev_level)]
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3593 - length[UNI_AC_ENC_INDEX(prev_run, prev_level)];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3594 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3595 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3596 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3597
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3598 score *= lambda;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3599
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3600 unquant_change= new_coeff - old_coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3601 assert((score < 100*lambda && score > -100*lambda) || lambda==0);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3602
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3603 score+= s->dsp.try_8x8basis(rem, weight, basis[j], unquant_change);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3604 if(score<best_score){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3605 best_score= score;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3606 best_coeff= i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3607 best_change= change;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3608 best_unquant_change= unquant_change;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3609 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3610 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3611 if(level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3612 prev_level= level + 64;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3613 if(prev_level&(~127))
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3614 prev_level= 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3615 prev_run= run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3616 run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3617 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3618 run++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3619 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3620 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3621 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3622 STOP_TIMER("iterative step")}
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3623 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3624
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3625 if(best_change){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3626 int j= perm_scantable[ best_coeff ];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3627
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3628 block[j] += best_change;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3629
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3630 if(best_coeff > last_non_zero){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3631 last_non_zero= best_coeff;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3632 assert(block[j]);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3633 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3634 after_last++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3635 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3636 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3637 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3638 if(block[j]){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3639 if(block[j] - best_change){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3640 if(FFABS(block[j]) > FFABS(block[j] - best_change)){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3641 raise++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3642 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3643 lower++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3644 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3645 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3646 from_zero++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3647 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3648 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3649 to_zero++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3650 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3651 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3652 for(; last_non_zero>=start_i; last_non_zero--){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3653 if(block[perm_scantable[last_non_zero]])
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3654 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3655 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3656 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3657 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3658 count++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3659 if(256*256*256*64 % count == 0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3660 printf("after_last:%d to_zero:%d from_zero:%d raise:%d lower:%d sign:%d xyp:%d/%d/%d\n", after_last, to_zero, from_zero, raise, lower, messed_sign, s->mb_x, s->mb_y, s->picture_number);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3661 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3662 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3663 run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3664 rle_index=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3665 for(i=start_i; i<=last_non_zero; i++){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3666 int j= perm_scantable[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3667 const int level= block[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3668
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3669 if(level){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3670 run_tab[rle_index++]=run;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3671 run=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3672 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3673 run++;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3674 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3675 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3676
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3677 s->dsp.add_8x8basis(rem, basis[j], best_unquant_change);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3678 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3679 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3680 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3681 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3682 #ifdef REFINE_STATS
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3683 if(last_non_zero>0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3684 STOP_TIMER("iterative search")
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3685 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3686 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3687 #endif
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3688
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3689 return last_non_zero;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3690 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3691
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3692 int dct_quantize_c(MpegEncContext *s,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3693 DCTELEM *block, int n,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3694 int qscale, int *overflow)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3695 {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3696 int i, j, level, last_non_zero, q, start_i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3697 const int *qmat;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3698 const uint8_t *scantable= s->intra_scantable.scantable;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3699 int bias;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3700 int max=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3701 unsigned int threshold1, threshold2;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3702
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3703 s->dsp.fdct (block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3704
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3705 if(s->dct_error_sum)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3706 s->denoise_dct(s, block);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3707
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3708 if (s->mb_intra) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3709 if (!s->h263_aic) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3710 if (n < 4)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3711 q = s->y_dc_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3712 else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3713 q = s->c_dc_scale;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3714 q = q << 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3715 } else
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3716 /* For AIC we skip quant/dequant of INTRADC */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3717 q = 1 << 3;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3718
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3719 /* note: block[0] is assumed to be positive */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3720 block[0] = (block[0] + (q >> 1)) / q;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3721 start_i = 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3722 last_non_zero = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3723 qmat = s->q_intra_matrix[qscale];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3724 bias= s->intra_quant_bias<<(QMAT_SHIFT - QUANT_BIAS_SHIFT);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3725 } else {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3726 start_i = 0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3727 last_non_zero = -1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3728 qmat = s->q_inter_matrix[qscale];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3729 bias= s->inter_quant_bias<<(QMAT_SHIFT - QUANT_BIAS_SHIFT);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3730 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3731 threshold1= (1<<QMAT_SHIFT) - bias - 1;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3732 threshold2= (threshold1<<1);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3733 for(i=63;i>=start_i;i--) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3734 j = scantable[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3735 level = block[j] * qmat[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3736
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3737 if(((unsigned)(level+threshold1))>threshold2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3738 last_non_zero = i;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3739 break;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3740 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3741 block[j]=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3742 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3743 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3744 for(i=start_i; i<=last_non_zero; i++) {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3745 j = scantable[i];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3746 level = block[j] * qmat[j];
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3747
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3748 // if( bias+level >= (1<<QMAT_SHIFT)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3749 // || bias-level >= (1<<QMAT_SHIFT)){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3750 if(((unsigned)(level+threshold1))>threshold2){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3751 if(level>0){
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3752 level= (bias + level)>>QMAT_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3753 block[j]= level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3754 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3755 level= (bias - level)>>QMAT_SHIFT;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3756 block[j]= -level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3757 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3758 max |=level;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3759 }else{
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3760 block[j]=0;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3761 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3762 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3763 *overflow= s->max_qcoeff < max; //overflow might have happened
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3764
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3765 /* we need this permutation so that we correct the IDCT, we only permute the !=0 elements */
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3766 if (s->dsp.idct_permutation_type != FF_NO_IDCT_PERM)
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3767 ff_block_permute(block, s->dsp.idct_permutation, scantable, last_non_zero);
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3768
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3769 return last_non_zero;
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3770 }
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3771
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3772 AVCodec h263_encoder = {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3773 "h263",
11560
8a4984c5cacc Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents: 11534
diff changeset
3774 AVMEDIA_TYPE_VIDEO,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3775 CODEC_ID_H263,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3776 sizeof(MpegEncContext),
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3777 MPV_encode_init,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3778 MPV_encode_picture,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3779 MPV_encode_end,
10146
38cfe222e1a4 Mark all pix_fmts and supported_framerates compound literals as const.
reimar
parents: 10034
diff changeset
3780 .pix_fmts= (const enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_NONE},
8665
5094ddbe4476 Make more descriptive the long names for the various variants of H.263.
stefano
parents: 8629
diff changeset
3781 .long_name= NULL_IF_CONFIG_SMALL("H.263 / H.263-1996"),
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3782 };
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3783
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3784 AVCodec h263p_encoder = {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3785 "h263p",
11560
8a4984c5cacc Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents: 11534
diff changeset
3786 AVMEDIA_TYPE_VIDEO,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3787 CODEC_ID_H263P,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3788 sizeof(MpegEncContext),
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3789 MPV_encode_init,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3790 MPV_encode_picture,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3791 MPV_encode_end,
10146
38cfe222e1a4 Mark all pix_fmts and supported_framerates compound literals as const.
reimar
parents: 10034
diff changeset
3792 .pix_fmts= (const enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_NONE},
8665
5094ddbe4476 Make more descriptive the long names for the various variants of H.263.
stefano
parents: 8629
diff changeset
3793 .long_name= NULL_IF_CONFIG_SMALL("H.263+ / H.263-1998 / H.263 version 2"),
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3794 };
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3795
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3796 AVCodec msmpeg4v1_encoder = {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3797 "msmpeg4v1",
11560
8a4984c5cacc Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents: 11534
diff changeset
3798 AVMEDIA_TYPE_VIDEO,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3799 CODEC_ID_MSMPEG4V1,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3800 sizeof(MpegEncContext),
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3801 MPV_encode_init,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3802 MPV_encode_picture,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3803 MPV_encode_end,
10146
38cfe222e1a4 Mark all pix_fmts and supported_framerates compound literals as const.
reimar
parents: 10034
diff changeset
3804 .pix_fmts= (const enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_NONE},
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 7034
diff changeset
3805 .long_name= NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 1"),
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3806 };
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3807
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3808 AVCodec msmpeg4v2_encoder = {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3809 "msmpeg4v2",
11560
8a4984c5cacc Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents: 11534
diff changeset
3810 AVMEDIA_TYPE_VIDEO,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3811 CODEC_ID_MSMPEG4V2,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3812 sizeof(MpegEncContext),
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3813 MPV_encode_init,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3814 MPV_encode_picture,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3815 MPV_encode_end,
10146
38cfe222e1a4 Mark all pix_fmts and supported_framerates compound literals as const.
reimar
parents: 10034
diff changeset
3816 .pix_fmts= (const enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_NONE},
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 7034
diff changeset
3817 .long_name= NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 2"),
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3818 };
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3819
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3820 AVCodec msmpeg4v3_encoder = {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3821 "msmpeg4",
11560
8a4984c5cacc Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents: 11534
diff changeset
3822 AVMEDIA_TYPE_VIDEO,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3823 CODEC_ID_MSMPEG4V3,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3824 sizeof(MpegEncContext),
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3825 MPV_encode_init,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3826 MPV_encode_picture,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3827 MPV_encode_end,
10146
38cfe222e1a4 Mark all pix_fmts and supported_framerates compound literals as const.
reimar
parents: 10034
diff changeset
3828 .pix_fmts= (const enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_NONE},
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 7034
diff changeset
3829 .long_name= NULL_IF_CONFIG_SMALL("MPEG-4 part 2 Microsoft variant version 3"),
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3830 };
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3831
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3832 AVCodec wmv1_encoder = {
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3833 "wmv1",
11560
8a4984c5cacc Define AVMediaType enum, and use it instead of enum CodecType, which
stefano
parents: 11534
diff changeset
3834 AVMEDIA_TYPE_VIDEO,
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3835 CODEC_ID_WMV1,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3836 sizeof(MpegEncContext),
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3837 MPV_encode_init,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3838 MPV_encode_picture,
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3839 MPV_encode_end,
10146
38cfe222e1a4 Mark all pix_fmts and supported_framerates compound literals as const.
reimar
parents: 10034
diff changeset
3840 .pix_fmts= (const enum PixelFormat[]){PIX_FMT_YUV420P, PIX_FMT_NONE},
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 7034
diff changeset
3841 .long_name= NULL_IF_CONFIG_SMALL("Windows Media Video 7"),
5204
1c3baf4323bb Description: split mpegvideo.c
benoit
parents:
diff changeset
3842 };