writeMailMeshFiles.py

import os

strut = {
    'ffx': {
        # 'xs': {
        #     'min': 91,
        #     'max': 121,
        #     'body': 40
        # },
        's': {
            'min': 116,
            'max': 152,
            'body': 55
        },
        'm': {
            'min': 143,
            'max': 205,
            'body': 82
        },
        'l': {
            'min': 195,
            'max': 311,
            'body': 136
        }
    },
    'std': {
        'xxs': {
            'min': 59,
            'max': 76
        },
        'xs': {
            'min': 75,
            'max': 96
        },
        's': {
            'min': 90,
            'max': 125
        },
        'm': {
            'min': 116,
            'max': 178
        },
        'l': {
            'min': 169,
            'max': 283
        }
    }
}

uJointBodyLength = 10
uJointRodLength = 13

# def elong(load, elastMod, area, len):
#     return ((load * len) / (elastMod * area))

eBody = 7.0e4; # N/mm2
eRod = 1.85e5; # N/mm2
knobLength = 15.0
bodyRadIn = 3.0

# bodyLength
# rodLength = 10.0

def range2(first, last, inc = 1, prec = 1.0e-4):
    x = first
    xs = []
    while (x <= last):
        xs.append(x)
        x += inc
    print(x, xs[-1], last)
    if ((last - xs[-1]) > prec):
        print(x)
        xs.append(last)
    return xs

def elemLengths(strutType, size, length, elemLength = 1.0):
    mm2m = 0.001
    # bodyLength  = strut[strutType][size]['min'] + knobLength
    bodyLength  = strut[strutType][size]['body'] + uJointBodyLength + 3.0 + knobLength
    rodLength  = length - bodyLength
    nelsBody = round(bodyLength / elemLength)  # number of elements on body
    nelsRod = round(rodLength / elemLength) # number of elements on rod
    # elemLengthAvg = round((bodyLength / nelsBody) + (rodLength / nelsRod) / 2)
    elBodyLength = bodyLength / nelsBody
    elRodLength = rodLength / nelsRod
    bodyLength *= mm2m
    rodLength *= mm2m
    elBodyLength *= mm2m
    elRodLength *= mm2m
    print(bodyLength, rodLength, elBodyLength, elRodLength)
    return bodyLength, rodLength, elBodyLength, elRodLength

# def strutCoords(strutType, size, length, nelsBody = 20, nelsRod = 20):    
def strutCoords(strutType, size, length, elemLength):    
    # elBodyLength, elRodLength = elemLengths(strutType, size, length, nelsBody, nelsRod)
    bodyLength, rodLength, elBodyLength, elRodLength = elemLengths(strutType, size, length, elemLength = elemLength)
    bodyCoords = []
    rodCoords = []
    mm2m = 0.001
    length *= mm2m

    # allCoords = []
    # for ii in range(nelsBody + 1):
    # bodyLength  = strut[strutType][size]['min'] + knobLength
    # rodLength  = length - bodyLength
    # print('length:', length)
    # print(bodyLength, rodLength, elBodyLength, elRodLength)
    for zBody in range2(0.0, bodyLength, elBodyLength):
        # zBody = ii * elBodyLength
        coords = (0.0, 0.0, zBody)
        bodyCoords.append(coords)
        # allCoords.append(coords)
    # for ii in range(nelsRod + 1):
    for zRod in range2(zBody + elRodLength, length, elRodLength):
        # coords = (0.0, 0.0, zBody + ii * elRodLength)
        coords = (0.0, 0.0, zRod)
        # allCoords.append(coords)
        rodCoords.append(coords)
    print(len(bodyCoords), len(rodCoords))
    return bodyCoords, rodCoords

# def strutMesh(strutType, size, length, separator = ' ', nelsBody = 20, nelsRod = 20):
def strutMesh(strutType, size, length, separator = ' ', elemLength = 5.0):
    # bodyCoords, rodCoords = strutCoords(strutType, size, length, nelsBody = 20, nelsRod = 20)
    bodyCoords, rodCoords = strutCoords(strutType, size, length, elemLength = elemLength)
    eol = '\r\n'
    nodesStr = 'COOR_3D' + eol
    elemsStr = 'SEG2' + eol
    # for ii in range(nelsBody):
    nodeCount = 1
    elemCount = 1
    elemNodeLabels  = []
    bodyElemGroup = []
    rodElemGroup = []
    for coords in bodyCoords:
        nodeLabel = 'N' + str(nodeCount)
        nodesStr += nodeLabel + separator
        # nodesStr += separator + str(coord)
        nodesStr += "{0: >014f}".format(coords[0]) + separator
        nodesStr += "{0: >014f}".format(coords[1]) + separator
        nodesStr += "{0: >014f}".format(coords[2])
        nodesStr += eol
        elemNodeLabels.append(nodeLabel)
        if (nodeCount > 1):            
            elemLabel = 'E' + str(elemCount)
            elemsStr += elemLabel + separator + elemNodeLabels[0] + separator + elemNodeLabels[1] + eol
            bodyElemGroup.append(elemLabel)
            # if (nodeCount % 2 == 0):            
            elemNodeLabels = [elemNodeLabels[1]]
        nodeCount += 1
        elemCount += 1
    for coords in rodCoords:
        nodeLabel = 'N' + str(nodeCount)
        nodesStr += nodeLabel
        for coord in coords:
            nodesStr += separator + str(coord)
        nodesStr += eol
        elemNodeLabels.append(nodeLabel)
        elemLabel = 'E' + str(elemCount)
        elemsStr += elemLabel + separator + elemNodeLabels[0] + separator + elemNodeLabels[1] + eol
        rodElemGroup.append(elemLabel)
        elemNodeLabels= [elemNodeLabels[1]]
        nodeCount += 1
        elemCount += 1
    nodesStr += 'FINSF' + eol
    elemsStr += 'FINSF' + eol
    
    bodyElemGroupStr = ''
    elemCount = 1
    for elemLabel in bodyElemGroup:
        if (elemCount % 8 == 0):
            bodyElemGroupStr += eol
        bodyElemGroupStr += elemLabel + separator
        elemCount += 1
    bodyElemGroupStr += eol
    # bodyElemGroupStr += '%' + eol

    rodElemGroupStr = ''
    elemCount = 1
    for elemLabel in rodElemGroup:
        if (elemCount % 8 == 0):
            rodElemGroupStr += eol
        rodElemGroupStr += elemLabel + separator
        elemCount += 1
    rodElemGroupStr += eol
    # rodElemGroupStr += '%' + eol

    # print('bodyElemGroupStr:', bodyElemGroupStr)
    # print('rodElemGroupStr:', rodElemGroupStr)
    groupNosStr = ''
    groupNosStr += 'GROUP_NO NOM = BCEnd' + eol
    # groupNosStr += 'BCEnd' + separator + 'N1' + eol
    groupNosStr += 'N1' + eol
    groupNosStr += 'FINSF' + eol
    groupNosStr += 'GROUP_NO NOM = LoadEnd' + eol
    # groupNosStr += 'LoadEnd' + separator + 'N' + str(len(bodyCoords) + len(rodCoords)) + eol
    groupNosStr += 'N' + str(len(bodyCoords) + len(rodCoords)) + eol
    groupNosStr += 'FINSF' + eol
    # rodElemGroup += 'FINSF' + eol
    # rodElemGroup += '' + eol

    return nodesStr, elemsStr, groupNosStr, bodyElemGroupStr, rodElemGroupStr

# nodesStr, elemsStr = strutMesh(strutType = 'ffx', size = 's', length = 120.0, separator = ' ', nelsBody = 20, nelsRod = 20)
def strutMeshStr(strutType = 'ffx', size = 's', length = 120.0, separator = ' ' , elemLength = 3.3):
    # nodesStr, elemsStr, groupNosStr, bodyElemsGroupStr, rodElemGroupStr = strutMesh(strutType = 'ffx', size = 's', length = 120.0, separator = ' ' , elemLength = 3.3)
    nodesStr, elemsStr, groupNosStr, bodyElemsGroupStr, rodElemGroupStr = strutMesh(strutType = strutType, size = size, length = length, separator = separator , elemLength = elemLength)
    eol = '\r\n'
    meshStr = ''
    # meshStr += 'TITRE'
    # meshStr += eol
    meshStr += '%'
    meshStr += eol
    meshStr += '% Beam mesh for TSF strut ' + strutType + separator + size + separator + str(length) + eol
    meshStr += '%'
    meshStr += eol
    meshStr += nodesStr
    meshStr += '%' + eol
    meshStr += elemsStr
    meshStr += '%' + eol
    meshStr += groupNosStr
    meshStr += '%' + eol
    meshStr += 'GROUP_MA NOM = Body'
    meshStr += eol
    # meshStr += 'Body' + eol
    meshStr += bodyElemsGroupStr
    # meshStr += '%' + eol
    meshStr += 'FINSF' + eol
    meshStr += 'GROUP_MA NOM = Rod'
    meshStr += eol
    meshStr += rodElemGroupStr
    meshStr += 'FINSF' + eol
    meshStr += '%' + eol
    meshStr += 'FIN'
    meshStr += eol
    # print(nodesStr, elemsStr)
    # print(meshStr)
    return meshStr

def meshFile(dirPath, strutType = 'ffx', size = 's', length = 120.0, separator = ' ' , elemLength = 1.0):
    # meshStr = strutMeshStr(strutType = 'ffx', size = 's', length = 120.0, separator = ' ' , elemLength = 3.3)
    meshStr = strutMeshStr(strutType = strutType, size = size, length = length, separator = separator , elemLength = elemLength)
    fileName =  'strut_' + strutType + '_' + size + '_' + str(length) + '.mail'
    filePath = dirPath + '/' + fileName
    with open(filePath, 'w') as f1:
        f1.write(meshStr)
    return filePath


# F:\far\uom\draft\TSF\struts\fe
# filePath = r'F:/far/uom/draft/TSF/struts/fe/test0.mail'
# meshFile(dirPath, strutType = 'ffx', size = 's', length = 130.0, separator = '  ' , elemLength = 2.0)


# results for small ffx struts at lengths of 116, 135 & 142 (mm): 13.7, 6.5, 5.3


def main(dirPath):
    ffxStrut = strut['ffx']
    for strutSize in ffxStrut.keys():
        ffxStrut0 = ffxStrut[strutSize]
        dirPath1 = os.path.join(dirPath, strutSize)
        if not os.path.isdir(dirPath1):
            os.mkdir(dirPath1)
        #print(dirPath1)
        for strutLength in range(ffxStrut0['min'], ffxStrut0['max'] + 1, 1):
            dirPath2 = os.path.join(dirPath1, str(strutLength))
            if not os.path.isdir(dirPath2):
                os.mkdir(dirPath2)
            meshFile(dirPath2, strutType = 'ffx', size = strutSize, length = strutLength, separator = '  ' , elemLength = 1.0)

dirPath = r'/home/fzf/projects/ca/py/tests/beam/buckling/strut'

main(dirPath)