/***************************************************************************** * Copyright (C) 2013 x265 project * * Authors: Steve Borho * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA. * * This program is also available under a commercial proprietary license. * For more information, contact us at license @ x265.com. *****************************************************************************/ #ifndef X265_ENTROPY_H #define X265_ENTROPY_H #include "common.h" #include "bitstream.h" #include "frame.h" #include "slice.h" #include "TLibCommon/ContextTables.h" namespace x265 { // private namespace struct SaoLcuParam; struct EstBitsSbac; class TComDataCU; class ScalingList; enum SplitType { DONT_SPLIT = 0, VERTICAL_SPLIT = 1, QUAD_SPLIT = 2, NUMBER_OF_SPLIT_MODES = 3 }; struct TURecurse { uint32_t section; uint32_t splitMode; uint32_t absPartIdxTURelCU; uint32_t absPartIdxStep; TURecurse(SplitType splitType, uint32_t _absPartIdxStep, uint32_t _absPartIdxTU) { static const uint32_t partIdxStepShift[NUMBER_OF_SPLIT_MODES] = { 0, 1, 2 }; section = 0; absPartIdxTURelCU = _absPartIdxTU; splitMode = (uint32_t)splitType; absPartIdxStep = _absPartIdxStep >> partIdxStepShift[splitMode]; } bool isNextSection() { if (splitMode == DONT_SPLIT) { section++; return false; } else { absPartIdxTURelCU += absPartIdxStep; section++; return section < (uint32_t)(1 << splitMode); } } bool isLastSection() const { return (section + 1) >= (uint32_t)(1 << splitMode); } }; struct EstBitsSbac { int significantCoeffGroupBits[NUM_SIG_CG_FLAG_CTX][2]; int significantBits[NUM_SIG_FLAG_CTX][2]; int lastXBits[10]; int lastYBits[10]; int greaterOneBits[NUM_ONE_FLAG_CTX][2]; int levelAbsBits[NUM_ABS_FLAG_CTX][2]; int blockCbpBits[NUM_QT_CBF_CTX][2]; int blockRootCbpBits[2]; }; class Entropy : public SyntaxElementWriter { public: uint64_t m_pad; uint8_t m_contextState[160]; // MAX_OFF_CTX_MOD + padding /* CABAC state */ uint32_t m_low; uint32_t m_range; uint32_t m_bufferedByte; int m_numBufferedBytes; int m_bitsLeft; uint64_t m_fracBits; EstBitsSbac m_estBitsSbac; Entropy(); void setBitstream(Bitstream* p) { m_bitIf = p; } bool isBitCounter() const { return !m_bitIf; } uint32_t getNumberOfWrittenBits() { X265_CHECK(!m_bitIf, "bit counting mode expected\n"); return uint32_t(m_fracBits >> 15); } void zeroFract() { m_fracBits = 0; } void resetBits(); void resetEntropy(Slice *slice); // SBAC RD void load(Entropy& src); void loadIntraDirModeLuma(Entropy& src); void store(Entropy& dest); void loadContexts(Entropy& src) { copyContextsFrom(src); } void copyState(Entropy& other); void codeVPS(VPS* vps); void codeSPS(SPS* sps, ScalingList *scalingList, ProfileTierLevel *ptl); void codePPS(PPS* pps); void codeVUI(VUI* vui); void codeAUD(Slice *slice); void codeHrdParameters(HRDInfo* hrd); void codeSliceHeader(Slice* slice); void codeSliceHeaderWPPEntryPoints(Slice* slice, uint32_t *substreamSizes, uint32_t maxOffset); void codeShortTermRefPicSet(RPS* rps); void codeSliceFinish() { finish(); } void codeTerminatingBit(uint32_t lsLast) { encodeBinTrm(lsLast); } void encodeCU(TComDataCU* cu); void codeSaoOffset(SaoLcuParam* saoLcuParam, uint32_t compIdx); void codeSaoUnitInterleaving(int compIdx, bool saoFlag, int rx, int ry, SaoLcuParam* saoLcuParam, int cuAddrInSlice, int cuAddrUpInSlice, int allowMergeLeft, int allowMergeUp); void codeSaoMerge(uint32_t code) { encodeBin(code, m_contextState[OFF_SAO_MERGE_FLAG_CTX]); } void codeCUTransquantBypassFlag(TComDataCU* cu, uint32_t absPartIdx); void codeSkipFlag(TComDataCU* cu, uint32_t absPartIdx); void codeMergeFlag(TComDataCU* cu, uint32_t absPartIdx); void codeMergeIndex(TComDataCU* cu, uint32_t absPartIdx); void codeSplitFlag(TComDataCU* cu, uint32_t absPartIdx, uint32_t depth); void codeMVPIdx(uint32_t symbol); void codeMvd(TComDataCU* cu, uint32_t absPartIdx, int list); void codePartSize(TComDataCU* cu, uint32_t absPartIdx, uint32_t depth); void codePredMode(TComDataCU* cu, uint32_t absPartIdx); void codePredInfo(TComDataCU* cu, uint32_t absPartIdx); void codeTransformSubdivFlag(uint32_t symbol, uint32_t ctx); void codeQtCbf(TComDataCU* cu, uint32_t absPartIdx, uint32_t absPartIdxStep, uint32_t width, uint32_t height, TextType ttype, uint32_t trDepth, bool lowestLevel); void codeQtCbf(TComDataCU* cu, uint32_t absPartIdx, TextType ttype, uint32_t trDepth); void codeQtCbfZero(TComDataCU* cu, TextType ttype, uint32_t trDepth); void codeQtRootCbfZero(TComDataCU* cu); void codeCoeff(TComDataCU* cu, uint32_t absPartIdx, uint32_t depth, bool& bCodeDQP); void codeCoeffNxN(TComDataCU* cu, coeff_t* coef, uint32_t absPartIdx, uint32_t log2TrSize, TextType ttype); void codeIntraDirLumaAng(TComDataCU* cu, uint32_t absPartIdx, bool isMultiple); void codeIntraDirChroma(TComDataCU* cu, uint32_t absPartIdx); // RDO functions void estBit(EstBitsSbac& estBitsSbac, uint32_t log2TrSize, bool bIsLuma); void estCBFBit(EstBitsSbac& estBitsSbac); void estSignificantCoeffGroupMapBit(EstBitsSbac& estBitsSbac, bool bIsLuma); void estSignificantMapBit(EstBitsSbac& estBitsSbac, uint32_t log2TrSize, bool bIsLuma); void estSignificantCoefficientsBit(EstBitsSbac& estBitsSbac, bool bIsLuma); private: /* CABAC private methods */ void start(); void finish(); void encodeBin(uint32_t binValue, uint8_t& ctxModel); void encodeBinEP(uint32_t binValue); void encodeBinsEP(uint32_t binValues, int numBins); void encodeBinTrm(uint32_t binValue); void encodeCU(TComDataCU* cu, uint32_t absPartIdx, uint32_t depth, bool bInsidePicture, bool& bEncodeDQP); void finishCU(TComDataCU* cu, uint32_t absPartIdx, uint32_t depth); void writeOut(); /* SBac private methods */ void writeUnaryMaxSymbol(uint32_t symbol, uint8_t* scmModel, int offset, uint32_t maxSymbol); void writeEpExGolomb(uint32_t symbol, uint32_t count); void writeCoefRemainExGolomb(uint32_t symbol, const uint32_t absGoRice); void codeProfileTier(ProfileTierLevel& ptl); void codeScalingList(ScalingList*); void codeScalingList(ScalingList* scalingList, uint32_t sizeId, uint32_t listId); void codePredWeightTable(Slice* slice); void codeInterDir(TComDataCU* cu, uint32_t absPartIdx); void codePUWise(TComDataCU* cu, uint32_t absPartIdx); void codeQtRootCbf(TComDataCU* cu, uint32_t absPartIdx); void codeRefFrmIdxPU(TComDataCU* cu, uint32_t absPartIdx, int list); void codeRefFrmIdx(TComDataCU* cu, uint32_t absPartIdx, int list); void codeSaoMaxUvlc(uint32_t code, uint32_t maxSymbol); void codeSaoTypeIdx(uint32_t code); void codeSaoUflc(uint32_t length, uint32_t code) { encodeBinsEP(code, length); } void codeSAOSign(uint32_t code) { encodeBinEP(code); } void codeDeltaQP(TComDataCU* cu, uint32_t absPartIdx); void codeLastSignificantXY(uint32_t posx, uint32_t posy, uint32_t log2TrSize, bool bIsLuma, uint32_t scanIdx); void codeTransformSkipFlags(TComDataCU* cu, uint32_t absPartIdx, uint32_t trSize, TextType ttype); struct CoeffCodeState { uint32_t bakAbsPartIdx; uint32_t bakChromaOffset; uint32_t bakAbsPartIdxCU; }; void encodeTransform(TComDataCU* cu, CoeffCodeState& state, uint32_t offsetLumaOffset, uint32_t offsetChroma, uint32_t absPartIdx, uint32_t absPartIdxStep, uint32_t depth, uint32_t log2TrSize, uint32_t uiTrIdx, bool& bCodeDQP); void copyFrom(Entropy& src); void copyContextsFrom(Entropy& src); }; } #endif // ifndef X265_ENTROPY_H