-
Notifications
You must be signed in to change notification settings - Fork 0
/
scene.py
40 lines (31 loc) · 951 Bytes
/
scene.py
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
import json
import numpy as np
from typing import List, TypedDict
class Location(TypedDict):
x: int
y: int
z: int
class Scene_JSON_Data(TypedDict):
boundaries: List[Location]
class Scene:
def __init__(self, json_data):
self.json_data: Scene_JSON_Data = json_data
def get_boundaries_numpy(self):
return np.array(
[
np.array(
[
np.int32(location["x"]),
np.int32(location["y"]),
np.int32(location["z"]),
]
)
for location in self.json_data["boundaries"]
],
dtype=np.int32,
)
def load_scene(scene_path: str) -> Scene:
# Remove .json extension if present
scene_path = "./scenes/" + scene_path.replace(".json", "") + ".json"
json_data = json.load(open(scene_path))
return Scene(json_data)