midas-civil 1.0.3__py3-none-any.whl → 1.0.5__py3-none-any.whl

midas_civil/_element.py

@@ -5,6 +5,7 @@ from ._group import _add_node_2_stGroup
 import numpy as np
 from scipy.interpolate import splev, splprep
 from math import hypot
+import math
 
 
 def _interpolateAlignment(pointsArray,n_seg=10,deg=1,mSize=0,includePoint:bool=True) -> list:
@@ -83,8 +84,30 @@ def _interpolateAlignment(pointsArray,n_seg=10,deg=1,mSize=0,includePoint:bool=T
     return align_fine_points
 
 
+def _nodeDIST(a:Node,b:Node):
+    return round(hypot((a.X-b.X),(a.Y-b.Y),(a.Z-b.Z)),6)
 
+def _nodeAngleVector(b:Node,a:Node):
 
+    Z_new = np.array([0.000001,0,1])
+    X_new = np.array([(a.X-b.X),(a.Y-b.Y),(a.Z-b.Z)])
+    Y_new = np.cross(Z_new, X_new)
+
+    Z_new = np.cross(X_new, Y_new) # Recomputing
+
+    X_new = X_new / (np.linalg.norm(X_new)+0.000001)
+    Y_new = Y_new / (np.linalg.norm(Y_new)+0.000001)
+    Z_new = Z_new / (np.linalg.norm(Z_new)+0.000001)
+
+
+    return [X_new,Y_new,Z_new]
+
+
+def _triangleAREA(a:Node,b:Node,c:Node):
+    v1 = np.array([a.X-b.X,a.Y-b.Y,a.Z-b.Z])
+    v2 = np.array([b.X-c.X,b.Y-c.Y,b.Z-c.Z])
+    mag = np.linalg.norm(np.cross(v1, v2))
+    return float(0.5 * mag) , np.cross(v1, v2)/mag
 
 def _ADD(self):
     """
@@ -112,6 +135,7 @@ def _ADD(self):
         self.ID = id
         Element.elements.append(self)
         Element.ids.append(int(self.ID))
+    Element.__elemDIC__[str(self.ID)] = self
     
     # ------------  Group assignment -----------------------
     if self._GROUP == "" :
@@ -232,6 +256,7 @@ class Element:
     """
     elements = []
     ids = []
+    __elemDIC__ = {}
 
     @classmethod
     def json(cls):
@@ -256,6 +281,7 @@ class Element:
         if a and 'ELEM' in a and a['ELEM']:
             Element.elements = []
             Element.ids = []
+            Element.__elemDIC__={}
             for elem_id, data in a['ELEM'].items():
                 _JS2Obj(elem_id, data)
 
@@ -264,12 +290,13 @@ class Element:
         MidasAPI("DELETE", "/db/ELEM")
         Element.elements = []
         Element.ids = []
+        Element.__elemDIC__={}
 
     # --- Element Type Subclasses ---
 
     class Beam(_common):
 
-        def __init__(self, i: int, j: int, mat: int = 1, sect: int = 1, angle: float = 0, group = "" , id: int = 0):
+        def __init__(self, i: int, j: int, mat: int = 1, sect: int = 1, angle: float = 0, group = "" , id: int = 0,bLocalAxis=False):
             """
             Creates a BEAM element for frame analysis.
             
@@ -304,10 +331,26 @@ class Element:
             self.NODE = [i, j]
             self.ANGLE = angle
             self._GROUP = group
+
+            _n1 = nodeByID(i)
+            _n2 = nodeByID(j)
+            self.LENGTH = _nodeDIST(_n1,_n2)
+
+            if bLocalAxis:
+                _tempAngle = _nodeAngleVector(_n1,_n2)
+                _n1.AXIS = np.add(_n1.AXIS,_tempAngle)
+                _n2.AXIS = np.add(_n2.AXIS,_tempAngle)
+
+                _norm1 = np.linalg.norm(_n1.AXIS ,axis=1,keepdims=True)
+                _n1.AXIS = _n1.AXIS /_norm1
+
+                _norm2 = np.linalg.norm(_n2.AXIS ,axis=1,keepdims=True)
+                _n2.AXIS = _n2.AXIS /_norm2
+
             _ADD(self)
 
         @staticmethod
-        def SDL(s_loc:list|Node,dir:list,l:float,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0): #CHANGE TO TUPLE
+        def SDL(s_loc:list,dir:list,l:float,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0,bLocalAxis=False): #CHANGE TO TUPLE
                 if isinstance(s_loc,Node):
                     s_loc = (s_loc.X,s_loc.Y,s_loc.Z)
 
@@ -324,17 +367,17 @@ class Element:
                 for i in range(n):
                     if id == 0 : id_new = 0
                     else: id_new = id+i
-                    beam_obj.append(Element.Beam(beam_nodes[i],beam_nodes[i+1],mat,sect,angle,group,id_new))
+                    beam_obj.append(Element.Beam(beam_nodes[i],beam_nodes[i+1],mat,sect,angle,group,id_new,bLocalAxis))
                 
                 return beam_obj
                     
 
         @staticmethod
-        def SE(s_loc:list|Node,e_loc:list|Node,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0):
+        def SE(s_loc:list,e_loc:list,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0,bLocalAxis=False):
                 if isinstance(s_loc,Node):
                     s_loc = (s_loc.X,s_loc.Y,s_loc.Z)
                 if isinstance(e_loc,Node):
-                    s_loc = (e_loc.X,e_loc.Y,e_loc.Z)
+                    e_loc = (e_loc.X,e_loc.Y,e_loc.Z)
 
                 beam_nodes =[]
                 beam_obj = []
@@ -346,12 +389,12 @@ class Element:
                 for i in range(n):
                     if id == 0 : id_new = 0
                     else: id_new = id+i
-                    beam_obj.append(Element.Beam(beam_nodes[i],beam_nodes[i+1],mat,sect,angle,group,id_new))
+                    beam_obj.append(Element.Beam(beam_nodes[i],beam_nodes[i+1],mat,sect,angle,group,id_new,bLocalAxis))
                 
                 return beam_obj
         
         @staticmethod
-        def PLine(points_loc:list,n_div:int=0,deg:int=1,includePoint:bool=True,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0):
+        def PLine(points_loc:list,n_div:int=0,deg:int=1,includePoint:bool=True,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0,bLocalAxis=False):
 
                 beam_nodes =[]
                 beam_obj = []
@@ -366,7 +409,7 @@ class Element:
                 for i in range(len(i_loc)-1):
                     if id == 0 : id_new = 0
                     else: id_new = id+i
-                    beam_obj.append(Element.Beam(beam_nodes[i],beam_nodes[i+1],mat,sect,angle,group,id_new))
+                    beam_obj.append(Element.Beam(beam_nodes[i],beam_nodes[i+1],mat,sect,angle,group,id_new,bLocalAxis))
                 
                 return beam_obj
 
@@ -403,10 +446,11 @@ class Element:
             self.NODE = [i, j]
             self.ANGLE = angle
             self._GROUP = group
+            self.LENGTH = _nodeDIST(nodeByID(i),nodeByID(j))
             _ADD(self)
 
         @staticmethod
-        def SDL(s_loc:list|Node,dir:list,l:float,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0):
+        def SDL(s_loc:list,dir:list,l:float,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0):
             if isinstance(s_loc,Node):
                     s_loc = (s_loc.X,s_loc.Y,s_loc.Z)
 
@@ -429,7 +473,7 @@ class Element:
                 
 
         @staticmethod
-        def SE(s_loc:list|Node,e_loc:list|Node,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0):
+        def SE(s_loc:list,e_loc:list,n:int=1,mat:int=1,sect:int=1,angle:float=0, group = "" , id: int = 0):
             
             if isinstance(s_loc,Node):
                 s_loc = (s_loc.X,s_loc.Y,s_loc.Z)
@@ -484,6 +528,17 @@ class Element:
             self.ANGLE = angle
             self.STYPE = stype
             self._GROUP = group
+
+            if len(nodes)==3:
+                self.AREA,self.NORMAL = _triangleAREA(nodeByID(nodes[0]),nodeByID(nodes[1]),nodeByID(nodes[2]))
+            elif len(nodes)==4:
+                a1 , n1 = _triangleAREA(nodeByID(nodes[0]),nodeByID(nodes[1]),nodeByID(nodes[2]))
+                a2 , n2 = _triangleAREA(nodeByID(nodes[2]),nodeByID(nodes[3]),nodeByID(nodes[0]))
+                self.AREA = a1+a2
+                self.NORMAL = (n1+n2)/np.linalg.norm((n1+n2))
+                
+
+
             _ADD(self)
             
     class Tension(_common):
@@ -528,6 +583,7 @@ class Element:
         self.ANGLE = angle
         self.STYPE = stype
         self._GROUP = group
+        self.LENGTH = _nodeDIST(nodeByID(i),nodeByID(j))
         
         # Handle subtype-specific parameters
         if stype == 1:  # Tension-only specific
@@ -588,6 +644,7 @@ class Element:
             self.ANGLE = angle
             self.STYPE = stype
             self._GROUP = group
+            self.LENGTH = _nodeDIST(nodeByID(i),nodeByID(j))
             
             # Handle subtype-specific parameters
             if stype == 1:  # Compression-only specific
@@ -806,12 +863,19 @@ class Element:
 
 # ---- GET ELEMENT OBJECT FROM ID ----------
 
-def elemByID(elemID:int) -> Element:
-    ''' Return Element object with the input ID '''
-    for elem in Element.elements:
-        if elem.ID == elemID:
-            return elem
+# def elemByID2(elemID:int) -> Element:
+#     ''' Return Element object with the input ID '''
+#     for elem in Element.elements:
+#         if elem.ID == elemID:
+#             return elem
         
-    print(f'There is no element with ID {elemID}')
-    return None
+#     print(f'There is no element with ID {elemID}')
+#     return None
 
+def elemByID(elemID:int) -> Element:
+    ''' Return Element object with the input ID '''
+    try:
+        return (Element.__elemDIC__[str(elemID)])
+    except:
+        print(f'There is no element with ID {elemID}')
+        return None

midas_civil/_group.py

@@ -7,8 +7,8 @@ from ._utils import sFlatten
 # ----------- HELPER FUNCTION -----------
     # --------   RETRIEVE NODE / ELEMENT FROM STRUCTURE GROUP -------
 
-def nodesInGroup(groupName:str|list,unique:bool=True) -> list:
-    ''' Returns Node ID list in a Structure Group '''
+def nodesInGroup(groupName:str,unique:bool=True) -> list:
+    ''' Returns Node ID list in a Structure Group or list of Structure groups'''
     groupNames = _convItem2List(groupName)
     nlist = []
     for gName in groupNames:
@@ -24,7 +24,7 @@ def nodesInGroup(groupName:str|list,unique:bool=True) -> list:
     return sFlatten(nlist)
 
 
-def elemsInGroup(groupName:str|list,unique:bool=True) -> list:
+def elemsInGroup(groupName:str,unique:bool=True) -> list:
     ''' Returns Element ID list in a Structure Group '''
     groupNames = _convItem2List(groupName)
     elist = []

midas_civil/_mapi.py

@@ -3,6 +3,8 @@ import sys
 import winreg
 
 
+
+
 def Midas_help():
     """MIDAS Documnetation : https://midas-rnd.github.io/midasapi-python """
     print("---"*22)
@@ -11,6 +13,13 @@ def Midas_help():
 
 
 
+
+class NX:
+    version_check = True
+    user_print = True
+    debug_print = False
+    
+
 class MAPI_COUNTRY:
     
     country = "US"
@@ -69,6 +78,7 @@ class MAPI_BASEURL:
 class MAPI_KEY:
     """MAPI key from Civil NX.\n\nEg: MAPI_Key("eadsfjaks568wqehhf.ajkgj345qfhh")"""
     data = ""
+    count = 1
     
     def __init__(self, mapi_key:str):
         MAPI_KEY.data = mapi_key
@@ -95,7 +105,7 @@ class MAPI_KEY:
 
 #2 midas API link code:
 def MidasAPI(method:str, command:str, body:dict={})->dict:
-    f"""Sends HTTP Request to MIDAS Civil NX
+    """Sends HTTP Request to MIDAS Civil NX
             Parameters:
                 Method: "PUT" , "POST" , "GET" or "DELETE"
                 Command: eg. "/db/NODE"
@@ -127,7 +137,15 @@ def MidasAPI(method:str, command:str, body:dict={})->dict:
         print(f"⚠️  Civil NX model is not connected.  Click on 'Apps> Connect' in Civil NX. \nMake sure the MAPI Key in python code is matching with the MAPI key in Civil NX.\n\n")
         sys.exit(0)
 
-    # print(method, command, response.status_code , "✅")
+    if NX.debug_print:
+        print(method, command, response.status_code , "✅")
+
+    if MAPI_KEY.count == 1:
+        if NX.user_print:
+            MAPI_KEY.count = 0
+            _checkUSER()
+    
+
 
     return response.json()
 
@@ -146,8 +164,11 @@ def _setUNIT(unitJS):
     MidasAPI('PUT','/db/UNIT',js)
 
 
-
-resp =  MidasAPI('GET','/config/ver',{})['VER']
-print(f"Connected to {resp["NAME"]}   🟢")
-print(f"USER : {resp["USER"]}    |   COMPANY : {resp["COMPANY"]}")
-print("-"*85)
+def _checkUSER():
+    try:
+        resp =  MidasAPI('GET','/config/ver',{})['VER']
+        print(f"Connected to {resp['NAME']}   🟢")
+        print(f"USER : {resp['USER']}    |   COMPANY : {resp['COMPANY']}")
+        print("-"*85)
+    except:
+        print("")

midas_civil/_model.py

@@ -247,6 +247,7 @@ class Model:
         if Thickness.thick!=[]: Thickness.create()
         if Node.nodes!=[]: Node.create()
         if Element.elements!=[] : Element.create()
+        if NodeLocalAxis.skew!=[] : NodeLocalAxis.create()
         Group.create()
         Boundary.create()
         Load.create()

midas_civil/_node.py

@@ -4,13 +4,11 @@ from ._utils import *
 from math import hypot
 from ._group import _add_node_2_stGroup
 
-
-
 def dist_tol(a,b):
     return hypot((a.X-b.X),(a.Y-b.Y),(a.Z-b.Z)) < 0.00001  #TOLERANCE BUILT IN (UNIT INDEP)
 
 def cell(point,size=1): #SIZE OF GRID
-    return (int(point.X//size),int(point.Y//size),int(point.Z//size))
+    return str(f"{int(point.X//size)},{int(point.Y//size)},{int(point.Z//size)}")
 
 
 # -------- FUNCTIONS ARE DEFINED BELOW TO RECOGNISE NODE CLASS ----------------
@@ -20,9 +18,10 @@ def cell(point,size=1): #SIZE OF GRID
 #5 Class to create nodes
 class Node:
     """X ordinate, Y ordinate, Z ordinate, Node ID (optional). \nSample: Node(1,0,5)"""
-    nodes = []
-    ids = []
-    Grid ={}
+    nodes = [] # Node object stores in a list
+    ids = []    # Node IDs used for auto increment of ID and replacement of nodes
+    Grid ={}    # Node object in cube grid
+    __nodeDic__ = {} # Stores
     def __init__(self,x,y,z,id=0,group='',merge=1):
         ''' Create Node object
 
@@ -61,6 +60,7 @@ class Node:
 
         #REPLACE - No merge check
         if id in Node.ids:
+
             index=Node.ids.index(id)
             n_orig = Node.nodes[index]
             loc_orig = str(cell(n_orig))
@@ -70,38 +70,51 @@ class Node:
             
             zz_add_to_dict(Node.Grid,loc_new,self)
             Node.nodes[index]=self
+            Node.__nodeDic__[str(id)] = self
 
 
         #CREATE NEW - Merge Check based on input
         else:
+            self.AXIS = [[0,0,0],[0,0,0],[0,0,0]]
             cell_loc = str(cell(self))      
 
             if cell_loc in Node.Grid:
+
                 if merge == 1:
                     chk=0   #OPTIONAL
                     for node in Node.Grid[cell_loc]:
                         if dist_tol(self,node):
+  
                             chk=1
                             self.ID=node.ID
+                            self.AXIS = node.AXIS
                     if chk==0:
+
+                        self.AXIS = [[0,0,0],[0,0,0],[0,0,0]]
                         Node.nodes.append(self)
                         Node.ids.append(self.ID)
                         Node.Grid[cell_loc].append(self)
+                        
+
                 else:
+
                     Node.nodes.append(self)
                     Node.ids.append(self.ID)
                     Node.Grid[cell_loc].append(self)
             else:
+
                 Node.Grid[cell_loc]=[]
                 Node.nodes.append(self)
                 Node.ids.append(self.ID)
                 Node.Grid[cell_loc].append(self)
+            Node.__nodeDic__[str(self.ID)] = self
             
         if group !="":
             _add_node_2_stGroup(self.ID,group)
 
         
-
+    def __str__(self):
+        return f"NODE ID : {self.ID} | X:{self.X} , Y:{self.Y} , Z:{self.Z} \n {self.__dict__}"
 
     @classmethod
     def json(cls):
@@ -123,32 +136,13 @@ class Node:
         Node.nodes=[]
         Node.ids=[]
         Node.Grid={}
+        Node.__nodeDic__ = {}
         a = Node.get()
         if a != {'message': ''}:
             if list(a['NODE'].keys()) != []:
                 for j in a['NODE'].keys():
                     Node(round(a['NODE'][j]['X'],6), round(a['NODE'][j]['Y'],6), round(a['NODE'][j]['Z'],6), id=int(j), group='', merge=0)
 
-    @staticmethod
-    def delete2(nodes_list):
-        if type(nodes_list)!=list:
-            nodes_list = [nodes_list]
-        url_add = arr2csv(nodes_list)
-
-        MidasAPI("DELETE",f"/db/NODE/{url_add}")
-
-
-    @staticmethod
-    def delete3(*args):
-        try:
-            args2=sFlatten(args)
-            url_add = arr2csv(args2)
-            MidasAPI("DELETE",f"/db/NODE/{url_add}") 
-        except:
-            MidasAPI("DELETE",f"/db/NODE/")
-            Node.nodes=[]
-            Node.ids=[]
-            Node.Grid={}
 
     @staticmethod
     def delete():
@@ -156,6 +150,7 @@ class Node:
         Node.nodes=[]
         Node.ids=[]
         Node.Grid={}
+        Node.__nodeDic__ = {}
 
 
 
@@ -166,18 +161,132 @@ class Node:
 
 # ---- GET NODE OBJECT FROM ID ----------
 
+# def nodeByID(nodeID:int) -> Node:
+#     ''' Return Node object with the input ID '''
+#     for node in Node.nodes:
+#         if node.ID == nodeID:
+#             return node
+        
+#     print(f'There is no node with ID {nodeID}')
+#     return None
+
+
+
 def nodeByID(nodeID:int) -> Node:
     ''' Return Node object with the input ID '''
-    for node in Node.nodes:
-        if node.ID == nodeID:
-            return node
-        
-    print(f'There is no node with ID {nodeID}')
-    return None
+    try:
+        return (Node.__nodeDic__[str(nodeID)])
+    except:
+        print(f'There is no node with ID {nodeID}')
+        return None
+
 
 
 
 
+class NodeLocalAxis:
+    skew = []
+    ids = [] 
 
+    def __init__(self,nodeID,type,angle):
+        '''
+        nodeID(int) : ID of the node
+        axis (str) : Axis of rotation, 'X' , 'Y' , 'Z' , 'XYZ' or 'Vector'
+        angle (float) : Angle of rotation if axis = 'X' , 'Y' or 'Z'  ;
+        angle (list : float) = [30,0,0] if type = 'XYZ'
+        angle (list : vector) -> node.AXIS = [[1,0,0],[0,1,0]] if type = 'Vector'
+        '''
+
+        self.ID = nodeID
+
+        if nodeID in NodeLocalAxis.ids:
+            index = NodeLocalAxis.ids.index(nodeID)
+            intial_angle = NodeLocalAxis.skew[index].ANGLE
+            if intial_angle == [[0,0,0],[0,0,0],[0,0,0]]:
+                intial_angle = [[1,0,0],[0,1,0],[0,0,1]]
+
+            if type == 'Vector':
+                self.TYPE = 'VEC'
+                self.VEC = angle
+            elif type == 'X':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = [angle,intial_angle[1],intial_angle[2]]
+            elif type == 'Y':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = [intial_angle[0],angle,intial_angle[2]]
+            elif type == 'Z':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = [intial_angle[0],intial_angle[1],angle]
+            elif type == 'XYZ':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = angle
+            NodeLocalAxis.skew[index] = self
+        else:
+            if type == 'Vector':
+                self.TYPE = 'VEC'
+                self.VEC = angle
+                self.ANGLE = [0,0,0]
+            elif type == 'X':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = [angle,0,0]
+            elif type == 'Y':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = [0,angle,0]
+            elif type == 'Z':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = [0,0,angle]
+            elif type == 'XYZ':
+                self.TYPE = 'ANGLE'
+                self.ANGLE = angle
+        
+            NodeLocalAxis.skew.append(self)
+            NodeLocalAxis.ids.append(self.ID)
+
+    @classmethod
+    def json(cls):
+        json = {"Assign":{}}
+        for i in cls.skew:
+            if i.TYPE == 'ANGLE':
+                json["Assign"][i.ID]={
+                                    "iMETHOD": 1,
+                                    "ANGLE_X": i.ANGLE[0],
+                                    "ANGLE_Y": i.ANGLE[1],
+                                    "ANGLE_Z": i.ANGLE[2]
+                                }
+            elif i.TYPE == 'VEC':
+                json["Assign"][i.ID]={
+                                    "iMETHOD": 3,
+                                    "V1X": i.VEC[0][0],
+                                    "V1Y": i.VEC[0][1],
+                                    "V1Z": i.VEC[0][2],
+                                    "V2X": i.VEC[1][0],
+                                    "V2Y": i.VEC[1][1],
+                                    "V2Z": i.VEC[1][2]
+                                }
+        return json
+    
+    @staticmethod
+    def create():
+        MidasAPI("PUT","/db/SKEW",NodeLocalAxis.json())
+
+    @staticmethod
+    def delete():
+        MidasAPI("DELETE","/db/SKEW/")
+        NodeLocalAxis.skew=[]
+        NodeLocalAxis.ids=[]
+
+    @staticmethod
+    def get():
+        return MidasAPI("GET","/db/SKEW")
+    
+    # @staticmethod
+    # def sync():
+    #     NodeLocalAxis.skew=[]
+    #     NodeLocalAxis.ids=[]
+    #     a = NodeLocalAxis.get()
+    #     if a != {'message': ''}:
+    #         if list(a['NODE'].keys()) != []:
 
+    #             for j in a['NODE'].keys():
 
+    #                 Node(round(a['NODE'][j]['X'],6), round(a['NODE'][j]['Y'],6), round(a['NODE'][j]['Z'],6), id=int(j), group='', merge=0)