121 lines
5.2 KiB
C++
121 lines
5.2 KiB
C++
/**********
|
|
This library is free software; you can redistribute it and/or modify it under
|
|
the terms of the GNU Lesser General Public License as published by the
|
|
Free Software Foundation; either version 3 of the License, or (at your
|
|
option) any later version. (See <http://www.gnu.org/copyleft/lesser.html>.)
|
|
|
|
This library 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 Lesser General Public License for
|
|
more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with this library; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
**********/
|
|
// "liveMedia"
|
|
// Copyright (c) 1996-2024 Live Networks, Inc. All rights reserved.
|
|
// A class for generating MPEG-2 Transport Stream from one or more input
|
|
// Elementary Stream data sources
|
|
// C++ header
|
|
|
|
#ifndef _MPEG2_TRANSPORT_STREAM_MULTIPLEXOR_HH
|
|
#define _MPEG2_TRANSPORT_STREAM_MULTIPLEXOR_HH
|
|
|
|
#ifndef _FRAMED_SOURCE_HH
|
|
#include "FramedSource.hh"
|
|
#endif
|
|
#ifndef _MPEG_1OR2_DEMUX_HH
|
|
#include "MPEG1or2Demux.hh" // for SCR
|
|
#endif
|
|
|
|
#define PID_TABLE_SIZE 0x2000 // 2^13
|
|
|
|
class LIVEMEDIA_API MPEG2TransportStreamMultiplexor: public FramedSource {
|
|
public:
|
|
typedef void (onEndOfSegmentFunc)(void* clientData, double segmentDuration);
|
|
void setTimedSegmentation(unsigned segmentationDuration,
|
|
onEndOfSegmentFunc* onEndOfSegmentFunc = NULL,
|
|
void* onEndOfSegmentClientData = NULL);
|
|
// Specifies that PAT and PMT packets should be output every "segmentationDuration" seconds.
|
|
// (If "segmentationDuration" is 0 (the default value), then PAT and PMT packets are output
|
|
// at a preset frequency.)
|
|
// The optional function "onEndOfSegmentFunc" is called after each segment is output.
|
|
double currentSegmentDuration() const { return fCurrentSegmentDuration; }
|
|
// Valid only if "setTimedSegmentation()" was previously called with "segmentationDuration" > 0
|
|
|
|
Boolean canDeliverNewFrameImmediately() const { return fInputBufferBytesUsed < fInputBufferSize; }
|
|
// Can be used by a downstream reader to test whether the next call to "doGetNextFrame()"
|
|
// will deliver data immediately).
|
|
|
|
protected:
|
|
MPEG2TransportStreamMultiplexor(UsageEnvironment& env);
|
|
virtual ~MPEG2TransportStreamMultiplexor();
|
|
|
|
virtual void awaitNewBuffer(unsigned char* oldBuffer) = 0;
|
|
// implemented by subclasses
|
|
|
|
void handleNewBuffer(unsigned char* buffer, unsigned bufferSize,
|
|
int mpegVersion, MPEG1or2Demux::SCR scr, int16_t PID = -1);
|
|
// called by "awaitNewBuffer()"
|
|
// Note: For MPEG-4 video, set "mpegVersion" to 4; for H.264 video, set "mpegVersion" to 5;
|
|
// for H.265 video, set "mpegVersion" to 6.
|
|
// For Opus audio, set "mpegVersion" to 3.
|
|
// The buffer is assumed to be a PES packet, with a proper PES header.
|
|
// If "PID" is not -1, then it (currently, only the low 8 bits) is used as the stream's PID,
|
|
// otherwise the "stream_id" in the PES header is reused to be the stream's PID.
|
|
|
|
private:
|
|
// Redefined virtual functions:
|
|
virtual Boolean isMPEG2TransportStreamMultiplexor() const;
|
|
virtual void doGetNextFrame();
|
|
|
|
private:
|
|
void deliverDataToClient(u_int16_t pid, unsigned char* buffer, unsigned bufferSize,
|
|
unsigned& startPositionInBuffer);
|
|
|
|
void deliverPATPacket();
|
|
void deliverPMTPacket(Boolean hasChanged);
|
|
|
|
void setProgramStreamMap(unsigned frameSize);
|
|
|
|
protected:
|
|
Boolean fHaveVideoStreams;
|
|
|
|
private:
|
|
unsigned fOutgoingPacketCounter;
|
|
unsigned fProgramMapVersion;
|
|
u_int8_t fPreviousInputProgramMapVersion, fCurrentInputProgramMapVersion;
|
|
// These two fields are used if we see "program_stream_map"s in the input.
|
|
struct {
|
|
unsigned counter;
|
|
u_int8_t streamType; // for use in Program Maps
|
|
} fPIDState[PID_TABLE_SIZE];
|
|
u_int16_t fPCR_PID, fCurrentPID; // only the low 13 bits are used
|
|
MPEG1or2Demux::SCR fPCR;
|
|
unsigned char* fInputBuffer;
|
|
unsigned fInputBufferSize, fInputBufferBytesUsed;
|
|
Boolean fIsFirstAdaptationField;
|
|
unsigned fSegmentationDuration;
|
|
// if nonzero, this is the number of seconds between successive 'segments'. Each 'segment'
|
|
// begins with a PAT, followed by a PMT.
|
|
// if zero (the default value), then the frequency of PATs and PMTs depends on the constants
|
|
// PAT_PERIOD_IF_UNTIMED and PMT_PERIOD_IF_UNTIMED, defined in the .cpp file.
|
|
Boolean segmentationIsTimed() const { return fSegmentationDuration > 0; }
|
|
u_int8_t fSegmentationIndication;
|
|
// used only if fSegmentationDuration > 0:
|
|
// 1 if a segment has just ended and the next packet is to be a PAT
|
|
// 2 if a segment has just ended and the following PAT has been sent; a PMT is next
|
|
// 0 otherwise
|
|
double fCurrentSegmentDuration, fPreviousPTS; // used only if fSegmentationDuration > 0
|
|
onEndOfSegmentFunc* fOnEndOfSegmentFunc; // used only if fSegmentationDuration > 0
|
|
void* fOnEndOfSegmentClientData; // ditto
|
|
};
|
|
|
|
|
|
// The CRC calculation function that Transport Streams use. We make this function public
|
|
// here in case it's useful elsewhere:
|
|
u_int32_t calculateCRC(u_int8_t const* data, unsigned dataLength, u_int32_t initialValue = 0xFFFFFFFF);
|
|
|
|
#endif
|