Add Support for Groups

Etabs_Adapter/CRUD/Create/Bar.cs

@@ -108,7 +108,7 @@ private bool CreateObject(Bar bhBar)
 
             bhBar.SetAdapterId(etabsIdFragment);
 
-            return SetObject(bhBar);
+            return SetObject(bhBar) && SetGroup(bhBar);
         }
 
         /***************************************************/
@@ -215,6 +215,48 @@ private bool SetObject(Bar bhBar)
 
         /***************************************************/
 
+        private bool SetGroup(Bar bhBar)
+        {
+            int ret = 0;
+            /* Get the ETABS name of the Bar */
+            string name = GetAdapterId<string>(bhBar);
+
+            try {
+                /* Get the list of unique groupNames assigned to the BHoM Bar */
+                List<string> groupNames = bhBar.Tags.Distinct().ToList();
+                /* Get the list of existing group names in the ETABS model */
+                int modelNumGroups = 0;
+                string[] modelGroupNames = null;
+                m_model.GroupDef.GetNameList(ref modelNumGroups, ref modelGroupNames);
+
+                /* Create any groups that do not already exist in the ETABS model */
+                foreach (string groupName in groupNames)
+                {
+                    if (!modelGroupNames.Contains(groupName))
+                    {
+                        ret = m_model.GroupDef.SetGroup_1(groupName);
+                        if (ret != 0)
+                        {
+                            Engine.Base.Compute.RecordError("Could not create the Group <" + groupName + "> assigned to the Bar. Group not created.");
+                            return false;
+                        }
+                    }
+                }
+
+                /* Assign the Bar to each group in the list */
+                groupNames.ToList().ForEach(groupName => m_model.FrameObj.SetGroupAssign(name, groupName));
+                } 
+            catch (Exception e)
+                {
+                    Engine.Base.Compute.RecordError("Could not assign input groups to the bar. Groups not assigned.");
+                    return false;
+                }
+
+            return true;
+        }
+
+        /***************************************************/
+
 #if Debug16 || Release16
 
         [Description("Returns a bar where the endpoints have been flipped without cloning the object")]

Etabs_Adapter/CRUD/Create/Link.cs

@@ -20,14 +20,15 @@
  * along with this code. If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.      
  */
 
-using System.Collections.Generic;
-using System.Linq;
 using BH.Engine.Adapter;
+using BH.Engine.Adapters.ETABS;
+using BH.Engine.Structure;
 using BH.oM.Adapters.ETABS;
-using BH.oM.Structure.Elements;
 using BH.oM.Structure.Constraints;
-using BH.Engine.Structure;
-using BH.Engine.Adapters.ETABS;
+using BH.oM.Structure.Elements;
+using System;
+using System.Collections.Generic;
+using System.Linq;
 
 
 namespace BH.Adapter.ETABS
@@ -70,6 +71,8 @@ private bool CreateObject(RigidLink bhLink)
             multiId.Id = linkIds;
             bhLink.SetAdapterId(multiId);
 
+            SetGroup(bhLink);
+
             return success;
         }
 
@@ -106,6 +109,51 @@ private bool CreateObject(LinkConstraint bhLinkConstraint)
         }
 
         /***************************************************/
+
+        private bool SetGroup(RigidLink bhLink)
+        {
+            int ret = 0;
+            /* Get the ETABS names of all the Links */
+            List<string> names = ((ETABSId)bhLink.Fragments[0]).Id as List<string>;
+
+            try
+            {
+                /* Get the list of unique groupNames assigned to the BHoM Link */
+                List<string> groupNames = bhLink.Tags.Distinct().ToList();
+                /* Get the list of existing group names in the ETABS model */
+                int modelNumGroups = 0;
+                string[] modelGroupNames = null;
+                m_model.GroupDef.GetNameList(ref modelNumGroups, ref modelGroupNames);
+
+                /* Create any groups that do not already exist in the ETABS model */
+                foreach (string groupName in groupNames)
+                {
+                    if (!modelGroupNames.Contains(groupName))
+                    {
+                        ret = m_model.GroupDef.SetGroup_1(groupName);
+                        if (ret != 0)
+                        {
+                            Engine.Base.Compute.RecordError("Could not create the Group <" + groupName + "> assigned to the Link. Group not created.");
+                            return false;
+                        }
+                    }
+                }
+
+                /* Assign the Links to each group in the list */
+                foreach (string name in names) { 
+                    groupNames.ToList().ForEach(groupName => m_model.LinkObj.SetGroupAssign(name, groupName));
+                }
+            }
+            catch (Exception e)
+            {
+                Engine.Base.Compute.RecordError("Could not assign input groups to the link. Groups not assigned.");
+                return false;
+            }
+
+            return true;
+        }
+
+        /***************************************************/
     }
 }
 

Etabs_Adapter/CRUD/Create/Node.cs

@@ -20,15 +20,16 @@
  * along with this code. If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.      
  */
 
-using System.Collections.Generic;
-using System.Linq;
 using BH.Engine.Adapter;
-using BH.oM.Adapters.ETABS;
-using BH.oM.Structure.Elements;
-using BH.Engine.Structure;
 using BH.Engine.Adapters.ETABS;
 using BH.Engine.Geometry;
+using BH.Engine.Structure;
+using BH.oM.Adapters.ETABS;
 using BH.oM.Geometry;
+using BH.oM.Structure.Elements;
+using System;
+using System.Collections.Generic;
+using System.Linq;
 
 namespace BH.Adapter.ETABS
 {
@@ -69,6 +70,7 @@ private bool CreateObject(Node bhNode)
 
                 bhNode.SetAdapterId(etabsid);
                 SetObject(bhNode, name);
+                SetGroup(bhNode);
             }
 
             return true;
@@ -106,6 +108,49 @@ private bool SetObject(Node bhNode, string name)
         }
 
         /***************************************************/
+
+        private bool SetGroup(Node bhNode)
+        {
+            int ret = 0;
+            /* Get the ETABS name of the Node */
+            string name = GetAdapterId<string>(bhNode);
+
+            try
+            {
+                /* Get the list of unique groupNames assigned to the BHoM Node */
+                List<string> groupNames = bhNode.Tags.Distinct().ToList();
+                /* Get the list of existing group names in the ETABS model */
+                int modelNumGroups = 0;
+                string[] modelGroupNames = null;
+                m_model.GroupDef.GetNameList(ref modelNumGroups, ref modelGroupNames);
+
+                /* Create any groups that do not already exist in the ETABS model */
+                foreach (string groupName in groupNames)
+                {
+                    if (!modelGroupNames.Contains(groupName))
+                    {
+                        ret = m_model.GroupDef.SetGroup_1(groupName);
+                        if (ret != 0)
+                        {
+                            Engine.Base.Compute.RecordError("Could not create the Group <" + groupName + "> assigned to the Node. Group not created.");
+                            return false;
+                        }
+                    }
+                }
+
+                /* Assign the Node to each group in the list */
+                groupNames.ToList().ForEach(groupName => m_model.PointObj.SetGroupAssign(name, groupName));
+            }
+            catch (Exception e)
+            {
+                Engine.Base.Compute.RecordError("Could not assign input groups to the node. Groups not assigned.");
+                return false;
+            }
+
+            return true;
+        }
+
+        /***************************************************/
     }
 }
 

Etabs_Adapter/CRUD/Create/Opening.cs

@@ -20,18 +20,19 @@
  * along with this code. If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.      
  */
 
-using System.Collections.Generic;
-using System.Linq;
 using BH.Engine.Adapter;
-using BH.oM.Adapters.ETABS;
-using BH.oM.Structure.Elements;
-using BH.Engine.Structure;
+using BH.Engine.Adapters.ETABS;
 using BH.Engine.Geometry;
 using BH.Engine.Spatial;
-using BH.Engine.Adapters.ETABS;
+using BH.Engine.Structure;
+using BH.oM.Adapters.ETABS;
 using BH.oM.Adapters.ETABS.Elements;
-using BH.oM.Geometry;
 using BH.oM.Analytical.Elements;
+using BH.oM.Geometry;
+using BH.oM.Structure.Elements;
+using System;
+using System.Collections.Generic;
+using System.Linq;
 
 
 namespace BH.Adapter.ETABS
@@ -105,10 +106,56 @@ private bool CreateObject(Opening bhOpening)
 
             m_model.AreaObj.SetOpening(openingName, true);
 
+            //Set Groups Assignment
+            SetGroup(bhOpening);
+
             return success;
         }
 
         /***************************************************/
 
+        private bool SetGroup(Opening bhOpening)
+        {
+            int ret = 0;
+            /* Get the ETABS name of the Opening */
+            string name = GetAdapterId<string>(bhOpening);
+
+            try
+            {
+                /* Get the list of unique groupNames assigned to the BHoM Opening */
+                List<string> groupNames = bhOpening.Tags.Distinct().ToList();
+                /* Get the list of existing group names in the ETABS model */
+                int modelNumGroups = 0;
+                string[] modelGroupNames = null;
+                m_model.GroupDef.GetNameList(ref modelNumGroups, ref modelGroupNames);
+
+                /* Create any groups that do not already exist in the ETABS model */
+                foreach (string groupName in groupNames)
+                {
+                    if (!modelGroupNames.Contains(groupName))
+                    {
+                        ret = m_model.GroupDef.SetGroup_1(groupName);
+                        if (ret != 0)
+                        {
+                            Engine.Base.Compute.RecordError("Could not create the Group <" + groupName + "> assigned to the Opening. Group not created.");
+                            return false;
+                        }
+                    }
+                }
+
+                /* Assign the Opening to each group in the list */
+                groupNames.ToList().ForEach(groupName => m_model.AreaObj.SetGroupAssign(name, groupName));
+            }
+            catch (Exception e)
+            {
+                Engine.Base.Compute.RecordError("Could not assign input groups to the openings. Groups not assigned.");
+                return false;
+            }
+
+            return true;
+        }
+
+        /***************************************************/
+
     }
 }

Etabs_Adapter/CRUD/Create/Panel.cs

@@ -20,17 +20,18 @@
  * along with this code. If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.      
  */
 
-using System.Collections.Generic;
-using System.Linq;
 using BH.Engine.Adapter;
-using BH.oM.Adapters.ETABS;
-using BH.oM.Structure.Elements;
-using BH.Engine.Structure;
+using BH.Engine.Adapters.ETABS;
 using BH.Engine.Geometry;
 using BH.Engine.Spatial;
-using BH.Engine.Adapters.ETABS;
+using BH.Engine.Structure;
+using BH.oM.Adapters.ETABS;
 using BH.oM.Adapters.ETABS.Elements;
 using BH.oM.Geometry;
+using BH.oM.Structure.Elements;
+using System;
+using System.Collections.Generic;
+using System.Linq;
 
 
 namespace BH.Adapter.ETABS
@@ -175,6 +176,10 @@ private bool CreateObject(Panel bhPanel)
             {
                 m_model.AreaObj.SetDiaphragm(name, diaphragm.Name);
             }
+
+            //Set Groups Assignment
+            SetGroup(bhPanel);
+
             return success;
         }
 
@@ -200,6 +205,49 @@ private static void NonLinearEdgesCheck(List<Edge> edges)
 
         /***************************************************/
 
+        private bool SetGroup(Panel bhPanel)
+        {
+            int ret = 0;
+            /* Get the ETABS name of the Panel */
+            string name = GetAdapterId<string>(bhPanel);
+
+            try
+            {
+                /* Get the list of unique groupNames assigned to the BHoM Panel */
+                List<string> groupNames = bhPanel.Tags.Distinct().ToList();
+                /* Get the list of existing group names in the ETABS model */
+                int modelNumGroups = 0;
+                string[] modelGroupNames = null;
+                m_model.GroupDef.GetNameList(ref modelNumGroups, ref modelGroupNames);
+
+                /* Create any groups that do not already exist in the ETABS model */
+                foreach (string groupName in groupNames)
+                {
+                    if (!modelGroupNames.Contains(groupName))
+                    {
+                        ret = m_model.GroupDef.SetGroup_1(groupName);
+                        if (ret != 0)
+                        {
+                            Engine.Base.Compute.RecordError("Could not create the Group <" + groupName + "> assigned to the Panel. Group not created.");
+                            return false;
+                        }
+                    }
+                }
+
+                /* Assign the Panel to each group in the list */
+                groupNames.ToList().ForEach(groupName => m_model.AreaObj.SetGroupAssign(name, groupName));
+            }
+            catch (Exception e)
+            {
+                Engine.Base.Compute.RecordError("Could not assign input groups to the panel. Groups not assigned.");
+                return false;
+            }
+
+            return true;
+        }
+
+        /***************************************************/
+
     }
 }