yirgacheffe 1.9.1__py3-none-any.whl → 1.9.2__py3-none-any.whl

yirgacheffe/_operators.py

@@ -339,6 +339,55 @@ class LayerMathMixin:
             round_down_pixels((y - area.top) / pixel_scale.ystep, builtins.abs(pixel_scale.ystep)),
         )
 
+    @property
+    def window(self) -> Window:
+        raise NotImplementedError("Must be overridden by subclass")
+
+    @property
+    def area(self) -> Area:
+        raise NotImplementedError("Must be overridden by subclass")
+
+    def read_array(self, _x, _y, _w, _h):
+        raise NotImplementedError("Must be overridden by subclass")
+
+    def show(self, ax=None, max_pixels: int | None =1000, **kwargs):
+        """Display data using matplotlib.
+
+        Args:
+            ax: Matplotlib axes object. If not provided, the default matplotlib context will be used.
+            max_pixels: How many pixels to downsample to. If None, raw pixels will be used.
+            **kwargs: Passed to matplotlib imshow.
+
+        Returns:
+            A matplotlib image.
+        """
+        import matplotlib.pyplot as plt # pylint: disable=C0415
+
+        if ax is None:
+            ax = plt.gca()
+
+        window = self.window
+
+        raw_data = self.read_array(
+            0,
+            0,
+            window.xsize,
+            window.ysize
+        )
+        if max_pixels:
+            downsample = max(window.xsize, window.ysize) // max_pixels
+            data = raw_data[::downsample,::downsample]
+        else:
+            data = raw_data
+        area = self.area
+        extent = [
+            area.left,
+            area.right,
+            area.bottom,
+            area.top,
+        ]
+        return ax.imshow(data, extent=extent, **kwargs)
+
 
 class LayerOperation(LayerMathMixin):
 
@@ -383,6 +432,7 @@ class LayerOperation(LayerMathMixin):
         self.kwargs = kwargs
         self.window_op = window_op
         self.buffer_padding = buffer_padding
+        self._forced_area = None
 
         if lhs is None:
             raise ValueError("LHS on operation should not be none")
@@ -450,6 +500,9 @@ class LayerOperation(LayerMathMixin):
         return self._get_operation_area(self.map_projection)
 
     def _get_operation_area(self, projection: MapProjection | None) -> Area:
+        if self._forced_area is not None:
+            return self._forced_area
+
         lhs_area = self.lhs._get_operation_area(projection)
         try:
             rhs_area = self.rhs._get_operation_area(projection)

{yirgacheffe-1.9.1.dist-info → yirgacheffe-1.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yirgacheffe
-Version: 1.9.1
+Version: 1.9.2
 Summary: Abstraction of gdal datasets for doing basic math operations
 Author-email: Michael Dales <mwd24@cam.ac.uk>
 License-Expression: ISC
@@ -30,6 +30,8 @@ Requires-Dist: h3
 Requires-Dist: pyproj
 Provides-Extra: mlx
 Requires-Dist: mlx; extra == "mlx"
+Provides-Extra: matplotlib
+Requires-Dist: matplotlib; extra == "matplotlib"
 Provides-Extra: dev
 Requires-Dist: mypy; extra == "dev"
 Requires-Dist: pylint; extra == "dev"

{yirgacheffe-1.9.1.dist-info → yirgacheffe-1.9.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 yirgacheffe/__init__.py,sha256=OOzfXtafPoDpAsNRC08BXjmwv0hBp-mNFCjwplGs9lY,668
 yirgacheffe/_core.py,sha256=AU6tlqovBV_l1dNZs6AlHSw59Z0U6pStUaQZvJGiLhM,5721
-yirgacheffe/_operators.py,sha256=7ReNOLW9MUGN5wI2wbEnAhyaeO5ZQGTnwTib72HB8y8,39656
+yirgacheffe/_operators.py,sha256=DjwWEGStNyliNeXNySe7SUjuouxsEUtZnpMKeZp0iJg,41208
 yirgacheffe/constants.py,sha256=bKUjOGNj19zwggV79lJgK7tiv51DH2-rgNOKswl2gvQ,293
 yirgacheffe/operators.py,sha256=nw-BpnAwTjCwFtjosa8wKd2MGUuC0PJR5jACFdLhqCg,412
 yirgacheffe/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -19,9 +19,9 @@ yirgacheffe/layers/h3layer.py,sha256=Rq1bFo7CApIh5NdBcV7hSj3hm-DszY79nhYsTRAvJ_g
 yirgacheffe/layers/rasters.py,sha256=zBE9uXm6LvAQF2_XdQzcOgJQOQWGmuPflY5JNDrUf3k,13527
 yirgacheffe/layers/rescaled.py,sha256=gEFbXeYxX1nVn7eQYmbGww90_yc5ENmgQrD_WxXxpQE,3352
 yirgacheffe/layers/vectors.py,sha256=A27kuTr0C9BZhHG0-cplNEa7aSNcse37Pm9xTjEzv-c,19990
-yirgacheffe-1.9.1.dist-info/licenses/LICENSE,sha256=dNSHwUCJr6axStTKDEdnJtfmDdFqlE3h1NPCveqPfnY,757
-yirgacheffe-1.9.1.dist-info/METADATA,sha256=aRcZJh0nb6tBa7W2GUXCjbtYqmPFDddbJ-nmONuqMHk,24186
-yirgacheffe-1.9.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-yirgacheffe-1.9.1.dist-info/entry_points.txt,sha256=j4KgHXbVGbGyfTySc1ypBdERpfihO4WNjppvCdE9HjE,52
-yirgacheffe-1.9.1.dist-info/top_level.txt,sha256=9DBFlKO2Ld3hG6TuE3qOTd3Tt8ugTiXil4AN4Wr9_y0,12
-yirgacheffe-1.9.1.dist-info/RECORD,,
+yirgacheffe-1.9.2.dist-info/licenses/LICENSE,sha256=dNSHwUCJr6axStTKDEdnJtfmDdFqlE3h1NPCveqPfnY,757
+yirgacheffe-1.9.2.dist-info/METADATA,sha256=dtcXyXfgUIXRNPseIm3AT7KxJQLzlflciYzic9t4zX0,24262
+yirgacheffe-1.9.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+yirgacheffe-1.9.2.dist-info/entry_points.txt,sha256=j4KgHXbVGbGyfTySc1ypBdERpfihO4WNjppvCdE9HjE,52
+yirgacheffe-1.9.2.dist-info/top_level.txt,sha256=9DBFlKO2Ld3hG6TuE3qOTd3Tt8ugTiXil4AN4Wr9_y0,12
+yirgacheffe-1.9.2.dist-info/RECORD,,