using System; using System.Collections.Generic; using UnityEditor.Graphing; using UnityEditor.ShaderGraph.Drawing.Slots; using UnityEditor.ShaderGraph.Internal; using UnityEngine; using UnityEngine.UIElements; namespace UnityEditor.ShaderGraph { [Serializable] class BooleanMaterialSlot : MaterialSlot, IMaterialSlotHasValue { [SerializeField] private bool m_Value; [SerializeField] private bool m_DefaultValue; public BooleanMaterialSlot() { } public BooleanMaterialSlot( int slotId, string displayName, string shaderOutputName, SlotType slotType, bool value, ShaderStageCapability stageCapability = ShaderStageCapability.All, bool hidden = false) : base(slotId, displayName, shaderOutputName, slotType, stageCapability, hidden) { m_DefaultValue = value; m_Value = value; } public override VisualElement InstantiateControl() { return new BooleanSlotControlView(this); } public bool defaultValue { get { return m_DefaultValue; } } public bool value { get { return m_Value; } set { m_Value = value; } } public override bool isDefaultValue => value.Equals(defaultValue); protected override string ConcreteSlotValueAsVariable() { return (value ? 1 : 0).ToString(); } public override void AddDefaultProperty(PropertyCollector properties, GenerationMode generationMode) { if (!generationMode.IsPreview()) return; var matOwner = owner as AbstractMaterialNode; if (matOwner == null) throw new Exception(string.Format("Slot {0} either has no owner, or the owner is not a {1}", this, typeof(AbstractMaterialNode))); var property = new BooleanShaderProperty() { overrideReferenceName = matOwner.GetVariableNameForSlot(id), generatePropertyBlock = false, value = value }; properties.AddShaderProperty(property); } public override SlotValueType valueType { get { return SlotValueType.Boolean; } } public override ConcreteSlotValueType concreteValueType { get { return ConcreteSlotValueType.Boolean; } } public override void GetPreviewProperties(List properties, string name) { var pp = new PreviewProperty(PropertyType.Boolean) { name = name, booleanValue = value }; properties.Add(pp); } public override void CopyValuesFrom(MaterialSlot foundSlot) { var slot = foundSlot as BooleanMaterialSlot; if (slot != null) value = slot.value; } public override void CopyDefaultValue(MaterialSlot other) { base.CopyDefaultValue(other); if (other is IMaterialSlotHasValue ms) { m_DefaultValue = ms.defaultValue; } } } }