node_elem_info.py

# -*- coding: utf-8 -*-
"""
Created on Wed Apr  6 19:49:33 2016

@author: peng69
"""

def node_info(nSG,line,unit_change_ratio=1.0,Xs=(0.0,0.0,0.0)):

    #nSG=1
    #line=['99', '           0.', '  0.345904797', ' -0.0876004323']
    #unit_change_ratio=2.0
    #Xs=(1.0,0.0,0.0)
    
    nid=int(line[0].strip())
    if nSG==1:
        x3=float(line[3].strip())
        if unit_change_ratio !=1.0:
            x3=    x3*unit_change_ratio
        if Xs !=(0.0,0.0,0.0):
            x3=    x3-    Xs[2]    
        node_info=[nid, x3]
    elif nSG==2:
        x2=float(line[2].strip())
        x3=float(line[3].strip())
        if unit_change_ratio !=1.0:
            x2=    x2*unit_change_ratio
            x3=    x3*unit_change_ratio
        if Xs !=(0.0,0.0,0.0):
            x2=    x2-    Xs[1]
            x3=    x3-    Xs[2]
        node_info=[nid, x2,x3]
    elif nSG==3:
        x1=float(line[1].strip())
        x2=float(line[2].strip())
        x3=float(line[3].strip())
        if unit_change_ratio !=1.0:
            x1=    x1*unit_change_ratio
            x2=    x2*unit_change_ratio
            x3=    x3*unit_change_ratio
        if Xs !=(0.0,0.0,0.0):
            x1=    x1-    Xs[0]
            x2=    x2-    Xs[1]
            x3=    x3-    Xs[2]
        node_info=[nid, x1,x2,x3]

    return node_info

#--------------------------------------------------

def elem_info(nSG,line):

#nSG=2
#line=['42', ' 102', ' 104', '  91', '  90','93','94']

    if nSG==1:
        print 'when nSG=1, other methods to generate the .sc file should be used, please contact the GUI developer.'
    elif nSG==2:
        nMaxnode_elem=9
        nSpecialElem=6
        zeroPosition=3
    elif nSG==3:
        nMaxnode_elem=20
        nSpecialElem=10
        zeroPosition=4
    
    
    eid=line[0].strip()
    ei_connt = [int(v.strip()) for v in line[1:]]    # {'element_id': [n_1, n_2, ...], ...}    
     nsc_connect=len(ei_connt) 

    
    if len(ei_connt)==nSpecialElem:
        ei_connt.insert(zeroPosition, 0)
        nsc_connect=len(ei_connt)
    
    zeros= [0] * (nMaxnode_elem - nsc_connect)
    ei_connt+= zeros
    ei_info=[eid,ei_connt]
    
    return ei_connt