65 lines
2.4 KiB
C#
65 lines
2.4 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.Binary;
|
|
|
|
public partial class DotProductModule : BinaryComputationalModule, IPolymorphismModule
|
|
{
|
|
private IDataPortGroup? VectorInputGroup { get; set; }
|
|
private IDataPortGroup? OutputGroup { get; set; }
|
|
private DataOutPort? Output1 { get; set; }
|
|
private DataOutPort? Output2 { get; set; }
|
|
private DataOutPort? Output3 { get; set; }
|
|
private DataOutPort? Output4 { 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");
|
|
VectorInputGroup = GlobalProvider.DataStructureProvider!.NewDataPortGroup(
|
|
this,
|
|
new IDataPort[] { Input1, Input2 },
|
|
"Vector Input Type",
|
|
EnigmosConstant.DataPortTypes.R2,
|
|
EnigmosConstant.DataPortTypes.VectorTypes
|
|
);
|
|
OutputGroup = GlobalProvider.DataStructureProvider.NewDataPortGroup(
|
|
this,
|
|
new IDataPort[] { Output1, Output2, Output3, Output4 },
|
|
"",
|
|
EnigmosConstant.DataPortTypes.Real,
|
|
Array.Empty<StringName>()
|
|
);
|
|
ConfigurablePortGroups = new HashSet<IDataPortGroup> { VectorInputGroup };
|
|
PostInit();
|
|
}
|
|
|
|
protected override void Compute(IDataPackage input1, IDataPackage input2)
|
|
{
|
|
IDataPackage res = GlobalProvider.PolymorphismProvider!.Dot(input1, input2, VectorInputGroup!.SelectedType);
|
|
foreach (IDataPort port in OutputGroup!)
|
|
(port as DataOutPort)!.ResultData.Assign(res, OutputGroup.SelectedType);
|
|
|
|
}
|
|
|
|
|
|
public void Inference()
|
|
{
|
|
if (GlobalProvider.DataPackageTypeProvider!.IsComplexTensorType(VectorInputGroup!.SelectedType))
|
|
OutputGroup!.SelectedType = EnigmosConstant.DataPortTypes.Complex;
|
|
else
|
|
OutputGroup!.SelectedType = EnigmosConstant.DataPortTypes.Real;
|
|
}
|
|
|
|
} |