mirror of
https://github.com/AdAstra-LD/DS-Pokemon-Rom-Editor.git
synced 2026-05-09 21:50:57 -05:00
337 lines
16 KiB
C#
337 lines
16 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Text;
|
|
using System.Resources;
|
|
using System.Reflection;
|
|
using System.Windows.Forms;
|
|
using DSPRE.Resources;
|
|
using static DSPRE.RomInfo;
|
|
|
|
namespace DSPRE.ROMFiles {
|
|
/// <summary>
|
|
/// Class to store message data from DS Pokémon games
|
|
/// </summary>
|
|
public class TextArchive : RomFile {
|
|
#region Fields (2)
|
|
public List<string> messages;
|
|
public int initialKey;
|
|
#endregion Fields
|
|
|
|
#region Constructors (1)
|
|
public TextArchive(FileStream messageStream, List<string> msg, bool discardLines = false) {
|
|
Dictionary<int, string> GetCharDictionary = TextDatabase.readTextDictionary;
|
|
BinaryReader readText = new BinaryReader(messageStream);
|
|
int stringCount;
|
|
|
|
try {
|
|
stringCount = readText.ReadUInt16();
|
|
} catch (EndOfStreamException) {
|
|
MessageBox.Show("Error loading text file.\n", "Unexpected EOF", MessageBoxButtons.OK, MessageBoxIcon.Error);
|
|
readText.Close();
|
|
return;
|
|
}
|
|
|
|
if (msg == null) {
|
|
messages = new List<string>();
|
|
} else {
|
|
messages = msg;
|
|
}
|
|
|
|
initialKey = readText.ReadUInt16();
|
|
|
|
if (!discardLines) {
|
|
int key1 = (initialKey * 0x2FD) & 0xFFFF;
|
|
bool specialCharON = false;
|
|
int[] currentOffset = new int[stringCount];
|
|
int[] currentSize = new int[stringCount];
|
|
bool compressed = new bool();
|
|
|
|
for (int i = 0; i < stringCount; i++) { // Reads and stores string offsets and sizes
|
|
int key2 = (key1 * (i + 1) & 0xFFFF);
|
|
int realKey = key2 | (key2 << 16);
|
|
currentOffset[i] = ((int)readText.ReadUInt32()) ^ realKey;
|
|
currentSize[i] = ((int)readText.ReadUInt32()) ^ realKey;
|
|
}
|
|
|
|
for (int i = 0; i < stringCount; i++) { // Adds new string
|
|
key1 = (0x91BD3 * (i + 1)) & 0xFFFF;
|
|
readText.BaseStream.Position = currentOffset[i];
|
|
StringBuilder pokemonText = new StringBuilder("");
|
|
|
|
for (int j = 0; j < currentSize[i]; j++) // Adds new characters to string
|
|
{
|
|
int car = (readText.ReadUInt16()) ^ key1;
|
|
|
|
switch (car) { // Special characters
|
|
case 0xE000:
|
|
pokemonText.Append(@"\n");
|
|
break;
|
|
case 0x25BC:
|
|
pokemonText.Append(@"\r");
|
|
break;
|
|
case 0x25BD:
|
|
pokemonText.Append(@"\f");
|
|
break;
|
|
case 0xF100:
|
|
compressed = true;
|
|
break;
|
|
case 0xFFFE:
|
|
pokemonText.Append(@"\v");
|
|
specialCharON = true;
|
|
break;
|
|
case 0xFFFF:
|
|
pokemonText.Append("");
|
|
break;
|
|
default:
|
|
if (specialCharON) {
|
|
pokemonText.Append(car.ToString("X4"));
|
|
specialCharON = false;
|
|
} else if (compressed) {
|
|
#region Compressed String
|
|
int shift = 0;
|
|
int trans = 0;
|
|
string uncomp = "";
|
|
while (true) {
|
|
int tmp = car >> shift;
|
|
int tmp1 = tmp;
|
|
if (shift >= 0xF) {
|
|
shift -= 0xF;
|
|
if (shift > 0) {
|
|
tmp1 = (trans | ((car << (9 - shift)) & 0x1FF));
|
|
if ((tmp1 & 0xFF) == 0xFF) {
|
|
break;
|
|
}
|
|
if (tmp1 != 0x0 && tmp1 != 0x1) {
|
|
string character = "";
|
|
if (!GetCharDictionary.TryGetValue(tmp1, out character)) {
|
|
pokemonText.Append(@"\x" + tmp1.ToString("X4"));
|
|
} else {
|
|
pokemonText.Append(character);
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
tmp1 = ((car >> shift) & 0x1FF);
|
|
if ((tmp1 & 0xFF) == 0xFF) {
|
|
break;
|
|
}
|
|
if (tmp1 != 0x0 && tmp1 != 0x1) {
|
|
string character = "";
|
|
if (!GetCharDictionary.TryGetValue(tmp1, out character))
|
|
pokemonText.Append(@"\x" + tmp1.ToString("X4"));
|
|
else {
|
|
pokemonText.Append(character);
|
|
}
|
|
}
|
|
shift += 9;
|
|
if (shift < 0xF) {
|
|
trans = ((car >> shift) & 0x1FF);
|
|
shift += 9;
|
|
}
|
|
key1 += 0x493D;
|
|
key1 &= 0xFFFF;
|
|
car = Convert.ToUInt16(readText.ReadUInt16() ^ key1);
|
|
j++;
|
|
}
|
|
}
|
|
#endregion
|
|
pokemonText.Append(uncomp);
|
|
} else {
|
|
if (GetCharDictionary.TryGetValue(car, out string character)) {
|
|
pokemonText.Append(character);
|
|
} else {
|
|
pokemonText.Append(@"\x" + car.ToString("X4"));
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
key1 += 0x493D;
|
|
key1 &= 0xFFFF;
|
|
}
|
|
messages.Add(pokemonText.ToString());
|
|
}
|
|
}
|
|
|
|
readText.Dispose();
|
|
}
|
|
public TextArchive(int ID, List<string> msg = null, bool discardLines = false) : this(new FileStream(RomInfo.gameDirs[DirNames.textArchives].unpackedDir + "\\" + ID.ToString("D4"), FileMode.Open), msg, discardLines) {
|
|
}
|
|
#endregion
|
|
|
|
#region Methods (2)
|
|
public int[] EncodeString(string currentMessage, int stringIndex, int stringSize) { // Converts string to hex characters
|
|
ResourceManager GetByte = new ResourceManager("DSPRE.Resources.WriteText", Assembly.GetExecutingAssembly());
|
|
|
|
int[] pokemonMessage = new int[stringSize - 1];
|
|
var charArray = currentMessage.ToCharArray();
|
|
int count = 0;
|
|
try {
|
|
for (int i = 0; i < currentMessage.Length; i++) {
|
|
if (charArray[i] == '\\') {
|
|
if (charArray[i + 1] == 'r') {
|
|
pokemonMessage[count] = 0x25BC;
|
|
i++;
|
|
} else {
|
|
if (charArray[i + 1] == 'n') {
|
|
pokemonMessage[count] = 0xE000;
|
|
i++;
|
|
} else {
|
|
if (charArray[i + 1] == 'f') {
|
|
pokemonMessage[count] = 0x25BD;
|
|
i++;
|
|
} else {
|
|
if (charArray[i + 1] == 'v') {
|
|
pokemonMessage[count] = 0xFFFE;
|
|
count++;
|
|
string characterID = ((char)charArray[i + 2]).ToString() + ((char)charArray[i + 3]).ToString() + ((char)charArray[i + 4]).ToString() + ((char)charArray[i + 5]).ToString();
|
|
pokemonMessage[count] = (int)Convert.ToUInt32(characterID, 16);
|
|
i += 5;
|
|
} else {
|
|
//This looks like it can be optimized
|
|
if (charArray[i + 1] == 'x' && charArray[i + 2] == '0' && charArray[i + 3] == '0' && charArray[i + 4] == '0' && charArray[i + 5] == '0') {
|
|
pokemonMessage[count] = 0x0000;
|
|
i += 5;
|
|
} else {
|
|
if (charArray[i + 1] == 'x' && charArray[i + 2] == '0' && charArray[i + 3] == '0' && charArray[i + 4] == '0' && charArray[i + 5] == '1') {
|
|
pokemonMessage[count] = 0x0001;
|
|
i += 5;
|
|
} else {
|
|
string characterID = ((char)charArray[i + 2]).ToString() + ((char)charArray[i + 3]).ToString() + ((char)charArray[i + 4]).ToString() + ((char)charArray[i + 5]).ToString();
|
|
pokemonMessage[count] = (int)Convert.ToUInt32(characterID, 16);
|
|
i += 5;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
if (charArray[i] == '[') {
|
|
if (charArray[i + 1] == 'P') {
|
|
pokemonMessage[count] = 0x01E0;
|
|
i += 3;
|
|
}
|
|
if (charArray[i + 1] == 'M') {
|
|
pokemonMessage[count] = 0x01E1;
|
|
i += 3;
|
|
}
|
|
} else {
|
|
pokemonMessage[count] = (int)Convert.ToUInt32(GetByte.GetString(((int)charArray[i]).ToString()), 16);
|
|
}
|
|
}
|
|
count++;
|
|
}
|
|
} catch (FormatException) {
|
|
MessageBox.Show("Format exception. Assembled so far: " + Environment.NewLine + pokemonMessage);
|
|
}
|
|
return pokemonMessage;
|
|
}
|
|
public int GetStringLength(string currentMessage) { // Calculates string length
|
|
int count = 0;
|
|
var charArray = currentMessage.ToCharArray();
|
|
for (int i = 0; i < currentMessage.Length; i++) {
|
|
if (charArray[i] == '\\') {
|
|
if (charArray[i + 1] == 'r') {
|
|
count++;
|
|
i++;
|
|
} else {
|
|
if (charArray[i + 1] == 'n') {
|
|
count++;
|
|
i++;
|
|
} else {
|
|
if (charArray[i + 1] == 'f') {
|
|
count++;
|
|
i++;
|
|
} else {
|
|
if (charArray[i + 1] == 'v') {
|
|
count += 2;
|
|
i += 5;
|
|
} else {
|
|
if (charArray[i + 1] == 'x' && charArray[i + 2] == '0' && charArray[i + 3] == '0' && charArray[i + 4] == '0' && charArray[i + 5] == '0') {
|
|
count++;
|
|
i += 5;
|
|
} else {
|
|
if (charArray[i + 1] == 'x' && charArray[i + 2] == '0' && charArray[i + 3] == '0' && charArray[i + 4] == '0' && charArray[i + 5] == '1') {
|
|
count++;
|
|
i += 5;
|
|
} else {
|
|
count++;
|
|
i += 5;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
if (charArray[i] == '[') {
|
|
if (charArray[i + 1] == 'P') {
|
|
count++;
|
|
i += 3;
|
|
}
|
|
if (charArray[i + 1] == 'M') {
|
|
count++;
|
|
i += 3;
|
|
}
|
|
} else {
|
|
count++;
|
|
}
|
|
}
|
|
}
|
|
count++;
|
|
return count;
|
|
}
|
|
private byte[] ToByteArray(List<string> msgSource) {
|
|
MemoryStream newData = new MemoryStream();
|
|
using (BinaryWriter writer = new BinaryWriter(newData)) {
|
|
writer.Write((ushort)msgSource.Count);
|
|
writer.Write((ushort)initialKey);
|
|
|
|
int key = (initialKey * 0x2FD) & 0xFFFF;
|
|
int key2 = 0;
|
|
int realKey = 0;
|
|
int offset = 0x4 + (msgSource.Count * 8);
|
|
int[] stringSize = new int[msgSource.Count];
|
|
|
|
for (int i = 0; i < msgSource.Count; i++) { // Reads and stores string offsets and sizes
|
|
key2 = (key * (i + 1) & 0xFFFF);
|
|
realKey = key2 | (key2 << 16);
|
|
writer.Write(offset ^ realKey);
|
|
int length = GetStringLength(msgSource[i]);
|
|
stringSize[i] = length;
|
|
writer.Write(length ^ realKey);
|
|
offset += length * 2;
|
|
}
|
|
|
|
for (int i = 0; i < msgSource.Count; i++) { // Encodes strings and writes them to file
|
|
key = (0x91BD3 * (i + 1)) & 0xFFFF;
|
|
int[] currentString = EncodeString(msgSource[i], i, stringSize[i]);
|
|
|
|
for (int j = 0; j < stringSize[i] - 1; j++) {
|
|
writer.Write((ushort)(currentString[j] ^ key));
|
|
key += 0x493D;
|
|
key &= 0xFFFF;
|
|
}
|
|
writer.Write((ushort)(0xFFFF ^ key));
|
|
}
|
|
}
|
|
return newData.ToArray();
|
|
}
|
|
|
|
public override string ToString() {
|
|
return string.Join(Environment.NewLine, messages);
|
|
}
|
|
|
|
public override byte[] ToByteArray() {
|
|
return this.ToByteArray(messages);
|
|
}
|
|
public void SaveToFileDefaultDir(int IDtoReplace, bool showSuccessMessage = true) {
|
|
SaveToFileDefaultDir(DirNames.textArchives, IDtoReplace, showSuccessMessage);
|
|
}
|
|
public void SaveToFileExplorePath(string suggestedFileName, bool showSuccessMessage = true) {
|
|
SaveToFileExplorePath("Gen IV Text Archive", "msg", suggestedFileName, showSuccessMessage);
|
|
}
|
|
#endregion
|
|
}
|
|
} |