mirror of
https://github.com/AdAstra-LD/DS-Pokemon-Rom-Editor.git
synced 2026-05-11 14:34:21 -05:00
755 lines
31 KiB
C#
755 lines
31 KiB
C#
using DSPRE.ROMFiles;
|
|
using Ekona.Images;
|
|
using Images;
|
|
using LibNDSFormats.NSBMD;
|
|
using LibNDSFormats.NSBTX;
|
|
using Microsoft.WindowsAPICodePack.Dialogs;
|
|
using NSMBe4.DSFileSystem;
|
|
using ScintillaNET;
|
|
using ScintillaNET.Utils;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Drawing;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Text;
|
|
using System.Windows.Forms;
|
|
using Tao.OpenGl;
|
|
using Tao.Platform.Windows;
|
|
using Velopack;
|
|
using Velopack.Sources;
|
|
using static DSPRE.RomInfo;
|
|
|
|
namespace DSPRE {
|
|
public static class Helpers {
|
|
static MainProgram MainProgram;
|
|
|
|
public static RomInfo romInfo;
|
|
public static bool hideBuildings = new bool();
|
|
|
|
public static NSBMDGlRenderer mapRenderer;
|
|
|
|
public static ToolStripProgressBar toolStripProgressBar { get { return MainProgram.toolStripProgressBar; } }
|
|
|
|
public static void Initialize(MainProgram mainProgram) {
|
|
MainProgram = mainProgram;
|
|
mapRenderer = new NSBMDGlRenderer();
|
|
}
|
|
|
|
public static void CheckForUpdates(bool silent = true)
|
|
{
|
|
AppLogger.Info("Checking for updates...");
|
|
var mgr = new UpdateManager(new GithubSource("https://github.com/Mixone-FinallyHere/DS-Pokemon-Rom-Editor", "", prerelease: false));
|
|
|
|
var newVersion = mgr.CheckForUpdates();
|
|
if (newVersion == null)
|
|
{
|
|
AppLogger.Info("No updates available.");
|
|
if (!silent)
|
|
MessageBox.Show("No update is available.", "Information", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
DialogResult update = MessageBox.Show($"A new DSPRE version is available: {newVersion.TargetFullRelease.Version}.\nDo you want to install it?", "New update", MessageBoxButtons.YesNo, MessageBoxIcon.Information);
|
|
if (update == DialogResult.Yes)
|
|
{
|
|
AppLogger.Info($"New version available: {newVersion.TargetFullRelease.Version} (Current: {mgr.CurrentVersion})");
|
|
mgr.DownloadUpdates(newVersion);
|
|
|
|
AppLogger.Info($"Installing update {newVersion.TargetFullRelease.Version} and restarting app...");
|
|
mgr.ApplyUpdatesAndRestart(newVersion);
|
|
}
|
|
else
|
|
{
|
|
AppLogger.Info("User declined to update the application.");
|
|
}
|
|
}
|
|
}
|
|
|
|
static bool disableHandlersOld;
|
|
static bool disableHandlers;
|
|
|
|
public static bool HandlersDisabled { get { return disableHandlers == true; } }
|
|
public static bool HandlersEnabled { get { return disableHandlers == false; } }
|
|
|
|
public static void BackUpDisableHandler() {
|
|
disableHandlersOld = disableHandlers;
|
|
}
|
|
|
|
public static void RestoreDisableHandler() {
|
|
disableHandlers = disableHandlersOld;
|
|
}
|
|
|
|
public static void DisableHandlers() {
|
|
disableHandlers = true;
|
|
}
|
|
|
|
public static void EnableHandlers() {
|
|
disableHandlers = false;
|
|
}
|
|
|
|
public static string GetDSPREVersion()
|
|
{
|
|
return "" + Assembly.GetExecutingAssembly().GetName().Version.Major + "." + Assembly.GetExecutingAssembly().GetName().Version.Minor +
|
|
"." + Assembly.GetExecutingAssembly().GetName().Version.Build;
|
|
}
|
|
|
|
public static void statusLabelMessage(string msg = "Ready") {
|
|
ToolStripStatusLabel statusLabel = MainProgram.statusLabel;
|
|
statusLabel.Text = msg;
|
|
statusLabel.Font = new Font(statusLabel.Font, FontStyle.Regular);
|
|
statusLabel.ForeColor = Color.Black;
|
|
statusLabel.Invalidate();
|
|
}
|
|
|
|
public static void statusLabelError(string errorMsg, bool severe = true) {
|
|
ToolStripStatusLabel statusLabel = MainProgram.statusLabel;
|
|
statusLabel.Text = errorMsg;
|
|
statusLabel.Font = new Font(statusLabel.Font, FontStyle.Bold);
|
|
statusLabel.ForeColor = severe ? Color.Red : Color.DarkOrange;
|
|
statusLabel.Invalidate();
|
|
}
|
|
|
|
//Locate File - buttons
|
|
public static void ExplorerSelect(string path) {
|
|
if (System.IO.File.Exists(path)) {
|
|
Process.Start("explorer.exe", "/select" + "," + "\"" + path + "\"");
|
|
}
|
|
}
|
|
|
|
public static string[] GetTrainerNames() {
|
|
List<string> trainerList = new List<string>();
|
|
|
|
/* Store all trainer names and classes */
|
|
TextArchive trainerClasses = new TextArchive(RomInfo.trainerClassMessageNumber);
|
|
TextArchive trainerNames = new TextArchive(RomInfo.trainerNamesMessageNumber);
|
|
|
|
int trainerCount = Filesystem.GetTrainerPropertiesCount();
|
|
for (int i = 0; i < trainerCount; i++) {
|
|
string path = Filesystem.GetTrainerPropertiesPath(i);
|
|
int classMessageID = BitConverter.ToUInt16(DSUtils.ReadFromFile(path, startOffset: 1, 2), 0);
|
|
string currentTrainerName;
|
|
|
|
if (i < trainerNames.messages.Count) {
|
|
currentTrainerName = trainerNames.messages[i];
|
|
} else {
|
|
currentTrainerName = TrainerFile.NAME_NOT_FOUND;
|
|
}
|
|
|
|
trainerList.Add("[" + i.ToString("D2") + "] " + trainerClasses.messages[classMessageID] + " " + currentTrainerName);
|
|
}
|
|
|
|
return trainerList.ToArray();
|
|
}
|
|
|
|
public static void MW_LoadModelTextures(NSBMD model, string textureFolder, int fileID) {
|
|
if (fileID < 0) {
|
|
return;
|
|
}
|
|
|
|
string texturePath = Filesystem.GetPath(textureFolder, fileID);
|
|
model.materials = NSBTXLoader.LoadNsbtx(new MemoryStream(System.IO.File.ReadAllBytes(texturePath)), out model.Textures, out model.Palettes);
|
|
try {
|
|
model.MatchTextures();
|
|
} catch {
|
|
}
|
|
}
|
|
|
|
public static void MW_LoadModelTextures(MapFile mapFile, int fileID) {
|
|
MW_LoadModelTextures(mapFile.mapModel, Filesystem.mapTextures, fileID);
|
|
}
|
|
|
|
public static void MW_LoadModelTextures(Building building, int fileID) {
|
|
MW_LoadModelTextures(building.NSBMDFile, Filesystem.buildingTextures, fileID);
|
|
}
|
|
|
|
public static void SetupRenderer(float ang, float dist, float elev, float perspective, int width, int height) {
|
|
//TODO: improve this
|
|
Gl.glEnable(Gl.GL_RESCALE_NORMAL);
|
|
Gl.glEnable(Gl.GL_COLOR_MATERIAL);
|
|
Gl.glEnable(Gl.GL_DEPTH_TEST);
|
|
Gl.glEnable(Gl.GL_NORMALIZE);
|
|
Gl.glDisable(Gl.GL_CULL_FACE);
|
|
Gl.glFrontFace(Gl.GL_CCW);
|
|
Gl.glClearDepth(1);
|
|
Gl.glEnable(Gl.GL_ALPHA_TEST);
|
|
Gl.glBlendFunc(Gl.GL_SRC_ALPHA, Gl.GL_ONE_MINUS_SRC_ALPHA);
|
|
Gl.glEnable(Gl.GL_BLEND);
|
|
Gl.glAlphaFunc(Gl.GL_GREATER, 0f);
|
|
Gl.glClearColor(51f / 255f, 51f / 255f, 51f / 255f, 1f);
|
|
Gl.glViewport(0, 0, width, height);
|
|
float aspect = width / height; //(vp[2] - vp[0]) / (vp[3] - vp[1]);
|
|
Gl.glMatrixMode(Gl.GL_PROJECTION);
|
|
Gl.glLoadIdentity();
|
|
Glu.gluPerspective(perspective, aspect, 0.2f, 500.0f); //0.02f, 32.0f);
|
|
Gl.glTranslatef(0, 0, -dist);
|
|
Gl.glRotatef(elev, 1, 0, 0);
|
|
Gl.glRotatef(ang, 0, 1, 0);
|
|
Gl.glMatrixMode(Gl.GL_MODELVIEW);
|
|
Gl.glLoadIdentity();
|
|
Gl.glTranslatef(0, 0, -dist);
|
|
Gl.glRotatef(elev, 1, 0, 0);
|
|
Gl.glRotatef(-ang, 0, 1, 0);
|
|
Gl.glLightfv(Gl.GL_LIGHT0, Gl.GL_POSITION, new float[] { 1, 1, 1, 0 });
|
|
Gl.glLightfv(Gl.GL_LIGHT1, Gl.GL_POSITION, new float[] { 1, 1, 1, 0 });
|
|
Gl.glLightfv(Gl.GL_LIGHT2, Gl.GL_POSITION, new float[] { 1, 1, 1, 0 });
|
|
Gl.glLightfv(Gl.GL_LIGHT3, Gl.GL_POSITION, new float[] { 1, 1, 1, 0 });
|
|
Gl.glLoadIdentity();
|
|
Gl.glBindTexture(Gl.GL_TEXTURE_2D, 0);
|
|
Gl.glColor3f(1.0f, 1.0f, 1.0f);
|
|
Gl.glDepthMask(Gl.GL_TRUE);
|
|
Gl.glClear(Gl.GL_COLOR_BUFFER_BIT | Gl.GL_DEPTH_BUFFER_BIT);
|
|
}
|
|
|
|
public static void RenderMap(ref MapFile mapFile, int width, int height, float ang, float dist, float elev, float perspective, bool mapTexturesON = true, bool buildingTexturesON = true) {
|
|
#region Useless variables that the rendering API still needs
|
|
|
|
MKDS_Course_Editor.NSBTA.NSBTA.NSBTA_File ani = new MKDS_Course_Editor.NSBTA.NSBTA.NSBTA_File();
|
|
MKDS_Course_Editor.NSBTP.NSBTP.NSBTP_File tp = new MKDS_Course_Editor.NSBTP.NSBTP.NSBTP_File();
|
|
MKDS_Course_Editor.NSBCA.NSBCA.NSBCA_File ca = new MKDS_Course_Editor.NSBCA.NSBCA.NSBCA_File();
|
|
int[] aniframeS = new int[0];
|
|
|
|
#endregion
|
|
|
|
/* Adjust rendering settings */
|
|
SetupRenderer(ang, dist, elev, perspective, width, height);
|
|
|
|
/* Render the map model */
|
|
NSBMD model = mapFile.mapModel;
|
|
mapRenderer.Model = model.models[0];
|
|
|
|
// int scale = 64;
|
|
float scale = 0.015625f;
|
|
Gl.glScalef(mapRenderer.Model.modelScale * scale, mapRenderer.Model.modelScale * scale, mapRenderer.Model.modelScale * scale);
|
|
|
|
/* Determine if map textures must be rendered */
|
|
if (mapTexturesON) {
|
|
Gl.glEnable(Gl.GL_TEXTURE_2D);
|
|
} else {
|
|
Gl.glDisable(Gl.GL_TEXTURE_2D);
|
|
}
|
|
|
|
// Render map model
|
|
mapRenderer.RenderModel("", ani, aniframeS, aniframeS, aniframeS, aniframeS, aniframeS, ca, false, -1, 0.0f, 0.0f, dist, elev, ang, true, tp, model);
|
|
|
|
if (hideBuildings) {
|
|
return;
|
|
}
|
|
|
|
if (buildingTexturesON) {
|
|
Gl.glEnable(Gl.GL_TEXTURE_2D);
|
|
} else {
|
|
Gl.glDisable(Gl.GL_TEXTURE_2D);
|
|
}
|
|
|
|
for (int i = 0; i < mapFile.buildings.Count; i++) {
|
|
Building building = mapFile.buildings[i];
|
|
model = building.NSBMDFile;
|
|
if (model is null) {
|
|
Console.WriteLine("Null building can't be rendered");
|
|
} else {
|
|
mapRenderer.Model = model.models[0];
|
|
ScaleTranslateRotateBuilding(building);
|
|
mapRenderer.RenderModel("", ani, aniframeS, aniframeS, aniframeS, aniframeS, aniframeS, ca, false, -1, 0.0f, 0.0f, dist, elev, ang, true, tp, model);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static Bitmap GrabMapScreenshot(int width, int height) {
|
|
Bitmap bmp = new Bitmap(width, height);
|
|
System.Drawing.Imaging.BitmapData data = bmp.LockBits(new Rectangle(0, 0, width, height), System.Drawing.Imaging.ImageLockMode.WriteOnly, System.Drawing.Imaging.PixelFormat.Format24bppRgb);
|
|
Gl.glReadPixels(0, 0, width, height, Gl.GL_BGR, Gl.GL_UNSIGNED_BYTE, data.Scan0);
|
|
bmp.UnlockBits(data);
|
|
bmp.RotateFlip(RotateFlipType.RotateNoneFlipY);
|
|
return bmp;
|
|
}
|
|
|
|
private static void ScaleTranslateRotateBuilding(Building building) {
|
|
float fullXcoord = building.xPosition + building.xFraction / 65536f;
|
|
float fullYcoord = building.yPosition + building.yFraction / 65536f;
|
|
float fullZcoord = building.zPosition + building.zFraction / 65536f;
|
|
|
|
float scaleFactor = building.NSBMDFile.models[0].modelScale / 1024;
|
|
float translateFactor = 256 / building.NSBMDFile.models[0].modelScale;
|
|
|
|
Gl.glScalef(scaleFactor * building.width, scaleFactor * building.height, scaleFactor * building.length);
|
|
Gl.glTranslatef(fullXcoord * translateFactor / building.width, fullYcoord * translateFactor / building.height, fullZcoord * translateFactor / building.length);
|
|
Gl.glRotatef(Building.U16ToDeg(building.xRotation), 1, 0, 0);
|
|
Gl.glRotatef(Building.U16ToDeg(building.yRotation), 0, 1, 0);
|
|
Gl.glRotatef(Building.U16ToDeg(building.zRotation), 0, 0, 1);
|
|
}
|
|
|
|
public static Image GetPokePic(int species, int w, int h, PaletteBase paletteBase, ImageBase imageBase, SpriteBase spriteBase) {
|
|
bool fiveDigits = false; // some extreme future proofing
|
|
try {
|
|
string path = Filesystem.GetMonIconPath(0);
|
|
paletteBase = new NCLR(path, 0, Path.GetFileName(path));
|
|
} catch (FileNotFoundException) {
|
|
string path = Filesystem.GetMonIconPath(0, "D5");
|
|
paletteBase = new NCLR(path, 0, Path.GetFileName(path));
|
|
fiveDigits = true;
|
|
}
|
|
|
|
// read arm9 table to grab pal ID
|
|
int paletteId = 0;
|
|
byte[] iconPalTableBuf;
|
|
|
|
switch (RomInfo.gameFamily) {
|
|
case RomInfo.GameFamilies.DP:
|
|
iconPalTableBuf = ARM9.ReadBytes(0x6B838, 4);
|
|
break;
|
|
case RomInfo.GameFamilies.Plat:
|
|
iconPalTableBuf = ARM9.ReadBytes(0x79F80, 4);
|
|
break;
|
|
case RomInfo.GameFamilies.HGSS:
|
|
default:
|
|
iconPalTableBuf = ARM9.ReadBytes(0x74408, 4);
|
|
break;
|
|
}
|
|
|
|
int iconPalTableAddress = (iconPalTableBuf[3] & 0xFF) << 24 | (iconPalTableBuf[2] & 0xFF) << 16 | (iconPalTableBuf[1] & 0xFF) << 8 | (iconPalTableBuf[0] & 0xFF) /* << 0 */;
|
|
string iconTablePath;
|
|
|
|
int iconPalTableOffsetFromFileStart;
|
|
if (iconPalTableAddress >= RomInfo.synthOverlayLoadAddress) {
|
|
// if the pointer shows the table was moved to the synthetic overlay
|
|
iconPalTableOffsetFromFileStart = iconPalTableAddress - (int)RomInfo.synthOverlayLoadAddress;
|
|
iconTablePath = Filesystem.expArmPath;
|
|
} else {
|
|
iconPalTableOffsetFromFileStart = iconPalTableAddress - 0x02000000;
|
|
iconTablePath = RomInfo.arm9Path;
|
|
}
|
|
|
|
using (DSUtils.EasyReader idReader = new DSUtils.EasyReader(iconTablePath, iconPalTableOffsetFromFileStart + species)) {
|
|
paletteId = idReader.ReadByte();
|
|
}
|
|
|
|
if (paletteId != 0) {
|
|
paletteBase.Palette[0] = paletteBase.Palette[paletteId]; // update pal 0 to be the new pal
|
|
}
|
|
|
|
// grab tiles
|
|
int spriteFileID = species + 7;
|
|
if (fiveDigits) {
|
|
string path = Filesystem.GetMonIconPath(spriteFileID, "D5");
|
|
imageBase = new NCGR(path, spriteFileID, Path.GetFileName(path));
|
|
} else {
|
|
string path = Filesystem.GetMonIconPath(spriteFileID);
|
|
imageBase = new NCGR(path, spriteFileID, Path.GetFileName(path));
|
|
}
|
|
|
|
// grab sprite
|
|
const int ncerFileId = 2;
|
|
if (fiveDigits) {
|
|
string path = Filesystem.GetMonIconPath(ncerFileId, "D5");
|
|
spriteBase = new NCER(path, ncerFileId, Path.GetFileName(path));
|
|
} else {
|
|
string path = Filesystem.GetMonIconPath(ncerFileId);
|
|
spriteBase = new NCER(path, ncerFileId, Path.GetFileName(path));
|
|
}
|
|
|
|
// copy this from the trainer
|
|
int bank0OAMcount = spriteBase.Banks[0].oams.Length;
|
|
int[] OAMenabled = new int[bank0OAMcount];
|
|
for (int i = 0; i < OAMenabled.Length; i++) {
|
|
OAMenabled[i] = i;
|
|
}
|
|
|
|
// finally compose image
|
|
try {
|
|
return spriteBase.Get_Image(imageBase, paletteBase, 0, w, h, false, false, false, true, true, -1, OAMenabled);
|
|
} catch (FormatException) {
|
|
return Properties.Resources.IconPokeball;
|
|
}
|
|
// default:
|
|
//partyPokemonPictureBoxList[partyPos].Image = cb.SelectedIndex > 0 ? (Image)Properties.PokePics.ResourceManager.GetObject(FixPokenameString(PokeDatabase.System.pokeNames[(ushort)cb.SelectedIndex])) : global::DSPRE.Properties.Resources.IconPokeball;
|
|
}
|
|
|
|
public static void GenerateKeystrokes(string keys, Scintilla textArea) {
|
|
//Example
|
|
//GenerateKeystrokes("+{TAB}");
|
|
HotKeyManager.Enable = false;
|
|
textArea.Focus();
|
|
SendKeys.Send(keys);
|
|
HotKeyManager.Enable = true;
|
|
}
|
|
|
|
public static void PictureBoxDisable(object sender, PaintEventArgs e) {
|
|
if (sender is PictureBox pict && pict.Image != null && (!pict.Enabled)) {
|
|
using (Bitmap img = new Bitmap(pict.Image, pict.ClientSize)) {
|
|
ControlPaint.DrawImageDisabled(e.Graphics, img, 0, 0, pict.BackColor);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static List<string> getHeaderListBoxNames() {
|
|
if (string.IsNullOrWhiteSpace(RomInfo.internalNamesPath)) {
|
|
return null;
|
|
}
|
|
|
|
List<string> headerListBoxNames = new List<string>();
|
|
|
|
using (DSUtils.EasyReader reader = new DSUtils.EasyReader(RomInfo.internalNamesPath)) {
|
|
int headerCount = RomInfo.GetHeaderCount();
|
|
for (int i = 0; i < headerCount; i++) {
|
|
byte[] row = reader.ReadBytes(RomInfo.internalNameLength);
|
|
string internalName = Encoding.ASCII.GetString(row); //.TrimEnd();
|
|
headerListBoxNames.Add(MapHeader.BuildName(i, internalName));
|
|
}
|
|
}
|
|
|
|
return headerListBoxNames;
|
|
}
|
|
|
|
public static List<string> getInternalNames() {
|
|
List<string> internalNames = new List<string>();
|
|
|
|
using (DSUtils.EasyReader reader = new DSUtils.EasyReader(RomInfo.internalNamesPath)) {
|
|
int headerCount = RomInfo.GetHeaderCount();
|
|
for (int i = 0; i < headerCount; i++) {
|
|
byte[] row = reader.ReadBytes(RomInfo.internalNameLength);
|
|
string internalName = Encoding.ASCII.GetString(row); //.TrimEnd();
|
|
internalNames.Add(internalName.TrimEnd('\0'));
|
|
}
|
|
}
|
|
|
|
return internalNames;
|
|
}
|
|
|
|
public static int CalculateTimeDifferenceInSeconds(int startHour, int startMinute, int startSecond, int endHour, int endMinute, int endSecond)
|
|
{
|
|
// Convert start time and end time to seconds since midnight
|
|
int startTimeInSeconds = (startHour * 3600) + (startMinute * 60) + startSecond;
|
|
int endTimeInSeconds = (endHour * 3600) + (endMinute * 60) + endSecond;
|
|
|
|
// Calculate difference
|
|
int timeDifference = endTimeInSeconds - startTimeInSeconds;
|
|
|
|
// If time difference is negative (end time is past midnight), adjust
|
|
if (timeDifference < 0)
|
|
{
|
|
timeDifference += 24 * 3600; // Add 24 hours in seconds
|
|
}
|
|
|
|
return timeDifference;
|
|
}
|
|
|
|
public static String formatTime(int time)
|
|
{
|
|
string stringTime = time.ToString();
|
|
if (time < 10)
|
|
{
|
|
stringTime = "0" + stringTime;
|
|
}
|
|
|
|
return stringTime;
|
|
}
|
|
|
|
[System.Runtime.InteropServices.DllImport("user32.dll", CharSet = System.Runtime.InteropServices.CharSet.Auto)]
|
|
extern static bool DestroyIcon(IntPtr handle);
|
|
|
|
|
|
public static void PopOutEditor<T>(T control, string title, Image icon, Action<T> onClose = null) where T : Control
|
|
{
|
|
var originalParent = control.Parent;
|
|
var originalIndex = originalParent?.Controls.IndexOf(control) ?? -1;
|
|
|
|
originalParent?.Controls.Remove(control);
|
|
Icon _icon = null;
|
|
if(icon != null)
|
|
{
|
|
Bitmap bitmap = new Bitmap(icon);
|
|
_icon = Icon.FromHandle(bitmap.GetHicon());
|
|
}
|
|
|
|
|
|
var form = new Form
|
|
{
|
|
Text = title,
|
|
StartPosition = FormStartPosition.CenterScreen,
|
|
FormBorderStyle = FormBorderStyle.FixedSingle,
|
|
MaximizeBox = false,
|
|
ClientSize = control.Size,
|
|
ShowIcon = icon == null ? false : true,
|
|
Icon = _icon
|
|
|
|
};
|
|
|
|
control.Dock = DockStyle.Fill;
|
|
form.Controls.Add(control);
|
|
|
|
form.FormClosing += (s, e) =>
|
|
{
|
|
form.Controls.Remove(control);
|
|
if(_icon != null)
|
|
{
|
|
DestroyIcon(_icon.Handle);
|
|
}
|
|
if (originalParent != null && !originalParent.IsDisposed)
|
|
{
|
|
if (originalIndex >= 0 && originalIndex <= originalParent.Controls.Count)
|
|
originalParent.Controls.Add(control);
|
|
else
|
|
originalParent.Controls.Add(control);
|
|
|
|
originalParent.Controls.SetChildIndex(control, originalIndex);
|
|
}
|
|
|
|
onClose?.Invoke(control);
|
|
};
|
|
|
|
form.Show();
|
|
}
|
|
|
|
public static void ExclusiveCBInvert(CheckBox cb)
|
|
{
|
|
if (Helpers.HandlersDisabled)
|
|
{
|
|
return;
|
|
}
|
|
|
|
Helpers.DisableHandlers();
|
|
|
|
if (cb.Checked)
|
|
{
|
|
cb.Checked = !cb.Checked;
|
|
}
|
|
|
|
Helpers.EnableHandlers();
|
|
}
|
|
|
|
public static void ContentBasedBatchRename(MainProgram parent, DirectoryInfo d = null)
|
|
{
|
|
(DirectoryInfo d, FileInfo[] files) dirData = OpenNonEmptyDir(d, title: "Content-Based Batch Rename Tool");
|
|
d = dirData.d;
|
|
FileInfo[] files = dirData.files;
|
|
|
|
if (d == null || files == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
DialogResult dr = MessageBox.Show("About to rename " + files.Length + " file" + (files.Length > 1 ? "s" : "") +
|
|
" from the input folder (taken in ascending order), according to their content.\n" +
|
|
"If a destination file already exists, DSPRE will append a number to its name.\n\n" +
|
|
"Do you want to proceed?", "Confirm operation", MessageBoxButtons.YesNo, MessageBoxIcon.Question);
|
|
|
|
if (dr.Equals(DialogResult.Yes))
|
|
{
|
|
List<string> enumerationFile = new List<string> {
|
|
"#============================================================================",
|
|
"# File enumeration definition for folder " + "\"" + d.Name + "\"",
|
|
"#============================================================================"
|
|
};
|
|
int initialLength = enumerationFile.Count;
|
|
|
|
const byte toRead = 16;
|
|
foreach (FileInfo f in files)
|
|
{
|
|
Console.WriteLine(f.Name);
|
|
|
|
string fileNameOnly = Path.GetFileNameWithoutExtension(f.FullName);
|
|
string dirNameOnly = Path.GetDirectoryName(f.FullName);
|
|
|
|
string destName = "";
|
|
byte[] b = DSUtils.ReadFromFile(f.FullName, 0, toRead);
|
|
|
|
if (b == null || b.Length < toRead)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
string magic = "";
|
|
|
|
if (b[0] == 'B' && b[3] == '0')
|
|
{ //B**0
|
|
ushort nameOffset;
|
|
|
|
destName = dirNameOnly + "\\"; //Full filename can be changed
|
|
nameOffset = (ushort)(52 + (4 * (BitConverter.ToUInt16(b, 0xE) - 1)));
|
|
|
|
if (b[1] == 'T' && b[2] == 'X')
|
|
{ //BTX0
|
|
#if false
|
|
nameOffset += 0xEC;
|
|
#else
|
|
destName = fileNameOnly;
|
|
#endif
|
|
}
|
|
|
|
string nameRead = Encoding.UTF8.GetString(DSUtils.ReadFromFile(f.FullName, nameOffset, 16)).TrimEnd(new char[] { (char)0 });
|
|
|
|
if (nameRead.Length <= 0 || nameRead.IndexOfAny(Path.GetInvalidPathChars()) >= 0)
|
|
{
|
|
destName = fileNameOnly; //Filename can't be changed, only extension
|
|
}
|
|
else
|
|
{
|
|
destName += nameRead;
|
|
}
|
|
|
|
destName += ".ns";
|
|
for (int i = 0; i < 3; i++)
|
|
{
|
|
magic += Char.ToLower((char)b[i]);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
destName = fileNameOnly + ".";
|
|
byte offset = 0;
|
|
|
|
if (b[5] == 'R' && b[8] == 'N')
|
|
{
|
|
offset = 5;
|
|
}
|
|
|
|
for (int i = 0; i < 4; i++)
|
|
{
|
|
magic += Char.ToLower((char)b[offset + i]);
|
|
}
|
|
}
|
|
|
|
if (string.IsNullOrWhiteSpace(magic) || !magic.All(char.IsLetterOrDigit))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
destName += magic;
|
|
|
|
if (string.IsNullOrWhiteSpace(destName))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
destName = MakeUniqueName(destName, fileNameOnly = null, dirNameOnly);
|
|
System.IO.File.Move(f.FullName, Path.Combine(Path.GetDirectoryName(f.FullName), Path.GetFileName(destName)));
|
|
|
|
enumerationFile.Add(Path.GetFileName(destName));
|
|
}
|
|
|
|
if (enumerationFile.Count > initialLength)
|
|
{
|
|
MessageBox.Show("Files inside folder \"" + d.FullName + "\" have been renamed according to their contents.", "Success", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
|
|
|
DialogResult response = MessageBox.Show("Do you want to save a file enumeration list?", "Waiting for user", MessageBoxButtons.YesNo, MessageBoxIcon.Question);
|
|
if (response.Equals(DialogResult.Yes))
|
|
{
|
|
MessageBox.Show("Choose where to save the output list file.", "Name your list file", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
|
|
|
SaveFileDialog sf = new SaveFileDialog
|
|
{
|
|
Filter = "List File (*.txt; *.list)|*.txt;*.list",
|
|
FileName = d.Name + ".list"
|
|
};
|
|
if (sf.ShowDialog(parent) != DialogResult.OK)
|
|
{
|
|
return;
|
|
}
|
|
|
|
System.IO.File.WriteAllLines(sf.FileName, enumerationFile);
|
|
MessageBox.Show("List file saved.", "Success", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
MessageBox.Show("No file content could be recognized.", "Operation terminated", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static (DirectoryInfo, FileInfo[]) OpenNonEmptyDir(DirectoryInfo d = null, string title = "Waiting for user")
|
|
{
|
|
/*==================================================================*/
|
|
if (d == null)
|
|
{
|
|
MessageBox.Show("Choose a source folder.", title, MessageBoxButtons.OK, MessageBoxIcon.Information);
|
|
CommonOpenFileDialog sourceDirDialog = new CommonOpenFileDialog
|
|
{
|
|
IsFolderPicker = true,
|
|
Multiselect = false
|
|
};
|
|
|
|
if (sourceDirDialog.ShowDialog() != CommonFileDialogResult.Ok)
|
|
{
|
|
return (null, null);
|
|
}
|
|
|
|
d = new DirectoryInfo(sourceDirDialog.FileName);
|
|
}
|
|
|
|
FileInfo[] tempfiles = d.GetFiles();
|
|
FileInfo[] files = tempfiles.OrderBy(n => System.Text.RegularExpressions.Regex.Replace(n.Name, @"\d+", e => e.Value.PadLeft(tempfiles.Length.ToString().Length, '0'))).ToArray();
|
|
|
|
if (files.Length <= 0)
|
|
{
|
|
MessageBox.Show("Folder " + "\"" + d.FullName + "\"" + " is empty.\nCan't proceed.", "Invalid folder", MessageBoxButtons.OK, MessageBoxIcon.Error);
|
|
return (null, null);
|
|
};
|
|
|
|
return (d, files);
|
|
}
|
|
|
|
public static string MakeUniqueName(string fileName, string fileNameOnly = null, string dirNameOnly = null, string extension = null)
|
|
{
|
|
if (fileNameOnly == null)
|
|
{
|
|
fileNameOnly = Path.GetFileNameWithoutExtension(fileName);
|
|
}
|
|
if (dirNameOnly == null)
|
|
{
|
|
dirNameOnly = Path.GetDirectoryName(fileName);
|
|
}
|
|
if (extension == null)
|
|
{
|
|
extension = Path.GetExtension(fileName);
|
|
}
|
|
|
|
int append = 1;
|
|
|
|
while (System.IO.File.Exists(Path.Combine(dirNameOnly, fileName)))
|
|
{
|
|
string tmp = fileNameOnly + "(" + (append++) + ")";
|
|
fileName = Path.Combine(dirNameOnly, tmp + extension);
|
|
}
|
|
return fileName;
|
|
}
|
|
|
|
public static void ExportTrainerUsageToCSV(Dictionary<string, Dictionary<string, int>> trainerUsage, string csvFilePath)
|
|
{
|
|
// Create the StreamWriter to write data to the CSV file
|
|
var sortedTrainerClasses = trainerUsage.Keys.OrderBy(className => className);
|
|
|
|
using (StreamWriter sw = new StreamWriter(csvFilePath))
|
|
{
|
|
// Write the header row
|
|
sw.WriteLine("Trainer Class;Pokemon Name;Occurrences");
|
|
|
|
// Iterate over the sorted trainer class names
|
|
foreach (string className in sortedTrainerClasses)
|
|
{
|
|
Dictionary<string, int> innerDict = trainerUsage[className];
|
|
|
|
// Sort the Pokemon names alphabetically
|
|
var sortedPokemonNames = innerDict.Keys.OrderByDescending(pokeName => innerDict[pokeName]);
|
|
|
|
// Iterate over the sorted mon names
|
|
foreach (string pokeName in sortedPokemonNames)
|
|
{
|
|
int occurrences = innerDict[pokeName];
|
|
|
|
// Write the data row
|
|
sw.WriteLine($"{className};{pokeName};{occurrences}");
|
|
}
|
|
sw.WriteLine($"-;-;-");
|
|
}
|
|
}
|
|
|
|
Console.WriteLine("CSV file exported successfully.");
|
|
}
|
|
|
|
}
|
|
} |