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;
}