jellyfin/BDInfo/TSStreamFile.cs

1552 lines
71 KiB
C#
Raw Normal View History

2013-02-21 01:33:05 +00:00
//============================================================================
// BDInfo - Blu-ray Video and Audio Analysis Tool
// Copyright © 2010 Cinema Squid
//
// 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 2.1 of the License, or (at your option) any later version.
//
// 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//=============================================================================
#undef DEBUG
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Text;
namespace BDInfo
{
public class TSStreamState
{
public ulong TransferCount = 0;
public string StreamTag = null;
public ulong TotalPackets = 0;
public ulong WindowPackets = 0;
public ulong TotalBytes = 0;
public ulong WindowBytes = 0;
public long PeakTransferLength = 0;
public long PeakTransferRate = 0;
public double TransferMarker = 0;
public double TransferInterval = 0;
public TSStreamBuffer StreamBuffer = new TSStreamBuffer();
public uint Parse = 0;
public bool TransferState = false;
public int TransferLength = 0;
public int PacketLength = 0;
public byte PacketLengthParse = 0;
public byte PacketParse = 0;
public byte PTSParse = 0;
public ulong PTS = 0;
public ulong PTSTemp = 0;
public ulong PTSLast = 0;
public ulong PTSPrev = 0;
public ulong PTSDiff = 0;
public ulong PTSCount = 0;
public ulong PTSTransfer = 0;
public byte DTSParse = 0;
public ulong DTSTemp = 0;
public ulong DTSPrev = 0;
public byte PESHeaderLength = 0;
public byte PESHeaderFlags = 0;
#if DEBUG
public byte PESHeaderIndex = 0;
public byte[] PESHeader = new byte[256 + 9];
#endif
}
public class TSPacketParser
{
public bool SyncState = false;
public byte TimeCodeParse = 4;
public byte PacketLength = 0;
public byte HeaderParse = 0;
public uint TimeCode;
public byte TransportErrorIndicator;
public byte PayloadUnitStartIndicator;
public byte TransportPriority;
public ushort PID;
public byte TransportScramblingControl;
public byte AdaptionFieldControl;
public bool AdaptionFieldState = false;
public byte AdaptionFieldParse = 0;
public byte AdaptionFieldLength = 0;
public ushort PCRPID = 0xFFFF;
public byte PCRParse = 0;
public ulong PreviousPCR = 0;
public ulong PCR = 0;
public ulong PCRCount = 0;
public ulong PTSFirst = ulong.MaxValue;
public ulong PTSLast = ulong.MinValue;
public ulong PTSDiff = 0;
public byte[] PAT = new byte[1024];
public bool PATSectionStart = false;
public byte PATPointerField = 0;
public uint PATOffset = 0;
public byte PATSectionLengthParse = 0;
public ushort PATSectionLength = 0;
public uint PATSectionParse = 0;
public bool PATTransferState = false;
public byte PATSectionNumber = 0;
public byte PATLastSectionNumber = 0;
public ushort TransportStreamId = 0xFFFF;
public List<TSDescriptor> PMTProgramDescriptors = new List<TSDescriptor>();
public ushort PMTPID = 0xFFFF;
public Dictionary<ushort, byte[]> PMT = new Dictionary<ushort, byte[]>();
public bool PMTSectionStart = false;
public ushort PMTProgramInfoLength = 0;
public byte PMTProgramDescriptor = 0;
public byte PMTProgramDescriptorLengthParse = 0;
public byte PMTProgramDescriptorLength = 0;
public ushort PMTStreamInfoLength = 0;
public uint PMTStreamDescriptorLengthParse = 0;
public uint PMTStreamDescriptorLength = 0;
public byte PMTPointerField = 0;
public uint PMTOffset = 0;
public uint PMTSectionLengthParse = 0;
public ushort PMTSectionLength = 0;
public uint PMTSectionParse = 0;
public bool PMTTransferState = false;
public byte PMTSectionNumber = 0;
public byte PMTLastSectionNumber = 0;
public byte PMTTemp = 0;
public TSStream Stream = null;
public TSStreamState StreamState = null;
public ulong TotalPackets = 0;
}
public class TSStreamDiagnostics
{
public ulong Bytes = 0;
public ulong Packets = 0;
public double Marker = 0;
public double Interval = 0;
public string Tag = null;
}
public class TSStreamFile
{
public FileInfo FileInfo = null;
public string Name = null;
public long Size = 0;
public double Length = 0;
public TSInterleavedFile InterleavedFile = null;
private Dictionary<ushort, TSStreamState> StreamStates =
new Dictionary<ushort, TSStreamState>();
public Dictionary<ushort, TSStream> Streams =
new Dictionary<ushort, TSStream>();
public Dictionary<ushort, List<TSStreamDiagnostics>> StreamDiagnostics =
new Dictionary<ushort, List<TSStreamDiagnostics>>();
private List<TSPlaylistFile> Playlists = null;
public TSStreamFile(FileInfo fileInfo)
{
FileInfo = fileInfo;
Name = fileInfo.Name.ToUpper();
}
public string DisplayName
{
get
{
if (BDInfoSettings.EnableSSIF &&
InterleavedFile != null)
{
return InterleavedFile.Name;
}
return Name;
}
}
private bool ScanStream(
TSStream stream,
TSStreamState streamState,
TSStreamBuffer buffer)
{
streamState.StreamTag = null;
long bitrate = 0;
if (stream.IsAudioStream &&
streamState.PTSTransfer > 0)
{
bitrate = (long)Math.Round(
(buffer.TransferLength * 8.0) /
((double)streamState.PTSTransfer / 90000));
if (bitrate > streamState.PeakTransferRate)
{
streamState.PeakTransferRate = bitrate;
}
}
if (buffer.TransferLength > streamState.PeakTransferLength)
{
streamState.PeakTransferLength = buffer.TransferLength;
}
buffer.BeginRead();
switch (stream.StreamType)
{
case TSStreamType.MPEG2_VIDEO:
TSCodecMPEG2.Scan(
(TSVideoStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.AVC_VIDEO:
TSCodecAVC.Scan(
(TSVideoStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.MVC_VIDEO:
TSCodecMVC.Scan(
(TSVideoStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.VC1_VIDEO:
TSCodecVC1.Scan(
(TSVideoStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.AC3_AUDIO:
TSCodecAC3.Scan(
(TSAudioStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.AC3_PLUS_AUDIO:
case TSStreamType.AC3_PLUS_SECONDARY_AUDIO:
TSCodecAC3.Scan(
(TSAudioStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.AC3_TRUE_HD_AUDIO:
TSCodecTrueHD.Scan(
(TSAudioStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.LPCM_AUDIO:
TSCodecLPCM.Scan(
(TSAudioStream)stream, buffer, ref streamState.StreamTag);
break;
case TSStreamType.DTS_AUDIO:
TSCodecDTS.Scan(
(TSAudioStream)stream, buffer, bitrate, ref streamState.StreamTag);
break;
case TSStreamType.DTS_HD_AUDIO:
case TSStreamType.DTS_HD_MASTER_AUDIO:
case TSStreamType.DTS_HD_SECONDARY_AUDIO:
TSCodecDTSHD.Scan(
(TSAudioStream)stream, buffer, bitrate, ref streamState.StreamTag);
break;
default:
stream.IsInitialized = true;
break;
}
buffer.EndRead();
streamState.StreamBuffer.Reset();
bool isAVC = false;
bool isMVC = false;
foreach (TSStream finishedStream in Streams.Values)
{
if (!finishedStream.IsInitialized)
{
return false;
}
if (finishedStream.StreamType == TSStreamType.AVC_VIDEO)
{
isAVC = true;
}
if (finishedStream.StreamType == TSStreamType.MVC_VIDEO)
{
isMVC = true;
}
}
if (isMVC && !isAVC)
{
return false;
}
return true;
}
private void UpdateStreamBitrates(
ushort PTSPID,
ulong PTS,
ulong PTSDiff)
{
if (Playlists == null) return;
foreach (ushort PID in StreamStates.Keys)
{
if (Streams.ContainsKey(PID) &&
Streams[PID].IsVideoStream &&
PID != PTSPID)
{
continue;
}
if (StreamStates[PID].WindowPackets == 0)
{
continue;
}
UpdateStreamBitrate(PID, PTSPID, PTS, PTSDiff);
}
foreach (TSPlaylistFile playlist in Playlists)
{
double packetSeconds = 0;
foreach (TSStreamClip clip in playlist.StreamClips)
{
if (clip.AngleIndex == 0)
{
packetSeconds += clip.PacketSeconds;
}
}
if (packetSeconds > 0)
{
foreach (TSStream playlistStream in playlist.SortedStreams)
{
if (playlistStream.IsVBR)
{
playlistStream.BitRate = (long)Math.Round(
((playlistStream.PayloadBytes * 8.0) / packetSeconds));
if (playlistStream.StreamType == TSStreamType.AC3_TRUE_HD_AUDIO &&
((TSAudioStream)playlistStream).CoreStream != null)
{
playlistStream.BitRate -=
((TSAudioStream)playlistStream).CoreStream.BitRate;
}
}
}
}
}
}
private void UpdateStreamBitrate(
ushort PID,
ushort PTSPID,
ulong PTS,
ulong PTSDiff)
{
if (Playlists == null) return;
TSStreamState streamState = StreamStates[PID];
double streamTime = (double)PTS / 90000;
double streamInterval = (double)PTSDiff / 90000;
double streamOffset = streamTime + streamInterval;
foreach (TSPlaylistFile playlist in Playlists)
{
foreach (TSStreamClip clip in playlist.StreamClips)
{
if (clip.Name != this.Name) continue;
if (streamTime == 0 ||
(streamTime >= clip.TimeIn &&
streamTime <= clip.TimeOut))
{
clip.PayloadBytes += streamState.WindowBytes;
clip.PacketCount += streamState.WindowPackets;
if (streamOffset > clip.TimeIn &&
streamOffset - clip.TimeIn > clip.PacketSeconds)
{
clip.PacketSeconds = streamOffset - clip.TimeIn;
}
Dictionary<ushort, TSStream> playlistStreams = playlist.Streams;
if (clip.AngleIndex > 0 &&
clip.AngleIndex < playlist.AngleStreams.Count + 1)
{
playlistStreams = playlist.AngleStreams[clip.AngleIndex - 1];
}
if (playlistStreams.ContainsKey(PID))
{
TSStream stream = playlistStreams[PID];
stream.PayloadBytes += streamState.WindowBytes;
stream.PacketCount += streamState.WindowPackets;
if (stream.IsVideoStream)
{
stream.PacketSeconds += streamInterval;
stream.ActiveBitRate = (long)Math.Round(
((stream.PayloadBytes * 8.0) /
stream.PacketSeconds));
}
if (stream.StreamType == TSStreamType.AC3_TRUE_HD_AUDIO &&
((TSAudioStream)stream).CoreStream != null)
{
stream.ActiveBitRate -=
((TSAudioStream)stream).CoreStream.BitRate;
}
}
}
}
}
if (Streams.ContainsKey(PID))
{
TSStream stream = Streams[PID];
stream.PayloadBytes += streamState.WindowBytes;
stream.PacketCount += streamState.WindowPackets;
if (stream.IsVideoStream)
{
TSStreamDiagnostics diag = new TSStreamDiagnostics();
diag.Marker = (double)PTS / 90000;
diag.Interval = (double)PTSDiff / 90000;
diag.Bytes = streamState.WindowBytes;
diag.Packets = streamState.WindowPackets;
diag.Tag = streamState.StreamTag;
StreamDiagnostics[PID].Add(diag);
stream.PacketSeconds += streamInterval;
}
}
streamState.WindowPackets = 0;
streamState.WindowBytes = 0;
}
public void Scan(List<TSPlaylistFile> playlists, bool isFullScan)
{
if (playlists == null || playlists.Count == 0)
{
return;
}
Playlists = playlists;
int dataSize = 16384;
FileStream fileStream = null;
try
{
string fileName;
if (BDInfoSettings.EnableSSIF &&
InterleavedFile != null)
{
fileName = InterleavedFile.FileInfo.FullName;
}
else
{
fileName = FileInfo.FullName;
}
fileStream = new FileStream(
fileName,
FileMode.Open,
FileAccess.Read,
FileShare.Read,
dataSize, false);
Size = 0;
Length = 0;
Streams.Clear();
StreamStates.Clear();
StreamDiagnostics.Clear();
TSPacketParser parser =
new TSPacketParser();
long fileLength = (uint)fileStream.Length;
byte[] buffer = new byte[dataSize];
int bufferLength = 0;
while ((bufferLength =
fileStream.Read(buffer, 0, buffer.Length)) > 0)
{
int offset = 0;
for (int i = 0; i < bufferLength; i++)
{
if (parser.SyncState == false)
{
if (parser.TimeCodeParse > 0)
{
parser.TimeCodeParse--;
switch (parser.TimeCodeParse)
{
case 3:
parser.TimeCode = 0;
parser.TimeCode |=
((uint)buffer[i] & 0x3F) << 24;
break;
case 2:
parser.TimeCode |=
((uint)buffer[i] & 0xFF) << 16;
break;
case 1:
parser.TimeCode |=
((uint)buffer[i] & 0xFF) << 8;
break;
case 0:
parser.TimeCode |=
((uint)buffer[i] & 0xFF);
break;
}
}
else if (buffer[i] == 0x47)
{
parser.SyncState = true;
parser.PacketLength = 187;
parser.TimeCodeParse = 4;
parser.HeaderParse = 3;
}
}
else if (parser.HeaderParse > 0)
{
parser.PacketLength--;
parser.HeaderParse--;
switch (parser.HeaderParse)
{
case 2:
{
parser.TransportErrorIndicator =
(byte)((buffer[i] >> 7) & 0x1);
parser.PayloadUnitStartIndicator =
(byte)((buffer[i] >> 6) & 0x1);
parser.TransportPriority =
(byte)((buffer[i] >> 5) & 0x1);
parser.PID =
(ushort)((buffer[i] & 0x1f) << 8);
}
break;
case 1:
{
parser.PID |= (ushort)buffer[i];
if (Streams.ContainsKey(parser.PID))
{
parser.Stream = Streams[parser.PID];
}
else
{
parser.Stream = null;
}
if (!StreamStates.ContainsKey(parser.PID))
{
StreamStates[parser.PID] = new TSStreamState();
}
parser.StreamState = StreamStates[parser.PID];
parser.StreamState.TotalPackets++;
parser.StreamState.WindowPackets++;
parser.TotalPackets++;
}
break;
case 0:
{
parser.TransportScramblingControl =
(byte)((buffer[i] >> 6) & 0x3);
parser.AdaptionFieldControl =
(byte)((buffer[i] >> 4) & 0x3);
if ((parser.AdaptionFieldControl & 0x2) == 0x2)
{
parser.AdaptionFieldState = true;
}
if (parser.PayloadUnitStartIndicator == 1)
{
if (parser.PID == 0)
{
parser.PATSectionStart = true;
}
else if (parser.PID == parser.PMTPID)
{
parser.PMTSectionStart = true;
}
else if (parser.StreamState != null &&
parser.StreamState.TransferState)
{
parser.StreamState.TransferState = false;
parser.StreamState.TransferCount++;
bool isFinished = ScanStream(
parser.Stream,
parser.StreamState,
parser.StreamState.StreamBuffer);
if (!isFullScan && isFinished)
{
return;
}
}
}
}
break;
}
}
else if (parser.AdaptionFieldState)
{
parser.PacketLength--;
parser.AdaptionFieldParse = buffer[i];
parser.AdaptionFieldLength = buffer[i];
parser.AdaptionFieldState = false;
}
else if (parser.AdaptionFieldParse > 0)
{
parser.PacketLength--;
parser.AdaptionFieldParse--;
if ((parser.AdaptionFieldLength - parser.AdaptionFieldParse) == 1)
{
if ((buffer[i] & 0x10) == 0x10)
{
parser.PCRParse = 6;
parser.PCR = 0;
}
}
else if (parser.PCRParse > 0)
{
parser.PCRParse--;
parser.PCR = (parser.PCR << 8) + (ulong)buffer[i];
if (parser.PCRParse == 0)
{
parser.PreviousPCR = parser.PCR;
parser.PCR = (parser.PCR & 0x1FF) +
((parser.PCR >> 15) * 300);
}
parser.PCRCount++;
}
if (parser.PacketLength == 0)
{
parser.SyncState = false;
}
}
else if (parser.PID == 0)
{
if (parser.PATTransferState)
{
if ((bufferLength - i) > parser.PATSectionLength)
{
offset = parser.PATSectionLength;
}
else
{
offset = (bufferLength - i);
}
if (parser.PacketLength <= offset)
{
offset = parser.PacketLength;
}
for (int k = 0; k < offset; k++)
{
parser.PAT[parser.PATOffset++] = buffer[i++];
parser.PATSectionLength--;
parser.PacketLength--;
} --i;
if (parser.PATSectionLength == 0)
{
parser.PATTransferState = false;
if (parser.PATSectionNumber == parser.PATLastSectionNumber)
{
for (int k = 0; k < (parser.PATOffset - 4); k += 4)
{
uint programNumber = (uint)
((parser.PAT[k] << 8) +
parser.PAT[k + 1]);
ushort programPID = (ushort)
(((parser.PAT[k + 2] & 0x1F) << 8) +
parser.PAT[k + 3]);
if (programNumber == 1)
{
parser.PMTPID = programPID;
}
}
}
}
}
else
{
--parser.PacketLength;
if (parser.PATSectionStart)
{
parser.PATPointerField = buffer[i];
if (parser.PATPointerField == 0)
{
parser.PATSectionLengthParse = 3;
}
parser.PATSectionStart = false;
}
else if (parser.PATPointerField > 0)
{
--parser.PATPointerField;
if (parser.PATPointerField == 0)
{
parser.PATSectionLengthParse = 3;
}
}
else if (parser.PATSectionLengthParse > 0)
{
--parser.PATSectionLengthParse;
switch (parser.PATSectionLengthParse)
{
case 2:
break;
case 1:
parser.PATSectionLength = (ushort)
((buffer[i] & 0xF) << 8);
break;
case 0:
parser.PATSectionLength |= buffer[i];
if (parser.PATSectionLength > 1021)
{
parser.PATSectionLength = 0;
}
else
{
parser.PATSectionParse = 5;
}
break;
}
}
else if (parser.PATSectionParse > 0)
{
--parser.PATSectionLength;
--parser.PATSectionParse;
switch (parser.PATSectionParse)
{
case 4:
parser.TransportStreamId = (ushort)
(buffer[i] << 8);
break;
case 3:
parser.TransportStreamId |= buffer[i];
break;
case 2:
break;
case 1:
parser.PATSectionNumber = buffer[i];
if (parser.PATSectionNumber == 0)
{
parser.PATOffset = 0;
}
break;
case 0:
parser.PATLastSectionNumber = buffer[i];
parser.PATTransferState = true;
break;
}
}
}
if (parser.PacketLength == 0)
{
parser.SyncState = false;
}
}
else if (parser.PID == parser.PMTPID)
{
if (parser.PMTTransferState)
{
if ((bufferLength - i) >= parser.PMTSectionLength)
{
offset = parser.PMTSectionLength;
}
else
{
offset = (bufferLength - i);
}
if (parser.PacketLength <= offset)
{
offset = parser.PacketLength;
}
if (!parser.PMT.ContainsKey(parser.PID))
{
parser.PMT[parser.PID] = new byte[1024];
}
byte[] PMT = parser.PMT[parser.PID];
for (int k = 0; k < offset; k++)
{
PMT[parser.PMTOffset++] = buffer[i++];
--parser.PMTSectionLength;
--parser.PacketLength;
} --i;
if (parser.PMTSectionLength == 0)
{
parser.PMTTransferState = false;
if (parser.PMTSectionNumber == parser.PMTLastSectionNumber)
{
//Console.WriteLine("PMT Start: " + parser.PMTTemp);
try
{
for (int k = 0; k < (parser.PMTOffset - 4); k += 5)
{
byte streamType = PMT[k];
ushort streamPID = (ushort)
(((PMT[k + 1] & 0x1F) << 8) +
PMT[k + 2]);
ushort streamInfoLength = (ushort)
(((PMT[k + 3] & 0xF) << 8) +
PMT[k + 4]);
/*
if (streamInfoLength == 2)
{
// TODO: Cleanup
//streamInfoLength = 0;
}
Console.WriteLine(string.Format(
"Type: {0} PID: {1} Length: {2}",
streamType, streamPID, streamInfoLength));
*/
if (!Streams.ContainsKey(streamPID))
{
List<TSDescriptor> streamDescriptors =
new List<TSDescriptor>();
/*
* TODO: Getting bad streamInfoLength
if (streamInfoLength > 0)
{
for (int d = 0; d < streamInfoLength; d++)
{
byte name = PMT[k + d + 5];
byte length = PMT[k + d + 6];
TSDescriptor descriptor =
new TSDescriptor(name, length);
for (int v = 0; v < length; v++)
{
descriptor.Value[v] =
PMT[k + d + v + 7];
}
streamDescriptors.Add(descriptor);
d += (length + 1);
}
}
*/
CreateStream(streamPID, streamType, streamDescriptors);
}
k += streamInfoLength;
}
}
catch (Exception ex)
{
// TODO
Console.WriteLine(ex.Message);
}
}
}
}
else
{
--parser.PacketLength;
if (parser.PMTSectionStart)
{
parser.PMTPointerField = buffer[i];
if (parser.PMTPointerField == 0)
{
parser.PMTSectionLengthParse = 3;
}
parser.PMTSectionStart = false;
}
else if (parser.PMTPointerField > 0)
{
--parser.PMTPointerField;
if (parser.PMTPointerField == 0)
{
parser.PMTSectionLengthParse = 3;
}
}
else if (parser.PMTSectionLengthParse > 0)
{
--parser.PMTSectionLengthParse;
switch (parser.PMTSectionLengthParse)
{
case 2:
if (buffer[i] != 0x2)
{
parser.PMTSectionLengthParse = 0;
}
break;
case 1:
parser.PMTSectionLength = (ushort)
((buffer[i] & 0xF) << 8);
break;
case 0:
parser.PMTSectionLength |= buffer[i];
if (parser.PMTSectionLength > 1021)
{
parser.PMTSectionLength = 0;
}
else
{
parser.PMTSectionParse = 9;
}
break;
}
}
else if (parser.PMTSectionParse > 0)
{
--parser.PMTSectionLength;
--parser.PMTSectionParse;
switch (parser.PMTSectionParse)
{
case 8:
case 7:
break;
case 6:
parser.PMTTemp = buffer[i];
break;
case 5:
parser.PMTSectionNumber = buffer[i];
if (parser.PMTSectionNumber == 0)
{
parser.PMTOffset = 0;
}
break;
case 4:
parser.PMTLastSectionNumber = buffer[i];
break;
case 3:
parser.PCRPID = (ushort)
((buffer[i] & 0x1F) << 8);
break;
case 2:
parser.PCRPID |= buffer[i];
break;
case 1:
parser.PMTProgramInfoLength = (ushort)
((buffer[i] & 0xF) << 8);
break;
case 0:
parser.PMTProgramInfoLength |= buffer[i];
if (parser.PMTProgramInfoLength == 0)
{
parser.PMTTransferState = true;
}
else
{
parser.PMTProgramDescriptorLengthParse = 2;
}
break;
}
}
else if (parser.PMTProgramInfoLength > 0)
{
--parser.PMTSectionLength;
--parser.PMTProgramInfoLength;
if (parser.PMTProgramDescriptorLengthParse > 0)
{
--parser.PMTProgramDescriptorLengthParse;
switch (parser.PMTProgramDescriptorLengthParse)
{
case 1:
parser.PMTProgramDescriptor = buffer[i];
break;
case 0:
parser.PMTProgramDescriptorLength = buffer[i];
parser.PMTProgramDescriptors.Add(
new TSDescriptor(
parser.PMTProgramDescriptor,
parser.PMTProgramDescriptorLength));
break;
}
}
else if (parser.PMTProgramDescriptorLength > 0)
{
--parser.PMTProgramDescriptorLength;
TSDescriptor descriptor = parser.PMTProgramDescriptors[
parser.PMTProgramDescriptors.Count - 1];
int valueIndex =
descriptor.Value.Length -
parser.PMTProgramDescriptorLength - 1;
descriptor.Value[valueIndex] = buffer[i];
if (parser.PMTProgramDescriptorLength == 0 &&
parser.PMTProgramInfoLength > 0)
{
parser.PMTProgramDescriptorLengthParse = 2;
}
}
if (parser.PMTProgramInfoLength == 0)
{
parser.PMTTransferState = true;
}
}
}
if (parser.PacketLength == 0)
{
parser.SyncState = false;
}
}
else if (parser.Stream != null &&
parser.StreamState != null &&
parser.TransportScramblingControl == 0)
{
TSStream stream = parser.Stream;
TSStreamState streamState = parser.StreamState;
streamState.Parse =
(streamState.Parse << 8) + buffer[i];
if (streamState.TransferState)
{
if ((bufferLength - i) >= streamState.PacketLength &&
streamState.PacketLength > 0)
{
offset = streamState.PacketLength;
}
else
{
offset = (bufferLength - i);
}
if (parser.PacketLength <= offset)
{
offset = parser.PacketLength;
}
streamState.TransferLength = offset;
if (!stream.IsInitialized ||
stream.IsVideoStream)
{
streamState.StreamBuffer.Add(
buffer, i, offset);
}
else
{
streamState.StreamBuffer.TransferLength += offset;
}
i += (int)(streamState.TransferLength - 1);
streamState.PacketLength -= streamState.TransferLength;
parser.PacketLength -= (byte)streamState.TransferLength;
streamState.TotalBytes += (ulong)streamState.TransferLength;
streamState.WindowBytes += (ulong)streamState.TransferLength;
if (streamState.PacketLength == 0)
{
streamState.TransferState = false;
streamState.TransferCount++;
bool isFinished = ScanStream(
stream,
streamState,
streamState.StreamBuffer);
if (!isFullScan && isFinished)
{
return;
}
}
}
else
{
--parser.PacketLength;
bool headerFound = false;
if (stream.IsVideoStream &&
streamState.Parse == 0x000001FD)
{
headerFound = true;
}
if (stream.IsVideoStream &&
streamState.Parse >= 0x000001E0 &&
streamState.Parse <= 0x000001EF)
{
headerFound = true;
}
if (stream.IsAudioStream &&
streamState.Parse == 0x000001BD)
{
headerFound = true;
}
if (stream.IsAudioStream &&
(streamState.Parse == 0x000001FA ||
streamState.Parse == 0x000001FD))
{
headerFound = true;
}
if (!stream.IsVideoStream &&
!stream.IsAudioStream &&
(streamState.Parse == 0x000001FA ||
streamState.Parse == 0x000001FD ||
streamState.Parse == 0x000001BD ||
(streamState.Parse >= 0x000001E0 &&
streamState.Parse <= 0x000001EF)))
{
headerFound = true;
}
if (headerFound)
{
streamState.PacketLengthParse = 2;
#if DEBUG
streamState.PESHeaderIndex = 0;
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)((streamState.Parse >> 24) & 0xFF);
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)((streamState.Parse >> 16) & 0xFF);
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)((streamState.Parse >> 8) & 0xFF);
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
}
else if (streamState.PacketLengthParse > 0)
{
--streamState.PacketLengthParse;
switch (streamState.PacketLengthParse)
{
case 1:
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 0:
streamState.PacketLength =
(int)(streamState.Parse & 0xFFFF);
streamState.PacketParse = 3;
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
}
}
else if (streamState.PacketParse > 0)
{
--streamState.PacketLength;
--streamState.PacketParse;
switch (streamState.PacketParse)
{
case 2:
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 1:
streamState.PESHeaderFlags =
(byte)(streamState.Parse & 0xFF);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 0:
streamState.PESHeaderLength =
(byte)(streamState.Parse & 0xFF);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
if ((streamState.PESHeaderFlags & 0xC0) == 0x80)
{
streamState.PTSParse = 5;
}
else if ((streamState.PESHeaderFlags & 0xC0) == 0xC0)
{
streamState.DTSParse = 10;
}
if (streamState.PESHeaderLength == 0)
{
streamState.TransferState = true;
}
break;
}
}
else if (streamState.PTSParse > 0)
{
--streamState.PacketLength;
--streamState.PESHeaderLength;
--streamState.PTSParse;
switch (streamState.PTSParse)
{
case 4:
streamState.PTSTemp =
((streamState.Parse & 0xE) << 29);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
break;
case 3:
streamState.PTSTemp |=
((streamState.Parse & 0xFF) << 22);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 2:
streamState.PTSTemp |=
((streamState.Parse & 0xFE) << 14);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 1:
streamState.PTSTemp |=
((streamState.Parse & 0xFF) << 7);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 0:
streamState.PTSTemp |=
((streamState.Parse & 0xFE) >> 1);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
streamState.PTS = streamState.PTSTemp;
if (streamState.PTS > streamState.PTSLast)
{
if (streamState.PTSLast > 0)
{
streamState.PTSTransfer = (streamState.PTS - streamState.PTSLast);
}
streamState.PTSLast = streamState.PTS;
}
streamState.PTSDiff = streamState.PTS - streamState.DTSPrev;
if (streamState.PTSCount > 0 &&
stream.IsVideoStream)
{
UpdateStreamBitrates(stream.PID, streamState.PTS, streamState.PTSDiff);
if (streamState.DTSTemp < parser.PTSFirst)
{
parser.PTSFirst = streamState.DTSTemp;
}
if (streamState.DTSTemp > parser.PTSLast)
{
parser.PTSLast = streamState.DTSTemp;
}
Length = (double)(parser.PTSLast - parser.PTSFirst) / 90000;
}
streamState.DTSPrev = streamState.PTS;
streamState.PTSCount++;
if (streamState.PESHeaderLength == 0)
{
streamState.TransferState = true;
}
break;
}
}
else if (streamState.DTSParse > 0)
{
--streamState.PacketLength;
--streamState.PESHeaderLength;
--streamState.DTSParse;
switch (streamState.DTSParse)
{
case 9:
streamState.PTSTemp =
((streamState.Parse & 0xE) << 29);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 8:
streamState.PTSTemp |=
((streamState.Parse & 0xFF) << 22);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 7:
streamState.PTSTemp |=
((streamState.Parse & 0xFE) << 14);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
break;
case 6:
streamState.PTSTemp |=
((streamState.Parse & 0xFF) << 7);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 5:
streamState.PTSTemp |=
((streamState.Parse & 0xFE) >> 1);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
streamState.PTS = streamState.PTSTemp;
if (streamState.PTS > streamState.PTSLast)
{
streamState.PTSLast = streamState.PTS;
}
break;
case 4:
streamState.DTSTemp =
((streamState.Parse & 0xE) << 29);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
break;
case 3:
streamState.DTSTemp |=
((streamState.Parse & 0xFF) << 22);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
break;
case 2:
streamState.DTSTemp |=
((streamState.Parse & 0xFE) << 14);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
break;
case 1:
streamState.DTSTemp |=
((streamState.Parse & 0xFF) << 7);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
break;
case 0:
streamState.DTSTemp |=
((streamState.Parse & 0xFE) >> 1);
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xff);
#endif
streamState.PTSDiff = streamState.DTSTemp - streamState.DTSPrev;
if (streamState.PTSCount > 0 &&
stream.IsVideoStream)
{
UpdateStreamBitrates(stream.PID, streamState.DTSTemp, streamState.PTSDiff);
if (streamState.DTSTemp < parser.PTSFirst)
{
parser.PTSFirst = streamState.DTSTemp;
}
if (streamState.DTSTemp > parser.PTSLast)
{
parser.PTSLast = streamState.DTSTemp;
}
Length = (double)(parser.PTSLast - parser.PTSFirst) / 90000;
}
streamState.DTSPrev = streamState.DTSTemp;
streamState.PTSCount++;
if (streamState.PESHeaderLength == 0)
{
streamState.TransferState = true;
}
break;
}
}
else if (streamState.PESHeaderLength > 0)
{
--streamState.PacketLength;
--streamState.PESHeaderLength;
#if DEBUG
streamState.PESHeader[streamState.PESHeaderIndex++] =
(byte)(streamState.Parse & 0xFF);
#endif
if (streamState.PESHeaderLength == 0)
{
streamState.TransferState = true;
}
}
}
if (parser.PacketLength == 0)
{
parser.SyncState = false;
}
}
else
{
parser.PacketLength--;
if ((bufferLength - i) >= parser.PacketLength)
{
i = i + parser.PacketLength;
parser.PacketLength = 0;
}
else
{
parser.PacketLength -= (byte)((bufferLength - i) + 1);
i = bufferLength;
}
if (parser.PacketLength == 0)
{
parser.SyncState = false;
}
}
}
Size += bufferLength;
}
ulong PTSLast = 0;
ulong PTSDiff = 0;
foreach (TSStream stream in Streams.Values)
{
if (!stream.IsVideoStream) continue;
if (StreamStates.ContainsKey(stream.PID) &&
StreamStates[stream.PID].PTSLast > PTSLast)
{
PTSLast = StreamStates[stream.PID].PTSLast;
PTSDiff = PTSLast - StreamStates[stream.PID].DTSPrev;
}
UpdateStreamBitrates(stream.PID, PTSLast, PTSDiff);
}
}
finally
{
if (fileStream != null)
{
fileStream.Close();
}
}
}
private TSStream CreateStream(
ushort streamPID,
byte streamType,
List<TSDescriptor> streamDescriptors)
{
TSStream stream = null;
switch ((TSStreamType)streamType)
{
case TSStreamType.MVC_VIDEO:
case TSStreamType.AVC_VIDEO:
case TSStreamType.MPEG1_VIDEO:
case TSStreamType.MPEG2_VIDEO:
case TSStreamType.VC1_VIDEO:
{
stream = new TSVideoStream();
}
break;
case TSStreamType.AC3_AUDIO:
case TSStreamType.AC3_PLUS_AUDIO:
case TSStreamType.AC3_PLUS_SECONDARY_AUDIO:
case TSStreamType.AC3_TRUE_HD_AUDIO:
case TSStreamType.DTS_AUDIO:
case TSStreamType.DTS_HD_AUDIO:
case TSStreamType.DTS_HD_MASTER_AUDIO:
case TSStreamType.DTS_HD_SECONDARY_AUDIO:
case TSStreamType.LPCM_AUDIO:
case TSStreamType.MPEG1_AUDIO:
case TSStreamType.MPEG2_AUDIO:
{
stream = new TSAudioStream();
}
break;
case TSStreamType.INTERACTIVE_GRAPHICS:
case TSStreamType.PRESENTATION_GRAPHICS:
{
stream = new TSGraphicsStream();
}
break;
case TSStreamType.SUBTITLE:
{
stream = new TSTextStream();
}
break;
default:
break;
}
if (stream != null &&
!Streams.ContainsKey(streamPID))
{
stream.PID = streamPID;
stream.StreamType = (TSStreamType)streamType;
stream.Descriptors = streamDescriptors;
Streams[stream.PID] = stream;
}
if (!StreamDiagnostics.ContainsKey(streamPID))
{
StreamDiagnostics[streamPID] =
new List<TSStreamDiagnostics>();
}
return stream;
}
}
}