PCK-Studio/PCK-Studio/Forms/Editor/ModelEditor.cs

using System;
using System.IO;
using System.Data;
using System.Linq;
using System.Drawing;
using System.Diagnostics;
using System.Windows.Forms;
using System.Collections.Generic;

using OMI.Formats.Model;
using MetroFramework.Forms;

using PckStudio.Controls;
using PckStudio.Interfaces;
using OMI.Formats.Material;
using PckStudio.ModelSupport;
using PckStudio.Core.Json;
using PckStudio.Core.Extensions;
using PckStudio.Internal.App;
using PckStudio.Core;

namespace PckStudio.Forms.Editor
{
    public partial class ModelEditor : EditorForm<ModelContainer>
    {
        private readonly ITryGetSet<string, Image> _textures;
        private readonly ITryGet<string, MaterialContainer.Material> _tryGetEntityMaterial;

        public ModelEditor(ModelContainer models, ISaveContext<ModelContainer> saveContext, ITryGetSet<string, Image> tryGetSetTextures, ITryGet<string, MaterialContainer.Material> tryGetEntityMaterial)
            : base(models, saveContext)
        {
            InitializeComponent();
            _textures = tryGetSetTextures;
            _tryGetEntityMaterial = tryGetEntityMaterial;
            modelTreeView.ImageList = new ImageList
            {
                ColorDepth = ColorDepth.Depth32Bit,
                ImageSize = new Size(32, 32)
            };
            modelTreeView.ImageList.Images.AddRange(ApplicationScope.EntityImages);
        }

        private const int InvalidImageIndex = 127;
        // TODO: move to json file. -miku
        private static Dictionary<string, int> ModelImageIndex = new Dictionary<string, int>()
        {
                ["bat"]              = 3,
                ["blaze"]            = 4,
                ["boat"]             = 5,
                ["cat"]              = 6,
                ["spider"]           = 107,
                ["chicken"]          = 9,
                ["cod"]              = 10,
                ["cow"]              = 12,
                ["creeper"]          = 13,
                ["creeper_head"]     = 13,
                ["dolphin"]          = 14,
                ["horse.v2"]         = 110,
                ["guardian"]         = 109,
                ["bed"]              = 108,
                ["dragon"]           = 21,
                ["dragon_head"]      = 21,
                ["enderman"]         = 23,
                ["ghast"]            = 34,
                ["irongolem"]        = 40,
                ["lavaslime"]        = 46,
                ["llama"]            = 44,
                ["llamaspit"]        = 45,
                ["minecart"]         = 47,
                ["ocelot"]           = 50,
                ["parrot"]           = 53,
                ["phantom"]          = 54,
                ["pig"]              = 55,
                ["pigzombie"]        = 94,
                ["polarbear"]        = 57,
                ["rabbit"]           = 60,
                ["sheep"]            = 63,
                ["sheep.sheared"]    = 113,
                ["shulker"]          = 64,
                ["silverfish"]       = 66,
                ["skeleton"]         = 67,
                ["skeleton_head"]    = 67,
                ["skeleton.stray"]   = 77,
                ["skeleton.wither"]  = 89,
                ["skeleton_wither_head"] = 89,
                ["slime"]            = 115,
                ["slime.armor"]      = 116,

                ["snowgolem"]        = 71,
                ["squid"]            = 76,
                ["trident"]          = 80,
                ["turtle"]           = 82,
                ["villager"]         = 84,
                ["villager.witch"]   = 87,

                ["vex"]              = 83,
                ["evoker"]           = 25,
                ["vindicator"]       = 25,
                ["witherBoss"]       = 88,
                ["wolf"]             = 91,
                ["zombie"]           = 92,
                ["zombie_head"]      = 92,
                ["zombie.husk"]      = 39,
                ["zombie.villager"]  = 95,
                ["zombie.drowned"]   = 17,
                ["endermite"]        = 24,
                ["pufferfish.small"] = 111,
                ["pufferfish.mid"]   = 112,
                ["pufferfish.large"] = 59,
                ["salmon"]           = 62,
                ["stray.armor"]      = 118,
                ["stray_armor"]      = 118,
                ["tropicalfish_a"]   = 81,
                ["tropicalfish_b"]   = 81,
                ["mooshroom"]        = 48,
                ["witherBoss.armor"] = 90,

                // 1.14 models
                ["panda"]              = 52,
                ["ravager"]            = 61,
                ["pillager"]           = 56,
                ["villager_v2"]        = 101,
                ["zombie.villager_v2"] = 102,
        };

        private static int GetModelImageIndex(string name) => ModelImageIndex.TryGetValue(name, out int index) ? index : InvalidImageIndex;

        private class ModelNode : TreeNode
        {
            private Model _model;
            public Model Model => _model;

            private ModelNode(Model model)
                : base(model.Name)
            {
                _model = model;
                ImageIndex = GetModelImageIndex(model.Name);
                SelectedImageIndex = GetModelImageIndex(model.Name);
                Nodes.AddRange(GetModelPartNodes(_model.GetParts()).ToArray());
            }
            private static IEnumerable<TreeNode> GetModelPartNodes(IEnumerable<ModelPart> parts) => parts.Select(ModelPartNode.Create);

            internal static ModelNode Create(Model model) => new ModelNode(model);
        }

        private class ModelPartNode : TreeNode
        {
            private ModelPart _part;

            public ModelPart Part => _part;

            private ModelPartNode(ModelPart part)
                : base($"{part.Name} Pivot:{part.Translation * -1} Rot:{part.Rotation + part.AdditionalRotation} ")
            {
                _part = part;
                ImageIndex = 126;
                SelectedImageIndex = 126;
                Nodes.AddRange(GetModelBoxNodes(part.GetBoxes()).ToArray());
            }
            private static IEnumerable<TreeNode> GetModelBoxNodes(IEnumerable<ModelBox> boxes) => boxes.Select(ModelBoxNode.Create);

            internal static ModelPartNode Create(ModelPart part) => new ModelPartNode(part);
        }

        private class ModelBoxNode : TreeNode
        {
            private ModelBox _modelBox;
            public ModelBox Box => _modelBox;
            private ModelBoxNode(ModelBox modelBox)
                : base($"Box: pos:{modelBox.Position} size:{modelBox.Size}")
            {
                ImageIndex = 126;
                SelectedImageIndex = 126;
                _modelBox = modelBox;
            }

            internal static ModelBoxNode Create(ModelBox modelBox) => new ModelBoxNode(modelBox);
        }

        private class NamedTextureTreeNode : TreeNode
        {
            private readonly NamedData<Image> _namedTexture;

            public NamedTextureTreeNode(NamedData<Image> namedTexture)
                : base(namedTexture.Name)
            {
                Tag = namedTexture;
                _namedTexture = namedTexture;
            }

            public Image GetTexture() => _namedTexture.Value;
        }

        private void LoadModels()
        {
            modelTreeView.Nodes.Clear();
            modelTreeView.Nodes.AddRange(EditorValue.Select(ModelNode.Create).ToArray());
        }

        protected override void OnLoad(EventArgs e)
        {
            base.OnLoad(e);
            LoadModels();
        }

        private void exportToolStripMenuItem_Click(object sender, EventArgs e)
        {
            if (modelTreeView.SelectedNode is ModelNode modelNode)
            {
                Model model = modelNode.Model;
                Debug.Write(model.Name + "; ");
                Debug.WriteLine(model.TextureSize);

                GameModelImporter.Default.ExportSettings.CreateModelOutline =
                    MessageBox.Show(
                        $"Do you wish to have all model parts contained in a group called '{model.Name}'?",
                        "Group model parts", MessageBoxButtons.YesNo, MessageBoxIcon.Question) == DialogResult.Yes;

                using SaveFileDialog openFileDialog = new SaveFileDialog();
                openFileDialog.FileName = model.Name;
                openFileDialog.Filter = GameModelImporter.Default.SupportedModelFileFormatsFilter;

                if (openFileDialog.ShowDialog(this) == DialogResult.OK)
                {
                    IEnumerable<NamedData<Image>> textures = GetModelTextures(model.Name);
                    var modelInfo = new GameModelInfo(model, textures);
                    GameModelImporter.Default.Export(openFileDialog.FileName, modelInfo);
                }
            }
        }

        private void modelTreeView_BeforeSelect(object sender, TreeViewCancelEventArgs e)
        {
            exportToolStripMenuItem.Visible = e.Node is ModelNode;
            removeToolStripMenuItem.Visible = e.Node is ModelNode;
            editToolStripMenuItem.Visible = e.Node is ModelBoxNode;
            //removeToolStripMenuItem.Visible = e.Node is ModelPartNode || e.Node is ModelBoxNode;
            if (e.Node is ModelNode modelNode && modelNode.Model.Name != modelViewport.CurrentModelName)
            {
                NamedData<Image>[] textures = GetModelTextures(modelNode.Model.Name).ToArray();
                
                textureImageList.Images.Clear();
                namedTexturesTreeView.Nodes.Clear();

                foreach ((int i, NamedData<Image> item) in textures.enumerate())
                {
                    textureImageList.Images.Add(item.Value);
                    namedTexturesTreeView.Nodes.Add(new NamedTextureTreeNode(item) { ImageIndex = i, SelectedImageIndex = i });
                }
                if (textures.Length != 0)
                    modelViewport.Texture = textures[0].Value;

                modelViewport.LoadModel(modelNode.Model);
                if (GameModelImporter.ModelMetaData.TryGetValue(modelNode.Model.Name, out JsonModelMetaData modelMetaData) && !string.IsNullOrEmpty(modelMetaData.MaterialName) &&
                    _tryGetEntityMaterial.TryGet(modelMetaData.MaterialName, out MaterialContainer.Material entityMaterial) ||
                    _tryGetEntityMaterial.TryGet(modelNode.Model.Name, out entityMaterial))
                {
                    modelViewport.SetModelMaterial(entityMaterial);
                }
                modelViewport.ResetCamera();
            }
            if (e.Node is ModelPartNode modelPartNode && modelPartNode.Parent is ModelNode parentNode && modelViewport.CurrentModelName == parentNode.Model.Name)
            {
                modelViewport.Highlight(modelPartNode.Part);
            }

            if (e.Node is ModelBoxNode modelBoxNode && modelBoxNode.Parent is ModelPartNode parentPartNode && parentPartNode.Parent is ModelNode parentNode1 &&
                modelViewport.CurrentModelName == parentNode1.Model.Name)
            {
                modelViewport.Highlight(modelBoxNode.Box, parentPartNode.Part);
            }
        }

        private IEnumerable<NamedData<Image>> GetModelTextures(string modelName)
        {
            if (!GameModelImporter.ModelMetaData.ContainsKey(modelName) || GameModelImporter.ModelMetaData[modelName]?.TextureLocations?.Length <= 0)
                yield break;
            foreach (var textureLocation in GameModelImporter.ModelMetaData[modelName].TextureLocations)
            {
                if (_textures.TryGet(textureLocation, out Image img))
                    yield return new NamedData<Image>(Path.GetFileName(textureLocation), img);
            }
            yield break;
        }

        private void importToolStripMenuItem1_Click(object sender, EventArgs e)
        {
            OpenFileDialog fileDialog = new OpenFileDialog();
            fileDialog.Filter = GameModelImporter.Default.SupportedModelFileFormatsFilter;
            fileDialog.Title = "Select model";
            if (fileDialog.ShowDialog() == DialogResult.OK)
            {
                GameModelInfo modelInfo = GameModelImporter.Default.Import(fileDialog.FileName);
                if (modelInfo is null)
                {
                    MessageBox.Show("Import failed.", ProductName);
                    return;
                }

                //if (models.Version < modelInfo.ModelVersion)
                //{
                //	MessageBox.Show("Model container version does not match with the model version.", ProductName, MessageBoxButtons.OK, MessageBoxIcon.Error);
                //	return;
                //}

                EditorValue.SetModel(modelInfo.Model);

                foreach (NamedData<Image> texture in modelInfo.Textures)
                {
                    _textures.TrySet(texture.Name, texture.Value);
                }

                LoadModels();
            }
        }

        private void saveToolStripMenuItem_Click(object sender, EventArgs e)
        {
            Save();
            DialogResult = DialogResult.OK;
        }

        private void namedTexturesTreeView_AfterSelect(object sender, TreeViewEventArgs e)
        {
            if (namedTexturesTreeView.SelectedNode is NamedTextureTreeNode namedTextureNode)
                modelViewport.Texture = namedTextureNode.GetTexture();
        }

        private void showModelBoundsToolStripMenuItem_CheckedChanged(object sender, EventArgs e)
        {
            modelViewport.RenderModelBounds = showModelBoundsToolStripMenuItem.Checked;
        }

        private void removeToolStripMenuItem_Click(object sender, EventArgs e)
        {
            if (modelTreeView?.SelectedNode is ModelNode modelNode && EditorValue.Remove(modelNode.Model))
            {
                modelNode.Remove();
            }
        }
    }
}