New tool: Invert lattice

InitGui.py

@@ -53,6 +53,8 @@ class LatticeWorkbench (Workbench):
         cmdsArrayTools = cmdsArrayTools + latticePolarArray.exportedCommands        
         import latticeArrayFromShape
         cmdsArrayTools = cmdsArrayTools + latticeArrayFromShape.exportedCommands        
+        import latticeInvert
+        cmdsArrayTools = cmdsArrayTools + latticeInvert.exportedCommands        
         import latticeApply
         cmdsArrayTools = cmdsArrayTools + latticeApply.exportedCommands
         import latticeCompose

latticeInvert.py

@@ -0,0 +1,155 @@
+#***************************************************************************
+#*                                                                         *
+#*   Copyright (c) 2015 - Victor Titov (DeepSOIC)                          *
+#*                                               <vv.titov@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                                                                   *
+#*                                                                         *
+#***************************************************************************
+
+__title__="Lattice Invert object: creates an array of placements from a compound."
+__author__ = "DeepSOIC"
+__url__ = ""
+
+import math
+
+import FreeCAD as App
+import Part
+
+from latticeCommon import *
+import latticeBaseFeature
+import latticeCompoundExplorer as LCE
+import latticeGeomUtils as Utils
+
+# -------------------------- document object --------------------------------------------------
+
+def makeLatticeInvert(name):
+    '''makeLatticeInvert(name): makes a LatticeInvert object.'''
+    return latticeBaseFeature.makeLatticeFeature(name, LatticeInvert, ViewProviderInvert)
+
+class LatticeInvert(latticeBaseFeature.LatticeFeature):
+    "The Lattice Invert object"
+    
+    def derivedInit(self,obj):
+        self.Type = "LatticeInvert"
+                
+        obj.addProperty("App::PropertyLink","Base","Lattice Invert","Lattice, all the placements of which are to be inverted.")
+                
+        obj.addProperty("App::PropertyEnumeration","TranslateMode","Lattice Invert","What to do with translation part of placements")
+        obj.TranslateMode = ['invert', 'keep', 'reset']
+        obj.TranslateMode = 'invert'
+        
+        obj.addProperty("App::PropertyEnumeration","OrientMode","Lattice Invert","what to do with orientation part of placements")
+        obj.OrientMode = ['invert', 'keep', 'reset']
+        obj.OrientMode = 'invert'
+
+    def derivedExecute(self,obj):
+        # cache stuff
+        base = obj.Base.Shape
+        if not latticeBaseFeature.isObjectLattice(obj.Base):
+            App.Console.PrintWarning(obj.Name+": Base is not a lattice, but lattice is expected. Results may be unexpected.\n")
+        baseChildren = LCE.AllLeaves(base)
+                        
+        #cache mode comparisons, for speed
+        posIsInvert = obj.TranslateMode == 'invert'
+        posIsKeep = obj.TranslateMode == 'keep'
+        posIsReset = obj.TranslateMode == 'reset'
+        
+        oriIsInvert = obj.OrientMode == 'invert'
+        oriIsKeep = obj.OrientMode == 'keep'
+        oriIsReset = obj.OrientMode == 'reset'
+        
+        # initialize output containers and loop variables
+        outputPlms = [] #list of placements
+        
+        # the essence
+        for child in baseChildren:
+            pos = App.Vector()
+            ori = App.Rotation()
+            inverted = child.Placement.inverse()
+            if posIsInvert:
+                pos = inverted.Base
+            elif posIsKeep:
+                pos = child.Placement.Base
+            elif posIsReset:
+                pass
+            
+            if oriIsInvert:
+                ori = inverted.Rotation
+            elif oriIsKeep:
+                ori = child.Placement.Rotation
+            elif oriIsReset:
+                pass
+                
+            plm = App.Placement(pos, ori)
+            outputPlms.append(plm)
+        return outputPlms
+
+
+class ViewProviderInvert(latticeBaseFeature.ViewProviderLatticeFeature):
+        
+    def getIcon(self):
+        return getIconPath('Lattice_Invert.svg')
+
+# -------------------------- /document object --------------------------------------------------
+
+# -------------------------- Gui command --------------------------------------------------
+
+def CreateLatticeInvert(name):
+    sel = FreeCADGui.Selection.getSelectionEx()
+    FreeCAD.ActiveDocument.openTransaction("Create LatticeInvert")
+    FreeCADGui.addModule("latticeInvert")
+    FreeCADGui.doCommand("f = latticeInvert.makeLatticeInvert(name='"+name+"')")
+    FreeCADGui.doCommand("f.Base = App.ActiveDocument."+sel[0].ObjectName)
+    FreeCADGui.doCommand("for child in f.ViewObject.Proxy.claimChildren():\n"+
+                         "    child.ViewObject.hide()")
+    FreeCADGui.doCommand("f.Proxy.execute(f)")
+    FreeCADGui.doCommand("f.purgeTouched()")
+    FreeCADGui.doCommand("f = None")
+    FreeCAD.ActiveDocument.commitTransaction()
+
+
+class _CommandLatticeInvert:
+    "Command to create LatticeInvert feature"
+    def GetResources(self):
+        return {'Pixmap'  : getIconPath("Lattice_Invert.svg"),
+                'MenuText': QtCore.QT_TRANSLATE_NOOP("Lattice_Invert","Invert lattice"),
+                'Accel': "",
+                'ToolTip': QtCore.QT_TRANSLATE_NOOP("Lattice_Invert","Lattice Invert: invert all placements in a lattice object.")}
+        
+    def Activated(self):
+        if len(FreeCADGui.Selection.getSelection()) == 1 :
+            CreateLatticeInvert(name = "Invert")
+        else:
+            mb = QtGui.QMessageBox()
+            mb.setIcon(mb.Icon.Warning)
+            mb.setText(translate("Lattice_Invert", "Please select one object, first. The object must be a lattice object (array of placements).", None))
+            mb.setWindowTitle(translate("Lattice_Invert","Bad selection", None))
+            mb.exec_()
+            
+    def IsActive(self):
+        if FreeCAD.ActiveDocument:
+            return True
+        else:
+            return False
+            
+FreeCADGui.addCommand('Lattice_Invert', _CommandLatticeInvert())
+
+exportedCommands = ['Lattice_Invert']
+
+# -------------------------- /Gui command --------------------------------------------------
+