using FeMM.Common.Models;
using FeMM.Grasshopper.DataTypes.FeMM;
using FeMM.Grasshopper.Helpers;
using Grasshopper.Kernel;
using Rhino.Geometry;
using System;
using System.Collections.Generic;
using System.Runtime.Versioning;

namespace FeMM.Grasshopper.Components.Attributes
{
#if NETCOREAPP
    [SupportedOSPlatform("windows")]
#endif
    public class BeamReleasesComponent : GH_Component
    {
        /// <summary>
        /// Initializes a new instance of the BeamReleasesComponent class.
        /// </summary>
        public BeamReleasesComponent()
          : base("Beam Releases", "BR", "Beam ends releases \n 0 or false = released; Null = fixed; Number = Partial released (0 = released)", CategoryNameConstants.CATEGORY_FEMM, CategoryNameConstants.SUBCATEGORY_ATTRIBUTES)
        {
        }

        /// <summary>
        /// Registers all the input parameters for this component.
        /// </summary>
        protected override void RegisterInputParams(GH_InputParamManager pManager)
        {
            pManager.AddGenericParameter("Beam", "B", "The beam where to define the ends releases", GH_ParamAccess.item);
            int[] idx =
            [
                pManager.AddNumberParameter("Axial N1", "AN1", "Axial load release starting node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Axial N2", "AN2", "Axial load release ending node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Shear 1 N1", "S1N1", "Shear force axis 1 release starting node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Shear 2 N1", "S2N1", "Shear force axis 2 release starting node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Shear 1 N2", "S1N2", "Shear force axis 1 release ending node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Shear 2 N2", "S2N2", "Shear force axis 2 release ending node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Torsion N1", "TN1", "Torsion release starting node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Torsion N2", "TN2", "Torsion release ending node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Moment 1 N1", "M1N1", "Moment axis 1 release starting node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Moment 2 N1", "M2N1", "Moment axis 2 release starting node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Moment 1 N2", "M1N2", "Moment axis 1 release ending node", GH_ParamAccess.item),
                pManager.AddNumberParameter("Moment 2 N2", "M2N2", "Moment axis 2 release ending node", GH_ParamAccess.item)
            ];
            foreach (int i in idx)
                pManager[i].Optional = true;
        }

        /// <summary>
        /// Registers all the output parameters for this component.
        /// </summary>
        protected override void RegisterOutputParams(GH_OutputParamManager pManager)
        {
            pManager.AddGenericParameter("Beam", "B", "The modified beam", GH_ParamAccess.item);
        }

        /// <summary>
        /// This is the method that actually does the work.
        /// </summary>
        /// <param name="DA">The DA object is used to retrieve from inputs and store in outputs.</param>
        protected override void SolveInstance(IGH_DataAccess DA)
        {
            GH_Beam beam = null;
            double axialN1 = 0;
            double axialN2 = 0;
            double shear1N1 = 0;
            double shear2N1 = 0;
            double shear1N2 = 0;
            double shear2N2 = 0;
            double torsionN1 = 0;
            double torsionN2 = 0;
            double moment1N1 = 0;
            double moment2N1 = 0;
            double moment1N2 = 0;
            double moment2N2 = 0;

            if (!DA.GetData("Beam", ref beam))
                return;

            var has_values = new Dictionary<string, bool>()
            {
                { "axialN1", DA.GetData(1, ref axialN1) },
                { "axialN2", DA.GetData(2, ref axialN2) },
                { "shear1N1", DA.GetData(3, ref shear1N1) },
                { "shear2N1", DA.GetData(4, ref shear2N1) },
                { "shear1N2", DA.GetData(5, ref shear1N2) },
                { "shear2N2", DA.GetData(6, ref shear2N2) },
                { "torsionN1", DA.GetData(7, ref torsionN1) },
                { "torsionN2", DA.GetData(8, ref torsionN2) },
                { "moment1N1", DA.GetData(9, ref moment1N1) },
                { "moment2N1", DA.GetData(10, ref moment2N1) },
                { "moment1N2", DA.GetData(11, ref moment1N2) },
                { "moment2N2", DA.GetData(12, ref moment2N2) }
            };

            var new_beam = new GH_Beam(beam);

            var data = new BeamReleasesAttributeModel.AttributeData()
            {
                TransNode1 = [has_values["shear1N1"], has_values["shear2N1"], has_values["axialN1"]],
                TransN1Partial = new Vector3d(shear1N1, shear2N1, axialN1),
                TransNode2 = [has_values["shear1N2"], has_values["shear2N2"], has_values["axialN2"]],
                TransN2Partial = new Vector3d(shear1N2, shear2N2, axialN2),
                RotNode1 = [has_values["moment1N1"], has_values["moment2N1"], has_values["torsionN1"]],
                RotN1Partial = new Vector3d(moment1N1, moment2N1, torsionN1),
                RotNode2 = [has_values["moment1N2"], has_values["moment2N2"], has_values["torsionN2"]],
                RotN2Partial = new Vector3d(moment1N2, moment2N2, torsionN2)
            };
            var attr = new BeamReleasesAttributeModel(data);
            new_beam.Value.Attributes.Add(attr);

            DA.SetData("Beam", new_beam);
        }

        public override GH_Exposure Exposure => GH_Exposure.secondary;

        /// <summary>
        /// Provides an Icon for the component.
        /// </summary>
        protected override System.Drawing.Bitmap Icon => Properties.Resources.BeamReleasesIcon;

        /// <summary>
        /// Gets the unique ID for this component. Do not change this ID after release.
        /// </summary>
        public override Guid ComponentGuid => new("be84a0e3-5c7b-45eb-b834-84a6c39b7e6d");
    }
}