Draft: split clone from Draft.py

.


.

src/Mod/Draft/draftobjects/clone.py

@@ -0,0 +1,131 @@
+# ***************************************************************************
+# *   Copyright (c) 2009, 2010 Yorik van Havre <yorik@uncreated.net>        *
+# *   Copyright (c) 2009, 2010 Ken Cline <cline@frii.com>                   *
+# *   Copyright (c) 2020 FreeCAD Developers                                 *
+# *                                                                         *
+# *   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                                                                   *
+# *                                                                         *
+# ***************************************************************************
+"""This module provides the object code for Draft Clone.
+"""
+## @package clone
+# \ingroup DRAFT
+# \brief This module provides the object code for Draft Clone.
+
+from PySide.QtCore import QT_TRANSLATE_NOOP
+
+import FreeCAD as App
+
+import DraftVecUtils
+
+from draftobjects.base import DraftObject
+
+
+class Clone(DraftObject):
+    """The Clone object"""
+
+    def __init__(self,obj):
+        super(Clone, self).__init__(obj, "Clone")
+
+        _tip = "The objects included in this clone"
+        obj.addProperty("App::PropertyLinkListGlobal", "Objects",
+                        "Draft", QT_TRANSLATE_NOOP("App::Property", _tip))
+
+        _tip = "The scale factor of this clone"
+        obj.addProperty("App::PropertyVector", "Scale",
+                        "Draft", QT_TRANSLATE_NOOP("App::Property", _tip))
+
+        _tip = "If Clones includes several objects,\n\
+                set True for fusion or False for compound"
+        obj.addProperty("App::PropertyBool", "Fuse",
+                        "Draft", QT_TRANSLATE_NOOP("App::Property", _tip))
+        
+        obj.Scale = App.Vector(1,1,1)
+
+    def join(self,obj,shapes):
+        fuse = getattr(obj, 'Fuse', False)
+        if fuse:
+            tmps = []
+            for s in shapes:
+                tmps += s.Solids
+            if not tmps:
+                for s in shapes:
+                    tmps += s.Faces
+                if not tmps:
+                    for s in shapes:
+                        tmps += s.Edges
+            shapes = tmps
+        if len(shapes) == 1:
+            return shapes[0]
+        import Part
+        if fuse:
+            try:
+                sh = shapes[0].multiFuse(shapes[1:])
+                sh = sh.removeSplitter()
+            except:
+                pass
+            else:
+                return sh
+        return Part.makeCompound(shapes)
+
+    def execute(self,obj):
+        import Part
+        pl = obj.Placement
+        shapes = []
+        if obj.isDerivedFrom("Part::Part2DObject"):
+            # if our clone is 2D, make sure all its linked geometry is 2D too
+            for o in obj.Objects:
+                if not o.getLinkedObject(True).isDerivedFrom("Part::Part2DObject"):
+                    App.Console.PrintWarning("Warning 2D Clone "+obj.Name+" contains 3D geometry")
+                    return
+        for o in obj.Objects:
+            sh = Part.getShape(o)
+            if not sh.isNull():
+                shapes.append(sh)
+        if shapes:
+            sh = self.join(obj, shapes)
+            m = App.Matrix()
+            if hasattr(obj,"Scale") and not sh.isNull():
+                sx,sy,sz = obj.Scale
+                if not DraftVecUtils.equals(obj.Scale,App.Vector(1, 1, 1)):
+                    op = sh.Placement
+                    sh.Placement = App.Placement()
+                    m.scale(obj.Scale)
+                    if sx == sy == sz:
+                        sh.transformShape(m)
+                    else:
+                        sh = sh.transformGeometry(m)
+                    sh.Placement = op
+            obj.Shape = sh
+
+        obj.Placement = pl
+        if hasattr(obj,"positionBySupport"):
+            obj.positionBySupport()
+
+    def getSubVolume(self,obj,placement=None):
+        # this allows clones of arch windows to return a subvolume too
+        if obj.Objects:
+            if hasattr(obj.Objects[0],"Proxy"):
+                if hasattr(obj.Objects[0].Proxy, "getSubVolume"):
+                    if not placement:
+                        # clones must displace the original subvolume too
+                        placement = obj.Placement
+                    return obj.Objects[0].Proxy.getSubVolume(obj.Objects[0], placement)
+        return None
+
+
+_Clone = Clone