113 lines
3.3 KiB
C++
113 lines
3.3 KiB
C++
#include "adts.h"
|
|
#include "h265.h"
|
|
#include "tinythread.h"
|
|
#include "ts_packet.h"
|
|
#include <deque>
|
|
#include <map>
|
|
#include <set>
|
|
|
|
#include "shared_memory.h"
|
|
#define TS_PTS_ROLLOVER 95443718
|
|
|
|
namespace TS{
|
|
enum codecType{
|
|
H264 = 0x1B,
|
|
AAC = 0x0F,
|
|
AC3 = 0x81,
|
|
MP3 = 0x03,
|
|
H265 = 0x24,
|
|
ID3 = 0x15,
|
|
MPEG2 = 0x02,
|
|
MP2 = 0x03,
|
|
META = 0x06
|
|
};
|
|
|
|
class ADTSRemainder{
|
|
private:
|
|
char *data;
|
|
uint64_t max;
|
|
uint64_t now;
|
|
uint64_t len;
|
|
uint64_t bpos;
|
|
|
|
public:
|
|
void setRemainder(const aac::adts &p, const void *source, uint32_t avail, uint64_t bPos);
|
|
|
|
ADTSRemainder();
|
|
~ADTSRemainder();
|
|
uint64_t getLength();
|
|
uint64_t getBpos();
|
|
uint64_t getTodo();
|
|
char *getData();
|
|
|
|
void append(const char *p, uint32_t pLen);
|
|
bool isComplete();
|
|
void clear();
|
|
};
|
|
|
|
class Stream{
|
|
public:
|
|
Stream(bool _threaded = false);
|
|
~Stream();
|
|
void add(char *newPack, uint64_t bytePos = 0);
|
|
void add(Packet &newPack, uint64_t bytePos = 0);
|
|
void parse(Packet &newPack, uint64_t bytePos);
|
|
void parse(char *newPack, uint64_t bytePos);
|
|
void parse(size_t tid);
|
|
void parseNal(size_t tid, const char *pesPayload, const char *packetPtr, bool &isKeyFrame);
|
|
bool hasPacketOnEachTrack() const;
|
|
bool hasPacket(size_t tid) const;
|
|
bool hasPacket() const;
|
|
void getPacket(size_t tid, DTSC::Packet &pack, size_t mappedAs = INVALID_TRACK_ID);
|
|
uint32_t getEarliestPID();
|
|
void getEarliestPacket(DTSC::Packet &pack);
|
|
void initializeMetadata(DTSC::Meta &meta, size_t tid = INVALID_TRACK_ID, size_t mappingId = INVALID_TRACK_ID);
|
|
void partialClear();
|
|
void clear();
|
|
void finish();
|
|
void eraseTrack(size_t tid);
|
|
bool isDataTrack(size_t tid) const;
|
|
void parseBitstream(size_t tid, const char *pesPayload, uint64_t realPayloadSize,
|
|
uint64_t timeStamp, int64_t timeOffset, uint64_t bPos, bool alignment);
|
|
std::set<size_t> getActiveTracks();
|
|
|
|
void setLastms(size_t tid, uint64_t timestamp);
|
|
|
|
private:
|
|
uint64_t lastPAT;
|
|
ProgramAssociationTable associationTable;
|
|
std::map<size_t, ADTSRemainder> remainders;
|
|
|
|
std::map<size_t, uint64_t> lastPMT;
|
|
std::map<size_t, ProgramMappingTable> mappingTable;
|
|
|
|
std::map<size_t, std::deque<Packet> > pesStreams;
|
|
std::deque<Packet> *psCache; /// Used only for internal speed optimizes.
|
|
uint32_t psCacheTid; /// Used only for internal speed optimizes.
|
|
std::map<size_t, std::deque<uint64_t> > pesPositions;
|
|
std::map<size_t, std::deque<DTSC::Packet> > outPackets;
|
|
std::map<size_t, DTSC::Packet> buildPacket;
|
|
std::map<size_t, uint32_t> pidToCodec;
|
|
std::map<size_t, aac::adts> adtsInfo;
|
|
std::map<size_t, std::string> spsInfo;
|
|
std::map<size_t, std::string> ppsInfo;
|
|
std::map<size_t, h265::initData> hevcInfo;
|
|
std::map<size_t, std::string> metaInit;
|
|
std::map<size_t, std::string> descriptors;
|
|
std::map<size_t, uint32_t> seenUnitStart;
|
|
std::map<size_t, std::string> mpeg2SeqHdr;
|
|
std::map<size_t, std::string> mpeg2SeqExt;
|
|
std::map<size_t, std::string> mp2Hdr;
|
|
|
|
std::map<size_t, size_t> rolloverCount;
|
|
std::map<size_t, unsigned long long> lastms;
|
|
|
|
mutable tthread::recursive_mutex tMutex;
|
|
|
|
bool threaded;
|
|
|
|
std::set<size_t> pmtTracks;
|
|
|
|
void parsePES(size_t tid, bool finished = false);
|
|
};
|
|
}// namespace TS
|