注:问号以及未注释部分 会在x265-1.8版本内更新
/*****************************************************************************
* Copyright (C) 2015 x265 project
*
* Authors: Steve Borho <steve@borho.org>
*
* 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_CUDATA_H
#define X265_CUDATA_H
#include "common.h"
#include "slice.h"
#include "mv.h"
namespace x265 {
// private namespace
class FrameData;
class Slice;
struct TUEntropyCodingParameters;
struct CUDataMemPool;
enum PartSize//PU划分方式
{
SIZE_2Nx2N, // symmetric motion partition, 2Nx2N
SIZE_2NxN, // symmetric motion partition, 2Nx N
SIZE_Nx2N, // symmetric motion partition, Nx2N
SIZE_NxN, // symmetric motion partition, Nx N
SIZE_2NxnU, // asymmetric motion partition, 2Nx( N/2) + 2Nx(3N/2)
SIZE_2NxnD, // asymmetric motion partition, 2Nx(3N/2) + 2Nx( N/2)
SIZE_nLx2N, // asymmetric motion partition, ( N/2)x2N + (3N/2)x2N
SIZE_nRx2N, // asymmetric motion partition, (3N/2)x2N + ( N/2)x2N
NUM_SIZES
};
enum PredMode//预测类型
{
MODE_NONE = 0,
MODE_INTER = (1 << 0),
MODE_INTRA = (1 << 1),
MODE_SKIP = (1 << 2) | MODE_INTER
};
// motion vector predictor direction used in AMVP
enum MVP_DIR
{
MD_LEFT = 0, // MVP of left block
MD_ABOVE, // MVP of above block
MD_ABOVE_RIGHT, // MVP of above right block
MD_BELOW_LEFT, // MVP of below left block
MD_ABOVE_LEFT, // MVP of above left block
MD_COLLOCATED // MVP of temporal neighbour
};
struct CUGeom//CU的几何信息 在FrameEncoder类被引用
{
enum {
INTRA = 1<<0, // 暂无用到 CU is intra predicted
PRESENT = 1<<1, // CU不是完全在图像外(只要有一部分在图像内就有此标志位)值 2 CU is not completely outside the frame
SPLIT_MANDATORY = 1<<2, // CU节点为被推荐划分(一般是处于边界情况,当前CU一部分内容在图像外面)值为4 CU split is mandatory if CU is inside frame and can be split
LEAF = 1<<3, // CU是叶子节点,即是8x8块(默认配置最小CU是8x8的情况下)值为 8 CU is a leaf node of the CTU
SPLIT = 1<<4, // 确定划分 SPLIT_MANDATORY 出现一定出现SPLIT CU is currently split in four child CUs.
};
// (1 + 4 + 16 + 64) = 85.
enum { MAX_GEOMS = 85 };//一个Geom的个数 (1 + 4 + 16 + 64) = 85.
uint32_t log2CUSize; // 当前的CUsize Log of the CU size.
uint32_t childOffset; // 表示从当前位置到第一个子cU的偏移量 offset of the first child CU from current CU
uint32_t absPartIdx; // 当前CU在LCU中4x4 zizag地址 Part index of this CU in terms of 4x4 blocks.
uint32_t numPartitions; // 当前CU有多少4x4块 Number of 4x4 blocks in the CU
uint32_t flags; // 当前CU的块状态 CU flags.
uint32_t depth; // 当前CU的深度 depth of this CU relative from CTU
};
/*
样例-Y
以MaxCU 64 MinCU 8 width 64 heigh 240, 为例
申请空间: allocGeoms * CUGeom::MAX_GEOMS = allocGeoms * 85 = 4*85
64x64:1
32x32:4
16x16:16
8x 8:64 1+4+16+64 = 85
数据详情:CUGeom.txt
数据排列方式是先64x64、32x32、32x32、32x32、32x32、16x16……
0—84:
64x64、32x32、16x16 的flag 为2 :PRESENT 8x8为:10 :1010, PRESENT和LEAF
85-169: 表示宽度不能整除CTU 。例如当前 416 = 6*64+32
64x64:flag为22 10110, SPLIT、PRESENT、SPLIT_MANDATORY
32x32:左边两个为 2: PRESENT ,右边两个为:0
16x16:最右边扩充的为0。其它为2
8x8:正常的为10 PRESENT和LEAF, 扩边的为8:LEAF
170-254: 表示高度不能整除CTU 。例如当前 240 = 3*64+48
64x64:flag为22 10110, SPLIT、PRESENT、SPLIT_MANDATORY
32x32:上边两个为 2: PRESENT ,下边两个为:22 10110, SPLIT、PRESENT、SPLIT_MANDATORY 因为当前不足32x32
16x16:最下边扩充的为0。其它为2
8x8:正常的为10 PRESENT和LEAF, 扩边的为8:LEAF
255-339: 表示宽度不能整除CTU 且高度不能整除CTU
64x64:flag为22 10110, SPLIT、PRESENT、SPLIT_MANDATORY
32x32:左上角一个为 2: PRESENT ,右边两个为0,左下角为:22 10110, SPLIT、PRESENT、SPLIT_MANDATORY 因为当前不足32x32
16x16:右边和下边扩充的为0。其它为2
8x8:正常的为10 PRESENT和LEAF, 扩边的为8:LEAF
**/
struct MVField
{
MV mv;
int refIdx;
};
// Structure that keeps the neighbour's MV information.
struct InterNeighbourMV
{
// Neighbour MV. The index represents the list.
MV mv[2];
// Collocated right bottom CU addr.
uint32_t cuAddr[2];
// For spatial prediction, this field contains the reference index
// in each list (-1 if not available).
//
// For temporal prediction, the first value is used for the
// prediction with list 0. The second value is used for the prediction
// with list 1. For each value, the first four bits are the reference index
// associated to the PMV, and the fifth bit is the list associated to the PMV.
// if both reference indices are -1, then unifiedRef is also -1
union { int16_t refIdx[2]; int32_t unifiedRef; };
};
typedef void(*cucopy_t)(uint8_t* dst, uint8_t* src); // dst and src are aligned to MIN(size, 32)
typedef void(*cubcast_t)(uint8_t* dst, uint8_t val); // dst is aligned to MIN(size, 32)
// Partition count table, index represents partitioning mode.
const uint32_t nbPartsTable[8] = { 1, 2, 2, 4, 2, 2, 2, 2 };
// Holds part data for a CU of a given size, from an 8x8 CU to a CTU
class CUData
{
public:
static cubcast_t s_partSet[NUM_FULL_DEPTH]; // 存储set函数指针根据CTU大小确定,如CTU 64 [0]存储一次性set 16x16(因为当前有16x16 个4x4块)大小的函数指针 [1]存储一次性set 8x8(因为当前有8x8 个4x4块)大小的函数指针 ....pointer to broadcast set functions per absolute depth
static uint32_t s_numPartInCUSize; // 存储一个CTU中每行或者列有多少4x4块
FrameData* m_encData;
const Slice* m_slice;
cucopy_t m_partCopy; // 当前CU的copy函数指针 pointer to function that copies m_numPartitions elements
cubcast_t m_partSet; // 当前CU的set函数指针 pointer to function that sets m_numPartitions elements
cucopy_t m_subPartCopy; // 当前CU子块的copy函数指针 pointer to function that copies m_numPartitions/4 elements, may be NULL
cubcast_t m_subPartSet; // 当前CU子块的set函数指针 pointer to function that sets m_numPartitions/4 elements, may be NULL
uint32_t m_cuAddr; // CTU在帧中的编号 address of CTU within the picture in raster order
uint32_t m_absIdxInCTU; // 当前CU左上角在CTU4x4块中的zigzag标号 address of CU within its CTU in Z scan order
uint32_t m_cuPelX; // 当前CU左上角像素点在图像中的横向偏移像素个数 CU position within the picture, in pixels (X)
uint32_t m_cuPelY; // 当前CU左上角像素点在图像中的纵向偏移像素个数CU position within the picture, in pixels (Y)
uint32_t m_numPartitions; // 当前CU的4x4块个数 maximum number of 4x4 partitions within this CU
uint32_t m_chromaFormat; // 如:420格式,色度宽高是亮度的1/2 ,则此处为1,表示需要右移1位
uint32_t m_hChromaShift; // 如:420格式,色度宽高是亮度的1/2 ,则此处为1,表示需要右移1位
uint32_t m_vChromaShift; // 如:420格式,色度宽高是亮度的1/2 ,则此处为1,表示需要右移1位
/* Per-part data, stored contiguously */
//以下所有空间在CUDataMemPool中(以下全部按照4x4块存储)
int8_t* m_qp; // 存储QP信息 array of QP values
uint8_t* m_log2CUSize; // 存储当前CU的CUsize 2^size = 真实大小 如16x16: 当前为4 array of cu log2Size TODO: seems redundant to depth
uint8_t* m_lumaIntraDir; // intra块的亮度预测方向 array of intra directions (luma)
uint8_t* m_tqBypass; // 存储当前是否应用transquant bypass 对残差不进行变换、量化及环路滤波 array of CU lossless flags
int8_t* m_refIdx[2]; // 存储在list中参考帧的序号 array of motion reference indices per list
uint8_t* m_cuDepth; // 存储当前CU所处的划分深度 array of depths
uint8_t* m_predMode; // 当前CU的类型:MODE_NONE、MODE_INTER、MODE_INTRA、MODE_SKIP array of prediction modes
uint8_t* m_partSize; // 当前CU的划分方式:SIZE_2Nx2N、SIZE_2NxN、SIZE_Nx2N、SIZE_NxN、SIZE_2NxnU、SIZE_2NxnD、SIZE_nLx2N、SIZE_nRx2Narray of partition sizes
uint8_t* m_mergeFlag; // 当前PU是否应用MERGE模式 array of merge flags
uint8_t* m_interDir; // 存储当前所用的list信息:0:intra、1:list0 2:list2 3:bi模式 array of inter directions
uint8_t* m_mvpIdx[2]; // 每个PU左上角的4x4块中存储了MVP选择的标号 [0]:list 0 [1] : list1 array of motion vector predictor candidates or merge candidate indices [0]
uint8_t* m_tuDepth; // 相对于当前CU下的TU深度 array of transform indices
uint8_t* m_transformSkip[3]; // 标记当前是否应用transform skip array of transform skipping flags per plane
uint8_t* m_cbf[3]; // 标记当前PU是否拥有残差 array of coded block flags (CBF) per plane
uint8_t* m_chromaIntraDir; // 存储intra色度块的预测方向 array of intra directions (chroma)
enum { BytesPerPartition = 21 }; // 各个模式的种数 combined sizeof() of all per-part data
coeff_t* m_trCoeff[3]; // transformed coefficient buffer per plane
MV* m_mv[2]; // 存储相应PU的MV array of motion vectors per list
MV* m_mvd[2]; // 存储相应PU的MVD(注:skip块此位置不存信息)array of coded motion vector deltas per list
enum { TMVP_UNIT_MASK = 0xF0 }; // mask for mapping index to into a compressed (reference) MV field
const CUData* m_cuAboveLeft; // pointer to above-left neighbor CTU
const CUData* m_cuAboveRight; // pointer to above-right neighbor CTU
const CUData* m_cuAbove; // pointer to above neighbor CTU
const CUData* m_cuLeft; // pointer to left neighbor CTU
CUData();//初始化
/** 函数功能 :初始化函数指针,获取CTU数据相应存储位置
/* 调用范围 :只在FrameData::create和Analysis::create函数中被调用
* \参数 dataPool :CU存储空间
* \参数 depth :CU划分深度(FrameData::create 传入 0 Analysis::create 传入相应深度)
* \参数 csp :图像格式
* \参数 instance :每个CU的标号
* \返回 :null* */
void initialize(const CUDataMemPool& dataPool, uint32_t depth, int csp, int instance);
/** 函数功能 : 计算CU的几何信息
/* 调用范围 : 只在FrameEncoder::initializeGeoms()函数中被调用
* \参数 ctuWidth : 宽度的余数
* \参数 ctuHeight : 高度的余数
* \参数 maxCUSize : 最大CU
* \参数 minCUSize : 最小CU
* \参数 cuDataArray: 存储位置
* 返回值 : null
**/
static void calcCTUGeoms(uint32_t ctuWidth, uint32_t ctuHeight, uint32_t maxCUSize, uint32_t minCUSize, CUGeom cuDataArray[CUGeom::MAX_GEOMS]);
void initCTU(const Frame& frame, uint32_t cuAddr, int qp);
/** 函数功能 : 根据cuGeom初始化当前子块的信息
* \参数 ctu : 上层大块
* \参数 cuGeom : 当前子块对应的一些几何(深度,位置等信息)信息
* \参数 qp : 当前的qp信息
* 返回值 : null**/
void initSubCU(const CUData& ctu, const CUGeom& cuGeom, int qp);
void initLosslessCU(const CUData& cu, const CUGeom& cuGeom);
/** 函数功能 : 将子块决策最优结果copy到父块对应位置中
* \参数 subCU : 当前子块
* \参数 childGeom : 当前子块的几何信息
* \参数 subPartIdx : 当前子块的标号
* 返回值 : null**/
void copyPartFrom(const CUData& cu, const CUGeom& childGeom, uint32_t subPartIdx);
/** 函数功能 : 当前子块全部在图像外,设置当前子块的depth和块大小值
* \参数 childGeom : 当前子块的几何信息
* \参数 subPartIdx : 当前子块的标号
* 返回值 : null**/
void setEmptyPart(const CUGeom& childGeom, uint32_t subPartIdx);
/** 函数功能 : 将当前cu决策信息赋值到CTU对应位置中
* \参数 depth : 当前的划分深度
* 返回值 : null**/
void copyToPic(uint32_t depth) const;
/* RD-0 methods called only from encodeResidue */
void copyFromPic(const CUData& ctu, const CUGeom& cuGeom);
void updatePic(uint32_t depth) const;
/** 函数功能 : 设置划分方式:SIZE_2Nx2N、SIZE_2NxN、SIZE_Nx2N、SIZE_NxN、SIZE_2NxnU、SIZE_2NxnD、SIZE_nLx2N、SIZE
* \参数 size : 划分方式:SIZE_2Nx2N、SIZE_2NxN、SIZE_Nx2N、SIZE_NxN、SIZE_2NxnU、SIZE_2NxnD、SIZE_nLx2N、SIZE
* 返回值 : null**/
void setPartSizeSubParts(PartSize size) { m_partSet(m_partSize, (uint8_t)size); }
/** 函数功能 : 设置CU类型:MODE_NONE、MODE_INTER、MODE_INTRA、MODE_SKIP
* \参数 mode : CU类型:MODE_NONE、MODE_INTER、MODE_INTRA、MODE_SKIP
* 返回值 : null**/
void setPredModeSubParts(PredMode mode) { m_partSet(m_predMode, (uint8_t)mode); }
void clearCbf() { m_partSet(m_cbf[0], 0); m_partSet(m_cbf[1], 0); m_partSet(m_cbf[2], 0); }
/* these functions all take depth as an absolute depth from CTU, it is used to calculate the number of parts to copy */
void setQPSubParts(int8_t qp, uint32_t absPartIdx, uint32_t depth) { s_partSet[depth]((uint8_t*)m_qp + absPartIdx, (uint8_t)qp); }
void setTUDepthSubParts(uint8_t tuDepth, uint32_t absPartIdx, uint32_t depth) { s_partSet[depth](m_tuDepth + absPartIdx, tuDepth); }
void setLumaIntraDirSubParts(uint8_t dir, uint32_t absPartIdx, uint32_t depth) { s_partSet[depth](m_lumaIntraDir + absPartIdx, dir); }
void setChromIntraDirSubParts(uint8_t dir, uint32_t absPartIdx, uint32_t depth) { s_partSet[depth](m_chromaIntraDir + absPartIdx, dir); }
void setCbfSubParts(uint8_t cbf, TextType ttype, uint32_t absPartIdx, uint32_t depth) { s_partSet[depth](m_cbf[ttype] + absPartIdx, cbf); }
void setCbfPartRange(uint8_t cbf, TextType ttype, uint32_t absPartIdx, uint32_t coveredPartIdxes) { memset(m_cbf[ttype] + absPartIdx, cbf, coveredPartIdxes); }
void setTransformSkipSubParts(uint8_t tskip, TextType ttype, uint32_t absPartIdx, uint32_t depth) { s_partSet[depth](m_transformSkip[ttype] + absPartIdx, tskip); }
void setTransformSkipPartRange(uint8_t tskip, TextType ttype, uint32_t absPartIdx, uint32_t coveredPartIdxes) { memset(m_transformSkip[ttype] + absPartIdx, tskip, coveredPartIdxes); }
bool setQPSubCUs(int8_t qp, uint32_t absPartIdx, uint32_t depth);
void setPUInterDir(uint8_t dir, uint32_t absPartIdx, uint32_t puIdx);
void setPUMv(int list, const MV& mv, int absPartIdx, int puIdx);
void setPURefIdx(int list, int8_t refIdx, int absPartIdx, int puIdx);
uint8_t getCbf(uint32_t absPartIdx, TextType ttype, uint32_t tuDepth) const { return (m_cbf[ttype][absPartIdx] >> tuDepth) & 0x1; }
uint8_t getQtRootCbf(uint32_t absPartIdx) const { return m_cbf[0][absPartIdx] || m_cbf[1][absPartIdx] || m_cbf[2][absPartIdx]; }
int8_t getRefQP(uint32_t currAbsIdxInCTU) const;
uint32_t getInterMergeCandidates(uint32_t absPartIdx, uint32_t puIdx, MVField (*candMvField)[2], uint8_t* candDir) const;
void clipMv(MV& outMV) const;
int getPMV(InterNeighbourMV *neighbours, uint32_t reference_list, uint32_t refIdx, MV* amvpCand, MV* pmv) const;
void getNeighbourMV(uint32_t puIdx, uint32_t absPartIdx, InterNeighbourMV* neighbours) const;
/** 函数功能 : 分别获取intraCU下最小TU,最大TU
* \参数 tuDepthRange[2]: 分别存储最小TU,最大TU
* \参数 absPartIdx : 在Entropy::encodeCU为当前CU的左上角标号,其它位置为0
* 返回值 : null
**/
void getIntraTUQtDepthRange(uint32_t tuDepthRange[2], uint32_t absPartIdx) const;
void getInterTUQtDepthRange(uint32_t tuDepthRange[2], uint32_t absPartIdx) const;
uint32_t getNumPartInter() const { return nbPartsTable[(int)m_partSize[0]]; }
bool isIntra(uint32_t absPartIdx) const { return m_predMode[absPartIdx] == MODE_INTRA; }
bool isInter(uint32_t absPartIdx) const { return !!(m_predMode[absPartIdx] & MODE_INTER); }
bool isSkipped(uint32_t absPartIdx) const { return m_predMode[absPartIdx] == MODE_SKIP; }
bool isBipredRestriction() const { return m_log2CUSize[0] == 3 && m_partSize[0] != SIZE_2Nx2N; }
void getPartIndexAndSize(uint32_t puIdx, uint32_t& absPartIdx, int& puWidth, int& puHeight) const;
void getMvField(const CUData* cu, uint32_t absPartIdx, int picList, MVField& mvField) const;
void getAllowedChromaDir(uint32_t absPartIdx, uint32_t* modeList) const;
int getIntraDirLumaPredictor(uint32_t absPartIdx, uint32_t* intraDirPred) const;
uint32_t getSCUAddr() const { return (m_cuAddr << g_unitSizeDepth * 2) + m_absIdxInCTU; }
uint32_t getCtxSplitFlag(uint32_t absPartIdx, uint32_t depth) const;
uint32_t getCtxSkipFlag(uint32_t absPartIdx) const;
void getTUEntropyCodingParameters(TUEntropyCodingParameters &result, uint32_t absPartIdx, uint32_t log2TrSize, bool bIsLuma) const;
const CUData* getPULeft(uint32_t& lPartUnitIdx, uint32_t curPartUnitIdx) const;
const CUData* getPUAbove(uint32_t& aPartUnitIdx, uint32_t curPartUnitIdx) const;
const CUData* getPUAboveLeft(uint32_t& alPartUnitIdx, uint32_t curPartUnitIdx) const;
const CUData* getPUAboveRight(uint32_t& arPartUnitIdx, uint32_t curPartUnitIdx) const;
const CUData* getPUBelowLeft(uint32_t& blPartUnitIdx, uint32_t curPartUnitIdx) const;
const CUData* getQpMinCuLeft(uint32_t& lPartUnitIdx, uint32_t currAbsIdxInCTU) const;
const CUData* getQpMinCuAbove(uint32_t& aPartUnitIdx, uint32_t currAbsIdxInCTU) const;
const CUData* getPUAboveRightAdi(uint32_t& arPartUnitIdx, uint32_t curPartUnitIdx, uint32_t partUnitOffset) const;
const CUData* getPUBelowLeftAdi(uint32_t& blPartUnitIdx, uint32_t curPartUnitIdx, uint32_t partUnitOffset) const;
protected:
template<typename T>
void setAllPU(T *p, const T& val, int absPartIdx, int puIdx);
int8_t getLastCodedQP(uint32_t absPartIdx) const;
int getLastValidPartIdx(int absPartIdx) const;
bool hasEqualMotion(uint32_t absPartIdx, const CUData& candCU, uint32_t candAbsPartIdx) const;
/* Check whether the current PU and a spatial neighboring PU are in same merge region */
bool isDiffMER(int xN, int yN, int xP, int yP) const { return ((xN >> 2) != (xP >> 2)) || ((yN >> 2) != (yP >> 2)); }
// add possible motion vector predictor candidates
bool getDirectPMV(MV& pmv, InterNeighbourMV *neighbours, uint32_t picList, uint32_t refIdx) const;
bool getIndirectPMV(MV& outMV, InterNeighbourMV *neighbours, uint32_t reference_list, uint32_t refIdx) const;
void getInterNeighbourMV(InterNeighbourMV *neighbour, uint32_t partUnitIdx, MVP_DIR dir) const;
bool getColMVP(MV& outMV, int& outRefIdx, int picList, int cuAddr, int absPartIdx) const;
bool getCollocatedMV(int cuAddr, int partUnitIdx, InterNeighbourMV *neighbour) const;
MV scaleMvByPOCDist(const MV& inMV, int curPOC, int curRefPOC, int colPOC, int colRefPOC) const;
void deriveLeftRightTopIdx(uint32_t puIdx, uint32_t& partIdxLT, uint32_t& partIdxRT) const;
uint32_t deriveCenterIdx(uint32_t puIdx) const;
uint32_t deriveRightBottomIdx(uint32_t puIdx) const;
uint32_t deriveLeftBottomIdx(uint32_t puIdx) const;
};
// TU settings for entropy encoding
struct TUEntropyCodingParameters
{
const uint16_t *scan;
const uint16_t *scanCG;
ScanType scanType;
uint32_t log2TrSizeCG;
uint32_t firstSignificanceMapContext;
};
//用于CU的空间申请和释放
struct CUDataMemPool //存在于FrameData类和ModeDepth中
{
uint8_t* charMemBlock; //用存储模式等数据 (空间大小为:当前CU的4x4块个数*一帧CTU个数*21 或者 当前CU的4x4块个数*预测模式个数*21)
coeff_t* trCoeffMemBlock;//存储系数数据 (空间大小为:一帧CTU个数*CTU像素个数 或者 预测模式个数*CTU像素个数)
MV* mvMemBlock; //存储MV数据 (空间大小为:当前CU的4x4块个数*4*一帧CTU个数:4表示: MV[0]、MV[1]、MVD[0]、MVD[1])
CUDataMemPool() { charMemBlock = NULL; trCoeffMemBlock = NULL; mvMemBlock = NULL; }//初始化
/** 函数功能 : 申请空间内存
/* 调用范围 : 只 FrameData::create和Analysis::create函数中被调用
* \参数 depth : CU划分深度(FrameData::create 传入 0 Analysis::create 传入相应深度)
* \参数 csp : 图像格式
* \参数numInstances: 需要申请内存的CU个数(FrameData::create 传入 一帧中CU的个数 Analysis::create 传入 预测模式个数:14 PRED_MERGE,PRED_SKIP,PRED_INTRA, PRED_2Nx2N,PRED_BIDIR,PRED_Nx2N,PRED_2NxN.....)
* \返回 : 是否申请内存成功 */
bool create(uint32_t depth, uint32_t csp, uint32_t numInstances)
{
uint32_t numPartition = NUM_4x4_PARTITIONS >> (depth * 2);//获取当前CU的4x4块个数
uint32_t cuSize = g_maxCUSize >> depth; //获取当前CU的size
uint32_t sizeL = cuSize * cuSize; //获取当前CU的像素个数
uint32_t sizeC = sizeL >> (CHROMA_H_SHIFT(csp) + CHROMA_V_SHIFT(csp));//色度块的像素个数
CHECKED_MALLOC(trCoeffMemBlock, coeff_t, (sizeL + sizeC * 2) * numInstances);//申请系数存储空间
CHECKED_MALLOC(charMemBlock, uint8_t, numPartition * numInstances * CUData::BytesPerPartition);//申请模式存储空间
CHECKED_MALLOC(mvMemBlock, MV, numPartition * 4 * numInstances);//申请MV存储空间
return true;
fail:
return false;
}
/** 函数功能 : 释放空间内存
/* 调用范围 : 只FrameData::destroy()和Analysis::destroy()函数中被调用
* \返回 : null */
void destroy()
{
X265_FREE(trCoeffMemBlock);
X265_FREE(mvMemBlock);
X265_FREE(charMemBlock);
}
};
}
#endif // ifndef X265_CUDATA_H