using FeMM.Common.Models;
using FeMM.Grasshopper.DataTypes.FeMM;
using FeMM.Grasshopper.Helpers;
using Grasshopper.Kernel;
using System;
using System.Runtime.Versioning;

namespace FeMM.Grasshopper.Components.Attributes
{
#if NETCOREAPP
    [SupportedOSPlatform("windows")]
#endif
    public class BeamModifiersNameComponent : GH_Component
    {
        /// <summary>
        /// Initializes a new instance of the BeamReleasesComponent class.
        /// </summary>
        public BeamModifiersNameComponent()
          : base("Beam Modifiers", "BM", "Beam modifiers (only for SAP/ETABS)", 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 define the beam modifiers", GH_ParamAccess.item);
            int[] idx =
            [
                pManager.AddNumberParameter("Area", "A", "Area Modifier", GH_ParamAccess.item, 1.0),
                pManager.AddNumberParameter("Shear Area Dir 2", "S2", "Shear Area Direction 2 Modifier", GH_ParamAccess.item, 1.0),
                pManager.AddNumberParameter("Shear Area Dir 3", "S3", "Shear Area Direction 3 Modifier", GH_ParamAccess.item, 1.0),
                pManager.AddNumberParameter("Torsional Constant", "T", "Torsional Constant Modifier", GH_ParamAccess.item, 1.0),
                pManager.AddNumberParameter("Intertia Dir 2", "I2", "Moment Of Intertia Direction 2 Modifier", GH_ParamAccess.item, 1.0),
                pManager.AddNumberParameter("Intertia Dir 3", "I2", "Moment Of Intertia Direction 3 Modifier", GH_ParamAccess.item, 1.0),
                pManager.AddNumberParameter("Mass", "M", "Mass Modifier", GH_ParamAccess.item, 1.0),
                pManager.AddNumberParameter("Weight", "W", "Weight Modifier", GH_ParamAccess.item, 1.0),
            ];
            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 areaModifier = 1.0;
            double shearAreaDir2Modifier = 1.0;
            double shearAreaDir3Modifier = 1.0;
            double torsionalConstantModifier = 1.0;
            double momentOfIntertiaDir2Modifier = 1.0;
            double momentOfIntertiaDir3Modifier = 1.0;
            double massModifier = 1.0;
            double weightModifier = 1.0;

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

            DA.GetData(1, ref areaModifier);
            DA.GetData(2, ref shearAreaDir2Modifier);
            DA.GetData(3, ref shearAreaDir3Modifier);
            DA.GetData(4, ref torsionalConstantModifier);
            DA.GetData(5, ref momentOfIntertiaDir2Modifier);
            DA.GetData(6, ref momentOfIntertiaDir3Modifier);
            DA.GetData(7, ref massModifier);
            DA.GetData(8, ref weightModifier);

            var data = new BeamModifiersModel.AttributeData()
            {
                AreaModifier = areaModifier,
                ShearAreaDir2Modifier = shearAreaDir2Modifier,
                ShearAreaDir3Modifier = shearAreaDir3Modifier,
                TorsionalConstantModifier = torsionalConstantModifier,
                MomentOfIntertiaDir2Modifier = momentOfIntertiaDir2Modifier,
                MomentOfIntertiaDir3Modifier = momentOfIntertiaDir3Modifier,
                MassModifier = massModifier,
                WeightModifier = weightModifier,
            };

            var beamModifiersModel = new BeamModifiersModel(data);

            beam.Value.Attributes.Add(beamModifiersModel);
            DA.SetData(0, 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.BeamModifiersIcon;

        /// <summary>
        /// Gets the unique ID for this component. Do not change this ID after release.
        /// </summary>
        public override Guid ComponentGuid => new("84d709bf-a0e6-4ae3-b910-1d5b330d4016");
    }
}
303 files24 directories