private void listBox_ExtractFiles_DragDrop(object sender, DragEventArgs e)
{
string[] files = (string[])e.Data.GetData(DataFormats.FileDrop);
for (int i = 0; i < files.Length; i++)
{
string file = files[i];
if (Path.GetExtension(file).ToUpper() != ".PKS")
{
continue;
}
PKS entry = new PKS();
entry.FilePath = file;
listBox_ExtractFiles.Items.Add(entry);
}
}
private void button_CreatePKS_Click(object sender, EventArgs e)
{
SaveFileDialog saveFileDialog = new SaveFileDialog();
saveFileDialog.Filter = "PKS File (*.pks)|*.pks";
if (saveFileDialog.ShowDialog() == DialogResult.OK)
{
PKS pks = new PKS();
pks.Compress = checkBox_Compress.Checked;
foreach (IPACEntry entry in listBox_ArchiveFiles.Items)
{
pks.IPAC.Entries.Add(entry);
}
pks.FilePath = saveFileDialog.FileName;
pks.Write(saveFileDialog.FileName);
}
}
/// <summary>
/// Extracts a given PKS file into the output directory provided
/// </summary>
/// <param name="path">Path to PKS</param>
/// <param name="folder">Path to output the extracted PKS</param>
static void ExtractPKS(string path, string folder)
{
if (!File.Exists(path))
{
Console.WriteLine("{0} does not exist as a file, falling back to batch mode.", path);
if (Directory.Exists(path))
{
var ext = new List <string> {
".pks", ".PKS"
};
var myFiles = Directory.GetFiles(path, "*.*", SearchOption.AllDirectories).Where(s => ext.Contains(Path.GetExtension(s)));
foreach (string file in myFiles)
{
var currentChildDir = folder + "\\" + Path.GetDirectoryName(file.Replace(path, ""));
string filename = Path.GetFileName(file);
string dest = currentChildDir + "\\_" + filename + "_\\";
string dir = Path.GetDirectoryName(dest);
if (!Directory.Exists(dir))
{
Directory.CreateDirectory(dir);
}
try
{
PKS pks = new PKS(file);
if (pks != null)
{
pks.Unpack(dest);
}
if (bVerbose)
{
Console.WriteLine("Unpacking {0} to {1}", file, dest);
}
}
catch (Exception e)
{
Console.WriteLine("Oops! {0} failed!\nException: {1}", file, e.ToString());
++iNumFailedOperations;
}
++iNumOperations;
}
}
else
{
Console.WriteLine("{0} does not exist. Cancelling operation.", path);
return;
}
}
else
{
try
{
PKS pks = new PKS(path);
if (pks != null)
{
pks.Unpack(folder);
}
if (bVerbose)
{
Console.WriteLine("Unpacked {0} to {1}", path, folder);
}
}
catch (Exception e)
{
Console.WriteLine("Oops! {0} failed!\nException: {1}", path, e.ToString());
++iNumFailedOperations;
}
++iNumOperations;
}
return;
}
private void ReadFileRecursive(byte[] data)
{
if (GZ.IsValid(data))
{
using (MemoryStream memstream = new MemoryStream(data))
{
GZ gz = new GZ(memstream);
ReadFileRecursive(gz.ContentBuffer);
}
}
if (AFS.IsValid(data))
{
using (MemoryStream memstream = new MemoryStream(data))
{
AFS afs = new AFS(memstream);
foreach (var entry in afs.Entries)
{
ReadFileRecursive(entry.Buffer);
}
}
}
if (PKF.IsValid(data))
{
using (MemoryStream memstream = new MemoryStream(data))
{
PKF pkf = new PKF(memstream);
foreach (var entry in pkf.Entries)
{
ReadFileRecursive(entry.Buffer);
}
}
}
if (PKS.IsValid(data))
{
using (MemoryStream memstream = new MemoryStream(data))
{
PKS pks = new PKS(memstream);
foreach (var entry in pks.IPAC.Entries)
{
ReadFileRecursive(entry.Buffer);
}
}
}
if (TEXN.IsValid(data))
{
using (MemoryStream memstream = new MemoryStream(data))
{
TEXN tex = new TEXN(memstream);
TexEntry entry = new TexEntry();
entry.texID = tex.TextureID;
entry.image = tex.Texture;
textures.Add(entry);
}
}
if (MT5.IsValid(data))
{
using (MemoryStream memstream = new MemoryStream(data))
{
MT5 mt5 = new MT5(memstream);
foreach (var tex in mt5.Textures)
{
if (tex.Image != null)
{
TexEntry entry = new TexEntry();
entry.texID = tex.TextureID;
entry.image = tex.Image;
textures.Add(entry);
}
}
}
}
if (MT7.IsValid(data))
{
using (MemoryStream memstream = new MemoryStream(data))
{
MT7 mt7 = new MT7(memstream);
foreach (var tex in mt7.Textures)
{
if (tex.Image != null)
{
TexEntry entry = new TexEntry();
entry.texID = tex.TextureID;
entry.image = tex.Image;
textures.Add(entry);
}
}
}
}
}
public static string GetExtensionFromBuffer(byte[] buffer)
{
//Archives
if (AFS.IsValid(buffer))
{
return("AFS");
}
if (GZ.IsValid(buffer))
{
return("GZ");
}
if (IDX.IsValid(buffer))
{
return("IDX");
}
if (IPAC.IsValid(buffer))
{
return("IPAC");
}
if (PKF.IsValid(buffer))
{
return("PKF");
}
if (PKS.IsValid(buffer))
{
return("PKS");
}
//if (SPR.IsValid(buffer)) return typeof(SPR); //same as TEXN skip and base identification on extension
if (TAD.IsValid(buffer))
{
return("TAD");
}
//Textures/Images
if (TEXN.IsValid(buffer))
{
return("TEXN");
}
if (PVRT.IsValid(buffer))
{
return("PVRT");
}
if (DDS.IsValid(buffer))
{
return("DDS");
}
//Models
if (MT5.IsValid(buffer))
{
return("MT5");
}
if (MT7.IsValid(buffer))
{
return("MT7");
}
//Subtitles
if (SUB.IsValid(buffer))
{
return("SUB");
}
return("UNKNOWN");
}
/// <summary>
/// Trys to find the fitting file type for the given file with the file signature.
/// </summary>
public static Type GetFileTypeFromSignature(Stream stream)
{
byte[] buffer = new byte[8];
stream.Read(buffer, 0, buffer.Length);
//Archives
if (AFS.IsValid(buffer))
{
return(typeof(AFS));
}
if (GZ.IsValid(buffer))
{
return(typeof(GZ));
}
if (IDX.IsValid(buffer))
{
return(typeof(IDX));
}
if (IPAC.IsValid(buffer))
{
return(typeof(IPAC));
}
if (PKF.IsValid(buffer))
{
return(typeof(PKF));
}
if (PKS.IsValid(buffer))
{
return(typeof(PKS));
}
//if (SPR.IsValid(buffer)) return typeof(SPR); //same as TEXN skip and base identification on extension
if (TAD.IsValid(buffer))
{
return(typeof(TAD));
}
//Textures/Images
if (TEXN.IsValid(buffer))
{
return(typeof(TEXN));
}
if (PVRT.IsValid(buffer))
{
return(typeof(PVRT));
}
if (DDS.IsValid(buffer))
{
return(typeof(DDS));
}
//Models
if (MT5.IsValid(buffer))
{
return(typeof(MT5));
}
if (MT7.IsValid(buffer))
{
return(typeof(MT7));
}
//Subtitles
if (SUB.IsValid(buffer))
{
return(typeof(SUB));
}
return(null);
}
