293 lines
8.4 KiB
C++
293 lines
8.4 KiB
C++
/// \file ts_packet.cpp
|
|
/// Holds all code for the TS namespace.
|
|
|
|
#include "ts_packet.h"
|
|
|
|
/// This constructor creates an empty TS::Packet, ready for use for either reading or writing.
|
|
/// All this constructor does is call TS::Packet::Clear().
|
|
TS::Packet::Packet() {
|
|
Clear( );
|
|
}
|
|
|
|
/// This constructor creates a filled TS::Packet, or creates an empty
|
|
/// packet if not enough Data provided.
|
|
/// It fills the content with the first 188 bytes of Data.
|
|
/// \param Data The data to be read into the packet.
|
|
TS::Packet::Packet( std::string & Data ) {
|
|
if( Data.size() < 188 ) {
|
|
Clear( );
|
|
} else {
|
|
for( int i = 0; i < 188; i++ ) {
|
|
Buffer[i] = Data[i];
|
|
}
|
|
Data.erase(0,188);
|
|
Free = 0;
|
|
}
|
|
}
|
|
|
|
/// The deconstructor deletes all space that may be occupied by a TS::Packet.
|
|
TS::Packet::~Packet() { }
|
|
|
|
/// Sets the PID of a single TS::Packet.
|
|
/// \param NewPID The new PID of the packet.
|
|
void TS::Packet::PID( int NewPID ) {
|
|
Buffer[1] = (Buffer[1] & 0xE0) + ((NewPID & 0x1F00) >> 8 );
|
|
Buffer[2] = (NewPID & 0x00FF);
|
|
Free = std::min( Free, 184 );
|
|
}
|
|
|
|
/// Gets the PID of a single TS::Packet.
|
|
/// \return The value of the PID.
|
|
int TS::Packet::PID() {
|
|
return (( Buffer[1] & 0x1F ) << 8 ) + Buffer[2];
|
|
}
|
|
|
|
/// Sets the Continuity Counter of a single TS::Packet
|
|
/// \param NewContinuity The new Continuity Counter of the packet.
|
|
void TS::Packet::ContinuityCounter( int NewContinuity ) {
|
|
Buffer[3] = ( Buffer[3] & 0xF0 ) + ( NewContinuity & 0x0F );
|
|
Free = std::min( Free, 184 );
|
|
}
|
|
|
|
/// Gets the Continuity Counter of a single TS::Packet.
|
|
/// \return The value of the Continuity Counter.
|
|
int TS::Packet::ContinuityCounter() {
|
|
return ( Buffer[3] & 0x0F );
|
|
}
|
|
|
|
/// Gets the amount of bytes that are not written yet in a TS::Packet
|
|
/// \return The amount of bytes that can still be written to this packet.
|
|
int TS::Packet::BytesFree( ) {
|
|
return Free;
|
|
}
|
|
|
|
/// Clears a TS::Packet
|
|
void TS::Packet::Clear( ) {
|
|
Free = 184;
|
|
Buffer[0] = 0x47;
|
|
for( int i = 1; i < 188; i++ ) {
|
|
Buffer[i] = 0x00;
|
|
}
|
|
AdaptationField( 1 );
|
|
}
|
|
|
|
/// Sets the selection value for an adaptationfield of a TS::Packet
|
|
/// \param NewSelector The new value of the selection bits.
|
|
/// - 1: No AdaptationField
|
|
/// - 2: AdaptationField Only
|
|
/// - 3: AdaptationField followed by Data
|
|
void TS::Packet::AdaptationField( int NewSelector ) {
|
|
Buffer[3] = ( Buffer[3] & 0xCF ) + ((NewSelector & 0x03) << 4);
|
|
Buffer[4] = 0;
|
|
Free = std::min( Free, 184 );
|
|
}
|
|
|
|
int TS::Packet::AdaptationField( ) {
|
|
return ((Buffer[3] & 0x30) >> 4 );
|
|
}
|
|
|
|
void TS::Packet::PCR( int64_t NewVal ) {
|
|
NewVal += (0xF618 * 300);
|
|
AdaptationField( 3 );
|
|
Buffer[4] = 7;
|
|
Buffer[5] = (Buffer[5] | 0x10 );
|
|
int64_t TmpVal = NewVal / 300;
|
|
Buffer[6] = (((TmpVal>>1)>>24) & 0xFF);
|
|
Buffer[7] = (((TmpVal>>1)>>16) & 0xFF);
|
|
Buffer[8] = (((TmpVal>>1)>>8) & 0xFF);
|
|
Buffer[9] = ((TmpVal>>1) & 0xFF);
|
|
int Remainder = NewVal % 300;
|
|
Buffer[10] = 0x7E + ((TmpVal & 0x01)<<7) + ((Remainder & 0x0100) >> 8 );
|
|
Buffer[11] = (Remainder & 0x00FF);
|
|
Free = std::min( Free, 176 );
|
|
};
|
|
|
|
int64_t TS::Packet::PCR( ) {
|
|
if( !AdaptationField() ) {
|
|
return -1;
|
|
}
|
|
if( !(Buffer[5] & 0x10 ) ) {
|
|
return -1;
|
|
}
|
|
int64_t Result = 0;
|
|
Result = (((((((Buffer[6] << 8) + Buffer[7]) << 8) + Buffer[8]) << 8) + Buffer[9]) << 1) + ( Buffer[10] & 0x80 );
|
|
Result = Result * 300;
|
|
Result += ((Buffer[10] & 0x01) << 8 + Buffer[11]);
|
|
return Result;
|
|
}
|
|
|
|
int TS::Packet::AdaptationFieldLen( ) {
|
|
if( !AdaptationField() ) {
|
|
return -1;
|
|
}
|
|
return (int)Buffer[4];
|
|
}
|
|
|
|
void TS::Packet::Print( ) {
|
|
std::cout << "TS Packet: " << (Buffer[0] == 0x47)
|
|
<< "\n\tNewUnit: " << UnitStart()
|
|
<< "\n\tPID: " << PID()
|
|
<< "\n\tContinuity Counter: " << ContinuityCounter()
|
|
<< "\n\tAdaption Field: " << AdaptationField() << "\n";
|
|
if( AdaptationField() ) {
|
|
std::cout << "\t\tAdaption Field Length: " << AdaptationFieldLen() << "\n";
|
|
if( AdaptationFieldLen() ) {
|
|
std::cout << "\t\tRandom Access: " << RandomAccess() << "\n";
|
|
}
|
|
if( PCR() != -1 ) {
|
|
std::cout << "\t\tPCR: " << PCR() << "( " << (double)PCR() / 27000000 << " s )\n";
|
|
}
|
|
}
|
|
}
|
|
|
|
int TS::Packet::UnitStart( ) {
|
|
return ( Buffer[1] & 0x40) >> 6;
|
|
}
|
|
|
|
void TS::Packet::UnitStart( int NewVal ) {
|
|
if( NewVal ) {
|
|
Buffer[1] = (Buffer[1] | 0x40);
|
|
} else {
|
|
Buffer[1] = (Buffer[1] & 0xBF);
|
|
}
|
|
}
|
|
|
|
int TS::Packet::RandomAccess( ) {
|
|
if( !AdaptationField() ) {
|
|
return -1;
|
|
}
|
|
return ( Buffer[5] & 0x40) >> 6;
|
|
}
|
|
|
|
void TS::Packet::RandomAccess( int NewVal ) {
|
|
if( AdaptationField() ) {
|
|
if( Buffer[4] == 0 ) {
|
|
Buffer[4] = 1;
|
|
}
|
|
if( NewVal ) {
|
|
Buffer[5] = (Buffer[5] | 0x40);
|
|
} else {
|
|
Buffer[5] = (Buffer[5] & 0xBF);
|
|
}
|
|
} else {
|
|
AdaptationField( 3 );
|
|
Buffer[4] = 1;
|
|
if( NewVal ) {
|
|
Buffer[5] = 0x40;
|
|
} else {
|
|
Buffer[5] = 0x00;
|
|
}
|
|
|
|
}
|
|
Free = std::min( Free, 182 );
|
|
}
|
|
|
|
void TS::Packet::DefaultPAT( ) {
|
|
static int MyCntr = 0;
|
|
std::copy( TS::PAT, TS::PAT + 188, Buffer );
|
|
ContinuityCounter( MyCntr );
|
|
Free = 0;
|
|
MyCntr = ( (MyCntr + 1) % 0x10);
|
|
}
|
|
|
|
void TS::Packet::DefaultPMT( ) {
|
|
static int MyCntr = 0;
|
|
std::copy( TS::PMT, TS::PMT + 188, Buffer );
|
|
ContinuityCounter( MyCntr );
|
|
Free = 0;
|
|
MyCntr = ( (MyCntr + 1) % 0x10);
|
|
}
|
|
|
|
std::string TS::Packet::ToString( ) {
|
|
std::string Result( Buffer, 188 );
|
|
std::cout.write( Buffer,188 );
|
|
return Result;
|
|
}
|
|
|
|
void TS::Packet::PESVideoLeadIn( int NewLen ) {
|
|
static int PTS = 27000000;
|
|
NewLen += 14;
|
|
int Offset = ( 188 - Free );
|
|
Buffer[Offset] = 0x00;//PacketStartCodePrefix
|
|
Buffer[Offset+1] = 0x00;//PacketStartCodePrefix (Cont)
|
|
Buffer[Offset+2] = 0x01;//PacketStartCodePrefix (Cont)
|
|
Buffer[Offset+3] = 0xe0;//StreamType Video
|
|
Buffer[Offset+4] = (NewLen & 0xFF00) >> 8;//PES PacketLength
|
|
Buffer[Offset+5] = (NewLen & 0x00FF);//PES PacketLength (Cont)
|
|
Buffer[Offset+6] = 0x80;//Reserved + Flags
|
|
Buffer[Offset+7] = 0x80;//PTSOnlyFlag + Flags
|
|
Buffer[Offset+8] = 0x05;//PESHeaderDataLength
|
|
Buffer[Offset+9] = 0x20 + ((PTS & 0x1C0000000) >> 29 ) + 1;//PTS
|
|
Buffer[Offset+10] = 0x00 + ((PTS & 0x03FC00000) >> 22 );//PTS (Cont)
|
|
Buffer[Offset+11] = 0x00 + ((PTS & 0x0003F8000) >> 14 ) + 1;//PTS (Cont)
|
|
Buffer[Offset+12] = 0x00 + ((PTS & 0x000007F80) >> 7 );//PTS (Cont)
|
|
Buffer[Offset+13] = 0x00 + ((PTS & 0x00000007F) << 1) + 1;//PTS (Cont)
|
|
|
|
//PesPacket-Wise Prepended Data
|
|
|
|
Buffer[Offset+14] = 0x00;//NALU StartCode
|
|
Buffer[Offset+15] = 0x00;//NALU StartCode (Cont)
|
|
Buffer[Offset+16] = 0x00;//NALU StartCode (Cont)
|
|
Buffer[Offset+17] = 0x01;//NALU StartCode (Cont)
|
|
Buffer[Offset+18] = 0x09;//NALU EndOfPacket (Einde Vorige Packet)
|
|
Buffer[Offset+19] = 0xF0;//NALU EndOfPacket (Cont)
|
|
Free = Free - 20;
|
|
PTS += 3003;
|
|
}
|
|
|
|
void TS::Packet::PESAudioLeadIn( int NewLen, uint64_t PTS ) {
|
|
PTS = PTS * 90;
|
|
NewLen += 8;
|
|
int Offset = ( 188 - Free ) - 2;
|
|
Buffer[Offset] = 0x00;//PacketStartCodePrefix
|
|
Buffer[Offset+1] = 0x00;//PacketStartCodePrefix (Cont)
|
|
Buffer[Offset+2] = 0x01;//PacketStartCodePrefix (Cont)
|
|
Buffer[Offset+3] = 0xc0;//StreamType Video
|
|
|
|
Buffer[Offset+4] = (NewLen & 0xFF00) >> 8;//PES PacketLength
|
|
Buffer[Offset+5] = (NewLen & 0x00FF);//PES PacketLength (Cont)
|
|
Buffer[Offset+6] = 0x80;//Reserved + Flags
|
|
Buffer[Offset+7] = 0x80;//PTSOnlyFlag + Flags
|
|
Buffer[Offset+8] = 0x05;//PESHeaderDataLength
|
|
Buffer[Offset+9] = 0x20 + ((PTS & 0x1C0000000) >> 29 ) + 1;//PTS
|
|
Buffer[Offset+10] = 0x00 + ((PTS & 0x03FC00000) >> 22 );//PTS (Cont)
|
|
Buffer[Offset+11] = 0x00 + ((PTS & 0x0003F8000) >> 14 ) + 1;//PTS (Cont)
|
|
Buffer[Offset+12] = 0x00 + ((PTS & 0x000007F80) >> 7 );//PTS (Cont)
|
|
Buffer[Offset+13] = 0x00 + ((PTS & 0x00000007F) << 1) + 1;//PTS (Cont)
|
|
Free = Free - 12;
|
|
}
|
|
|
|
void TS::Packet::FillFree( std::string & NewVal ) {
|
|
int Offset = 188 - Free;
|
|
std::copy( NewVal.begin(), NewVal.begin() + Free, Buffer + Offset );
|
|
NewVal.erase(0,Free);
|
|
Free = 0;
|
|
}
|
|
|
|
void TS::Packet::AddStuffing( int NumBytes ) {
|
|
if( NumBytes <= 0 ) { return; }
|
|
if( AdaptationField( ) == 3 ) {
|
|
int Offset = Buffer[4];
|
|
Buffer[4] = Offset + NumBytes - 1;
|
|
for( int i = 0; i < ( NumBytes -2 ); i ++ ) {
|
|
Buffer[6+Offset+i] = 0xFF;
|
|
}
|
|
Free -= NumBytes;
|
|
} else {
|
|
AdaptationField( 3 );
|
|
Buffer[4] = NumBytes - 1;
|
|
Buffer[5] = 0x00;
|
|
for( int i = 0; i < ( NumBytes -2 ); i ++ ) {
|
|
Buffer[6+i] = 0xFF;
|
|
}
|
|
Free -= NumBytes;
|
|
}
|
|
}
|
|
|
|
void TS::Packet::FFMpegHeader( ) {
|
|
static int MyCntr = 0;
|
|
std::copy( TS::SDT, TS::SDT + 188, Buffer );
|
|
ContinuityCounter( MyCntr );
|
|
Free = 0;
|
|
MyCntr = ( (MyCntr + 1) % 0x10);
|
|
}
|