不多说,直接上代码,主要是参考K0103提案看的。
/* The copyright in this software is being made available under the BSD
* License, included below. This software may be subject to other third party
* and contributor rights, including patent rights, and no such rights are
* granted under this license.
*
* Copyright (c) 2010-2013, ITU/ISO/IEC
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
* be used to endorse or promote products derived from this software without
* specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
/** \file TEncRateCtrl.h
\brief Rate control manager class
*/
#ifndef __TENCRATECTRL__
#define __TENCRATECTRL__
#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include "../TLibCommon/CommonDef.h"
#include "../TLibCommon/TComDataCU.h"
#include <vector>
#include <algorithm>
using namespace std;
//! \ingroup TLibEncoder
//! \{
#if RATE_CONTROL_LAMBDA_DOMAIN
#include "../TLibEncoder/TEncCfg.h"
#include <list>
#include <cassert>
const Int g_RCInvalidQPValue = -999;
const Int g_RCSmoothWindowSize = 40;
const Int g_RCMaxPicListSize = 32;
const Double g_RCWeightPicTargetBitInGOP = 0.9;
const Double g_RCWeightPicRargetBitInBuffer = 1.0 - g_RCWeightPicTargetBitInGOP;
#if M0036_RC_IMPROVEMENT
const Int g_RCIterationNum = 20;
const Double g_RCWeightHistoryLambda = 0.5;
const Double g_RCWeightCurrentLambda = 1.0 - g_RCWeightHistoryLambda;
const Int g_RCLCUSmoothWindowSize = 4;
const Double g_RCAlphaMinValue = 0.05;
const Double g_RCAlphaMaxValue = 500.0;
const Double g_RCBetaMinValue = -3.0;
const Double g_RCBetaMaxValue = -0.1;
#endif
#if RATE_CONTROL_INTRA
#define ALPHA 6.7542;
#define BETA1 1.2517
#define BETA2 1.7860
#endif
struct TRCLCU
{
Int m_actualBits;
Int m_QP; // QP of skip mode is set to g_RCInvalidQPValue
Int m_targetBits;
Double m_lambda;
#if M0036_RC_IMPROVEMENT
Double m_bitWeight;
#else
Double m_MAD;
#endif
Int m_numberOfPixel;//指当前LCU的像素数
#if RATE_CONTROL_INTRA
Double m_costIntra;
Int m_targetBitsLeft;
#endif
};
struct TRCParameter
{
Double m_alpha;
Double m_beta;
};
class TEncRCSeq
{
public:
TEncRCSeq();
~TEncRCSeq();
public:
//主要是初始化EC这个类
#if M0036_RC_IMPROVEMENT
Void create( Int totalFrames, Int targetBitrate, Int frameRate, Int GOPSize, Int picWidth, Int picHeight, Int LCUWidth, Int LCUHeight, Int numberOfLevel, Bool useLCUSeparateModel, Int adaptiveBit );
#else
Void create( Int totalFrames, Int targetBitrate, Int frameRate, Int GOPSize, Int picWidth, Int picHeight, Int LCUWidth, Int LCUHeight, Int numberOfLevel, Bool useLCUSeparateModel );
#endif
Void destroy();
Void initBitsRatio( Int bitsRatio[] );
Void initGOPID2Level( Int GOPID2Level[] );
Void initPicPara( TRCParameter* picPara = NULL ); // NULL to initial