/// <summary>
/// PCCObject class constructor. It also loads namelist, importlist, exportinfo, and exportdata from pcc file
/// </summary>
/// <param name="pccFilePath">full path + file name of desired pcc file.</param>
private ME3Package(string pccFilePath)
{
FileName = Path.GetFullPath(pccFilePath);
MemoryStream inStream;
using (FileStream pccStream = File.OpenRead(FileName))
{
header = pccStream.ReadBytes(headerSize);
if (magic != ZBlock.magic &&
magic.Swap() != ZBlock.magic)
{
throw new FormatException("Not an Unreal package!");
}
if (lowVers != 684 && highVers != 194)
{
throw new FormatException("Not an ME3 Package!");
}
if (IsCompressed)
{
inStream = CompressionHelper.DecompressME3(pccStream);
//read uncompressed header
inStream.Seek(0, SeekOrigin.Begin);
inStream.Read(header, 0, header.Length); //load uncompressed header
}
else
{
inStream = new MemoryStream();
pccStream.Seek(0, SeekOrigin.Begin);
pccStream.CopyTo(inStream);
}
}
names = new List <string>();
inStream.Seek(NameOffset, SeekOrigin.Begin);
for (int i = 0; i < NameCount; i++)
{
int strLength = inStream.ReadValueS32();
string str = inStream.ReadString(strLength * -2, true, Encoding.Unicode);
names.Add(str);
}
imports = new List <ImportEntry>();
inStream.Seek(ImportOffset, SeekOrigin.Begin);
for (int i = 0; i < ImportCount; i++)
{
ImportEntry imp = new ImportEntry(this, inStream);
imp.Index = i;
imp.PropertyChanged += importChanged;
imports.Add(imp);
}
exports = new List <IExportEntry>();
inStream.Seek(ExportOffset, SeekOrigin.Begin);
for (int i = 0; i < ExportCount; i++)
{
ME3ExportEntry e = new ME3ExportEntry(this, inStream);
e.Index = i;
e.PropertyChanged += exportChanged;
exports.Add(e);
}
}
public void addExport(ME3ExportEntry exportEntry)
{
if (exportEntry.FileRef != this)
{
throw new Exception("you cannot add a new export entry from another pcc file, it has invalid references!");
}
exportEntry.DataChanged = true;
exportEntry.Index = exports.Count;
exportEntry.PropertyChanged += exportChanged;
exports.Add(exportEntry);
ExportCount = exports.Count;
updateTools(PackageChange.ExportAdd, ExportCount - 1);
}
public IExportEntry Clone()
{
ME3ExportEntry newExport = new ME3ExportEntry(FileRef as ME3Package);
newExport.header = (byte[])this.header.Clone();
newExport.headerOffset = 0;
newExport.Data = this.Data;
int index = 0;
string name = ObjectName;
foreach (IExportEntry ent in FileRef.Exports)
{
if (name == ent.ObjectName && ent.indexValue > index)
{
index = ent.indexValue;
}
}
index++;
newExport.indexValue = index;
return(newExport);
}
/// <summary>
/// This method is an alternate way of saving PCCs
/// Instead of reconstructing the PCC from the data taken, it instead copies across the existing
/// data, appends the name list and import list, appends changed and new exports, and then appends the export list.
/// Changed exports with the same datasize or smaller are updaed in place.
/// </summary>
/// <param name="newFileName">The filename to write to</param>
private void appendSave(string newFileName)
{
IEnumerable <ME3ExportEntry> replaceExports;
IEnumerable <ME3ExportEntry> appendExports;
int lastDataOffset;
int max;
if (IsAppend)
{
replaceExports = exports.Where(export => export.DataChanged && export.DataOffset < NameOffset && export.DataSize <= export.OriginalDataSize);
appendExports = exports.Where(export => export.DataOffset > NameOffset || (export.DataChanged && export.DataSize > export.OriginalDataSize));
max = exports.Where(exp => exp.DataOffset < NameOffset).Max(e => e.DataOffset);
}
else
{
IEnumerable <ME3ExportEntry> changedExports;
changedExports = exports.Where(export => export.DataChanged);
replaceExports = changedExports.Where(export => export.DataSize <= export.OriginalDataSize);
appendExports = changedExports.Except(replaceExports);
max = exports.Max(maxExport => maxExport.DataOffset);
}
ME3ExportEntry lastExport = exports.Find(export => export.DataOffset == max);
lastDataOffset = lastExport.DataOffset + lastExport.DataSize;
byte[] oldPCC = new byte[lastDataOffset];
if (IsCompressed)
{
oldPCC = CompressionHelper.Decompress(FileName).Take(lastDataOffset).ToArray();
IsCompressed = false;
}
else
{
using (FileStream oldPccStream = new FileStream(this.FileName, FileMode.Open))
{
//Read the original data up to the last export
oldPccStream.Read(oldPCC, 0, lastDataOffset);
}
}
//Start writing the new file
using (FileStream newPCCStream = new FileStream(newFileName, FileMode.Create))
{
newPCCStream.Seek(0, SeekOrigin.Begin);
//Write the original file up til the last original export (note that this leaves in all the original exports)
newPCCStream.Write(oldPCC, 0, lastDataOffset);
//write the in-place export updates
foreach (ME3ExportEntry export in replaceExports)
{
newPCCStream.Seek(export.DataOffset, SeekOrigin.Begin);
export.DataSize = export.Data.Length;
newPCCStream.WriteBytes(export.Data);
}
newPCCStream.Seek(lastDataOffset, SeekOrigin.Begin);
//Set the new nameoffset and namecounts
NameOffset = (int)newPCCStream.Position;
NameCount = names.Count;
//Write out the namelist
foreach (string name in names)
{
newPCCStream.WriteValueS32(-(name.Length + 1));
newPCCStream.WriteString(name + "\0", (uint)(name.Length + 1) * 2, Encoding.Unicode);
}
//Write the import list
ImportOffset = (int)newPCCStream.Position;
ImportCount = imports.Count;
foreach (ImportEntry import in imports)
{
newPCCStream.WriteBytes(import.header);
}
//Append the new data
foreach (ME3ExportEntry export in appendExports)
{
export.DataOffset = (int)newPCCStream.Position;
export.DataSize = export.Data.Length;
//update offsets for pcc-stored audio in wwisestreams
if (export.ClassName == "WwiseStream" && export.GetProperty <NameProperty>("Filename") == null)
{
byte[] binData = export.getBinaryData();
binData.OverwriteRange(12, BitConverter.GetBytes(export.DataOffset + export.propsEnd() + 16));
export.setBinaryData(binData);
}
newPCCStream.WriteBytes(export.Data);
}
//Write the export list
ExportOffset = (int)newPCCStream.Position;
ExportCount = exports.Count;
foreach (ME3ExportEntry export in exports)
{
newPCCStream.WriteBytes(export.header);
}
IsAppend = true;
//write the updated header
newPCCStream.Seek(0, SeekOrigin.Begin);
newPCCStream.WriteBytes(header);
}
AfterSave();
}
/// <summary>
/// save PCCObject to file by reconstruction from data
/// </summary>
/// <param name="path">full path + file name.</param>
/// <param name="compress">true if you want a zlib compressed pcc file.</param>
public void saveByReconstructing(string path, bool compress)
{
try
{
this.IsCompressed = false;
MemoryStream m = new MemoryStream();
m.WriteBytes(header);
//name table
NameOffset = (int)m.Position;
NameCount = names.Count;
foreach (string s in names)
{
string text = s;
if (!text.EndsWith("\0"))
{
text += "\0";
}
m.WriteValueS32(-text.Length);
foreach (char c in text)
{
m.WriteByte((byte)c);
m.WriteByte(0);
}
}
//import table
ImportOffset = (int)m.Position;
ImportCount = imports.Count;
foreach (ImportEntry e in imports)
{
m.WriteBytes(e.header);
}
//export table
ExportOffset = (int)m.Position;
ExportCount = exports.Count;
for (int i = 0; i < exports.Count; i++)
{
ME3ExportEntry e = exports[i];
e.headerOffset = (uint)m.Position;
m.WriteBytes(e.header);
}
//freezone
int FreeZoneSize = FreeZoneEnd - FreeZoneStart;
FreeZoneStart = (int)m.Position;
m.Write(new byte[FreeZoneSize], 0, FreeZoneSize);
FreeZoneEnd = expDataBegOffset = (int)m.Position;
//export data
for (int i = 0; i < exports.Count; i++)
{
ME3ExportEntry e = exports[i];
e.DataOffset = (int)m.Position;
e.DataSize = e.Data.Length;
//update offsets for pcc-stored audio in wwisestreams
if (e.ClassName == "WwiseStream" && e.GetProperty <NameProperty>("Filename") == null)
{
byte[] binData = e.getBinaryData();
binData.OverwriteRange(12, BitConverter.GetBytes(e.DataOffset + e.propsEnd() + 16));
e.setBinaryData(binData);
}
m.WriteBytes(e.Data);
long pos = m.Position;
m.Seek(e.headerOffset + 32, SeekOrigin.Begin);
m.WriteValueS32(e.DataSize);
m.WriteValueS32(e.DataOffset);
m.Seek(pos, SeekOrigin.Begin);
}
//update header
m.Seek(0, SeekOrigin.Begin);
m.WriteBytes(header);
if (compress)
{
CompressionHelper.CompressAndSave(m, path);
}
else
{
File.WriteAllBytes(path, m.ToArray());
}
AfterSave();
}
catch (Exception ex)
{
MessageBox.Show("PCC Save error:\n" + ex.Message);
}
}
/// <summary>
/// PCCObject class constructor. It also loads namelist, importlist, exportinfo, and exportdata from pcc file
/// </summary>
/// <param name="pccFilePath">full path + file name of desired pcc file.</param>
private ME3Package(string pccFilePath)
{
FileName = Path.GetFullPath(pccFilePath);
MemoryStream listsStream;
names = new List <string>();
imports = new List <ImportEntry>();
exports = new List <ME3ExportEntry>();
using (FileStream pccStream = File.OpenRead(FileName))
{
header = pccStream.ReadBytes(headerSize);
if (magic != ZBlock.magic &&
magic.Swap() != ZBlock.magic)
{
throw new FormatException("not a pcc file");
}
if (lowVers != 684 && highVers != 194)
{
throw new FormatException("unsupported version");
}
if (IsCompressed)
{
listsStream = CompressionHelper.DecompressME3(pccStream);
//correcting the header
listsStream.Seek(0, SeekOrigin.Begin);
listsStream.Read(header, 0, header.Length);
}
else
{
listsStream = new MemoryStream();
pccStream.Seek(0, SeekOrigin.Begin);
pccStream.CopyTo(listsStream);
}
}
// fill names list
listsStream.Seek(NameOffset, SeekOrigin.Begin);
for (int i = 0; i < NameCount; i++)
{
long currOffset = listsStream.Position;
int strLength = listsStream.ReadValueS32();
string str = listsStream.ReadString(strLength * -2, true, Encoding.Unicode);
names.Add(str);
}
// fill import list
listsStream.Seek(ImportOffset, SeekOrigin.Begin);
for (int i = 0; i < ImportCount; i++)
{
long offset = listsStream.Position;
ImportEntry imp = new ImportEntry(this, listsStream);
imp.Index = i;
imp.PropertyChanged += importChanged;
imports.Add(imp);
}
// fill export list
listsStream.Seek(ExportOffset, SeekOrigin.Begin);
byte[] buffer;
for (int i = 0; i < ExportCount; i++)
{
uint expInfoOffset = (uint)listsStream.Position;
listsStream.Seek(44, SeekOrigin.Current);
int count = listsStream.ReadValueS32();
listsStream.Seek(-48, SeekOrigin.Current);
int expInfoSize = 68 + (count * 4);
buffer = new byte[expInfoSize];
listsStream.Read(buffer, 0, buffer.Length);
ME3ExportEntry e = new ME3ExportEntry(this, buffer, expInfoOffset);
long headerEnd = listsStream.Position;
buffer = new byte[e.DataSize];
listsStream.Seek(e.DataOffset, SeekOrigin.Begin);
listsStream.Read(buffer, 0, buffer.Length);
e.Data = buffer;
e.DataChanged = false;
e.Index = i;
e.PropertyChanged += exportChanged;
exports.Add(e);
listsStream.Seek(headerEnd, SeekOrigin.Begin);
}
}
