From 90ea0c2195a0900f6707d857f1d36e2f494c2201 Mon Sep 17 00:00:00 2001 From: Jan Petykiewicz Date: Fri, 30 Jul 2021 23:24:58 -0700 Subject: [PATCH] fix read_point_list when using numpy --- fatamorgana/basic.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/fatamorgana/basic.py b/fatamorgana/basic.py index 960346c..6194ab0 100644 --- a/fatamorgana/basic.py +++ b/fatamorgana/basic.py @@ -1568,17 +1568,23 @@ def read_point_list(stream: io.BufferedIOBase, dy = sum(y for _x, y in points) if list_type == 0: - points += [[-dx, 0], [0, -dy]] + close_points = [[-dx, 0], [0, -dy]] elif list_type == 1: - points += [[0, -dy], [-dx, 0]] + close_points = [[0, -dy], [-dx, 0]] elif list_type == 2: assert (dx == 0) or (dy == 0) - points.append([-dx, -dy]) + close_points = [[-dx, -dy]] elif list_type == 3: assert (dx == 0) or (dy == 0) or (dx == dy) or (dx == -dy) - points.append([-dx, -dy]) + close_points = [[-dx, -dy]] else: - points.append([-dx, -dy]) + close_points = [[-dx, -dy]] + + if _USE_NUMPY: + points = numpy.vstack((points, close_points)) + else: + points.append(close_points) + return points