# *************************************************************************** # * Copyright (c) 2020 Bernd Hahnebach * # * * # * This file is part of the FreeCAD CAx development system. * # * * # * 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__ = "FEM geometry tools" __author__ = "Bernd Hahnebach" __url__ = "https://www.freecad.org" import FreeCAD # ************************************************************************************************ def find_element_in_shape( aShape, anElement ): # import Part ele_st = anElement.ShapeType if ele_st == "Solid" or ele_st == "CompSolid": for index, solid in enumerate(aShape.Solids): # FreeCAD.Console.PrintMessage("{}\n".format(is_same_geometry(solid, anElement))) if is_same_geometry(solid, anElement): # FreeCAD.Console.PrintMessage("{}\n".format(index)) # Part.show(aShape.Solids[index]) ele = ele_st + str(index + 1) return ele FreeCAD.Console.PrintError( "Solid " + str(anElement) + " not found in: " + str(aShape) + "\n" ) if ele_st == "Solid" and aShape.ShapeType == "Solid": message_part = ( "We have been searching for a Solid in a Solid and we have not found it. " "In most cases this should be searching for a Solid inside a CompSolid. " "Check the ShapeType of your Part to mesh." ) FreeCAD.Console.PrintMessage(message_part + "\n") # Part.show(anElement) # Part.show(aShape) elif ele_st == "Face" or ele_st == "Shell": for index, face in enumerate(aShape.Faces): # FreeCAD.Console.PrintMessage("{}\n".format(is_same_geometry(face, anElement))) if is_same_geometry(face, anElement): # FreeCAD.Console.PrintMessage("{}\n".format(index)) # Part.show(aShape.Faces[index]) ele = ele_st + str(index + 1) return ele elif ele_st == "Edge" or ele_st == "Wire": for index, edge in enumerate(aShape.Edges): # FreeCAD.Console.PrintMessage("{}\n".format(is_same_geometry(edge, anElement))) if is_same_geometry(edge, anElement): # FreeCAD.Console.PrintMessage(index, "\n") # Part.show(aShape.Edges[index]) ele = ele_st + str(index + 1) return ele elif ele_st == "Vertex": for index, vertex in enumerate(aShape.Vertexes): # FreeCAD.Console.PrintMessage("{}\n".format(is_same_geometry(vertex, anElement))) if is_same_geometry(vertex, anElement): # FreeCAD.Console.PrintMessage("{}\n".format(index)) # Part.show(aShape.Vertexes[index]) ele = ele_st + str(index + 1) return ele elif ele_st == "Compound": FreeCAD.Console.PrintError("Compound is not supported.\n") # ************************************************************************************************ def get_vertexes_by_element( aShape, anElement ): # we're going to extend the method find_element_in_shape and return the vertexes # import Part ele_vertexes = [] ele_st = anElement.ShapeType if ele_st == "Solid" or ele_st == "CompSolid": for index, solid in enumerate(aShape.Solids): if is_same_geometry(solid, anElement): for vele in aShape.Solids[index].Vertexes: for i, v in enumerate(aShape.Vertexes): if vele.isSame(v): # use isSame, because orientation could be different ele_vertexes.append(i) # FreeCAD.Console.PrintMessage(" " + str(sorted(ele_vertexes)), "\n") return ele_vertexes FreeCAD.Console.PrintError( "Error, Solid " + str(anElement) + " not found in: " + str(aShape) + "\n" ) elif ele_st == "Face" or ele_st == "Shell": for index, face in enumerate(aShape.Faces): if is_same_geometry(face, anElement): for vele in aShape.Faces[index].Vertexes: for i, v in enumerate(aShape.Vertexes): if vele.isSame(v): # use isSame, because orientation could be different ele_vertexes.append(i) # FreeCAD.Console.PrintMessage(" " + str(sorted(ele_vertexes)) + "\n") return ele_vertexes elif ele_st == "Edge" or ele_st == "Wire": for index, edge in enumerate(aShape.Edges): if is_same_geometry(edge, anElement): for vele in aShape.Edges[index].Vertexes: for i, v in enumerate(aShape.Vertexes): if vele.isSame(v): # use isSame, because orientation could be different ele_vertexes.append(i) # FreeCAD.Console.PrintMessage(" " + str(sorted(ele_vertexes)) + "\n") return ele_vertexes elif ele_st == "Vertex": for index, vertex in enumerate(aShape.Vertexes): if is_same_geometry(vertex, anElement): ele_vertexes.append(index) # FreeCAD.Console.PrintMessage(" " + str(sorted(ele_vertexes)) + "\n") return ele_vertexes elif ele_st == "Compound": FreeCAD.Console.PrintError("Compound is not supported.\n") # ************************************************************************************************ def is_same_geometry( shape1, shape2 ): # the vertexes and the CenterOfMass are compared # it is a hack, but I do not know any better ! # check of Volume and Area before starting with the vertices could be added # BoundBox is possible too, but is BB calculations robust?! # FreeCAD.Console.PrintMessage("{}\n".format(shape1)) # FreeCAD.Console.PrintMessage("{}\n".format(shape2)) same_Vertexes = 0 if len(shape1.Vertexes) == len(shape2.Vertexes) and len(shape1.Vertexes) > 1: # compare CenterOfMass if shape1.CenterOfMass != shape2.CenterOfMass: return False else: # compare the Vertexes for vs1 in shape1.Vertexes: for vs2 in shape2.Vertexes: if vs1.X == vs2.X and vs1.Y == vs2.Y and vs1.Z == vs2.Z: same_Vertexes += 1 continue # FreeCAD.Console.PrintMessage("{}\n".(same_Vertexes)) if same_Vertexes == len(shape1.Vertexes): return True else: return False if len(shape1.Vertexes) == len(shape2.Vertexes) and len(shape1.Vertexes) == 1: vs1 = shape1.Vertexes[0] vs2 = shape2.Vertexes[0] if vs1.X == vs2.X and vs1.Y == vs2.Y and vs1.Z == vs2.Z: return True else: return False else: return False # ************************************************************************************************ def get_element( part, element ): if element.startswith("Solid"): index = int(element.lstrip("Solid")) - 1 if index >= len(part.Shape.Solids): FreeCAD.Console.PrintError( "Index out of range. This Solid does not exist in the Shape!\n" ) return None else: return part.Shape.Solids[index] # Solid else: return part.Shape.getElement(element) # Face, Edge, Vertex # ************************************************************************************************ def get_rectangular_coords( obj ): from math import cos, sin, radians A = [1, 0, 0] B = [0, 1, 0] a_x = A[0] a_y = A[1] a_z = A[2] b_x = B[0] b_y = B[1] b_z = B[2] x_rot = radians(obj.X_rot) y_rot = radians(obj.Y_rot) z_rot = radians(obj.Z_rot) if obj.X_rot != 0: a_y = A[1] * cos(x_rot) + A[2] * sin(x_rot) a_z = A[2] * cos(x_rot) - A[1] * sin(x_rot) b_y = B[1] * cos(x_rot) + B[2] * sin(x_rot) b_z = B[2] * cos(x_rot) - B[1] * sin(x_rot) if obj.Y_rot != 0: a_x = A[0] * cos(y_rot) - A[2] * sin(y_rot) a_z = A[2] * cos(y_rot) + A[0] * sin(y_rot) b_x = B[0] * cos(y_rot) - B[2] * sin(y_rot) b_z = B[2] * cos(y_rot) + B[0] * sin(z_rot) if obj.Z_rot != 0: a_x = A[0] * cos(z_rot) + A[1] * sin(z_rot) a_y = A[1] * cos(z_rot) - A[0] * sin(z_rot) b_x = B[0] * cos(z_rot) + B[1] * sin(z_rot) b_y = B[1] * cos(z_rot) - B[0] * sin(z_rot) return (a_x, a_y, a_z, b_x, b_y, b_z) # ************************************************************************************************ def get_cylindrical_coords( obj ): vec = obj.Axis base = obj.BasePoint a_x = base[0] + 10 * vec[0] a_y = base[1] + 10 * vec[1] a_z = base[2] + 10 * vec[2] b_x = base[0] - 10 * vec[0] b_y = base[1] - 10 * vec[1] b_z = base[2] - 10 * vec[2] return (a_x, a_y, a_z, b_x, b_y, b_z)