You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
507 lines
31 KiB
507 lines
31 KiB
4 years ago
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.IO;
|
||
|
|
using System.Text;
|
||
|
|
|
||
|
|
using VGMToolbox.util;
|
||
|
|
|
||
|
|
namespace VGMToolbox.format
|
||
|
|
{
|
||
|
|
public abstract class MpegStream
|
||
|
|
{
|
||
|
|
protected static readonly byte[] PacketStartBytes = new byte[] { 0x00, 0x00, 0x01, 0xBA };
|
||
|
|
protected static readonly byte[] PacketEndBytes = new byte[] { 0x00, 0x00, 0x01, 0xB9 };
|
||
|
|
|
||
|
|
public MpegStream(string path)
|
||
|
|
{
|
||
|
|
this.FilePath = path;
|
||
|
|
this.UsesSameIdForMultipleAudioTracks = false;
|
||
|
|
this.SubTitleExtractionSupported = false;
|
||
|
|
this.BlockSizeIsLittleEndian = false;
|
||
|
|
|
||
|
|
//********************
|
||
|
|
// Add Slice Packets
|
||
|
|
//********************
|
||
|
|
byte[] sliceBytes;
|
||
|
|
uint sliceBytesValue;
|
||
|
|
BlockSizeStruct blockSize = new BlockSizeStruct(PacketSizeType.Static, 0xE);
|
||
|
|
|
||
|
|
for (byte i = 0; i <= 0xAF; i++)
|
||
|
|
{
|
||
|
|
sliceBytes = new byte[] { 0x00, 0x00, 0x01, i };
|
||
|
|
sliceBytesValue = BitConverter.ToUInt32(sliceBytes, 0);
|
||
|
|
this.BlockIdDictionary.Add(sliceBytesValue, blockSize);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public enum PacketSizeType
|
||
|
|
{
|
||
|
|
Static,
|
||
|
|
SizeBytes,
|
||
|
|
Eof
|
||
|
|
}
|
||
|
|
|
||
|
|
public struct MpegDemuxOptions
|
||
|
|
{
|
||
|
|
public bool AddHeader { set; get; }
|
||
|
|
}
|
||
|
|
|
||
|
|
public struct BlockSizeStruct
|
||
|
|
{
|
||
|
|
public PacketSizeType SizeType;
|
||
|
|
public int Size;
|
||
|
|
|
||
|
|
public BlockSizeStruct(PacketSizeType sizeTypeValue, int sizeValue)
|
||
|
|
{
|
||
|
|
this.SizeType = sizeTypeValue;
|
||
|
|
this.Size = sizeValue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public struct DemuxOptionsStruct
|
||
|
|
{
|
||
|
|
public bool ExtractVideo { set; get; }
|
||
|
|
public bool ExtractAudio { set; get; }
|
||
|
|
|
||
|
|
public bool AddHeader { set; get; }
|
||
|
|
public bool SplitAudioStreams { set; get; }
|
||
|
|
public bool AddPlaybackHacks { set; get; }
|
||
|
|
}
|
||
|
|
|
||
|
|
#region Dictionary Initialization
|
||
|
|
|
||
|
|
protected Dictionary<uint, BlockSizeStruct> BlockIdDictionary =
|
||
|
|
new Dictionary<uint, BlockSizeStruct>
|
||
|
|
{
|
||
|
|
//********************
|
||
|
|
// System Packets
|
||
|
|
//********************
|
||
|
|
{BitConverter.ToUInt32(MpegStream.PacketEndBytes, 0), new BlockSizeStruct(PacketSizeType.Eof, -1)}, // Program End
|
||
|
|
{BitConverter.ToUInt32(MpegStream.PacketStartBytes, 0), new BlockSizeStruct(PacketSizeType.Static, 0xE)}, // Pack Header
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xBB }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // System Header, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xBD }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Private Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xBE }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Padding Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xBF }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Private Stream, two bytes following equal length (Big Endian)
|
||
|
|
|
||
|
|
//****************************
|
||
|
|
// Audio Streams
|
||
|
|
//****************************
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC0 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC1 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC2 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC3 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC4 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC5 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC6 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC7 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC8 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xC9 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xCA }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xCB }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xCC }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xCD }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xCE }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xCF }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD0 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD1 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD2 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD3 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD4 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD5 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD6 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD7 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD8 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xD9 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xDA }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xDB }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xDC }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xDD }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xDE }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xDF }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Audio Stream, two bytes following equal length (Big Endian)
|
||
|
|
|
||
|
|
//****************************
|
||
|
|
// Video Streams
|
||
|
|
//****************************
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE0 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE1 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE2 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE3 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE4 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE5 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE6 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE7 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE8 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xE9 }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xEA }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xEB }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xEC }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xED }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xEE }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
{BitConverter.ToUInt32(new byte[] { 0x00, 0x00, 0x01, 0xEF }, 0), new BlockSizeStruct(PacketSizeType.SizeBytes, 2)}, // Video Stream, two bytes following equal length (Big Endian)
|
||
|
|
};
|
||
|
|
#endregion
|
||
|
|
|
||
|
|
public string FilePath { get; set; }
|
||
|
|
public string FileExtensionAudio { get; set; }
|
||
|
|
public string FileExtensionVideo { get; set; }
|
||
|
|
|
||
|
|
public bool HasAudio { get; set; }
|
||
|
|
public string FinalAudioExtension { get; set; }
|
||
|
|
|
||
|
|
protected Dictionary<byte, string> StreamIdFileType = new Dictionary<byte, string>();
|
||
|
|
|
||
|
|
public bool UsesSameIdForMultipleAudioTracks { set; get; } // for PMF/PAM/DVD, who use 000001BD for all audio tracks
|
||
|
|
public bool SubTitleExtractionSupported { set; get; } // assume not supported.
|
||
|
|
|
||
|
|
public bool BlockSizeIsLittleEndian { set; get; }
|
||
|
|
|
||
|
|
protected virtual byte[] GetPacketStartBytes() { return MpegStream.PacketStartBytes; }
|
||
|
|
|
||
|
|
protected virtual byte[] GetPacketEndBytes() { return MpegStream.PacketEndBytes; }
|
||
|
|
|
||
|
|
protected abstract int GetAudioPacketHeaderSize(Stream readStream, long currentOffset);
|
||
|
|
|
||
|
|
protected virtual int GetAudioPacketSubHeaderSize(Stream readStream, long currentOffset, byte streamId) { return 0; }
|
||
|
|
|
||
|
|
protected abstract int GetVideoPacketHeaderSize(Stream readStream, long currentOffset);
|
||
|
|
|
||
|
|
protected virtual int GetAudioPacketFooterSize(Stream readStream, long currentOffset) { return 0; }
|
||
|
|
|
||
|
|
protected virtual int GetVideoPacketFooterSize(Stream readStream, long currentOffset) { return 0; }
|
||
|
|
|
||
|
|
protected virtual bool IsThisAnAudioBlock(byte[] blockToCheck)
|
||
|
|
{
|
||
|
|
return ((blockToCheck[3] >= 0xC0) &&
|
||
|
|
(blockToCheck[3] <= 0xDF));
|
||
|
|
}
|
||
|
|
|
||
|
|
protected virtual bool IsThisAVideoBlock(byte[] blockToCheck)
|
||
|
|
{
|
||
|
|
return ((blockToCheck[3] >= 0xE0) && (blockToCheck[3] <= 0xEF));
|
||
|
|
}
|
||
|
|
|
||
|
|
protected virtual bool IsThisASubPictureBlock(byte[] blockToCheck)
|
||
|
|
{
|
||
|
|
return ((blockToCheck[3] >= 0xE0) && (blockToCheck[3] <= 0xEF));
|
||
|
|
}
|
||
|
|
|
||
|
|
protected virtual string GetAudioFileExtension(Stream readStream, long currentOffset)
|
||
|
|
{
|
||
|
|
return this.FileExtensionAudio;
|
||
|
|
}
|
||
|
|
|
||
|
|
protected virtual string GetVideoFileExtension(Stream readStream, long currentOffset)
|
||
|
|
{
|
||
|
|
return this.FileExtensionVideo;
|
||
|
|
}
|
||
|
|
|
||
|
|
protected virtual byte GetStreamId(Stream readStream, long currentOffset) { return 0; }
|
||
|
|
|
||
|
|
protected virtual long GetStartOffset(Stream readStream, long currentOffset) { return 0; }
|
||
|
|
|
||
|
|
protected virtual void DoFinalTasks(FileStream sourceFileStream, Dictionary<uint, FileStream> outputFiles, bool addHeader)
|
||
|
|
{
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
public virtual void DemultiplexStreams(DemuxOptionsStruct demuxOptions)
|
||
|
|
{
|
||
|
|
using (FileStream fs = File.OpenRead(this.FilePath))
|
||
|
|
{
|
||
|
|
long fileSize = fs.Length;
|
||
|
|
long currentOffset = 0;
|
||
|
|
|
||
|
|
byte[] currentBlockId;
|
||
|
|
uint currentBlockIdVal;
|
||
|
|
byte[] currentBlockIdNaming;
|
||
|
|
|
||
|
|
BlockSizeStruct blockStruct = new BlockSizeStruct();
|
||
|
|
byte[] blockSizeArray;
|
||
|
|
uint blockSize;
|
||
|
|
|
||
|
|
int audioBlockSkipSize;
|
||
|
|
int videoBlockSkipSize;
|
||
|
|
|
||
|
|
int audioBlockFooterSize;
|
||
|
|
int videoBlockFooterSize;
|
||
|
|
|
||
|
|
int cutSize;
|
||
|
|
|
||
|
|
bool eofFlagFound = false;
|
||
|
|
|
||
|
|
Dictionary<uint, FileStream> streamOutputWriters = new Dictionary<uint, FileStream>();
|
||
|
|
string outputFileName;
|
||
|
|
|
||
|
|
byte streamId = 0; // for types that have multiple streams in the same block ID
|
||
|
|
uint currentStreamKey; // hash key for each file
|
||
|
|
bool isAudioBlock;
|
||
|
|
string audioFileExtension;
|
||
|
|
|
||
|
|
// look for first packet
|
||
|
|
currentOffset = this.GetStartOffset(fs, currentOffset);
|
||
|
|
currentOffset = ParseFile.GetNextOffset(fs, currentOffset, this.GetPacketStartBytes());
|
||
|
|
|
||
|
|
if (currentOffset != -1)
|
||
|
|
{
|
||
|
|
while (currentOffset < fileSize)
|
||
|
|
{
|
||
|
|
#if DEBUG
|
||
|
|
//if (currentOffset == 0x414080e)
|
||
|
|
//{
|
||
|
|
// int gggg = 1;
|
||
|
|
//}
|
||
|
|
|
||
|
|
//// hack for bad data (ni no kuni s09.pam)
|
||
|
|
//if ((currentOffset & 1) == 1)
|
||
|
|
//{
|
||
|
|
// currentOffset = MathUtil.RoundUpToByteAlignment(currentOffset, 0x800);
|
||
|
|
//}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
try
|
||
|
|
{
|
||
|
|
// get the current block
|
||
|
|
currentBlockId = ParseFile.ParseSimpleOffset(fs, currentOffset, 4);
|
||
|
|
|
||
|
|
// get value to use as key to hash table
|
||
|
|
currentBlockIdVal = BitConverter.ToUInt32(currentBlockId, 0);
|
||
|
|
|
||
|
|
if (BlockIdDictionary.ContainsKey(currentBlockIdVal))
|
||
|
|
{
|
||
|
|
// get info about this block type
|
||
|
|
blockStruct = BlockIdDictionary[currentBlockIdVal];
|
||
|
|
|
||
|
|
switch (blockStruct.SizeType)
|
||
|
|
{
|
||
|
|
/////////////////////
|
||
|
|
// Static Block Size
|
||
|
|
/////////////////////
|
||
|
|
case PacketSizeType.Static:
|
||
|
|
currentOffset += blockStruct.Size; // skip this block
|
||
|
|
break;
|
||
|
|
|
||
|
|
//////////////////
|
||
|
|
// End of Stream
|
||
|
|
//////////////////
|
||
|
|
case PacketSizeType.Eof:
|
||
|
|
eofFlagFound = true; // set EOF block found so we can exit the loop
|
||
|
|
break;
|
||
|
|
|
||
|
|
//////////////////////
|
||
|
|
// Varying Block Size
|
||
|
|
//////////////////////
|
||
|
|
case PacketSizeType.SizeBytes:
|
||
|
|
|
||
|
|
// Get the block size
|
||
|
|
blockSizeArray = ParseFile.ParseSimpleOffset(fs, currentOffset + currentBlockId.Length, blockStruct.Size);
|
||
|
|
|
||
|
|
if (!this.BlockSizeIsLittleEndian)
|
||
|
|
{
|
||
|
|
Array.Reverse(blockSizeArray);
|
||
|
|
}
|
||
|
|
|
||
|
|
switch (blockStruct.Size)
|
||
|
|
{
|
||
|
|
case 4:
|
||
|
|
blockSize = (uint)BitConverter.ToUInt32(blockSizeArray, 0);
|
||
|
|
break;
|
||
|
|
case 2:
|
||
|
|
blockSize = (uint)BitConverter.ToUInt16(blockSizeArray, 0);
|
||
|
|
break;
|
||
|
|
case 1:
|
||
|
|
blockSize = (uint)blockSizeArray[0];
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
throw new ArgumentOutOfRangeException(String.Format("Unhandled size block size.{0}", Environment.NewLine));
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
// if block type is audio or video, extract it
|
||
|
|
isAudioBlock = this.IsThisAnAudioBlock(currentBlockId);
|
||
|
|
|
||
|
|
if ((demuxOptions.ExtractAudio && isAudioBlock) ||
|
||
|
|
(demuxOptions.ExtractVideo && this.IsThisAVideoBlock(currentBlockId)))
|
||
|
|
{
|
||
|
|
// reset stream id
|
||
|
|
streamId = 0;
|
||
|
|
|
||
|
|
// if audio block, get the stream number from the queue
|
||
|
|
if (isAudioBlock && this.UsesSameIdForMultipleAudioTracks)
|
||
|
|
{
|
||
|
|
streamId = this.GetStreamId(fs, currentOffset);
|
||
|
|
currentStreamKey = (streamId | currentBlockIdVal);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
currentStreamKey = currentBlockIdVal;
|
||
|
|
}
|
||
|
|
|
||
|
|
// check if we've already started parsing this stream
|
||
|
|
if (!streamOutputWriters.ContainsKey(currentStreamKey))
|
||
|
|
{
|
||
|
|
// convert block id to little endian for naming
|
||
|
|
currentBlockIdNaming = BitConverter.GetBytes(currentStreamKey);
|
||
|
|
Array.Reverse(currentBlockIdNaming);
|
||
|
|
|
||
|
|
// build output file name
|
||
|
|
outputFileName = Path.GetFileNameWithoutExtension(this.FilePath);
|
||
|
|
outputFileName = outputFileName + "_" + BitConverter.ToUInt32(currentBlockIdNaming, 0).ToString("X8");
|
||
|
|
|
||
|
|
// add proper extension
|
||
|
|
if (this.IsThisAnAudioBlock(currentBlockId))
|
||
|
|
{
|
||
|
|
audioFileExtension = this.GetAudioFileExtension(fs, currentOffset);
|
||
|
|
outputFileName += audioFileExtension;
|
||
|
|
|
||
|
|
if (!this.StreamIdFileType.ContainsKey(streamId))
|
||
|
|
{
|
||
|
|
this.StreamIdFileType.Add(streamId, audioFileExtension);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
this.FileExtensionVideo = this.GetVideoFileExtension(fs, currentOffset);
|
||
|
|
outputFileName += this.FileExtensionVideo;
|
||
|
|
}
|
||
|
|
|
||
|
|
// add output directory
|
||
|
|
outputFileName = Path.Combine(Path.GetDirectoryName(this.FilePath), outputFileName);
|
||
|
|
|
||
|
|
// add an output stream for writing
|
||
|
|
streamOutputWriters[currentStreamKey] = new FileStream(outputFileName, FileMode.Create, FileAccess.ReadWrite);
|
||
|
|
}
|
||
|
|
|
||
|
|
// write the block
|
||
|
|
if (this.IsThisAnAudioBlock(currentBlockId))
|
||
|
|
{
|
||
|
|
// write audio
|
||
|
|
audioBlockSkipSize = this.GetAudioPacketHeaderSize(fs, currentOffset) + GetAudioPacketSubHeaderSize(fs, currentOffset, streamId);
|
||
|
|
audioBlockFooterSize = this.GetAudioPacketFooterSize(fs, currentOffset);
|
||
|
|
cutSize = (int)(blockSize - audioBlockSkipSize - audioBlockFooterSize);
|
||
|
|
if (cutSize > 0)
|
||
|
|
{
|
||
|
|
streamOutputWriters[currentStreamKey].Write(ParseFile.ParseSimpleOffset(fs, currentOffset + currentBlockId.Length + blockSizeArray.Length + audioBlockSkipSize, (int)(blockSize - audioBlockSkipSize)), 0, cutSize);
|
||
|
|
}
|
||
|
|
#if DEBUG
|
||
|
|
//else
|
||
|
|
//{
|
||
|
|
// int aaa = 1;
|
||
|
|
//}
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// write video
|
||
|
|
videoBlockSkipSize = this.GetVideoPacketHeaderSize(fs, currentOffset);
|
||
|
|
videoBlockFooterSize = this.GetVideoPacketFooterSize(fs, currentOffset);
|
||
|
|
cutSize = (int)(blockSize - videoBlockSkipSize - videoBlockFooterSize);
|
||
|
|
if (cutSize > 0)
|
||
|
|
{
|
||
|
|
streamOutputWriters[currentStreamKey].Write(ParseFile.ParseSimpleOffset(fs, currentOffset + currentBlockId.Length + blockSizeArray.Length + videoBlockSkipSize, (int)(blockSize - videoBlockSkipSize)), 0, cutSize);
|
||
|
|
}
|
||
|
|
#if DEBUG
|
||
|
|
//else
|
||
|
|
//{
|
||
|
|
// int vvv = 1;
|
||
|
|
//}
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// move to next block
|
||
|
|
currentOffset += currentBlockId.Length + blockSizeArray.Length + blockSize;
|
||
|
|
blockSizeArray = new byte[] { };
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else // this is an undexpected block type
|
||
|
|
{
|
||
|
|
this.closeAllWriters(streamOutputWriters);
|
||
|
|
Array.Reverse(currentBlockId);
|
||
|
|
throw new FormatException(String.Format("Block ID at 0x{0} not found in table: 0x{1}", currentOffset.ToString("X8"), BitConverter.ToUInt32(currentBlockId, 0).ToString("X8")));
|
||
|
|
}
|
||
|
|
|
||
|
|
// exit loop if EOF block found
|
||
|
|
if (eofFlagFound)
|
||
|
|
{
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
catch (Exception _ex)
|
||
|
|
{
|
||
|
|
this.closeAllWriters(streamOutputWriters);
|
||
|
|
throw new Exception(String.Format("Error parsing file at offset {0), '{1}'", currentOffset.ToString("X8"), _ex.Message), _ex);
|
||
|
|
}
|
||
|
|
} // while (currentOffset < fileSize)
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
this.closeAllWriters(streamOutputWriters);
|
||
|
|
throw new FormatException(String.Format("Cannot find Pack Header for file: {0}{1}", Path.GetFileName(this.FilePath), Environment.NewLine));
|
||
|
|
}
|
||
|
|
|
||
|
|
///////////////////////////////////
|
||
|
|
// Perform any final tasks needed
|
||
|
|
///////////////////////////////////
|
||
|
|
this.DoFinalTasks(fs, streamOutputWriters, demuxOptions.AddHeader);
|
||
|
|
|
||
|
|
//////////////////////////
|
||
|
|
// close all open writers
|
||
|
|
//////////////////////////
|
||
|
|
this.closeAllWriters(streamOutputWriters);
|
||
|
|
|
||
|
|
} // using (FileStream fs = File.OpenRead(path))
|
||
|
|
}
|
||
|
|
|
||
|
|
private void closeAllWriters(Dictionary<uint, FileStream> writers)
|
||
|
|
{
|
||
|
|
//////////////////////////
|
||
|
|
// close all open writers
|
||
|
|
//////////////////////////
|
||
|
|
foreach (uint b in writers.Keys)
|
||
|
|
{
|
||
|
|
if (writers[b].CanRead)
|
||
|
|
{
|
||
|
|
writers[b].Close();
|
||
|
|
writers[b].Dispose();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public static int GetMpegStreamType(string path)
|
||
|
|
{
|
||
|
|
int mpegType = -1;
|
||
|
|
|
||
|
|
using (FileStream fs = File.OpenRead(path))
|
||
|
|
{
|
||
|
|
// look for first packet
|
||
|
|
long currentOffset = ParseFile.GetNextOffset(fs, 0, MpegStream.PacketStartBytes);
|
||
|
|
|
||
|
|
if (currentOffset != -1)
|
||
|
|
{
|
||
|
|
currentOffset += 4;
|
||
|
|
fs.Position = currentOffset;
|
||
|
|
byte idByte = (byte)fs.ReadByte();
|
||
|
|
|
||
|
|
if ((int)ByteConversion.GetHighNibble(idByte) == 2)
|
||
|
|
{
|
||
|
|
mpegType = 1;
|
||
|
|
}
|
||
|
|
else if ((int)ByteConversion.GetHighNibble(idByte) == 4)
|
||
|
|
{
|
||
|
|
mpegType = 2;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
throw new FormatException(String.Format("Cannot find Pack Header for file: {0}{1}", Path.GetFileName(path), Environment.NewLine));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return mpegType;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|