diff options
| author | Casimir666 <casimir666@users.sourceforge.net> | 2008-01-05 14:02:10 +0300 |
|---|---|---|
| committer | Casimir666 <casimir666@users.sourceforge.net> | 2008-01-05 14:02:10 +0300 |
| commit | 2dc57897ae7300f113fa08d3f6806daab54321c5 (patch) | |
| tree | e293f518c4e2ef60e52b807d1c044c6e2418f83a /src/filters/transform/mpcvideodec/DXVADecoderH264.cpp | |
| parent | f4dbaea1a57ad2e0a255e4c64023f972a613f05c (diff) | |
Decode slice header modify, code cleanup
git-svn-id: https://mpc-hc.svn.sourceforge.net/svnroot/mpc-hc/trunk@360 10f7b99b-c216-0410-bff0-8a66a9350fd8
Diffstat (limited to 'src/filters/transform/mpcvideodec/DXVADecoderH264.cpp')
| -rw-r--r-- | src/filters/transform/mpcvideodec/DXVADecoderH264.cpp | 1176 |
1 files changed, 22 insertions, 1154 deletions
diff --git a/src/filters/transform/mpcvideodec/DXVADecoderH264.cpp b/src/filters/transform/mpcvideodec/DXVADecoderH264.cpp index 629fad95c..a33dec943 100644 --- a/src/filters/transform/mpcvideodec/DXVADecoderH264.cpp +++ b/src/filters/transform/mpcvideodec/DXVADecoderH264.cpp @@ -24,6 +24,9 @@ #include "DXVADecoderH264.h"
#include "MPCVideoDecFilter.h"
+#include "PODtypes.h"
+#include "avcodec.h"
+
extern "C"
{
#include "FfmpegContext.h"
@@ -66,21 +69,13 @@ CDXVADecoderH264::CDXVADecoderH264 (CMPCVideoDecFilter* pFilter, IDirectXVideoDe }
-CDXVADecoderH264::~CDXVADecoderH264()
-{
-// Flush();
-}
-
-
void CDXVADecoderH264::Init()
{
memset (&m_DXVAPicParams, 0, sizeof(m_DXVAPicParams));
memset (&m_SliceShort, 0, sizeof(m_SliceShort));
- memset (&m_PicParam, 0, sizeof(m_PicParam));
- memset (&m_SeqParam, 0, sizeof(m_SeqParam));
- memset (&m_Slice, 0, sizeof(m_Slice));
m_nCurRefFrame = 0;
+ m_nNALLength = 4;
switch (GetMode())
{
@@ -103,7 +98,7 @@ void CDXVADecoderH264::CopyBitstream(BYTE* pDXVABuffer, BYTE* pBuffer, UINT& nSi nSize = 0;
do
{
- ReadNalu (&Nalu, pDataSlice, nSliceSize, 4);
+ ReadNalu (&Nalu, pDataSlice, nSliceSize, m_nNALLength);
switch (Nalu.nal_unit_type)
{
@@ -116,7 +111,7 @@ void CDXVADecoderH264::CopyBitstream(BYTE* pDXVABuffer, BYTE* pBuffer, UINT& nSi pDXVABuffer[0]=pDXVABuffer[1]=pDXVABuffer[2]=0; pDXVABuffer[3]=1;
// Copy NALU
- memcpy (pDXVABuffer+4, (BYTE*)pDataSlice+4, Nalu.len-4);
+ memcpy (pDXVABuffer+4, (BYTE*)pDataSlice+m_nNALLength, Nalu.len-m_nNALLength);
pDXVABuffer += Nalu.len;
nSize += Nalu.len;
break;
@@ -137,6 +132,8 @@ void CDXVADecoderH264::CopyBitstream(BYTE* pDXVABuffer, BYTE* pBuffer, UINT& nSi void CDXVADecoderH264::Flush()
{
ClearRefFramesList();
+ m_DXVAPicParams.UsedForReferenceFlags = 0;
+
__super::Flush();
}
@@ -151,10 +148,10 @@ HRESULT CDXVADecoderH264::DecodeFrame (BYTE* pDataIn, UINT nSize, REFERENCE_TIME CComPtr<IMediaSample> pSampleToDeliver;
bool bSliceFound = false;
- int x;
+
while (!bSliceFound)
{
- ReadNalu (&Nalu, pDataSlice, nSliceSize, 4);
+ ReadNalu (&Nalu, pDataSlice, nSliceSize, m_nNALLength);
switch (Nalu.nal_unit_type)
{
case NALU_TYPE_SLICE:
@@ -163,21 +160,15 @@ HRESULT CDXVADecoderH264::DecodeFrame (BYTE* pDataIn, UINT nSize, REFERENCE_TIME break;
case NALU_TYPE_PPS :
- //ReadPPS(&m_PicParam, Nalu.data, RBSPtoSODB(Nalu.data, Nalu.data_len));
- //UpdatePictureParams(false);
-
m_pFilter->DecodeData (pDataSlice, nSliceSize);
- FFUpdatePictureParam (false, &m_DXVAPicParams, &m_DXVAScalingMatrix, m_pFilter->GetAVCtx());
+ FFH264UpdatePictureParam (false, &m_DXVAPicParams, &m_DXVAScalingMatrix, m_pFilter->GetAVCtx());
pDataSlice += Nalu.len;
nSliceSize -= Nalu.len;
break;
case NALU_TYPE_SPS :
- //ReadSPS(&m_SeqParam, Nalu.data, Nalu.data_len); // TODO : buffer size correct ??
- //UpdatePictureParams(false);
-
m_pFilter->DecodeData (pDataSlice, nSliceSize);
- FFUpdatePictureParam (false, &m_DXVAPicParams, &m_DXVAScalingMatrix, m_pFilter->GetAVCtx());
+ FFH264UpdatePictureParam (false, &m_DXVAPicParams, &m_DXVAScalingMatrix, m_pFilter->GetAVCtx());
pDataSlice += Nalu.len;
nSliceSize -= Nalu.len;
@@ -190,28 +181,12 @@ HRESULT CDXVADecoderH264::DecodeFrame (BYTE* pDataIn, UINT nSize, REFERENCE_TIME }
}
- // Step 1 : Parse bitstream and initialize Picture Parameters
- m_Slice.idr_flag = (Nalu.nal_unit_type == NALU_TYPE_IDR);
- ReadSliceHeader (&m_Slice, pDataSlice + 4, nSliceSize - 4);
-
+ // Parse slice header and set DX destination surface
+ CHECK_HR (FFH264ReadSlideHeader (&m_DXVAPicParams, m_pFilter->GetAVCtx(), pDataSlice+ m_nNALLength, nSliceSize - m_nNALLength));
CHECK_HR (GetFreeSurfaceIndex (nSurfaceIndex, &pSampleToDeliver, rtStart, rtStop));
-
- m_DXVAPicParams.field_pic_flag = m_Slice.field_pic_flag;
- m_DXVAPicParams.RefPicFlag = (Nalu.nal_reference_idc != 0);
- m_DXVAPicParams.IntraPicFlag = (m_Slice.slice_type == I_FRAME);
- m_DXVAPicParams.MbaffFrameFlag = (m_SeqParam.mb_adaptive_frame_field_flag && (m_Slice.field_pic_flag==0));
- m_DXVAPicParams.frame_num = m_Slice.frame_num;
-
- // Fill CurrPic parameters
- if (m_Slice.field_pic_flag)
- m_DXVAPicParams.CurrPic.AssociatedFlag = m_Slice.delta_pic_order_cnt_bottom;
m_DXVAPicParams.CurrPic.bPicEntry = nSurfaceIndex;
m_DXVAPicParams.CurrPic.Index7Bits = nSurfaceIndex;
- m_DXVAPicParams.CurrFieldOrderCnt[0] = m_Slice.pic_order_cnt_lsb;
- m_DXVAPicParams.CurrFieldOrderCnt[1] = m_Slice.pic_order_cnt_lsb;
-
-
CHECK_HR (BeginFrame(nSurfaceIndex, pSampleToDeliver));
// Send picture parameters
@@ -219,127 +194,30 @@ HRESULT CDXVADecoderH264::DecodeFrame (BYTE* pDataIn, UINT nSize, REFERENCE_TIME m_DXVAPicParams.StatusReportFeedbackNumber++;
CHECK_HR (Execute());
- // Add bitstream
+ // Add bitstream, slice control and quantization matrix
CHECK_HR (AddExecuteBuffer (DXVA2_BitStreamDateBufferType, nSize, pDataIn, &nSize));
m_SliceShort.SliceBytesInBuffer = nSize;
-
- // Decode bitstream
CHECK_HR (AddExecuteBuffer (DXVA2_SliceControlBufferType, sizeof (m_SliceShort), &m_SliceShort));
CHECK_HR (AddExecuteBuffer (DXVA2_InverseQuantizationMatrixBufferType, sizeof (DXVA_Qmatrix_H264), (void*)&m_DXVAScalingMatrix));
+ // Decode bitstream
m_DXVAPicParams.StatusReportFeedbackNumber++;
CHECK_HR (Execute());
CHECK_HR (EndFrame(nSurfaceIndex));
- UpdateRefFramesList (m_Slice.frame_num, (Nalu.nal_reference_idc != 0));
+ UpdateRefFramesList (m_DXVAPicParams.frame_num, (Nalu.nal_reference_idc != 0));
AddToStore (nSurfaceIndex, pSampleToDeliver, (Nalu.nal_reference_idc != 0), /*m_Slice.framepoc/2,*/ rtStart, rtStop);
return DisplayNextFrame();
}
-void CDXVADecoderH264::UpdatePictureParams(bool bInit)
-{
- if (bInit)
- {
- m_DXVAPicParams.wBitFields = 0;
- m_DXVAPicParams.MbsConsecutiveFlag = 1; // TODO : always activate ??
- m_DXVAPicParams.Reserved16Bits = 0;
- m_DXVAPicParams.ContinuationFlag = 1;
- m_DXVAPicParams.Reserved8BitsA = 0;
- m_DXVAPicParams.Reserved8BitsB = 0;
- m_DXVAPicParams.MinLumaBipredSize8x8Flag = 1; // Improve accelerator performances
-
- for (int i =0; i<16; i++)
- {
- m_DXVAPicParams.RefFrameList[i].AssociatedFlag = 1;
- m_DXVAPicParams.RefFrameList[i].bPicEntry = 255;
- m_DXVAPicParams.RefFrameList[i].Index7Bits = 127;
- }
- }
-
- m_DXVAPicParams.wFrameWidthInMbsMinus1 = m_SeqParam.pic_width_in_mbs_minus1;
- m_DXVAPicParams.wFrameHeightInMbsMinus1 = m_SeqParam.pic_height_in_map_units_minus1;
- m_DXVAPicParams.num_ref_frames = m_SeqParam.num_ref_frames;
-// m_DXVAPicParams.field_pic_flag = SET IN DecodeFrame;
-// m_DXVAPicParams.MbaffFrameFlag = SET IN DecodeFrame;
- m_DXVAPicParams.residual_colour_transform_flag = m_SeqParam.residual_colour_transform_flag;
-// m_DXVAPicParams.sp_for_switch_flag
- m_DXVAPicParams.chroma_format_idc = m_SeqParam.chroma_format_idc;
-// m_DXVAPicParams.RefPicFlag = SET IN DecodeFrame;
- m_DXVAPicParams.constrained_intra_pred_flag = m_PicParam.constrained_intra_pred_flag;
- m_DXVAPicParams.weighted_pred_flag = m_PicParam.weighted_pred_flag;
- m_DXVAPicParams.weighted_bipred_idc = m_PicParam.weighted_bipred_idc;
- m_DXVAPicParams.frame_mbs_only_flag = m_SeqParam.frame_mbs_only_flag;
- m_DXVAPicParams.transform_8x8_mode_flag = m_PicParam.transform_8x8_mode_flag;
-// m_DXVAPicParams.IntraPicFlag = SET IN DecodeFrame;
-
- m_DXVAPicParams.bit_depth_luma_minus8 = m_SeqParam.bit_depth_luma_minus8;
- m_DXVAPicParams.bit_depth_chroma_minus8 = m_SeqParam.bit_depth_chroma_minus8;
-// m_DXVAPicParams.StatusReportFeedbackNumber = SET IN DecodeFrame;
-
-
-// m_DXVAPicParams.CurrFieldOrderCnt = SET IN UpdateRefFramesList;
-// m_DXVAPicParams.FieldOrderCntList = SET IN UpdateRefFramesList;
-// m_DXVAPicParams.FrameNumList = SET IN UpdateRefFramesList;
-// m_DXVAPicParams.UsedForReferenceFlags = SET IN UpdateRefFramesList;
-// m_DXVAPicParams.NonExistingFrameFlags
-// m_DXVAPicParams.SliceGroupMap
-
-
- m_DXVAPicParams.log2_max_frame_num_minus4 = m_SeqParam.log2_max_frame_num_minus4;
- m_DXVAPicParams.pic_order_cnt_type = m_SeqParam.pic_order_cnt_type;
- m_DXVAPicParams.log2_max_pic_order_cnt_lsb_minus4 = m_SeqParam.log2_max_pic_order_cnt_lsb_minus4;
- m_DXVAPicParams.delta_pic_order_always_zero_flag = m_SeqParam.delta_pic_order_always_zero_flag;
- m_DXVAPicParams.direct_8x8_inference_flag = m_SeqParam.direct_8x8_inference_flag;
- m_DXVAPicParams.entropy_coding_mode_flag = m_PicParam.entropy_coding_mode_flag;
- m_DXVAPicParams.pic_order_present_flag = m_PicParam.pic_order_present_flag;
- m_DXVAPicParams.num_slice_groups_minus1 = m_PicParam.num_slice_groups_minus1;
- m_DXVAPicParams.slice_group_map_type = m_PicParam.slice_group_map_type;
- m_DXVAPicParams.deblocking_filter_control_present_flag = m_PicParam.deblocking_filter_control_present_flag;
- m_DXVAPicParams.redundant_pic_cnt_present_flag = m_PicParam.redundant_pic_cnt_present_flag;
- m_DXVAPicParams.slice_group_change_rate_minus1 = m_PicParam.slice_group_change_rate_minus1;
-
- m_DXVAPicParams.pic_init_qp_minus26 = m_PicParam.pic_init_qp_minus26;
- m_DXVAPicParams.chroma_qp_index_offset = m_PicParam.chroma_qp_index_offset;
- m_DXVAPicParams.second_chroma_qp_index_offset = m_PicParam.second_chroma_qp_index_offset;
- m_DXVAPicParams.num_ref_idx_l0_active_minus1 = m_PicParam.num_ref_idx_l0_active_minus1;
- m_DXVAPicParams.num_ref_idx_l1_active_minus1 = m_PicParam.num_ref_idx_l1_active_minus1;
-
- memcpy (&m_DXVAScalingMatrix, &m_PicParam.QuantMatrix, sizeof(DXVA_Qmatrix_H264));
-}
-
-
-
void CDXVADecoderH264::SetExtraData (BYTE* pDataIn, UINT nSize)
{
- NALU Nalu;
-
- while (nSize > 0)
- {
- ReadNalu (&Nalu, pDataIn, nSize, 2);
- switch (Nalu.nal_unit_type)
- {
- case NALU_TYPE_PPS :
- ReadPPS(&m_PicParam, Nalu.data, RBSPtoSODB(Nalu.data, Nalu.data_len));
- break;
- case NALU_TYPE_SPS :
- ReadSPS(&m_SeqParam, Nalu.data, Nalu.data_len); // TODO : buffer size correct ??
- break;
- }
-
- pDataIn += Nalu.len;
- nSize -= Nalu.len;
- }
- UpdatePictureParams(true);
-
- DXVA_PicParams_H264 Test;
- FFUpdatePictureParam (true, &Test, &m_DXVAScalingMatrix, m_pFilter->GetAVCtx());
- int x = memcmp (&m_DXVAPicParams, &Test, sizeof(DXVA_PicParams_H264));
- ASSERT (x == 0);
-
- FFUpdatePictureParam (true, &m_DXVAPicParams, &m_DXVAScalingMatrix, m_pFilter->GetAVCtx());
+ AVCodecContext* pAVCtx = m_pFilter->GetAVCtx();
+ m_nNALLength = pAVCtx->nal_length_size;
+ FFH264UpdatePictureParam (true, &m_DXVAPicParams, &m_DXVAScalingMatrix, pAVCtx);
}
@@ -371,7 +249,7 @@ void CDXVADecoderH264::UpdateRefFramesList (int nFrameNum, bool bRefFrame) int i;
// Reset when new picture group detected
- if (m_Slice.frame_num == 0) ClearRefFramesList();
+ if (nFrameNum == 0) ClearRefFramesList();
if (bRefFrame)
{
@@ -380,7 +258,7 @@ void CDXVADecoderH264::UpdateRefFramesList (int nFrameNum, bool bRefFrame) {
if (m_DXVAPicParams.RefFrameList[0].bPicEntry != 255)
RemoveRefFrame (m_DXVAPicParams.RefFrameList[0].bPicEntry);
-// m_pCurRefSample[m_DXVAPicParams.RefFrameList[0].bPicEntry] = NULL;
+
for (i=1; i<m_DXVAPicParams.num_ref_frames; i++)
{
m_DXVAPicParams.FrameNumList[i-1] = m_DXVAPicParams.FrameNumList[i];
@@ -407,617 +285,6 @@ void CDXVADecoderH264::UpdateRefFramesList (int nFrameNum, bool bRefFrame) }
-// === H264 bitstream parsing (based on JM / reference decoder)
-#pragma region
-
-/*!
- ************************************************************************
- * \brief
- * read one exp-golomb VLC symbol
- *
- * \param buffer
- * containing VLC-coded data bits
- * \param totbitoffset
- * bit offset from start of partition
- * \param info
- * returns the value of the symbol
- * \param bytecount
- * buffer length
- * \return
- * bits read
- ************************************************************************
- */
-int CDXVADecoderH264::GetVLCSymbol (BYTE* buffer,int totbitoffset,int *info, int bytecount)
-{
-
- register int inf;
- long byteoffset = (totbitoffset >> 3); // byte from start of buffer
- int bitoffset = (7 - (totbitoffset & 0x07)); // bit from start of byte
- int bitcounter = 1;
- int len = 0;
- byte *cur_byte = &(buffer[byteoffset]);
- int ctr_bit = ((*cur_byte) >> (bitoffset)) & 0x01; // control bit for current bit posision
-
- while (ctr_bit == 0)
- { // find leading 1 bit
- len++;
- bitcounter++;
- bitoffset--;
- bitoffset &= 0x07;
- cur_byte += (bitoffset == 7);
- byteoffset+= (bitoffset == 7);
- ctr_bit = ((*cur_byte) >> (bitoffset)) & 0x01;
- }
-
- if (byteoffset + ((len + 7) >> 3) > bytecount)
- return -1;
-
- // make infoword
- inf = 0; // shortest possible code is 1, then info is always 0
-
- while (len--)
- {
- bitoffset --;
- bitoffset &= 0x07;
- cur_byte += (bitoffset == 7);
- bitcounter++;
- inf <<= 1;
- inf |= ((*cur_byte) >> (bitoffset)) & 0x01;
- }
-
- *info = inf;
- return bitcounter; // return absolute offset in bit from start of frame
-}
-
-
-/*!
- ************************************************************************
- * \brief
- * Reads bits from the bitstream buffer
- *
- * \param buffer
- * containing VLC-coded data bits
- * \param totbitoffset
- * bit offset from start of partition
- * \param info
- * returns value of the read bits
- * \param bytecount
- * total bytes in bitstream
- * \param numbits
- * number of bits to read
- *
- ************************************************************************
- */
-int CDXVADecoderH264::GetBits (byte buffer[],int totbitoffset,int *info, int bytecount,
- int numbits)
-{
- register int inf;
- int bitoffset = (totbitoffset & 0x07); // bit from start of byte
- long byteoffset = (totbitoffset >> 3); // byte from start of buffer
- int bitcounter = numbits;
- static byte *curbyte;
-
- if ((byteoffset) + ((numbits + bitoffset)>> 3) > bytecount)
- return -1;
-
- curbyte = &(buffer[byteoffset]);
-
- bitoffset = 7 - bitoffset;
-
- inf=0;
-
- while (numbits--)
- {
- inf <<=1;
- inf |= ((*curbyte)>> (bitoffset--)) & 0x01;
- //curbyte += (bitoffset >> 3) & 0x01;
- curbyte -= (bitoffset >> 3);
- bitoffset &= 0x07;
- //curbyte += (bitoffset == 7);
- }
-
- *info = inf;
- return bitcounter; // return absolute offset in bit from start of frame
-}
-
-int CDXVADecoderH264::ReadSyntaxElement_VLC(SyntaxElement *sym, BYTE* pBuffer, UINT nBufferLength, UINT& nBitOffset)
-{
- sym->len = GetVLCSymbol (pBuffer, nBitOffset, &(sym->inf), nBufferLength);
- if (sym->len == -1)
- return -1;
- nBitOffset += sym->len;
- sym->mapping(sym->len,sym->inf,&(sym->value1),&(sym->value2));
-
- return 1;
-}
-
-int CDXVADecoderH264::ReadSyntaxElement_FLC(SyntaxElement *sym, BYTE* pBuffer, UINT nBufferLength, UINT& nBitOffset)
-{
- if ((GetBits(pBuffer, nBitOffset, &(sym->inf), nBufferLength, sym->len)) < 0)
- return -1;
-
- sym->value1 = sym->inf;
- nBitOffset += sym->len; // move bitstream pointer
-
- return 1;
-}
-
-
-
-/*!
- ************************************************************************
- * \brief
- * mapping rule for ue(v) syntax elements
- * \par Input:
- * lenght and info
- * \par Output:
- * number in the code table
- ************************************************************************
- */
-void CDXVADecoderH264::LInfo_Ue(int len, int info, int *value1, int *dummy)
-{
- ASSERT ((len >> 1) < 32);
- *value1 = (1 << (len >> 1)) + info - 1;
-}
-
-/*!
- ************************************************************************
- * \brief
- * mapping rule for se(v) syntax elements
- * \par Input:
- * lenght and info
- * \par Output:
- * signed mvd
- ************************************************************************
- */
-void CDXVADecoderH264::LInfo_Se(int len, int info, int *value1, int *dummy)
-{
- int n;
- ASSERT ((len >> 1) < 32);
- n = (1 << (len >> 1)) + info - 1;
- *value1 = (n + 1) >> 1;
- if((n & 0x01) == 0) // lsb is signed bit
- *value1 = -*value1;
-}
-
-int CDXVADecoderH264::ue_v (BYTE* pBuffer, UINT nBufferLength, UINT& nBitOffset)
-{
- SyntaxElement symbol;
-
- symbol.type = SE_HEADER;
- symbol.mapping = LInfo_Ue; // Mapping rule
-
- ReadSyntaxElement_VLC (&symbol, pBuffer, nBufferLength, nBitOffset);
- return symbol.value1;
-}
-
-
-/*!
- *************************************************************************************
- * \brief
- * ue_v, reads an se(v) syntax element, the length in bits is stored in
- * the global UsedBits variable
- *
- * \param tracestring
- * the string for the trace file
- *
- * \param bitstream
- * the stream to be read from
- *
- * \return
- * the value of the coded syntax element
- *
- *************************************************************************************
- */
-int CDXVADecoderH264::se_v (BYTE* pBuffer, UINT nBufferLength, UINT& nBitOffset)
-{
- SyntaxElement symbol;
-
- symbol.type = SE_HEADER;
- symbol.mapping = LInfo_Se; // Mapping rule: signed integer
-
- ReadSyntaxElement_VLC (&symbol, pBuffer, nBufferLength, nBitOffset);
- return symbol.value1;
-}
-
-
-/*!
- *************************************************************************************
- * \brief
- * ue_v, reads an u(1) syntax element, the length in bits is stored in
- * the global UsedBits variable
- *
- * \param tracestring
- * the string for the trace file
- *
- * \param bitstream
- * the stream to be read from
- *
- * \return
- * the value of the coded syntax element
- *
- *************************************************************************************
- */
-bool CDXVADecoderH264::u_1 (BYTE* pBuffer, UINT nBufferLength, UINT& nBitOffset)
-{
- return (bool) !!u_v (1, pBuffer, nBufferLength, nBitOffset);
-}
-
- //*************************************************************************************
- //* \brief
- //* ue_v, reads an u(v) syntax element, the length in bits is stored in
- //* the global UsedBits variable
- //*
- //* \param LenInBits
- //* length of the syntax element
- //*
- //* \param tracestring
- //* the string for the trace file
- //*
- //* \param bitstream
- //* the stream to be read from
- //*
- //* \return
- //* the value of the coded syntax element
- //*
- //*************************************************************************************
-int CDXVADecoderH264::u_v (int LenInBits, BYTE* pBuffer, UINT nBufferLength, UINT& nBitOffset)
-{
- SyntaxElement symbol;
- symbol.inf = 0;
-
- symbol.type = SE_HEADER;
- symbol.mapping = LInfo_Ue; // Mapping rule
- symbol.len = LenInBits;
-
- ReadSyntaxElement_FLC (&symbol, pBuffer, nBufferLength, nBitOffset);
-
- return symbol.inf;
-}
-
-
-void CDXVADecoderH264::ReadSliceHeader(SLICE_PARAMETER* pSlice, BYTE* pBuffer, UINT nBufferLength)
-{
- UINT nBitOffset = 8;
- int tmp;
-
- pSlice->first_mb_in_slice = ue_v (pBuffer, nBufferLength, nBitOffset);
-
- tmp = ue_v (pBuffer, nBufferLength, nBitOffset);
- if (tmp>4) tmp -=5;
- pSlice->slice_type = (FRAME_TYPE) tmp;
-
- pSlice->pic_parameter_set_id = ue_v (pBuffer, nBufferLength, nBitOffset);
-
- if (m_SeqParam.residual_colour_transform_flag)
- pSlice->colour_plane_id = u_v (2, pBuffer, nBufferLength, nBitOffset);
-
- pSlice->frame_num = (UINT)u_v (m_SeqParam.log2_max_frame_num_minus4 + 4, pBuffer, nBufferLength, nBitOffset);
-
- if (m_SeqParam.frame_mbs_only_flag)
- {
- pSlice->field_pic_flag = 0;
- }
- else
- {
- pSlice->field_pic_flag = u_1(pBuffer, nBufferLength, nBitOffset);
- if (pSlice->field_pic_flag)
- pSlice->bottom_field_flag = u_1(pBuffer, nBufferLength, nBitOffset);
- // TODO : pb sur le else !!
- }
-
- if (pSlice->idr_flag)
- pSlice->idr_pic_id = ue_v(pBuffer, nBufferLength, nBitOffset);
-
- if (m_SeqParam.pic_order_cnt_type == 0)
- {
- pSlice->pic_order_cnt_lsb = u_v(m_SeqParam.log2_max_pic_order_cnt_lsb_minus4 + 4, pBuffer, nBufferLength, nBitOffset);
- if( m_PicParam.pic_order_present_flag == 1 && !pSlice->field_pic_flag )
- pSlice->delta_pic_order_cnt_bottom = se_v(pBuffer, nBufferLength, nBitOffset);
- else
- pSlice->delta_pic_order_cnt_bottom = 0;
- }
- if (m_SeqParam.pic_order_cnt_type == 1 && !m_SeqParam.delta_pic_order_always_zero_flag )
- {
- pSlice->delta_pic_order_cnt[ 0 ] = se_v(pBuffer, nBufferLength, nBitOffset);
- if( m_PicParam.pic_order_present_flag == 1 && !pSlice->field_pic_flag )
- pSlice->delta_pic_order_cnt[ 1 ] = se_v(pBuffer, nBufferLength, nBitOffset);
- }
- else
- {
- if (m_SeqParam.pic_order_cnt_type == 1)
- {
- pSlice->delta_pic_order_cnt[ 0 ] = 0;
- pSlice->delta_pic_order_cnt[ 1 ] = 0;
- }
- }
-
- // TODO : read sp_for_switch_flag !!
-
- //! redundant_pic_cnt is missing here
-/*
- if (active_pps->redundant_pic_cnt_present_flag)
- {
- img->redundant_pic_cnt = ue_v ("SH: redundant_pic_cnt", currStream);
- }
-
- if(img->type==B_SLICE)
- {
- img->direct_spatial_mv_pred_flag = u_1 ("SH: direct_spatial_mv_pred_flag", currStream);
- }
-
- img->num_ref_idx_l0_active = active_pps->num_ref_idx_l0_active_minus1 + 1;
- img->num_ref_idx_l1_active = active_pps->num_ref_idx_l1_active_minus1 + 1;
-
- if(img->type==P_SLICE || img->type == SP_SLICE || img->type==B_SLICE)
- {
- val = u_1 ("SH: num_ref_idx_override_flag", currStream);
- if (val)
- {
- img->num_ref_idx_l0_active = 1 + ue_v ("SH: num_ref_idx_l0_active_minus1", currStream);
-
- if(img->type==B_SLICE)
- {
- img->num_ref_idx_l1_active = 1 + ue_v ("SH: num_ref_idx_l1_active_minus1", currStream);
- }
- }
- }
- if (img->type!=B_SLICE)
- {
- img->num_ref_idx_l1_active = 0;
- }
-
- ref_pic_list_reordering();
-
- img->apply_weights = ((active_pps->weighted_pred_flag && (currSlice->picture_type == P_SLICE || currSlice->picture_type == SP_SLICE) )
- || ((active_pps->weighted_bipred_idc > 0 ) && (currSlice->picture_type == B_SLICE)));
-
- if ((active_pps->weighted_pred_flag&&(img->type==P_SLICE|| img->type == SP_SLICE))||
- (active_pps->weighted_bipred_idc==1 && (img->type==B_SLICE)))
- {
- pred_weight_table();
- }
-
- if (img->nal_reference_idc)
- dec_ref_pic_marking(currStream);
-
- if (active_pps->entropy_coding_mode_flag && img->type!=I_SLICE && img->type!=SI_SLICE)
- {
- img->model_number = ue_v("SH: cabac_init_idc", currStream);
- }
- else
- {
- img->model_number = 0;
- }
-
- val = se_v("SH: slice_qp_delta", currStream);
- currSlice->qp = img->qp = 26 + active_pps->pic_init_qp_minus26 + val;
- if ((img->qp < -img->bitdepth_luma_qp_scale) || (img->qp > 51))
- error ("slice_qp_delta makes slice_qp_y out of range", 500);
-
-
- currSlice->slice_qp_delta = val;
-
- if(img->type==SP_SLICE || img->type == SI_SLICE)
- {
- if(img->type==SP_SLICE)
- {
- img->sp_switch = u_1 ("SH: sp_for_switch_flag", currStream);
- }
- val = se_v("SH: slice_qs_delta", currStream);
- img->qpsp = 26 + active_pps->pic_init_qs_minus26 + val;
- if ((img->qpsp < 0) || (img->qpsp > 51))
- error ("slice_qs_delta makes slice_qs_y out of range", 500);
- }
-*/
-
- DecodePOC (pSlice);
-}
-
-
-void CDXVADecoderH264::DecodePOC(SLICE_PARAMETER* pSlice)
-{
- int i;
- // for POC mode 0:
- unsigned int MaxPicOrderCntLsb = 1<<(m_SeqParam.log2_max_pic_order_cnt_lsb_minus4+4);
- unsigned int MaxFrameNum = 1<<(m_SeqParam.log2_max_frame_num_minus4+4);
-
- switch ( m_SeqParam.pic_order_cnt_type )
- {
- case 0: // POC MODE 0
- // 1st
- if(pSlice->idr_flag)
- {
- pSlice->PrevPicOrderCntMsb = 0;
- pSlice->PrevPicOrderCntLsb = 0;
- }
- else
- {
- //if (pSlice->last_has_mmco_5)
- //{
- // if (pSlice->last_pic_bottom_field)
- // {
- // pSlice->PrevPicOrderCntMsb = 0;
- // pSlice->PrevPicOrderCntLsb = 0;
- // }
- // else
- // {
- // pSlice->PrevPicOrderCntMsb = 0;
- // pSlice->PrevPicOrderCntLsb = pSlice->toppoc;
- // }
- //}
- }
-
- // Calculate the MSBs of current picture
- if ( pSlice->pic_order_cnt_lsb < pSlice->PrevPicOrderCntLsb &&
- ( pSlice->PrevPicOrderCntLsb - pSlice->pic_order_cnt_lsb ) >= ( MaxPicOrderCntLsb / 2 ) )
- pSlice->PicOrderCntMsb = pSlice->PrevPicOrderCntMsb + MaxPicOrderCntLsb;
- else if ( pSlice->pic_order_cnt_lsb > pSlice->PrevPicOrderCntLsb &&
- ( pSlice->pic_order_cnt_lsb - pSlice->PrevPicOrderCntLsb ) > ( MaxPicOrderCntLsb / 2 ) )
- pSlice->PicOrderCntMsb = pSlice->PrevPicOrderCntMsb - MaxPicOrderCntLsb;
- else
- pSlice->PicOrderCntMsb = pSlice->PrevPicOrderCntMsb;
-
- // 2nd
- if(pSlice->field_pic_flag==0)
- { //frame pix
- pSlice->toppoc = pSlice->PicOrderCntMsb + pSlice->pic_order_cnt_lsb;
- pSlice->bottompoc = pSlice->toppoc + pSlice->delta_pic_order_cnt_bottom;
- pSlice->ThisPOC = pSlice->framepoc = (pSlice->toppoc < pSlice->bottompoc)? pSlice->toppoc : pSlice->bottompoc; // POC200301
- }
- else if (pSlice->bottom_field_flag==0)
- { //top field
- pSlice->ThisPOC= pSlice->toppoc = pSlice->PicOrderCntMsb + pSlice->pic_order_cnt_lsb;
- }
- else
- { //bottom field
- pSlice->ThisPOC= pSlice->bottompoc = pSlice->PicOrderCntMsb + pSlice->pic_order_cnt_lsb;
- }
- pSlice->framepoc=pSlice->ThisPOC;
-
- if ( pSlice->frame_num!=pSlice->PreviousFrameNum)
- pSlice->PreviousFrameNum=pSlice->frame_num;
-
- if(pSlice->nal_reference_idc)
- {
- pSlice->PrevPicOrderCntLsb = pSlice->pic_order_cnt_lsb;
- pSlice->PrevPicOrderCntMsb = pSlice->PicOrderCntMsb;
- }
-
- break;
-
- case 1: // POC MODE 1
- // 1st
- if(pSlice->idr_flag)
- {
- pSlice->FrameNumOffset=0; // first pix of IDRGOP,
- pSlice->delta_pic_order_cnt[0]=0; // ignore first delta
- if(pSlice->frame_num)
- {
- TRACE("frame_num not equal to zero in IDR picture");
- ASSERT (FALSE);
- return;
- }
- }
- else
- {
- //if (pSlice->last_has_mmco_5)
- //{
- // pSlice->PreviousFrameNumOffset = 0;
- // pSlice->PreviousFrameNum = 0;
- //}
- if (pSlice->frame_num<pSlice->PreviousFrameNum)
- { //not first pix of IDRGOP
- pSlice->FrameNumOffset = pSlice->PreviousFrameNumOffset + MaxFrameNum;
- }
- else
- {
- pSlice->FrameNumOffset = pSlice->PreviousFrameNumOffset;
- }
- }
-
- // 2nd
- if(m_SeqParam.num_ref_frames_in_pic_order_cnt_cycle)
- pSlice->AbsFrameNum = pSlice->FrameNumOffset+pSlice->frame_num;
- else
- pSlice->AbsFrameNum=0;
- if( (!pSlice->nal_reference_idc) && pSlice->AbsFrameNum>0)
- pSlice->AbsFrameNum--;
-
- // 3rd
- pSlice->ExpectedDeltaPerPicOrderCntCycle=0;
-
- if(m_SeqParam.num_ref_frames_in_pic_order_cnt_cycle)
- for(i=0;i<(int) m_SeqParam.num_ref_frames_in_pic_order_cnt_cycle;i++)
- pSlice->ExpectedDeltaPerPicOrderCntCycle += m_SeqParam.offset_for_ref_frame[i];
-
- if(pSlice->AbsFrameNum)
- {
- pSlice->PicOrderCntCycleCnt = (pSlice->AbsFrameNum-1)/m_SeqParam.num_ref_frames_in_pic_order_cnt_cycle;
- pSlice->FrameNumInPicOrderCntCycle = (pSlice->AbsFrameNum-1)%m_SeqParam.num_ref_frames_in_pic_order_cnt_cycle;
- pSlice->ExpectedPicOrderCnt = pSlice->PicOrderCntCycleCnt*pSlice->ExpectedDeltaPerPicOrderCntCycle;
- for(i=0;i<=(int)pSlice->FrameNumInPicOrderCntCycle;i++)
- pSlice->ExpectedPicOrderCnt += m_SeqParam.offset_for_ref_frame[i];
- }
- else
- pSlice->ExpectedPicOrderCnt=0;
-
- if(!pSlice->nal_reference_idc)
- pSlice->ExpectedPicOrderCnt += m_SeqParam.offset_for_non_ref_pic;
-
- if(pSlice->field_pic_flag==0)
- { //frame pix
- pSlice->toppoc = pSlice->ExpectedPicOrderCnt + pSlice->delta_pic_order_cnt[0];
- pSlice->bottompoc = pSlice->toppoc + m_SeqParam.offset_for_top_to_bottom_field + pSlice->delta_pic_order_cnt[1];
- pSlice->ThisPOC = pSlice->framepoc = (pSlice->toppoc < pSlice->bottompoc)? pSlice->toppoc : pSlice->bottompoc; // POC200301
- }
- else if (pSlice->bottom_field_flag==0)
- { //top field
- pSlice->ThisPOC = pSlice->toppoc = pSlice->ExpectedPicOrderCnt + pSlice->delta_pic_order_cnt[0];
- }
- else
- { //bottom field
- pSlice->ThisPOC = pSlice->bottompoc = pSlice->ExpectedPicOrderCnt + m_SeqParam.offset_for_top_to_bottom_field + pSlice->delta_pic_order_cnt[0];
- }
- pSlice->framepoc=pSlice->ThisPOC;
-
- pSlice->PreviousFrameNum=pSlice->frame_num;
- pSlice->PreviousFrameNumOffset=pSlice->FrameNumOffset;
-
- break;
-
-
- case 2: // POC MODE 2
- if(pSlice->idr_flag) // IDR picture
- {
- pSlice->FrameNumOffset=0; // first pix of IDRGOP,
- pSlice->ThisPOC = pSlice->framepoc = pSlice->toppoc = pSlice->bottompoc = 0;
- if(pSlice->frame_num)
- {
- TRACE("frame_num not equal to zero in IDR picture");
- ASSERT(FALSE);
- return;
- }
- }
- else
- {
- //if (pSlice->last_has_mmco_5)
- //{
- // pSlice->PreviousFrameNum = 0;
- // pSlice->PreviousFrameNumOffset = 0;
- //}
- if (pSlice->frame_num<pSlice->PreviousFrameNum)
- pSlice->FrameNumOffset = pSlice->PreviousFrameNumOffset + MaxFrameNum;
- else
- pSlice->FrameNumOffset = pSlice->PreviousFrameNumOffset;
-
-
- pSlice->AbsFrameNum = pSlice->FrameNumOffset+pSlice->frame_num;
- if(!pSlice->nal_reference_idc)
- pSlice->ThisPOC = (2*pSlice->AbsFrameNum - 1);
- else
- pSlice->ThisPOC = (2*pSlice->AbsFrameNum);
-
- if (pSlice->field_pic_flag==0)
- pSlice->toppoc = pSlice->bottompoc = pSlice->framepoc = pSlice->ThisPOC;
- else if (pSlice->bottom_field_flag==0)
- pSlice->toppoc = pSlice->framepoc = pSlice->ThisPOC;
- else pSlice->bottompoc = pSlice->framepoc = pSlice->ThisPOC;
- }
-
- pSlice->PreviousFrameNum=pSlice->frame_num;
- pSlice->PreviousFrameNumOffset=pSlice->FrameNumOffset;
- break;
-
-
- default:
- //error must occurs
- ASSERT(FALSE);
- break;
- }
-}
-
-
void CDXVADecoderH264::ReadNalu (NALU* pNalu, BYTE* pBuffer, UINT nBufferLength, UINT NbBytesForSize)
{
pNalu->data = pBuffer;
@@ -1036,402 +303,3 @@ void CDXVADecoderH264::ReadNalu (NALU* pNalu, BYTE* pBuffer, UINT nBufferLength, pNalu->data++;
pNalu->data_len--;
}
-
-
-/*!
- ************************************************************************
- * \brief
- * test if bit buffer contains only stop bit
- *
- * \param buffer
- * buffer containing VLC-coded data bits
- * \param totbitoffset
- * bit offset from start of partition
- * \param bytecount
- * buffer length
- * \return
- * true if more bits available
- ************************************************************************
- */
-int CDXVADecoderH264::more_rbsp_data (BYTE* pBuffer,int totbitoffset,int bytecount)
-{
- int bitoffset = (7 - (totbitoffset & 0x07)); // bit from start of byte
- long byteoffset = (totbitoffset >> 3); // byte from start of pBuffer
- byte *cur_byte = &(pBuffer[byteoffset]);
- int ctr_bit = 0; // control bit for current bit posision
- int cnt = 0;
-
- ASSERT (byteoffset<bytecount);
-
- // there is more until we're in the last byte
- if (byteoffset < (bytecount - 1)) return TRUE;
-
- // read one bit
- ctr_bit = ((*cur_byte)>> (bitoffset--)) & 0x01;
-
- // a stop bit has to be one
- if (ctr_bit==0) return TRUE;
-
- while (bitoffset>=0 && !cnt)
- {
- cnt |= ((*cur_byte)>> (bitoffset--)) & 0x01; // set up control bit
- }
-
- return (cnt);
-}
-
- /*!
- ************************************************************************
- * \brief
- * Converts RBSP to string of data bits
- * \param pBuffer
- * pointer to buffer containing data
- * \param last_byte_pos
- * position of the last byte containing data.
- * \return last_byte_pos
- * position of the last byte pos. If the last-byte was entirely a stuffing byte,
- * it is removed, and the last_byte_pos is updated.
- *
-************************************************************************/
-
-int CDXVADecoderH264::RBSPtoSODB(BYTE* pBuffer, int last_byte_pos)
-{
- int ctr_bit, bitoffset;
-
- bitoffset = 0;
- //find trailing 1
- ctr_bit = (pBuffer[last_byte_pos-1] & (0x01<<bitoffset)); // set up control bit
-
- while (ctr_bit==0)
- { // find trailing 1 bit
- bitoffset++;
- if(bitoffset == 8)
- {
- if(last_byte_pos == 0)
- printf(" Panic: All zero data sequence in RBSP \n");
- ASSERT(last_byte_pos != 0);
- last_byte_pos -= 1;
- bitoffset = 0;
- }
- ctr_bit= pBuffer[last_byte_pos-1] & (0x01<<(bitoffset));
- }
-
-
- // We keep the stop bit for now
-/* if (remove_stop)
- {
- pBuffer[last_byte_pos-1] -= (0x01<<(bitoffset));
- if(bitoffset == 7)
- return(last_byte_pos-1);
- else
- return(last_byte_pos);
- }
-*/
- return(last_byte_pos);
-
-}
-
-void CDXVADecoderH264::Scaling_List(UCHAR* scalingList, int sizeOfScalingList, BYTE* pBuffer, UINT nBufferLength, UINT& nBitOffset)
-{
- int j, scanj;
- int delta_scale, lastScale, nextScale;
-
- lastScale = 8;
- nextScale = 8;
-
- for(j=0; j<sizeOfScalingList; j++)
- {
- scanj = (sizeOfScalingList==16) ? ZZ_SCAN[j]:ZZ_SCAN8[j];
-
- if(nextScale!=0)
- {
- delta_scale = se_v (pBuffer, nBufferLength, nBitOffset);
- nextScale = (lastScale + delta_scale + 256) % 256;
-// *UseDefaultScalingMatrix = (Boolean) (scanj==0 && nextScale==0);
- }
-
- scalingList[scanj] = (nextScale==0) ? lastScale:nextScale;
- lastScale = scalingList[scanj];
- }
-}
-
-
-void CDXVADecoderH264::ReadPPS(PIC_PARAMETER_SET_RBSP* pps, BYTE* pBuffer, UINT nBufferLength)
-{
- UINT nBitOffset = 0;
- int NumberBitsPerSliceGroupId;
-// int chroma_format_idc;
- unsigned n_ScalingList;
- unsigned i;
-
- memcpy (&pps->QuantMatrix, (void*)&g_QMatrixH264[Flat16], sizeof(DXVA_Qmatrix_H264));
-
- pps->pic_parameter_set_id = ue_v (pBuffer, nBufferLength, nBitOffset);
- pps->seq_parameter_set_id = ue_v (pBuffer, nBufferLength, nBitOffset);
- pps->entropy_coding_mode_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
-
- //! Note: as per JVT-F078 the following bit is unconditional. If F078 is not accepted, then
- //! one has to fetch the correct SPS to check whether the bit is present (hopefully there is
- //! no consistency problem :-(
- //! The current encoder code handles this in the same way. When you change this, don't forget
- //! the encoder! StW, 12/8/02
- pps->pic_order_present_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
-
- pps->num_slice_groups_minus1 = ue_v (pBuffer, nBufferLength, nBitOffset);
-
- // FMO stuff begins here
- if (pps->num_slice_groups_minus1 > 0)
- {
- pps->slice_group_map_type = ue_v (pBuffer, nBufferLength, nBitOffset);
- if (pps->slice_group_map_type == 0)
- {
- for (i=0; i<=pps->num_slice_groups_minus1; i++)
- pps->run_length_minus1 [i] = ue_v (pBuffer, nBufferLength, nBitOffset);
- }
- else if (pps->slice_group_map_type == 2)
- {
- for (i=0; i<pps->num_slice_groups_minus1; i++)
- {
- //! JVT-F078: avoid reference of SPS by using ue(v) instead of u(v)
- pps->top_left [i] = ue_v (pBuffer, nBufferLength, nBitOffset);
- pps->bottom_right [i] = ue_v (pBuffer, nBufferLength, nBitOffset);
- }
- }
- else if (pps->slice_group_map_type == 3 ||
- pps->slice_group_map_type == 4 ||
- pps->slice_group_map_type == 5)
- {
- pps->slice_group_change_direction_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- pps->slice_group_change_rate_minus1 = ue_v (pBuffer, nBufferLength, nBitOffset);
- }
- else if (pps->slice_group_map_type == 6)
- {
- if (pps->num_slice_groups_minus1+1 >4)
- NumberBitsPerSliceGroupId = 3;
- else if (pps->num_slice_groups_minus1+1 > 2)
- NumberBitsPerSliceGroupId = 2;
- else
- NumberBitsPerSliceGroupId = 1;
-
- //! JVT-F078, exlicitly signal number of MBs in the map
- pps->num_slice_group_map_units_minus1 = ue_v (pBuffer, nBufferLength, nBitOffset);
- for (i=0; i<=pps->num_slice_group_map_units_minus1; i++)
- pps->slice_group_id[i] = u_v (NumberBitsPerSliceGroupId, pBuffer, nBufferLength, nBitOffset);
- }
- }
-
- // End of FMO stuff
-
- pps->num_ref_idx_l0_active_minus1 = ue_v (pBuffer, nBufferLength, nBitOffset);
- pps->num_ref_idx_l1_active_minus1 = ue_v (pBuffer, nBufferLength, nBitOffset);
- pps->weighted_pred_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- pps->weighted_bipred_idc = u_v ( 2, pBuffer, nBufferLength, nBitOffset);
- pps->pic_init_qp_minus26 = se_v (pBuffer, nBufferLength, nBitOffset);
- pps->pic_init_qs_minus26 = se_v (pBuffer, nBufferLength, nBitOffset);
-
- pps->chroma_qp_index_offset = se_v (pBuffer, nBufferLength, nBitOffset);
-
- pps->deblocking_filter_control_present_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- pps->constrained_intra_pred_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- pps->redundant_pic_cnt_present_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
-
- if(more_rbsp_data(pBuffer, nBitOffset, nBufferLength)) // more_data_in_rbsp()
- {
- //Fidelity Range Extensions Stuff
- pps->transform_8x8_mode_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- pps->pic_scaling_matrix_present_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
-
- if(pps->pic_scaling_matrix_present_flag)
- {
- // TODO : refuse 444 profiles !!
- // chroma_format_idc = SeqParSet[pps->seq_parameter_set_id].chroma_format_idc;
- // n_ScalingList = 6 + ((chroma_format_idc != YUV444) ? 2 : 6) * pps->transform_8x8_mode_flag;
- n_ScalingList = 6 + 2 * pps->transform_8x8_mode_flag;
-
- for(i=0; i<n_ScalingList; i++)
- {
- pps->pic_scaling_list_present_flag[i]= u_1 (pBuffer, nBufferLength, nBitOffset);
-
- if(pps->pic_scaling_list_present_flag[i])
- {
- if(i<6)
- Scaling_List(pps->QuantMatrix.bScalingLists4x4[i], 16, pBuffer, nBufferLength, nBitOffset);
- else
- Scaling_List(pps->QuantMatrix.bScalingLists8x8[i-6], 64, pBuffer, nBufferLength, nBitOffset);
- //if(i<6)
- // Scaling_List(pps->ScalingList4x4[i], 16, &pps->UseDefaultScalingMatrix4x4Flag[i], s);
- //else
- // Scaling_List(pps->ScalingList8x8[i-6], 64, &pps->UseDefaultScalingMatrix8x8Flag[i-6], s);
- }
- }
- }
- pps->second_chroma_qp_index_offset = se_v (pBuffer, nBufferLength, nBitOffset);
- }
- else
- {
- pps->second_chroma_qp_index_offset = pps->chroma_qp_index_offset;
- }
-
- pps->Valid = TRUE;
-}
-
-
-
-void CDXVADecoderH264::ReadSPS(SEQ_PARAMETER_SET_RBSP* sps, BYTE* pBuffer, UINT nBufferLength)
-{
- UINT nBitOffset = 0;
- unsigned i;
- unsigned n_ScalingList;
- int reserved_zero;
-
-
- sps->profile_idc = u_v (8, pBuffer, nBufferLength, nBitOffset);
-
- if ((sps->profile_idc!=66 ) &&
- (sps->profile_idc!=77 ) &&
- (sps->profile_idc!=88 ) &&
- (sps->profile_idc!=FREXT_HP ) &&
- (sps->profile_idc!=FREXT_Hi10P ) &&
- (sps->profile_idc!=FREXT_Hi422 ) &&
- (sps->profile_idc!=FREXT_Hi444 ) &&
- (sps->profile_idc!=FREXT_CAVLC444 ))
- {
- return;
- }
-
- sps->constrained_set0_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- sps->constrained_set1_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- sps->constrained_set2_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- sps->constrained_set3_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- reserved_zero = u_v (4, pBuffer, nBufferLength, nBitOffset);
- ASSERT (reserved_zero==0);
-
- sps->level_idc = u_v (8, pBuffer, nBufferLength, nBitOffset);
-
- sps->seq_parameter_set_id = ue_v (pBuffer, nBufferLength, nBitOffset);
-
- // Fidelity Range Extensions stuff
- sps->chroma_format_idc = 1;
- sps->bit_depth_luma_minus8 = 0;
- sps->bit_depth_chroma_minus8 = 0;
-// img->lossless_qpprime_flag = 0;
- sps->residual_colour_transform_flag = 0;
-
- if((sps->profile_idc==FREXT_HP ) ||
- (sps->profile_idc==FREXT_Hi10P) ||
- (sps->profile_idc==FREXT_Hi422) ||
- (sps->profile_idc==FREXT_Hi444) ||
- (sps->profile_idc==FREXT_CAVLC444))
- {
- sps->chroma_format_idc = ue_v (pBuffer, nBufferLength, nBitOffset);
-
- if(sps->chroma_format_idc == YUV444)
- {
- sps->residual_colour_transform_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- }
-
- sps->bit_depth_luma_minus8 = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->bit_depth_chroma_minus8 = ue_v (pBuffer, nBufferLength, nBitOffset);
- /*img->lossless_qpprime_flag = */ u_1 (pBuffer, nBufferLength, nBitOffset);
-
- sps->seq_scaling_matrix_present_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
-
- if(sps->seq_scaling_matrix_present_flag)
- {
- n_ScalingList = (sps->chroma_format_idc != YUV444) ? 8 : 12;
- for(i=0; i<n_ScalingList; i++)
- {
- sps->seq_scaling_list_present_flag[i] = u_1 (pBuffer, nBufferLength, nBitOffset);
- if(sps->seq_scaling_list_present_flag[i])
- {
- //if(i<6)
- // Scaling_List(sps->ScalingList4x4[i], 16, &sps->UseDefaultScalingMatrix4x4Flag[i], s);
- //else
- // Scaling_List(sps->ScalingList8x8[i-6], 64, &sps->UseDefaultScalingMatrix8x8Flag[i-6], s);
- }
- }
- }
- }
-
- sps->log2_max_frame_num_minus4 = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->pic_order_cnt_type = ue_v (pBuffer, nBufferLength, nBitOffset);
-
- if (sps->pic_order_cnt_type == 0)
- sps->log2_max_pic_order_cnt_lsb_minus4 = ue_v (pBuffer, nBufferLength, nBitOffset);
- else if (sps->pic_order_cnt_type == 1)
- {
- sps->delta_pic_order_always_zero_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- sps->offset_for_non_ref_pic = se_v (pBuffer, nBufferLength, nBitOffset);
- sps->offset_for_top_to_bottom_field = se_v (pBuffer, nBufferLength, nBitOffset);
- sps->num_ref_frames_in_pic_order_cnt_cycle = ue_v (pBuffer, nBufferLength, nBitOffset);
- for(i=0; i<sps->num_ref_frames_in_pic_order_cnt_cycle; i++)
- sps->offset_for_ref_frame[i] = se_v (pBuffer, nBufferLength, nBitOffset);
- }
- sps->num_ref_frames = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->gaps_in_frame_num_value_allowed_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- sps->pic_width_in_mbs_minus1 = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->pic_height_in_map_units_minus1 = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->frame_mbs_only_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- if (!sps->frame_mbs_only_flag)
- {
- sps->mb_adaptive_frame_field_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- }
- sps->direct_8x8_inference_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
- sps->frame_cropping_flag = u_1 (pBuffer, nBufferLength, nBitOffset);
-
- if (sps->frame_cropping_flag)
- {
- sps->frame_cropping_rect_left_offset = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->frame_cropping_rect_right_offset = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->frame_cropping_rect_top_offset = ue_v (pBuffer, nBufferLength, nBitOffset);
- sps->frame_cropping_rect_bottom_offset = ue_v (pBuffer, nBufferLength, nBitOffset);
- }
- sps->vui_parameters_present_flag = (bool) u_1 (pBuffer, nBufferLength, nBitOffset);
-
- sps->Valid = TRUE;
-}
-
-#pragma endregion
-
-
-
-
-
-/*
-
- //char strFile[MAX_PATH];
- //static int nNb = 1;
- //sprintf (strFile, "D:\\Sources\\mpc-hc\\Samples\\DXVA2\\Trace\\Prison Break S03 - E07\\BitStream_mpc_%d.bin", nNb++);
- //FILE* hOutFile = fopen (strFile, "wb");
- //if (hOutFile)
- //{
- // fwrite (pDataIn,1, nSize, hOutFile);
- // fclose (hOutFile);
- //}
- //return S_OK;
-
-
- //USES_CONVERSION;
- //static BYTE* pDurBuff = NULL;
- //static int nBuff = 1;
- //FILE* hBitstream = NULL;
- //CString strPath;
- //if (!pDurBuff) pDurBuff = new BYTE[400000];
- //strPath.Format (_T("D:\\Sources\\mpc-hc\\Samples\\DXVA2\\TraceLost S03 - E19 - The Brig\\BitStream_%d.bin"), nBuff++);
- //hBitstream = fopen (W2A(strPath), "rb");
- //nSize = fread (pDurBuff, 1, 400000, hBitstream) + 4;
- //pDataIn = pDurBuff - 4;
- //fclose(hBitstream);
-
-
- //static FILE* hPict = NULL;
- //static DXVA_PicParams_H264* pParams;
- //static int nParamIndex = 0;
- //if (!hPict)
- //{
- // hPict = fopen ("PicParam.bin", "rb");
- // pParams = (DXVA_PicParams_H264*)new BYTE [500000];
- // int nByteRead = fread (pParams, 500000, 1, hPict);
- //}
- //memcpy (&m_DXVAPicParams, &pParams[nParamIndex++], sizeof(m_DXVAPicParams));
-*/
\ No newline at end of file |