Switch-Toolbox/Switch_FileFormatsMain/GUI/AAMP/AampV1Editor.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Drawing;
using System.Data;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using AampV1Library;
using Syroot.Maths;
using Switch_Toolbox.Library;

namespace FirstPlugin.Forms
{
    public partial class AampV1Editor : UserControl
    {
        public AampV1Editor()
        {
            InitializeComponent();
        }

        public void LoadImages(TreeView treeView, TreeNode parentNode)
        {
            foreach (TreeNode nodes in TreeViewExtensions.Collect(parentNode.Nodes))
            {
                if (nodes is EditableEntry)
                {
                    if (((EditableEntry)nodes).entry.ParamType == ParamType.Color4F)
                    {
                        nodes.ImageIndex += treeView.ImageList.Images.Count;
                        nodes.SelectedImageIndex = nodes.ImageIndex;
                    }
                }
            }

            foreach (var image in Images)
                treeView.ImageList.Images.Add(image);
        }

        public void LoadFile(AampFile aampFile, TreeNode parentNode)
        {
            LoadChildNodes(aampFile.RootNode, parentNode);
        }

        public void LoadChildNodes(ParamList paramList, TreeNode parentNode)
        {
            TreeNode newNode = new TreeNode(paramList.HashString);
     
            parentNode.Nodes.Add(newNode);

            //Add lists and objects if exits
            if (paramList.childParams.Length > 0)
                newNode.Nodes.Add("list", "Lists {}");
            if (paramList.paramObjects.Length > 0)
                newNode.Nodes.Add("obj", "Objects {}");

            //Add child nodes
            foreach (var child in paramList.childParams)
                LoadChildNodes(child, newNode.Nodes["list"]);

            //Add object nodes
            foreach (var obj in paramList.paramObjects)
                SetObjNode(obj, newNode.Nodes["obj"]);
        }

        List<Bitmap> Images = new List<Bitmap>();

        void SetObjNode(ParamObject paramObj, TreeNode parentNode)
        {
            string name = paramObj.HashString;
            string groupName = paramObj.GroupHashString;

            var objNode = new TreeNode(name);
            parentNode.Nodes.Add(objNode);

            foreach (var entry in paramObj.paramEntries)
            {
                var entryNode = new EditableEntry($"{entry.HashString}");
                entryNode.ImageIndex = 0;
                entryNode.ImageKey = "empty";
                entryNode.SelectedImageKey = "empty";

                switch (entry.ParamType)
                {
                    case ParamType.Boolean:
                    case ParamType.Float:
                    case ParamType.Int:
                    case ParamType.Uint:
                    case ParamType.String64:
                    case ParamType.String32:
                    case ParamType.String256:
                    case ParamType.StringRef:
                        entryNode.Text = $"{entry.HashString} {entry.Value}";
                        break;
                    case ParamType.Vector2F:
                        var vec2 = (Vector2F)entry.Value;
                        entryNode.Text = $"{entry.HashString} {vec2.X} {vec2.Y}";
                        break;
                    case ParamType.Vector3F:
                        var vec3 = (Vector3F)entry.Value;
                        entryNode.Text = $"{entry.HashString} {vec3.X} {vec3.Y} {vec3.Z}";
                        break;
                    case ParamType.Vector4F:
                        var vec4 = (Vector4F)entry.Value;
                        entryNode.Text = $"{entry.HashString} {vec4.X} {vec4.Y} {vec4.Z} {vec4.W}";
                        break;
                    case ParamType.Color4F:
                        var col = (Vector4F)entry.Value;
                        entryNode.Text = $"{entry.HashString} {col.X} {col.Y} {col.Z} {col.W}";

                        int ImageIndex = Images.Count;
                        entryNode.ImageIndex = ImageIndex;

                        var color = System.Drawing.Color.FromArgb(
                        EditBox.FloatToIntClamp(col.W),
                        EditBox.FloatToIntClamp(col.X),
                        EditBox.FloatToIntClamp(col.Y),
                        EditBox.FloatToIntClamp(col.Z));

                        Bitmap bmp = new Bitmap(32, 32);
                        Graphics g = Graphics.FromImage(bmp);
                        g.Clear(color);

                        Images.Add(bmp);
                        break;
                    default:
                        break;
                }

                entryNode.entry = entry;
                objNode.Nodes.Add(entryNode);
            }
        }

        public class EditableEntry : TreeNode
        {
            public ParamEntry entry;
            public EditableEntry(string name)
            {
                Text = name;

                ContextMenu = new ContextMenu();
                ContextMenu.MenuItems.Add(new MenuItem("Edit", OpenEditor));
            }

            private void OpenEditor(object sender, EventArgs e)
            {
                EditBox editor = new EditBox();
                editor.LoadEntry(entry);

                if (editor.ShowDialog() == DialogResult.OK)
                {
                    editor.SaveEntry();

                    switch (entry.ParamType)
                    {
                        case ParamType.Boolean:
                        case ParamType.Float:
                        case ParamType.Int:
                        case ParamType.Uint:
                        case ParamType.String64:
                        case ParamType.String32:
                        case ParamType.String256:
                        case ParamType.StringRef:
                            Text = $"{entry.HashString} {entry.Value}";
                            break;
                        case ParamType.Vector2F:
                            var vec2 = (Vector2F)entry.Value;
                            Text = $"{entry.HashString} {vec2.X} {vec2.Y}";
                            break;
                        case ParamType.Vector3F:
                            var vec3 = (Vector3F)entry.Value;
                            Text = $"{entry.HashString} {vec3.X} {vec3.Y} {vec3.Z}";
                            break;
                        case ParamType.Vector4F:
                            var vec4 = (Vector4F)entry.Value;
                            Text = $"{entry.HashString} {vec4.X} {vec4.Y} {vec4.Z} {vec4.W}";
                            break;
                        case ParamType.Color4F:
                            var col = (Vector4F)entry.Value;
                            Text = $"{entry.HashString} {col.X} {col.Y} {col.Z} {col.W}";
                            break;
                    }
                }
            }
        }
    }
}