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 ScalarDivisionModule : BinaryComputationalModule, IPolymorphismModule, IErrorHandlerModule
{
    private IDataPortGroup? ScalarInputGroup { get; set; }
    private IDataPortGroup? TensorInputGroup { 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.77852142d;
    public override double MaintenanceBeta => 0.9544432d;
    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");
        TensorInputGroup = GlobalProvider.DataStructureProvider!.NewDataPortGroup(
            this,
            new IDataPort[] { Input1 },
            "Tensor Input Type",
            EnigmosConstant.DataPortTypes.Real,
            EnigmosConstant.DataPortTypes.VectorTypes
        );
        ScalarInputGroup =GlobalProvider.DataStructureProvider.NewDataPortGroup(
            this,
            new IDataPort[] { Input2 },
            "Scalar 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> { ScalarInputGroup, TensorInputGroup };
        SelectedOption = 0;
        PostInit();
    }

    protected override void Compute(IDataPackage input1, IDataPackage input2)
    {
        try
        {
            IDataPackage res =
                GlobalProvider.PolymorphismProvider!.ScalarDiv(
                    input1,
                    input2,
                    TensorInputGroup!.SelectedType,
                    ScalarInputGroup!.SelectedType
                );
            foreach (IDataPort port in OutputGroup)
                (port as DataOutPort)!.ResultData.Assign(res, OutputGroup.SelectedType);
               
        }
        catch (Exception e)
        {
            ErrorHandler(e, SelectedOption);
        }

        
    }

    public void Inference()
    {
        if (GlobalProvider.DataPackageTypeProvider.IsComplexTensorType(ScalarInputGroup.SelectedType))
            OutputGroup.SelectedType = GlobalProvider.DataPackageTypeProvider.ComplexVersionOf(TensorInputGroup.SelectedType);
        else
            OutputGroup.SelectedType = TensorInputGroup.SelectedType;
    }

    public void ErrorHandler(Exception error, int idx)
    {
        switch (idx)
        {
            case 0:
                foreach (IDataPort port in OutputGroup!)
                    (port as DataOutPort)!.ResultData
                        .Assign(GlobalProvider.DataStructureProvider!.DefaultDataPackage, OutputGroup.SelectedType);
                break;
            case 1:
                foreach (IDataPort port in OutputGroup!)
                    (port as DataOutPort)!.ResultData
                        .Assign(GlobalProvider.DataStructureProvider!.DefaultDataPackage, OutputGroup.SelectedType);
                break;
            case 2:
                foreach (IDataPort port in OutputGroup!)
                    (port as DataOutPort)!.ResultData
                        .Assign(GlobalProvider.DataStructureProvider!.DefaultDataPackage, OutputGroup.SelectedType);
                break;
            default:
                return;
        }
    }

    public string[] HandlingOptions() =>
        new[]
        {
            "Reset Circuit State",
            "Return Previous Valid Value",
            "Return Default Value"
        };

    public int SelectedOption { get; set; }
}