ICS-SCADA-APP-WEN/Library/PackageCache/com.unity.2d.tilemap.extras/Editor/Tiles/RuleOverrideTile/PopulateRuleOverrideTileWiz...

using UnityEngine;
using UnityEngine.Tilemaps;
using System.Linq;
using System.Text.RegularExpressions;
using UnityEngine.Scripting.APIUpdating;

namespace UnityEditor.Tilemaps
{
    /// <summary>
    /// Wizard for populating RuleOverrideTile from a SpriteSheet
    /// </summary>
    [MovedFrom(true, "UnityEditor")]
    public class PopulateRuleOverideTileWizard : ScriptableWizard 
    {
        [MenuItem("CONTEXT/RuleOverrideTile/Populate From Sprite Sheet")]
        static void MenuOption(MenuCommand menuCommand)
        {
            PopulateRuleOverideTileWizard.CreateWizard(menuCommand.context as RuleOverrideTile);
        }
        [MenuItem("CONTEXT/RuleOverrideTile/Populate From Sprite Sheet", true)]
        static bool MenuOptionValidation(MenuCommand menuCommand)
        {
            RuleOverrideTile tile = menuCommand.context as RuleOverrideTile;
            return tile.m_Tile;
        }

        /// <summary>
        /// The Texture2D containing the Sprites to override with
        /// </summary>
        public Texture2D m_spriteSet;

        private RuleOverrideTile m_tileset;

        /// <summary>
        /// Creates a wizard for the target RuleOverrideTIle
        /// </summary>
        /// <param name="target">The RuleOverrideTile to be edited by the wizard</param>
        public static void CreateWizard(RuleOverrideTile target) {
            PopulateRuleOverideTileWizard wizard = DisplayWizard<PopulateRuleOverideTileWizard>("Populate Override", "Populate");
            wizard.m_tileset = target;
        }

        /// <summary>
        /// Creates a new PopulateRuleOverideTileWizard and copies the settings from an existing PopulateRuleOverideTileWizard
        /// </summary>
        /// <param name="oldWizard">The wizard to copy settings from</param>
        public static void CloneWizard(PopulateRuleOverideTileWizard oldWizard) {
            PopulateRuleOverideTileWizard wizard = DisplayWizard<PopulateRuleOverideTileWizard>("Populate Override", "Populate");
            wizard.m_tileset = oldWizard.m_tileset;
            wizard.m_spriteSet = oldWizard.m_spriteSet;
        }

        private void OnWizardUpdate() {
            isValid = m_tileset != null && m_spriteSet != null;
        }

        private void OnWizardCreate() {
            try {
                Populate();
            }
            catch(System.Exception ex) {
                EditorUtility.DisplayDialog("Auto-populate failed!", ex.Message, "Ok");
                CloneWizard(this);
            }
        }

        /// <summary>
        /// Attempts to populate the selected override tile using the chosen sprite.
        /// The assumption here is that the sprite set selected by the user has the same
        /// naming scheme as the original sprite. That is to say, they should both have the same number
        /// of sprites, each sprite ends in an underscore followed by a number, and that they are
        /// intended to be equivalent in function.
        /// </summary>
        private void Populate() {
            string spriteSheet = AssetDatabase.GetAssetPath(m_spriteSet);
            Sprite[] overrideSprites = AssetDatabase.LoadAllAssetsAtPath(spriteSheet).OfType<Sprite>().ToArray();

            bool finished = false;

            try {
                Undo.RecordObject(m_tileset, "Auto-populate " + m_tileset.name);

                foreach(RuleTile.TilingRule rule in m_tileset.m_Tile.m_TilingRules) {
                    foreach(Sprite originalSprite in rule.m_Sprites) {
                        string spriteName = originalSprite.name;
                        string spriteNumber = Regex.Match(spriteName, @"_\d+$").Value;

                        Sprite matchingOverrideSprite = overrideSprites.First(sprite => sprite.name.EndsWith(spriteNumber));

                        m_tileset[originalSprite] = matchingOverrideSprite;
                    }
                }

                finished = true;
            }
            catch(System.InvalidOperationException ex) {
                throw (new System.ArgumentOutOfRangeException("Sprite sheet mismatch", ex));
            }
            finally {
                // We handle the undo like this in case we end up catching more exceptions.
                // We want this to ALWAYS happen unless we complete the population.
                if(!finished) {
                    Undo.PerformUndo();
                }
            }
        }

    }
}