# *************************************************************************** # * 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 math import FreeCAD import TestSketcherApp App = FreeCAD class TestMultiTransform(unittest.TestCase): def setUp(self): self.Doc = FreeCAD.newDocument("PartDesignTestMultiTransform") FreeCAD.ConfigSet("SuppressRecomputeRequiredDialog", "True") def testMultiTransform(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") # Make first offset cube Pad self.PadSketch = self.Doc.addObject("Sketcher::SketchObject", "SketchPad") self.Body.addObject(self.PadSketch) TestSketcherApp.CreateRectangleSketch(self.PadSketch, (0, 0), (10, 10)) self.Doc.recompute() self.Pad = self.Doc.addObject("PartDesign::Pad", "Pad") self.Body.addObject(self.Pad) self.Pad.Profile = self.PadSketch self.Pad.Length = 10 self.Doc.recompute() self.MultiTransform = self.Doc.addObject("PartDesign::MultiTransform", "MultiTransform") self.Doc.recompute() self.MultiTransform.Originals = [self.Pad] self.MultiTransform.Shape = self.Pad.Shape self.Body.addObject(self.MultiTransform) self.Doc.recompute() self.Mirrored = self.Doc.addObject("PartDesign::Mirrored", "Mirrored") self.Mirrored.MirrorPlane = (self.PadSketch, ["H_Axis"]) self.Body.addObject(self.Mirrored) self.LinearPattern = self.Doc.addObject("PartDesign::LinearPattern", "LinearPattern") self.LinearPattern.Direction = (self.PadSketch, ["H_Axis"]) self.LinearPattern.Length = 20 self.LinearPattern.Occurrences = 3 self.Body.addObject(self.LinearPattern) self.PolarPattern = self.Doc.addObject("PartDesign::PolarPattern", "PolarPattern") self.PolarPattern.Axis = (self.PadSketch, ["N_Axis"]) self.PolarPattern.Angle = 360 self.PolarPattern.Occurrences = 4 self.Body.addObject(self.PolarPattern) self.MultiTransform.Transformations = [self.Mirrored, self.LinearPattern, self.PolarPattern] self.Doc.recompute() self.assertAlmostEqual(self.MultiTransform.Shape.Volume, 20000) def testMultiTransformDressup(self): # Arrange Doc = self.Doc Body = Doc.addObject("PartDesign::Body", "Body") Body.AllowCompound = False # Make first offset cube Pad PadSketch = Doc.addObject("Sketcher::SketchObject", "SketchPad") Body.addObject(PadSketch) xw = yw = zw = 10 TestSketcherApp.CreateRectangleSketch(PadSketch, (0, 0), (xw, yw)) Doc.recompute() Pad = Doc.addObject("PartDesign::Pad", "Pad") Body.addObject(Pad) Pad.Profile = PadSketch Pad.Length = zw Doc.recompute() MultiTransform = Doc.addObject("PartDesign::MultiTransform", "MultiTransform") Doc.recompute() MultiTransform.Originals = [Pad] MultiTransform.Shape = Pad.Shape Body.addObject(MultiTransform) Doc.recompute() Mirrored = Doc.addObject("PartDesign::Mirrored", "Mirrored") Mirrored.MirrorPlane = (Doc.getObject("XY_Plane"), [""]) Mirrored.Refine = True Body.addObject(Mirrored) Mirrored2 = Doc.addObject("PartDesign::Mirrored", "Mirrored") Mirrored2.MirrorPlane = (Doc.getObject("XZ_Plane"), [""]) Mirrored2.Refine = True Body.addObject(Mirrored2) MultiTransform.Transformations = [Mirrored, Mirrored2] MultiTransform.Refine = True Doc.recompute() Fillet = Doc.addObject("PartDesign::Fillet", "Fillet") Fillet.Base = (MultiTransform, ["Face1", "Face2"]) radius = 3 Fillet.Radius = radius Body.addObject(Fillet) # Broken out calculation of volume with two adjacent filleted faces = 5 long edges, 2 short edges, # 2 fully rounded corners and 4 corners with only 2 fillets meeting cubeVolume = xw * yw * zw * 2 * 2 # Mirrored and mirrored again. filletOuter = radius**2 * (xw - radius * 2) # Volume of the rect prisms the fillets are in. filletCorner = radius**3 # Volume of the rect prism corners qRoundArea = math.pi * radius**2 / 4 # Area of the quarter round fillet profile filletPrism = qRoundArea * (xw - radius * 2) # Volume of fillet minus corners fillet3Corner = ( math.pi * radius**3 * 4 / 3 / 8 ) # Volume of a fully rounded corner ( Sphere / 8 ) fillet2Corner = radius**2 * 2 # Volume of corner with two fillets intersecting fillet1Corner = ( math.pi * radius**2 / 4 * radius ) # Volume of corner with stopped single fillet extraFillet = qRoundArea * xw # extra fillet in a mirrored direction filletOuterExt = radius**2 * 10 # extra rect prim surrounding fillet rectBox = ( cubeVolume - (4 + 3) * (filletOuter + filletCorner) - 5 * filletOuterExt + filletCorner ) fillets = ( (4 + 3) * filletPrism + 5 * extraFillet + fillet3Corner * 2 + fillet2Corner * 4 + fillet1Corner * 0 ) volume = rectBox + fillets # Act Link = Doc.addObject("App::Link", "Link001") Link.setLink(Doc.Body) Link.Label = "Body001" Doc.recompute() # Assert self.assertAlmostEqual(abs(Body.Shape.Volume), volume, 6) self.assertAlmostEqual(abs(Link.Shape.Volume), volume, 6) def testMultiTransformBody(self): pass # TODO: Someone who understands the second mode added to transform needs to # write a test here. Maybe close to testMultiTransform but with # self.Mirrored.TransformMode="Transform body" instead of # self.Mirrored.TransformMode="Transform tools" def tearDown(self): # closing doc FreeCAD.closeDocument("PartDesignTestMultiTransform") FreeCAD.ConfigSet("SuppressRecomputeRequiredDialog", "") # print ("omit closing document for debugging")