FreeCAD

1
# -*- coding: utf-8 -*-
2
# ***************************************************************************
3
# *   Copyright (c) 2019 sliptonic <shopinthewoods@gmail.com>               *
4
# *                                                                         *
5
# *   This program is free software; you can redistribute it and/or modify  *
6
# *   it under the terms of the GNU Lesser General Public License (LGPL)    *
7
# *   as published by the Free Software Foundation; either version 2 of     *
8
# *   the License, or (at your option) any later version.                   *
9
# *   for detail see the LICENCE text file.                                 *
10
# *                                                                         *
11
# *   This program is distributed in the hope that it will be useful,       *
12
# *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
13
# *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
14
# *   GNU Library General Public License for more details.                  *
15
# *                                                                         *
16
# *   You should have received a copy of the GNU Library General Public     *
17
# *   License along with this program; if not, write to the Free Software   *
18
# *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  *
19
# *   USA                                                                   *
20
# *                                                                         *
21
# ***************************************************************************
22

23
import Draft
24
import FreeCAD
25
import Path
26
import Path.Main.Job as PathJob
27
import Path.Op.Helix as PathHelix
28
import Tests.PathTestUtils as PathTestUtils
29

30
Path.Log.setLevel(Path.Log.Level.INFO, Path.Log.thisModule())
31
# Path.Log.trackModule(Path.Log.thisModule())
32

33

34
class TestPathHelix(PathTestUtils.PathTestBase):
35
    RotateBy = 45
36

37
    def setUp(self):
38
        self.clone = None
39
        self.doc = FreeCAD.open(FreeCAD.getHomePath() + "Mod/CAM/Tests/test_holes00.fcstd")
40
        self.job = PathJob.Create("Job", [self.doc.Body])
41

42
    def tearDown(self):
43
        FreeCAD.closeDocument(self.doc.Name)
44

45
    def test00(self):
46
        """Verify Helix does not throw an exception."""
47

48
        op = PathHelix.Create("Helix")
49
        op.Proxy.execute(op)
50

51
    def test01(self):
52
        """Verify Helix generates proper holes from model"""
53

54
        op = PathHelix.Create("Helix")
55
        proxy = op.Proxy
56
        for base in op.Base:
57
            model = base[0]
58
            for sub in base[1]:
59
                pos = proxy.holePosition(op, model, sub)
60
                self.assertRoughly(round(pos.Length / 10, 0), proxy.holeDiameter(op, model, sub))
61

62
    def test02(self):
63
        """Verify Helix generates proper holes for rotated model"""
64

65
        self.job.Tools.Group[0].Tool.Diameter = 0.5
66

67
        op = PathHelix.Create("Helix")
68
        proxy = op.Proxy
69
        model = self.job.Model.Group[0]
70

71
        for deg in range(self.RotateBy, 360, self.RotateBy):
72
            model.Placement.Rotation = FreeCAD.Rotation(deg, 0, 0)
73
            for base in op.Base:
74
                model = base[0]
75
                for sub in base[1]:
76
                    pos = proxy.holePosition(op, model, sub)
77
                    # Path.Log.track(deg, pos, pos.Length)
78
                    self.assertRoughly(
79
                        round(pos.Length / 10, 0), proxy.holeDiameter(op, model, sub)
80
                    )
81

82
    def test03(self):
83
        """Verify Helix generates proper holes for rotated base model"""
84

85
        for deg in range(self.RotateBy, 360, self.RotateBy):
86
            self.tearDown()
87
            self.doc = FreeCAD.open(FreeCAD.getHomePath() + "Mod/CAM/Tests/test_holes00.fcstd")
88
            self.doc.Body.Placement.Rotation = FreeCAD.Rotation(deg, 0, 0)
89

90
            self.job = PathJob.Create("Job", [self.doc.Body])
91
            self.job.Tools.Group[0].Tool.Diameter = 0.5
92

93
            op = PathHelix.Create("Helix")
94
            proxy = op.Proxy
95
            model = self.job.Model.Group[0]
96

97
            for base in op.Base:
98
                model = base[0]
99
                for sub in base[1]:
100
                    pos = proxy.holePosition(op, model, sub)
101
                    # Path.Log.track(deg, pos, pos.Length)
102
                    self.assertRoughly(
103
                        round(pos.Length / 10, 0), proxy.holeDiameter(op, model, sub)
104
                    )
105

106
    def test04(self):
107
        """Verify Helix generates proper holes for rotated clone base model"""
108
        for deg in range(self.RotateBy, 360, self.RotateBy):
109
            self.tearDown()
110
            self.doc = FreeCAD.open(FreeCAD.getHomePath() + "Mod/CAM/Tests/test_holes00.fcstd")
111
            self.clone = Draft.clone(self.doc.Body)
112
            self.clone.Placement.Rotation = FreeCAD.Rotation(deg, 0, 0)
113

114
            self.job = PathJob.Create("Job", [self.clone])
115
            self.job.Tools.Group[0].Tool.Diameter = 0.5
116

117
            op = PathHelix.Create("Helix")
118
            proxy = op.Proxy
119
            model = self.job.Model.Group[0]
120

121
            for base in op.Base:
122
                model = base[0]
123
                for sub in base[1]:
124
                    pos = proxy.holePosition(op, model, sub)
125
                    # Path.Log.track(deg, pos, pos.Length)
126
                    self.assertRoughly(
127
                        round(pos.Length / 10, 0), proxy.holeDiameter(op, model, sub)
128
                    )
129