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 AdditionModule : BinaryComputationalModule, IPolymorphismModule
{
	private IDataPortGroup? InputGroup { 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 override double MaintenanceAlpha => 0.19572021d;
	public override double MaintenanceBeta => 0.20151779d;
	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, Input2 },
			"",
			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 input2)
	{
		IDataPackage res = GlobalProvider.PolymorphismProvider!.Add(input1, input2, OutputGroup!.SelectedType);
		foreach (IDataPort port in OutputGroup)
			(port as DataOutPort)!.ResultData.Assign(res, OutputGroup.SelectedType);
		
	}

	public void Inference() => InputGroup!.SelectedType = OutputGroup!.SelectedType;

	public override string GetDescription => GlobalProvider.EnigmosProvider.ModuleDescription<AdditionModule>();
}