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
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
|
# Copyright (c) 2024, Manfred Moitzi
# License: MIT License
from __future__ import annotations
import pytest
import ezdxf
from ezdxf.layouts import Modelspace
from ezdxf import select
@pytest.fixture(scope="module")
def msp():
doc = ezdxf.new()
msp_ = doc.modelspace()
msp_.add_point((0, 1))
msp_.add_circle((0, 0), radius=5)
msp_.add_line((-1, -1), (1, 1))
msp_.add_lwpolyline([(-2, -2), (2, -2), (2, 2), (-2, 2)], close=True)
return msp_
class TestBoundingBoxVsWindow:
"""The selection functions are testing only the bounding boxes of entities.
This is a design choice: performance and simplicity over accuracy
"""
def test_all_inside(self, msp: Modelspace):
window = select.Window((-5, -5), (5, 5))
selection = select.bbox_inside(window, msp)
assert len(selection) == 4
def test_inside_select_point(self, msp: Modelspace):
window = select.Window((-1, 0), (1, 2))
selection = select.bbox_inside(window, msp)
assert len(selection) == 1
assert selection[0].dxftype() == "POINT"
def test_inside_select_point_and_line(self, msp: Modelspace):
window = select.Window((-1, -1), (1, 1))
selection = select.bbox_inside(window, msp)
assert len(selection) == 2
assert selection[0].dxftype() == "POINT"
assert selection[1].dxftype() == "LINE"
def test_outside_none(self, msp: Modelspace):
window = select.Window((-5, -5), (5, 5))
selection = select.bbox_outside(window, msp)
assert len(selection) == 0
def test_outside_all(self, msp: Modelspace):
window = select.Window((6, 6), (7, 7))
selection = select.bbox_outside(window, msp)
assert len(selection) == 4
def test_overlaps_all(self, msp: Modelspace):
"""The window just overlaps the bounding boxes of the CIRCLE and the
LWPOLYLINE not the geometry itself.
"""
window = select.Window((0, 0), (2, 2))
selection = select.bbox_overlap(window, msp)
assert len(selection) == 4
def test_overlap_selects_touching_entities(self, msp: Modelspace):
"""The window touches just the bounding box of the CIRCLE, not the curve itself."""
window = select.Window((5, 5), (6, 6))
selection = select.bbox_overlap(window, msp)
assert len(selection) == 1
assert selection[0].dxftype() == "CIRCLE"
class TestBoundingBoxVsCircle:
"""The selection functions are testing only the bounding boxes of selection.
This is a design choice: performance and simplicity over accuracy
"""
def test_all_inside(self, msp: Modelspace):
circle = select.Circle((0, 0), 10)
selection = select.bbox_inside(circle, msp)
assert len(selection) == 4
def test_all_outside(self, msp: Modelspace):
circle = select.Circle((7, 7), 1)
selection = select.bbox_outside(circle, msp)
assert len(selection) == 4
def test_none_outside(self, msp: Modelspace):
circle = select.Circle((0, 0), 1)
selection = select.bbox_outside(circle, msp)
assert len(selection) == 0
def test_outside_corner_case_point(self, msp: Modelspace):
"""Bounding boxes do overlap but point is outside circle.
point = (0, 1)
"""
circle = select.Circle((1, 0), 1)
selection = select.bbox_outside(circle, msp.query("POINT"))
assert len(selection) == 1
def test_outside_corner_case_polyline(self, msp: Modelspace):
"""All vertices are outside circle but polyline is not outside.
xCCx
CCCC
CCCC
xCCx
polyline corners = (-2, -2) ... (2, 2)
"""
circle = select.Circle((0, 0), 2)
selection = select.bbox_outside(circle, msp.query("LWPOLYLINE"))
assert len(selection) == 0
def test_overlap_all(self, msp: Modelspace):
circle = select.Circle((0, 0), 1)
selection = select.bbox_overlap(circle, msp)
assert len(selection) == 4
def test_overlap_none(self, msp: Modelspace):
circle = select.Circle((7, 0), 1)
selection = select.bbox_overlap(circle, msp)
assert len(selection) == 0
def test_is_overlapping_point(self, msp: Modelspace):
circle = select.Circle((0, 0), 1)
selection = select.bbox_overlap(circle, msp.query("POINT"))
assert len(selection) == 1
def test_is_overlapping_line(self, msp: Modelspace):
"""Bounding boxes do overlap but circle does not intersect line.
overlap tests are performed on the bounding box of the line!
CC.
C#x
.xx
line = (-1, -1) (1, 1)
"""
circle = select.Circle((-1, 1), 1)
selection = select.bbox_overlap(circle, msp.query("LINE"))
assert len(selection) == 1
class TestBoundingBoxVsPolygon:
"""The selection functions are testing only the bounding boxes of selection.
This is a design choice: performance and simplicity over accuracy
"""
@pytest.mark.parametrize(
"vertices",
[
[],
[(0, 0)],
[(0, 0), (0, 0)],
[(0, 0), (0, 0), (0, 0)],
],
)
def test_invalid_vertex_count(self, vertices):
with pytest.raises(ValueError):
select.Polygon(vertices)
def test_all_inside(self, msp: Modelspace):
polygon = select.Polygon([(-5, -5), (5, -5), (5, 5), (-5, 5)])
selection = select.bbox_inside(polygon, msp)
assert len(selection) == 4
def test_circle_not_inside(self, msp: Modelspace):
"""Bounding boxes do overlap but all corner vertices of the CIRCLE bbox are
outside.
"""
polygon = select.Polygon([(-2, -2), (2, -2), (2, 2), (-2, 2)])
selection = select.bbox_inside(polygon, msp.query("CIRCLE"))
assert len(selection) == 0
def test_circle_is_not_inside_concave_polygon(self, msp: Modelspace):
"""Bounding box of CIRCLE is completely inside the polygon, all edge vertices
are inside the polygon but some edges do intersect with the polygon.
CIRCLE is not complete inside!
"""
polygon = select.Polygon([(-10, -10), (10, -10), (10, 10), (0, 0), (-10, 10)])
selection = select.bbox_inside(polygon, msp.query("CIRCLE"))
assert len(selection) == 0
selection = select.bbox_outside(polygon, msp.query("CIRCLE"))
assert len(selection) == 0
def test_all_outside(self, msp: Modelspace):
polygon = select.Polygon([(6, 6), (7, 6), (7, 7), (6, 7)])
selection = select.bbox_outside(polygon, msp)
assert len(selection) == 4
def test_point_is_outside(self, msp: Modelspace):
"""Bounding boxes do overlap, all corner vertices of the POINT bbox are
outside and POINT is outside the polygon.
point: (0, 1)
"""
polygon = select.Polygon([(0, 0), (1, 0), (1, 1)])
selection = select.bbox_outside(polygon, msp.query("POINT"))
assert len(selection) == 1
def test_line_is_not_outside(self, msp: Modelspace):
"""Bounding boxes do overlap all corner vertices of the LINE bbox are
outside but the LINE is overlap the polygon.
line: (-2, -2) (2, 2)
"""
polygon = select.Polygon([(-1, -3), (1, -3), (0, 3)])
selection = select.bbox_outside(polygon, msp.query("LINE"))
assert len(selection) == 0
def test_all_overlap(self, msp: Modelspace):
"""All inside is also all overlap."""
polygon = select.Polygon([(-5, -5), (5, -5), (5, 5), (-5, 5)])
selection = select.bbox_overlap(polygon, msp)
assert len(selection) == 4
def test_point_not_overlaps(self, msp: Modelspace):
"""Bounding boxes do overlap, all corner vertices of the POINT bbox are
outside and POINT is outside the polygon.
point: (0, 1)
"""
polygon = select.Polygon([(0, 0), (1, 0), (1, 1)])
selection = select.bbox_overlap(polygon, msp.query("POINT"))
assert len(selection) == 0
def test_line_overlaps(self, msp: Modelspace):
"""Bounding boxes do overlap, all corner vertices of the LINE bbox are
outside and the LINE is overlap the polygon.
line: (-2, -2) (2, 2)
"""
polygon = select.Polygon([(-1, -3), (1, -3), (0, 3)])
selection = select.bbox_overlap(polygon, msp.query("LINE"))
assert len(selection) == 1
def test_circle_overlaps(self, msp: Modelspace):
"""All corner vertices of the CIRCLE bbox are outside and CIRCLE does overlap
with polygon. This is different to overlap selection in CAD applications!
circle: (0, 0) radius=5
"""
polygon = select.Polygon([(-1, -1), (1, -1), (1, 1), (-1, 1)])
selection = select.bbox_overlap(polygon, msp.query("CIRCLE"))
assert len(selection) == 1
def test_concave(self, msp: Modelspace):
polygon = select.Polygon(
[(-10, -10), (-10, -20), (20, -20), (20, 20), (10, 20), (10, -10)]
)
# ...PP
# .O.PP
# ...PP
# PPPPP
# PPPPP
selection = select.bbox_inside(polygon, msp.query("CIRCLE"))
assert len(selection) == 0
selection = select.bbox_outside(polygon, msp.query("CIRCLE"))
assert len(selection) == 1
selection = select.bbox_overlap(polygon, msp.query("CIRCLE"))
assert len(selection) == 0
class TestBoundingBoxIntersectsFence:
@pytest.mark.parametrize("vertices", [[], [(0, 0)]])
def test_invalid_vertex_count(self, vertices, msp: Modelspace):
with pytest.raises(ValueError):
select.bbox_crosses_fence(vertices, msp)
def test_overlap_all_except_point(self, msp: Modelspace):
selection = select.bbox_crosses_fence([(-5, 0), (5, 0)], msp)
assert len(selection) == 3
def test_can_not_select_point(self, msp: Modelspace):
"""Fence cannot select POINT entities by definition.
point: (0, 1)
"""
selection = select.bbox_crosses_fence([(-5, 1), (5, 1)], msp.query("POINT"))
assert len(selection) == 0
def test_select_nothing_inside_a_closed_fence(self, msp: Modelspace):
"""A closed fence does NOT work like a polygon selection!"""
selection = select.bbox_crosses_fence(
[(-9, -9), (-9, 9), (9, 9), (-9, 9), (-9, -9)], msp
)
assert len(selection) == 0
def test_select_by_touching_bbox_corner(self, msp: Modelspace):
"""CIRCLE is selected by fence touching the CIRCLE bbox at one corner point.
Does not cross the curve itself!
circle: (0, 0), radius = 5
"""
selection = select.bbox_crosses_fence([(0, 10), (10, 0)], msp)
assert len(selection) == 1
def test_point_selects_all(msp: Modelspace):
"""The point is inside of all entity bounding boxes."""
selection = select.point_in_bbox((0, 1), msp)
assert len(selection) == 4
def test_all_entities_chained_by_bounding_boxes():
doc = ezdxf.new()
msp = doc.modelspace()
for x in range(10):
msp.add_line((x, 0), (x + 1.01, 1))
start = msp[-1]
selected = select.bbox_chained(start, msp)
assert len(selected) == 10
class TestPlanarSearchIndex:
def test_circle_select_radius_1(self, msp: Modelspace):
psi = select.PlanarSearchIndex(msp)
result = psi.detection_point_in_circle((0, 0), 1)
assert len(result) == 1
assert result[0].dxftype() == "POINT"
# CIRCLE is not selected because, none of the bounding box vertices of the
# CIRCLE is inside the section circle
def test_circle_select_radius_2(self, msp: Modelspace):
psi = select.PlanarSearchIndex(msp)
result = psi.detection_point_in_circle((0, 0), 2)
assert len(result) == 2
types = {e.dxftype() for e in result}
assert "POINT" in types
assert "LINE" in types
def test_circle_select_all(self, msp: Modelspace):
psi = select.PlanarSearchIndex(msp)
result = psi.detection_point_in_circle((0, 0), 8)
assert len(result) == 4
def test_rect_select(self, msp: Modelspace):
psi = select.PlanarSearchIndex(msp)
result = psi.detection_point_in_rect((-1, -1), (1, 1))
assert len(result) == 2
types = {e.dxftype() for e in result}
assert "POINT" in types
assert "LINE" in types
def test_rect_select_all(self, msp: Modelspace):
psi = select.PlanarSearchIndex(msp)
result = psi.detection_point_in_rect((-5, -5), (5, 5))
assert len(result) == 4
if __name__ == "__main__":
pytest.main([__file__])
|
