FEM: calculix writer, use generic method to write sets

This commit is contained in:
Bernd Hahnebach
2021-07-05 09:15:44 +02:00
parent 9b04818dcb
commit 73cf10c4e7
2 changed files with 140 additions and 235 deletions

View File

@@ -265,29 +265,52 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
return inpfile
# ********************************************************************************************
# constraints fixed
def write_node_sets_constraints_fixed(self, f):
if not self.fixed_objects:
# write constraint node sets, constraint face sets, constraint element sets
def write_constraints_sets(
self,
f,
femobjs,
analysis_types,
sets_getter_method,
write_name,
sets_writer_method,
caller_method_name="",
):
if not femobjs:
return
# write for all analysis types
# get nodes
self.get_constraints_fixed_nodes()
if analysis_types != "all" and self.analysis_type not in analysis_types:
return
write_name = "constraints_fixed_node_sets"
f.write("\n***********************************************************\n")
# get the sets
sets_getter_method()
f.write("\n{}\n".format(59 * "*"))
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
f.write("** written by {} function\n".format(caller_method_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
file_name_split = "{}_{}.inp".format(self.mesh_name, write_name)
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_node_sets_nodes_constraints_fixed(inpfile_split)
sets_writer_method(inpfile_split)
inpfile_split.close()
else:
self.write_node_sets_nodes_constraints_fixed(f)
sets_writer_method(f)
# ********************************************************************************************
# constraints fixed
def write_node_sets_constraints_fixed(self, f):
self.write_constraints_sets(
f,
femobjs=self.fixed_objects,
analysis_types="all", # write for all analysis types
sets_getter_method=self.get_constraints_fixed_nodes,
write_name="constraints_fixed_node_sets",
sets_writer_method=self.write_node_sets_nodes_constraints_fixed,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_node_sets_nodes_constraints_fixed(self, f):
# write nodes to file
@@ -354,27 +377,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints displacement
def write_node_sets_constraints_displacement(self, f):
if not self.displacement_objects:
return
# write for all analysis types
# get nodes
self.get_constraints_displacement_nodes()
write_name = "constraints_displacement_node_sets"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_node_sets_nodes_constraints_displacement(inpfile_split)
inpfile_split.close()
else:
self.write_node_sets_nodes_constraints_displacement(f)
self.write_constraints_sets(
f,
femobjs=self.displacement_objects,
analysis_types="all", # write for all analysis types
sets_getter_method=self.get_constraints_displacement_nodes,
write_name="constraints_displacement_node_sets",
sets_writer_method=self.write_node_sets_nodes_constraints_displacement,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_node_sets_nodes_constraints_displacement(self, f):
# write nodes to file
@@ -432,27 +443,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints planerotation
def write_node_sets_constraints_planerotation(self, f):
if not self.planerotation_objects:
return
# write for all analysis types
# get nodes
self.get_constraints_planerotation_nodes()
write_name = "constraints_planerotation_node_sets"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_node_sets_nodes_constraints_planerotation(inpfile_split)
inpfile_split.close()
else:
self.write_node_sets_nodes_constraints_planerotation(f)
self.write_constraints_sets(
f,
femobjs=self.planerotation_objects,
analysis_types="all", # write for all analysis types
sets_getter_method=self.get_constraints_planerotation_nodes,
write_name="constraints_planerotation_node_sets",
sets_writer_method=self.write_node_sets_nodes_constraints_planerotation,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_node_sets_nodes_constraints_planerotation(self, f):
# write nodes to file
@@ -515,27 +514,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints contact
def write_surfaces_constraints_contact(self, f):
if not self.contact_objects:
return
# write for all analysis types
# get faces
self.get_constraints_contact_faces()
write_name = "constraints_contact_surface_sets"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_surfacefaces_constraints_contact(inpfile_split)
inpfile_split.close()
else:
self.write_surfacefaces_constraints_contact(f)
self.write_constraints_sets(
f,
femobjs=self.contact_objects,
analysis_types="all", # write for all analysis types
sets_getter_method=self.get_constraints_contact_faces,
write_name="constraints_contact_surface_sets",
sets_writer_method=self.write_surfacefaces_constraints_contact,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_surfacefaces_constraints_contact(self, f):
# write faces to file
@@ -585,27 +572,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints tie
def write_surfaces_constraints_tie(self, f):
if not self.tie_objects:
return
# write for all analysis types
# get faces
self.get_constraints_tie_faces()
write_name = "constraints_tie_surface_sets"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_surfacefaces_constraints_tie(inpfile_split)
inpfile_split.close()
else:
self.write_surfacefaces_constraints_tie(f)
self.write_constraints_sets(
f,
femobjs=self.tie_objects,
analysis_types="all", # write for all analysis types
sets_getter_method=self.get_constraints_tie_faces,
write_name="constraints_tie_surface_sets",
sets_writer_method=self.write_surfacefaces_constraints_tie,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_surfacefaces_constraints_tie(self, f):
# write faces to file
@@ -647,26 +622,18 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints sectionprint
def write_surfaces_constraints_sectionprint(self, f):
if not self.sectionprint_objects:
return
# write for all analysis types
write_name = "constraints_sectionprint_surface_sets"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_surfacefaces_constraints_sectionprint(inpfile_split)
inpfile_split.close()
else:
self.write_surfacefaces_constraints_sectionprint(f)
self.write_constraints_sets(
f,
femobjs=self.sectionprint_objects,
analysis_types="all", # write for all analysis types
sets_getter_method=self.get_constraints_sectionprint_faces,
write_name="constraints_sectionprint_surface_sets",
sets_writer_method=self.write_surfacefaces_constraints_sectionprint,
caller_method_name=sys._getframe().f_code.co_name,
)
# TODO move code parts from this method to base writer module
# into get_constraints_sectionprint_faces method
def write_surfacefaces_constraints_sectionprint(self, f):
# get surface nodes and write them to file
obj = 0
@@ -724,27 +691,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints transform
def write_node_sets_constraints_transform(self, f):
if not self.transform_objects:
return
# write for all analysis types
# get nodes
self.get_constraints_transform_nodes()
write_name = "constraints_transform_node_sets"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_node_sets_nodes_constraints_transform(inpfile_split)
inpfile_split.close()
else:
self.write_node_sets_nodes_constraints_transform(f)
self.write_constraints_sets(
f,
femobjs=self.transform_objects,
analysis_types="all", # write for all analysis types
sets_getter_method=self.get_constraints_transform_nodes,
write_name="constraints_transform_node_sets",
sets_writer_method=self.write_node_sets_nodes_constraints_transform,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_node_sets_nodes_constraints_transform(self, f):
# write nodes to file
@@ -798,28 +753,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints temperature
def write_node_sets_constraints_temperature(self, f):
if not self.temperature_objects:
return
if not self.analysis_type == "thermomech":
return
# get nodes
self.get_constraints_temperature_nodes()
write_name = "constraints_temperature_node_sets"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_node_sets_nodes_constraints_temperature(inpfile_split)
inpfile_split.close()
else:
self.write_node_sets_nodes_constraints_temperature(f)
self.write_constraints_sets(
f,
femobjs=self.temperature_objects,
analysis_types=["thermomech"],
sets_getter_method=self.get_constraints_temperature_nodes,
write_name="constraints_temperature_node_sets",
sets_writer_method=self.write_node_sets_nodes_constraints_temperature,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_node_sets_nodes_constraints_temperature(self, f):
# write nodes to file
@@ -862,7 +804,7 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
def write_constraints_initialtemperature(self, f):
if not self.initialtemperature_objects:
return
if not self.analysis_type == "thermomech":
if self.analysis_type not in ["thermomech"]:
return
# write constraint to file
@@ -880,11 +822,7 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
def write_constraints_selfweight(self, f):
if not self.selfweight_objects:
return
if not (
self.analysis_type == "static"
or self.analysis_type == "thermomech"
or self.analysis_type == "buckling"
):
if self.analysis_type not in ["buckling", "static", "thermomech"]:
return
# write constraint to file
@@ -916,32 +854,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints force
def write_constraints_force(self, f):
if not self.force_objects:
return
if not (
self.analysis_type == "static"
or self.analysis_type == "thermomech"
or self.analysis_type == "buckling"
):
return
# check shape type of reference shape and get node loads
self.get_constraints_force_nodeloads()
write_name = "constraints_force_node_loads"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_nodeloads_constraints_force(inpfile_split)
inpfile_split.close()
else:
self.write_nodeloads_constraints_force(f)
self.write_constraints_sets(
f,
femobjs=self.force_objects,
analysis_types=["buckling", "static", "thermomech"],
sets_getter_method=self.get_constraints_force_nodeloads,
write_name="constraints_force_node_loads",
sets_writer_method=self.write_nodeloads_constraints_force,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_nodeloads_constraints_force(self, f):
# write node loads to file
@@ -969,32 +890,15 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints pressure
def write_constraints_pressure(self, f):
if not self.pressure_objects:
return
if not (
self.analysis_type == "static"
or self.analysis_type == "thermomech"
or self.analysis_type == "buckling"
):
return
# get the faces and face numbers
self.get_constraints_pressure_faces()
write_name = "constraints_pressure_element_face_loads"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_faceloads_constraints_pressure(inpfile_split)
inpfile_split.close()
else:
self.write_faceloads_constraints_pressure(f)
self.write_constraints_sets(
f,
femobjs=self.pressure_objects,
analysis_types=["buckling", "static", "thermomech"],
sets_getter_method=self.get_constraints_pressure_faces,
write_name="constraints_pressure_element_face_loads",
sets_writer_method=self.write_faceloads_constraints_pressure,
caller_method_name=sys._getframe().f_code.co_name,
)
def write_faceloads_constraints_pressure(self, f):
# write face loads to file
@@ -1024,26 +928,18 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints heatflux
def write_constraints_heatflux(self, f):
if not self.heatflux_objects:
return
if not self.analysis_type == "thermomech":
return
write_name = "constraints_heatflux_element_face_heatflux"
f.write("\n***********************************************************\n")
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("** written by {} function\n".format(sys._getframe().f_code.co_name))
if self.split_inpfile is True:
file_name_split = self.mesh_name + "_" + write_name + ".inp"
f.write("** {}\n".format(write_name.replace("_", " ")))
f.write("*INCLUDE,INPUT={}\n".format(file_name_split))
inpfile_split = open(join(self.dir_name, file_name_split), "w")
self.write_faceheatflux_constraints_heatflux(inpfile_split)
inpfile_split.close()
else:
self.write_faceheatflux_constraints_heatflux(f)
self.write_constraints_sets(
f,
femobjs=self.heatflux_objects,
analysis_types=["thermomech"],
sets_getter_method=self.get_constraints_heatflux_faces,
write_name="constraints_heatflux_element_face_heatflux",
sets_writer_method=self.write_faceheatflux_constraints_heatflux,
caller_method_name=sys._getframe().f_code.co_name,
)
# TODO move code parts from this method to base writer module
# into get_constraints_heatflux_faces method
def write_faceheatflux_constraints_heatflux(self, f):
# write heat flux faces to file
for hfobj in self.heatflux_objects:
@@ -1083,10 +979,11 @@ class FemInputWriterCcx(writerbase.FemInputWriter):
# ********************************************************************************************
# constraints fluidsection
# TODO: split method into separate methods and move some part into base writer
def write_constraints_fluidsection(self, f):
if not self.fluidsection_objects:
return
if not self.analysis_type == "thermomech":
if self.analysis_type not in ["thermomech"]:
return
# write constraint to file

View File

@@ -394,6 +394,14 @@ class FemInputWriter():
# FreeCAD.Console.PrintLog("{}\n".format(femobj["ContactSlaveFaces"]))
# FreeCAD.Console.PrintLog("{}\n".format(femobj["ContactMasterFaces"]))
def get_constraints_sectionprint_faces(self):
pass
# TODO implement
def get_constraints_heatflux_faces(self):
pass
# TODO implement
# ********************************************************************************************
# ********************************************************************************************
# element sets