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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/*
* Copyright (C) 2013 Jean-Luc Barriere
* This file is part of Kodi - https://kodi.tv
*
* SPDX-License-Identifier: GPL-2.0-or-later
* See LICENSES/README.md for more information.
*/
#ifndef TSDEMUXER_H
#define TSDEMUXER_H
#include "tsPacket.h"
#include "elementaryStream.h"
#include "mutex.h"
#include <map>
#include <vector>
#define FLUTS_NORMAL_TS_PACKETSIZE 188
#define FLUTS_M2TS_TS_PACKETSIZE 192
#define FLUTS_DVB_ASI_TS_PACKETSIZE 204
#define FLUTS_ATSC_TS_PACKETSIZE 208
#define AV_CONTEXT_PACKETSIZE 208
#define TS_CHECK_MIN_SCORE 2
#define TS_CHECK_MAX_SCORE 10
namespace TSDemux
{
class TSDemuxer
{
public:
virtual bool ReadAV(uint64_t pos, unsigned char* buffer, size_t len) = 0;
};
enum {
AVCONTEXT_TS_ERROR = -3,
AVCONTEXT_IO_ERROR = -2,
AVCONTEXT_TS_NOSYNC = -1,
AVCONTEXT_CONTINUE = 0,
AVCONTEXT_PROGRAM_CHANGE = 1,
AVCONTEXT_STREAM_PID_DATA = 2,
AVCONTEXT_DISCONTINUITY = 3
};
class AVContext
{
public:
AVContext(TSDemuxer* const demux, uint64_t pos, uint16_t channel);
void Reset(void);
uint16_t GetPID() const;
PACKET_TYPE GetPIDType() const;
uint16_t GetPIDChannel() const;
bool HasPIDStreamData() const;
bool HasPIDPayload() const;
ElementaryStream* GetPIDStream();
std::vector<ElementaryStream*> GetStreams();
void StartStreaming(uint16_t pid);
void StopStreaming(uint16_t pid);
ElementaryStream* GetStream(uint16_t pid) const;
uint16_t GetChannel(uint16_t pid) const;
void ResetPackets();
// TS parser
int TSResync();
uint64_t GoNext();
uint64_t Shift();
void GoPosition(uint64_t pos, bool rp);
uint64_t GetPosition() const;
uint64_t GetRecoveryPos() { return payload_unit_pos ? payload_unit_pos : av_pos; };
uint64_t GetNextPosition() const;
int ProcessTSPacket();
int ProcessTSPayload();
private:
AVContext(const AVContext&);
AVContext& operator=(const AVContext&);
int configure_ts();
static STREAM_TYPE get_stream_type(uint8_t pes_type);
static uint8_t av_rb8(const unsigned char* p);
static uint16_t av_rb16(const unsigned char* p);
static uint32_t av_rb32(const unsigned char* p);
static uint64_t decode_pts(const unsigned char* p);
void clear_pmt();
void clear_pes(uint16_t channel);
int parse_ts_psi();
static STREAM_INFO parse_pes_descriptor(const unsigned char* p, size_t len, STREAM_TYPE* st);
int parse_ts_pes();
// Critical section
mutable PLATFORM::CMutex mutex;
// AV stream owner
TSDemuxer* m_demux;
// Raw packet buffer
uint64_t av_pos;
uint64_t payload_unit_pos;
uint64_t prev_payload_unit_pos;
size_t av_data_len;
size_t av_pkt_size;
unsigned char av_buf[AV_CONTEXT_PACKETSIZE];
// TS Streams context
bool is_configured;
uint16_t channel;
std::map<uint16_t, Packet> packets;
// Packet context
uint16_t pid;
bool transport_error;
bool has_payload;
bool payload_unit_start;
bool discontinuity;
const unsigned char* payload;
size_t payload_len;
Packet* packet;
};
}
#endif /* TSDEMUXER_H */
|
