ExposedParameter.cs 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428
  1. using System;
  2. using System.Collections.Generic;
  3. using UnityEngine;
  4. namespace GraphProcessor
  5. {
  6. [Serializable]
  7. public class ExposedParameter : ISerializationCallbackReceiver
  8. {
  9. [Serializable]
  10. public class Settings
  11. {
  12. public bool isHidden = false;
  13. public bool expanded = false;
  14. [SerializeField]
  15. internal string guid = null;
  16. public override bool Equals(object obj)
  17. {
  18. if (obj is Settings s && s != null)
  19. return Equals(s);
  20. else
  21. return false;
  22. }
  23. public virtual bool Equals(Settings param)
  24. => isHidden == param.isHidden && expanded == param.expanded;
  25. public override int GetHashCode() => base.GetHashCode();
  26. }
  27. public string guid; // unique id to keep track of the parameter
  28. public string name;
  29. [Obsolete("Use GetValueType()")]
  30. public string type;
  31. [Obsolete("Use value instead")]
  32. public SerializableObject serializedValue;
  33. public bool input = true;
  34. [SerializeReference]
  35. public Settings settings;
  36. public string shortType => GetValueType()?.Name;
  37. public void Initialize(string name, object value)
  38. {
  39. guid = Guid.NewGuid().ToString(); // Generated once and unique per parameter
  40. settings = CreateSettings();
  41. settings.guid = guid;
  42. this.name = name;
  43. this.value = value;
  44. }
  45. void ISerializationCallbackReceiver.OnAfterDeserialize()
  46. {
  47. // SerializeReference migration step:
  48. #pragma warning disable CS0618
  49. if (serializedValue?.value != null) // old serialization system can't serialize null values
  50. {
  51. value = serializedValue.value;
  52. Debug.Log("Migrated: " + serializedValue.value + " | " + serializedValue.serializedName);
  53. serializedValue.value = null;
  54. }
  55. #pragma warning restore CS0618
  56. }
  57. void ISerializationCallbackReceiver.OnBeforeSerialize() {}
  58. protected virtual Settings CreateSettings() => new Settings();
  59. public virtual object value { get; set; }
  60. public virtual Type GetValueType() => value == null ? typeof(object) : value.GetType();
  61. static Dictionary<Type, Type> exposedParameterTypeCache = new Dictionary<Type, Type>();
  62. internal ExposedParameter Migrate()
  63. {
  64. if (exposedParameterTypeCache.Count == 0)
  65. {
  66. foreach (var type in AppDomain.CurrentDomain.GetAllTypes())
  67. {
  68. if (type.IsSubclassOf(typeof(ExposedParameter)) && !type.IsAbstract)
  69. {
  70. var paramType = Activator.CreateInstance(type) as ExposedParameter;
  71. exposedParameterTypeCache[paramType.GetValueType()] = type;
  72. }
  73. }
  74. }
  75. #pragma warning disable CS0618 // Use of obsolete fields
  76. var oldType = Type.GetType(type);
  77. #pragma warning restore CS0618
  78. if (oldType == null || !exposedParameterTypeCache.TryGetValue(oldType, out var newParamType))
  79. return null;
  80. var newParam = Activator.CreateInstance(newParamType) as ExposedParameter;
  81. newParam.guid = guid;
  82. newParam.name = name;
  83. newParam.input = input;
  84. newParam.settings = newParam.CreateSettings();
  85. newParam.settings.guid = guid;
  86. return newParam;
  87. }
  88. public static bool operator ==(ExposedParameter param1, ExposedParameter param2)
  89. {
  90. if (ReferenceEquals(param1, null) && ReferenceEquals(param2, null))
  91. return true;
  92. if (ReferenceEquals(param1, param2))
  93. return true;
  94. if (ReferenceEquals(param1, null))
  95. return false;
  96. if (ReferenceEquals(param2, null))
  97. return false;
  98. return param1.Equals(param2);
  99. }
  100. public static bool operator !=(ExposedParameter param1, ExposedParameter param2) => !(param1 == param2);
  101. public bool Equals(ExposedParameter parameter) => guid == parameter.guid;
  102. public override bool Equals(object obj)
  103. {
  104. if ((obj == null) || !this.GetType().Equals(obj.GetType()))
  105. return false;
  106. else
  107. return Equals((ExposedParameter)obj);
  108. }
  109. public override int GetHashCode() => guid.GetHashCode();
  110. public ExposedParameter Clone()
  111. {
  112. var clonedParam = Activator.CreateInstance(GetType()) as ExposedParameter;
  113. clonedParam.guid = guid;
  114. clonedParam.name = name;
  115. clonedParam.input = input;
  116. clonedParam.settings = settings;
  117. clonedParam.value = value;
  118. return clonedParam;
  119. }
  120. }
  121. // Due to polymorphic constraints with [SerializeReference] we need to explicitly create a class for
  122. // every parameter type available in the graph (i.e. templating doesn't work)
  123. [System.Serializable]
  124. public class ColorParameter : ExposedParameter
  125. {
  126. public enum ColorMode
  127. {
  128. Default,
  129. HDR
  130. }
  131. [Serializable]
  132. public class ColorSettings : Settings
  133. {
  134. public ColorMode mode;
  135. public override bool Equals(Settings param)
  136. => base.Equals(param) && mode == ((ColorSettings)param).mode;
  137. }
  138. [SerializeField] Color val;
  139. public override object value { get => val; set => val = (Color)value; }
  140. protected override Settings CreateSettings() => new ColorSettings();
  141. }
  142. [System.Serializable]
  143. public class FloatParameter : ExposedParameter
  144. {
  145. public enum FloatMode
  146. {
  147. Default,
  148. Slider,
  149. }
  150. [Serializable]
  151. public class FloatSettings : Settings
  152. {
  153. public FloatMode mode;
  154. public float min = 0;
  155. public float max = 1;
  156. public override bool Equals(Settings param)
  157. => base.Equals(param) && mode == ((FloatSettings)param).mode && min == ((FloatSettings)param).min && max == ((FloatSettings)param).max;
  158. }
  159. [SerializeField] float val;
  160. public override object value { get => val; set => val = (float)value; }
  161. protected override Settings CreateSettings() => new FloatSettings();
  162. }
  163. [System.Serializable]
  164. public class Vector2Parameter : ExposedParameter
  165. {
  166. public enum Vector2Mode
  167. {
  168. Default,
  169. MinMaxSlider,
  170. }
  171. [Serializable]
  172. public class Vector2Settings : Settings
  173. {
  174. public Vector2Mode mode;
  175. public float min = 0;
  176. public float max = 1;
  177. public override bool Equals(Settings param)
  178. => base.Equals(param) && mode == ((Vector2Settings)param).mode && min == ((Vector2Settings)param).min && max == ((Vector2Settings)param).max;
  179. }
  180. [SerializeField] Vector2 val;
  181. public override object value { get => val; set => val = (Vector2)value; }
  182. protected override Settings CreateSettings() => new Vector2Settings();
  183. }
  184. [System.Serializable]
  185. public class Vector3Parameter : ExposedParameter
  186. {
  187. [SerializeField] Vector3 val;
  188. public override object value { get => val; set => val = (Vector3)value; }
  189. }
  190. [System.Serializable]
  191. public class Vector4Parameter : ExposedParameter
  192. {
  193. [SerializeField] Vector4 val;
  194. public override object value { get => val; set => val = (Vector4)value; }
  195. }
  196. [System.Serializable]
  197. public class IntParameter : ExposedParameter
  198. {
  199. public enum IntMode
  200. {
  201. Default,
  202. Slider,
  203. }
  204. [Serializable]
  205. public class IntSettings : Settings
  206. {
  207. public IntMode mode;
  208. public int min = 0;
  209. public int max = 10;
  210. public override bool Equals(Settings param)
  211. => base.Equals(param) && mode == ((IntSettings)param).mode && min == ((IntSettings)param).min && max == ((IntSettings)param).max;
  212. }
  213. [SerializeField] int val;
  214. public override object value { get => val; set => val = (int)value; }
  215. protected override Settings CreateSettings() => new IntSettings();
  216. }
  217. [System.Serializable]
  218. public class Vector2IntParameter : ExposedParameter
  219. {
  220. [SerializeField] Vector2Int val;
  221. public override object value { get => val; set => val = (Vector2Int)value; }
  222. }
  223. [System.Serializable]
  224. public class Vector3IntParameter : ExposedParameter
  225. {
  226. [SerializeField] Vector3Int val;
  227. public override object value { get => val; set => val = (Vector3Int)value; }
  228. }
  229. [System.Serializable]
  230. public class DoubleParameter : ExposedParameter
  231. {
  232. [SerializeField] Double val;
  233. public override object value { get => val; set => val = (Double)value; }
  234. }
  235. [System.Serializable]
  236. public class LongParameter : ExposedParameter
  237. {
  238. [SerializeField] long val;
  239. public override object value { get => val; set => val = (long)value; }
  240. }
  241. [System.Serializable]
  242. public class StringParameter : ExposedParameter
  243. {
  244. [SerializeField] string val;
  245. public override object value { get => val; set => val = (string)value; }
  246. public override Type GetValueType() => typeof(String);
  247. }
  248. [System.Serializable]
  249. public class RectParameter : ExposedParameter
  250. {
  251. [SerializeField] Rect val;
  252. public override object value { get => val; set => val = (Rect)value; }
  253. }
  254. [System.Serializable]
  255. public class RectIntParameter : ExposedParameter
  256. {
  257. [SerializeField] RectInt val;
  258. public override object value { get => val; set => val = (RectInt)value; }
  259. }
  260. [System.Serializable]
  261. public class BoundsParameter : ExposedParameter
  262. {
  263. [SerializeField] Bounds val;
  264. public override object value { get => val; set => val = (Bounds)value; }
  265. }
  266. [System.Serializable]
  267. public class BoundsIntParameter : ExposedParameter
  268. {
  269. [SerializeField] BoundsInt val;
  270. public override object value { get => val; set => val = (BoundsInt)value; }
  271. }
  272. [System.Serializable]
  273. public class AnimationCurveParameter : ExposedParameter
  274. {
  275. [SerializeField] AnimationCurve val;
  276. public override object value { get => val; set => val = (AnimationCurve)value; }
  277. public override Type GetValueType() => typeof(AnimationCurve);
  278. }
  279. [System.Serializable]
  280. public class GradientParameter : ExposedParameter
  281. {
  282. public enum GradientColorMode
  283. {
  284. Default,
  285. HDR,
  286. }
  287. [Serializable]
  288. public class GradientSettings : Settings
  289. {
  290. public GradientColorMode mode;
  291. public override bool Equals(Settings param)
  292. => base.Equals(param) && mode == ((GradientSettings)param).mode;
  293. }
  294. [SerializeField] Gradient val;
  295. [SerializeField, GradientUsage(true)] Gradient hdrVal;
  296. public override object value { get => val; set => val = (Gradient)value; }
  297. public override Type GetValueType() => typeof(Gradient);
  298. protected override Settings CreateSettings() => new GradientSettings();
  299. }
  300. [System.Serializable]
  301. public class GameObjectParameter : ExposedParameter
  302. {
  303. [SerializeField] GameObject val;
  304. public override object value { get => val; set => val = (GameObject)value; }
  305. public override Type GetValueType() => typeof(GameObject);
  306. }
  307. [System.Serializable]
  308. public class BoolParameter : ExposedParameter
  309. {
  310. [SerializeField] bool val;
  311. public override object value { get => val; set => val = (bool)value; }
  312. }
  313. [System.Serializable]
  314. public class Texture2DParameter : ExposedParameter
  315. {
  316. [SerializeField] Texture2D val;
  317. public override object value { get => val; set => val = (Texture2D)value; }
  318. public override Type GetValueType() => typeof(Texture2D);
  319. }
  320. [System.Serializable]
  321. public class RenderTextureParameter : ExposedParameter
  322. {
  323. [SerializeField] RenderTexture val;
  324. public override object value { get => val; set => val = (RenderTexture)value; }
  325. public override Type GetValueType() => typeof(RenderTexture);
  326. }
  327. [System.Serializable]
  328. public class MeshParameter : ExposedParameter
  329. {
  330. [SerializeField] Mesh val;
  331. public override object value { get => val; set => val = (Mesh)value; }
  332. public override Type GetValueType() => typeof(Mesh);
  333. }
  334. [System.Serializable]
  335. public class MaterialParameter : ExposedParameter
  336. {
  337. [SerializeField] Material val;
  338. public override object value { get => val; set => val = (Material)value; }
  339. public override Type GetValueType() => typeof(Material);
  340. }
  341. }