Spaces:
Running
Running
File size: 2,057 Bytes
5f028d6 |
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 |
import numpy as np
def checkerboard_geometry(
length=12.0,
color0=[172/255, 172/255, 172/255],
color1=[215/255, 215/255, 215/255],
tile_width=0.5,
alpha=1.0,
up="y",
c1=0.0,
c2=0.0,
):
assert up == "y" or up == "z"
color0 = np.array(color0 + [alpha])
color1 = np.array(color1 + [alpha])
radius = length / 2.0
num_rows = num_cols = max(2, int(length / tile_width))
vertices = []
vert_colors = []
faces = []
face_colors = []
for i in range(num_rows):
for j in range(num_cols):
u0, v0 = j * tile_width - radius, i * tile_width - radius
us = np.array([u0, u0, u0 + tile_width, u0 + tile_width])
vs = np.array([v0, v0 + tile_width, v0 + tile_width, v0])
zs = np.zeros(4)
if up == "y":
cur_verts = np.stack([us, zs, vs], axis=-1) # (4, 3)
cur_verts[:, 0] += c1
cur_verts[:, 2] += c2
else:
cur_verts = np.stack([us, vs, zs], axis=-1) # (4, 3)
cur_verts[:, 0] += c1
cur_verts[:, 1] += c2
cur_faces = np.array(
[[0, 1, 3], [1, 2, 3], [0, 3, 1], [1, 3, 2]], dtype=np.int64
)
cur_faces += 4 * (i * num_cols + j) # the number of previously added verts
use_color0 = (i % 2 == 0 and j % 2 == 0) or (i % 2 == 1 and j % 2 == 1)
cur_color = color0 if use_color0 else color1
cur_colors = np.array([cur_color, cur_color, cur_color, cur_color])
vertices.append(cur_verts)
faces.append(cur_faces)
vert_colors.append(cur_colors)
face_colors.append(cur_colors)
vertices = np.concatenate(vertices, axis=0).astype(np.float32)
vert_colors = np.concatenate(vert_colors, axis=0).astype(np.float32)
faces = np.concatenate(faces, axis=0).astype(np.float32)
face_colors = np.concatenate(face_colors, axis=0).astype(np.float32)
return vertices, faces, vert_colors, face_colors |