Merge pull request #38 from u-anurag/OpenSeesPy-Plotting

Version Bump to Get_Rendering

openseespy-pip/.gitignore

@@ -10,3 +10,4 @@ dist
 .idea
 *.dylib
 .vscode
+test_ODB/

openseespy-pip/openseespy/postprocessing/internal_database_functions.py

@@ -0,0 +1,319 @@
+import numpy as np
+import os
+import warnings
+import openseespy.opensees as ops
+
+
+
+def _getNodesandElements():
+    """
+    This function returns the nodes and elments for an active model, in a 
+    standardized format. The OpenSees model must be active in order for the 
+    function to work.
+    
+    Returns
+    -------
+    nodes : 2dArray
+        An array of all nodes in the model.
+        Returns nodes in the shape:
+        [Nodes, 3] in 2d and [Nodes, 4]
+        For each node the information is tored as follows:
+        [NodeID, x, y] or [NodeID, x, y, z]
+    elements : Array 
+        An list of all elements in. Each entry in the list is it's own'
+        [element1, element2,...],   element1 = [element#, node1, node2,...]
+    """
+
+    # Get nodes and elements
+    nodeList = ops.getNodeTags()
+    eleList = ops.getEleTags()   
+
+    # Check Number of dimensions and intialize variables
+    ndm = len(ops.nodeCoord(nodeList[0]))
+    Nnodes = len(nodeList)
+    nodes = np.zeros([Nnodes, ndm + 1])
+
+    # Get Node list
+    for ii, node in enumerate(nodeList):
+        nodes[ii,0] = node
+        nodes[ii,1:] = ops.nodeCoord(nodeList[ii])           
+
+    Nele = len(eleList)
+    elements = [None]*Nele
+
+    # Generate the element list by looping through all emenemts
+    for ii, ele in enumerate(eleList):
+        tempNodes = ops.eleNodes(ele)
+
+        tempNnodes = len(tempNodes)
+        tempEle = np.zeros(tempNnodes + 1)
+
+        tempEle[0] = int(ele)
+        tempEle[1:] = tempNodes
+
+        elements[ii] = tempEle       
+
+    return nodes, elements
+
+def _saveNodesandElements(ModelName):
+    """   
+    This file saves the node and element information for the structure. 
+    For each node information is saved in the following format:
+        Nodes:    [NodeID, xcord, ycord] or [NodeID, xcord, ycord, zcord]
+    
+    For elements, the element is saved with the element connectivity. 
+    A different file is created for each type of element
+    each possible element type.
+        Elements: [EleID, eleNode1, eleNode2, ... , eleNodeN]
+    Parameters
+    ----------
+    nodeName : str, optional
+        The name of the file to be saved. The default is 'Nodes'.
+    eleName : str, optional
+        The name of the . The default is 'Elements'.
+    delim : str, optional
+        The delimeter for the output file. The default is ','.
+    fmt : str, optional
+        the format of the file to be saved in. The default is '%.5e'.
+    """
+
+
+    # Consider making these optional arguements
+    nodeName = 'Nodes'
+    eleName = 'Elements'
+    delim = ' '
+    fmt = '%.5e'
+    ftype = '.out'
+
+    ODBdir = ModelName+"_ODB"		# ODB Dir name
+
+    # Read noades and elements
+    nodes, elements = _getNodesandElements()
+
+    # Sort through the element arrays
+    ele2Node = np.array([ele for ele in elements if len(ele) == 3])
+    ele3Node = np.array([ele for ele in elements if len(ele) == 4])
+    ele4Node = np.array([ele for ele in elements if len(ele) == 5])
+    ele8Node = np.array([ele for ele in elements if len(ele) == 9])
+
+
+    nodeFile = os.path.join(ODBdir, nodeName + ftype)
+
+    ele2File = os.path.join(ODBdir, eleName + "_2Node" + ftype)
+    ele3File = os.path.join(ODBdir, eleName + "_3Node" + ftype)
+    ele4File = os.path.join(ODBdir, eleName + "_4Node"  + ftype)
+    ele8File = os.path.join(ODBdir, eleName + "_8Node"  + ftype)
+
+    # SaveNodes
+    np.savetxt(nodeFile, nodes, delimiter = delim, fmt = fmt)
+
+    # Save element arrays
+    np.savetxt(ele2File, ele2Node, delimiter = delim, fmt = fmt)
+    np.savetxt(ele3File, ele3Node, delimiter = delim, fmt = fmt)
+    np.savetxt(ele4File, ele4Node, delimiter = delim, fmt = fmt)
+    np.savetxt(ele8File, ele8Node, delimiter = delim, fmt = fmt)
+
+
+def _readNodesandElements(ModelName):
+    """   
+    This function reads input node/element information, assuming it is in the 
+    standard format. 
+    If outputDir == False, the base directory will be used.    
+    
+    Parameters
+    ----------
+    nodeName : str, optional
+        The base name for the node file. It will be appended to include
+        the file type. The default is 'Nodes.out'.
+    eleName : str, optional
+        The base nae for the element files. The default is 'Elements.out'.
+    delim : str, optional
+        The delimiter for files to be read. The default is ','.
+    dtype : TYPE, optional
+        The data type to read in. The default is 'float32'.
+    Returns
+    -------
+    nodes : Array
+        An output vector in standard format
+    elements : List
+        An output Element vector in standard format.
+        elements = [ele1, ele2,..., elen], 
+        ele1 = [element, node 1, node 2, ... , node n]
+    """
+
+    # Consider making these optional arguements
+    nodeName = 'Nodes'
+    eleName = 'Elements'
+    delim = ' '
+    dtype ='float32' 
+    ftype = '.out'
+
+    ODBdir = ModelName+"_ODB"		# ODB Dir name
+
+    # Check if output database exists
+    if not os.path.exists(ODBdir):
+        print('No directory found for nodes and elements')
+
+    # Generate the file names
+    nodeFile = os.path.join(ODBdir, nodeName + ftype)
+    ele2File = os.path.join(ODBdir, eleName + "_2Node" + ftype)
+    ele3File = os.path.join(ODBdir, eleName + "_3Node" + ftype)
+    ele4File = os.path.join(ODBdir, eleName + "_4Node"  + ftype)
+    ele8File = os.path.join(ODBdir, eleName + "_8Node"  + ftype)     
+
+    eleFileNames = [ele2File, ele3File, ele4File, ele8File]    
+
+    ## Load Node information
+    try:
+        nodes = np.loadtxt(nodeFile, dtype, delimiter = delim, unpack=False)
+    except:
+        print("Reading node data from a OpenSees Tcl model")
+        nodes = np.transpose(np.loadtxt(nodeFile, dtype=float, delimiter=None, converters=None, unpack=True))
+
+    # Populate an array with the input element information
+    TempEle = [[]]*4
+
+    # Check if the file exists, read it if it does. Ignore warnings if the files are empty
+    for ii, FileName in enumerate(eleFileNames):
+        if os.path.isfile(FileName):
+            with warnings.catch_warnings():
+                warnings.simplefilter("ignore")
+                try:
+                    TempEle[ii] = np.loadtxt(FileName, dtype,  delimiter = delim, skiprows=0, ndmin=2, unpack=False)
+                except:
+                    print("Reading element data from a OpenSees Tcl model")
+                    TempEle[ii] = np.transpose(np.loadtxt(FileName, dtype=float, delimiter=None,  skiprows=0, ndmin=2,converters=None, unpack=True))
+
+    # define the final element array
+    elements = [*TempEle[0],*TempEle[1],*TempEle[2],*TempEle[3]]
+
+    # Check if any files were read
+    if elements is []:
+        raise Exception('No files were found!')
+
+    return nodes, elements
+
+
+################ ModeShapes #############################
+
+def _getModeShapeData(modeNumber):
+
+	# Get nodes and elements
+    nodeList = ops.getNodeTags()
+
+    # Check Number of dimensions and intialize variables
+    ndm = len(ops.nodeCoord(nodeList[0]))
+    Nnodes = len(nodeList)
+    nodes_modeshape = np.zeros([Nnodes, ndm + 1])
+
+    for ii, node in enumerate(nodeList):
+        nodes_modeshape[ii,0] = node
+        tempData = ops.nodeEigenvector(nodeList[ii], modeNumber)
+        nodes_modeshape[ii,1:] = tempData[0:ndm]
+
+    return nodes_modeshape
+
+
+def _saveModeShapeData(ModelName,modeNumber):
+
+    nodes_modeshape = _getModeShapeData(modeNumber)
+
+	# Consider making these optional arguements
+    modeName = "ModeShape"
+    delim = ' '
+    fmt = '%.5e'
+    ftype = '.out'
+
+    ODBdir = ModelName+"_ODB"		# ODB Dir name
+    ModeShapeDir = os.path.join(ODBdir,"ModeShapes")
+    modeFile = os.path.join(ModeShapeDir, modeName+str(modeNumber)+ftype)
+
+	## ModeShapeDir is a default name
+    np.savetxt(modeFile, nodes_modeshape, delimiter = delim, fmt = fmt)    
+
+
+def _readModeShapeData(ModelName,modeNumber):
+
+    # Consider making these optional arguements
+    modeName = "ModeShape"
+    delim = ' '
+    fmt = '%.5e'
+    dtype ='float32'
+    ftype = '.out'
+
+    ODBdir = ModelName+"_ODB"		# ODB Dir name
+    ModeShapeDir = os.path.join(ODBdir,"ModeShapes")
+
+    # Check if output database exists
+    if not os.path.exists(ModeShapeDir):
+        print('Error: No directory found for modeshapes. Use recordODB() command to save modeshapes.')
+
+    modeFile = os.path.join(ModeShapeDir, modeName+str(modeNumber)+ftype)
+    modeTFile = os.path.join(ModeShapeDir, "ModalPeriods.out")
+
+    ## Read modal period data to display
+    periods = np.loadtxt(modeTFile, dtype, delimiter = delim, unpack=False)
+
+	## Load Node information
+    try:
+        nodes_modeshape = np.loadtxt(modeFile, dtype, delimiter = delim, unpack=False)
+    except:
+        print("Reading modeshape data from a OpenSees Tcl model")
+        nodes_modeshape = np.transpose(np.loadtxt(modeFile, dtype=float, delimiter=None, converters=None, unpack=True))
+
+    return nodes_modeshape, periods
+
+
+############## Node Displacement Data ######################################
+
+def _readNodeDispData(ModelName,LoadCaseName):
+
+    ODBdir = ModelName+"_ODB"		# ODB Dir name
+    LoadCaseDir = os.path.join(ODBdir, LoadCaseName)
+
+    # Get number of nodes in the model to set a node displacement array
+    nodes,elements = _readNodesandElements(ModelName)
+    Nnodes = len(nodes)
+    ndm = len(nodes[0,1:])
+
+    NodeDispFile = os.path.join(LoadCaseDir,"NodeDisp_All.out")
+    Disp = np.transpose(np.loadtxt(NodeDispFile, dtype=float, delimiter=None, converters=None, unpack=True))
+
+    timeSteps = Disp[:,0]
+    Ntime = len(Disp[:,0])
+
+    tempDisp = np.zeros([Ntime,Nnodes,ndm])
+    tempDisp[:,:,0] = Disp[:,1::ndm]
+    tempDisp[:,:,1] = Disp[:,2::ndm]
+
+    if ndm == 3:
+         tempDisp[:,:,2] = Disp[:,3::ndm]
+
+    nodes_displacement = tempDisp
+
+    return timeSteps, nodes_displacement
+
+
+#### Read fibre data
+
+def _readFiberData2D(ModelName, LoadCaseName, eleNumber, sectionNumber):
+
+    # Consider making these optional arguements
+    FibreName = "FiberData"
+    delim = ' '
+    # fmt = '%.5e'
+    dtype ='float32'
+    ftype = '.out'    
+
+    ODBdir = ModelName+"_ODB"		# ODB Dir name
+    FibreFileName = FibreName  + '_ele_' + str(eleNumber) + '_section_' + str(sectionNumber) + ftype
+    FiberDir = os.path.join(ODBdir, LoadCaseName, FibreFileName)
+    # Check if output database exists
+    if not os.path.exists(FiberDir):
+        print('Error: No file for Fiber data. Use saveFiberData2D() to create a recorder.')    
+
+    FiberData = np.loadtxt(FiberDir, dtype=dtype, delimiter=delim)
+    timeSteps = FiberData[:,0]
+    FiberData = FiberData[:,1:]
+
+    return timeSteps, FiberData