def get_lines()

in threedod/benchmark_scripts/utils/visual_utils.py [0:0]

• 22 lines of code
• 2 McCabe index (conditional complexity)

def get_lines(box, color=np.array([1.0, 0.0, 0.0])):
    """
    Args:
        box: np.array (8, 3)
            8 corners
        color: line color
    Returns:
        o3d.Linset()
    """
    points = box
    lines = [
        [0, 1],
        [0, 3],
        [1, 2],
        [2, 3],
        [4, 5],
        [4, 7],
        [5, 6],
        [6, 7],
        [0, 4],
        [1, 5],
        [2, 6],
        [3, 7],
    ]
    colors = [color for i in range(len(lines))]
    line_set = o3d.geometry.LineSet()
    line_set.points = o3d.utility.Vector3dVector(points)
    line_set.lines = o3d.utility.Vector2iVector(lines)
    line_set.colors = o3d.utility.Vector3dVector(colors)
    return line_set