handle 2d centering

jigli · Jun 20, 2021 · f0657f9 · f0657f9
1 parent b5d4d9b
commit f0657f9
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/include/igl/opengl/glfw/Viewer.cpp b/include/igl/opengl/glfw/Viewer.cpp
@@ -667,13 +667,11 @@ namespace glfw
 
 
     // Initialization code for the trackball
-    Eigen::RowVector3d center;
-    if (data().V.rows() == 0)
+    Eigen::RowVector3d center = Eigen::RowVector3d(0,0,0);
+    if(data().V.rows() > 0)
     {
-      center << 0,0,0;
-    }else
-    {
-      center = data().V.colwise().sum()/data().V.rows();
+      // be careful that V may be 2D
+      center.head(data().V.cols()) = data().V.colwise().sum()/data().V.rows();
     }
 
     Eigen::Vector3f coord =