diff options
Diffstat (limited to 'src/DSUtil/H264Nalu.cpp')
| -rw-r--r-- | src/DSUtil/H264Nalu.cpp | 108 |
1 files changed, 54 insertions, 54 deletions
diff --git a/src/DSUtil/H264Nalu.cpp b/src/DSUtil/H264Nalu.cpp index 7f5015b57..8286ce9b8 100644 --- a/src/DSUtil/H264Nalu.cpp +++ b/src/DSUtil/H264Nalu.cpp @@ -25,79 +25,79 @@ void CH264Nalu::SetBuffer(BYTE* pBuffer, int nSize, int nNALSize)
{
- m_pBuffer = pBuffer;
- m_nSize = nSize;
- m_nNALSize = nNALSize;
- m_nCurPos = 0;
- m_nNextRTP = 0;
+ m_pBuffer = pBuffer;
+ m_nSize = nSize;
+ m_nNALSize = nNALSize;
+ m_nCurPos = 0;
+ m_nNextRTP = 0;
- m_nNALStartPos = 0;
- m_nNALDataPos = 0;
+ m_nNALStartPos = 0;
+ m_nNALDataPos = 0;
- if (!nNALSize && nSize) {
- MoveToNextAnnexBStartcode();
- }
+ if (!nNALSize && nSize) {
+ MoveToNextAnnexBStartcode();
+ }
}
bool CH264Nalu::MoveToNextAnnexBStartcode()
{
- int nBuffEnd = m_nSize - 4;
+ int nBuffEnd = m_nSize - 4;
- for (int i=m_nCurPos; i<nBuffEnd; i++) {
- if ((*((DWORD*)(m_pBuffer+i)) & 0x00FFFFFF) == 0x00010000) {
- // Find next AnnexB Nal
- m_nCurPos = i;
- return true;
- }
- }
+ for (int i = m_nCurPos; i < nBuffEnd; i++) {
+ if ((*((DWORD*)(m_pBuffer + i)) & 0x00FFFFFF) == 0x00010000) {
+ // Find next AnnexB Nal
+ m_nCurPos = i;
+ return true;
+ }
+ }
- m_nCurPos = m_nSize;
- return false;
+ m_nCurPos = m_nSize;
+ return false;
}
bool CH264Nalu::MoveToNextRTPStartcode()
{
- if (m_nNextRTP < m_nSize) {
- m_nCurPos = m_nNextRTP;
- return true;
- }
+ if (m_nNextRTP < m_nSize) {
+ m_nCurPos = m_nNextRTP;
+ return true;
+ }
- m_nCurPos = m_nSize;
- return false;
+ m_nCurPos = m_nSize;
+ return false;
}
bool CH264Nalu::ReadNext()
{
- if ((m_nCurPos >= m_nSize) || (m_nCurPos<0)) {
- return false;
- }
+ if ((m_nCurPos >= m_nSize) || (m_nCurPos < 0)) {
+ return false;
+ }
- if ((m_nNALSize != 0) && (m_nCurPos == m_nNextRTP)) {
- // RTP Nalu type : (XX XX) XX XX NAL..., with XX XX XX XX or XX XX equal to NAL size
- m_nNALStartPos = m_nCurPos;
- m_nNALDataPos = m_nCurPos + m_nNALSize;
- unsigned nTemp = 0;
- for (int i=0; i<m_nNALSize; i++) {
- nTemp = (nTemp << 8) + m_pBuffer[m_nCurPos++];
- }
- m_nNextRTP += nTemp + m_nNALSize;
- MoveToNextRTPStartcode();
- } else {
- // Remove trailing bits
- while (m_pBuffer[m_nCurPos]==0x00 && ((*((DWORD*)(m_pBuffer+m_nCurPos)) & 0x00FFFFFF) != 0x00010000)) {
- m_nCurPos++;
- }
+ if ((m_nNALSize != 0) && (m_nCurPos == m_nNextRTP)) {
+ // RTP Nalu type : (XX XX) XX XX NAL..., with XX XX XX XX or XX XX equal to NAL size
+ m_nNALStartPos = m_nCurPos;
+ m_nNALDataPos = m_nCurPos + m_nNALSize;
+ unsigned nTemp = 0;
+ for (int i = 0; i < m_nNALSize; i++) {
+ nTemp = (nTemp << 8) + m_pBuffer[m_nCurPos++];
+ }
+ m_nNextRTP += nTemp + m_nNALSize;
+ MoveToNextRTPStartcode();
+ } else {
+ // Remove trailing bits
+ while (m_pBuffer[m_nCurPos] == 0x00 && ((*((DWORD*)(m_pBuffer + m_nCurPos)) & 0x00FFFFFF) != 0x00010000)) {
+ m_nCurPos++;
+ }
- // AnnexB Nalu : 00 00 01 NAL...
- m_nNALStartPos = m_nCurPos;
- m_nCurPos += 3;
- m_nNALDataPos = m_nCurPos;
- MoveToNextAnnexBStartcode();
- }
+ // AnnexB Nalu : 00 00 01 NAL...
+ m_nNALStartPos = m_nCurPos;
+ m_nCurPos += 3;
+ m_nNALDataPos = m_nCurPos;
+ MoveToNextAnnexBStartcode();
+ }
- forbidden_bit = (m_pBuffer[m_nNALDataPos]>>7) & 1;
- nal_reference_idc = (m_pBuffer[m_nNALDataPos]>>5) & 3;
- nal_unit_type = (NALU_TYPE) (m_pBuffer[m_nNALDataPos] & 0x1f);
+ forbidden_bit = (m_pBuffer[m_nNALDataPos] >> 7) & 1;
+ nal_reference_idc = (m_pBuffer[m_nNALDataPos] >> 5) & 3;
+ nal_unit_type = (NALU_TYPE)(m_pBuffer[m_nNALDataPos] & 0x1f);
- return true;
+ return true;
}
|