6.8301-Project/configs/gen_corners.py
2024-04-12 19:09:21 -04:00

37 lines
991 B
Python

import json
import sys # hacky way to import utils for now
sys.path.append("..")
from utils import rgb_to_hex
corner_shape = [ # giving all 4 corners have the same shape
[0, 0, 0, 0],
[0, 1, 1, 0],
[0, 1, 1, 0],
[0, 0, 0, 0],
]
center_color = [0, 0, 0]
corner_colors = { # assuming each corner will only have two colors (center_color and a corner-specific color)
0: [255, 0, 0],
1: [0, 255, 0],
2: [0, 0, 255],
3: [255, 255, 255,]
}
center_color = rgb_to_hex(center_color)
corner_colors = {key: rgb_to_hex(val) for key, val in corner_colors.items()}
corner_colored_shapes = {
corner_ind: [[corner_color if cell else center_color for cell in row] for row in corner_shape]
for corner_ind, corner_color in corner_colors.items()
}
with open("corners_hollow4x4_v0.json", "w") as f:
json.dump({
"corner_colors": corner_colored_shapes,
"corner_width": len(corner_shape[0]),
"corner_height": len(corner_shape),
}, f)