diff options
Diffstat (limited to 'command/demux.h')
| -rw-r--r-- | command/demux.h | 331 |
1 files changed, 307 insertions, 24 deletions
diff --git a/command/demux.h b/command/demux.h index 0c91c7a..4213479 100644 --- a/command/demux.h +++ b/command/demux.h @@ -8,39 +8,322 @@ #ifndef __demux_h_ #define __demux_h_ -#ifndef TS_SIZE -#define TS_SIZE 188 +#include <stdint.h> + +#ifndef uchar +typedef unsigned char uchar; #endif -#define PESHDRSIZE 6 +#define PACKET_MASK 0xF0 +#define PACKET_VIDEO 0x10 +#define PACKET_H262 0x10 // 0x00 0x00 0x01 (PES / H262) +#define PACKET_H264 0x11 // 0x00 0x00 0x00 0x01 (H264) +#define PACKET_AUDIO 0x20 +#define PACKET_AC3 0x20 +#define PACKET_MP2 0x21 + +typedef struct AvPacket +{ + uchar *Data; + int Length; + int Type; +} AvPacket; + +struct TSHDR +{ +unsigned Sync: + 8; +unsigned PidH: + 5; +unsigned Priority: + 1; +unsigned PayloadStart: + 1; +unsigned TError: + 1; +unsigned PidL: + 8; +unsigned Counter: + 4; +unsigned AFC: + 2; +unsigned TSC: + 2; +}; + +struct PESHDR +{ + uchar Sync1; + uchar Sync2; + uchar Sync3; + uchar StreamID; + uchar LenH; + uchar LenL; +}; + +#pragma pack(1) +struct PESHDROPT +{ +unsigned OOC: + 1; +unsigned CY: + 1; +unsigned DAI: + 1; +unsigned PESP: + 1; +unsigned PESSC: + 2; +unsigned MarkerBits: + 2; +unsigned EXT: + 1; +unsigned CRC: + 1; +unsigned ACI: + 1; +unsigned TM: + 1; +unsigned RATE: + 1; +unsigned ESCR: + 1; +unsigned PTSDTS: + 2; +unsigned Length: + 8; +}; +#pragma pack() + +// ---------------------------------------------------------------------------- + +class cPaketQueue +{ + struct MP2HDR + { +unsigned Sync1: + 8; +unsigned Protection: + 1; +unsigned Layer: + 2; +unsigned MpegID: + 2; +unsigned Sync2: + 3; +unsigned Private: + 1; +unsigned Padding: + 1; +unsigned SampleRateIndex: + 2; +unsigned BitRateIndex: + 4; +unsigned Emphasis: + 2; +unsigned Original: + 1; +unsigned Copyright: + 1; +unsigned ModeExt: + 2; +unsigned Mode: + 2; + }; + +#pragma pack(1) + struct AC3HDR + { +unsigned Sync1: + 8; +unsigned Sync2: + 8; +unsigned CRC1: + 8; +unsigned CRC2: + 8; +unsigned FrameSizeIndex: + 6; +unsigned SampleRateIndex: + 2; + }; +#pragma pack() + +private: + const char *name; + struct pktinfo + { + int pkthdr; + int pktsyncsize; + int streamsize; + bool ispes; + } pktinfo; + + int percent; + int mpercent; // max percentage use + + uchar *buffer; + int maxqueue; + int inptr; + int outptr; + + int skipped; -#include "queue.h" -#include "ts2pkt.h" -#include "pes2es.h" + uint32_t scanner; + int scannerstart; -class cMarkAdDemux + int findaudioheader(int start, int *framesize, int *headersize, bool ac3); + int findpktheader(int start, int *streamsize,int *headersize, bool longstartcode, bool pesonly); +public: + cPaketQueue(const char *Name, int Size=32768); + ~cPaketQueue(); + int Length() + { + return inptr-outptr; + } + const char *Name() + { + return name; + } + void Clear(); + int Skipped() + { + return skipped; + } + bool Put(uchar *Data, int Size); + uchar *Get(int *Size); + uchar *Peek(int Size); + uchar *GetPacket(int *Size, int Type); + void Resize(int NewSize, const char *NewName=NULL); + bool NeedMoreData() + { + if (scannerstart==-1) return true; + if (pktinfo.streamsize) + { + return (((inptr-pktinfo.pkthdr)-outptr)<=(pktinfo.streamsize+6)); + } + else + { + return ((scannerstart==inptr) || (scannerstart==outptr)); + } + } + int FindPesHeader(int Start); +}; + +// ---------------------------------------------------------------------------- + +class cTS2Pkt { +private: + cPaketQueue *queue; + int counter; + bool sync; + bool firstsync; + int skipped; + int pid; + bool h264; + bool noticeFILLER; + bool noticeSEQUENCE; + bool noticeSTREAM; +public: + cTS2Pkt(int Pid, const char *QueueName="TS2Pkt", int QueueSize=32768, bool H264=false); + ~cTS2Pkt(); + void Clear(); + int Skipped() + { + return skipped; + } + bool Process(uchar *TSData, int TSSize, AvPacket *Pkt); + void Resize(int NewQueueSize, const char *NewQueueName) + { + queue->Resize(NewQueueSize, NewQueueName); + } + bool NeedMoreData() + { + if (queue) + { + return queue->NeedMoreData(); + } + else + { + return false; + } + } +}; +// ---------------------------------------------------------------------------- + +class cPES2ES +{ private: - cMarkAdTS2Pkt *ts2pkt; - cMarkAdPES2ES *pes2audioes; - cMarkAdPES2ES *pes2videoes; - cMarkAdPaketQueue *queue; - - bool pause; - int pause_retval; - int min_needed; - int skip; - - int GetMinNeeded(MarkAdPid Pid, uchar *Data, int Count, bool *Offcnt); - void ProcessTS(MarkAdPid Pid, uchar *Data, int Count, MarkAdPacket *pkt); - void ProcessVDR(MarkAdPid Pid, uchar *Data, int Count, MarkAdPacket *pkt); - void VDRTSAddPATPMT2Offset(MarkAdPid Pid, uchar *Data, int Count, bool *Offcnt); + cPaketQueue *queue; + int skipped; + int ptype; + bool h264; +public: + cPES2ES(int PacketType, const char *QueueName="PES2ES", int QueueSize=32768); + ~cPES2ES(); + void Clear(); + bool Process(uchar *PESData, int PESSize, AvPacket *ESPkt); + int Skipped() + { + return skipped; + } + bool NeedMoreData() + { + if (queue) + { + return queue->NeedMoreData(); + } + else + { + return false; + } + } +}; + +// ---------------------------------------------------------------------------- + +class cDemux +{ +private: + int vpid,dpid,apid; + int stream_or_pid; + int skipped; + bool h264; + bool TS; + uint64_t offset; + int from_oldfile; + int last_bplen; + + bool vdrcount; + int vdroffset; + bool vdraddpatpmt(uchar *data, int count); + + void addoffset(); + + cPaketQueue *queue; + cPES2ES *pes2videoes; + cPES2ES *pes2audioes_mp2; + cPES2ES *pes2audioes_ac3; + cTS2Pkt *ts2pkt_vpid; + cTS2Pkt *ts2pkt_dpid; + cTS2Pkt *ts2pkt_apid; + bool needmoredata(); + int checkts(uchar *data, int count, int &pid); + bool isvideopes(uchar *data, int count); + int fillqueue(uchar *data, int count, int &stream_or_pid, int &packetsize, int &readout); public: - cMarkAdDemux(); - ~cMarkAdDemux(); + cDemux(int VPid, int DPid, int APid, bool H264=false, bool VDRCount=false); + ~cDemux(); void Clear(); - int Process(MarkAdPid Pid, uchar *Data, int Count, MarkAdPacket *pkt); + int Skipped(); + void NewFile(); + uint64_t Offset() + { + return offset; + } + int Process(uchar *Data, int Count, AvPacket *pkt); }; #endif |