Simplify affine_transform on C++ side

Author: QuLogic

lib/matplotlib/tests/test_transforms.py

@@ -287,20 +287,17 @@ def test_translate_plus_other(self):
 
     def test_invalid_arguments(self):
         t = mtransforms.Affine2D()
-        # The wrong number of dimensions and a wrong shape with a possible number of
-        # dimensions are both caught by pybind11, which always raises the less precise
-        # RuntimeError.
-        with pytest.raises(RuntimeError):
+        with pytest.raises(TypeError):
             t.transform(1)
-        with pytest.raises(RuntimeError):
+        with pytest.raises(TypeError):
             t.transform([[[1]]])
-        with pytest.raises(RuntimeError):
+        with pytest.raises(TypeError):
             t.transform([])
-        with pytest.raises(RuntimeError):
+        with pytest.raises(TypeError):
             t.transform([1])
-        with pytest.raises(RuntimeError):
+        with pytest.raises(TypeError):
             t.transform([[1]])
-        with pytest.raises(RuntimeError):
+        with pytest.raises(TypeError):
             t.transform([[1, 2, 3]])
 
     def test_copy(self):

lib/matplotlib/transforms.py

@@ -1880,9 +1880,14 @@ def to_values(self):
     def transform_affine(self, values):
         mtx = self._get_eigen_matrix()
         if isinstance(values, np.ma.MaskedArray):
-            tpoints = mtx.affine_transform(values.data)
-            return np.ma.MaskedArray(tpoints, mask=np.ma.getmask(values))
-        return mtx.affine_transform(values)
+            tpoints = mtx.affine_transform(values.data.T).T
+            result = np.ma.MaskedArray(tpoints, mask=np.ma.getmask(values))
+        else:
+            values = np.asarray(values)
+            result = mtx.affine_transform(values.T).T
+        if len(values.shape) == 2:
+            result = np.atleast_2d(result)
+        return result
 
     if DEBUG:
         _transform_affine = transform_affine

src/_eigen.cpp

@@ -126,10 +126,8 @@ PYBIND11_MODULE(_eigen, m) {
             }
         )
         .def("affine_transform",
-            [](const Eigen::Affine2d& self, py::array_t<double> vertices_arr) {
-                auto vertices = vertices_arr.attr("transpose")().cast<Eigen::Ref<const Eigen::Matrix2Xd>>();
-                auto result = py::cast(self * vertices, py::return_value_policy::move);
-                return result.attr("transpose")();
+            [](const Eigen::Affine2d& self, Eigen::Ref<const Eigen::Matrix2Xd> vertices) {
+                return self * vertices;
             }
         )