1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
/*
* $Id$
*
* (C) 2006-2010 see AUTHORS
*
* This file is part of mplayerc.
*
* Mplayerc 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 3 of the License, or
* (at your option) any later version.
*
* Mplayerc 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, see <http://www.gnu.org/licenses/>.
*
*/
#pragma once
typedef enum
{
NALU_TYPE_SLICE = 1,
NALU_TYPE_DPA = 2,
NALU_TYPE_DPB = 3,
NALU_TYPE_DPC = 4,
NALU_TYPE_IDR = 5,
NALU_TYPE_SEI = 6,
NALU_TYPE_SPS = 7,
NALU_TYPE_PPS = 8,
NALU_TYPE_AUD = 9,
NALU_TYPE_EOSEQ = 10,
NALU_TYPE_EOSTREAM = 11,
NALU_TYPE_FILL = 12
} NALU_TYPE;
class CH264Nalu
{
private :
int forbidden_bit; //! should be always FALSE
int nal_reference_idc; //! NALU_PRIORITY_xxxx
NALU_TYPE nal_unit_type; //! NALU_TYPE_xxxx
int m_nNALStartPos; //! NALU start (including startcode / size)
int m_nNALDataPos; //! Useful part
unsigned m_nDataLen; //! Length of the NAL unit (Excluding the start code, which does not belong to the NALU)
BYTE* m_pBuffer;
int m_nCurPos;
int m_nNextRTP;
int m_nSize;
int m_nNALSize;
bool MoveToNextStartcode();
public :
NALU_TYPE GetType() const
{
return nal_unit_type;
};
bool IsRefFrame() const
{
return (nal_reference_idc != 0);
};
int GetDataLength() const
{
return m_nCurPos - m_nNALDataPos;
};
BYTE* GetDataBuffer()
{
return m_pBuffer + m_nNALDataPos;
};
int GetRoundedDataLength() const
{
int nSize = m_nCurPos - m_nNALDataPos;
return nSize + 128 - (nSize % 128);
}
int GetLength() const
{
return m_nCurPos - m_nNALStartPos;
};
BYTE* GetNALBuffer()
{
return m_pBuffer + m_nNALStartPos;
};
bool IsEOF() const
{
return m_nCurPos >= m_nSize;
};
void SetBuffer(BYTE* pBuffer, int nSize, int nNALSize);
bool ReadNext();
};
|
