PCK-Studio/PckStudio.Core/Extensions/SkinExtensions.cs

using OMI.Formats.Languages;
using OMI.Formats.Model;
using OMI.Formats.Pck;
using OpenTK;
using PckStudio.Core.Skin;
using System;
using System.Collections.Generic;
using System.Drawing;
using System.Drawing.Drawing2D;
using System.Linq;

namespace PckStudio.Core.Extensions
{
    public static class SkinExtensions
    {
        public static PckAsset CreateFile(this Skin.Skin skin, LOCFile localizationFile)
        {
            string skinId = skin.Identifier.ToString("d08");
            PckAsset skinFile = new PckAsset($"dlcskin{skinId}.png", PckAssetType.SkinFile);

            skinFile.AddParameter("DISPLAYNAME", skin.MetaData.Name);
            if (localizationFile is not null)
            {
                string skinLocKey = $"IDS_dlcskin{skinId}_DISPLAYNAME";
                skinFile.AddParameter("DISPLAYNAMEID", skinLocKey);
                localizationFile.AddLocKey(skinLocKey, skin.MetaData.Name);
            }

            if (!string.IsNullOrEmpty(skin.MetaData.Theme))
            {
                skinFile.AddParameter("THEMENAME", skin.MetaData.Theme);
                if (localizationFile is not null)
                {
                    skinFile.AddParameter("THEMENAMEID", $"IDS_dlcskin{skinId}_THEMENAME");
                    localizationFile.AddLocKey($"IDS_dlcskin{skinId}_THEMENAME", skin.MetaData.Theme);
                }
            }

            if (skin.HasCape)
            {
                skinFile.AddParameter("CAPEPATH", $"dlccape{skinId}.png");
            }

            skinFile.AddParameter("ANIM", skin.Anim);
            skinFile.AddParameter("GAME_FLAGS", skin.GameFlags);
            skinFile.AddParameter("FREE", "1");

            foreach (SkinBOX box in skin.Model.AdditionalBoxes)
            {
                skinFile.AddParameter(box.ToParameter());
            }
            foreach (SkinPartOffset offset in skin.Model.PartOffsets)
            {
                skinFile.AddParameter(offset.ToParameter());
            }

            skinFile.SetTexture(skin.Texture);

            return skinFile;
        }

        public static PckAsset CreateCapeFile(this Skin.Skin skin)
        {
            if (!skin.HasCape)
                throw new InvalidOperationException("Skin does not contain a cape.");
            string skinId = skin.Identifier.ToString("d08");
            PckAsset capeFile = new PckAsset($"dlccape{skinId}.png", PckAssetType.CapeFile);
            capeFile.SetTexture(skin.CapeTexture);
            return capeFile;
        }

        // Function to create a paper doll from the skin for use in 2D contexts
        public static Image DrawPaperDoll(this Skin.Skin skin, float pixelScale = 10.0f, int xmlVersion = 0, bool bustCrop = false)
        {
            //pixel scale set to 10 so inflated parts can be seen as properly as possible

            Image skinTexture = skin.Texture;
            if (skinTexture.Width != skinTexture.Height && skinTexture.Width / skinTexture.Height != 2)
            {
                return null;
            }

            bool isWideSkin = skin.Anim.GetFlag(SkinAnimFlag.MODERN_WIDE_MODEL);
            bool isSlimSkin = skin.Anim.GetFlag(SkinAnimFlag.SLIM_MODEL);
            bool isModernSkin = isWideSkin || isSlimSkin;
            bool isDinnerbone = skin.Anim.GetFlag(SkinAnimFlag.DINNERBONE);
            bool isStatueOfLiberty = skin.Anim.GetFlag(SkinAnimFlag.STATUE_OF_LIBERTY);

            int skinHeight = isModernSkin ? 64 : 32;

            // fix for abnormal skin textures that are smaller than 64
            if (skinTexture.Width != 64 || skinTexture.Height != skinHeight)
            {
                Bitmap resized = new Bitmap(64, skinHeight);

                using (Graphics gfx = Graphics.FromImage(resized))
                {
                    gfx.InterpolationMode = InterpolationMode.NearestNeighbor;
                    gfx.PixelOffsetMode = PixelOffsetMode.Half;
                    gfx.SmoothingMode = SmoothingMode.None;
                    gfx.CompositingMode = CompositingMode.SourceCopy;

                    gfx.DrawImage(
                        skinTexture,
                        new Rectangle(0, 0, 64, skinHeight)
                    );
                }

                skinTexture = resized;
            }

            float textureScaleX = skinTexture.Width / 64f; // minecraft skins always have a width of 64
            float textureScaleY = skinTexture.Height / (isModernSkin ? 64f : 32f);

            // start with a large canvas and crop down later
            Image paperDoll = new Bitmap(512, 512);

            using (Graphics gfx = Graphics.FromImage(paperDoll))
            {
                gfx.InterpolationMode = InterpolationMode.NearestNeighbor;
                gfx.PixelOffsetMode = PixelOffsetMode.HighQuality;
                gfx.SmoothingMode = SmoothingMode.None;
                gfx.CompositingMode = CompositingMode.SourceOver;

                SkinBOX baseHeadBox = new SkinBOX("HEAD", new(-4, -8, -4), new(8), new(0, 0));
                SkinBOX baseTorsoBox = new SkinBOX("BODY", new(-4, 0, -2), new(8, 12, 4), new(16, 16));

                List<SkinBOX> baseBoxes = new List<SkinBOX>();
                List<SkinBOX> skinBoxes = new List<SkinBOX>();

                if (!skin.Anim.GetFlag(SkinAnimFlag.HEAD_DISABLED))
                {
                    baseBoxes.Add(baseHeadBox);
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.HEADWEAR_DISABLED))
                {
                    baseBoxes.Add(new SkinBOX("HEADWEAR", new(-4, -8, -4), new(8), new(32, 0), scale: 0.5f));
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.TORSO_DISABLED))
                {
                    baseBoxes.Add(baseTorsoBox);
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.JACKET_DISABLED) && isModernSkin)
                {
                    baseBoxes.Add(new SkinBOX("JACKET", new(-4, 0, -2), new(8, 12, 4), new(16, 32), scale: 0.25f));
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.RIGHT_ARM_DISABLED))
                {
                    baseBoxes.Add(new SkinBOX("ARM0", new(isSlimSkin ? -2 : -3, -2, -2), new(isSlimSkin ? 3 : 4, 12, 4), new(40, 16)));
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.RIGHT_SLEEVE_DISABLED) && isModernSkin)
                {
                    baseBoxes.Add(new SkinBOX("SLEEVE0", new(isSlimSkin ? -2 : -3, -2, -2), new(isSlimSkin ? 3 : 4, 12, 4), new(40, 32), scale: 0.25f));
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.LEFT_ARM_DISABLED))
                {
                    if (!isModernSkin)
                    {
                        baseBoxes.Add(new SkinBOX("ARM1", new(-1, -2, -2), new(4, 12, 4), new(40, 16), mirror: true));
                    }
                    else
                    {
                        baseBoxes.Add(new SkinBOX("ARM1", new(-1, -2, -2), new(isSlimSkin ? 3 : 4, 12, 4), new(32, 48)));
                    }
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.LEFT_SLEEVE_DISABLED) && isModernSkin)
                {
                    baseBoxes.Add(new SkinBOX("SLEEVE1", new(-1, -2, -2), new(isSlimSkin ? 3 : 4, 12, 4), new(48, 48), scale: 0.25f));
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.RIGHT_LEG_DISABLED))
                {
                    baseBoxes.Add(new SkinBOX("LEG0", new(-2, 0, -2), new(4, 12, 4), new(0, 16)));
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.RIGHT_PANTS_DISABLED) && isModernSkin)
                {
                    baseBoxes.Add(new SkinBOX("PANTS0", new(-2, 0, -2), new(4, 12, 4), new(0, 32), scale: 0.25f));
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.LEFT_LEG_DISABLED))
                {
                    if (!isModernSkin)
                    {
                        baseBoxes.Add(new SkinBOX("LEG1", new(-2, 0, -2), new(4, 12, 4), new(0, 16), mirror: true));
                    }
                    else
                    {
                        baseBoxes.Add(new SkinBOX("LEG1", new(-2, 0, -2), new(4, 12, 4), new(16, 48)));
                    }
                }
                if (!skin.Anim.GetFlag(SkinAnimFlag.LEFT_PANTS_DISABLED) && isModernSkin)
                {
                    baseBoxes.Add(new SkinBOX("PANTS1", new(-2, 0, -2), new(4, 12, 4), new(0, 48), scale: 0.25f));
                }

                HashSet<SkinBOX> baseBoxSet = new(baseBoxes);

                baseBoxes.Clear();

                skinBoxes.AddRange(baseBoxSet);
                skinBoxes.AddRange(skin.Model.AdditionalBoxes);

                skinBoxes.Sort((first, second) =>
                {
                    float aZPos = isStatueOfLiberty && first.Type == "ARM0"
                        ? -first.Pos.Z
                        : first.Pos.Z;

                    float bZPos = isStatueOfLiberty && second.Type == "ARM0"
                        ? -second.Pos.Z
                        : second.Pos.Z;

                    int result = bZPos.CompareTo(aZPos);

                    return result != 0
                        ? result
                        : first.Scale.CompareTo(second.Scale);
                });

                float canvasWidth = 0;
                float canvasHeight = 0;
                float canvasMinX = float.MaxValue;
                float canvasMinY = float.MaxValue;
                float canvasMaxX = float.MinValue;
                float canvasMaxY = float.MinValue;

                // used to center the render around the torso to best use space
                float torsoCenterY = baseTorsoBox.Pos.Y + (baseTorsoBox.Size.Y / 2f);

                float defaultDrawOffsetX = paperDoll.Width / 2f; // draw at center of the canvas
                float defaultDrawOffsetY = paperDoll.Height / 2f - torsoCenterY * pixelScale; // draw from center of the torso box

                var drawFunctions = new List<(
                    Image BackFace,
                    Image FrontFace,
                    RectangleF DrawRect,
                    bool StatueOfLibertyArm
                )>();

                var partOffsets = skin.Model.PartOffsets
                    .ToDictionary(x => x.Type, x => x.Value);

                foreach (SkinBOX box in skinBoxes)
                {
                    float boxWidth = box.Size.X;
                    float boxHeight = box.Size.Y;
                    float boxDepth = box.Size.Z;

                    if (boxWidth <= 0 || boxHeight <= 0) // if either of these are 0, skip because drawing is impossible
                        continue;

                    bool isStatueOfLibertyArm = isStatueOfLiberty && (box.Type == "ARM0" || box.Type == "SLEEVE0" || box.Type == "ARMARMOR0");

                    box.UV = new (box.UV.X % 64, box.UV.Y % (isModernSkin ? 64f : 32f));

                    // this math is basically to ensure the face is stretched if the texture is improper
                    Rectangle faceRect = new Rectangle(
                        isStatueOfLibertyArm // get back of ARM 0 box if Statue of Liberty
                            ? (int)((box.UV.X + boxWidth + 2 * boxDepth) * textureScaleX) // depth + width + depth = offset for back of box
                            : (int)((box.UV.X + boxDepth) * textureScaleX),
                        (int)((box.UV.Y + boxDepth) * textureScaleY),
                        (int)(boxWidth * textureScaleX),
                        (int)(boxHeight * textureScaleY)
                    );

                    if(faceRect.Size.Height == 0)
                    {
                        faceRect.Size = new Size(faceRect.X, 1);
                    }

                    // get back of the face to display behind just incase transparent parts exist on the front layer
                    Rectangle backFaceRect = new Rectangle(
                        !isStatueOfLibertyArm // get front of ARM 0 box if Statue of Liberty
                            ? (int)((box.UV.X + boxWidth + 2 * boxDepth) * textureScaleX) // depth + width + depth = offset for back of box
                            : (int)((box.UV.X + boxDepth) * textureScaleX),
                        (int)((box.UV.Y + boxDepth) * textureScaleY),
                        (int)(boxWidth * textureScaleX),
                        (int)(boxHeight * textureScaleY)
                    );

                    if (backFaceRect.Size.Height == 0)
                    {
                        backFaceRect.Size = new Size(backFaceRect.X, 1);
                    }

                    Image boxFace = skinTexture.GetArea(faceRect);
                    Image backBoxFace = skinTexture.GetArea(backFaceRect);

                    if (box.Mirror)
                    {
                        boxFace.RotateFlip(RotateFlipType.RotateNoneFlipX);
                    }
                    else
                    {
                        backBoxFace.RotateFlip(RotateFlipType.RotateNoneFlipX);
                    }

                    if (isStatueOfLibertyArm)
                    {
                        boxFace.RotateFlip(RotateFlipType.RotateNoneFlipY);
                        backBoxFace.RotateFlip(RotateFlipType.RotateNoneFlipY);
                    }

                    // these values are added to later
                    float drawOffsetX = defaultDrawOffsetX; // draw at center of the canvas
                    float drawOffsetY = defaultDrawOffsetY;

                    // offset by 4 pixels so feet align with where the head should be for the bust crop. This might not be perfect, but it gets the job done for the 2 "vanilla" upside-down skins
                    if (isDinnerbone)
                        drawOffsetY -= 4 * pixelScale;

                    float armXOffset = GameConstants.SkinLeftArmTranslation.X * pixelScale;
                    float armYOffset = GameConstants.SkinLeftArmTranslation.Y * pixelScale;
                    float legXOffset = GameConstants.SkinLeftLegTranslation.X * pixelScale;
                    float legYOffset = GameConstants.SkinLeftLegTranslation.Y * pixelScale;

                    float offset = 0;

                    switch (box.Type)
                    {
                        case "HEAD":
                        case "HEADWEAR":
                            partOffsets.TryGetValue("HEAD", out offset);
                            break;

                        case "BODY":
                        case "WAIST":
                        case "JACKET":
                        case "BELT":
                        case "BODYARMOR":
                            partOffsets.TryGetValue("BODY", out offset);
                            break;

                        case "ARM0":
                        case "SLEEVE0":
                        case "ARMARMOR0":
                            partOffsets.TryGetValue("ARM0", out offset);
                            drawOffsetX -= armXOffset;
                            drawOffsetY += armYOffset;
                            break;

                        case "ARM1":
                        case "SLEEVE1":
                        case "ARMARMOR1":
                            partOffsets.TryGetValue("ARM1", out offset);
                            drawOffsetX += armXOffset;
                            drawOffsetY += armYOffset;
                            break;

                        case "LEG0":
                        case "PANTS0":
                        case "LEGGING0":
                        case "SOCK0":
                        case "BOOT0":
                            partOffsets.TryGetValue("LEG0", out offset);
                            drawOffsetX -= legXOffset;
                            drawOffsetY += legYOffset;
                            break;

                        case "LEG1":
                        case "PANTS1":
                        case "LEGGING1":
                        case "SOCK1":
                        case "BOOT1":
                            partOffsets.TryGetValue("LEG1", out offset);
                            drawOffsetX += legXOffset;
                            drawOffsetY += legYOffset;
                            break;
                    }

                    drawOffsetY += offset * pixelScale;

                    float boxPosY = box.Pos.Y;

                    if (isStatueOfLibertyArm)
                    {
                        boxPosY = -(box.Pos.Y + box.Size.Y);
                    }

                    float drawX = box.Pos.X * pixelScale;
                    float drawY = boxPosY * pixelScale;

                    float drawWidth = boxWidth * pixelScale;
                    float drawHeight = boxHeight * pixelScale;

                    // handle Box scale
                    float boxInflate =
                        // if XMLVersion 3 or a base box; then return box scale
                        (xmlVersion == 3 || baseBoxSet.Contains(box)) ? box.Scale
                            // if not; then check that it is an overlay box
                            : (xmlVersion > 0 && xmlVersion < 3 && SkinBOX.OverlayTypes.Contains(box.Type))
                                // if so, return hardcoded box inflate value
                                ? (box.Type == "HEADWEAR" ? 0.5f : 0.25f)
                                : 0f; // finally, if not caught anywhere else, return no scale

                    float drawInflate = boxInflate * pixelScale * 1.5f; // no idea why 1.5 seems to fix this :,)
                    float halfInflation = drawInflate * 0.5f;

                    float finalDrawX = drawOffsetX + drawX - halfInflation;
                    float finalDrawY = drawOffsetY + drawY - halfInflation;

                    float finalDrawWidth = drawWidth + drawInflate;
                    float finalDrawHeight = drawHeight + drawInflate;

                    RectangleF boxRect = new RectangleF(finalDrawX, finalDrawY, finalDrawWidth, finalDrawHeight);

                    if (isStatueOfLiberty && box.Type == "ARM0")
                    {
                        float pivotX = defaultDrawOffsetX + GameConstants.SkinRightArmPivot.X * pixelScale;
                        float pivotY = defaultDrawOffsetY + GameConstants.SkinRightArmPivot.Y * pixelScale;

                        const float Angle = -8f * (float)Math.PI / 180f;

                        float cos = (float)Math.Cos(Angle);
                        float sin = (float)Math.Sin(Angle);

                        // used to calculate if the bounds of the arm go beyond the current bounds of the image so that the entire skin is shown
                        PointF[] armBounds =
                        {
                                new PointF(finalDrawX, finalDrawY),
                                new PointF(finalDrawX + finalDrawWidth, finalDrawY),
                                new PointF(finalDrawX, finalDrawY + finalDrawHeight),
                                new PointF(finalDrawX + finalDrawWidth, finalDrawY + finalDrawHeight)
                            };

                        foreach (var bound in armBounds)
                        {
                            float finalX = bound.X - pivotX;
                            float finalY = bound.Y - pivotY;

                            float rotateX = pivotX + (finalX * cos - finalY * sin);
                            float rotateY = pivotY + (finalX * sin + finalY * cos);

                            canvasMinX = Math.Min(canvasMinX, rotateX);
                            canvasMinY = Math.Min(canvasMinY, rotateY);
                            canvasMaxX = Math.Max(canvasMaxX, rotateX);
                            canvasMaxY = Math.Max(canvasMaxY, rotateY);
                        }
                    }
                    else
                    {
                        canvasMinX = Math.Min(canvasMinX, finalDrawX);
                        canvasMinY = Math.Min(canvasMinY, finalDrawY);
                        canvasMaxX = Math.Max(canvasMaxX, finalDrawX + finalDrawWidth);
                        canvasMaxY = Math.Max(canvasMaxY, finalDrawY + finalDrawHeight);
                    }

                    drawFunctions.Add((
                        backBoxFace,
                        boxFace,
                        boxRect,
                        isStatueOfLiberty && box.Type == "ARM0"
                    ));
                }

                // draw back faces first
                for (int i = drawFunctions.Count - 1; i >= 0; i--)
                {
                    var function = drawFunctions[i];

                    if (function.StatueOfLibertyArm)
                    {
                        float pivotX = defaultDrawOffsetX + GameConstants.SkinRightArmPivot.X * pixelScale;
                        float pivotY = defaultDrawOffsetY + GameConstants.SkinRightArmPivot.Y * pixelScale;

                        var state = gfx.Save();

                        gfx.TranslateTransform(pivotX, pivotY);
                        gfx.RotateTransform(-8f);
                        gfx.TranslateTransform(-pivotX, -pivotY);

                        gfx.DrawImage(function.BackFace, function.DrawRect);

                        gfx.Restore(state);
                    }
                    else
                    {
                        gfx.DrawImage(function.BackFace, function.DrawRect);
                    }
                }

                // then draw all fronts overtop
                foreach (var function in drawFunctions)
                {
                    if (function.StatueOfLibertyArm)
                    {
                        float pivotX = defaultDrawOffsetX + GameConstants.SkinRightArmPivot.X * pixelScale;
                        float pivotY = defaultDrawOffsetY + GameConstants.SkinRightArmPivot.Y * pixelScale;

                        var state = gfx.Save();

                        gfx.TranslateTransform(pivotX, pivotY);
                        gfx.RotateTransform(-8f);
                        gfx.TranslateTransform(-pivotX, -pivotY);

                        gfx.DrawImage(function.FrontFace, function.DrawRect);

                        gfx.Restore(state);
                    }
                    else
                    {
                        gfx.DrawImage(function.FrontFace, function.DrawRect);
                    }
                }

                canvasWidth = canvasMaxX - canvasMinX;
                canvasHeight = canvasMaxY - canvasMinY;

                Rectangle cropRect = new Rectangle(
                    (int)Math.Floor(canvasMinX),
                    (int)Math.Floor(canvasMinY),
                    (int)Math.Ceiling(canvasMaxX - canvasMinX),
                    (int)Math.Ceiling(canvasMaxY - canvasMinY)
                );

                if (bustCrop)
                {
                    if(isDinnerbone)
                        paperDoll.RotateFlip(RotateFlipType.RotateNoneFlipY);

                    float headCenterY = baseHeadBox.Pos.Y + (baseHeadBox.Size.Y / 2f);

                    float cropX = defaultDrawOffsetX;
                    float cropY = defaultDrawOffsetY + (headCenterY * pixelScale);

                    const int size = 256;

                    cropRect = new Rectangle(
                        (int)Math.Floor(cropX - size / 2f),
                        (int)Math.Floor(cropY - size / 2f),
                        size,
                        size
                    );
                }

                paperDoll = paperDoll.GetArea(cropRect);

                if (isDinnerbone && !bustCrop)
                    paperDoll.RotateFlip(RotateFlipType.RotateNoneFlipY);

                return paperDoll;
            }
        }
    }
}