FreeCAD
151 строка · 6.7 Кб
1# ***************************************************************************
2# * (c) 2009, 2010 Yorik van Havre <yorik@uncreated.net> *
3# * (c) 2009, 2010 Ken Cline <cline@frii.com> *
4# * (c) 2020 Eliud Cabrera Castillo <e.cabrera-castillo@tum.de> *
5# * *
6# * This file is part of the FreeCAD CAx development system. *
7# * *
8# * This program is free software; you can redistribute it and/or modify *
9# * it under the terms of the GNU Lesser General Public License (LGPL) *
10# * as published by the Free Software Foundation; either version 2 of *
11# * the License, or (at your option) any later version. *
12# * for detail see the LICENCE text file. *
13# * *
14# * FreeCAD is distributed in the hope that it will be useful, *
15# * but WITHOUT ANY WARRANTY; without even the implied warranty of *
16# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
17# * GNU Library General Public License for more details. *
18# * *
19# * You should have received a copy of the GNU Library General Public *
20# * License along with FreeCAD; if not, write to the Free Software *
21# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
22# * USA *
23# * *
24# ***************************************************************************
25"""Provides GUI tools to convert Draft objects to Sketches and back.
26
27Many Draft objects will be converted to a single non-constrained Sketch.
28
29However, a single sketch with disconnected traces will be converted
30into several individual Draft objects.
31"""
32## @package gui_draft2sketch
33# \ingroup draftguitools
34# \brief Provides GUI tools to convert Draft objects to Sketches and back.
35
36## \addtogroup draftguitools
37# @{
38from PySide.QtCore import QT_TRANSLATE_NOOP
39import FreeCADGui as Gui
40import Draft_rc
41import draftguitools.gui_base_original as gui_base_original
42import draftguitools.gui_tool_utils as gui_tool_utils
43from draftutils.messages import _msg
44from draftutils.translate import translate
45
46# The module is used to prevent complaints from code checkers (flake8)
47True if Draft_rc.__name__ else False
48
49
50class Draft2Sketch(gui_base_original.Modifier):
51"""Gui Command for the Draft2Sketch tool."""
52
53def GetResources(self):
54"""Set icon, menu and tooltip."""
55
56return {'Pixmap': 'Draft_Draft2Sketch',
57'MenuText': QT_TRANSLATE_NOOP("Draft_Draft2Sketch", "Draft to sketch"),
58'ToolTip': QT_TRANSLATE_NOOP("Draft_Draft2Sketch", "Convert bidirectionally between Draft objects and Sketches.\nMany Draft objects will be converted into a single non-constrained Sketch.\nHowever, a single sketch with disconnected traces will be converted into several individual Draft objects.")}
59
60def Activated(self):
61"""Execute when the command is called."""
62super().Activated(name="Convert Draft/Sketch")
63if not self.ui:
64return
65if not Gui.Selection.getSelection():
66self.ui.selectUi(on_close_call=self.finish)
67_msg(translate("draft", "Select an object to convert."))
68self.call = self.view.addEventCallback("SoEvent", gui_tool_utils.selectObject)
69else:
70self.proceed()
71
72def proceed(self):
73"""Proceed with the command if one object was selected."""
74if self.call is not None:
75self.end_callbacks(self.call)
76sel = Gui.Selection.getSelection()
77allSketches = True
78allDraft = True
79Gui.addModule("Draft")
80for obj in sel:
81if obj.isDerivedFrom("Sketcher::SketchObject"):
82allDraft = False
83elif (obj.isDerivedFrom("Part::Part2DObjectPython")
84or obj.isDerivedFrom("Part::Feature")):
85allSketches = False
86else:
87allDraft = False
88allSketches = False
89
90if not sel:
91return
92elif allDraft:
93_cmd = "Draft.make_sketch"
94_cmd += "("
95_cmd += "FreeCADGui.Selection.getSelection(), "
96_cmd += "autoconstraints=True"
97_cmd += ")"
98_cmd_list = ['sk = ' + _cmd,
99'FreeCAD.ActiveDocument.recompute()']
100self.commit(translate("draft", "Convert to Sketch"),
101_cmd_list)
102elif allSketches:
103n = 0
104_cmd_list = list()
105for o in sel:
106_cmd = "Draft.draftify"
107_cmd += "("
108_cmd += "FreeCAD.ActiveDocument." + o.Name + ", "
109_cmd += "delete=False"
110_cmd += ")"
111_cmd_list.append("df" + str(n) + " = " + _cmd)
112n += 1
113
114_cmd_list.append('FreeCAD.ActiveDocument.recompute()')
115self.commit(translate("draft", "Convert to Draft"),
116_cmd_list)
117else:
118_cmd_list = list()
119n = 0
120for obj in sel:
121_cmd_df = "Draft.draftify"
122_cmd_df += "("
123_cmd_df += "FreeCAD.ActiveDocument." + obj.Name + ", "
124_cmd_df += "delete=False"
125_cmd_df += ")"
126
127_cmd_sk = "Draft.make_sketch"
128_cmd_sk += "("
129_cmd_sk += "FreeCAD.ActiveDocument." + obj.Name + ", "
130_cmd_sk += "autoconstraints=True"
131_cmd_sk += ")"
132
133if obj.isDerivedFrom("Sketcher::SketchObject"):
134_cmd_list.append("obj" + str(n) + " = " + _cmd_df)
135elif (obj.isDerivedFrom("Part::Part2DObjectPython")
136or obj.isDerivedFrom("Part::Feature")):
137_cmd_list.append("obj" + str(n) + " = " + _cmd_sk)
138#elif obj.isDerivedFrom("Part::Feature"):
139# # if (len(obj.Shape.Wires) == 1
140# # or len(obj.Shape.Edges) == 1):
141# _cmd_list.append("obj" + str(n) + " = " + _cmd_sk)
142n += 1
143_cmd_list.append('FreeCAD.ActiveDocument.recompute()')
144self.commit(translate("draft", "Convert Draft/Sketch"),
145_cmd_list)
146self.finish()
147
148
149Gui.addCommand('Draft_Draft2Sketch', Draft2Sketch())
150
151## @}
152