fgm24/Assets/Scripts/PathFinding/NavMeshComponents/Editor/NavMeshModifierTilemapEditor.cs

using NavMeshPlus.Components;
using NavMeshPlus.Extensions;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using UnityEditor;
using UnityEditor.UIElements;
using UnityEditor.AI;
using UnityEngine.AI;
using UnityEngine.Tilemaps;

//***********************************************************************************
// Contributed by author jl-randazzo github.com/jl-randazzo
//***********************************************************************************
namespace NavMeshPlus.Editors.Components
{
    [CanEditMultipleObjects]
    [CustomEditor(typeof(NavMeshModifierTilemap))]
    class NavMeshModifierTilemapEditor : Editor
    {
        SerializedProperty m_TileModifiers;

        void OnEnable()
        {
            m_TileModifiers = serializedObject.FindProperty("m_TileModifiers");
        }

        public override void OnInspectorGUI()
        {
            NavMeshModifierTilemap modifierTilemap = target as NavMeshModifierTilemap;

            serializedObject.Update();
            EditorGUILayout.PropertyField(m_TileModifiers);

            if (modifierTilemap.HasDuplicateTileModifiers())
            {
                EditorGUILayout.HelpBox("There are duplicate Tile entries in the tilemap modifiers! Only the first will be used.", MessageType.Warning);
            }

            EditorGUILayout.Space();

            Tilemap tilemap = modifierTilemap.GetComponent<Tilemap>();
            if (tilemap)
            {
                if (GUILayout.Button("Add Used Tiles"))
                {
                    AddUsedTiles(tilemap, modifierTilemap);
                }
            }
            else
            {
                EditorGUILayout.HelpBox("Missing required component 'Tilemap'", MessageType.Error);
            }

            if (serializedObject.ApplyModifiedProperties())
            {
                modifierTilemap.CacheModifiers();
            }
        }

        private void AddUsedTiles(Tilemap tilemap, NavMeshModifierTilemap modifierTilemap)
        {
            Dictionary<TileBase, NavMeshModifierTilemap.TileModifier> tileModifiers = modifierTilemap.GetModifierMap();

            BoundsInt bounds = tilemap.cellBounds;
            for (int i = bounds.xMin; i <= bounds.xMax; i++)
            {
                for (int j = bounds.yMin; j <= bounds.yMax; j++)
                {
                    for (int k = bounds.zMin; k <= bounds.zMax; k++)
                    {
                        if (tilemap.GetTile(new Vector3Int(i, j, k)) is TileBase tileBase)
                        {
                            if (!tileModifiers.ContainsKey(tileBase))
                            {
                                tileModifiers.Add(tileBase, new NavMeshModifierTilemap.TileModifier());

                                int idx = m_TileModifiers.arraySize;
                                m_TileModifiers.InsertArrayElementAtIndex(idx);
                                var newElem = m_TileModifiers.GetArrayElementAtIndex(idx);
                                var tileProperty = newElem.FindPropertyRelative(nameof(NavMeshModifierTilemap.TileModifier.tile));
                                tileProperty.objectReferenceValue = tileBase;
                            }
                        }
                    }
                }
            }
        }

        [CustomPropertyDrawer(typeof(NavMeshModifierTilemap.TileModifier))]
        class TileModifierPropertyDrawer : PropertyDrawer
        {
            private Rect ClaimAdvance(ref Rect position, float height)
            {
                Rect retVal = position;
                retVal.height = height;
                position.y += height;
                position.height -= height;
                return retVal;
            }

            public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
            {
                Rect expandRect = ClaimAdvance(ref position, 20);
                property.isExpanded = EditorGUI.Foldout(expandRect, property.isExpanded, label);
                if (property.isExpanded)
                {
                    var tileProperty = property.FindPropertyRelative(nameof(NavMeshModifierTilemap.TileModifier.tile));
                    Rect tileRect = ClaimAdvance(ref position, 40);
                    tileRect.width -= 40;

                    Rect previewRect = tileRect;
                    previewRect.width = 40;
                    previewRect.x += tileRect.width;
                    tileRect.height /= 2;

                    // Adding the tile selector and a preview image.
                    EditorGUI.PropertyField(tileRect, tileProperty);
                    TileBase tileBase = tileProperty.objectReferenceValue as TileBase;
                    TileData tileData = new TileData();
                    tileBase?.GetTileData(Vector3Int.zero, null, ref tileData);
                    if (tileData.sprite)
                    {
                        EditorGUI.DrawPreviewTexture(previewRect, tileData.sprite?.texture, null, ScaleMode.ScaleToFit, 0);
                    }

                    Rect toggleRect = ClaimAdvance(ref position, 20);
                    var overrideAreaProperty = property.FindPropertyRelative(nameof(NavMeshModifierTilemap.TileModifier.overrideArea));
                    EditorGUI.PropertyField(toggleRect, overrideAreaProperty);

                    if (overrideAreaProperty.boolValue)
                    {
                        Rect areaRect = ClaimAdvance(ref position, 20);
                        var areaProperty = property.FindPropertyRelative(nameof(NavMeshModifierTilemap.TileModifier.area));
                        EditorGUI.indentLevel++;
                        EditorGUI.PropertyField(areaRect, areaProperty);
                        EditorGUI.indentLevel--;
                    }
                }
            }

            public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
            {
                if (property.isExpanded)
                {
                    var overrideAreaProperty = property.FindPropertyRelative(nameof(NavMeshModifierTilemap.TileModifier.overrideArea));
                    if (overrideAreaProperty.boolValue)
                    {
                        return 100;
                    }
                    return 80;
                }
                return 20;

            }
        }
    }
}