Welcome to mirror list, hosted at ThFree Co, Russian Federation.

H264Nalu.h « DSUtil « src - github.com/mpc-hc/mpc-hc.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 67548a264c0ce755ebbdcd4e4c64960d8e766187 (plain)
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
/* 
 * $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();
};