Merge pull request #373 from kbwbe/temp

merge constraintServices from develbranch
kbwbe · Apr 4, 2020 · a2bffc9 · a2bffc9
2 parents d1a0315 + 1f7c39b
commit a2bffc9
Show file tree

Hide file tree

Showing 7 changed files with 878 additions and 9 deletions.
diff --git a/InitGui.py b/InitGui.py
@@ -26,7 +26,7 @@
 __title__ = 'A2plus assembly Workbench - InitGui file'
 __author__ = 'kbwbe'
 
-A2P_VERSION = 'V0.4.45c'
+A2P_VERSION = 'V0.4.46'
 
 import FreeCAD
 import FreeCADGui
@@ -66,6 +66,7 @@ def Initialize(self):
         import a2p_constraintDialog
         import a2p_constraintcommands
         import a2p_bom # bom == bill of materials == partslist
+        import a2p_constraintServices
 
         if a2plib.getRecursiveUpdateEnabled():
             partCommands = [
@@ -97,6 +98,7 @@ def Initialize(self):
             constraintCommands = [
                 'a2p_ConstraintDialogCommand',
                 'a2p_EditConstraintCommand',
+                'a2p_reAdjustConstraintDirectionsCommand',
                 'a2p_DeleteConnectionsCommand',
                 'a2p_PointIdentityConstraintCommand',
                 'a2p_PointOnLineConstraintCommand',
@@ -117,6 +119,7 @@ def Initialize(self):
             constraintCommands = [
                 'a2p_ConstraintDialogCommand',
                 'a2p_EditConstraintCommand',
+                'a2p_reAdjustConstraintDirectionsCommand',
                 'a2p_DeleteConnectionsCommand',
                 ]
 

diff --git a/a2p_Resources2.py b/a2p_Resources2.py
diff --git a/a2p_Resources3.py b/a2p_Resources3.py
diff --git a/a2p_constraintServices.py b/a2p_constraintServices.py
@@ -0,0 +1,110 @@
+#***************************************************************************
+#*                                                                         *
+#*   Copyright (c) 2020 kbwbe                                              *
+#*                                                                         *
+#*   Portions of code based on hamish's assembly 2                         *
+#*                                                                         *
+#*   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                                                                   *
+#*                                                                         *
+#***************************************************************************
+
+import FreeCAD
+import FreeCADGui
+from PySide import QtGui
+import a2plib
+import a2p_constraints
+
+#==============================================================================
+def redAdjustConstraintDirections(doc):
+    '''
+    recalculate value of property 'direction' and the sign of property 'offset' of all
+    a2p-constraints of a document, in order to reach a solvable state if 
+    possible, especially used after updating of imported parts.
+    '''
+    unknown_constraints = []
+    constraints = [ obj for obj in doc.Objects if 'ConstraintInfo' in obj.Content]
+    for c in constraints:
+        try: #process as much constraints as possible
+            if c.Type == 'pointIdentity':
+                a2p_constraints.PointIdentityConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'pointOnLine':
+                a2p_constraints.PointOnLineConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'pointOnPlane':
+                a2p_constraints.PointOnPlaneConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'circularEdge':
+                a2p_constraints.CircularEdgeConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'axial':
+                a2p_constraints.AxialConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'axisParallel':
+                a2p_constraints.AxisParallelConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'axisPlaneParallel':
+                a2p_constraints.AxisPlaneParallelConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'axisPlaneAngle':
+                a2p_constraints.AxisPlaneAngleConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'axisPlaneNormal':
+                a2p_constraints.AxisPlaneNormalConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'planesParallel':
+                a2p_constraints.PlanesParallelConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'plane':
+                a2p_constraints.PlaneConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'angledPlanes':
+                a2p_constraints.AngledPlanesConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'sphereCenterIdent':
+                a2p_constraints.SphericalConstraint.recalculateMatingDirection(c)
+            elif c.Type == 'CenterOfMass':
+                a2p_constraints.CenterOfMassConstraint.recalculateMatingDirection(c)
+            else:
+                unknown_constraints.append(c.Type)
+        except:
+            print("Errors occured during processing of {}".format(c.Label))
+
+    if len(unknown_constraints) > 0:
+        print("redefineConstraintDirections(): Found unknown constraints: {}".format(
+            set(unknown_constraints)
+            )
+        )
+#==============================================================================
+class a2p_reAdjustConstraintDirectionsCommand:
+    def Activated(self):
+        flags = QtGui.QMessageBox.StandardButton.Yes | QtGui.QMessageBox.StandardButton.No
+        response = QtGui.QMessageBox.information(
+            QtGui.QApplication.activeWindow(), 
+            u'Recalculate direction of constraints', 
+            u'Do you really want to recalculate\nthe directions of all constraints?', 
+            flags
+            )
+        if response == QtGui.QMessageBox.Yes:
+            doc = FreeCAD.activeDocument()
+            doc.openTransaction("Readjust constraint's directions")
+            redAdjustConstraintDirections(doc)
+            doc.commitTransaction()
+
+    def IsActive(self):
+        if FreeCAD.activeDocument() is None: return False
+        return True
+
+    def GetResources(self):
+        return {
+            'Pixmap'  : a2plib.pathOfModule()+'/icons/a2p_ReAdjustConstraints.svg',
+            'MenuText': 'Readjust directions of all constraints',
+            'ToolTip': 'Readjust directions of all constraints to fit best'
+            }
+FreeCADGui.addCommand('a2p_reAdjustConstraintDirectionsCommand', a2p_reAdjustConstraintDirectionsCommand())
+#==============================================================================
+
+
+