misc: updated SetLampEnabledUnit to support IMultiInput

Author: jsm174

Editor/Descriptors/SetLampEnabledUnitDescriptor.cs

@@ -40,13 +40,14 @@ protected override void DefinedPort(IUnitPort port, UnitPortDescription desc)
 		{
 			base.DefinedPort(port, desc);
 
-			switch (port.key) {
-				case nameof(SetLampEnabledUnit.Id):
-					desc.summary = "The ID of the lamp to toggle.";
-					break;
-				case nameof(SetLampEnabledUnit.IsEnabled):
-					desc.summary = "Whether to turn on or off.";
-					break;
+			if (port.key == nameof(SetLampEnabledUnit.IsEnabled)) {
+				desc.summary = "Whether to turn on or off.";
+			}
+			else if (int.TryParse(port.key, out int id)) {
+				id += 1;
+
+				desc.label = $"Lamp ID {id}";
+				desc.summary = $"Lamp ID {id} of the lamp to toggle.";
 			}
 		}
 	}

Editor/Widgets/SetLampEnabledUnitWidget.cs

@@ -26,19 +26,28 @@ namespace VisualPinball.Unity.VisualScripting.Editor
 	[Widget(typeof(SetLampEnabledUnit))]
 	public sealed class SetLampEnabledUnitWidget : GleUnitWidget<SetLampEnabledUnit>
 	{
-		private VariableNameInspector _lampIdInspector;
-		private readonly Func<Metadata, VariableNameInspector> _setLampInspectorConstructor;
+		private readonly List<Func<Metadata, VariableNameInspector>> _lampIdInspectorConstructorList;
 
 		public SetLampEnabledUnitWidget(FlowCanvas canvas, SetLampEnabledUnit unit) : base(canvas, unit)
 		{
-			_setLampInspectorConstructor = meta => new VariableNameInspector(meta, GetNameSuggestions);
+			_lampIdInspectorConstructorList = new List<Func<Metadata, VariableNameInspector>>();
 		}
 
 		public override Inspector GetPortInspector(IUnitPort port, Metadata meta)
 		{
-			if (port == unit.Id) {
-				InspectorProvider.instance.Renew(ref _lampIdInspector, meta, _setLampInspectorConstructor);
-				return _lampIdInspector;
+			if (_lampIdInspectorConstructorList.Count() < unit.inputCount) {
+				for (var index = 0; index < unit.inputCount - _lampIdInspectorConstructorList.Count(); index++) {
+					_lampIdInspectorConstructorList.Add(meta => new VariableNameInspector(meta, GetNameSuggestions));
+				}
+			}
+
+			for (var index = 0; index < unit.inputCount; index++) {
+				if (unit.multiInputs[index] == port) {
+					VariableNameInspector lampIdInspector = new VariableNameInspector(meta, GetNameSuggestions);
+					InspectorProvider.instance.Renew(ref lampIdInspector, meta, _lampIdInspectorConstructorList[index]);
+
+					return lampIdInspector;
+				}
 			}
 
 			return base.GetPortInspector(port, meta);

Runtime/Nodes/Lamps/SetLampEnabledUnit.cs

@@ -14,6 +14,8 @@
 // You should have received a copy of the GNU General Public License
 // along with this program. If not, see <https://www.gnu.org/licenses/>.
 
+using System.Collections.Generic;
+using System.Collections.ObjectModel;
 using Unity.VisualScripting;
 using UnityEngine;
 
@@ -23,7 +25,7 @@ namespace VisualPinball.Unity.VisualScripting
 	[UnitTitle("Set Lamp (ID, on/off)")]
 	[UnitSurtitle("Gamelogic Engine")]
 	[UnitCategory("Visual Pinball")]
-	public class SetLampEnabledUnit : GleUnit
+	public class SetLampEnabledUnit : GleUnit, IMultiInputUnit
 	{
 		[DoNotSerialize]
 		[PortLabelHidden]
@@ -33,9 +35,19 @@ public class SetLampEnabledUnit : GleUnit
 		[PortLabelHidden]
 		public ControlOutput OutputTrigger;
 
+		[SerializeAs(nameof(inputCount))]
+		private int _inputCount = 1;
+
+		[DoNotSerialize]
+		[Inspectable, UnitHeaderInspectable("Lamp IDs")]
+		public int inputCount
+		{
+			get => _inputCount;
+			set => _inputCount = Mathf.Clamp(value, 1, 10);
+		}
+
 		[DoNotSerialize]
-		[PortLabel("Lamp ID")]
-		public ValueInput Id { get; private set; }
+		public ReadOnlyCollection<ValueInput> multiInputs { get; private set; }
 
 		[DoNotSerialize]
 		[PortLabel("Is Enabled")]
@@ -46,10 +58,19 @@ protected override void Definition()
 			InputTrigger = ControlInput(nameof(InputTrigger), Process);
 			OutputTrigger = ControlOutput(nameof(OutputTrigger));
 
-			Id = ValueInput(nameof(Id), string.Empty);
+			var _multiInputs = new List<ValueInput>();
+
+			multiInputs = _multiInputs.AsReadOnly();
+
+			for (var i = 0; i < inputCount; i++) {
+				var input = ValueInput(i.ToString(), string.Empty);
+				_multiInputs.Add(input);
+
+				Requirement(input, InputTrigger);
+			}
+
 			IsEnabled = ValueInput(nameof(IsEnabled), false);
 
-			Requirement(Id, InputTrigger);
 			Succession(InputTrigger, OutputTrigger);
 		}
 
@@ -60,10 +81,12 @@ private ControlOutput Process(Flow flow)
 				return OutputTrigger;
 			}
 
-			var id = flow.GetValue<string>(Id);
 			var isEnabled = flow.GetValue<bool>(IsEnabled);
 
-			Gle.SetLamp(id, isEnabled ? 255f : 0f);
+			foreach (var input in multiInputs) {
+				var lampId = flow.GetValue<string>(input);
+				Gle.SetLamp(lampId, isEnabled ? 255f : 0f);
+			}
 
 			return OutputTrigger;
 		}

Runtime/Nodes/Lamps/SetLampUnit.cs

@@ -62,8 +62,7 @@ protected override void Definition()
 
 			multiInputs = _multiInputs.AsReadOnly();
 
-			for (var i = 0; i < inputCount; i++)
-			{
+			for (var i = 0; i < inputCount; i++) {
 				var input = ValueInput(i.ToString(), string.Empty);
 				_multiInputs.Add(input);
 

Runtime/Nodes/Switches/AllSwitchesEnabledEventUnit.cs

@@ -66,8 +66,8 @@ protected override bool ShouldTrigger(Flow flow, SwitchEventArgs2 args)
 			}
 
 			var validSwitch = false;
-			foreach(var item in multiInputs) {
-				var swId = flow.GetValue<string>(item);
+			foreach(var input in multiInputs) {
+				var swId = flow.GetValue<string>(input);
 				if (swId == args.Id) {
 					validSwitch = true;
 				}