using Enigmos.Ports;
using Enigmos.Ports.DataPorts;
using Nocturnis.DataStructures;
using Nocturnis.Enigmos.Modules;
using TabulaSmaragdina;
using TabulaSmaragdina.Constants;

namespace Enigmos.Modules.ComputationalModules.Ternary;

public partial class SelectorModule : TernaryComputationalModule, IPolymorphismModule
{
    private DataOutPort? Output1 { get; set; }
    private DataOutPort? Output2 { get; set; }
    private DataOutPort? Output3 { get; set; }
    private IDataPortGroup? DataTypeGroup { get; set; }
    public HashSet<IDataPortGroup> ConfigurablePortGroups { get; set; } = new();
    public override IEnumerable<BasePort> Ports => base.Ports.Union(new[] { Output1, Output2, Output3 })!;
    public override void Init()
    {
        base.Init();
        Output1 = GetPort<DataOutPort>("Output1");
        Output2 = GetPort<DataOutPort>("Output2");
        Output3 = GetPort<DataOutPort>("Output3");
        DataTypeGroup = GlobalProvider.DataStructureProvider.NewDataPortGroup(
            this,
            new DataPort[] { Input2, Input3, Output1, Output2, Output3 },
            "Data Type",
            EnigmosConstant.DataPortTypes.Real,
            EnigmosConstant.DataPortTypes.AnyType
        );
        Input1.SetDataType(EnigmosConstant.DataPortTypes.Bit);
        PostInit();
    }

    protected override void Compute(IDataPackage input1, IDataPackage input2, IDataPackage input3)
    {

        Output1!.ResultData.Assign(input1.Bit ? input2 : input3, DataTypeGroup.SelectedType);
        Output2!.ResultData.Assign(input1.Bit ? input2 : input3, DataTypeGroup.SelectedType);
        Output3!.ResultData.Assign(input1.Bit ? input2 : input3, DataTypeGroup.SelectedType);
    }

    public void Inference()
    {
    }
}