# *************************************************************************** # * Copyright (c) 2011 Juergen Riegel * # * * # * 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")