Files
Enigmos/Modules/ComputationalModules/Unary/NegationModule.cs
2024-06-30 01:52:44 +08:00

58 lines
1.9 KiB
C#

using Enigmos.Ports;
using Enigmos.Ports.DataPorts;
using Godot;
using Nocturnis.DataStructures;
using Nocturnis.Enigmos.Modules;
using Nocturnis.Enigmos.Ports;
using TabulaSmaragdina;
using TabulaSmaragdina.Constants;
namespace Enigmos.Modules.ComputationalModules.Unary;
public partial class NegationModule : UnaryComputationalModule, IPolymorphismModule
{
private DataOutPort? Output1 { get; set; }
private DataOutPort? Output2 { get; set; }
private DataOutPort? Output3 { get; set; }
private DataOutPort? Output4 { get; set; }
private IDataPortGroup? InputGroup { get; set; }
private IDataPortGroup? OutputGroup { get; set; }
public override IEnumerable<BasePort> Ports => base.Ports.Union(new[] { Output1, Output2, Output3, Output4 })!;
public HashSet<IDataPortGroup> ConfigurablePortGroups { get; set; } = new();
public override void Init()
{
base.Init();
Output1 = GetPort<DataOutPort>("Output1");
Output2 = GetPort<DataOutPort>("Output2");
Output3 = GetPort<DataOutPort>("Output3");
Output4 = GetPort<DataOutPort>("Output4");
InputGroup = GlobalProvider.DataStructureProvider.NewDataPortGroup(
this,
new IDataPort[] {Input1},
"",
EnigmosConstant.DataPortTypes.Real,
Array.Empty<StringName>()
);
OutputGroup = GlobalProvider.DataStructureProvider.NewDataPortGroup(
this,
new IDataPort[] { Output1, Output2, Output3, Output4 },
"Output Port Type",
EnigmosConstant.DataPortTypes.Real,
EnigmosConstant.DataPortTypes.AnyTensor
);
ConfigurablePortGroups = new HashSet<IDataPortGroup> { OutputGroup };
PostInit();
}
protected override void Compute(IDataPackage input1)
{
IDataPackage res = GlobalProvider.PolymorphismProvider.Neg(input1, OutputGroup.SelectedType);
foreach (IDataPort port in OutputGroup)
(port as DataOutPort)!.ResultData.Assign(res, OutputGroup.SelectedType);
}
public void Inference() => InputGroup.SelectedType = OutputGroup.SelectedType;
}