FModel/FModel/Creator/Bases/BaseIcon.cs

using System;
using System.Collections.Generic;
using System.Windows;
using FModel.Creator.Icons;
using FModel.Creator.Rarities;
using FModel.Creator.Stats;
using FModel.Creator.Texts;
using FModel.PakReader;
using FModel.PakReader.Parsers.Class;
using FModel.PakReader.Parsers.PropertyTagData;
using FModel.Properties;
using FModel.Utils;
using SkiaSharp;

namespace FModel.Creator.Bases
{
    public class BaseIcon : IBase
    {
        public SKBitmap FallbackImage;
        public SKBitmap IconImage;
        public SKBitmap RarityBackgroundImage;
        public SKBitmap[] UserFacingFlags;
        public SKColor[] RarityBackgroundColors;
        public SKColor[] RarityBorderColor;
        public string DisplayName;
        public string Description;
        public string ShortDescription;
        public string CosmeticSource;
        public int Size = 512;     // keep it 512 (or a multiple of 512) if you don't want blurry icons
        public int AdditionalSize; // must be increased if there are weapon stats, hero abilities or more to draw/show
        public int Margin = 2;
        public List<Statistic> Stats;

        public BaseIcon()
        {
            FallbackImage = SKBitmap.Decode(Application.GetResourceStream(new Uri("pack://application:,,,/Resources/T_Placeholder_Item_Image.png"))?.Stream);
            IconImage = FallbackImage;
            RarityBackgroundImage = null;
            UserFacingFlags = null;
            RarityBackgroundColors = new[] { SKColor.Parse("5EBC36"), SKColor.Parse("305C15") };
            RarityBorderColor = new[] { SKColor.Parse("74EF52"), SKColor.Parse("74EF52") };
            DisplayName = "";
            Description = "";
            ShortDescription = "";
            CosmeticSource = "";
            Stats = new List<Statistic>();
        }

        public BaseIcon(IUExport export, string assetName, bool forceHR) : this()
        {
            if (export.GetExport<ObjectProperty>("Series") is { } series)
                Serie.GetRarity(this, series);
            else if (Settings.Default.UseGameColors) // override default green
                Rarity.GetInGameRarity(this, export.GetExport<EnumProperty>("Rarity")); // uncommon will be triggered by Rarity being null
            else if (export.GetExport<EnumProperty>("Rarity") is { } rarity)
                Rarity.GetHardCodedRarity(this, rarity);

            if (export.GetExport<ObjectProperty>("HeroDefinition", "WeaponDefinition") is { } itemDef)
                LargeSmallImage.GetPreviewImage(this, itemDef, assetName, forceHR);
            else if (export.GetExport<SoftObjectProperty>(forceHR ? "LargePreviewImage" : "SmallPreviewImage", forceHR ? "ItemDisplayAsset" : "SmallImage", forceHR ? "SidePanelIcon" : "ToastIcon") is { } previewImage)
                LargeSmallImage.GetPreviewImage(this, previewImage);
            else if (export.GetExport<ObjectProperty>("access_item") is { } accessItem)
            {
                Package p = Utils.GetPropertyPakPackage(accessItem.Value.Resource.OuterIndex.Resource.ObjectName.String);
                if (p.HasExport() && !p.Equals(default))
                {
                    var d = p.GetExport<UObject>();
                    if (d != null)
                    {
                        IconImage = new BaseIcon(d, accessItem.Value.Resource.ObjectName.String + ".uasset", false).IconImage;
                    }
                }
            }

            if (export.GetExport<TextProperty>("DisplayName", "DefaultHeaderText", "UIDisplayName") is { } displayName)
                DisplayName = Text.GetTextPropertyBase(displayName);
        }

        /// <summary>
        /// Order:
        ///     1. Rarity
        ///     2. Image
        ///     3. Text
        ///         1. DisplayName
        ///         2. Description
        ///         3. Misc
        ///     4. GameplayTags
        ///         1. order doesn't matter
        ///         2. the importance here is to get the description before gameplay tags
        /// </summary>
        public BaseIcon(IUExport export, string exportType, ref string assetName) : this()
        {
            // rarity
            if (export.GetExport<ObjectProperty>("Series") is { } series)
                Serie.GetRarity(this, series);
            else if (Settings.Default.UseGameColors) // override default green
                Rarity.GetInGameRarity(this, export.GetExport<EnumProperty>("Rarity")); // uncommon will be triggered by Rarity being null
            else if (export.GetExport<EnumProperty>("Rarity") is { } rarity)
                Rarity.GetHardCodedRarity(this, rarity);

            // image
            if (Settings.Default.UseItemShopIcon &&
                DisplayAssetImage.GetDisplayAssetImage(this, export, ref assetName))
            { } // ^^^^ will return false if image not found, if so, we try to get the normal icon
            else if (export.GetExport<ObjectProperty>("HeroDefinition", "WeaponDefinition") is { } itemDef)
                LargeSmallImage.GetPreviewImage(this, itemDef, assetName);
            else if (export.GetExport<SoftObjectProperty>("LargePreviewImage", "SidePanelIcon", "EntryListIcon", "SmallPreviewImage", "ItemDisplayAsset") is { } previewImage)
                LargeSmallImage.GetPreviewImage(this, previewImage);
            else if (export.GetExport<ObjectProperty>("SmallPreviewImage", "ToastIcon") is { } smallPreviewImage)
                IconImage = Utils.GetObjectTexture(smallPreviewImage);
            else if (export.GetExport<StructProperty>("IconBrush") is { } iconBrush) // abilities
                LargeSmallImage.GetPreviewImage(this, iconBrush);

            // text
            if (export.GetExport<TextProperty>("DisplayName", "DefaultHeaderText", "UIDisplayName") is { } displayName)
                DisplayName = Text.GetTextPropertyBase(displayName);
            if (export.GetExport<TextProperty>("Description", "GeneralDescription", "DefaultBodyText", "UIDescription", "UIDisplayDescription") is { } description)
                Description = Text.GetTextPropertyBase(description);
            else if (export.GetExport<ArrayProperty>("Description") is { } arrayDescription) // abilities
                Description = Text.GetTextPropertyBase(arrayDescription);
            if (export.GetExport<StructProperty>("MaxStackSize") is { } maxStackSize)
                ShortDescription = Text.GetMaxStackSize(maxStackSize);
            else if (export.GetExport<StructProperty>("XpRewardAmount") is { } xpRewardAmount)
                ShortDescription = Text.GetXpRewardAmount(xpRewardAmount);
            else if (export.GetExport<TextProperty>("ShortDescription", "UIDisplaySubName") is { } shortDescription)
                ShortDescription = Text.GetTextPropertyBase(shortDescription);
            else if (exportType.Equals("AthenaItemWrapDefinition")) // if no ShortDescription it's most likely a wrap
                ShortDescription = Localizations.GetLocalization("Fort.Cosmetics", "ItemWrapShortDescription", "Wrap");

            // gameplaytags
            if (export.GetExport<StructProperty>("GameplayTags") is { } gameplayTags)
                GameplayTag.GetGameplayTags(this, gameplayTags, exportType);
            else if (export.GetExport<ObjectProperty>("cosmetic_item") is { } cosmeticItem) // variants
                CosmeticSource = cosmeticItem.Value.Resource.ObjectName.String;

            if (Settings.Default.DrawStats && export.GetExport<SoftObjectProperty>("AmmoData") is { } ammoData)
                Statistics.GetAmmoData(this, ammoData);
            if (Settings.Default.DrawStats && export.GetExport<StructProperty>("WeaponStatHandle") is { } weaponStatHandle &&
                (exportType.Equals("FortWeaponMeleeItemDefinition") ||
                (export.GetExport<SoftObjectProperty>("StatList") is { } statList &&
                !statList.Value.AssetPathName.String.StartsWith("/Game/UI/Tooltips/NoTooltipStats"))))
            {
                Statistics.GetWeaponStats(this, weaponStatHandle);
            }
            if (Settings.Default.DrawStats && export.GetExport<ObjectProperty>("HeroGameplayDefinition") is { } heroGameplayDefinition)
                Statistics.GetHeroStats(this, heroGameplayDefinition);

            /* Please do not add Schematics support because it takes way too much memory */
            /* Thank the STW Dev Team for using a 5,69Mb file to get... Oh nvm, they all left */

            AdditionalSize = 48 * Stats.Count;
        }

        SKBitmap IBase.FallbackImage => FallbackImage;
        SKBitmap IBase.IconImage => IconImage;
        SKColor[] IBase.RarityBackgroundColors => RarityBackgroundColors;
        SKColor[] IBase.RarityBorderColor => RarityBorderColor;
        string IBase.DisplayName => DisplayName;
        string IBase.Description => Description;
        int IBase.Width => Size;
        int IBase.Height => Size;
        int IBase.Margin => Margin;
    }
}