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.Linq;

namespace FeMM.Grasshopper.Components.Results
{
    public class BeamResultsComponent : GH_Component
    {
        /// <summary>
        /// Initializes a new instance of the BeamResultsComponent class.
        /// </summary>
        public BeamResultsComponent()
          : base("Beam Results", "BR", "The Beam analysis results", CategoryNameConstants.CATEGORY_FEMM, CategoryNameConstants.SUBCATEGORY_RESULTS)
        {
        }

        /// <summary>
        /// Registers all the input parameters for this component.
        /// </summary>
        protected override void RegisterInputParams(GH_InputParamManager pManager)
        {
            pManager.AddGenericParameter("Beam", "B", "The beam with results", GH_ParamAccess.item);
        }

        /// <summary>
        /// Registers all the output parameters for this component.
        /// </summary>
        protected override void RegisterOutputParams(GH_OutputParamManager pManager)
        {
            pManager.AddTextParameter("Load cases", "LC", "The load case name", GH_ParamAccess.list);
            pManager.AddNumberParameter("Stations", "ST", "The stations", GH_ParamAccess.list);
            pManager.AddNumberParameter("Steps", "S", "The steps", GH_ParamAccess.list);
            pManager.AddVectorParameter("Shear Forces", "SF", "The shear forces in the principal direction 1 and 2 (>0 st7)", GH_ParamAccess.list);
            pManager.AddVectorParameter("Bending Moments", "BM", "The bending moments in the principal direction 1 and 2 (>0 st7)", GH_ParamAccess.list);
            pManager.AddNumberParameter("Axial Forces", "AF", "The axial forces", GH_ParamAccess.list);
            pManager.AddNumberParameter("Torques", "T", "The torque (>0 st7)", GH_ParamAccess.list);
        }

        /// <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_BeamResults beam = null;
            if (!DA.GetData(0, ref beam))
                return;

            var lc = new List<string>();
            var stations = new List<double>();
            var steps = new List<double>();
            var shearForces = new List<Vector2d>();
            var bendingMoments = new List<Vector2d>();
            var axialForces = new List<double>();
            var torques = new List<double>();

            foreach (BeamForceResultModel result in beam.Value.Where(r => r is BeamForceResultModel).Cast<BeamForceResultModel>())
            {
                lc.Add(result.LoadCase.Name);
                steps.Add(result.StepNum);
                stations.Add(result.Station);
                shearForces.Add(result.ShearForce);
                bendingMoments.Add(result.BendingMoment);
                axialForces.Add(result.AxialForce);
                torques.Add(result.Torque);
            }

            DA.SetDataList(0, lc);
            DA.SetDataList(1, stations);
            DA.SetDataList(2, steps);
            DA.SetDataList(3, shearForces);
            DA.SetDataList(4, bendingMoments);
            DA.SetDataList(5, axialForces);
            DA.SetDataList(6, torques);
        }

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

        /// <summary>
        /// Gets the unique ID for this component. Do not change this ID after release.
        /// </summary>
        public override Guid ComponentGuid => new("ee27d853-f6a8-41e1-8e81-83d5b49ad1a2");
    }
}