raytrace-box-visualization/ray-trace-box-visualization.py at master · CanNotCode2/raytrace-box-visualization · GitHub

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
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.art3d import Poly3DCollection
import numpy as np

def parse_vector(vector_str):
    origin_str, direction_str = vector_str.split('direction:')
    origin_str = origin_str.replace('origin:', '').strip()
    direction_str = direction_str.strip()

    origin = np.array([float(x) for x in origin_str.split(',')])
    direction = np.array([float(x) for x in direction_str.split(',')])

    return origin, direction

def parse_box(box_str):
    box_str = box_str.replace('SimpleCollisionBox', '').strip('{}')
    components = [comp.split('=')[1] for comp in box_str.split(',')]
    min_x, min_y, min_z, max_x, max_y, max_z = map(float, components[:6])

    return min_x, min_y, min_z, max_x, max_y, max_z

def draw_vector(ax, origin, direction, length=10, **kwargs):
    ax.quiver(
        origin[0], origin[2], origin[1],  # Swap y and z
        direction[0], direction[2], direction[1],  # Swap y and z
        length=length, **kwargs
    )

def draw_box(ax, min_x, min_y, min_z, max_x, max_y, max_z, **kwargs):
    r = [min_x, max_x]
    X, Y = np.meshgrid(r, r)
    ax.scatter([min_x, max_x], [min_z, max_z], [min_y, max_y], c='r')  # Swap y and z

    # Draw faces
    ax.add_collection3d(Poly3DCollection([
        list(zip([min_x, min_x, max_x, max_x], [min_z, min_z, max_z, max_z], [min_y, max_y, max_y, min_y])),  # Swap y and z
        list(zip([min_x, min_x, max_x, max_x], [min_z, min_z, max_z, max_z], [max_y, max_y, max_y, max_y])),  # Swap y and z
        list(zip([min_x, min_x, min_x, min_x], [min_z, max_z, max_z, min_z], [min_y, min_y, max_y, max_y])),  # Swap y and z
        list(zip([max_x, max_x, max_x, max_x], [min_z, max_z, max_z, min_z], [min_y, min_y, max_y, max_y])),  # Swap y and z
        list(zip([min_x, max_x, max_x, min_x], [min_z, min_z, min_z, min_z], [min_y, max_y, max_y, min_y])),  # Swap y and z
        list(zip([min_x, max_x, max_x, min_x], [max_z, max_z, max_z, max_z], [min_y, max_y, max_y, min_y]))  # Swap y and z
    ], **kwargs))

def main():
    vector_str = "origin: 234.94632132425156,6.27,292.58369401837393 direction: -0.10548005814118344,-0.07040894031524658,0.9919256235540885"
    box_str = "SimpleCollisionBox{minX=233.9998, minY=5.9998, minZ=294.9998, maxX=235.0002, maxY=7.0002, maxZ=296.0002, isFullBlock=true}"

    origin, direction = parse_vector(vector_str)
    min_x, min_y, min_z, max_x, max_y, max_z = parse_box(box_str)

    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')

    draw_vector(ax, origin, direction, color='blue')
    draw_box(ax, min_x, min_y, min_z, max_x, max_y, max_z, alpha=0.3, color='red')

    ax.set_xlabel('X')
    ax.set_ylabel('Z')  # Change label from 'Y' to 'Z'
    ax.set_zlabel('Y')  # Change label from 'Z' to 'Y'

    # Determine the center and maximum range across all axes
    x_mid = 0.5 * (min_x + max_x)
    y_mid = 0.5 * (min_y + max_y)
    z_mid = 0.5 * (min_z + max_z)
    max_range = max(max_x - min_x, max_y - min_y, max_z - min_z)

    # Ensure each axis has at least 30 units of range
    min_range = 30
    max_range = max(max_range, min_range)

    ax.set_xlim([x_mid - max_range / 2, x_mid + max_range / 2])
    ax.set_ylim([z_mid - max_range / 2, z_mid + max_range / 2])  # Use z limits for y
    ax.set_zlim([y_mid - max_range / 2, y_mid + max_range / 2])  # Use y limits for z

    plt.show()

if __name__ == "__main__":
    main()