143 lines
5.3 KiB
Python
143 lines
5.3 KiB
Python
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2016 - Bernd Hahnebach <bernd@bimstatik.org> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * This program is distributed in the hope that it will be useful, *
|
|
# * but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
# * GNU Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with this program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
__title__ = "FreeCAD Z88 Disp Reader"
|
|
__author__ = "Bernd Hahnebach"
|
|
__url__ = "http://www.freecadweb.org"
|
|
|
|
## @package importZ88O2Results
|
|
# \ingroup FEM
|
|
# \brief FreeCAD Z88 Disp Reader for FEM workbench
|
|
|
|
import FreeCAD
|
|
import os
|
|
|
|
|
|
Debug = False
|
|
|
|
|
|
########## generic FreeCAD import and export methods ##########
|
|
if open.__module__ == '__builtin__':
|
|
# because we'll redefine open below (Python2)
|
|
pyopen = open
|
|
elif open.__module__ == 'io':
|
|
# because we'll redefine open below (Python3)
|
|
pyopen = open
|
|
|
|
|
|
def open(filename):
|
|
"called when freecad opens a file"
|
|
docname = os.path.splitext(os.path.basename(filename))[0]
|
|
insert(filename, docname)
|
|
|
|
|
|
def insert(filename, docname):
|
|
"called when freecad wants to import a file"
|
|
try:
|
|
doc = FreeCAD.getDocument(docname)
|
|
except NameError:
|
|
doc = FreeCAD.newDocument(docname)
|
|
FreeCAD.ActiveDocument = doc
|
|
import_z88_disp(filename)
|
|
|
|
|
|
########## module specific methods ##########
|
|
def import_z88_disp(filename, analysis=None, result_name_prefix=None):
|
|
'''insert a FreeCAD FEM mechanical result object in the ActiveDocument
|
|
'''
|
|
import importToolsFem
|
|
import ObjectsFem
|
|
if result_name_prefix is None:
|
|
result_name_prefix = ''
|
|
m = read_z88_disp(filename)
|
|
if(len(m['Nodes']) > 0):
|
|
if analysis is None:
|
|
analysis_name = os.path.splitext(os.path.basename(filename))[0]
|
|
analysis_object = ObjectsFem.makeAnalysis('Analysis')
|
|
analysis_object.Label = analysis_name
|
|
else:
|
|
analysis_object = analysis # see if statement few lines later, if not analysis -> no FemMesh object is created !
|
|
|
|
for result_set in m['Results']:
|
|
results_name = result_name_prefix + 'results'
|
|
results = ObjectsFem.makeResultMechanical(results_name)
|
|
for m in analysis_object.Member: # TODO analysis could have multiple mesh objects in the future
|
|
if m.isDerivedFrom("Fem::FemMeshObject"):
|
|
results.Mesh = m
|
|
break
|
|
results = importToolsFem.fill_femresult_mechanical(results, result_set, 0)
|
|
analysis_object.Member = analysis_object.Member + [results]
|
|
|
|
if(FreeCAD.GuiUp):
|
|
import FemGui
|
|
FemGui.setActiveAnalysis(analysis_object)
|
|
|
|
|
|
def read_z88_disp(z88_disp_input):
|
|
'''
|
|
read a z88 disp file and extract the nodes and elements
|
|
z88 Displacment output file is z88o2.txt
|
|
works with Z88OS14
|
|
|
|
pure usage:
|
|
import FemToolsZ88
|
|
fea = FemToolsZ88.FemToolsZ88()
|
|
import importZ88O2Results
|
|
disp_file = '/pathtofile/z88o2.txt'
|
|
importZ88O2Results.import_z88_disp(disp_file , fea.analysis)
|
|
|
|
The FreeCAD file needs to have an Analysis and an appropriate FEM Mesh
|
|
'''
|
|
nodes = {}
|
|
mode_disp = {}
|
|
mode_results = {}
|
|
results = []
|
|
|
|
z88_disp_file = pyopen(z88_disp_input, "r")
|
|
|
|
for no, line in enumerate(z88_disp_file):
|
|
lno = no + 1
|
|
linelist = line.split()
|
|
|
|
if lno >= 6:
|
|
# disp line
|
|
# print(linelist)
|
|
node_no = int(linelist[0])
|
|
mode_disp_x = float(linelist[1])
|
|
mode_disp_y = float(linelist[2])
|
|
if len(linelist) > 3:
|
|
mode_disp_z = float(linelist[3])
|
|
else:
|
|
mode_disp_z = 0.0
|
|
mode_disp[node_no] = FreeCAD.Vector(mode_disp_x, mode_disp_y, mode_disp_z)
|
|
nodes[node_no] = node_no
|
|
|
|
mode_results['disp'] = mode_disp
|
|
results.append(mode_results)
|
|
|
|
if Debug:
|
|
for r in results[0]['disp']:
|
|
print(r, ' --> ', results[0]['disp'][r])
|
|
|
|
z88_disp_file.close()
|
|
return {'Nodes': nodes, 'Results': results}
|