CodeWalker/Tools/ExtractTexForm.cs

268 lines
11 KiB
C#

using CodeWalker.GameFiles;
using CodeWalker.Properties;
using CodeWalker.Utils;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
namespace CodeWalker
{
public partial class ExtractTexForm : Form
{
private volatile bool KeysLoaded = false;
private volatile bool InProgress = false;
private volatile bool AbortOperation = false;
public ExtractTexForm()
{
InitializeComponent();
}
private void ExtractTexForm_Load(object sender, EventArgs e)
{
FolderTextBox.Text = GTAFolder.CurrentGTAFolder;
OutputFolderTextBox.Text = Settings.Default.ExtractedTexturesFolder;
try
{
GTA5Keys.LoadFromPath(GTAFolder.CurrentGTAFolder, Settings.Default.Key);
KeysLoaded = true;
UpdateExtractStatus("Ready to extract.");
}
catch
{
UpdateExtractStatus("Keys not found! This shouldn't happen.");
}
}
private void OutputFolderTextBox_TextChanged(object sender, EventArgs e)
{
Settings.Default.ExtractedTexturesFolder = OutputFolderTextBox.Text;
}
private void FolderBrowseButton_Click(object sender, EventArgs e)
{
GTAFolder.UpdateGTAFolder(false);
FolderTextBox.Text = GTAFolder.CurrentGTAFolder;
}
private void OutputFolderBrowseButton_Click(object sender, EventArgs e)
{
FolderBrowserDialog.SelectedPath = OutputFolderTextBox.Text;
DialogResult res = FolderBrowserDialog.ShowDialog();
if (res == DialogResult.OK)
{
OutputFolderTextBox.Text = FolderBrowserDialog.SelectedPath;
}
}
private void UpdateExtractStatus(string text)
{
try
{
if (InvokeRequired)
{
Invoke(new Action(() => { UpdateExtractStatus(text); }));
}
else
{
ExtractStatusLabel.Text = text;
}
}
catch { }
}
private void ExtractButton_Click(object sender, EventArgs e)
{
if (InProgress) return;
if (!KeysLoaded)
{
MessageBox.Show("Please scan a GTA 5 exe dump for keys first, or include key files in this app's folder!");
return;
}
if (!Directory.Exists(FolderTextBox.Text))
{
MessageBox.Show("Folder doesn't exist: " + FolderTextBox.Text);
return;
}
if (!Directory.Exists(OutputFolderTextBox.Text))
{
MessageBox.Show("Folder doesn't exist: " + OutputFolderTextBox.Text);
return;
}
//if (Directory.GetFiles(OutputFolderTextBox.Text, "*.ysc", SearchOption.AllDirectories).Length > 0)
//{
// if (MessageBox.Show("Output folder already contains .ysc files. Are you sure you want to continue?", "Output folder already contains .ysc files", MessageBoxButtons.OKCancel) != DialogResult.OK)
// {
// return;
// }
//}
InProgress = true;
AbortOperation = false;
string searchpath = FolderTextBox.Text;
string outputpath = OutputFolderTextBox.Text;
string replpath = searchpath + "\\";
bool bytd = YtdChecBox.Checked;
bool bydr = YdrCheckBox.Checked;
bool bydd = YddCheckBox.Checked;
bool byft = YftCheckBox.Checked;
Task.Run(() =>
{
UpdateExtractStatus("Keys loaded.");
RpfManager rpfman = new RpfManager();
rpfman.Init(searchpath, UpdateExtractStatus, UpdateExtractStatus);
UpdateExtractStatus("Beginning texture extraction...");
StringBuilder errsb = new StringBuilder();
foreach (RpfFile rpf in rpfman.AllRpfs)
{
foreach (RpfEntry entry in rpf.AllEntries)
{
if (AbortOperation)
{
UpdateExtractStatus("Operation aborted");
InProgress = false;
return;
}
try
{
if (bytd && entry.NameLower.EndsWith(".ytd"))
{
UpdateExtractStatus(entry.Path);
YtdFile ytd = rpfman.GetFile<YtdFile>(entry);
if (ytd == null) throw new Exception("Couldn't load file.");
if (ytd.TextureDict == null) throw new Exception("Couldn't load texture dictionary.");
if (ytd.TextureDict.Textures == null) throw new Exception("Couldn't load texture dictionary texture array.");
if (ytd.TextureDict.Textures.data_items == null) throw new Exception("Texture dictionary had no entries...");
foreach (var tex in ytd.TextureDict.Textures.data_items)
{
SaveTexture(tex, entry, outputpath);
}
}
else if (bydr && entry.NameLower.EndsWith(".ydr"))
{
UpdateExtractStatus(entry.Path);
YdrFile ydr = rpfman.GetFile<YdrFile>(entry);
if (ydr == null) throw new Exception("Couldn't load file.");
if (ydr.Drawable == null) throw new Exception("Couldn't load drawable.");
if (ydr.Drawable.ShaderGroup != null)
{
var ydrtd = ydr.Drawable.ShaderGroup.TextureDictionary;
if ((ydrtd != null) && (ydrtd.Textures != null) && (ydrtd.Textures.data_items != null))
{
foreach (var tex in ydrtd.Textures.data_items)
{
SaveTexture(tex, entry, outputpath);
}
}
}
}
else if (bydd && entry.NameLower.EndsWith(".ydd"))
{
UpdateExtractStatus(entry.Path);
YddFile ydd = rpfman.GetFile<YddFile>(entry);
if (ydd == null) throw new Exception("Couldn't load file.");
//if (ydd.DrawableDict == null) throw new Exception("Couldn't load drawable dictionary.");
//if (ydd.DrawableDict.Drawables == null) throw new Exception("Drawable dictionary had no items...");
//if (ydd.DrawableDict.Drawables.data_items == null) throw new Exception("Drawable dictionary had no items...");
if ((ydd.Dict==null)||(ydd.Dict.Count==0)) throw new Exception("Drawable dictionary had no items...");
foreach (var drawable in ydd.Dict.Values)
{
if (drawable.ShaderGroup != null)
{
var ydrtd = drawable.ShaderGroup.TextureDictionary;
if ((ydrtd != null) && (ydrtd.Textures != null) && (ydrtd.Textures.data_items != null))
{
foreach (var tex in ydrtd.Textures.data_items)
{
SaveTexture(tex, entry, outputpath);
}
}
}
}
}
else if (byft && entry.NameLower.EndsWith(".yft"))
{
UpdateExtractStatus(entry.Path);
YftFile yft = rpfman.GetFile<YftFile>(entry);
if (yft == null) throw new Exception("Couldn't load file.");
if (yft.Fragment == null) throw new Exception("Couldn't load fragment.");
if (yft.Fragment.Drawable != null)
{
if (yft.Fragment.Drawable.ShaderGroup != null)
{
var ydrtd = yft.Fragment.Drawable.ShaderGroup.TextureDictionary;
if ((ydrtd != null) && (ydrtd.Textures != null) && (ydrtd.Textures.data_items != null))
{
foreach (var tex in ydrtd.Textures.data_items)
{
SaveTexture(tex, entry, outputpath);
}
}
}
}
}
}
catch (Exception ex)
{
string err = entry.Name + ": " + ex.Message;
UpdateExtractStatus(err);
errsb.AppendLine(err);
}
}
}
File.WriteAllText(outputpath + "\\_errors.txt", errsb.ToString());
UpdateExtractStatus("Complete.");
InProgress = false;
});
}
private void SaveTexture(Texture tex, RpfEntry entry, string folder)
{
//DirectXTex
byte[] dds = DDSIO.GetDDSFile(tex);
string bpath = folder + "\\" + entry.Name + "_" + tex.Name;
string fpath = bpath + ".dds";
int c = 1;
while (File.Exists(fpath))
{
fpath = bpath + "_Copy" + c.ToString() + ".dds";
c++;
}
File.WriteAllBytes(fpath, dds);
}
private void AbortButton_Click(object sender, EventArgs e)
{
AbortOperation = true;
}
}
}