jan/gridlock
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix handling of 3d polys

Browse Source
This commit is contained in:
Jan Petykiewicz 2024-07-29 01:47:12 -07:00
parent c32d94ed85
commit e256f56f2b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
gridlock/draw.py
View File

@ -60,12 +60,14 @@ class GridDrawMixin(GridPosMixin):
# Check polygons, and remove redundant coordinates # Check polygons, and remove redundant coordinates
surface = numpy.delete(range(3), surface_normal) surface = numpy.delete(range(3), surface_normal)
for i, polygon in enumerate(poly_list): for ii in range(len(poly_list)):
malformed = f'Malformed polygon: ({i})' polygon = poly_list[ii]
malformed = f'Malformed polygon: ({ii})'
if polygon.shape[1] not in (2, 3): if polygon.shape[1] not in (2, 3):
raise GridError(malformed + 'must be a Nx2 or Nx3 ndarray') raise GridError(malformed + 'must be a Nx2 or Nx3 ndarray')
if polygon.shape[1] == 3: if polygon.shape[1] == 3:
polygon = polygon[surface, :] polygon = polygon[surface, :]
poly_list[ii] = polygon
if not polygon.shape[0] > 2: if not polygon.shape[0] > 2:
raise GridError(malformed + 'must consist of more than 2 points') raise GridError(malformed + 'must consist of more than 2 points')