fixed point transformation

Author: timtreis

src/spatialdata_plot/pl/render.py

@@ -220,7 +220,6 @@ def _render_points(
             coords.extend(color)
 
         points = points[coords].compute()
-        # points[color[0]].cat.set_categories(render_params.groups, inplace=True)
         if render_params.groups is not None:
             points = points[points[color].isin(render_params.groups).values]
             points[color[0]] = points[color[0]].cat.set_categories(render_params.groups)
@@ -260,6 +259,10 @@ def _render_points(
         if color_source_vector is None and render_params.transfunc is not None:
             color_vector = render_params.transfunc(color_vector)
 
+        trans = get_transformation(sdata.points[e], get_all=True)[coordinate_system]
+        affine_trans = trans.to_affine_matrix(input_axes=("x", "y"), output_axes=("x", "y"))
+        trans = mtransforms.Affine2D(matrix=affine_trans) + ax.transData
+
         norm = copy(render_params.cmap_params.norm)
         _cax = ax.scatter(
             adata[:, 0].X.flatten(),
@@ -270,17 +273,11 @@ def _render_points(
             cmap=render_params.cmap_params.cmap,
             norm=norm,
             alpha=render_params.alpha,
+            transform=trans
             # **kwargs,
         )
         cax = ax.add_collection(_cax)
 
-        trans = get_transformation(sdata.points[e], get_all=True)[coordinate_system]
-        affine_trans = trans.to_affine_matrix(input_axes=("x", "y"), output_axes=("x", "y"))
-        trans = mtransforms.Affine2D(matrix=affine_trans)
-
-        for path in _cax.get_paths():
-            path.vertices = trans.transform(path.vertices)
-
         if not (
             len(set(color_vector)) == 1 and list(set(color_vector))[0] == to_hex(render_params.cmap_params.na_color)
         ):

tests/_images/Extent_correct_plot_after_transformations.png

@@ -111,7 +111,7 @@ def test_plot_correct_plot_after_transformations(self):
             for element_name in [f"circles_pi{i}", f"polygons_pi{i}", f"multipolygons_pi{i}", f"points_pi{i}"]:
                 set_transformation(element=sdata[element_name], transformation=rotation, to_coordinate_system=f"pi{i}")
 
-        fig, axs = plt.subplots(ncols=3, nrows=4, figsize=(7, 9))
+        _, axs = plt.subplots(ncols=3, nrows=4, figsize=(7, 9))
 
         for cs_idx, cs in enumerate(["global", "pi3", "pi4"]):
             if cs == "global":
@@ -136,7 +136,7 @@ def test_plot_correct_plot_after_transformations(self):
                 coordinate_systems=cs, ax=axs[2, cs_idx], title=""
             )
             sdata.pl.render_points(elements=points_name, size=10).pl.show(
-                coordinate_systems=cs, ax=axs[3, cs_idx], title="", pad_extent=0.05
+                coordinate_systems=cs, ax=axs[3, cs_idx], title="", pad_extent=0.02
             )
 
         plt.tight_layout()