Make array_view.size return 0 for empty arrays

jkseppan · jkseppan · commit cae71f6a13f1 · 2015-10-15T15:28:32.000+03:00
diff --git a/src/numpy_cpp.h b/src/numpy_cpp.h
@@ -539,7 +539,17 @@ class array_view : public detail::array_view_accessors<array_view, T, ND>
 
     size_t size() const
     {
-        return (size_t)dim(0);
+        bool empty = (ND == 0);
+        for (size_t i = 0; i < ND; i++) {
+            if (m_shape[i] == 0) {
+                empty = true;
+            }
+        }
+        if (empty) {
+            return 0;
+        } else {
+            return (size_t)dim(0);
+        }
     }
 
     bool empty() const
