using SAP2000v1;
using FeMM.Grasshopper.Helpers;
using Grasshopper;
using Grasshopper.Kernel;
using Grasshopper.Kernel.Data;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.Versioning;
namespace FeMM.Grasshopper.Components.SAPExtra
{
#if NETCOREAPP
[SupportedOSPlatform("windows")]
#endif
public class SAPGetFrameForceComponent : GH_Component
{
bool _run;
/// <summary>
/// Initializes a new instance of the SAPAreaLoaderComponent class.
/// </summary>
public SAPGetFrameForceComponent()
: base("SAP2000 Frame Force", "FFR", "Read the result for selected load case", CategoryNameConstants.CATEGORY_CHECKS, CategoryNameConstants.SUBCATEGORY_SAPEXTRA)
{
_run = false;
}
public override void CreateAttributes()
{
var attr = new ComponentAttributes.ComponentOneButtonAttributes(this, "Run");
attr.ButtonPressed += () =>
{
_run = true;
ExpireSolution(true);
};
m_attributes = attr;
}
/// <summary>
/// Registers all the input parameters for this component.
/// </summary>
protected override void RegisterInputParams(GH_InputParamManager pManager)
{
pManager.AddTextParameter("Model Path", "MP", "Sap2000 model path to open. If empty it will open an empty file", GH_ParamAccess.item, "");
pManager[0].Optional = true;
pManager.AddBooleanParameter("Attach to Instance", "AI", "Attach to instance. If true, 'model path' will be ignored", GH_ParamAccess.item, false);
pManager[1].Optional = true;
pManager.AddTextParameter("SapExe", "S", "sap2000 exe fullpath", GH_ParamAccess.item, "");
pManager[2].Optional = true;
pManager.AddTextParameter("Groups", "G", "Groups selected for output. If empty, all beams are exported", GH_ParamAccess.list, "ALL");
pManager[3].Optional = true;
pManager.AddTextParameter("Load cases", "LC", "Load cases selected for output. If empty, cases selected in SAP are exported", GH_ParamAccess.list);
pManager[4].Optional = true;
}
/// <summary>
/// Registers all the output parameters for this component.
/// </summary>
protected override void RegisterOutputParams(GH_OutputParamManager pManager)
{
pManager.AddTextParameter("Beam IDs", "BI", "The beam element IDs", GH_ParamAccess.list);
pManager.AddNumberParameter("Beam stations", "BS", "The beam element stations", GH_ParamAccess.tree);
pManager.AddTextParameter("Combos", "LC", "The load cases or the load combinations", GH_ParamAccess.tree);
pManager.AddNumberParameter("Axial Force", "N", "The beam element results", GH_ParamAccess.tree);
pManager.AddNumberParameter("Shear 2", "V2", "The beam element results", GH_ParamAccess.tree);
pManager.AddNumberParameter("Shear 3", "V3", "The beam element results", GH_ParamAccess.tree);
pManager.AddNumberParameter("Torsion", "T", "The beam element results", GH_ParamAccess.tree);
pManager.AddNumberParameter("Bending 2", "M2", "The beam element results", GH_ParamAccess.tree);
pManager.AddNumberParameter("Bending 3", "M3", "The beam element results", GH_ParamAccess.tree);
pManager.AddTextParameter("Step Type", "ST", "The beam element results", GH_ParamAccess.tree);
}
protected override void SolveInstance(IGH_DataAccess DA)
{
string sapModel = "";
bool attach = false;
string sapExe = "";
var groups = new List<string>();
var loadCases = new List<string>();
DA.GetData(0, ref sapModel);
DA.GetData(1, ref attach);
DA.GetData(2, ref sapExe);
DA.GetDataList(3, groups);
DA.GetDataList(4, loadCases);
if (_run)
{
if (!System.IO.File.Exists(sapModel) && !attach) // model path non valido, Se path modello non valido fa partire con stringa vuota (file vuoto)
{
AddRuntimeMessage(GH_RuntimeMessageLevel.Error, "Sap model path not valid, opening an empty file");
_run = false;
return;
}
ExtraSapHelper.InitializeModel(!attach, sapExe, sapModel, out cOAPI mySapObject, out cSapModel mySapModel, out cHelper myHelper);
if (mySapObject != null && mySapModel != null)
{
if (!mySapModel.GetModelIsLocked())
{
AddRuntimeMessage(GH_RuntimeMessageLevel.Error, "Please run the analysis");
_run = false;
return;
}
}
eUnits units = mySapModel.GetPresentUnits();
Message = units.ToString();
// Lista degli ID delle travi di cui richiedere i risultati
var eulerBeams = new HashSet<string>();
if (groups.Count > 0)
{
int groupNumber = 0;
string[] groupNames = [];
mySapModel.GroupDef.GetNameList(ref groupNumber, ref groupNames);
for (int i = 0; i < groups.Count; i++)
{
int numberItems = 0;
int[] objectType = [];
string[] objectName = [];
mySapModel.GroupDef.GetAssignments(groups[i], ref numberItems, ref objectType, ref objectName);
for (int j = 0; j < numberItems; j++)
{
// objectType == Frame object
if (objectType[j] == 2)
{
string sectProp = "";
string auto = "";
mySapModel.FrameObj.GetSection(objectName[j], ref sectProp, ref auto);
if (sectProp == "None")
continue;
eulerBeams.Add(objectName[j]);
}
}
}
}
else
{
int numberFrames = 0;
string[] frameNames = [];
mySapModel.FrameObj.GetNameList(ref numberFrames, ref frameNames);
for (int i = 0; i < numberFrames; i++)
{
string sectProp = "";
string auto = "";
mySapModel.FrameObj.GetSection(frameNames[i], ref sectProp, ref auto);
if (sectProp == "None")
continue;
eulerBeams.Add(frameNames[i]);
}
}
if (eulerBeams.Count == 0)
{
AddRuntimeMessage(GH_RuntimeMessageLevel.Error, "No beams in selected groups");
_run = false;
return;
}
int totalLoadPatterns = 0;
string[] loadPatternNames = [];
if (mySapModel.LoadPatterns.GetNameList(ref totalLoadPatterns, ref loadPatternNames) == 0)
{
if (mySapModel.DatabaseTables.SetLoadPatternsSelectedForDisplay(ref loadPatternNames) != 0)
{
AddRuntimeMessage(GH_RuntimeMessageLevel.Error, "Fail to set load patterns for output");
_run = false;
return;
}
}
else
{
AddRuntimeMessage(GH_RuntimeMessageLevel.Error, "Fail to set load patterns for output");
_run = false;
return;
}
if (loadCases.Count != 0)
{
//deselect all cases and combos
mySapModel.Results.Setup.DeselectAllCasesAndCombosForOutput();
//set cases selected for output
for (int i = 0; i < loadCases.Count; i++)
{
if (mySapModel.Results.Setup.SetComboSelectedForOutput($"{loadCases[i]}") != 0)
if (mySapModel.Results.Setup.SetCaseSelectedForOutput($"{loadCases[i]}") != 0)
AddRuntimeMessage(GH_RuntimeMessageLevel.Warning, $"Fail to set case: {loadCases[i]}");
}
}
int beamCount = eulerBeams.Count;
int[] number_result = new int[beamCount];
string[][] loadCase = new string[beamCount][];
double[][] objStation = new double[beamCount][];
double[][] elementStation = new double[beamCount][];
double[][] N = new double[beamCount][];
double[][] V2 = new double[beamCount][];
double[][] V3 = new double[beamCount][];
double[][] T = new double[beamCount][];
double[][] M2 = new double[beamCount][];
double[][] M3 = new double[beamCount][];
string[][] stepTypes = new string[beamCount][];
for (int b = 0; b < eulerBeams.Count; b++)
{
string beamName = eulerBeams.ElementAt(b);
string[] obj = [];
objStation[b] = [];
string[] element = [];
elementStation[b] = [];
loadCase[b] = [];
stepTypes[b] = [];
double[] step_num = [];
N[b] = [];
V2[b] = [];
V3[b] = [];
T[b] = [];
M2[b] = [];
M3[b] = [];
try
{
if (mySapModel.Results.FrameForce(beamName, eItemTypeElm.ObjectElm, ref number_result[b], ref obj, ref objStation[b], ref element, ref elementStation[b],
ref loadCase[b], ref stepTypes[b], ref step_num, ref N[b], ref V2[b], ref V3[b], ref T[b], ref M2[b], ref M3[b]) != 0)
{
AddRuntimeMessage(GH_RuntimeMessageLevel.Warning, $"Fail to read results for beam {beamName}");
}
}
catch (Exception ex)
{
AddRuntimeMessage(GH_RuntimeMessageLevel.Warning, $"Fail to read results for beam {beamName}: {ex.Message}");
}
}
// NOTA: conversione tra SAP e Model delle forze
// N => N V2 => V2 M2 => M2
// T => T V3 => V1 M3 => M1
double[][] stationPositions = new double[eulerBeams.Count][];
for (int b = 0; b < eulerBeams.Count; b++)
{
int numberOfStations = 0;
double distance = 0;
double length = objStation[b].Last();
for (int l = 0; l < objStation[b].Length; l++)
{
numberOfStations++;
if (Math.Abs(objStation[b][l] - length) < 0.001)
break;
}
stationPositions[b] = new double[numberOfStations];
for (int j = 0; j < numberOfStations; j++)
{
if (j == 0)
stationPositions[b][j] = objStation[b][j];
else
{
if (objStation[b][j] == 0)
stationPositions[b][j] = distance;
else
{
distance += objStation[b][j] - objStation[b][j - 1];
stationPositions[b][j] = distance;
}
}
}
};
var stationPositionsDataTree = new DataTree<double>();
for (int i = 0; i < stationPositions.GetLength(0); i++)
stationPositionsDataTree.AddRange(stationPositions[i], new GH_Path(i));
var loadCaseDataTree = new DataTree<string>();
for (int i = 0; i < loadCase.GetLength(0); i++)
loadCaseDataTree.AddRange(loadCase[i], new GH_Path(i));
var NDataTree = new DataTree<double>();
for (int i = 0; i < N.GetLength(0); i++)
NDataTree.AddRange(N[i], new GH_Path(i));
var V2DataTree = new DataTree<double>();
for (int i = 0; i < V2.GetLength(0); i++)
V2DataTree.AddRange(V2[i], new GH_Path(i));
var V3DataTree = new DataTree<double>();
for (int i = 0; i < V3.GetLength(0); i++)
V3DataTree.AddRange(V3[i], new GH_Path(i));
var TDataTree = new DataTree<double>();
for (int i = 0; i < T.GetLength(0); i++)
TDataTree.AddRange(T[i], new GH_Path(i));
var M2DataTree = new DataTree<double>();
for (int i = 0; i < M2.GetLength(0); i++)
M2DataTree.AddRange(M2[i], new GH_Path(i));
var M3DataTree = new DataTree<double>();
for (int i = 0; i < M3.GetLength(0); i++)
M3DataTree.AddRange(M3[i], new GH_Path(i));
var stepTypeDataTree = new DataTree<string>();
for (int i = 0; i < stepTypes.GetLength(0); i++)
stepTypeDataTree.AddRange(stepTypes[i], new GH_Path(i));
DA.SetDataList(0, eulerBeams.ToList());
DA.SetDataTree(1, stationPositionsDataTree);
DA.SetDataTree(2, loadCaseDataTree);
DA.SetDataTree(3, NDataTree);
DA.SetDataTree(4, V2DataTree);
DA.SetDataTree(5, V3DataTree);
DA.SetDataTree(6, TDataTree);
DA.SetDataTree(7, M2DataTree);
DA.SetDataTree(8, M3DataTree);
DA.SetDataTree(9, stepTypeDataTree);
}
_run = false;
}
protected override System.Drawing.Bitmap Icon => Properties.Resources.SAPBeamLoaderIcon;
public override Guid ComponentGuid => new("6c193c77-0e61-4f82-9182-b2735d0e689b");
}
}