Implement GetAreaOfPlates method and add Nodal Load functions in Load class; enhance tools with safe array input functions

Author: OpenStaad

openstaad/geometry.py

@@ -480,21 +480,17 @@ def DeletePlate(self,plate):
         self._geometry.DeletePlate(plate)
 
 
-    def _GetAreaOfPlate(self,plate):
-        safe_n1 = make_safe_array_double(1)
-        x = make_variant_vt_ref(safe_n1,  automation.VT_ARRAY |  automation.VT_R8)
-
-        self._geometry.GetAreaOfPlates([plate],x)
+    def GetAreaOfPlates(self,plates: list) -> list[float]:
 
-        return x.value[0][0]
-    
-    def GetAreaOfPlates(self,plates):
-        areas = []
-        for plate in plates:
-            area = self._GetAreaOfPlate(plate)
-            areas.append(area)
-
-        return areas
+        safe_palte_list = make_safe_array_int_input(plates)
+        plate_list = make_variant_vt_ref(safe_palte_list, automation.VT_ARRAY | automation.VT_I4)
+        
+        safe_area = make_safe_array_double(len(plates))
+        area = make_variant_vt_ref(safe_area,  automation.VT_ARRAY | automation.VT_R8)
+        
+        self._geometry.GetAreaOfPlates(plate_list,area)
+        
+        return area.value[0]
 
 
 

openstaad/load.py

@@ -21,7 +21,9 @@ def __init__(self):
             "GetLoadCaseTitle",
             "GetReferenceLoadCaseCount",
             "SetLoadActive",
-            "SetReferenceLoadActive"
+            "SetReferenceLoadActive",
+            "AddNodalLoad",
+            "ClearReferenceLoadCase"
         ]
 
         for function_name in self._functions:
@@ -156,6 +158,17 @@ def SetReferenceLoadActive(self,load_case:int):
         """
         return self._load.SetReferenceLoadActive(load_case)
 
+    def AddNodalLoad(self, nodes : list[int], fx : float, fy:float, fz:float, mx:float, my:float, mz:float):
+        def make_safe_array_long(array):
+            return automation._midlSAFEARRAY(ctypes.c_long).create(array)
+
+        safe_list = make_safe_array_long(nodes)
+        varNodeNo = make_variant_vt_ref(safe_list, automation.VT_ARRAY | automation.VT_I4)
+
+        self._load.AddNodalLoad(varNodeNo,fx, fy, fz, mx, my, mz)
+
+    def ClearReferenceLoadCase(self,reference_load_number: int):
+        self._load.ClearReferenceLoadCase(reference_load_number)
 
 
 

openstaad/root.py

@@ -18,7 +18,9 @@ def __init__(self):
             'GetSTAADFile',
             'GetSTAADFileFolder',
             'NewSTAADFile',
-            'SaveModel'
+            'SaveModel',
+            'CloseSTAADFile',
+            'OpenSTAADFile'
         ]
 
         for function_name in self._functions:
@@ -255,3 +257,13 @@ def SaveModel(self,silent:int):
         self._root.SaveModel(silent)
 
 
+    def CloseSTAADFile(self):
+        self._root.CloseSTAADFile()
+
+    def OpenSTAADFile(self, file_name: str):
+        """
+        Opens a .STD file.
+        """
+        self._root.OpenSTAADFile(file_name)
+
+    

openstaad/tools.py

@@ -25,6 +25,9 @@ def make_variant_vt_ref(obj, var_type):
 def make_safe_array_long_input(lista): 
     return automation._midlSAFEARRAY(ctypes.c_long).create(lista)
 
+def make_safe_array_int_input(lista):
+    return automation._midlSAFEARRAY(ctypes.c_int).create(lista)
+
 
 APICALL = {'file':'',
             'geometry':'Geometry',