From b99e2e3e6b32cfb284fa744ac42e23c06f525e35 Mon Sep 17 00:00:00 2001 From: Zhongshi Date: Mon, 10 May 2021 10:03:38 -0400 Subject: [PATCH] update_object support for 2d --- meshplot/Viewer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/meshplot/Viewer.py b/meshplot/Viewer.py index ca1d93a..1cd406b 100644 --- a/meshplot/Viewer.py +++ b/meshplot/Viewer.py @@ -376,6 +376,10 @@ def reset(self): def update_object(self, oid=0, vertices=None, colors=None, faces=None): obj = self.__objects[oid] if type(vertices) != type(None): + if vertices.shape[1] == 2: + vertices3 = np.zeros((vertices.shape[0],3)) + vertices3[:,:2] = vertices + vertices = vertices3 if obj["coloring"] == "FaceColors": f = obj["arrays"][1] verts = np.zeros((f.shape[0]*3, 3), dtype="float32")