mirror of
https://git.huckle.dev/Huckles-Minecraft-Archive/PCK-Studio.git
synced 2026-05-22 16:16:30 +00:00
369 lines
18 KiB
C#
369 lines
18 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Drawing;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Text.RegularExpressions;
|
|
using System.Threading.Tasks;
|
|
using System.Windows.Documents;
|
|
using OMI;
|
|
using OMI.Formats.GameRule;
|
|
using OMI.Formats.Languages;
|
|
using OMI.Formats.Pck;
|
|
using OMI.Workers.GameRule;
|
|
using OMI.Workers.Language;
|
|
using OMI.Workers.Pck;
|
|
using PckStudio.Core.Deserializer;
|
|
using PckStudio.Core.Extensions;
|
|
using PckStudio.Core.Interfaces;
|
|
|
|
namespace PckStudio.Core.DLC
|
|
{
|
|
public sealed class DLCManager
|
|
{
|
|
public static DLCManager Default { get; } = new DLCManager(default, AppLanguage.SystemDefault);
|
|
|
|
internal const string DEFAULT_TEXTURE_PACK_FILENAME = "TexturePack.pck";
|
|
internal const string DEFAULT_MINIGAME_PACK_FILENAME = "WorldPack.pck";
|
|
internal const string DATA_DIRECTORY_NAME = "Data";
|
|
internal const string PACKAGE_DISPLAYNAME_ID = "IDS_DISPLAY_NAME";
|
|
|
|
public ByteOrder ByteOrder => _byteOrder;
|
|
|
|
public ConsolePlatform Platform => _platform;
|
|
|
|
/// <summary>
|
|
/// See <see cref="AvailableLanguages"/> for details.
|
|
/// </summary>
|
|
public string PreferredLanguage { get; private set; }
|
|
|
|
private readonly DLCPackageRegistry _packageRegistry = new DLCPackageRegistry();
|
|
private readonly Random _rng = new Random();
|
|
private ByteOrder _byteOrder;
|
|
private ConsolePlatform _platform;
|
|
|
|
|
|
/// <param name="byteOrder"></param>
|
|
/// <param name="platform"></param>
|
|
/// <param name="preferredLanguage">See <see cref="AvailableLanguages"/> for details.</param>
|
|
public DLCManager(ConsolePlatform platform, AppLanguage preferredLanguage)
|
|
{
|
|
_platform = platform;
|
|
_byteOrder = GetByteOrderForPlatform(Platform);
|
|
PreferredLanguage = GetPreferredLanguage(preferredLanguage);
|
|
}
|
|
|
|
private static ByteOrder GetByteOrderForPlatform(ConsolePlatform platform)
|
|
{
|
|
return platform switch
|
|
{
|
|
ConsolePlatform.Switch => ByteOrder.LittleEndian,
|
|
ConsolePlatform.PS4 => ByteOrder.LittleEndian,
|
|
_ => ByteOrder.BigEndian
|
|
};
|
|
}
|
|
|
|
public IDLCPackage CreateNewPackage(string name, DLCPackageType packageType)
|
|
{
|
|
int identifier = _rng.Next(8000, GameConstants.MAX_PACK_ID);
|
|
IDLCPackage package = packageType switch
|
|
{
|
|
DLCPackageType.SkinPack => DLCSkinPackage.CreateEmpty(name, identifier),
|
|
DLCPackageType.TexturePack => DLCTexturePackage.CreateDefaultPackage(name, "", identifier),
|
|
DLCPackageType.MashUpPack => new DLCMashUpPackage(name, "", identifier),
|
|
//! TODO: implemnt minigame dlc packages -null
|
|
DLCPackageType.MG01 => new DLCBattlePackage(name, identifier),
|
|
DLCPackageType.MG02 => new DLCMiniGamePackage(name, identifier, packageType, MiniGameId.Tumble),
|
|
DLCPackageType.MG03 => new DLCMiniGamePackage(name, identifier, packageType, MiniGameId.Glide),
|
|
DLCPackageType.Invalid => InvalidDLCPackage.Instance,
|
|
_ => throw new ArgumentException("Unable to create DLC Package of 'Unknown' type."),
|
|
};
|
|
|
|
LOCFile localisation = new LOCFile();
|
|
localisation.AddLanguage(PreferredLanguage);
|
|
localisation.AddLocKey(PACKAGE_DISPLAYNAME_ID, name);
|
|
_packageRegistry.RegisterPackage(identifier, package, localisation);
|
|
|
|
return package;
|
|
}
|
|
|
|
public IDLCPackage OpenDLCPackage(string filepath) => OpenDLCPackage(new FileInfo(filepath));
|
|
|
|
public IDLCPackage OpenDLCPackage(FileInfo fileInfo)
|
|
{
|
|
if (Platform == ConsolePlatform.Unknown)
|
|
throw new Exception($"{nameof(Platform)} is Unknown.");
|
|
|
|
if (!IsValidPckFile(fileInfo))
|
|
return InvalidDLCPackage.Instance;
|
|
|
|
using Stream stream = fileInfo.OpenRead();
|
|
|
|
PckFileReader fileReader = new PckFileReader(ByteOrder);
|
|
|
|
PckFile pckFile = fileReader.FromStream(stream);
|
|
|
|
if (!pckFile.TryGetAsset("0", PckAssetType.InfoFile, out PckAsset zeroAsset))
|
|
{
|
|
Trace.TraceError("Could not find asset named:'0'.");
|
|
return new UnknownDLCPackage(fileInfo.Name, pckFile);
|
|
}
|
|
|
|
int identifier = (zeroAsset?.HasProperty("PACKID") ?? default) ? zeroAsset.GetProperty("PACKID", int.Parse) : -1;
|
|
if (identifier <= 0 || identifier > GameConstants.MAX_PACK_ID)
|
|
{
|
|
Trace.TraceError($"{nameof(identifier)}({identifier}) was out of range!");
|
|
return new UnknownDLCPackage(fileInfo.Name, pckFile);
|
|
}
|
|
|
|
if (_packageRegistry.ContainsPackage(identifier))
|
|
return _packageRegistry[identifier];
|
|
|
|
LOCFile localisation = pckFile.GetAssetsByType(PckAssetType.LocalisationFile).FirstOrDefault()?.GetData(new LOCFileReader());
|
|
if (localisation is null)
|
|
return new UnknownDLCPackage(fileInfo.Name, pckFile);
|
|
|
|
IDLCPackage package = ScanForPackageType(fileInfo, identifier, pckFile, localisation, fileReader);
|
|
if (package.GetDLCPackageType() != DLCPackageType.Invalid)
|
|
{
|
|
_packageRegistry.RegisterPackage(identifier, package, localisation);
|
|
}
|
|
return package;
|
|
}
|
|
|
|
internal LOCFile GetLocalisation(int identifier)
|
|
{
|
|
return _packageRegistry.ContainsPackage(identifier) ? _packageRegistry.GetLocalisation(identifier) : default;
|
|
}
|
|
|
|
private bool IsValidPckFile(FileInfo fileInfo)
|
|
{
|
|
if (fileInfo is null)
|
|
throw new ArgumentNullException(nameof(fileInfo));
|
|
|
|
if (!fileInfo.Exists)
|
|
throw new FileNotFoundException(fileInfo.FullName);
|
|
|
|
if (fileInfo.Extension != ".pck")
|
|
throw new FileFormatException("File does not end with '.pck'.");
|
|
return true;
|
|
}
|
|
|
|
private IDLCPackage ScanForPackageType(FileInfo fileInfo, int identifier, PckFile pckFile, LOCFile localisation, PckFileReader fileReader)
|
|
{
|
|
bool hasLanguage = localisation?.Languages?.Contains(PreferredLanguage) ?? default;
|
|
|
|
string name = hasLanguage && (localisation?.HasLocEntry(PACKAGE_DISPLAYNAME_ID) ?? default)
|
|
? localisation.GetLocEntry(PACKAGE_DISPLAYNAME_ID, PreferredLanguage) : fileInfo.Name;
|
|
|
|
string description = hasLanguage && (localisation?.HasLocEntry(DLCTexturePackage.TEXTUREPACK_DESCRIPTION_ID) ?? default)
|
|
? localisation.GetLocEntry(DLCTexturePackage.TEXTUREPACK_DESCRIPTION_ID, PreferredLanguage) : string.Empty;
|
|
|
|
bool couldBeTexturePack = fileInfo.Name == DEFAULT_TEXTURE_PACK_FILENAME;
|
|
bool couldBeMiniGamePack = fileInfo.Name == DEFAULT_MINIGAME_PACK_FILENAME;
|
|
|
|
bool hasSkins = pckFile.Contains(PckAssetType.SkinFile) || pckFile.Contains(PckAssetType.SkinDataFile);
|
|
|
|
DLCPackageType dlcPackageType = hasSkins ? DLCPackageType.SkinPack : DLCPackageType.Unknown;
|
|
|
|
DirectoryInfo dataDirectoryInfo = fileInfo.Directory.EnumerateDirectories().Where(dirInfo => dirInfo.Name == DATA_DIRECTORY_NAME).FirstOrDefault();
|
|
|
|
PckAsset texturePackInfo = pckFile.GetAssetsByType(PckAssetType.TexturePackInfoFile).FirstOrDefault();
|
|
string dataPath = texturePackInfo is not null ? texturePackInfo.GetProperty("DATAPATH") : string.Empty;
|
|
|
|
if (dataDirectoryInfo is null)
|
|
{
|
|
return skinPackage;
|
|
}
|
|
|
|
if (!string.IsNullOrWhiteSpace(dataPath))
|
|
{
|
|
PckFile infoPck = texturePackInfo.GetData(fileReader);
|
|
FileInfo texturePackFileInfo = dataDirectoryInfo.EnumerateFiles().Where(fileInfo => fileInfo.Name == dataPath).FirstOrDefault();
|
|
if (IsValidPckFile(texturePackFileInfo))
|
|
{
|
|
using Stream texturePackFileStream = texturePackFileInfo.OpenRead();
|
|
PckFile texturePackPck = fileReader.FromStream(texturePackFileStream);
|
|
IDLCPackage texturePackage = GetTexturePackageFromPckFile(infoPck, texturePackPck);
|
|
dlcPackageType = DLCPackageType.TexturePack;
|
|
}
|
|
}
|
|
|
|
IEnumerable<FileInfo> audioFiles = dataDirectoryInfo.EnumerateFiles("*.binka");
|
|
IDictionary<string, byte[]> audios = new Dictionary<string, byte[]>();
|
|
foreach (FileInfo audioFile in audioFiles)
|
|
{
|
|
MemoryStream dataStream = new MemoryStream();
|
|
using (Stream audioStream = audioFile.OpenRead())
|
|
{
|
|
audioStream.CopyTo(dataStream);
|
|
}
|
|
audios.Add(audioFile.Name, dataStream.ToArray());
|
|
}
|
|
|
|
IDictionary<string, IDictionary<string, byte[]>> saves = new Dictionary<string, IDictionary<string, byte[]>>();
|
|
foreach (FileInfo worldFile in dataDirectoryInfo.EnumerateFiles("*.mcs"))
|
|
{
|
|
IDictionary<string, byte[]> save = MapReader.OpenSave(worldFile.OpenRead());
|
|
saves.Add(worldFile.Name, save);
|
|
}
|
|
|
|
if (pckFile.Contains("GameRules.grf", PckAssetType.GameRulesFile) && pckFile.TryGetAsset("GameRules.grf", PckAssetType.GameRulesFile, out PckAsset gameRuleAsset))
|
|
{
|
|
GameRuleFile gameRuleFile = gameRuleAsset.GetData(new GameRuleFileReader(GetPlatformCompressionType()));
|
|
dlcPackageType = DLCPackageType.MashUpPack;
|
|
}
|
|
|
|
Debug.WriteLine(dlcPackageType);
|
|
return new UnknownDLCPackage(name, pckFile);
|
|
}
|
|
|
|
private GameRuleFile.CompressionType GetPlatformCompressionType()
|
|
{
|
|
switch (Platform)
|
|
{
|
|
case ConsolePlatform.Xbox360:
|
|
return GameRuleFile.CompressionType.XMem;
|
|
|
|
case ConsolePlatform.PS3:
|
|
return GameRuleFile.CompressionType.Deflate;
|
|
|
|
case ConsolePlatform.XboxOne:
|
|
case ConsolePlatform.PS4:
|
|
case ConsolePlatform.PSVita:
|
|
case ConsolePlatform.WiiU:
|
|
case ConsolePlatform.Switch:
|
|
return GameRuleFile.CompressionType.Zlib;
|
|
|
|
case ConsolePlatform.Unknown:
|
|
default:
|
|
return GameRuleFile.CompressionType.Unknown;
|
|
}
|
|
}
|
|
|
|
private IDLCPackage GetTexturePackageFromPckFile(PckFile infoPck, PckFile dataPck)
|
|
{
|
|
if (!infoPck.TryGetAsset("comparison.png", PckAssetType.TextureFile, out PckAsset comparisonAsset))
|
|
{
|
|
Trace.TraceError($"Could not find 'comparison.png'.");
|
|
return InvalidDLCPackage.Instance;
|
|
}
|
|
if (!infoPck.TryGetAsset("icon.png", PckAssetType.TextureFile, out PckAsset iconnAsset))
|
|
{
|
|
Trace.TraceError($"Could not find 'icon.png'.");
|
|
return InvalidDLCPackage.Instance;
|
|
}
|
|
|
|
Image comparisonImg = comparisonAsset.GetTexture();
|
|
Image iconImg = iconnAsset.GetTexture();
|
|
DLCTexturePackage.MetaData metaData = new DLCTexturePackage.MetaData(comparisonImg, iconImg);
|
|
|
|
bool hasTerrainAtlas = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.BlockAtlas, out Atlas terrainAtlas);
|
|
bool hasItemAtlas = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.ItemAtlas, out Atlas itemAtlas);
|
|
bool hasParticleAtlas = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.ParticleAtlas, out Atlas particleAtlas);
|
|
bool hasPaintingAtlas = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.PaintingAtlas, out Atlas paintingAtlas);
|
|
|
|
string itemAnimationResourcePath = ResourceLocation.GetPathFromCategory(ResourceCategory.ItemAnimation);
|
|
if (dataPck != null &&
|
|
dataPck.TryGetAsset(itemAnimationResourcePath + "/compass.png", PckAssetType.TextureFile, out PckAsset compassAsset) &&
|
|
dataPck.TryGetAsset(itemAnimationResourcePath + "/clock.png", PckAssetType.TextureFile, out PckAsset clockAsset))
|
|
{
|
|
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private bool TryGetAtlasFromResourceCategory(PckFile pck, ResourceCategory resourceCategory, out Atlas atlas)
|
|
{
|
|
ResourceLocation resourceLocation = ResourceLocation.GetFromCategory(resourceCategory);
|
|
if (!pck.TryGetAsset(resourceLocation.ToString(), PckAssetType.TextureFile, out PckAsset asset))
|
|
{
|
|
Trace.TraceWarning($"Could not find '{resourceLocation}'.");
|
|
atlas = null;
|
|
return false;
|
|
}
|
|
atlas = asset.GetDeserializedData(new AtlasDeserializer(resourceLocation));
|
|
return true;
|
|
}
|
|
|
|
private IDLCPackage GetDLCSkinPackage(string name, int identifier, PckFile pck, PckFileReader fileReader, IDLCPackage parentPackage = null)
|
|
{
|
|
Skin.Skin GetSkinWithCape(PckAsset skinAsset)
|
|
{
|
|
Skin.Skin skin = skinAsset.GetSkin();
|
|
if (skinAsset.TryGetProperty("CAPEPATH", out string capeAssetPath) && pck.TryGetAsset(capeAssetPath, PckAssetType.CapeFile, out PckAsset capeAsset))
|
|
skin.CapeTexture = capeAsset.GetTexture();
|
|
return skin;
|
|
}
|
|
|
|
IEnumerable<Skin.Skin> skins = pck.GetAssetsByType(PckAssetType.SkinFile).Select(GetSkinWithCape);
|
|
|
|
skins = skins.Concat(pck.GetAssetsByType(PckAssetType.SkinDataFile)
|
|
.Select(asset => asset.GetData(fileReader))
|
|
.SelectMany(pck => pck.GetAssetsByType(PckAssetType.SkinFile))
|
|
.Select(GetSkinWithCape)
|
|
);
|
|
|
|
return new DLCSkinPackage(name, identifier, skins, null, parentPackage);
|
|
}
|
|
|
|
private static string GetPreferredLanguage(AppLanguage appLanguage)
|
|
{
|
|
return appLanguage switch
|
|
{
|
|
AppLanguage.SystemDefault => LOCFile.ValidLanguages.Contains(CultureInfo.CurrentUICulture.Name) ? CultureInfo.CurrentUICulture.Name : AvailableLanguages.English,
|
|
AppLanguage.CzechCzechia => AvailableLanguages.CzechCzechia,
|
|
AppLanguage.Czechia => AvailableLanguages.Czechia,
|
|
AppLanguage.Danish => AvailableLanguages.Danish,
|
|
AppLanguage.DenmarkDanish => AvailableLanguages.DenmarkDanish,
|
|
AppLanguage.GermanAustria => AvailableLanguages.GermanAustria,
|
|
AppLanguage.German => AvailableLanguages.German,
|
|
AppLanguage.GreekGreece => AvailableLanguages.GreekGreece,
|
|
AppLanguage.Greece => AvailableLanguages.Greece,
|
|
AppLanguage.EnglishAustralia => AvailableLanguages.EnglishAustralia,
|
|
AppLanguage.EnglishCanada => AvailableLanguages.EnglishCanada,
|
|
AppLanguage.English => AvailableLanguages.English,
|
|
AppLanguage.EnglishUnitedKingdom => AvailableLanguages.EnglishUnitedKingdom,
|
|
AppLanguage.EnglishIreland => AvailableLanguages.EnglishIreland,
|
|
AppLanguage.EnglishNewZealand => AvailableLanguages.EnglishNewZealand,
|
|
AppLanguage.EnglishUnitedStatesOfAmerica => AvailableLanguages.EnglishUnitedStatesOfAmerica,
|
|
AppLanguage.SpanishSpain => AvailableLanguages.SpanishSpain,
|
|
AppLanguage.SpanishMexico => AvailableLanguages.SpanishMexico,
|
|
AppLanguage.FinnishFinland => AvailableLanguages.FinnishFinland,
|
|
AppLanguage.FrenchFrance => AvailableLanguages.FrenchFrance,
|
|
AppLanguage.FrenchCanada => AvailableLanguages.FrenchCanada,
|
|
AppLanguage.ItalianItaly => AvailableLanguages.ItalianItaly,
|
|
AppLanguage.JapaneseJapan => AvailableLanguages.JapaneseJapan,
|
|
AppLanguage.KoreanSouthKorea => AvailableLanguages.KoreanSouthKorea,
|
|
AppLanguage.Latin => AvailableLanguages.Latin,
|
|
AppLanguage.NorwegianNorway => AvailableLanguages.NorwegianNorway,
|
|
AppLanguage.NorwegianBokmålNorway => AvailableLanguages.NorwegianBokmålNorway,
|
|
AppLanguage.DutchNetherlands => AvailableLanguages.DutchNetherlands,
|
|
AppLanguage.DutchBelgium => AvailableLanguages.DutchBelgium,
|
|
AppLanguage.PolishPoland => AvailableLanguages.PolishPoland,
|
|
AppLanguage.PortugueseBrazil => AvailableLanguages.PortugueseBrazil,
|
|
AppLanguage.PortuguesePortugal => AvailableLanguages.PortuguesePortugal,
|
|
AppLanguage.RussianRussia => AvailableLanguages.RussianRussia,
|
|
AppLanguage.SlovakSlovakia => AvailableLanguages.SlovakSlovakia,
|
|
AppLanguage.SwedishSweden => AvailableLanguages.SwedishSweden,
|
|
AppLanguage.TurkishTurkey => AvailableLanguages.TurkishTurkey,
|
|
AppLanguage.ChineseChina => AvailableLanguages.ChineseChina,
|
|
AppLanguage.ChineseHongKong => AvailableLanguages.ChineseHongKong,
|
|
AppLanguage.ChineseSingapore => AvailableLanguages.ChineseSingapore,
|
|
AppLanguage.ChineseTaiwan => AvailableLanguages.ChineseTaiwan,
|
|
_ => AvailableLanguages.English,
|
|
};
|
|
}
|
|
|
|
public void SetPreferredLanguage(AppLanguage lang) => PreferredLanguage = GetPreferredLanguage(lang);
|
|
|
|
public void SetPlatform(ConsolePlatform platform)
|
|
{
|
|
_platform = platform;
|
|
_byteOrder = GetByteOrderForPlatform(platform);
|
|
}
|
|
}
|
|
} |