Assets/Scripts/Imported/Tilemap/GridInformation/Scripts/GridInformation.cs

   1 using System;
   2 using System.Collections;
   3 using System.Collections.Generic;
   4 using System.Linq;
   5
   6 namespace UnityEngine.Tilemaps
   7 {
   8     [Serializable]
   9     internal enum GridInformationType
  10     {
  11         Integer,
  12         String,
  13         Float,
  14         Double,
  15         UnityObject,
  16         Color
  17     }
  18
  19     [Serializable]
  20     [AddComponentMenu("Tilemap/Grid Information")]
  21     public class GridInformation : MonoBehaviour, ISerializationCallbackReceiver
  22     {
  23         internal struct GridInformationValue
  24         {
  25             public GridInformationType type;
  26             public object data;
  27         }
  28
  29         [Serializable]
  30         internal struct GridInformationKey
  31         {
  32             public Vector3Int position;
  33             public String name;
  34         }
  35
  36         private Dictionary<GridInformationKey, GridInformationValue> m_PositionProperties = new Dictionary<GridInformationKey, GridInformationValue>();
  37         internal Dictionary<GridInformationKey, GridInformationValue> PositionProperties
  38         {
  39             get { return m_PositionProperties; }
  40         }
  41
  42         [SerializeField]
  43         [HideInInspector]
  44         private List<GridInformationKey> m_PositionIntKeys = new List<GridInformationKey>();
  45
  46         [SerializeField]
  47         [HideInInspector]
  48         private List<int> m_PositionIntValues = new List<int>();
  49
  50         [SerializeField]
  51         [HideInInspector]
  52         private List<GridInformationKey> m_PositionStringKeys = new List<GridInformationKey>();
  53
  54         [SerializeField]
  55         [HideInInspector]
  56         private List<String> m_PositionStringValues = new List<String>();
  57
  58         [SerializeField]
  59         [HideInInspector]
  60         private List<GridInformationKey> m_PositionFloatKeys = new List<GridInformationKey>();
  61
  62         [SerializeField]
  63         [HideInInspector]
  64         private List<float> m_PositionFloatValues = new List<float>();
  65
  66         [SerializeField]
  67         [HideInInspector]
  68         private List<GridInformationKey> m_PositionDoubleKeys = new List<GridInformationKey>();
  69
  70         [SerializeField]
  71         [HideInInspector]
  72         private List<Double> m_PositionDoubleValues = new List<Double>();
  73
  74         [SerializeField]
  75         [HideInInspector]
  76         private List<GridInformationKey> m_PositionObjectKeys = new List<GridInformationKey>();
  77
  78         [SerializeField]
  79         [HideInInspector]
  80         private List<Object> m_PositionObjectValues = new List<Object>();
  81
  82         [SerializeField]
  83         [HideInInspector]
  84         private List<GridInformationKey> m_PositionColorKeys = new List<GridInformationKey>();
  85
  86         [SerializeField]
  87         [HideInInspector]
  88         private List<Color> m_PositionColorValues = new List<Color>();
  89
  90         void ISerializationCallbackReceiver.OnBeforeSerialize()
  91         {
  92             Grid grid = GetComponentInParent<Grid>();
  93             if (grid == null)
  94                 return;
  95
  96             m_PositionIntKeys.Clear();
  97             m_PositionIntValues.Clear();
  98             m_PositionStringKeys.Clear();
  99             m_PositionStringValues.Clear();
 100             m_PositionFloatKeys.Clear();
 101             m_PositionFloatValues.Clear();
 102             m_PositionDoubleKeys.Clear();
 103             m_PositionDoubleValues.Clear();
 104             m_PositionObjectKeys.Clear();
 105             m_PositionObjectValues.Clear();
 106             m_PositionColorKeys.Clear();
 107             m_PositionColorValues.Clear();
 108
 109             foreach (var kvp in m_PositionProperties)
 110             {
 111                 switch (kvp.Value.type)
 112                 {
 113                     case GridInformationType.Integer:
 114                         m_PositionIntKeys.Add(kvp.Key);
 115                         m_PositionIntValues.Add((int)kvp.Value.data);
 116                         break;
 117                     case GridInformationType.String:
 118                         m_PositionStringKeys.Add(kvp.Key);
 119                         m_PositionStringValues.Add(kvp.Value.data as String);
 120                         break;
 121                     case GridInformationType.Float:
 122                         m_PositionFloatKeys.Add(kvp.Key);
 123                         m_PositionFloatValues.Add((float)kvp.Value.data);
 124                         break;
 125                     case GridInformationType.Double:
 126                         m_PositionDoubleKeys.Add(kvp.Key);
 127                         m_PositionDoubleValues.Add((double)kvp.Value.data);
 128                         break;
 129                     case GridInformationType.Color:
 130                         m_PositionColorKeys.Add(kvp.Key);
 131                         m_PositionColorValues.Add((Color)kvp.Value.data);
 132                         break;
 133                     default:
 134                         m_PositionObjectKeys.Add(kvp.Key);
 135                         m_PositionObjectValues.Add(kvp.Value.data as Object);
 136                         break;
 137                 }
 138             }
 139         }
 140
 141         void ISerializationCallbackReceiver.OnAfterDeserialize()
 142         {
 143             m_PositionProperties.Clear();
 144             for (int i = 0; i != Math.Min(m_PositionIntKeys.Count, m_PositionIntValues.Count); i++)
 145             {
 146                 GridInformationValue positionValue;
 147                 positionValue.type = GridInformationType.Integer;
 148                 positionValue.data = m_PositionIntValues[i];
 149                 m_PositionProperties.Add(m_PositionIntKeys[i], positionValue);
 150             }
 151             for (int i = 0; i != Math.Min(m_PositionStringKeys.Count, m_PositionStringValues.Count); i++)
 152             {
 153                 GridInformationValue positionValue;
 154                 positionValue.type = GridInformationType.String;
 155                 positionValue.data = m_PositionStringValues[i];
 156                 m_PositionProperties.Add(m_PositionStringKeys[i], positionValue);
 157             }
 158             for (int i = 0; i != Math.Min(m_PositionFloatKeys.Count, m_PositionFloatValues.Count); i++)
 159             {
 160                 GridInformationValue positionValue;
 161                 positionValue.type = GridInformationType.Float;
 162                 positionValue.data = m_PositionFloatValues[i];
 163                 m_PositionProperties.Add(m_PositionFloatKeys[i], positionValue);
 164             }
 165             for (int i = 0; i != Math.Min(m_PositionDoubleKeys.Count, m_PositionDoubleValues.Count); i++)
 166             {
 167                 GridInformationValue positionValue;
 168                 positionValue.type = GridInformationType.Double;
 169                 positionValue.data = m_PositionDoubleValues[i];
 170                 m_PositionProperties.Add(m_PositionDoubleKeys[i], positionValue);
 171             }
 172             for (int i = 0; i != Math.Min(m_PositionObjectKeys.Count, m_PositionObjectValues.Count); i++)
 173             {
 174                 GridInformationValue positionValue;
 175                 positionValue.type = GridInformationType.UnityObject;
 176                 positionValue.data = m_PositionObjectValues[i];
 177                 m_PositionProperties.Add(m_PositionObjectKeys[i], positionValue);
 178             }
 179             for (int i = 0; i != Math.Min(m_PositionColorKeys.Count, m_PositionColorValues.Count); i++)
 180             {
 181                 GridInformationValue positionValue;
 182                 positionValue.type = GridInformationType.Color;
 183                 positionValue.data = m_PositionColorValues[i];
 184                 m_PositionProperties.Add(m_PositionColorKeys[i], positionValue);
 185             }
 186         }
 187
 188         public bool SetPositionProperty<T>(Vector3Int position, String name, T positionProperty)
 189         {
 190             throw new NotImplementedException("Storing this type is not accepted in GridInformation");
 191         }
 192
 193         public bool SetPositionProperty(Vector3Int position, String name, int positionProperty)
 194         {
 195             return SetPositionProperty(position, name, GridInformationType.Integer, positionProperty);
 196         }
 197
 198         public bool SetPositionProperty(Vector3Int position, String name, string positionProperty)
 199         {
 200             return SetPositionProperty(position, name, GridInformationType.String, positionProperty);
 201         }
 202
 203         public bool SetPositionProperty(Vector3Int position, String name, float positionProperty)
 204         {
 205             return SetPositionProperty(position, name, GridInformationType.Float, positionProperty);
 206         }
 207
 208         public bool SetPositionProperty(Vector3Int position, String name, double positionProperty)
 209         {
 210             return SetPositionProperty(position, name, GridInformationType.Double, positionProperty);
 211         }
 212
 213         public bool SetPositionProperty(Vector3Int position, String name, UnityEngine.Object positionProperty)
 214         {
 215             return SetPositionProperty(position, name, GridInformationType.UnityObject, positionProperty);
 216         }
 217
 218         public bool SetPositionProperty(Vector3Int position, String name, Color positionProperty)
 219         {
 220             return SetPositionProperty(position, name, GridInformationType.Color, positionProperty);
 221         }
 222
 223         private bool SetPositionProperty(Vector3Int position, String name, GridInformationType dataType, System.Object positionProperty)
 224         {
 225             Grid grid = GetComponentInParent<Grid>();
 226             if (grid != null && positionProperty != null)
 227             {
 228                 GridInformationKey positionKey;
 229                 positionKey.position = position;
 230                 positionKey.name = name;
 231
 232                 GridInformationValue positionValue;
 233                 positionValue.type = dataType;
 234                 positionValue.data = positionProperty;
 235
 236                     m_PositionProperties[positionKey] = positionValue;
 237                 return true;
 238             }
 239             return false;
 240         }
 241
 242         public T GetPositionProperty<T>(Vector3Int position, String name, T defaultValue) where T : UnityEngine.Object
 243         {
 244             GridInformationKey positionKey;
 245             positionKey.position = position;
 246             positionKey.name = name;
 247
 248             GridInformationValue positionValue;
 249             if (m_PositionProperties.TryGetValue(positionKey, out positionValue))
 250             {
 251                 if (positionValue.type != GridInformationType.UnityObject)
 252                     throw new InvalidCastException("Value stored in GridInformation is not of the right type");
 253                 return positionValue.data as T;
 254             }
 255             return defaultValue;
 256         }
 257
 258         public int GetPositionProperty(Vector3Int position, String name, int defaultValue)
 259         {
 260             GridInformationKey positionKey;
 261             positionKey.position = position;
 262             positionKey.name = name;
 263
 264             GridInformationValue positionValue;
 265             if (m_PositionProperties.TryGetValue(positionKey, out positionValue))
 266             {
 267                 if (positionValue.type != GridInformationType.Integer)
 268                     throw new InvalidCastException("Value stored in GridInformation is not of the right type");
 269                 return (int)positionValue.data;
 270             }
 271             return defaultValue;
 272         }
 273
 274         public string GetPositionProperty(Vector3Int position, String name, string defaultValue)
 275         {
 276             GridInformationKey positionKey;
 277             positionKey.position = position;
 278             positionKey.name = name;
 279
 280             GridInformationValue positionValue;
 281             if (m_PositionProperties.TryGetValue(positionKey, out positionValue))
 282             {
 283                 if (positionValue.type != GridInformationType.String)
 284                     throw new InvalidCastException("Value stored in GridInformation is not of the right type");
 285                 return (string)positionValue.data;
 286             }
 287             return defaultValue;
 288         }
 289
 290         public float GetPositionProperty(Vector3Int position, String name, float defaultValue)
 291         {
 292             GridInformationKey positionKey;
 293             positionKey.position = position;
 294             positionKey.name = name;
 295
 296             GridInformationValue positionValue;
 297             if (m_PositionProperties.TryGetValue(positionKey, out positionValue))
 298             {
 299                 if (positionValue.type != GridInformationType.Float)
 300                     throw new InvalidCastException("Value stored in GridInformation is not of the right type");
 301                 return (float)positionValue.data;
 302             }
 303             return defaultValue;
 304         }
 305
 306         public double GetPositionProperty(Vector3Int position, String name, double defaultValue)
 307         {
 308             GridInformationKey positionKey;
 309             positionKey.position = position;
 310             positionKey.name = name;
 311
 312             GridInformationValue positionValue;
 313             if (m_PositionProperties.TryGetValue(positionKey, out positionValue))
 314             {
 315                 if (positionValue.type != GridInformationType.Double)
 316                     throw new InvalidCastException("Value stored in GridInformation is not of the right type");
 317                 return (double)positionValue.data;
 318             }
 319             return defaultValue;
 320         }
 321
 322         public Color GetPositionProperty(Vector3Int position, String name, Color defaultValue)
 323         {
 324             GridInformationKey positionKey;
 325             positionKey.position = position;
 326             positionKey.name = name;
 327
 328             GridInformationValue positionValue;
 329             if (m_PositionProperties.TryGetValue(positionKey, out positionValue))
 330             {
 331                 if (positionValue.type != GridInformationType.Color)
 332                     throw new InvalidCastException("Value stored in GridInformation is not of the right type");
 333                 return (Color)positionValue.data;
 334             }
 335             return defaultValue;
 336         }
 337
 338         public bool ErasePositionProperty(Vector3Int position, String name)
 339         {
 340             GridInformationKey positionKey;
 341             positionKey.position = position;
 342             positionKey.name = name;
 343             return m_PositionProperties.Remove(positionKey);
 344         }
 345
 346         public virtual void Reset()
 347         {
 348             m_PositionProperties.Clear();
 349         }
 350
 351         public Vector3Int[] GetAllPositions(string propertyName)
 352         {
 353             return m_PositionProperties.Keys.ToList().FindAll(x => x.name == propertyName).Select(x => x.position).ToArray();
 354         }
 355     }
 356 }