170 lines
7.9 KiB
Python
170 lines
7.9 KiB
Python
#***************************************************************************
|
|
#* Copyright (c) 2011 Juergen Riegel <FreeCAD@juergen-riegel.net> *
|
|
#* *
|
|
#* 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 unittest
|
|
|
|
import FreeCAD
|
|
from pivy import coin
|
|
|
|
App = FreeCAD
|
|
|
|
class TestBoolean(unittest.TestCase):
|
|
def setUp(self):
|
|
self.Doc = FreeCAD.newDocument("PartDesignTestBoolean")
|
|
|
|
def testBooleanFuseCase(self):
|
|
self.Body = self.Doc.addObject('PartDesign::Body','Body')
|
|
self.Box = self.Doc.addObject('PartDesign::AdditiveBox','Box')
|
|
self.Box.Length=10
|
|
self.Box.Width=10
|
|
self.Box.Height=10
|
|
self.Body.addObject(self.Box)
|
|
self.Doc.recompute()
|
|
self.Body001 = self.Doc.addObject('PartDesign::Body','Body001')
|
|
self.Box001 = self.Doc.addObject('PartDesign::AdditiveBox','Box001')
|
|
self.Box001.Length=10
|
|
self.Box001.Width=10
|
|
self.Box001.Height=10
|
|
self.Box001.Placement.Base = App.Vector(-5,0,0)
|
|
self.Body001.addObject(self.Box001)
|
|
self.Doc.recompute()
|
|
self.BooleanFuse = self.Doc.addObject('PartDesign::Boolean','BooleanFuse')
|
|
self.Body001.addObject(self.BooleanFuse)
|
|
self.Doc.recompute()
|
|
self.BooleanFuse.setObjects([self.Body,])
|
|
self.BooleanFuse.Type = 0
|
|
self.Doc.recompute()
|
|
self.assertAlmostEqual(self.BooleanFuse.Shape.Volume, 1500)
|
|
|
|
def testBooleanCutCase(self):
|
|
self.Body = self.Doc.addObject('PartDesign::Body','Body')
|
|
self.Box = self.Doc.addObject('PartDesign::AdditiveBox','Box')
|
|
self.Box.Length=10
|
|
self.Box.Width=10
|
|
self.Box.Height=10
|
|
self.Body.addObject(self.Box)
|
|
self.Doc.recompute()
|
|
self.Body001 = self.Doc.addObject('PartDesign::Body','Body001')
|
|
self.Box001 = self.Doc.addObject('PartDesign::AdditiveBox','Box001')
|
|
self.Box001.Length=10
|
|
self.Box001.Width=10
|
|
self.Box001.Height=10
|
|
self.Box001.Placement.Base = App.Vector(-5,0,0)
|
|
self.Body001.addObject(self.Box001)
|
|
self.Doc.recompute()
|
|
self.BooleanCut = self.Doc.addObject('PartDesign::Boolean','BooleanCut')
|
|
self.Body001.addObject(self.BooleanCut)
|
|
self.Doc.recompute()
|
|
self.BooleanCut.setObjects([self.Body,])
|
|
self.BooleanCut.Type = 1
|
|
self.Doc.recompute()
|
|
self.assertAlmostEqual(self.BooleanCut.Shape.Volume, 500)
|
|
|
|
def testBooleanCommonCase(self):
|
|
self.Body = self.Doc.addObject('PartDesign::Body','Body')
|
|
self.Box = self.Doc.addObject('PartDesign::AdditiveBox','Box')
|
|
self.Box.Length=10
|
|
self.Box.Width=10
|
|
self.Box.Height=10
|
|
self.Body.addObject(self.Box)
|
|
self.Doc.recompute()
|
|
self.Body001 = self.Doc.addObject('PartDesign::Body','Body001')
|
|
self.Box001 = self.Doc.addObject('PartDesign::AdditiveBox','Box001')
|
|
self.Box001.Length=10
|
|
self.Box001.Width=10
|
|
self.Box001.Height=10
|
|
self.Box001.Placement.Base = App.Vector(-5,0,0)
|
|
self.Body001.addObject(self.Box001)
|
|
self.Doc.recompute()
|
|
self.BooleanCommon = self.Doc.addObject('PartDesign::Boolean','BooleanCommon')
|
|
self.Body001.addObject(self.BooleanCommon)
|
|
self.Doc.recompute()
|
|
self.BooleanCommon.setObjects([self.Body,])
|
|
self.BooleanCommon.Type = 2
|
|
self.Doc.recompute()
|
|
self.assertAlmostEqual(self.BooleanCommon.Shape.Volume, 500)
|
|
|
|
def testBooleanSelectUI(self):
|
|
if not App.GuiUp:
|
|
return
|
|
# Arrange
|
|
self.Body = self.Doc.addObject('PartDesign::Body','Body')
|
|
self.Box = self.Doc.addObject('PartDesign::AdditiveBox','Box')
|
|
self.Box.Length=10
|
|
self.Box.Width=10
|
|
self.Box.Height=10
|
|
self.Body.addObject(self.Box)
|
|
self.Doc.recompute()
|
|
self.Body001 = self.Doc.addObject('PartDesign::Body','Body001')
|
|
self.Box001 = self.Doc.addObject('PartDesign::AdditiveBox','Box001')
|
|
self.Box001.Length=10
|
|
self.Box001.Width=10
|
|
self.Box001.Height=10
|
|
self.Box001.Placement.Base = App.Vector(-5,0,0)
|
|
self.Body001.addObject(self.Box001)
|
|
self.Body002 = self.Doc.addObject('PartDesign::Body','Body002')
|
|
self.BooleanFuse = self.Doc.addObject('PartDesign::Boolean','BooleanFuse')
|
|
self.BooleanFuse.Group = [ self.Body, self.Body001 ]
|
|
self.BooleanFuse.ViewObject.Display="Result" # TODO: This is a required redundancy or it doesn't init right
|
|
self.Body002.addObject(self.BooleanFuse)
|
|
self.Doc.recompute()
|
|
# Act
|
|
App.Gui.Selection.addSelection("", self.Body002.Name, "Face2")
|
|
App.Gui.Selection.addSelection("", self.Body002.Name, "Face6")
|
|
App.Gui.Selection.addSelection("", self.Body002.Name, "Face10")
|
|
App.Gui.Selection.addSelection("", self.Body002.Name, "Face3")
|
|
App.Gui.Selection.addSelection("", self.Body002.Name, "Face7")
|
|
App.Gui.Selection.addSelection("", self.Body002.Name, "Face11")
|
|
App.Gui.Selection.addSelection("", "Body002", "Face14")
|
|
App.Gui.updateGui()
|
|
# Assert
|
|
self.assertEqual(
|
|
len(App.Gui.Selection.getSelectionEx("", 0)[0].SubElementNames), 7
|
|
)
|
|
self.assertEqual(
|
|
App.Gui.Selection.getSelectionEx("", 0)[0].SubElementNames[0][-8:],
|
|
",F.Face2",
|
|
)
|
|
# Ideally we would be able to check and see the selection color set, but there is currently no way to
|
|
# see this either from python of c++ without either tapping into the coin GL renderer or writing a new
|
|
# test renderer. Prototypes of code moving in this direction would be:
|
|
# compSel = App.Gui.Selection.getCompleteSelection()
|
|
# selection0 = compSel[0]
|
|
# selection00Object = selection0.Object.ViewObject
|
|
# path = coin.SoPath()
|
|
# foundCount = App.Gui.ActiveDocument.Box.getDetailPath("Face14.",path,False)
|
|
# pathLength = path.getLength()
|
|
# # path.getHead() or path.getTail() or path.getNode(n)
|
|
# childCount = path.getHead().getNumChildren()
|
|
# children = path.getHead().getChildren() # Some pivy objects support this.
|
|
# child = children.get(0)
|
|
# childOpt2 = path.getHead().getChild(0) # And others support this. And some support both.
|
|
# rgb = App.Gui.ActiveDocument.Body002.RootNode.getChild(2).getChild(1).getChild(1).getChild(0).diffuseColor.getValues()[0].getValue()
|
|
# # Assert the rgb color value.
|
|
|
|
def tearDown(self):
|
|
if hasattr(App, "KeepTestDoc") and App.KeepTestDoc:
|
|
return
|
|
#closing doc.
|
|
FreeCAD.closeDocument("PartDesignTestBoolean")
|
|
|
|
|