private void MuxPesPacketToTs(PesPacket pp, bool priority)
{
byte[] data = new byte[Constants.TS_SIZE];
int j = 0;
int i = 0;
byte[] pes = pp.GetData();
// take care of the first packet
data[0] = Constants.SYNC_BYTE;
data[1] = 0x40; // payload start
data[2] = 0;
if (pes.Length < Constants.TS_PAYLOAD_SIZE)
{
data[3] = 0x30; // adaptation and payload
int stufLength = Constants.TS_PAYLOAD_SIZE - pes.Length - 1;
data[4] = (byte)stufLength;
i = 5;
if (stufLength > 0)
{
data[i] = 0;
i++;
stufLength--;
}
for (; i < (6 + stufLength); i++)
data[i] = 0xff;
for (; i < Constants.TS_SIZE; i++)
{
data[i] = pes[j];
j++;
}
}
else
{
data[3] = 0x10; // no adaptation, payload only
for (i = 4; i < data.Length; i++)
{
data[i] = pes[j];
j++;
}
}
TsPacket ts = new TsPacket();
ts.SetData(data, 0);
ushort pid = pidMappings[pp.PID];
ts.PID = pid;
ts.Priority = priority;
ts.ContinuityCounter = Continuities[pid];
ts.IncrementContinuityCounter();
Continuities[pid] = ts.ContinuityCounter;
buffer.Add(ts);
while (j < ((pes.Length / Constants.TS_PAYLOAD_SIZE) * Constants.TS_PAYLOAD_SIZE))
{
// take care of the other packets
data[0] = Constants.SYNC_BYTE;
data[1] = 0x00; // no payload start
data[2] = 0;
data[3] = 0x10; // no adaptation, payload only
for (i = 4; i < data.Length; i++)
{
data[i] = pes[j];
j++;
}
ts = new TsPacket();
ts.SetData(data, 0);
ts.PID = pid;
ts.Priority = priority;
ts.ContinuityCounter = Continuities[pid];
ts.IncrementContinuityCounter();
Continuities[pid] = ts.ContinuityCounter;
buffer.Add(ts);
}
// take care of the last packet
if (j < pes.Length)
{
data[0] = Constants.SYNC_BYTE;
data[1] = 0x00; // no payload start
data[2] = 0;
data[3] = 0x30; // adaptation and payload
int stufLength = Constants.TS_PAYLOAD_SIZE - (pes.Length - j) - 1;
data[4] = (byte)stufLength;
i = 5;
if (stufLength > 0)
{
data[i] = 0;
i++;
stufLength--;
for (; i < (6 + stufLength); i++)
data[i] = 0xff;
}
for (; i < Constants.TS_SIZE; i++)
{
data[i] = pes[j];
j++;
}
ts = new TsPacket();
ts.SetData(data, 0);
ts.PID = pid;
ts.Priority = priority;
ts.ContinuityCounter = Continuities[pid];
ts.IncrementContinuityCounter();
Continuities[pid] = ts.ContinuityCounter;
buffer.Add(ts);
}
}
private PesPacket CheckAndFixDiscontinuities(PesPacket pp)
{
// checks for PTS/DTS discontinuities
PesHeader ph = pp.GetHeader();
byte[] data = pp.GetData();
int len = ph.TotalHeaderLength;
int i = len;
UInt32 marker = 0xffffffff;
if (Constants.DEFAULT_VIDEO_PID == pidMappings[pp.PID])
{
// checks for end of stream headers
switch (VideoType)
{
case (byte)ElementaryStreamTypes.VIDEO_STREAM_VC1:
for (; i < data.Length; i++)
{
marker <<= 8;
marker |= data[i];
if (marker == Constants.VC1_END_OF_STREAM)
break;
}
if (i < data.Length)
{
// we have an end of stream marker
i -= 3;
PesPacket pnew = new PesPacket(data, 0, i, pp.PID);
i += 4;
pnew.AddData(data, i, data.Length - i);
pp = pnew;
ph = pp.GetHeader();
data = pp.GetData();
}
break;
case (byte)ElementaryStreamTypes.VIDEO_STREAM_MPEG2:
for (; i < data.Length; i++)
{
marker <<= 8;
marker |= data[i];
if (marker == Constants.MPEG2_SEQ_END)
break;
}
if (i < data.Length)
{
// we have an end of stream marker
i -= 3;
PesPacket pnew = new PesPacket(data, 0, i, pp.PID);
i += 4;
pnew.AddData(data, i, data.Length - i);
pp = pnew;
ph = pp.GetHeader();
data = pp.GetData();
}
break;
case (byte)ElementaryStreamTypes.VIDEO_STREAM_H264:
for (; i < data.Length; i++)
{
marker <<= 8;
marker |= data[i];
if ((marker & 0xffffff9f) == Constants.H264_END_OF_STREAM)
break;
}
if (i < data.Length)
{
// we have an end of stream marker
i -= 3;
PesPacket pnew = new PesPacket(data, 0, i, pp.PID);
i += 4;
pnew.AddData(data, i, data.Length - i);
pp = pnew;
ph = pp.GetHeader();
data = pp.GetData();
}
break;
}
if (ph.HasPts)
{
lastPts = currentPts;
currentPts = ph.Pts;
ptsCount = ptsDelta;
ptsDelta = currentPts - lastPts;
if (lastPts != -1)
{
if (ptsDelta < (0 - (Constants.PTS_CLOCK_RATE << 2)) ||
ptsDelta > (Constants.PTS_CLOCK_RATE << 2))
{
ptsOffset += (lastPts + ptsCount - currentPts);
}
}
}
// build EP Map info
marker = 0xffffffff;
switch (VideoType)
{
case (byte)ElementaryStreamTypes.VIDEO_STREAM_VC1:
for (i = ph.TotalHeaderLength; i < data.Length; i++)
{
marker <<= 8;
marker |= data[i];
if (marker == Constants.VC1_SEQ_SC && ph.HasPts)
{
EpElement ep = new EpElement(ph.Pts, this.CurrentPacketNumber);
epData.Add(ep);
break;
}
}
break;
case (byte)ElementaryStreamTypes.VIDEO_STREAM_MPEG2:
for (i = ph.TotalHeaderLength; i < data.Length; i++)
{
marker <<= 8;
marker |= data[i];
if (marker == Constants.MPEG2_SEQ_CODE && ph.HasPts)
{
EpElement ep = new EpElement(ph.Pts, this.CurrentPacketNumber);
epData.Add(ep);
break;
}
}
break;
case (byte)ElementaryStreamTypes.VIDEO_STREAM_H264:
for (i = ph.TotalHeaderLength; i < data.Length; i++)
{
marker <<= 8;
marker |= data[i];
if ((marker & 0xffffff9f) == Constants.H264_PREFIX && ph.HasPts)
{
EpElement ep = new EpElement(ph.Pts, this.CurrentPacketNumber);
epData.Add(ep);
break;
}
}
break;
}
}
if (ph.HasPts && ptsOffset != 0)
{
Int64 time = ph.Pts + ptsOffset;
if (time < 0)
time += Constants.MAX_PTS_CLOCK;
else if (time > Constants.MAX_PTS_CLOCK)
time -= Constants.MAX_PTS_CLOCK;
ph.Pts = time;
for (i = 9; i < 14; i++)
pp[i] = ph[i]; // copy PTS
if (ph.HasDts)
{
time = ph.Dts + ptsOffset;
if (time < 0)
time += Constants.MAX_PTS_CLOCK;
else if (time > Constants.MAX_PTS_CLOCK)
time -= Constants.MAX_PTS_CLOCK;
ph.Dts = time;
for (i = 14; i < 19; i++)
pp[i] = ph[i]; // copy DTS
}
}
lastPacket = pp;
return pp;
}
/*
public UInt32 CurrentPacketNumber
{
get { return packetCount + (UInt32)buffer.Count; }
}
public EpElement[] EpData
{
get
{
return epData.ToArray();
}
}
public StreamInfo[] Psi
{
get { return pmtStreams.ToArray(); }
}
*/
public override void MuxPacket(PesPacket pp)
{
if (fileType == TsFileType.TS || fileType == TsFileType.M2TS || fileType == TsFileType.BLU_RAY)
MuxTsPacket(pp);
else if (fileType == TsFileType.SUP_ELEMENTARY && pidMappings.ContainsKey(pp.PID))
{
byte[] data = pp.GetData();
PesHeader ph = pp.GetHeader();
if (ph.HasPts)
{
Int64 Pts = ph.Pts;
Int64 Dts = 0;
if (ph.HasDts)
Dts = ph.Dts;
supHeader[2] = (byte)((Pts >> 24) & 0xff);
supHeader[3] = (byte)((Pts >> 16) & 0xff);
supHeader[4] = (byte)((Pts >> 8) & 0xff);
supHeader[5] = (byte)(Pts & 0xff);
supHeader[6] = (byte)((Dts >> 24) & 0xff);
supHeader[7] = (byte)((Dts >> 16) & 0xff);
supHeader[8] = (byte)((Dts >> 8) & 0xff);
supHeader[9] = (byte)(Dts & 0xff);
tsiow.Write(supHeader, 0, supHeader.Length);
}
tsiow.Write(data, ph.HeaderLength + 9, data.Length - (ph.HeaderLength + 9));
}
else if (fileType == TsFileType.PES_ELEMENTARY && pidMappings.ContainsKey(pp.PID))
{
byte[] data = pp.GetData();
tsiow.Write(data, 0, data.Length);
}
else if (fileType == TsFileType.ELEMENTARY && pidMappings.ContainsKey(pp.PID))
{
byte[] data = pp.GetData();
PesHeader ph = pp.GetHeader();
tsiow.Write(data, ph.HeaderLength + 9, data.Length - (ph.HeaderLength + 9));
}
}
public override void MuxPacket(PesPacket pp)
{
// TODO: Fix, just QnD ...
bool bKeep = false;
foreach (StreamInfo si in _StreamsToKeep)
{
if (si.ElementaryPID == pp.PID)
bKeep = true;
}
if (bKeep)
{
byte[] data = pp.GetData();
_FileStream.Write(data, 0, data.Length);
}
}
// If PesPacket contains valid stream data, store it
// in the respective output stream.
public override void MuxPacket(PesPacket pp)
{
foreach (DeMuxStream S in _Streams)
{
if (pp.PID == S.PID)
{
byte[] data = pp.GetData();
S.FileStream.Write(data, 0, data.Length);
}
}
}
public override PesPacket[] GetNextPesPackets()
{
// there was a big optimization opportunity here:
// this method added to partial packets one byte at a time
// the calls to PesPacket.AddData used about 40% of total CPU
byte[] inData = new byte[1];
byte[] inData2 = new byte[128];
int bytesRead = 0;
int bytesToRead = 0;
int bytesToGo = 0;
UInt32 marker = 0xffffffff;
PesPacket pp = null;
PesPacket partial = null;
int i = 0;
while (tsior.Read(inData, 0, inData.Length) == inData.Length)
{
marker = marker << 8;
marker &= 0xffffff00;
marker += inData[0];
if ((marker & 0xffffff00) == 0x00000100)
{
if ((marker & 0xff) == Constants.PACK_ID)
{
// pack start code
UInt64 header = 0;
for (i = 4; i < 10; i++)
{
if (tsior.Read(inData, 0, inData.Length) != inData.Length)
goto done;
header <<= 8;
header |= inData[0];
}
for (i = 10; i < 14; i++) // skip mux rate etc.
{
if (tsior.Read(inData, 0, inData.Length) != inData.Length)
goto done;
}
Int64 pcr = (Int64)((header & 0x380000000000) >> 13);
pcr |= (Int64)((header & 0x3fff8000000) >> 12);
pcr |= (Int64)((header & 0x3fff800) >> 11);
pcr *= 300;
pcr |= (Int64)((header & 0x3fe) >> 1);
Int64 delta = pcr - lastPcr;
if (delta > Constants.MPEG2TS_CLOCK_RATE / 9)
{
if (pcrDelegate != null)
pcrDelegate(pcr);
lastPcr = pcr;
}
}
else if (((marker & 0xff) == Constants.SYS_ID)
|| ((marker & 0xff) == Constants.MAP_ID)
|| ((marker & 0xff) == Constants.PAD_ID)
|| ((marker & 0xff) == Constants.DIR_ID))
{
ushort Length = 0;
for (i = 4; i < 6; i++)
{
if (tsior.Read(inData, 0, inData.Length) != inData.Length)
goto done;
Length <<= 8;
Length |= inData[0];
}
Length += 6;
for (i = 6; i < Length; i++)
{
if (tsior.Read(inData, 0, inData.Length) != inData.Length)
goto done;
}
}
else
{
byte end = inData[0];
inData[0] = 0;
partial = new PesPacket(inData, 0, 1, 0);
partial.AddData(inData, 0, 1);
inData[0] = 1;
partial.AddData(inData, 0, 1);
inData[0] = end;
partial.AddData(inData, 0, 1);
ushort Length = 0;
for (i = 4; i < 6; i++)
{
if (tsior.Read(inData, 0, inData.Length) != inData.Length)
goto done;
partial.AddData(inData, 0, 1);
Length <<= 8;
Length |= inData[0];
}
Length += 6;
// we don't need this byte-by-byte loop here, as we have just gotten the length:
/*
for (i = 6; i < Length; i++)
{
if (tsior.Read(inData, 0, inData.Length) != inData.Length)
goto done;
partial.AddData(inData, 0, 1);
}
*/
bytesToGo = Length - 6;
while (bytesToGo > 0)
{
bytesToRead = (bytesToGo > inData2.Length) ? inData2.Length : bytesToGo;
if ((bytesRead = tsior.Read(inData2, 0, bytesToRead)) != bytesToRead)
goto done;
partial.AddData(inData2, 0, bytesRead);
bytesToGo -= bytesRead;
}
pp = partial;
PesHeader ph = partial.GetHeader();
if (partial.BaseId == Constants.PES_PRIVATE1)
{
if ((partial.ExtendedId & 0xf0) == Constants.PES_PRIVATE_AC3_PLUS ||
(partial.ExtendedId & 0xf8) == Constants.PES_PRIVATE_AC3)
{
if (soundFrames.ContainsKey(partial.ExtendedId) == false)
soundFrames.Add(partial.ExtendedId, new List<byte>());
byte[] tempd = partial.GetPayload();
for (int j = 4; j < tempd.Length; j++)
soundFrames[partial.ExtendedId].Add(tempd[j]);
tempd = soundFrames[partial.ExtendedId].ToArray();
int len = 0;
ushort mk2 = 0xffff;
if (tempd.Length > 2 && tempd[0] == (byte)(Constants.AC3_SYNC >> 8) &&
tempd[1] == (byte)(Constants.AC3_SYNC & 0xff))
{
pp = null;
while (tempd.Length > 2 && tempd[0] == (byte)(Constants.AC3_SYNC >> 8)
&& tempd[1] == (byte)(Constants.AC3_SYNC & 0xff))
{
AC3Info ac3 = new AC3Info(tempd, 0);
if (ac3.Valid == false || ac3.FrameLength == 0 || tempd.Length < ac3.FrameLength)
break;
if (pp == null)
pp = new PesPacket(partial.GetData(), 0, ph.TotalHeaderLength, partial.ExtendedId);
pp.AddData(tempd, 0, ac3.FrameLength);
soundFrames[partial.ExtendedId].RemoveRange(0, ac3.FrameLength);
tempd = soundFrames[partial.ExtendedId].ToArray();
}
if (pp != null)
{
pp.Complete = true;
goto done;
}
}
for (int j = 2; j < tempd.Length; j++)
{
mk2 <<= 8;
mk2 |= tempd[j];
if (mk2 == Constants.AC3_SYNC)
{
len = j - 1;
mk2 = 0xffff;
break;
}
}
if (len == 0)
len = tempd.Length;
soundFrames[partial.ExtendedId].RemoveRange(0, len);
pp = new PesPacket(partial.GetData(), 0, ph.TotalHeaderLength, partial.ExtendedId);
pp.AddData(tempd, 0, len);
pp.Complete = true;
}
else if ((partial.ExtendedId & 0xf8) == Constants.PES_PRIVATE_AC3_TRUE_HD)
{
if (soundFrames.ContainsKey(partial.ExtendedId) == false)
soundFrames.Add(partial.ExtendedId, new List<byte>());
byte[] tempd = partial.GetPayload();
for (int j = 5; j < tempd.Length; j++)
soundFrames[partial.ExtendedId].Add(tempd[j]);
tempd = soundFrames[partial.ExtendedId].ToArray();
int len = tempd.Length;
/*
UInt32 mk2 = 0xffffffff;
for (int j = 5; j < tempd.Length; j++)
{
mk2 <<= 8;
mk2 |= tempd[j];
if (mk2 == Constants.MLP_SYNC)
{
len = j - 3;
mk2 = 0xffffffff;
}
}
if (len == 0)
{
len = tempd.Length;
}
*/
soundFrames[partial.ExtendedId].RemoveRange(0, len);
pp = new PesPacket(partial.GetData(), 0, ph.TotalHeaderLength, partial.ExtendedId);
pp.AddData(tempd, 0, len);
pp.Complete = true;
}
else if ((partial.ExtendedId & 0xf8) == Constants.PES_PRIVATE_LPCM)
{
if (soundFrames.ContainsKey(partial.ExtendedId) == false)
soundFrames.Add(partial.ExtendedId, new List<byte>());
byte[] tempd = partial.GetPayload();
for (int j = 7; j < tempd.Length; j++)
soundFrames[partial.ExtendedId].Add(tempd[j]);
tempd = soundFrames[partial.ExtendedId].ToArray();
int len = tempd.Length;
/*
UInt32 mk2 = 0xffffffff;
for (int j = 5; j < tempd.Length; j++)
{
mk2 <<= 8;
mk2 |= tempd[j];
if (mk2 == Constants.MLP_SYNC)
{
len = j - 3;
mk2 = 0xffffffff;
}
}
if (len == 0)
{
len = tempd.Length;
}
*/
soundFrames[partial.ExtendedId].RemoveRange(0, len);
pp = new PesPacket(partial.GetData(), 0, ph.TotalHeaderLength, partial.ExtendedId);
pp.AddData(tempd, 0, len);
pp.Complete = true;
}
else if ((partial.ExtendedId & 0xf8) == Constants.PES_PRIVATE_DTS_HD)
{
if (soundFrames.ContainsKey(partial.ExtendedId) == false)
soundFrames.Add(partial.ExtendedId, new List<byte>());
byte[] tempd = partial.GetPayload();
for (int j = 4; j < tempd.Length; j++)
soundFrames[partial.ExtendedId].Add(tempd[j]);
tempd = soundFrames[partial.ExtendedId].ToArray();
int len = 0;
UInt32 mk2 = 0xffffffff;
for (int j = 4; j < tempd.Length; j++)
{
mk2 <<= 8;
mk2 |= tempd[j];
if (mk2 == Constants.DTS_SYNC)
{
len = j - 3;
mk2 = 0xffffffff;
}
}
if (len == 0)
{
DtsInfo dts = new DtsInfo(tempd, 0);
if (dts.Valid && (int)dts.FrameSize < tempd.Length &&
(tempd.Length - (int)dts.FrameSize < 4))
len = (int)dts.FrameSize;
else
len = tempd.Length;
}
soundFrames[partial.ExtendedId].RemoveRange(0, len);
pp = new PesPacket(partial.GetData(), 0, ph.TotalHeaderLength, partial.ExtendedId);
pp.AddData(tempd, 0, len);
pp.Complete = true;
}
else
pp.PID = pp.ExtendedId;
}
else if (pp.BaseId == Constants.PES_PADDING || pp.BaseId == Constants.PES_PRIVATE2)
{
marker = 0xffffffff;
continue;
}
else if (((pp.BaseId & 0xe0) == Constants.PES_AUDIO_MPEG)
|| ((pp.BaseId & 0xf0) == Constants.PES_VIDEO))
pp.PID = pp.BaseId;
else if (pp.BaseId == Constants.PES_VIDEO_VC1 && ph != null)
pp.PID = ph.Extention2;
if (ph != null && ph.HasPts)
{
if (ptsDelegate != null)
ptsDelegate(ph.Pts, pp.PID);
}
goto done;
}
marker = 0xffffffff;
}
}
done:
if (null != pp)
{
PesPacket[] ppa = new PesPacket[1];
ppa[0] = pp;
return ppa;
}
else
return null;
}