using Godot;
using Nocturnis.DataStructures.DataPortGroups;
using Nocturnis.DataStructures.DataTypes;
using Nocturnis.Enigmos.Modules;
using Nocturnis.Enigmos.Modules.ComputationalModules;
using Nocturnis.GlobalManagement.Constants;
using Nocturnis.GlobalManagement.Providers;
using Skeleton.DataStructure;

namespace Enigmos.Modules.ComputationalModules.Binary;

public abstract partial class DotProductModule : BinaryComputationalModule, IPolymorphismModule, IDuplicateOutputModule
{
    private IDataInGroup? VectorInputGroup { get; set; }
    private IDataOutGroup? OutputGroup { get; set; }
    public HashSet<IDataPortGroup> ConfigurablePortGroups { get; set; } = new();

    public override void Init()
    {
        base.Init();
        this.DataOutInit("Output", 4);
        VectorInputGroup = GlobalProvider.DataStructureProvider!.NewDataInGroup(
            this,
            DataInPorts,
            "Vector Input Type",
            DataTypeConstant.BaseDataTypes.R2,
            DataTypeConstant.DataTypeOptions.VectorTypes
        );
        OutputGroup = GlobalProvider.DataStructureProvider.NewDataOutGroup(
            this,
            DataOutPorts,
            "",
            DataTypeConstant.BaseDataTypes.Real,
            DataTypeConstant.DataTypeOptions.ScalarTypes
        );
        ConfigurablePortGroups = new HashSet<IDataPortGroup> { VectorInputGroup };
        PostInit();
    }

    public override void Define()
    {
        (object, DataType) Func(CacheItem cache) =>
            GlobalProvider.PolymorphismProvider!.Dot(this.X(cache), this.Y(cache));
        this.Define(Func);
    }

    public void Inference()
    {
        if (GlobalProvider.DataTypeProvider!.IsComplexTensorType(VectorInputGroup!.SelectedType))
            OutputGroup!.SelectedType = DataTypeConstant.BaseDataTypes.Complex;
        else
            OutputGroup!.SelectedType = DataTypeConstant.BaseDataTypes.Real;
    }

}