1*53ee8cc1Swenshuai.xi //<MStar Software>
2*53ee8cc1Swenshuai.xi //******************************************************************************
3*53ee8cc1Swenshuai.xi // MStar Software
4*53ee8cc1Swenshuai.xi // Copyright (c) 2010 - 2012 MStar Semiconductor, Inc. All rights reserved.
5*53ee8cc1Swenshuai.xi // All software, firmware and related documentation herein ("MStar Software") are
6*53ee8cc1Swenshuai.xi // intellectual property of MStar Semiconductor, Inc. ("MStar") and protected by
7*53ee8cc1Swenshuai.xi // law, including, but not limited to, copyright law and international treaties.
8*53ee8cc1Swenshuai.xi // Any use, modification, reproduction, retransmission, or republication of all
9*53ee8cc1Swenshuai.xi // or part of MStar Software is expressly prohibited, unless prior written
10*53ee8cc1Swenshuai.xi // permission has been granted by MStar.
11*53ee8cc1Swenshuai.xi //
12*53ee8cc1Swenshuai.xi // By accessing, browsing and/or using MStar Software, you acknowledge that you
13*53ee8cc1Swenshuai.xi // have read, understood, and agree, to be bound by below terms ("Terms") and to
14*53ee8cc1Swenshuai.xi // comply with all applicable laws and regulations:
15*53ee8cc1Swenshuai.xi //
16*53ee8cc1Swenshuai.xi // 1. MStar shall retain any and all right, ownership and interest to MStar
17*53ee8cc1Swenshuai.xi // Software and any modification/derivatives thereof.
18*53ee8cc1Swenshuai.xi // No right, ownership, or interest to MStar Software and any
19*53ee8cc1Swenshuai.xi // modification/derivatives thereof is transferred to you under Terms.
20*53ee8cc1Swenshuai.xi //
21*53ee8cc1Swenshuai.xi // 2. You understand that MStar Software might include, incorporate or be
22*53ee8cc1Swenshuai.xi // supplied together with third party`s software and the use of MStar
23*53ee8cc1Swenshuai.xi // Software may require additional licenses from third parties.
24*53ee8cc1Swenshuai.xi // Therefore, you hereby agree it is your sole responsibility to separately
25*53ee8cc1Swenshuai.xi // obtain any and all third party right and license necessary for your use of
26*53ee8cc1Swenshuai.xi // such third party`s software.
27*53ee8cc1Swenshuai.xi //
28*53ee8cc1Swenshuai.xi // 3. MStar Software and any modification/derivatives thereof shall be deemed as
29*53ee8cc1Swenshuai.xi // MStar`s confidential information and you agree to keep MStar`s
30*53ee8cc1Swenshuai.xi // confidential information in strictest confidence and not disclose to any
31*53ee8cc1Swenshuai.xi // third party.
32*53ee8cc1Swenshuai.xi //
33*53ee8cc1Swenshuai.xi // 4. MStar Software is provided on an "AS IS" basis without warranties of any
34*53ee8cc1Swenshuai.xi // kind. Any warranties are hereby expressly disclaimed by MStar, including
35*53ee8cc1Swenshuai.xi // without limitation, any warranties of merchantability, non-infringement of
36*53ee8cc1Swenshuai.xi // intellectual property rights, fitness for a particular purpose, error free
37*53ee8cc1Swenshuai.xi // and in conformity with any international standard. You agree to waive any
38*53ee8cc1Swenshuai.xi // claim against MStar for any loss, damage, cost or expense that you may
39*53ee8cc1Swenshuai.xi // incur related to your use of MStar Software.
40*53ee8cc1Swenshuai.xi // In no event shall MStar be liable for any direct, indirect, incidental or
41*53ee8cc1Swenshuai.xi // consequential damages, including without limitation, lost of profit or
42*53ee8cc1Swenshuai.xi // revenues, lost or damage of data, and unauthorized system use.
43*53ee8cc1Swenshuai.xi // You agree that this Section 4 shall still apply without being affected
44*53ee8cc1Swenshuai.xi // even if MStar Software has been modified by MStar in accordance with your
45*53ee8cc1Swenshuai.xi // request or instruction for your use, except otherwise agreed by both
46*53ee8cc1Swenshuai.xi // parties in writing.
47*53ee8cc1Swenshuai.xi //
48*53ee8cc1Swenshuai.xi // 5. If requested, MStar may from time to time provide technical supports or
49*53ee8cc1Swenshuai.xi // services in relation with MStar Software to you for your use of
50*53ee8cc1Swenshuai.xi // MStar Software in conjunction with your or your customer`s product
51*53ee8cc1Swenshuai.xi // ("Services").
52*53ee8cc1Swenshuai.xi // You understand and agree that, except otherwise agreed by both parties in
53*53ee8cc1Swenshuai.xi // writing, Services are provided on an "AS IS" basis and the warranty
54*53ee8cc1Swenshuai.xi // disclaimer set forth in Section 4 above shall apply.
55*53ee8cc1Swenshuai.xi //
56*53ee8cc1Swenshuai.xi // 6. Nothing contained herein shall be construed as by implication, estoppels
57*53ee8cc1Swenshuai.xi // or otherwise:
58*53ee8cc1Swenshuai.xi // (a) conferring any license or right to use MStar name, trademark, service
59*53ee8cc1Swenshuai.xi // mark, symbol or any other identification;
60*53ee8cc1Swenshuai.xi // (b) obligating MStar or any of its affiliates to furnish any person,
61*53ee8cc1Swenshuai.xi // including without limitation, you and your customers, any assistance
62*53ee8cc1Swenshuai.xi // of any kind whatsoever, or any information; or
63*53ee8cc1Swenshuai.xi // (c) conferring any license or right under any intellectual property right.
64*53ee8cc1Swenshuai.xi //
65*53ee8cc1Swenshuai.xi // 7. These terms shall be governed by and construed in accordance with the laws
66*53ee8cc1Swenshuai.xi // of Taiwan, R.O.C., excluding its conflict of law rules.
67*53ee8cc1Swenshuai.xi // Any and all dispute arising out hereof or related hereto shall be finally
68*53ee8cc1Swenshuai.xi // settled by arbitration referred to the Chinese Arbitration Association,
69*53ee8cc1Swenshuai.xi // Taipei in accordance with the ROC Arbitration Law and the Arbitration
70*53ee8cc1Swenshuai.xi // Rules of the Association by three (3) arbitrators appointed in accordance
71*53ee8cc1Swenshuai.xi // with the said Rules.
72*53ee8cc1Swenshuai.xi // The place of arbitration shall be in Taipei, Taiwan and the language shall
73*53ee8cc1Swenshuai.xi // be English.
74*53ee8cc1Swenshuai.xi // The arbitration award shall be final and binding to both parties.
75*53ee8cc1Swenshuai.xi //
76*53ee8cc1Swenshuai.xi //******************************************************************************
77*53ee8cc1Swenshuai.xi //<MStar Software>
78*53ee8cc1Swenshuai.xi //////////////////////////////////////////////////////////////////////////
79*53ee8cc1Swenshuai.xi // MStar simple rate control
80*53ee8cc1Swenshuai.xi //////////////////////////////////////////////////////////////////////////
81*53ee8cc1Swenshuai.xi
82*53ee8cc1Swenshuai.xi #include "MFE_chip.h"
83*53ee8cc1Swenshuai.xi #include "mfe_type.h"
84*53ee8cc1Swenshuai.xi #include "mfe_common.h"
85*53ee8cc1Swenshuai.xi #include "ms_dprintf.h"
86*53ee8cc1Swenshuai.xi
87*53ee8cc1Swenshuai.xi #include "mdrv_mfe_math.h"
88*53ee8cc1Swenshuai.xi
89*53ee8cc1Swenshuai.xi #include "msRateCtrl.h"
90*53ee8cc1Swenshuai.xi #if defined(_MFE_T8_)&&defined(_MIPS_PLATFORM_)&&!defined(_KERNEL_MODE_)
91*53ee8cc1Swenshuai.xi #include <stdlib.h>
92*53ee8cc1Swenshuai.xi #include <stdio.h>
93*53ee8cc1Swenshuai.xi #endif
94*53ee8cc1Swenshuai.xi
95*53ee8cc1Swenshuai.xi
96*53ee8cc1Swenshuai.xi
97*53ee8cc1Swenshuai.xi // H264
98*53ee8cc1Swenshuai.xi // Spec limitation is [1, 51]
99*53ee8cc1Swenshuai.xi #define MSRC_MAX_QP_H264 48UL
100*53ee8cc1Swenshuai.xi #define MSRC_MIN_QP_H264 12UL
101*53ee8cc1Swenshuai.xi #define FRAMESKIP_THR_QP_H264 40UL
102*53ee8cc1Swenshuai.xi // MPEG-4, H263
103*53ee8cc1Swenshuai.xi // Spec limitation is [1, 31]
104*53ee8cc1Swenshuai.xi #define MSRC_MAX_QP_MPEG4 28UL
105*53ee8cc1Swenshuai.xi #define MSRC_MIN_QP_MPEG4 1UL
106*53ee8cc1Swenshuai.xi #define FRAMESKIP_THR_QP_MPEG4 20UL
107*53ee8cc1Swenshuai.xi
108*53ee8cc1Swenshuai.xi //! How many seconds of frames are responsible for compensation of bitrate usage.
109*53ee8cc1Swenshuai.xi #define CBR_DEPUTY_SECOND 1UL
110*53ee8cc1Swenshuai.xi #define CVBR_DEPUTY_SECOND 3UL
111*53ee8cc1Swenshuai.xi #define VBR_DEPUTY_SECOND 10UL
112*53ee8cc1Swenshuai.xi
113*53ee8cc1Swenshuai.xi #define MIN_DEPUTY_FACTOR 10UL
114*53ee8cc1Swenshuai.xi
115*53ee8cc1Swenshuai.xi //! Default I-frame weighting over inter-frame
116*53ee8cc1Swenshuai.xi // #define IFRAME_WEIGHT 2UL
117*53ee8cc1Swenshuai.xi // #define PFRAME_WEIGHT 1UL
118*53ee8cc1Swenshuai.xi #define IFRAME_WEIGHT 8UL
119*53ee8cc1Swenshuai.xi #define PFRAME_WEIGHT 4UL
120*53ee8cc1Swenshuai.xi #define BFRAME_WEIGHT 3UL
121*53ee8cc1Swenshuai.xi
122*53ee8cc1Swenshuai.xi
123*53ee8cc1Swenshuai.xi
124*53ee8cc1Swenshuai.xi #ifdef _SUPPORT_MBLEVEL_RC_
125*53ee8cc1Swenshuai.xi #define MBACT_NORMALIZATION_FACTOR 2UL
126*53ee8cc1Swenshuai.xi //#define FLAT_AREA_VAR_THR 5UL
127*53ee8cc1Swenshuai.xi #define SPEC_QP_DIFF_LIMIT 2UL // Spec limitation
128*53ee8cc1Swenshuai.xi #define USE_INT_PREC
129*53ee8cc1Swenshuai.xi // Below are used when USE_INT_PREC
130*53ee8cc1Swenshuai.xi #define SCALE_BITS 6UL
131*53ee8cc1Swenshuai.xi #define SCALE_ACT 6UL
132*53ee8cc1Swenshuai.xi #endif
133*53ee8cc1Swenshuai.xi
134*53ee8cc1Swenshuai.xi //////////////////////////////////////////////////////////////////////////
135*53ee8cc1Swenshuai.xi // global variables
136*53ee8cc1Swenshuai.xi //////////////////////////////////////////////////////////////////////////
137*53ee8cc1Swenshuai.xi #if 0//ndef _USE_EXT_RCDATA_
138*53ee8cc1Swenshuai.xi CVBRRateControl g_cvbrContext;
139*53ee8cc1Swenshuai.xi CVBRRCInfo g_cvbrInfo;
140*53ee8cc1Swenshuai.xi MS_S32 g_cvbrFrameSkip;
141*53ee8cc1Swenshuai.xi #endif
142*53ee8cc1Swenshuai.xi
143*53ee8cc1Swenshuai.xi //private:
144*53ee8cc1Swenshuai.xi MS_S32 cvbr_ComputeFrameQStep(CVBRRateControl* ct, MS_S32 nFrameType); // Return the target 'QStep'
145*53ee8cc1Swenshuai.xi MS_S32 cvbr_ComputeMbQP(CVBRRateControl* ct, MS_S32 mb_idx); // Return the target 'QP'
146*53ee8cc1Swenshuai.xi
147*53ee8cc1Swenshuai.xi MS_S32 rcQP2Qstep(CVBRRateControl* ct, MS_S32 QP);
148*53ee8cc1Swenshuai.xi MS_S32 rcQstep2QP(CVBRRateControl* ct, MS_S32 QstepX32);
149*53ee8cc1Swenshuai.xi
150*53ee8cc1Swenshuai.xi /*
151*53ee8cc1Swenshuai.xi void output_MBR_reg(void* hd, CVBRRateControl* ct)
152*53ee8cc1Swenshuai.xi {
153*53ee8cc1Swenshuai.xi if (ct->m_nVPMbRow>0 && ct->m_nVPSize<=0)
154*53ee8cc1Swenshuai.xi sd_output(hd, 0, 2, "reg_mfe_g_er_mode");
155*53ee8cc1Swenshuai.xi else if (ct->m_nVPMbRow<=0 && ct->m_nVPSize>0)
156*53ee8cc1Swenshuai.xi sd_output(hd, 1, 2, "reg_mfe_g_er_mode");
157*53ee8cc1Swenshuai.xi else if (ct->m_nVPMbRow>0 && ct->m_nVPSize>0)
158*53ee8cc1Swenshuai.xi sd_output(hd, 2, 2, "reg_mfe_g_er_mode");
159*53ee8cc1Swenshuai.xi else
160*53ee8cc1Swenshuai.xi sd_output(hd, 3, 2, "reg_mfe_g_er_mode");
161*53ee8cc1Swenshuai.xi if (ct->m_nVPMbRow==1)
162*53ee8cc1Swenshuai.xi sd_output(hd, 0, 2, "reg_mfe_g_er_mby");
163*53ee8cc1Swenshuai.xi else if (ct->m_nVPMbRow==2)
164*53ee8cc1Swenshuai.xi sd_output(hd, 1, 2, "reg_mfe_g_er_mby");
165*53ee8cc1Swenshuai.xi else if (ct->m_nVPMbRow==4)
166*53ee8cc1Swenshuai.xi sd_output(hd, 2, 2, "reg_mfe_g_er_mby");
167*53ee8cc1Swenshuai.xi else if (ct->m_nVPMbRow==8)
168*53ee8cc1Swenshuai.xi sd_output(hd, 3, 2, "reg_mfe_g_er_mby");
169*53ee8cc1Swenshuai.xi else
170*53ee8cc1Swenshuai.xi sd_output(hd, 0, 2, "reg_mfe_g_er_mby");
171*53ee8cc1Swenshuai.xi }
172*53ee8cc1Swenshuai.xi */
173*53ee8cc1Swenshuai.xi
cvbr_InitRateControl(CVBRRateControl * ct,CVBRRCInfo * pRCInfo)174*53ee8cc1Swenshuai.xi void cvbr_InitRateControl(CVBRRateControl* ct, CVBRRCInfo* pRCInfo)
175*53ee8cc1Swenshuai.xi {
176*53ee8cc1Swenshuai.xi MS_S32 i, j;
177*53ee8cc1Swenshuai.xi
178*53ee8cc1Swenshuai.xi ct->m_nFrameCount = 0;
179*53ee8cc1Swenshuai.xi
180*53ee8cc1Swenshuai.xi MS_ASSERT(pRCInfo->nWidth<=MAX_MB_WIDTH*16);
181*53ee8cc1Swenshuai.xi if (pRCInfo->rcMethod==CONST_QUALITY)
182*53ee8cc1Swenshuai.xi MS_ASSERT(pRCInfo->rcGranularity==FRAMELEVELRC);
183*53ee8cc1Swenshuai.xi // Copy parameters, no error checking yet.
184*53ee8cc1Swenshuai.xi ct->m_nCodecType = pRCInfo->nCodecType;
185*53ee8cc1Swenshuai.xi ct->m_nWidth = pRCInfo->nWidth;
186*53ee8cc1Swenshuai.xi ct->m_nHeight = pRCInfo->nHeight;
187*53ee8cc1Swenshuai.xi /*
188*53ee8cc1Swenshuai.xi if (pRCInfo->fTargetFrameRate - (MS_S32)pRCInfo->fTargetFrameRate != 0) {
189*53ee8cc1Swenshuai.xi switch ((MS_S32)pRCInfo->fTargetFrameRate) {
190*53ee8cc1Swenshuai.xi case 14: // 14.985
191*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateNum = 15000;
192*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateDeNum = 1001;
193*53ee8cc1Swenshuai.xi break;
194*53ee8cc1Swenshuai.xi case 23: // 23.976
195*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateNum = 24000;
196*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateDeNum = 1001;
197*53ee8cc1Swenshuai.xi break;
198*53ee8cc1Swenshuai.xi case 29: // 29.97
199*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateNum = 30000;
200*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateDeNum = 1001;
201*53ee8cc1Swenshuai.xi break;
202*53ee8cc1Swenshuai.xi default:
203*53ee8cc1Swenshuai.xi MS_ASSERT(0);
204*53ee8cc1Swenshuai.xi }
205*53ee8cc1Swenshuai.xi }
206*53ee8cc1Swenshuai.xi */
207*53ee8cc1Swenshuai.xi //2010/01/21 because LG GP3 no support MS_FLOAT operation, so only support 29.97 now.
208*53ee8cc1Swenshuai.xi //if fps is 29.97, fTargetFrameRate_is_float =1
209*53ee8cc1Swenshuai.xi if (pRCInfo->fTargetFrameRate_is_float==1) {
210*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateNum = 30000;
211*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateDeNum = 1001;
212*53ee8cc1Swenshuai.xi }
213*53ee8cc1Swenshuai.xi else {
214*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateNum = pRCInfo->int_fTargetFrameRate;
215*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateDeNum = 1;
216*53ee8cc1Swenshuai.xi }
217*53ee8cc1Swenshuai.xi
218*53ee8cc1Swenshuai.xi ct->m_nTargetFrameRateInt = ct->m_nTargetFrameRateNum / ct->m_nTargetFrameRateDeNum;
219*53ee8cc1Swenshuai.xi ct->m_nBitrate = pRCInfo->nBitrate;
220*53ee8cc1Swenshuai.xi ct->m_nMaxBitrate = pRCInfo->nMaxBitrate;
221*53ee8cc1Swenshuai.xi ct->m_nConstQP = pRCInfo->nConstQP;
222*53ee8cc1Swenshuai.xi ct->m_nVPSize = pRCInfo->nVPSize;
223*53ee8cc1Swenshuai.xi ct->m_nVPMbRow = pRCInfo->nVPMbRow;
224*53ee8cc1Swenshuai.xi ct->m_bFixedFrameRate = pRCInfo->bFixedFrameRate;
225*53ee8cc1Swenshuai.xi ct->m_nPCount = pRCInfo->nPCount;
226*53ee8cc1Swenshuai.xi ct->m_nBCount = pRCInfo->nBCount;
227*53ee8cc1Swenshuai.xi ct->m_rcGranularity = pRCInfo->rcGranularity;
228*53ee8cc1Swenshuai.xi ct->m_rcMethod = pRCInfo->rcMethod;
229*53ee8cc1Swenshuai.xi // More bitrate checking
230*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONSTRAINED_VARIABLE_BITRATE) {
231*53ee8cc1Swenshuai.xi if (ct->m_nBitrate==0)
232*53ee8cc1Swenshuai.xi ct->m_nBitrate = (MS_S32)(ct->m_nMaxBitrate *6/10);
233*53ee8cc1Swenshuai.xi if (ct->m_nMaxBitrate==0)
234*53ee8cc1Swenshuai.xi ct->m_nMaxBitrate = (MS_S32)(ct->m_nBitrate * 14/10);
235*53ee8cc1Swenshuai.xi ct->m_nMaxOffset = (MS_S32)MFE_DIV_S64((MS_S64)(ct->m_nMaxBitrate - ct->m_nBitrate)*ct->m_nTargetFrameRateDeNum, ct->m_nTargetFrameRateNum) ;
236*53ee8cc1Swenshuai.xi // ct->m_nMaxOffset = (MS_S32)((MS_S64)(ct->m_nMaxBitrate - ct->m_nBitrate)*ct->m_nTargetFrameRateDeNum / ct->m_nTargetFrameRateNum);
237*53ee8cc1Swenshuai.xi
238*53ee8cc1Swenshuai.xi if (ct->m_rcGranularity==FRAMELEVELRC)
239*53ee8cc1Swenshuai.xi ct->m_nMaxOffset = (ct->m_nMaxOffset) >> 2;
240*53ee8cc1Swenshuai.xi }
241*53ee8cc1Swenshuai.xi else {
242*53ee8cc1Swenshuai.xi ct->m_nMaxBitrate = 0; // Don't care
243*53ee8cc1Swenshuai.xi }
244*53ee8cc1Swenshuai.xi
245*53ee8cc1Swenshuai.xi // Derived
246*53ee8cc1Swenshuai.xi ct->m_nAvgBitsPerFrame = (MS_S32)MFE_DIV_S64((MS_S64)ct->m_nBitrate*ct->m_nTargetFrameRateDeNum, ct->m_nTargetFrameRateNum) ;
247*53ee8cc1Swenshuai.xi //ct->m_nAvgBitsPerFrame = (MS_S32)((MS_S64)ct->m_nBitrate *ct->m_nTargetFrameRateDeNum / ct->m_nTargetFrameRateNum);
248*53ee8cc1Swenshuai.xi
249*53ee8cc1Swenshuai.xi if (ct->m_nPCount == -1) {
250*53ee8cc1Swenshuai.xi // infinite P, no B
251*53ee8cc1Swenshuai.xi // lim((1 + m_nPCount) / (IFRAME_WEIGHT + PFRAME_WEIGHT * m_nPCount)) = 1 / 4
252*53ee8cc1Swenshuai.xi i = 1;
253*53ee8cc1Swenshuai.xi j = 4;
254*53ee8cc1Swenshuai.xi }
255*53ee8cc1Swenshuai.xi else {
256*53ee8cc1Swenshuai.xi i = 1+ct->m_nPCount*(1+ct->m_nBCount);
257*53ee8cc1Swenshuai.xi j = IFRAME_WEIGHT + PFRAME_WEIGHT*ct->m_nPCount + BFRAME_WEIGHT*(ct->m_nPCount*ct->m_nBCount);
258*53ee8cc1Swenshuai.xi }
259*53ee8cc1Swenshuai.xi
260*53ee8cc1Swenshuai.xi ct->m_nBitsPerFrame[0] = (MS_S32)MFE_DIV_S64((MS_S64)ct->m_nBitrate*ct->m_nTargetFrameRateDeNum*IFRAME_WEIGHT*i, (MS_S64)j*ct->m_nTargetFrameRateNum);
261*53ee8cc1Swenshuai.xi ct->m_nBitsPerFrame[1] = (MS_S32)MFE_DIV_S64((MS_S64)ct->m_nBitrate*ct->m_nTargetFrameRateDeNum*PFRAME_WEIGHT*i, (MS_S64)j*ct->m_nTargetFrameRateNum);
262*53ee8cc1Swenshuai.xi ct->m_nBitsPerFrame[2] = (MS_S32)MFE_DIV_S64((MS_S64)ct->m_nBitrate*ct->m_nTargetFrameRateDeNum*BFRAME_WEIGHT*i, (MS_S64)j*ct->m_nTargetFrameRateNum);
263*53ee8cc1Swenshuai.xi #ifdef _SUPPORT_MBLEVEL_RC_
264*53ee8cc1Swenshuai.xi ct->m_nMbWidth = (ct->m_nWidth+15)>>4;
265*53ee8cc1Swenshuai.xi #endif
266*53ee8cc1Swenshuai.xi
267*53ee8cc1Swenshuai.xi // QP, QStep: min, max
268*53ee8cc1Swenshuai.xi if (ct->m_nCodecType==REG_ENC_MODE_MPG4 || ct->m_nCodecType==REG_ENC_MODE_H263) {
269*53ee8cc1Swenshuai.xi ct->m_nMinQP = MSRC_MIN_QP_MPEG4;
270*53ee8cc1Swenshuai.xi ct->m_nMaxQP = MSRC_MAX_QP_MPEG4;
271*53ee8cc1Swenshuai.xi ct->m_nMinQStep = rcQP2Qstep(ct, MSRC_MIN_QP_MPEG4)-1;
272*53ee8cc1Swenshuai.xi ct->m_nMaxQStep = rcQP2Qstep(ct, MSRC_MAX_QP_MPEG4);
273*53ee8cc1Swenshuai.xi ct->m_nFrameSkipThrQP = FRAMESKIP_THR_QP_MPEG4;
274*53ee8cc1Swenshuai.xi }
275*53ee8cc1Swenshuai.xi else {
276*53ee8cc1Swenshuai.xi ct->m_nMinQP = MSRC_MIN_QP_H264;
277*53ee8cc1Swenshuai.xi ct->m_nMaxQP = MSRC_MAX_QP_H264;
278*53ee8cc1Swenshuai.xi ct->m_nMinQStep = rcQP2Qstep(ct, MSRC_MIN_QP_H264)-1;
279*53ee8cc1Swenshuai.xi ct->m_nMaxQStep = rcQP2Qstep(ct, MSRC_MAX_QP_H264);
280*53ee8cc1Swenshuai.xi ct->m_nFrameSkipThrQP = FRAMESKIP_THR_QP_H264;
281*53ee8cc1Swenshuai.xi MS_ASSERT(ct->m_nCodecType==REG_ENC_MODE_H264);
282*53ee8cc1Swenshuai.xi }
283*53ee8cc1Swenshuai.xi
284*53ee8cc1Swenshuai.xi // Bitrate usage monitoring
285*53ee8cc1Swenshuai.xi ct->m_nMinDeputyCount = (ct->m_nTargetFrameRateNum*CBR_DEPUTY_SECOND)/ct->m_nTargetFrameRateDeNum;
286*53ee8cc1Swenshuai.xi ct->m_nTargetFullness = ct->m_nBitrate >> 1;
287*53ee8cc1Swenshuai.xi ct->m_nBufFullness = ct->m_nTargetFullness;
288*53ee8cc1Swenshuai.xi switch (ct->m_rcMethod) {
289*53ee8cc1Swenshuai.xi case VARIABLE_BITRATE:
290*53ee8cc1Swenshuai.xi ct->m_nDeputyCount = (ct->m_nTargetFrameRateNum*VBR_DEPUTY_SECOND)/ct->m_nTargetFrameRateDeNum;
291*53ee8cc1Swenshuai.xi break;
292*53ee8cc1Swenshuai.xi case CONSTRAINED_VARIABLE_BITRATE:
293*53ee8cc1Swenshuai.xi ct->m_BitrateGauge = MfeDrvMemMalloc(ct->m_nTargetFrameRateInt * sizeof(MS_S32), (const MS_S8*)("m_BitrateGauge"));
294*53ee8cc1Swenshuai.xi if (ct->m_BitrateGauge) {
295*53ee8cc1Swenshuai.xi memset(ct->m_BitrateGauge, 0, ct->m_nTargetFrameRateInt * sizeof(MS_S32));
296*53ee8cc1Swenshuai.xi
297*53ee8cc1Swenshuai.xi ct->m_BitrateGauge[0] = ct->m_nBitsPerFrame[0];
298*53ee8cc1Swenshuai.xi for (i = 1; i < ct->m_nTargetFrameRateInt; i++) {
299*53ee8cc1Swenshuai.xi ct->m_BitrateGauge[i] = ct->m_nBitsPerFrame[1];
300*53ee8cc1Swenshuai.xi for (j = 0; j < ct->m_nBCount; j++, i++) {
301*53ee8cc1Swenshuai.xi if (i == ct->m_nTargetFrameRateInt)
302*53ee8cc1Swenshuai.xi break;
303*53ee8cc1Swenshuai.xi ct->m_BitrateGauge[i] = ct->m_nBitsPerFrame[2];
304*53ee8cc1Swenshuai.xi }
305*53ee8cc1Swenshuai.xi }
306*53ee8cc1Swenshuai.xi }
307*53ee8cc1Swenshuai.xi ct->m_nDeputyCount = (ct->m_nTargetFrameRateNum*CVBR_DEPUTY_SECOND)/ct->m_nTargetFrameRateDeNum;
308*53ee8cc1Swenshuai.xi ct->m_nGaugeCount = ct->m_nTargetFrameRateInt;
309*53ee8cc1Swenshuai.xi ct->m_nGaugeIndex = 0;
310*53ee8cc1Swenshuai.xi ct->m_nGaugeBitrate = (MS_S32)MFE_DIV_S64((MS_S64)ct->m_nBitrate*ct->m_nTargetFrameRateDeNum*ct->m_nGaugeCount, ct->m_nTargetFrameRateNum) ;
311*53ee8cc1Swenshuai.xi //ct->m_nGaugeBitrate = (MS_S32)((MS_S64)ct->m_nBitrate*ct->m_nTargetFrameRateDeNum*ct->m_nGaugeCount) / ct->m_nTargetFrameRateNum;
312*53ee8cc1Swenshuai.xi break;
313*53ee8cc1Swenshuai.xi case CONST_BITRATE:
314*53ee8cc1Swenshuai.xi default:
315*53ee8cc1Swenshuai.xi ct->m_nDeputyCount = (MS_S32)MFE_DIV_S64((MS_S64)ct->m_nTargetFrameRateNum*CBR_DEPUTY_SECOND,ct->m_nTargetFrameRateDeNum) ;
316*53ee8cc1Swenshuai.xi //ct->m_nDeputyCount = (MS_S32)((MS_S64)ct->m_nTargetFrameRateNum*CBR_DEPUTY_SECOND)/ct->m_nTargetFrameRateDeNum;
317*53ee8cc1Swenshuai.xi break;
318*53ee8cc1Swenshuai.xi }
319*53ee8cc1Swenshuai.xi
320*53ee8cc1Swenshuai.xi ct->m_nFrameCount = 0;
321*53ee8cc1Swenshuai.xi ct->m_nTotalBits = 0;
322*53ee8cc1Swenshuai.xi ct->m_nLastFrameAvgQStep[0] = ct->m_nLastFrameAvgQStep[1] = ct->m_nLastFrameBits = 0;
323*53ee8cc1Swenshuai.xi ct->m_nLongTermQP64 = 0;
324*53ee8cc1Swenshuai.xi
325*53ee8cc1Swenshuai.xi ct->m_nTotalMB = (ct->m_nWidth>>4)*(ct->m_nHeight>>4);
326*53ee8cc1Swenshuai.xi #if 0//ndef _USE_EXT_RCDATA_
327*53ee8cc1Swenshuai.xi g_cvbrFrameSkip = 0;
328*53ee8cc1Swenshuai.xi #endif
329*53ee8cc1Swenshuai.xi
330*53ee8cc1Swenshuai.xi
331*53ee8cc1Swenshuai.xi // ms_dprintk(DRV_L3,"[CVBRRC] InitRateControl: w=%d, h=%d, FPS=%2.3f, FixedFPS=%d, Bitrate=%d\n",
332*53ee8cc1Swenshuai.xi // ct->m_nWidth, ct->m_nHeight, (MS_FLOAT)ct->m_nTargetFrameRateNum/ct->m_nTargetFrameRateDeNum,
333*53ee8cc1Swenshuai.xi // ct->m_bFixedFrameRate, ct->m_nBitrate);
334*53ee8cc1Swenshuai.xi
335*53ee8cc1Swenshuai.xi
336*53ee8cc1Swenshuai.xi
337*53ee8cc1Swenshuai.xi
338*53ee8cc1Swenshuai.xi }
339*53ee8cc1Swenshuai.xi
340*53ee8cc1Swenshuai.xi
cvbr_CloseRateControl(CVBRRateControl * ct)341*53ee8cc1Swenshuai.xi void cvbr_CloseRateControl(CVBRRateControl* ct)
342*53ee8cc1Swenshuai.xi {
343*53ee8cc1Swenshuai.xi if (ct->m_rcMethod == CONSTRAINED_VARIABLE_BITRATE) {
344*53ee8cc1Swenshuai.xi if (ct->m_BitrateGauge) {
345*53ee8cc1Swenshuai.xi MfeDrvMemFree((void**)&ct->m_BitrateGauge, (const MS_S8*)("m_BitrateGauge"));
346*53ee8cc1Swenshuai.xi ct->m_BitrateGauge = NULL;
347*53ee8cc1Swenshuai.xi }
348*53ee8cc1Swenshuai.xi }
349*53ee8cc1Swenshuai.xi }
350*53ee8cc1Swenshuai.xi
351*53ee8cc1Swenshuai.xi // FieldType: 0--Progressive, 1--First field, 2--second field.
cvbr_InitFrame(CVBRRateControl * ct,MS_S32 nFrameType,MS_S8 FieldType)352*53ee8cc1Swenshuai.xi MS_S32 cvbr_InitFrame(CVBRRateControl* ct, MS_S32 nFrameType, MS_S8 FieldType)
353*53ee8cc1Swenshuai.xi {
354*53ee8cc1Swenshuai.xi MS_S32 nDeputyCount = 0;
355*53ee8cc1Swenshuai.xi MS_S32 delta = 0, nRtnQP;
356*53ee8cc1Swenshuai.xi
357*53ee8cc1Swenshuai.xi if (FieldType==0||FieldType==1)
358*53ee8cc1Swenshuai.xi ct->m_bIsBotField = 0;
359*53ee8cc1Swenshuai.xi else
360*53ee8cc1Swenshuai.xi ct->m_bIsBotField = 1; // FieldType is 2
361*53ee8cc1Swenshuai.xi
362*53ee8cc1Swenshuai.xi ct->m_nFrameType = nFrameType;
363*53ee8cc1Swenshuai.xi ct->m_nLastVPBits = 0;
364*53ee8cc1Swenshuai.xi ct->m_nNewPacket = 0;
365*53ee8cc1Swenshuai.xi
366*53ee8cc1Swenshuai.xi // Target frame bitcount
367*53ee8cc1Swenshuai.xi if (FieldType!=2)
368*53ee8cc1Swenshuai.xi {
369*53ee8cc1Swenshuai.xi if (ct->m_nFrameCount>0) {
370*53ee8cc1Swenshuai.xi // 1. Determine the number of future frame to compensate for current bitrate mismatch.
371*53ee8cc1Swenshuai.xi if (ct->m_nFrameCount>ct->m_nDeputyCount*MIN_DEPUTY_FACTOR)
372*53ee8cc1Swenshuai.xi nDeputyCount = ct->m_nDeputyCount;
373*53ee8cc1Swenshuai.xi else if (ct->m_nFrameCount<ct->m_nMinDeputyCount)
374*53ee8cc1Swenshuai.xi nDeputyCount = ct->m_nMinDeputyCount;
375*53ee8cc1Swenshuai.xi else
376*53ee8cc1Swenshuai.xi nDeputyCount = ct->m_nMinDeputyCount +
377*53ee8cc1Swenshuai.xi (ct->m_nFrameCount-ct->m_nMinDeputyCount)*(ct->m_nDeputyCount-ct->m_nMinDeputyCount)/(ct->m_nDeputyCount*MIN_DEPUTY_FACTOR-ct->m_nMinDeputyCount);
378*53ee8cc1Swenshuai.xi // 2. Calculate the bitcount that this frame should be compensate for.
379*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONST_BITRATE) {
380*53ee8cc1Swenshuai.xi delta = (MS_S32)MFE_DIV_S64((MS_S64)(ct->m_nBufFullness-ct->m_nTargetFullness)*ct->m_nTargetFrameRateDeNum, ct->m_nTargetFrameRateNum) ;
381*53ee8cc1Swenshuai.xi //delta = (MS_S32)((MS_S64)(ct->m_nBufFullness-ct->m_nTargetFullness)*ct->m_nTargetFrameRateDeNum)/ct->m_nTargetFrameRateNum;
382*53ee8cc1Swenshuai.xi
383*53ee8cc1Swenshuai.xi
384*53ee8cc1Swenshuai.xi // delta = (ct->m_nBufFullness>ct->m_nTargetFullness) ?
385*53ee8cc1Swenshuai.xi // (MS_S32)((ct->m_nBufFullness-ct->m_nTargetFullness)/ct->m_fTargetFrameRate) : ct->m_nBufFullness-ct->m_nTargetFullness;
386*53ee8cc1Swenshuai.xi }
387*53ee8cc1Swenshuai.xi else if (ct->m_rcMethod==CONSTRAINED_VARIABLE_BITRATE) {
388*53ee8cc1Swenshuai.xi delta = (ct->m_nBufFullness-ct->m_nTargetFullness) / nDeputyCount;
389*53ee8cc1Swenshuai.xi if (delta<-ct->m_nMaxOffset)
390*53ee8cc1Swenshuai.xi delta = -ct->m_nMaxOffset;
391*53ee8cc1Swenshuai.xi }
392*53ee8cc1Swenshuai.xi else if (ct->m_rcMethod==VARIABLE_BITRATE) {
393*53ee8cc1Swenshuai.xi delta = (ct->m_nBufFullness-ct->m_nTargetFullness) / nDeputyCount;
394*53ee8cc1Swenshuai.xi if (delta>0 && ((ct->m_nLastFrameAvgQStep[0]<<6)>ct->m_nLongTermQP64))
395*53ee8cc1Swenshuai.xi delta = delta>>1; // Make it more variable bitrate to allow better quality
396*53ee8cc1Swenshuai.xi }
397*53ee8cc1Swenshuai.xi // 3. Finally, calculate the target bitcount.
398*53ee8cc1Swenshuai.xi if (ct->m_nPCount==0) {
399*53ee8cc1Swenshuai.xi ct->m_nTargetBits = ct->m_nBitsPerFrame[0] - delta;
400*53ee8cc1Swenshuai.xi }
401*53ee8cc1Swenshuai.xi else {
402*53ee8cc1Swenshuai.xi switch (nFrameType) {
403*53ee8cc1Swenshuai.xi case I_VOP:
404*53ee8cc1Swenshuai.xi ct->m_nTargetBits = ct->m_nBitsPerFrame[0] - delta;
405*53ee8cc1Swenshuai.xi break;
406*53ee8cc1Swenshuai.xi case P_VOP:
407*53ee8cc1Swenshuai.xi ct->m_nTargetBits = ct->m_nBitsPerFrame[1] - delta;
408*53ee8cc1Swenshuai.xi break;
409*53ee8cc1Swenshuai.xi case B_VOP:
410*53ee8cc1Swenshuai.xi ct->m_nTargetBits = ct->m_nBitsPerFrame[2] - delta;
411*53ee8cc1Swenshuai.xi break;
412*53ee8cc1Swenshuai.xi }
413*53ee8cc1Swenshuai.xi }
414*53ee8cc1Swenshuai.xi if (ct->m_nTargetBits<=(ct->m_nAvgBitsPerFrame>>3))
415*53ee8cc1Swenshuai.xi ct->m_nTargetBits = (ct->m_nAvgBitsPerFrame>>3); // Target bitcount must>0 for ComputeFrameQP()
416*53ee8cc1Swenshuai.xi }
417*53ee8cc1Swenshuai.xi else {
418*53ee8cc1Swenshuai.xi ct->m_nTargetBits = ct->m_nBitsPerFrame[0]; // Must be I-frame
419*53ee8cc1Swenshuai.xi }
420*53ee8cc1Swenshuai.xi
421*53ee8cc1Swenshuai.xi /* Return initial frame QP */
422*53ee8cc1Swenshuai.xi
423*53ee8cc1Swenshuai.xi ct->m_nFrameQStep = cvbr_ComputeFrameQStep(ct, nFrameType);
424*53ee8cc1Swenshuai.xi
425*53ee8cc1Swenshuai.xi ct->m_nLastFrameAvgQStep[0] = ct->m_nLastFrameAvgQStep[1] = 0;
426*53ee8cc1Swenshuai.xi ct->m_nLastFrameBits = 0;
427*53ee8cc1Swenshuai.xi
428*53ee8cc1Swenshuai.xi #ifdef _SUPPORT_MBLEVEL_RC_
429*53ee8cc1Swenshuai.xi ct->m_nTargetMbBits = ct->m_nTargetBits / ct->m_nTotalMB;
430*53ee8cc1Swenshuai.xi #endif
431*53ee8cc1Swenshuai.xi }
432*53ee8cc1Swenshuai.xi else { // bottom field
433*53ee8cc1Swenshuai.xi MS_S32 nTargetBits;
434*53ee8cc1Swenshuai.xi ct->m_nFrameQStep = ct->m_nLastFrameAvgQStep[0] / ((ct->m_nTotalMB>>1)-1);
435*53ee8cc1Swenshuai.xi #ifdef _SUPPORT_MBLEVEL_RC_
436*53ee8cc1Swenshuai.xi nTargetBits = ct->m_nTargetBits - ct->m_nLastFrameBits;
437*53ee8cc1Swenshuai.xi if (nTargetBits<ct->m_nTargetBits>>3)
438*53ee8cc1Swenshuai.xi nTargetBits = ct->m_nTargetBits>>3;
439*53ee8cc1Swenshuai.xi ct->m_nTargetMbBits = nTargetBits / (ct->m_nTotalMB>>1);
440*53ee8cc1Swenshuai.xi #endif
441*53ee8cc1Swenshuai.xi }
442*53ee8cc1Swenshuai.xi
443*53ee8cc1Swenshuai.xi #ifdef _SUPPORT_MBLEVEL_RC_
444*53ee8cc1Swenshuai.xi ct->m_nMBN = 0;
445*53ee8cc1Swenshuai.xi if (ct->m_nTargetMbBits<1)
446*53ee8cc1Swenshuai.xi ct->m_nTargetMbBits = 1;
447*53ee8cc1Swenshuai.xi #endif
448*53ee8cc1Swenshuai.xi
449*53ee8cc1Swenshuai.xi nRtnQP = rcQstep2QP(ct, ct->m_nFrameQStep);
450*53ee8cc1Swenshuai.xi //ct->m_nFrameQStep = rcQP2Qstep(ct, nRtnQP);
451*53ee8cc1Swenshuai.xi
452*53ee8cc1Swenshuai.xi // ms_dprintk(DRV_L3,"[CVBRRC] %5d%s[%s] InitFrame: TargetBits %7d InitQP %2d Buffer %7d Deputy=%d\n",
453*53ee8cc1Swenshuai.xi // ct->m_nFrameCount, FieldType==0?"F":(FieldType==1?"T":"B"), nFrameType==I_VOP?"I":(nFrameType==P_VOP?"P":"B"),
454*53ee8cc1Swenshuai.xi // (MS_S32)(ct->m_nTargetBits), nRtnQP, ct->m_nBufFullness, nDeputyCount);
455*53ee8cc1Swenshuai.xi
456*53ee8cc1Swenshuai.xi return nRtnQP;
457*53ee8cc1Swenshuai.xi }
458*53ee8cc1Swenshuai.xi
cvbr_UpdateFrame(CVBRRateControl * ct,MS_S32 totalUsedBits,MS_S8 bDummyFrame,MS_S8 FieldType)459*53ee8cc1Swenshuai.xi MS_S32 cvbr_UpdateFrame(CVBRRateControl* ct, MS_S32 totalUsedBits, MS_S8 bDummyFrame, MS_S8 FieldType)
460*53ee8cc1Swenshuai.xi {
461*53ee8cc1Swenshuai.xi MS_S32 frameskip = 0;
462*53ee8cc1Swenshuai.xi
463*53ee8cc1Swenshuai.xi // Update counter
464*53ee8cc1Swenshuai.xi ct->m_nTotalBits += totalUsedBits;
465*53ee8cc1Swenshuai.xi ct->m_nLastFrameBits += totalUsedBits;
466*53ee8cc1Swenshuai.xi if(ct->m_nBufFullness < MAX_BUF_FULL)
467*53ee8cc1Swenshuai.xi ct->m_nBufFullness += totalUsedBits;
468*53ee8cc1Swenshuai.xi if (FieldType!=1) {
469*53ee8cc1Swenshuai.xi ct->m_nFrameCount++;
470*53ee8cc1Swenshuai.xi ct->m_nLastTargetBits = ct->m_nTargetBits;
471*53ee8cc1Swenshuai.xi
472*53ee8cc1Swenshuai.xi if (!bDummyFrame) {
473*53ee8cc1Swenshuai.xi ct->m_nLastFrameAvgQStep[0] = (ct->m_nLastFrameAvgQStep[0]+ct->m_nLastFrameAvgQStep[1]) / (FieldType==0 ? ct->m_nTotalMB-1 : ct->m_nTotalMB-2);
474*53ee8cc1Swenshuai.xi // Variable bitrate
475*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==VARIABLE_BITRATE)
476*53ee8cc1Swenshuai.xi ct->m_nLongTermQP64 += ((ct->m_nLastFrameAvgQStep[0]<<6)-ct->m_nLongTermQP64) / ct->m_nFrameCount;
477*53ee8cc1Swenshuai.xi else if (ct->m_rcMethod==CONSTRAINED_VARIABLE_BITRATE) {
478*53ee8cc1Swenshuai.xi if (ct->m_nFrameCount==1)
479*53ee8cc1Swenshuai.xi ct->m_nLongTermQP64 = ct->m_nLastFrameAvgQStep[0]<<6;
480*53ee8cc1Swenshuai.xi else
481*53ee8cc1Swenshuai.xi ct->m_nLongTermQP64 = (ct->m_nLongTermQP64*(ct->m_nDeputyCount-1) + (ct->m_nLastFrameAvgQStep[0]<<6)) / ct->m_nDeputyCount;
482*53ee8cc1Swenshuai.xi }
483*53ee8cc1Swenshuai.xi }
484*53ee8cc1Swenshuai.xi else {
485*53ee8cc1Swenshuai.xi ct->m_nLastFrameAvgQStep[0] = ct->m_nFrameQStep;
486*53ee8cc1Swenshuai.xi }
487*53ee8cc1Swenshuai.xi
488*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONSTRAINED_VARIABLE_BITRATE) {
489*53ee8cc1Swenshuai.xi ct->m_nGaugeBitrate -= ct->m_BitrateGauge[ct->m_nGaugeIndex];
490*53ee8cc1Swenshuai.xi ct->m_nGaugeBitrate += totalUsedBits;
491*53ee8cc1Swenshuai.xi ct->m_BitrateGauge[ct->m_nGaugeIndex] = totalUsedBits;
492*53ee8cc1Swenshuai.xi ct->m_nGaugeIndex++;
493*53ee8cc1Swenshuai.xi if (ct->m_nGaugeIndex==ct->m_nGaugeCount)
494*53ee8cc1Swenshuai.xi ct->m_nGaugeIndex = 0;
495*53ee8cc1Swenshuai.xi }
496*53ee8cc1Swenshuai.xi
497*53ee8cc1Swenshuai.xi // Update buffer status
498*53ee8cc1Swenshuai.xi ct->m_nBufFullness -= ct->m_nAvgBitsPerFrame;
499*53ee8cc1Swenshuai.xi }
500*53ee8cc1Swenshuai.xi
501*53ee8cc1Swenshuai.xi
502*53ee8cc1Swenshuai.xi // Check if next skipped frame(s) needed
503*53ee8cc1Swenshuai.xi if (FieldType!=1 && !ct->m_bFixedFrameRate && !bDummyFrame)
504*53ee8cc1Swenshuai.xi {
505*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONSTRAINED_VARIABLE_BITRATE || ct->m_rcMethod==VARIABLE_BITRATE) {
506*53ee8cc1Swenshuai.xi if ( ct->m_nLongTermQP64>(rcQP2Qstep(ct, ct->m_nFrameSkipThrQP)<<6)
507*53ee8cc1Swenshuai.xi && ct->m_nLastFrameBits >= (ct->m_nLastTargetBits<<1) ) {
508*53ee8cc1Swenshuai.xi //frameskip = (ct->m_nLastFrameBits / ct->m_nLastTargetBits)-1;
509*53ee8cc1Swenshuai.xi frameskip = (MS_S32)((ct->m_nLastFrameBits - ct->m_nLastTargetBits) / ct->m_nAvgBitsPerFrame - 1);
510*53ee8cc1Swenshuai.xi if (frameskip<0)
511*53ee8cc1Swenshuai.xi frameskip = 0;
512*53ee8cc1Swenshuai.xi else if (frameskip>ct->m_nMaxFrozenFrame)
513*53ee8cc1Swenshuai.xi frameskip = ct->m_nMaxFrozenFrame;
514*53ee8cc1Swenshuai.xi }
515*53ee8cc1Swenshuai.xi }
516*53ee8cc1Swenshuai.xi else if (ct->m_rcMethod==CONST_BITRATE) {
517*53ee8cc1Swenshuai.xi if (ct->m_nLastFrameAvgQStep[0]>rcQP2Qstep(ct, ct->m_nFrameSkipThrQP)) {
518*53ee8cc1Swenshuai.xi // Actual fullness is updated after encoding dummy-P frame
519*53ee8cc1Swenshuai.xi MS_S32 nBufFullness = ct->m_nBufFullness;
520*53ee8cc1Swenshuai.xi while (nBufFullness > ct->m_nTargetFullness) {
521*53ee8cc1Swenshuai.xi nBufFullness = (MS_S32)(nBufFullness - ct->m_nAvgBitsPerFrame);
522*53ee8cc1Swenshuai.xi frameskip += 1;
523*53ee8cc1Swenshuai.xi }
524*53ee8cc1Swenshuai.xi }
525*53ee8cc1Swenshuai.xi }
526*53ee8cc1Swenshuai.xi }
527*53ee8cc1Swenshuai.xi /*
528*53ee8cc1Swenshuai.xi #ifdef DEBUG
529*53ee8cc1Swenshuai.xi if (frameskip > 255)
530*53ee8cc1Swenshuai.xi {
531*53ee8cc1Swenshuai.xi fprintf (stderr, "Warning: frameskip > 255\n");
532*53ee8cc1Swenshuai.xi }
533*53ee8cc1Swenshuai.xi if (FieldType!=1){
534*53ee8cc1Swenshuai.xi MS_S64 tmp_bitrate ;
535*53ee8cc1Swenshuai.xi tmp_bitrate= (MS_S32)MFE_DIV_S64((MS_S64)ct->m_nTotalBits*ct->m_nTargetFrameRateNum, ct->m_nFrameCount*ct->m_nTargetFrameRateDeNum) ;
536*53ee8cc1Swenshuai.xi
537*53ee8cc1Swenshuai.xi // ms_dprintk(DRV_L3,"[CVBRRC] UpdateFrame(%7d bits, %3d%%) AvgQ=%2d LTQ=%2d Bitrate=%8d frameskip=%2d\n",
538*53ee8cc1Swenshuai.xi // ct->m_nLastFrameBits, (ct->m_nLastFrameBits-ct->m_nTargetBits)*100/ct->m_nTargetBits, rcQstep2QP(ct, ct->m_nLastFrameAvgQStep[0]),
539*53ee8cc1Swenshuai.xi // (MS_S32)(ct->m_nLongTermQP64+32)>>6,(MS_S32)tmp_bitrate, frameskip);
540*53ee8cc1Swenshuai.xi }
541*53ee8cc1Swenshuai.xi
542*53ee8cc1Swenshuai.xi // DEBUG_RC(("[CVBRRC] UpdateFrame(%7d bits, %3d%%) AvgQ=%2d LTQ=%2d Bitrate=%8d frameskip=%2d\n",
543*53ee8cc1Swenshuai.xi // totalUsedBits, (totalUsedBits-ct->m_nTargetBits)*100/ct->m_nTargetBits, ct->m_nLastFrameAvgQStep[0],
544*53ee8cc1Swenshuai.xi // (MS_S32)ct->m_fLongTermQP, ct->m_nGaugeBitrate, frameskip));
545*53ee8cc1Swenshuai.xi #endif
546*53ee8cc1Swenshuai.xi */
547*53ee8cc1Swenshuai.xi return frameskip;
548*53ee8cc1Swenshuai.xi }
549*53ee8cc1Swenshuai.xi
550*53ee8cc1Swenshuai.xi #define SMOOTH_PERIOD 1UL
551*53ee8cc1Swenshuai.xi #define INIT_QP_FACTOR 720UL
552*53ee8cc1Swenshuai.xi #define MIN_INIT_QP 1UL
553*53ee8cc1Swenshuai.xi #define MAX_INIT_QP 15UL
554*53ee8cc1Swenshuai.xi
555*53ee8cc1Swenshuai.xi /* Return target QPStep */
cvbr_ComputeFrameQStep(CVBRRateControl * ct,MS_S32 nFrameType)556*53ee8cc1Swenshuai.xi MS_S32 cvbr_ComputeFrameQStep(CVBRRateControl* ct, MS_S32 nFrameType)
557*53ee8cc1Swenshuai.xi {
558*53ee8cc1Swenshuai.xi MS_S32 newQPStep=0;
559*53ee8cc1Swenshuai.xi MS_S64 buf_rest;
560*53ee8cc1Swenshuai.xi MS_S32 buf_rest_pic;
561*53ee8cc1Swenshuai.xi MS_S32 frames_left;
562*53ee8cc1Swenshuai.xi MS_S32 nAdjust;
563*53ee8cc1Swenshuai.xi
564*53ee8cc1Swenshuai.xi MS_S32 bitrate = ct->m_nBitrate;
565*53ee8cc1Swenshuai.xi MS_S32 targetFPS_DeNum = ct->m_nTargetFrameRateDeNum;
566*53ee8cc1Swenshuai.xi MS_S32 targetFPS_Num = ct->m_nTargetFrameRateNum;
567*53ee8cc1Swenshuai.xi MS_S32 TotalMB = ct->m_nTotalMB;
568*53ee8cc1Swenshuai.xi MS_S32 frame_count = ct->m_nFrameCount;
569*53ee8cc1Swenshuai.xi
570*53ee8cc1Swenshuai.xi // For the very first frame, guess one qp!
571*53ee8cc1Swenshuai.xi if (ct->m_nFrameCount==0) {
572*53ee8cc1Swenshuai.xi MS_S32 nbpMb, newQP;
573*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONST_QUALITY) {
574*53ee8cc1Swenshuai.xi newQP = ct->m_nConstQP;
575*53ee8cc1Swenshuai.xi newQPStep = rcQP2Qstep(ct, newQP); // So that frame qp will be exactly ct->m_nConstQP
576*53ee8cc1Swenshuai.xi }
577*53ee8cc1Swenshuai.xi else {
578*53ee8cc1Swenshuai.xi nbpMb= (MS_S32)MFE_DIV_S64((MS_S64)bitrate*targetFPS_DeNum, TotalMB*targetFPS_Num) +1 ;
579*53ee8cc1Swenshuai.xi //nbpMb = (MS_S32)((MS_S64)ct->m_nBitrate*ct->m_nTargetFrameRateDeNum)/(ct->m_nTotalMB*ct->m_nTargetFrameRateNum) + 1;
580*53ee8cc1Swenshuai.xi newQP = INIT_QP_FACTOR / nbpMb;
581*53ee8cc1Swenshuai.xi if (newQP<MIN_INIT_QP)
582*53ee8cc1Swenshuai.xi newQP = MIN_INIT_QP;
583*53ee8cc1Swenshuai.xi else if (newQP>MAX_INIT_QP)
584*53ee8cc1Swenshuai.xi newQP = MAX_INIT_QP;
585*53ee8cc1Swenshuai.xi newQPStep = newQP<<QS_SHIFT_FACTOR;
586*53ee8cc1Swenshuai.xi }
587*53ee8cc1Swenshuai.xi
588*53ee8cc1Swenshuai.xi ct->m_nMaxFrozenFrame = newQP>>1;
589*53ee8cc1Swenshuai.xi if (ct->m_nMaxFrozenFrame>15)
590*53ee8cc1Swenshuai.xi ct->m_nMaxFrozenFrame=15;
591*53ee8cc1Swenshuai.xi return newQPStep;
592*53ee8cc1Swenshuai.xi }
593*53ee8cc1Swenshuai.xi
594*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONST_QUALITY)
595*53ee8cc1Swenshuai.xi return rcQP2Qstep(ct, ct->m_nConstQP);
596*53ee8cc1Swenshuai.xi
597*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONST_BITRATE) {
598*53ee8cc1Swenshuai.xi buf_rest= MFE_DIV_S64((MS_S64)frame_count*targetFPS_DeNum*bitrate, targetFPS_Num) ;
599*53ee8cc1Swenshuai.xi //buf_rest = (((MS_S64)ct->m_nFrameCount*ct->m_nTargetFrameRateDeNum*ct->m_nBitrate)/ct->m_nTargetFrameRateNum);
600*53ee8cc1Swenshuai.xi
601*53ee8cc1Swenshuai.xi buf_rest += (SMOOTH_PERIOD*ct->m_nBitrate) - ct->m_nTotalBits;
602*53ee8cc1Swenshuai.xi
603*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLastFrameAvgQStep[0];
604*53ee8cc1Swenshuai.xi frames_left = (SMOOTH_PERIOD * ct->m_nTargetFrameRateNum) / ct->m_nTargetFrameRateDeNum;
605*53ee8cc1Swenshuai.xi //if (frames_left > 0)
606*53ee8cc1Swenshuai.xi {
607*53ee8cc1Swenshuai.xi MS_S32 dQP;
608*53ee8cc1Swenshuai.xi buf_rest_pic = (MS_S32)MFE_DIV_S64(buf_rest , frames_left);
609*53ee8cc1Swenshuai.xi dQP = ct->m_nLastFrameAvgQStep[0]>>3;
610*53ee8cc1Swenshuai.xi if (ct->m_nLastFrameBits > (buf_rest_pic*9)>>3) {
611*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLastFrameAvgQStep[0]+dQP;
612*53ee8cc1Swenshuai.xi }
613*53ee8cc1Swenshuai.xi else if (ct->m_nLastFrameBits < (buf_rest_pic*7)>>3) {
614*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLastFrameAvgQStep[0]-dQP;
615*53ee8cc1Swenshuai.xi }
616*53ee8cc1Swenshuai.xi }
617*53ee8cc1Swenshuai.xi }
618*53ee8cc1Swenshuai.xi
619*53ee8cc1Swenshuai.xi else if (ct->m_rcMethod==CONSTRAINED_VARIABLE_BITRATE) {
620*53ee8cc1Swenshuai.xi MS_S32 nLowBound, nHighBound;
621*53ee8cc1Swenshuai.xi nAdjust = ct->m_nLongTermQP64>>2;
622*53ee8cc1Swenshuai.xi nLowBound = (ct->m_nLongTermQP64 - nAdjust) >> 6;
623*53ee8cc1Swenshuai.xi nHighBound = (ct->m_nLongTermQP64 + nAdjust) >> 6;
624*53ee8cc1Swenshuai.xi if (ct->m_nPCount>0 && nFrameType==I_VOP) {
625*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLastFrameAvgQStep[0];
626*53ee8cc1Swenshuai.xi if (ct->m_nGaugeBitrate<ct->m_nBitrate)
627*53ee8cc1Swenshuai.xi newQPStep = newQPStep-(1<<QS_SHIFT_FACTOR);
628*53ee8cc1Swenshuai.xi newQPStep = MSRC_MAX((1<<QS_SHIFT_FACTOR), newQPStep);
629*53ee8cc1Swenshuai.xi newQPStep = MSRC_MIN((12<<QS_SHIFT_FACTOR), newQPStep);
630*53ee8cc1Swenshuai.xi }
631*53ee8cc1Swenshuai.xi else {
632*53ee8cc1Swenshuai.xi MS_S64 tmp;
633*53ee8cc1Swenshuai.xi tmp= MFE_DIV_S64((MS_S64)ct->m_nLongTermQP64 * MFE_DIV_S64(ct->m_nTotalBits,ct->m_nFrameCount), ct->m_nTargetBits) ;
634*53ee8cc1Swenshuai.xi //tmp = ct->m_nLongTermQP64 * (ct->m_nTotalBits/ct->m_nFrameCount) /ct->m_nTargetBits);
635*53ee8cc1Swenshuai.xi MS_ASSERT((tmp >> 6) < (((MS_S64)1)<<31));
636*53ee8cc1Swenshuai.xi newQPStep= (MS_S32)(
637*53ee8cc1Swenshuai.xi MFE_DIV_S64((MS_S64)ct->m_nLongTermQP64 * MFE_DIV_S64(ct->m_nTotalBits,ct->m_nFrameCount),
638*53ee8cc1Swenshuai.xi ct->m_nTargetBits) >>6) ;
639*53ee8cc1Swenshuai.xi //newQPStep = (MS_S32)((ct->m_nLongTermQP64 * (ct->m_nTotalBits/ct->m_nFrameCount) / ct->m_nTargetBits) >> 6);
640*53ee8cc1Swenshuai.xi if (ct->m_nLastFrameBits>ct->m_nLastTargetBits) {
641*53ee8cc1Swenshuai.xi nAdjust = ((ct->m_nLastFrameBits-ct->m_nLastTargetBits)/ct->m_nMaxOffset) + (1<<QS_SHIFT_FACTOR);
642*53ee8cc1Swenshuai.xi if (nAdjust>(3<<QS_SHIFT_FACTOR)) nAdjust=(3<<QS_SHIFT_FACTOR);
643*53ee8cc1Swenshuai.xi if ((ct->m_nLastFrameAvgQStep[0]<<6) > ct->m_nLongTermQP64) { // Danger! Make it more aggressive
644*53ee8cc1Swenshuai.xi nHighBound = ct->m_nLastFrameAvgQStep[0]+nAdjust;
645*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLastFrameAvgQStep[0]+nAdjust;
646*53ee8cc1Swenshuai.xi }
647*53ee8cc1Swenshuai.xi else {
648*53ee8cc1Swenshuai.xi nHighBound += nAdjust;
649*53ee8cc1Swenshuai.xi newQPStep += nAdjust;
650*53ee8cc1Swenshuai.xi }
651*53ee8cc1Swenshuai.xi }
652*53ee8cc1Swenshuai.xi else if (ct->m_nGaugeBitrate>ct->m_nBitrate) {
653*53ee8cc1Swenshuai.xi if (newQPStep < ct->m_nLastFrameAvgQStep[0])
654*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLastFrameAvgQStep[0];
655*53ee8cc1Swenshuai.xi if ((newQPStep<<6) < ct->m_nLongTermQP64)
656*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLongTermQP64>>6;
657*53ee8cc1Swenshuai.xi }
658*53ee8cc1Swenshuai.xi else {
659*53ee8cc1Swenshuai.xi if ((newQPStep<<6) >= ct->m_nLongTermQP64)
660*53ee8cc1Swenshuai.xi newQPStep = (ct->m_nLongTermQP64>>6)-1;
661*53ee8cc1Swenshuai.xi if (ct->m_nTargetFullness>ct->m_nBufFullness) {
662*53ee8cc1Swenshuai.xi nAdjust = (ct->m_nTargetFullness-ct->m_nBufFullness) / (MS_S32)ct->m_nBitrate;
663*53ee8cc1Swenshuai.xi newQPStep -= nAdjust;
664*53ee8cc1Swenshuai.xi }
665*53ee8cc1Swenshuai.xi }
666*53ee8cc1Swenshuai.xi }
667*53ee8cc1Swenshuai.xi newQPStep = MSRC_MIN(nHighBound, newQPStep);
668*53ee8cc1Swenshuai.xi newQPStep = MSRC_MAX(nLowBound, newQPStep);
669*53ee8cc1Swenshuai.xi }
670*53ee8cc1Swenshuai.xi else if (ct->m_rcMethod==VARIABLE_BITRATE) {
671*53ee8cc1Swenshuai.xi MS_S32 nLowBound, nHighBound;
672*53ee8cc1Swenshuai.xi if (ct->m_nPCount>0 && nFrameType==I_VOP) {
673*53ee8cc1Swenshuai.xi newQPStep = ct->m_nLastFrameAvgQStep[0];
674*53ee8cc1Swenshuai.xi if ((ct->m_nLastFrameAvgQStep[0]<<6) > ct->m_nLongTermQP64)
675*53ee8cc1Swenshuai.xi newQPStep = newQPStep-(1<<QS_SHIFT_FACTOR);
676*53ee8cc1Swenshuai.xi }
677*53ee8cc1Swenshuai.xi else {
678*53ee8cc1Swenshuai.xi MS_S32 nAdjLTQ;
679*53ee8cc1Swenshuai.xi MS_S64 tmp;
680*53ee8cc1Swenshuai.xi if (ct->m_nFrameCount>=ct->m_nTargetFrameRateInt || ct->m_nPCount==0) {
681*53ee8cc1Swenshuai.xi tmp= MFE_DIV_S64(
682*53ee8cc1Swenshuai.xi (MS_S64)ct->m_nLongTermQP64 * MFE_DIV_S64(ct->m_nTotalBits,ct->m_nFrameCount),
683*53ee8cc1Swenshuai.xi ct->m_nAvgBitsPerFrame) ;
684*53ee8cc1Swenshuai.xi //tmp = ct->m_nLongTermQP64 * (ct->m_nTotalBits/ct->m_nFrameCount) / ct->m_nAvgBitsPerFrame;
685*53ee8cc1Swenshuai.xi MS_ASSERT((tmp >> 6) < (((MS_S64)1)<<31));
686*53ee8cc1Swenshuai.xi
687*53ee8cc1Swenshuai.xi nAdjLTQ= (MS_S32)(MFE_DIV_S64(
688*53ee8cc1Swenshuai.xi (MS_S64)ct->m_nLongTermQP64 * MFE_DIV_S64(ct->m_nTotalBits,ct->m_nFrameCount),
689*53ee8cc1Swenshuai.xi ct->m_nAvgBitsPerFrame) >>6) ;
690*53ee8cc1Swenshuai.xi //nAdjLTQ = (MS_S32)((ct->m_nLongTermQP64 * (ct->m_nTotalBits/ct->m_nFrameCount) / ct->m_nAvgBitsPerFrame) >> 6);
691*53ee8cc1Swenshuai.xi
692*53ee8cc1Swenshuai.xi }
693*53ee8cc1Swenshuai.xi else {
694*53ee8cc1Swenshuai.xi nAdjLTQ = ct->m_nLongTermQP64>>6; // Wait for stabilization
695*53ee8cc1Swenshuai.xi }
696*53ee8cc1Swenshuai.xi MS_ASSERT(ct->m_nTargetBits>0);
697*53ee8cc1Swenshuai.xi newQPStep = (nAdjLTQ * ct->m_nAvgBitsPerFrame) / ct->m_nTargetBits;
698*53ee8cc1Swenshuai.xi
699*53ee8cc1Swenshuai.xi nAdjust = MSRC_MAX((2<<QS_SHIFT_FACTOR), (MS_S32)(nAdjLTQ)>>2);
700*53ee8cc1Swenshuai.xi nLowBound = (MS_S32)(nAdjLTQ) - nAdjust;
701*53ee8cc1Swenshuai.xi nHighBound = (MS_S32)(nAdjLTQ) + nAdjust;
702*53ee8cc1Swenshuai.xi if (ct->m_nLastFrameBits>ct->m_nLastTargetBits) {
703*53ee8cc1Swenshuai.xi nAdjust = (MS_S32)(ct->m_nLastFrameBits/ct->m_nLastTargetBits);
704*53ee8cc1Swenshuai.xi if (nAdjust>2) nAdjust=2;
705*53ee8cc1Swenshuai.xi nHighBound += nAdjust;
706*53ee8cc1Swenshuai.xi }
707*53ee8cc1Swenshuai.xi
708*53ee8cc1Swenshuai.xi if (ct->m_nAvgBitsPerFrame>ct->m_nTargetBits) {
709*53ee8cc1Swenshuai.xi newQPStep = MSRC_MIN(nHighBound, newQPStep);
710*53ee8cc1Swenshuai.xi }
711*53ee8cc1Swenshuai.xi else {
712*53ee8cc1Swenshuai.xi newQPStep = MSRC_MAX(nLowBound, newQPStep);
713*53ee8cc1Swenshuai.xi }
714*53ee8cc1Swenshuai.xi }
715*53ee8cc1Swenshuai.xi }
716*53ee8cc1Swenshuai.xi
717*53ee8cc1Swenshuai.xi return newQPStep;
718*53ee8cc1Swenshuai.xi }
719*53ee8cc1Swenshuai.xi
720*53ee8cc1Swenshuai.xi // MS_S32 QP2QSTEP32[52] = { // QP2Qstep(QP) * 32
721*53ee8cc1Swenshuai.xi // 20,
722*53ee8cc1Swenshuai.xi // 22, 26, 28, 32, 36, 40, 44, 52, 56, 64,
723*53ee8cc1Swenshuai.xi // 72, 80, 88, 104, 112, 128, 144, 160, 176, 208,
724*53ee8cc1Swenshuai.xi // 224, 256, 288, 320, 352, 416, 448, 512, 576, 640,
725*53ee8cc1Swenshuai.xi // 704, 832, 896, 1024, 1152, 1280, 1408, 1664, 1792, 2048,
726*53ee8cc1Swenshuai.xi // 2304, 2560, 2816, 3328, 3584, 4096, 4608, 5120, 5632, 6656,
727*53ee8cc1Swenshuai.xi // 7168,
728*53ee8cc1Swenshuai.xi // };
rcQP2Qstep(CVBRRateControl * ct,MS_S32 QP)729*53ee8cc1Swenshuai.xi MS_S32 rcQP2Qstep(CVBRRateControl* ct, MS_S32 QP)
730*53ee8cc1Swenshuai.xi {
731*53ee8cc1Swenshuai.xi if (ct->m_nCodecType==REG_ENC_MODE_H264)
732*53ee8cc1Swenshuai.xi {
733*53ee8cc1Swenshuai.xi MS_S32 i;
734*53ee8cc1Swenshuai.xi MS_S32 Qstep;
735*53ee8cc1Swenshuai.xi static const MS_S32 QP2QSTEP[6] = { 20, 22, 26, 28, 32, 36 };
736*53ee8cc1Swenshuai.xi
737*53ee8cc1Swenshuai.xi Qstep = QP2QSTEP[QP % 6];
738*53ee8cc1Swenshuai.xi for( i=0; i<(QP/6); i++)
739*53ee8cc1Swenshuai.xi Qstep *= 2;
740*53ee8cc1Swenshuai.xi
741*53ee8cc1Swenshuai.xi return Qstep;
742*53ee8cc1Swenshuai.xi }
743*53ee8cc1Swenshuai.xi else
744*53ee8cc1Swenshuai.xi {
745*53ee8cc1Swenshuai.xi return QP<<QS_SHIFT_FACTOR;
746*53ee8cc1Swenshuai.xi }
747*53ee8cc1Swenshuai.xi
748*53ee8cc1Swenshuai.xi return 0; // Should never entering this.
749*53ee8cc1Swenshuai.xi }
750*53ee8cc1Swenshuai.xi
rcQstep2QP(CVBRRateControl * ct,MS_S32 QstepX32)751*53ee8cc1Swenshuai.xi MS_S32 rcQstep2QP(CVBRRateControl* ct, MS_S32 QstepX32)
752*53ee8cc1Swenshuai.xi {
753*53ee8cc1Swenshuai.xi if (ct->m_nCodecType==REG_ENC_MODE_H264)
754*53ee8cc1Swenshuai.xi {
755*53ee8cc1Swenshuai.xi MS_S32 q_per = 0, q_rem = 0;
756*53ee8cc1Swenshuai.xi
757*53ee8cc1Swenshuai.xi // MS_ASSERT( QstepX32 >= QP2QSTEP[0] && QstepX32 <= QP2QSTEP[51]);
758*53ee8cc1Swenshuai.xi if( QstepX32 <= ct->m_nMinQStep)
759*53ee8cc1Swenshuai.xi return ct->m_nMinQP;
760*53ee8cc1Swenshuai.xi else if (QstepX32 > ct->m_nMaxQStep)
761*53ee8cc1Swenshuai.xi return ct->m_nMaxQP;
762*53ee8cc1Swenshuai.xi
763*53ee8cc1Swenshuai.xi while( QstepX32 > rcQP2Qstep(ct, 5) )
764*53ee8cc1Swenshuai.xi {
765*53ee8cc1Swenshuai.xi QstepX32 >>= 1;
766*53ee8cc1Swenshuai.xi q_per += 1;
767*53ee8cc1Swenshuai.xi }
768*53ee8cc1Swenshuai.xi
769*53ee8cc1Swenshuai.xi if (QstepX32 <= 21)
770*53ee8cc1Swenshuai.xi {
771*53ee8cc1Swenshuai.xi //QstepX32 = 20;
772*53ee8cc1Swenshuai.xi q_rem = 0;
773*53ee8cc1Swenshuai.xi }
774*53ee8cc1Swenshuai.xi else if (QstepX32 <= 24)
775*53ee8cc1Swenshuai.xi {
776*53ee8cc1Swenshuai.xi //QstepX32 = 22;
777*53ee8cc1Swenshuai.xi q_rem = 1;
778*53ee8cc1Swenshuai.xi }
779*53ee8cc1Swenshuai.xi else if (QstepX32 <= 27)
780*53ee8cc1Swenshuai.xi {
781*53ee8cc1Swenshuai.xi //QstepX32 = 26;
782*53ee8cc1Swenshuai.xi q_rem = 2;
783*53ee8cc1Swenshuai.xi }
784*53ee8cc1Swenshuai.xi else if (QstepX32 <= 30)
785*53ee8cc1Swenshuai.xi {
786*53ee8cc1Swenshuai.xi //QstepX32 = 28;
787*53ee8cc1Swenshuai.xi q_rem = 3;
788*53ee8cc1Swenshuai.xi }
789*53ee8cc1Swenshuai.xi else if (QstepX32 <= 34)
790*53ee8cc1Swenshuai.xi {
791*53ee8cc1Swenshuai.xi //QstepX32 = 32;
792*53ee8cc1Swenshuai.xi q_rem = 4;
793*53ee8cc1Swenshuai.xi }
794*53ee8cc1Swenshuai.xi else
795*53ee8cc1Swenshuai.xi {
796*53ee8cc1Swenshuai.xi //QstepX32 = 36;
797*53ee8cc1Swenshuai.xi q_rem = 5;
798*53ee8cc1Swenshuai.xi }
799*53ee8cc1Swenshuai.xi
800*53ee8cc1Swenshuai.xi return (q_per * 6 + q_rem);
801*53ee8cc1Swenshuai.xi }
802*53ee8cc1Swenshuai.xi else if (ct->m_nCodecType==REG_ENC_MODE_MPG4 || ct->m_nCodecType==REG_ENC_MODE_H263)
803*53ee8cc1Swenshuai.xi {
804*53ee8cc1Swenshuai.xi if( QstepX32 <= ct->m_nMinQStep)
805*53ee8cc1Swenshuai.xi return ct->m_nMinQP;
806*53ee8cc1Swenshuai.xi else if (QstepX32 > ct->m_nMaxQStep)
807*53ee8cc1Swenshuai.xi return ct->m_nMaxQP;
808*53ee8cc1Swenshuai.xi
809*53ee8cc1Swenshuai.xi return QstepX32>>QS_SHIFT_FACTOR;
810*53ee8cc1Swenshuai.xi }
811*53ee8cc1Swenshuai.xi
812*53ee8cc1Swenshuai.xi return 0; // Should never entering this.
813*53ee8cc1Swenshuai.xi }
814*53ee8cc1Swenshuai.xi
815*53ee8cc1Swenshuai.xi #ifdef _SUPPORT_MBLEVEL_RC_
816*53ee8cc1Swenshuai.xi
817*53ee8cc1Swenshuai.xi #define DUMP_FRAME_NO -1L //140
818*53ee8cc1Swenshuai.xi // Each MB must call this routine once
cvbr_InitMB(CVBRRateControl * ct,MS_S32 nVar,const MS_S32 nPrevQP,const MS_S32 nBits,MS_S32 IsIntra,MS_S32 IsP4MV,MS_S32 BPredType,MS_S32 nResetDQ)819*53ee8cc1Swenshuai.xi MS_S32 cvbr_InitMB(CVBRRateControl* ct, MS_S32 nVar, const MS_S32 nPrevQP, const MS_S32 nBits, MS_S32 IsIntra, MS_S32 IsP4MV, MS_S32 BPredType, MS_S32 nResetDQ)
820*53ee8cc1Swenshuai.xi {
821*53ee8cc1Swenshuai.xi #ifdef USE_INT_PREC
822*53ee8cc1Swenshuai.xi MS_S32 nAvgAct, nActj, nNBits=0;
823*53ee8cc1Swenshuai.xi #else
824*53ee8cc1Swenshuai.xi MS_FLOAT fAvgAct, fActj, fNBits;
825*53ee8cc1Swenshuai.xi #endif
826*53ee8cc1Swenshuai.xi MS_S32 QStep, RtnQP, nVariance;
827*53ee8cc1Swenshuai.xi // MS_S32 TgtQP;
828*53ee8cc1Swenshuai.xi MS_BOOL bChangeQP = 0;
829*53ee8cc1Swenshuai.xi MS_S32 nMBX, nMBY;
830*53ee8cc1Swenshuai.xi
831*53ee8cc1Swenshuai.xi #ifdef RAND_INPUT
832*53ee8cc1Swenshuai.xi nVariance = nVar = rand()%128;
833*53ee8cc1Swenshuai.xi //IsIntra = (rand()%4)==0 ? 1 : 0;
834*53ee8cc1Swenshuai.xi //Is4MV = (rand()%4)==0 ? 1 : 0;
835*53ee8cc1Swenshuai.xi #else
836*53ee8cc1Swenshuai.xi nVariance = nVar;
837*53ee8cc1Swenshuai.xi #endif
838*53ee8cc1Swenshuai.xi
839*53ee8cc1Swenshuai.xi // if (ct->m_nFrameCount == DUMP_FRAME_NO)
840*53ee8cc1Swenshuai.xi // printf("ct->m_nMBN = %d (MBX, MBY) = (%2d, %2d), nBits = %d\n", ct->m_nMBN, ct->m_nMBN%ct->m_nMbWidth, ct->m_nMBN/ct->m_nMbWidth, nBits);
841*53ee8cc1Swenshuai.xi
842*53ee8cc1Swenshuai.xi // if (ct->m_rcGranularity==FRAMELEVELRC) {
843*53ee8cc1Swenshuai.xi // return rcQstep2QP(ct->m_nFrameQStep);
844*53ee8cc1Swenshuai.xi // }
845*53ee8cc1Swenshuai.xi
846*53ee8cc1Swenshuai.xi if (ct->m_nMBN==0) { // Initialization
847*53ee8cc1Swenshuai.xi ct->m_nSumAct = 0;
848*53ee8cc1Swenshuai.xi ct->m_nTargetUsedBits = 0;
849*53ee8cc1Swenshuai.xi }
850*53ee8cc1Swenshuai.xi
851*53ee8cc1Swenshuai.xi if (ct->m_nMBN>0)
852*53ee8cc1Swenshuai.xi { // Previous MB updating: Last MB is not counted!
853*53ee8cc1Swenshuai.xi // Update QP
854*53ee8cc1Swenshuai.xi ct->m_nPrevTopQP[(ct->m_nMBN-1)%ct->m_nMbWidth] = nPrevQP;
855*53ee8cc1Swenshuai.xi ct->m_nLastFrameAvgQStep[ct->m_bIsBotField] += rcQP2Qstep(ct, nPrevQP);
856*53ee8cc1Swenshuai.xi
857*53ee8cc1Swenshuai.xi // SW simulation hw pipeline: Update bit usage history
858*53ee8cc1Swenshuai.xi if (ct->m_nMBN<3)
859*53ee8cc1Swenshuai.xi ct->m_nUsedBits[ct->m_nMBN-1] = nBits;
860*53ee8cc1Swenshuai.xi else {
861*53ee8cc1Swenshuai.xi ct->m_nUsedBits[0] = ct->m_nUsedBits[1];
862*53ee8cc1Swenshuai.xi ct->m_nUsedBits[1] = nBits;
863*53ee8cc1Swenshuai.xi }
864*53ee8cc1Swenshuai.xi }
865*53ee8cc1Swenshuai.xi
866*53ee8cc1Swenshuai.xi if (ct->m_nMBN>1) // HW pipeline: See below nNBits (or fNBits) calculation
867*53ee8cc1Swenshuai.xi ct->m_nTargetUsedBits += ct->m_nTargetMbBits;
868*53ee8cc1Swenshuai.xi
869*53ee8cc1Swenshuai.xi // Update history
870*53ee8cc1Swenshuai.xi if (nVariance==0) nVariance = 1;
871*53ee8cc1Swenshuai.xi ct->m_nSumAct += nVariance;
872*53ee8cc1Swenshuai.xi
873*53ee8cc1Swenshuai.xi // QP calculation
874*53ee8cc1Swenshuai.xi ct->m_nNewPacket = 0; // Default
875*53ee8cc1Swenshuai.xi if (ct->m_nMBN<2) {
876*53ee8cc1Swenshuai.xi RtnQP = rcQstep2QP(ct, ct->m_nFrameQStep);
877*53ee8cc1Swenshuai.xi }
878*53ee8cc1Swenshuai.xi else
879*53ee8cc1Swenshuai.xi {
880*53ee8cc1Swenshuai.xi if (ct->m_nCodecType==REG_ENC_MODE_MPG4 && (!IsIntra) && ((ct->m_nFrameType==P_VOP&&IsP4MV)||(ct->m_nFrameType==B_VOP&&BPredType==0))) {
881*53ee8cc1Swenshuai.xi RtnQP = nPrevQP; // Not allowing DQ!=0
882*53ee8cc1Swenshuai.xi }
883*53ee8cc1Swenshuai.xi else {
884*53ee8cc1Swenshuai.xi #ifdef USE_INT_PREC
885*53ee8cc1Swenshuai.xi nNBits = (ct->m_nUsedBits[0]<<SCALE_BITS) / ct->m_nTargetUsedBits;
886*53ee8cc1Swenshuai.xi //MS_ASSERT(nNBits<(1<<(SCALE_BITS+6)));
887*53ee8cc1Swenshuai.xi if (nNBits>=(1<<(SCALE_BITS+6)))
888*53ee8cc1Swenshuai.xi nNBits = (1<<(SCALE_BITS+6)) - 1;
889*53ee8cc1Swenshuai.xi nAvgAct = ct->m_nSumAct / (ct->m_nMBN+1);
890*53ee8cc1Swenshuai.xi nActj = (((nVariance<<1) + nAvgAct) << SCALE_ACT) / (nVariance + (nAvgAct<<1));
891*53ee8cc1Swenshuai.xi // Target QStep
892*53ee8cc1Swenshuai.xi if (ct->m_nFrameType==I_VOP)
893*53ee8cc1Swenshuai.xi QStep = (MS_S32)(((MS_S64)ct->m_nFrameQStep * nActj) >> SCALE_ACT);
894*53ee8cc1Swenshuai.xi else
895*53ee8cc1Swenshuai.xi QStep = (MS_S32)(((MS_S64)ct->m_nFrameQStep * nActj * nNBits) >> (SCALE_ACT+SCALE_BITS));
896*53ee8cc1Swenshuai.xi #else
897*53ee8cc1Swenshuai.xi fNBits = (MS_FLOAT)ct->m_nUsedBits[0] / ct->m_nTargetUsedBits;
898*53ee8cc1Swenshuai.xi fAvgAct = (MS_FLOAT)ct->m_nSumAct / (ct->m_nMBN+1);
899*53ee8cc1Swenshuai.xi fActj = ((nVariance*MBACT_NORMALIZATION_FACTOR) + fAvgAct) / (nVariance + (fAvgAct*MBACT_NORMALIZATION_FACTOR));
900*53ee8cc1Swenshuai.xi // Target QStep
901*53ee8cc1Swenshuai.xi if (ct->m_nFrameType==I_VOP)
902*53ee8cc1Swenshuai.xi QStep = (MS_S32)(ct->m_nFrameQStep * fActj);
903*53ee8cc1Swenshuai.xi else
904*53ee8cc1Swenshuai.xi QStep = (MS_S32)(ct->m_nFrameQStep * fActj * fNBits);
905*53ee8cc1Swenshuai.xi #endif
906*53ee8cc1Swenshuai.xi // if (QStep<(1<<QS_SHIFT_FACTOR))
907*53ee8cc1Swenshuai.xi // QStep = (1<<QS_SHIFT_FACTOR);
908*53ee8cc1Swenshuai.xi MS_ASSERT(QStep>=0 && QStep<(1<<20));
909*53ee8cc1Swenshuai.xi
910*53ee8cc1Swenshuai.xi // Target QP
911*53ee8cc1Swenshuai.xi RtnQP = rcQstep2QP(ct, QStep);
912*53ee8cc1Swenshuai.xi // Refinement: Phase 1, top and left MB-QP limitation
913*53ee8cc1Swenshuai.xi if (ct->m_nMBN >= ct->m_nMbWidth) {
914*53ee8cc1Swenshuai.xi if (RtnQP > ct->m_nPrevTopQP[ct->m_nMBN%ct->m_nMbWidth] + TOP_QP_DIFF_LIMIT) {
915*53ee8cc1Swenshuai.xi RtnQP = ct->m_nPrevTopQP[ct->m_nMBN%ct->m_nMbWidth] + SPEC_QP_DIFF_LIMIT;
916*53ee8cc1Swenshuai.xi bChangeQP = 1;
917*53ee8cc1Swenshuai.xi }
918*53ee8cc1Swenshuai.xi else if (RtnQP < ct->m_nPrevTopQP[ct->m_nMBN%ct->m_nMbWidth] - TOP_QP_DIFF_LIMIT) {
919*53ee8cc1Swenshuai.xi RtnQP = ct->m_nPrevTopQP[ct->m_nMBN%ct->m_nMbWidth] - SPEC_QP_DIFF_LIMIT;
920*53ee8cc1Swenshuai.xi bChangeQP = 1;
921*53ee8cc1Swenshuai.xi }
922*53ee8cc1Swenshuai.xi }
923*53ee8cc1Swenshuai.xi if (RtnQP > nPrevQP + LEFT_QP_DIFF_LIMIT) {
924*53ee8cc1Swenshuai.xi RtnQP = nPrevQP + SPEC_QP_DIFF_LIMIT;
925*53ee8cc1Swenshuai.xi bChangeQP = 1;
926*53ee8cc1Swenshuai.xi }
927*53ee8cc1Swenshuai.xi else if (RtnQP < nPrevQP - LEFT_QP_DIFF_LIMIT) {
928*53ee8cc1Swenshuai.xi RtnQP = nPrevQP - SPEC_QP_DIFF_LIMIT;
929*53ee8cc1Swenshuai.xi bChangeQP = 1;
930*53ee8cc1Swenshuai.xi }
931*53ee8cc1Swenshuai.xi // Refinement: Phase 2, spec QP range limitation
932*53ee8cc1Swenshuai.xi RtnQP = MSRC_MAX(ct->m_nMinQP, RtnQP);
933*53ee8cc1Swenshuai.xi RtnQP = MSRC_MIN(ct->m_nMaxQP, RtnQP);
934*53ee8cc1Swenshuai.xi #if 0 // Don't need this because QP is bounded by rcQstep2QP(),
935*53ee8cc1Swenshuai.xi // and if TOP_QP_DIFF_LIMIT and BOTTOM_QP_DIFF_LIMIT >=2, there is no way to have +-1.
936*53ee8cc1Swenshuai.xi // Refinement: Phase 3, MPEG-4 B-frame limitation
937*53ee8cc1Swenshuai.xi if (ct->m_nCodecType==REG_ENC_MODE_MPG4 && ct->m_nFrameType==B_VOP /*&& RtnQP!=nPrevQP*/) {
938*53ee8cc1Swenshuai.xi if (RtnQP-nPrevQP==-1 || RtnQP-nPrevQP==1)
939*53ee8cc1Swenshuai.xi RtnQP = nPrevQP;
940*53ee8cc1Swenshuai.xi }
941*53ee8cc1Swenshuai.xi #endif
942*53ee8cc1Swenshuai.xi // Avoid frequent QP changing
943*53ee8cc1Swenshuai.xi if (!bChangeQP)
944*53ee8cc1Swenshuai.xi RtnQP = nPrevQP; // Reset
945*53ee8cc1Swenshuai.xi }
946*53ee8cc1Swenshuai.xi
947*53ee8cc1Swenshuai.xi //DEBUG_RC((" [CVBRRC] %d : NBits=%d, AvgAct=%d\n", ct->m_nMBN, (MS_S32)fNBits, (MS_S32)fAvgAct ));
948*53ee8cc1Swenshuai.xi
949*53ee8cc1Swenshuai.xi //////////////////////////////////////////////////////////////////////////
950*53ee8cc1Swenshuai.xi // er_en
951*53ee8cc1Swenshuai.xi if (ct->m_nVPSize) {
952*53ee8cc1Swenshuai.xi if (ct->m_nUsedBits[0]-ct->m_nLastVPBits>=ct->m_nVPSize) {
953*53ee8cc1Swenshuai.xi if (ct->m_nVPMbRow) {
954*53ee8cc1Swenshuai.xi nMBX = ct->m_nMBN%ct->m_nMbWidth;
955*53ee8cc1Swenshuai.xi nMBY = ct->m_nMBN/ct->m_nMbWidth;
956*53ee8cc1Swenshuai.xi if (nMBX==0 && nMBY>0 && (nMBY%ct->m_nVPMbRow)==0) {
957*53ee8cc1Swenshuai.xi ct->m_nNewPacket = 1;
958*53ee8cc1Swenshuai.xi ct->m_nLastVPBits = ct->m_nUsedBits[0];
959*53ee8cc1Swenshuai.xi }
960*53ee8cc1Swenshuai.xi }
961*53ee8cc1Swenshuai.xi else {
962*53ee8cc1Swenshuai.xi ct->m_nNewPacket = 1;
963*53ee8cc1Swenshuai.xi ct->m_nLastVPBits = ct->m_nUsedBits[0];
964*53ee8cc1Swenshuai.xi }
965*53ee8cc1Swenshuai.xi }
966*53ee8cc1Swenshuai.xi }
967*53ee8cc1Swenshuai.xi else if (ct->m_nVPMbRow) {
968*53ee8cc1Swenshuai.xi nMBX = ct->m_nMBN%ct->m_nMbWidth;
969*53ee8cc1Swenshuai.xi nMBY = ct->m_nMBN/ct->m_nMbWidth;
970*53ee8cc1Swenshuai.xi if (nMBX==0 && nMBY>0 && (nMBY%ct->m_nVPMbRow)==0) {
971*53ee8cc1Swenshuai.xi ct->m_nNewPacket = 1;
972*53ee8cc1Swenshuai.xi ct->m_nLastVPBits = ct->m_nUsedBits[0];
973*53ee8cc1Swenshuai.xi }
974*53ee8cc1Swenshuai.xi }
975*53ee8cc1Swenshuai.xi }
976*53ee8cc1Swenshuai.xi
977*53ee8cc1Swenshuai.xi if (ct->m_rcGranularity==FRAMELEVELRC) {
978*53ee8cc1Swenshuai.xi RtnQP = rcQstep2QP(ct, ct->m_nFrameQStep);
979*53ee8cc1Swenshuai.xi }
980*53ee8cc1Swenshuai.xi
981*53ee8cc1Swenshuai.xi ct->m_nMBN++;
982*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONST_QUALITY)
983*53ee8cc1Swenshuai.xi return ct->m_nConstQP;
984*53ee8cc1Swenshuai.xi MS_ASSERT(RtnQP<=ct->m_nMaxQP && RtnQP>=ct->m_nMinQP);
985*53ee8cc1Swenshuai.xi return RtnQP;
986*53ee8cc1Swenshuai.xi }
987*53ee8cc1Swenshuai.xi
988*53ee8cc1Swenshuai.xi #else // _SUPPORT_MBLEVEL_RC_ not defined
989*53ee8cc1Swenshuai.xi
cvbr_InitMB(CVBRRateControl * ct,MS_S32 nVar,const MS_S32 nPrevQP,const MS_S32 nBits,MS_S32 IsIntra,MS_S32 IsP4MV,MS_S32 BPredType,MS_S32 nResetDQ)990*53ee8cc1Swenshuai.xi MS_S32 cvbr_InitMB(CVBRRateControl* ct, MS_S32 nVar, const MS_S32 nPrevQP, const MS_S32 nBits, MS_S32 IsIntra, MS_S32 IsP4MV, MS_S32 BPredType, MS_S32 nResetDQ)
991*53ee8cc1Swenshuai.xi {
992*53ee8cc1Swenshuai.xi if (ct->m_nMBN>0)
993*53ee8cc1Swenshuai.xi ct->m_nLastFrameAvgQStep[ct->m_bIsBotField] += rcQP2Qstep(nPrevQP);
994*53ee8cc1Swenshuai.xi if (ct->m_rcMethod==CONST_QUALITY)
995*53ee8cc1Swenshuai.xi return ct->m_nConstQP;
996*53ee8cc1Swenshuai.xi return rcQstep2QP(ct->m_nFrameQStep);
997*53ee8cc1Swenshuai.xi }
998*53ee8cc1Swenshuai.xi
999*53ee8cc1Swenshuai.xi #endif // _SUPPORT_MBLEVEL_RC_
1000