File size: 10,346 Bytes
985c397 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 | # SPDX-License-Identifier: LGPL-2.1-or-later
# ***************************************************************************
# * *
# * Copyright (c) 2025 Furgo *
# * *
# * This file is part of FreeCAD. *
# * *
# * FreeCAD is free software: you can redistribute it and/or modify it *
# * under the terms of the GNU Lesser General Public License as *
# * published by the Free Software Foundation, either version 2.1 of the *
# * License, or (at your option) any later version. *
# * *
# * FreeCAD 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 *
# * Lesser General Public License for more details. *
# * *
# * You should have received a copy of the GNU Lesser General Public *
# * License along with FreeCAD. If not, see *
# * <https://www.gnu.org/licenses/>. *
# * *
# ***************************************************************************
import FreeCAD
import Arch
import Part
from bimtests import TestArchBase
class TestArchRebar(TestArchBase.TestArchBase):
def _create_sketch(self, add_line=True, length=500.0):
"""Helper function to create a basic sketch."""
sketch = self.document.addObject("Sketcher::SketchObject")
if add_line:
sketch.addGeometry(
Part.LineSegment(FreeCAD.Vector(0, 0, 0), FreeCAD.Vector(length, 0, 0)), False
)
self.document.recompute()
return sketch
def _create_host_structure(self, length=1000.0, width=200.0, height=300.0):
"""Helper function to create a basic host structure."""
host = Arch.makeStructure(length=length, width=width, height=height)
self.document.recompute()
return host
def test_makeRebar_default_name(self):
"""Test makeRebar creates rebar with default 'Rebar' Label.
See https://github.com/FreeCAD/FreeCAD/issues/21670
"""
sketch = self._create_sketch()
rebar = Arch.makeRebar(sketch=sketch, diameter=6)
self.document.recompute()
self.assertIsNotNone(rebar, "makeRebar failed to create a rebar object.")
self.assertEqual(rebar.Label, "Rebar", "Rebar default label should be 'Rebar'.")
self.assertTrue(
rebar.Name.startswith("Rebar"), "Rebar internal name should start with 'Rebar'."
)
def test_makeRebar_custom_name(self):
"""Test makeRebar with a custom name sets Label and Mark correctly."""
sketch = self._create_sketch()
custom_name = "MyCustomRebar"
rebar = Arch.makeRebar(sketch=sketch, diameter=16, amount=5, name=custom_name)
self.document.recompute()
self.assertIsNotNone(rebar, "makeRebar failed to create a rebar object.")
self.assertEqual(rebar.Label, custom_name, "Rebar label is incorrect.")
self.assertEqual(rebar.Mark, custom_name, "Rebar mark should match label by default.")
def test_makeRebar_with_sketch_and_host(self):
"""Test makeRebar with both sketch and host object links them correctly."""
sketch_len = 800.0
host_len = 1000.0
sketch = self._create_sketch(length=sketch_len)
host = self._create_host_structure(length=host_len)
rebar = Arch.makeRebar(baseobj=host, sketch=sketch, diameter=10, amount=3, offset=20)
self.document.recompute()
self.assertIsNotNone(rebar, "makeRebar failed.")
self.assertEqual(rebar.Host, host, "Rebar.Host is not set correctly.")
self.assertEqual(rebar.Base, sketch, "Rebar.Base (sketch) is not set correctly.")
self.assertAlmostEqual(rebar.Diameter.Value, 10.0, delta=1e-9)
self.assertEqual(rebar.Amount, 3)
self.assertAlmostEqual(rebar.OffsetStart.Value, 20.0, delta=1e-9)
self.assertTrue(hasattr(rebar, "Shape"), "Rebar should have a Shape attribute.")
self.assertIsNotNone(rebar.Shape, "Rebar.Shape should not be None after recompute.")
self.assertTrue(rebar.Shape.isValid(), "Rebar.Shape should be valid.")
self.assertGreater(rebar.Shape.Length, 0, "Rebar shape seems to have no length.")
self.assertGreater(
rebar.TotalLength.Value, 0, "Rebar total length should be greater than 0."
)
def test_makeRebar_with_sketch_only(self):
"""Test makeRebar with only sketch results in no host and valid geometry."""
sketch = self._create_sketch()
rebar = Arch.makeRebar(sketch=sketch, diameter=8)
self.document.recompute()
self.assertIsNotNone(rebar, "makeRebar failed.")
self.assertIsNone(rebar.Host, "Rebar.Host should be None.")
self.assertEqual(rebar.Base, sketch, "Rebar.Base (sketch) is not set correctly.")
self.assertAlmostEqual(rebar.Diameter.Value, 8.0, delta=1e-9)
self.assertIsNotNone(rebar.Shape)
self.assertTrue(rebar.Shape.isValid())
self.assertGreater(rebar.Shape.Length, 0)
def test_makeRebar_with_baseobj_only_shape_conversion(self):
"""Test makeRebar with baseobj only copies shape."""
host_shape_provider = self._create_host_structure()
rebar = Arch.makeRebar(baseobj=host_shape_provider)
self.document.recompute()
self.assertIsNotNone(rebar, "makeRebar failed.")
self.assertIsNone(rebar.Host, "Rebar.Host should be None in this mode.")
self.assertIsNone(rebar.Base, "Rebar.Base should be None in this mode.")
self.assertIsNotNone(rebar.Shape, "Rebar.Shape should be set from baseobj.")
self.assertTrue(rebar.Shape.isValid())
self.assertGreater(rebar.Shape.Volume, 0, "Converted rebar shape should have volume.")
def test_makeRebar_properties_set_correctly(self):
"""Test makeRebar sets dimensional and calculated properties correctly."""
sketch = self._create_sketch(length=900)
host = self._create_host_structure(length=1200)
rebar = Arch.makeRebar(baseobj=host, sketch=sketch, diameter=12.5, amount=7, offset=30.0)
# Default label "Rebar" will be used, and Mark will also be "Rebar"
self.document.recompute()
self.assertEqual(rebar.Mark, "Rebar")
self.assertAlmostEqual(rebar.Diameter.Value, 12.5, delta=1e-9)
self.assertEqual(rebar.Amount, 7)
self.assertAlmostEqual(rebar.OffsetStart.Value, 30.0, delta=1e-9)
self.assertAlmostEqual(rebar.OffsetEnd.Value, 30.0, delta=1e-9)
self.assertEqual(rebar.Host, host)
self.assertEqual(rebar.Base, sketch)
self.assertTrue(hasattr(rebar, "Length"), "Rebar should have Length property.")
self.assertTrue(hasattr(rebar, "TotalLength"), "Rebar should have TotalLength property.")
self.assertAlmostEqual(
rebar.Length.Value,
sketch.Shape.Length,
delta=1e-9,
msg="Single rebar length should match sketch length.",
)
expected_total_length = rebar.Length.Value * rebar.Amount
self.assertAlmostEqual(
rebar.TotalLength.Value,
expected_total_length,
delta=1e-9,
msg="TotalLength calculation is incorrect.",
)
def test_makeRebar_no_sketch_no_baseobj(self):
"""Test makeRebar with no sketch/baseobj creates object with no meaningful shape."""
rebar = Arch.makeRebar(diameter=10, amount=1)
self.document.recompute()
self.assertIsNotNone(rebar, "makeRebar should create an object even with minimal inputs.")
self.assertIsNone(rebar.Base, "Rebar.Base should be None.")
self.assertIsNone(rebar.Host, "Rebar.Host should be None.")
self.assertAlmostEqual(rebar.Diameter.Value, 10.0, delta=1e-9)
self.assertEqual(rebar.Amount, 1)
# A Part::FeaturePython object always has a .Shape attribute.
# For a rebar with no Base, its execute() returns early, so no geometry is made.
# The Shape should be the default null/empty shape.
self.assertIsNotNone(rebar.Shape, "Rebar object should always have a Shape attribute.")
self.assertTrue(
rebar.Shape.isNull(), "Rebar.Shape should be a null shape if no Base is provided."
)
def test_RebarMe(self):
"""
Tests the creation of an Arch Rebar
"""
import Sketcher
self.printTestMessage("Checking Arch Rebar...\n")
s = Arch.makeStructure(length=2, width=3, height=5)
sk = self.document.addObject("Sketcher::SketchObject", "Sketch")
sk.AttachmentSupport = (s, ["Face6"])
sk.addGeometry(
Part.LineSegment(FreeCAD.Vector(-0.85, 1.25, 0), FreeCAD.Vector(0.75, 1.25, 0))
)
sk.addGeometry(
Part.LineSegment(FreeCAD.Vector(0.75, 1.25, 0), FreeCAD.Vector(0.75, -1.20, 0))
)
sk.addGeometry(
Part.LineSegment(FreeCAD.Vector(0.75, -1.20, 0), FreeCAD.Vector(-0.85, -1.20, 0))
)
sk.addGeometry(
Part.LineSegment(FreeCAD.Vector(-0.85, -1.20, 0), FreeCAD.Vector(-0.85, 1.25, 0))
)
sk.addConstraint(Sketcher.Constraint("Coincident", 0, 2, 1, 1))
sk.addConstraint(Sketcher.Constraint("Coincident", 1, 2, 2, 1))
sk.addConstraint(Sketcher.Constraint("Coincident", 2, 2, 3, 1))
sk.addConstraint(Sketcher.Constraint("Coincident", 3, 2, 0, 1))
self.document.recompute()
r = Arch.makeRebar(s, sk, diameter=0.1, amount=2)
self.document.recompute()
self.assertTrue(r, "Arch Rebar creation failed")
self.assertFalse(r.Shape.isNull(), "Rebar shape is null")
|