////////////////////////////////////////////////////////////////////////////
///
/// Handle a block of data that is not frame aligned.
///
/// There may be the end of a frame, whole frames, the start of a frame or even just
/// the middle of the frame.
///
/// If we have incomplete blocks we build up a complete one in the saved data,
/// in order to process we need to acquire a frame plus the next header (for sync check)
/// we can end up with the save buffer having a frame + part of a header, and a secondary
/// save with just part of a header.
///
/// \return Collator status code, CollatorNoError indicates success.
///
CollatorStatus_t Collator_PesAudio_c::HandleElementaryStream(unsigned int Length, unsigned char *Data)
{
CollatorStatus_t Status;
//
if (DiscardPesPacket)
{
COLLATOR_DEBUG("Discarding %d bytes of elementary stream\n", Length);
return CodecNoError;
}
//
// Copy our arguments to class members so the utility functions can
// act upon it.
//
RemainingElementaryOrigin = Data;
RemainingElementaryData = Data;
RemainingElementaryLength = Length;
//
// Continually handle units of input until all input is exhausted (or an error occurs).
//
// Be aware that our helper functions may, during their execution, cause state changes
// that result in a different branch being taken next time round the loop.
//
Status = CollatorNoError;
while (Status == CollatorNoError && RemainingElementaryLength != 0)
{
COLLATOR_DEBUG("ES loop has %d bytes remaining\n", RemainingElementaryLength);
//report_dump_hex( severity_note, RemainingElementaryData, min(RemainingElementaryLength, 188), 32, 0);
switch (CollatorState)
{
case SeekingSyncWord:
//
// Try to lock to incoming frame headers
//
Status = SearchForSyncWord();
break;
case GotSynchronized:
case SeekingFrameEnd:
//
// Read in the remains of the frame header
//
Status = ReadPartialFrameHeader();
break;
case ReadSubFrame:
case SkipSubFrame:
//
// Squirrel away the frame
//
Status = ReadFrame();
break;
case GotCompleteFrame:
//
// Pass the accumulated subframes to the frame parser
//
InternalFrameFlush();
CollatorState = ReadSubFrame;
break;
default:
// should not occur...
COLLATOR_DEBUG("General failure; wrong collator state");
Status = CollatorError;
break;
}
}
if (Status != CollatorNoError)
{
// if anything when wrong then we need to resynchronize
COLLATOR_DEBUG("General failure; seeking new synchronization sequence\n");
DiscardAccumulatedData();
CollatorState = SeekingSyncWord;
}
return Status;
}
//}}}
//{{{ InternalFrameFlush
// /////////////////////////////////////////////////////////////////////////
//
// The Frame Flush functions
//
CollatorStatus_t Collator_PesFrame_c::InternalFrameFlush(bool FlushedByStreamTerminate)
{
CodedFrameParameters->FollowedByStreamTerminate = FlushedByStreamTerminate;
return InternalFrameFlush();
}
//}}}
//{{{ Input
////////////////////////////////////////////////////////////////////////////
///
/// Extract Frame length from Pes Private Data area if present.
///
/// \return Collator status code, CollatorNoError indicates success.
///
CollatorStatus_t Collator_PesVideoMjpeg_c::Input(PlayerInputDescriptor_t *Input,
unsigned int DataLength,
void *Data,
bool NonBlocking,
unsigned int *DataLengthRemaining)
{
CollatorStatus_t Status = CollatorNoError;
unsigned char *DataBlock = (unsigned char *)Data;
unsigned char *PesHeader;
unsigned int PesLength;
unsigned int PayloadLength;
unsigned int Offset;
unsigned int CodeOffset;
unsigned int Code;
AssertComponentState("Collator_PacketPes_c::Input", ComponentRunning);
COLLATOR_ASSERT(!NonBlocking);
InputEntry(Input, DataLength, Data, NonBlocking);
Offset = 0;
RemainingData = (unsigned char *)Data;
RemainingLength = DataLength;
TerminationFlagIsSet = false;
Offset = 0;
while (Offset < DataLength)
{
// Read the length of the payload
PesHeader = DataBlock + Offset;
PesLength = (PesHeader[4] << 8) + PesHeader[5];
if (PesLength != 0)
PayloadLength = PesLength - PesHeader[8] - 3;
else
PayloadLength = 0;
COLLATOR_DEBUG("DataLength %d, PesLength %d; PayloadLength %d, Offset %d\n", DataLength, PesLength, PayloadLength, Offset);
Offset += PesLength + 6; // PES packet is PesLength + 6 bytes long
Bits.SetPointer(PesHeader + 9); // Set bits pointer ready to process optional fields
if ((PesHeader[7] & 0x80) == 0x80) // PTS present?
//{{{ read PTS
{
Bits.FlushUnseen(4);
PlaybackTime = (unsigned long long)(Bits.Get(3)) << 30;
Bits.FlushUnseen(1);
PlaybackTime |= Bits.Get(15) << 15;
Bits.FlushUnseen(1);
PlaybackTime |= Bits.Get(15);
Bits.FlushUnseen(1);
PlaybackTimeValid = true;
COLLATOR_DEBUG("PTS %llu.\n", PlaybackTime);
}
//}}}
if ((PesHeader[7] & 0xC0) == 0xC0) // DTS present?
//{{{ read DTS
{
Bits.FlushUnseen(4);
DecodeTime = (unsigned long long)(Bits.Get(3)) << 30;
Bits.FlushUnseen(1);
DecodeTime |= Bits.Get(15) << 15;
Bits.FlushUnseen(1);
DecodeTime |= Bits.Get(15);
Bits.FlushUnseen(1);
DecodeTimeValid = true;
}
//}}}
else if ((PesHeader[7] & 0xC0) == 0x40)
{
COLLATOR_ERROR("Malformed pes header contains DTS without PTS.\n");
DiscardAccumulatedData(); // throw away previous frame as incomplete
InputExit();
return CollatorError;
}
RemainingData = PesHeader + (PesLength + 6 - PayloadLength);
RemainingLength = PayloadLength;
while (RemainingLength > 0)
{
Status = FindNextStartCode(&CodeOffset);
if (Status != CollatorNoError) // Error indicates no start code found
{
Status = AccumulateData(RemainingLength, RemainingData);
if (Status != CollatorNoError)
DiscardAccumulatedData();
RemainingLength = 0;
break;
}
// Got one accumulate up to and including it
Status = AccumulateData(CodeOffset + 2, RemainingData);
if (Status != CollatorNoError)
{
DiscardAccumulatedData();
break;
}
Code = RemainingData[CodeOffset + 1];
RemainingLength -= CodeOffset + 2;
RemainingData += CodeOffset + 2;
// Is it a block terminate code
if ((Code == Configuration.BlockTerminateCode) && (AccumulatedDataSize > 2))
{
AccumulatedDataSize -= 2;
Status = InternalFrameFlush((Configuration.StreamTerminateFlushesFrame && (Code == Configuration.StreamTerminationCode)));
if (Status != CollatorNoError)
break;
BufferBase[0] = 0xff;
BufferBase[1] = Code;
AccumulatedDataSize = 2;
}
// Accumulate the start code
Status = AccumulateStartCode(PackStartCode(AccumulatedDataSize - 2, Code));
if (Status != CollatorNoError)
{
DiscardAccumulatedData();
InputExit();
return Status;
}
}
}
InputExit();
return CollatorNoError;
}
//{{{ Input
////////////////////////////////////////////////////////////////////////////
///
/// Extract Frame length from Pes Private Data area if present.
///
/// \return Collator status code, CollatorNoError indicates success.
///
CollatorStatus_t Collator_PesFrame_c::Input(PlayerInputDescriptor_t *Input,
unsigned int DataLength,
void *Data,
bool NonBlocking,
unsigned int *DataLengthRemaining)
{
CollatorStatus_t Status = CollatorNoError;
bool PrivateDataPresent;
unsigned char *DataBlock = (unsigned char *)Data;
unsigned char *PesHeader;
unsigned char *PayloadStart;
unsigned int PayloadLength;
unsigned int PesLength;
unsigned int Offset;
COLLATOR_ASSERT(!NonBlocking);
AssertComponentState("Collator_PesFrame_c::Input", ComponentRunning);
InputEntry(Input, DataLength, Data, NonBlocking);
Offset = 0;
while (Offset < DataLength)
{
// Read the length of the payload
PrivateDataPresent = false;
PesHeader = DataBlock + Offset;
PesLength = (PesHeader[4] << 8) + PesHeader[5];
if (PesLength != 0)
PayloadLength = PesLength - PesHeader[8] - 3;
else
PayloadLength = 0;
COLLATOR_DEBUG("DataLength %d, PesLength %d; PayloadLength %d, Offset %d\n", DataLength, PesLength, PayloadLength, Offset);
Offset += PesLength + 6; // PES packet is PesLength + 6 bytes long
Bits.SetPointer(PesHeader + 9); // Set bits pointer ready to process optional fields
if ((PesHeader[7] & 0x80) == 0x80) // PTS present?
//{{{ read PTS
{
Bits.FlushUnseen(4);
PlaybackTime = (unsigned long long)(Bits.Get(3)) << 30;
Bits.FlushUnseen(1);
PlaybackTime |= Bits.Get(15) << 15;
Bits.FlushUnseen(1);
PlaybackTime |= Bits.Get(15);
Bits.FlushUnseen(1);
PlaybackTimeValid = true;
COLLATOR_DEBUG("PTS %llu.\n", PlaybackTime);
}
//}}}
if ((PesHeader[7] & 0xC0) == 0xC0) // DTS present?
//{{{ read DTS
{
Bits.FlushUnseen(4);
DecodeTime = (unsigned long long)(Bits.Get(3)) << 30;
Bits.FlushUnseen(1);
DecodeTime |= Bits.Get(15) << 15;
Bits.FlushUnseen(1);
DecodeTime |= Bits.Get(15);
Bits.FlushUnseen(1);
DecodeTimeValid = true;
}
//}}}
else if ((PesHeader[7] & 0xC0) == 0x40)
{
COLLATOR_ERROR("Malformed pes header contains DTS without PTS.\n");
DiscardAccumulatedData(); // throw away previous frame as incomplete
InputExit();
return CollatorError;
}
//}}}
//{{{ walk down optional bits
if ((PesHeader[7] & 0x20) == 0x20) // ESCR present
Bits.FlushUnseen(48); // Size of ESCR
if ((PesHeader[7] & 0x10) == 0x10) // ES Rate present
Bits.FlushUnseen(24); // Size of ES Rate
if ((PesHeader[7] & 0x08) == 0x08) // Trick mode control present
Bits.FlushUnseen(8); // Size of Trick mode control
if ((PesHeader[7] & 0x04) == 0x04) // Additional copy info present
Bits.FlushUnseen(8); // Size of additional copy info
if ((PesHeader[7] & 0x02) == 0x02) // PES CRC present
Bits.FlushUnseen(16); // Size of previous packet CRC
if ((PesHeader[7] & 0x01) == 0x01) // PES Extension flag
{
PrivateDataPresent = Bits.Get(1);
Bits.FlushUnseen(7); // Size of Pes extension data
}
//}}}
if (PrivateDataPresent)
{
if (RemainingDataLength != 0)
{
COLLATOR_ERROR("%s: Warning new frame indicated but %d bytes missing\n", __FUNCTION__, RemainingDataLength);
DiscardAccumulatedData(); // throw away previous frame as incomplete
}
FrameSize = Bits.Get(8) + (Bits.Get(8) << 8) + (Bits.Get(8) << 16);
RemainingDataLength = FrameSize;
COLLATOR_DEBUG("%s: PlaybackTimeValid %d, PlaybackTime %llx, FrameSize %d\n", __FUNCTION__, PlaybackTimeValid, PlaybackTime, FrameSize);
}
if ((int)PayloadLength > RemainingDataLength) // Too much data - have some packets been lost?
{
if (RemainingDataLength != 0)
{
COLLATOR_ERROR("%s: Warning packet contains more bytes than needed %d bytes missing?\n", __FUNCTION__, RemainingDataLength);
DiscardAccumulatedData(); // throw away previous frame as incomplete
//RemainingDataLength = 0;
//InputExit();
//return CollatorError;
}
RemainingDataLength = PayloadLength; // assume new packet is stand alone frame
}
AccumulateStartCode(PackStartCode(AccumulatedDataSize, 0x42));
PayloadStart = PesHeader + (PesLength + 6 - PayloadLength);
Status = AccumulateData(PayloadLength, (unsigned char *)PayloadStart);
if (Status != CollatorNoError)
{
InputExit();
return Status;
}
RemainingDataLength -= PayloadLength;
if (RemainingDataLength <= 0)
{
Status = InternalFrameFlush(); // flush out collected frame
if (Status != CollatorNoError)
{
InputExit();
return Status;
}
}
COLLATOR_DEBUG("%s PrivateDataPresent %d, RemainingDataLength %d, PayloadLength %d\n",
__FUNCTION__, PrivateDataPresent, RemainingDataLength, PayloadLength);
}
InputExit();
return CollatorNoError;
}
CollatorStatus_t Collator_PesVideoRaw_c::Input(PlayerInputDescriptor_t *Input,
unsigned int DataLength,
void *Data,
bool NonBlocking,
unsigned int *DataLengthRemaining)
{
PlayerStatus_t Status = PlayerNoError;
COLLATOR_ASSERT(!NonBlocking);
AssertComponentState("Collator_PesRaw_c::Input", ComponentRunning);
InputEntry(Input, DataLength, Data, NonBlocking);
// Extract the descriptor timing information
ActOnInputDescriptor(Input);
COLLATOR_DEBUG("DataLength : %d\n", DataLength);
if (DataRemaining == 0)
//{{{ Treat the packet as a header, read metadata and build stream info
{
BufferStructure_t BufferStructure;
class Buffer_c* Buffer;
unsigned int Format;
unsigned char* DataBlock;
unsigned char* PesHeader = (unsigned char*)Data;
unsigned int PesLength;
unsigned int PesHeaderLength;
unsigned int PayloadLength;
COLLATOR_DEBUG("Header : %d\n", DataLength);
//{{{ Read pes header
// Read the length of the payload
PesLength = (PesHeader[4] << 8) + PesHeader[5];
PesHeaderLength = PesHeader[8];
if (PesLength != 0)
PayloadLength = PesLength - PesHeaderLength - 3;
else
PayloadLength = 0;
COLLATOR_DEBUG("DataLength %d, PesLength %d, PesHeaderLength %d, PayloadLength %d\n", DataLength, PesLength, PesHeaderLength, PayloadLength);
Bits.SetPointer(PesHeader + 9); // Set bits pointer ready to process optional fields
if ((PesHeader[7] & 0x80) == 0x80) // PTS present?
//{{{ read PTS
{
Bits.FlushUnseen(4);
PlaybackTime = (unsigned long long)(Bits.Get(3)) << 30;
Bits.FlushUnseen(1);
PlaybackTime |= Bits.Get(15) << 15;
Bits.FlushUnseen(1);
PlaybackTime |= Bits.Get(15);
Bits.FlushUnseen(1);
PlaybackTimeValid = true;
COLLATOR_DEBUG("PTS %llu.\n", PlaybackTime);
}
//}}}
if ((PesHeader[7] & 0xC0) == 0xC0) // DTS present?
//{{{ read DTS
{
Bits.FlushUnseen(4);
DecodeTime = (unsigned long long)(Bits.Get(3)) << 30;
Bits.FlushUnseen(1);
DecodeTime |= Bits.Get(15) << 15;
Bits.FlushUnseen(1);
DecodeTime |= Bits.Get(15);
Bits.FlushUnseen(1);
DecodeTimeValid = true;
}
//}}}
else if ((PesHeader[7] & 0xC0) == 0x40)
{
COLLATOR_ERROR("Malformed pes header contains DTS without PTS.\n");
InputExit();
return CollatorError;
}
//}}}
DataBlock = PesHeader + 9 + PesHeaderLength;
//{{{ Trace
#if 0
report(severity_info, "(%d)\n%06d: ", PayloadLength, 0);
for (int i = 0; i < PayloadLength; i++)
{
report(severity_info, "%02x ", DataBlock[i]);
if (((i + 1) & 0x1f) == 0)
report(severity_info, "\n%06d: ", i + 1);
}
report(severity_info, "\n");
#endif
//}}}
// Check that this is what we think it is
if (strcmp((char*)DataBlock, "STMicroelectronics") != 0)
{
//COLLATOR_TRACE("Id : %s\n", Id);
InputExit();
return FrameParserNoError;
}
DataBlock += strlen((char*)DataBlock) + 1;
// Fill in stream info for frame parser
memcpy(&StreamInfo.width, DataBlock, sizeof(unsigned int));
DataBlock += sizeof(unsigned int);
memcpy(&StreamInfo.height, DataBlock, sizeof(unsigned int));
DataBlock += sizeof(unsigned int);
memcpy(&StreamInfo.FrameRateNumerator, DataBlock, sizeof(unsigned long long));
DataBlock += sizeof(unsigned long long);
memcpy(&StreamInfo.FrameRateDenominator, DataBlock, sizeof(unsigned long long));
DataBlock += sizeof(unsigned long long);
memcpy(&Format, DataBlock, sizeof(unsigned int));
DataBlock += sizeof(unsigned int);
//memcpy (&StreamInfo.interlaced, DataBlock, sizeof (unsigned int));
//DataBlock += sizeof (unsigned int);
memcpy(&DataRemaining, DataBlock, sizeof(unsigned int));
DataBlock += sizeof(unsigned int);
StreamInfo.interlaced = 0;
StreamInfo.pixel_aspect_ratio.Numerator = 1;
StreamInfo.pixel_aspect_ratio.Denominator = 1;
//StreamInfo.FrameRateNumerator = 25;
//StreamInfo.FrameRateDenominator = 1;
StreamInfo.InputWindow.X = 0;
StreamInfo.InputWindow.Y = 0;
StreamInfo.InputWindow.Width = StreamInfo.width;
StreamInfo.InputWindow.Height = StreamInfo.height;
StreamInfo.OutputWindow = StreamInfo.InputWindow;
memset(&StreamInfo.OutputWindow, 0, sizeof(StreamInfo.OutputWindow));
if (DecodeBuffer == NULL)
{
// Fill out the buffer structure request
memset(&BufferStructure, 0x00, sizeof(BufferStructure_t));
BufferStructure.DimensionCount = 2;
BufferStructure.Dimension[0] = StreamInfo.width;
BufferStructure.Dimension[1] = StreamInfo.height;
//{{{ determine buffer format
switch (Format)
{
case CodeToInteger('N', 'V', '2', '2'):
BufferStructure.Format = FormatVideo422_Raster;
break;
case CodeToInteger('R', 'G', 'B', 'P'):
BufferStructure.Format = FormatVideo565_RGB;
break;
case CodeToInteger('R', 'G', 'B', '3'):
BufferStructure.Format = FormatVideo888_RGB;
break;
case CodeToInteger('R', 'G', 'B', '4'):
BufferStructure.Format = FormatVideo8888_ARGB;
break;
case CodeToInteger('Y', 'U', 'Y', 'V'):
BufferStructure.Format = FormatVideo422_YUYV;
break;
default:
COLLATOR_ERROR("Unsupported decode buffer format request %.4s\n", (char*) &Format);
InputExit();
return CollatorError;
}
//}}}
// Ask the manifestor for the buffer
Status = Manifestor->GetDecodeBuffer(&BufferStructure, &Buffer);
if (Status != ManifestorNoError)
{
COLLATOR_ERROR("Failed to get decode buffer\n");
InputExit();
return CollatorError;
}
StreamInfo.buffer_class = (unsigned int*)Buffer;
// Get physical address of data buffer (not actually used later in pipeline)
Status = Buffer->ObtainDataReference(NULL, NULL, (void**)&StreamInfo.buffer, PhysicalAddress);
if (Status != BufferNoError)
{
COLLATOR_ERROR("Failed to get decode buffer data reference\n");
InputExit();
return CollatorError;
}
// Remember cached address so we can write to it
Status = Buffer->ObtainDataReference(NULL, NULL, (void**)&DecodeBuffer, CachedAddress);
if (Status != BufferNoError)
{
COLLATOR_ERROR("Failed to get decode buffer data reference\n");
InputExit();
return CollatorError;
}
CodedFrameBuffer->AttachBuffer(Buffer); // Attach to decode buffer (so it will be freed at the same time)
Buffer->DecrementReferenceCount(); // and release ownership of the buffer to the decode buffer
}
COLLATOR_DEBUG("%s: ImageSize %6d\n", __FUNCTION__, DataRemaining);
COLLATOR_DEBUG("%s: ImageWidth %6d\n", __FUNCTION__, StreamInfo.width);
COLLATOR_DEBUG("%s: ImageHeight %6d\n", __FUNCTION__, StreamInfo.height);
COLLATOR_DEBUG("%s: Format %.4s\n", __FUNCTION__, (char *) &Format);
Status = CollatorNoError;
}
//}}}
else
//{{{ Assume packet is part of data - copy to decode buffer
{
if (DecodeBuffer == NULL)
{
COLLATOR_ERROR("No decode buffer available\n");
InputExit();
return CodecError;
}
// Transfer the packet to the next coded data frame and pass on
memcpy(DecodeBuffer + DataCopied, Data, DataLength);
DataRemaining -= DataLength;
DataCopied += DataLength;
if (DataRemaining <= 0)
{
Status = AccumulateData(sizeof(StreamInfo_t), (unsigned char*)&StreamInfo);
DataRemaining = 0;
DataCopied = 0;
DecodeBuffer = NULL;
if (Status != CollatorNoError)
{
COLLATOR_ERROR("Failed to accumulate StreamInfo\n");
InputExit();
return Status;
}
Status = InternalFrameFlush();
if (Status != CollatorNoError)
COLLATOR_ERROR("Failed to flush frame\n");
}
}
//}}}
InputExit();
return Status;
}
