forked from jaheyns/CfdOF
-
Notifications
You must be signed in to change notification settings - Fork 0
/
CfdMesh.py
194 lines (156 loc) · 8.27 KB
/
CfdMesh.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
# ***************************************************************************
# * *
# * Copyright (c) 2016 - Bernd Hahnebach <bernd@bimstatik.org> *
# * Copyright (c) 2019 Oliver Oxtoby <oliveroxtoby@gmail.com> *
# * *
# * 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 *
# * *
# ***************************************************************************
from __future__ import print_function
import FreeCAD
import FreeCADGui
from PySide import QtCore
import CfdTools
from CfdTools import addObjectProperty
import os
def makeCfdMesh(name="CFDMesh"):
obj = FreeCAD.ActiveDocument.addObject("App::DocumentObjectGroupPython", name)
_CfdMesh(obj)
if FreeCAD.GuiUp:
_ViewProviderCfdMesh(obj.ViewObject)
return obj
class _CommandCfdMeshFromShape:
def GetResources(self):
icon_path = os.path.join(CfdTools.get_module_path(), "Gui", "Resources", "icons", "mesh.png")
return {'Pixmap': icon_path,
'MenuText': QtCore.QT_TRANSLATE_NOOP("Cfd_MeshFromShape",
"CFD mesh"),
'ToolTip': QtCore.QT_TRANSLATE_NOOP("Cfd_MeshFromShape",
"Create a mesh using cfMesh, snappyHexMesh or gmsh")}
def IsActive(self):
sel = FreeCADGui.Selection.getSelection()
analysis = CfdTools.getActiveAnalysis()
return analysis is not None and sel and len(sel) == 1 and sel[0].isDerivedFrom("Part::Feature")
def Activated(self):
FreeCAD.ActiveDocument.openTransaction("Create CFD mesh")
analysis_obj = CfdTools.getActiveAnalysis()
if analysis_obj:
meshObj = CfdTools.getMesh(analysis_obj)
if not meshObj:
sel = FreeCADGui.Selection.getSelection()
if len(sel) == 1:
if sel[0].isDerivedFrom("Part::Feature"):
mesh_obj_name = sel[0].Name + "_Mesh"
FreeCADGui.doCommand("")
FreeCADGui.addModule("CfdMesh")
FreeCADGui.doCommand("CfdMesh.makeCfdMesh('" + mesh_obj_name + "')")
FreeCADGui.doCommand("App.ActiveDocument.ActiveObject.Part = App.ActiveDocument." + sel[0].Name)
if CfdTools.getActiveAnalysis():
FreeCADGui.addModule("CfdTools")
FreeCADGui.doCommand(
"CfdTools.getActiveAnalysis().addObject(App.ActiveDocument.ActiveObject)")
FreeCADGui.ActiveDocument.setEdit(FreeCAD.ActiveDocument.ActiveObject.Name)
else:
print("ERROR: You cannot have more than one mesh object")
FreeCADGui.Selection.clearSelection()
class _CfdMesh:
""" CFD mesh properties """
# they will be used from the task panel too, thus they need to be outside of the __init__
known_element_dimensions = ['2D', '3D']
known_mesh_utility = ['cfMesh', 'snappyHexMesh', 'gmsh']
def __init__(self, obj):
self.Type = "CfdMesh"
self.Object = obj
obj.Proxy = self
self.initProperties(obj)
def initProperties(self, obj):
addObjectProperty(obj, 'CaseName', "meshCase", "App::PropertyString", "",
"Name of directory in which the mesh is created")
addObjectProperty(obj, 'ShapeFaceNames', [], "App::PropertyStringList", "", "Mesh face names")
addObjectProperty(obj, 'STLLinearDeflection', 0.05, "App::PropertyFloat", "", "STL linear deflection")
addObjectProperty(obj, 'NumberCores', 1, "App::PropertyInteger", "", "Number of parallel cores "
"(only applicable when using snappyHexMesh) ")
addObjectProperty(obj, "Part", None, "App::PropertyLink", "Mesh Parameters", "Part object to mesh")
if addObjectProperty(obj, "MeshUtility", _CfdMesh.known_mesh_utility, "App::PropertyEnumeration",
"Mesh Parameters", "Meshing utilities"):
obj.MeshUtility = 'cfMesh'
addObjectProperty(obj, "CharacteristicLengthMax", "0 m", "App::PropertyLength", "Mesh Parameters",
"Max mesh element size (0.0 = infinity)")
addObjectProperty(obj, 'PointInMesh', {"x": '0 m', "y": '0 m', "z": '0 m'}, "App::PropertyMap",
"Mesh Parameters",
"Location vector inside the region to be meshed (must not coincide with a cell face)")
addObjectProperty(obj, 'CellsBetweenLevels', 3, "App::PropertyInteger", "Mesh Parameters",
"Number of cells between each level of refinement")
addObjectProperty(obj, 'EdgeRefinement', 1, "App::PropertyFloat", "Mesh Parameters",
"Relative edge (feature) refinement")
if addObjectProperty(obj, 'ElementDimension', _CfdMesh.known_element_dimensions, "App::PropertyEnumeration",
"Mesh Parameters", "Dimension of mesh elements (Default 3D)"):
obj.ElementDimension = '3D'
def onDocumentRestored(self, obj):
self.initProperties(obj)
def __getstate__(self):
return self.Type
def __setstate__(self, state):
if state:
self.Type = state
class _ViewProviderCfdMesh:
""" A View Provider for the CfdMesh object """
def __init__(self, vobj):
vobj.Proxy = self
def getIcon(self):
icon_path = os.path.join(CfdTools.get_module_path(), "Gui", "Resources", "icons", "mesh.png")
return icon_path
def attach(self, vobj):
self.ViewObject = vobj
self.Object = vobj.Object
def updateData(self, obj, prop):
return
def onChanged(self, vobj, prop):
CfdTools.setCompSolid(vobj)
return
def setEdit(self, vobj, mode):
for obj in FreeCAD.ActiveDocument.Objects:
if hasattr(obj, 'Proxy') and isinstance(obj.Proxy, _CfdMesh):
obj.ViewObject.show()
import _TaskPanelCfdMesh
taskd = _TaskPanelCfdMesh._TaskPanelCfdMesh(self.Object)
taskd.obj = vobj.Object
FreeCADGui.Control.showDialog(taskd)
return True
def unsetEdit(self, vobj, mode):
FreeCADGui.Control.closeDialog()
return
def doubleClicked(self, vobj):
if FreeCADGui.activeWorkbench().name() != 'CfdOFWorkbench':
FreeCADGui.activateWorkbench("CfdOFWorkbench")
gui_doc = FreeCADGui.getDocument(vobj.Object.Document)
if not gui_doc.getInEdit():
gui_doc.setEdit(vobj.Object.Name)
else:
FreeCAD.Console.PrintError('Task dialog already open\n')
return True
def onDelete(self, feature, subelements):
try:
for obj in self.Object.Group:
obj.ViewObject.show()
except Exception as err:
FreeCAD.Console.PrintError("Error in onDelete: " + str(err))
return True
def __getstate__(self):
return None
def __setstate__(self, state):
return None