hongchen/Floorplan/test_overlap.py

import sys
sys.path.insert(0, '/media/dage/KESU/temp/floorplan')
from pixel_to_world import segment_bbox_intersections

# 测试用例：seg[28] 和 seg[29] 与门 bbox 的重叠
# 门 bbox: [-3.41, -2.0, -3.17, -0.7]
# seg[28]: (-3.33,-0.18)->(-3.33,-1.40) 垂直线段
# seg[29]: (-3.33,-1.08)->(-3.33,-2.02) 垂直线段

bbox = [-3.41, -2.0, -3.17, -0.7]

# seg[28]
va = (-3.33, -0.18)
vb = (-3.33, -1.40)
dx, dy = vb[0] - va[0], vb[1] - va[1]
seg_len = (dx*dx + dy*dy) ** 0.5
print(f'seg[28]: {va}->{vb}, len={seg_len:.3f}')

pts = segment_bbox_intersections([va[0], va[1], vb[0], vb[1]], bbox)
print(f'  交点数：{len(pts)}')
if len(pts) >= 2:
    dx0 = pts[0][0]-pts[1][0]
    dy0 = pts[0][1]-pts[1][1]
    overlap = (dx0*dx0 + dy0*dy0) ** 0.5
    print(f'  重叠长度：{overlap:.3f}')
elif len(pts) == 1:
    print(f'  只有一个交点，需要特殊处理')
else:
    print(f'  无交点，可能线段完全在 bbox 内或完全在外')

# seg[29]
va = (-3.33, -1.08)
vb = (-3.33, -2.02)
dx, dy = vb[0] - va[0], vb[1] - va[1]
seg_len = (dx*dx + dy*dy) ** 0.5
print(f'seg[29]: {va}->{vb}, len={seg_len:.3f}')

pts = segment_bbox_intersections([va[0], va[1], vb[0], vb[1]], bbox)
print(f'  交点数：{len(pts)}')
if len(pts) >= 2:
    dx0 = pts[0][0]-pts[1][0]
    dy0 = pts[0][1]-pts[1][1]
    overlap = (dx0*dx0 + dy0*dy0) ** 0.5
    print(f'  重叠长度：{overlap:.3f}')
elif len(pts) == 1:
    print(f'  只有一个交点，需要特殊处理')
else:
    print(f'  无交点，可能线段完全在 bbox 内或完全在外')