xref: /utopia/UTPA2-700.0.x/modules/mfe/drv/mfe_ex/cModel/msRateCtrl.c (revision 53ee8cc121a030b8d368113ac3e966b4705770ef)
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