GradientSlotControlView.cs 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. using System;
  2. using UnityEditor.Graphing;
  3. using UnityEngine;
  4. using UnityEditor.ShaderGraph;
  5. using UnityEditor.ShaderGraph.Drawing.Controls;
  6. using Object = UnityEngine.Object;
  7. using UnityEditor.UIElements;
  8. using UnityEngine.UIElements;
  9. using UnityEngine.UIElements.StyleSheets;
  10. namespace UnityEditor.ShaderGraph.Drawing.Slots
  11. {
  12. class GradientSlotControlView : VisualElement
  13. {
  14. GradientInputMaterialSlot m_Slot;
  15. [SerializeField]
  16. GradientObject m_GradientObject;
  17. [SerializeField]
  18. SerializedObject m_SerializedObject;
  19. public GradientSlotControlView(GradientInputMaterialSlot slot)
  20. {
  21. m_Slot = slot;
  22. styleSheets.Add(Resources.Load<StyleSheet>("Styles/Controls/GradientSlotControlView"));
  23. m_GradientObject = ScriptableObject.CreateInstance<GradientObject>();
  24. m_GradientObject.gradient = new Gradient();
  25. m_SerializedObject = new SerializedObject(m_GradientObject);
  26. m_GradientObject.gradient.SetKeys(m_Slot.value.colorKeys, m_Slot.value.alphaKeys);
  27. m_GradientObject.gradient.mode = m_Slot.value.mode;
  28. var gradientField = new GradientField() { value = m_GradientObject.gradient };
  29. gradientField.RegisterValueChangedCallback(OnValueChanged);
  30. Add(gradientField);
  31. }
  32. void OnValueChanged(ChangeEvent<Gradient> evt)
  33. {
  34. m_SerializedObject.Update();
  35. if (!evt.newValue.Equals(m_Slot.value))
  36. {
  37. m_Slot.owner.owner.owner.RegisterCompleteObjectUndo("Change Gradient");
  38. m_GradientObject.gradient.SetKeys(evt.newValue.colorKeys, evt.newValue.alphaKeys);
  39. m_GradientObject.gradient.mode = evt.newValue.mode;
  40. m_SerializedObject.ApplyModifiedProperties();
  41. m_Slot.value = m_GradientObject.gradient;
  42. m_Slot.owner.Dirty(ModificationScope.Node);
  43. }
  44. }
  45. }
  46. }