kuva-reader 1.1.1__py3-none-any.whl → 1.1.3__py3-none-any.whl

kuva_reader/__init__.py

@@ -22,7 +22,7 @@ Dependencies
   visualization.
 """
 
-__version__ = "0.1.0"
+__version__ = "1.1.2"
 
 from .reader.image import (
     image_footprint,

kuva_reader/reader/level0.py

@@ -61,11 +61,12 @@ class Level0Product(ProductBase[MetadataLevel0]):
     ) -> None:
         super().__init__(image_path, metadata, target_ureg)
 
-        self.images = {
+        self._images = {
             camera: cast(
                 rio.DatasetReader,
                 rio.open(
                     self.image_path / (cube.camera.name + ".tif"),
+                    num_threads='16',
                 ),
             )
             for camera, cube in self.metadata.image.data_cubes.items()  # type: ignore
@@ -97,6 +98,12 @@ class Level0Product(ProductBase[MetadataLevel0]):
         """Return the datarray for the chosen camera."""
         return self.images[camera]
 
+    @property
+    def images(self) -> dict[str, rio.DatasetReader]:
+        if self._images is None:
+            raise RuntimeError("Images has been released.")
+        return self._images
+
     def keys(self) -> list[str]:
         """Easy access to the camera keys."""
         return list(self.images.keys())
@@ -230,11 +237,14 @@ class Level0Product(ProductBase[MetadataLevel0]):
         """Explicitely closes the Rasterio DatasetReaders and releases the memory of
         the `images` variable.
         """
-        for k in self.images.keys():
-            self.images[k].close()
-
-        del self.images
-        self.images = None
+        if self._images is not None:
+            for k in self._images.keys():
+                self._images[k].close()
+
+            del self._images
+            # We know that images are not None as long as somebody doesn't call
+            # this function beforehand....
+            self._images = None
 
 
 def generate_level_0_metafile():

kuva_reader/reader/level1.py

@@ -50,9 +50,9 @@ class Level1ABProduct(ProductBase[MetadataLevel1AB]):
     ) -> None:
         super().__init__(image_path, metadata, target_ureg)
 
-        self.image = cast(
+        self._image = cast(
             rio.DatasetReader,
-            rio.open(self.image_path / "L1B.tif"),
+            rio.open(self.image_path / "L1B.tif", num_threads='16'),
         )
 
         self.data_tags = self.image.tags()
@@ -73,6 +73,12 @@ class Level1ABProduct(ProductBase[MetadataLevel1AB]):
         else:
             return f"{self.__class__.__name__} loaded from '{self.image_path}'"
 
+    @property
+    def image(self) -> rio.DatasetReader:
+        if self._image is None:
+            raise RuntimeError("Images has been released.")
+        return self._image
+
     def footprint(self, crs="") -> Polygon:
         """The product footprint as a Shapely polygon."""
         return image_footprint(self.image, crs)
@@ -130,8 +136,35 @@ class Level1ABProduct(ProductBase[MetadataLevel1AB]):
         """Explicitely closes the Rasterio DatasetReader and releases the memory of
         the `image` variable.
         """
-        del self.image
-        self.image = None
+        if self._image is not None:
+            self._image.close()
+            del self._image
+            self._image = None
+
+    def generate_metadata_file(self) -> None:
+        """Write the sidecar files next to the product."""
+        metadata_file_name = self.image_path.name + ".json"
+
+        with rio.open(self.image_path / "L1B.tif") as src:
+            shape = (src.height, src.width)
+            crs_epsg = src.crs.to_epsg()
+            geotransform = src.transform
+            gsd_w, gsd_h = src.res
+
+
+        with (self.image_path / metadata_file_name).open("w") as fh:
+            fh.write(
+                self.metadata.model_dump_json(
+                    indent=2,
+                    context={
+                        "shape": shape,
+                        "epsg": crs_epsg,
+                        "transform": geotransform,
+                        "gsd_w": gsd_w,
+                        "gsd_h": gsd_h,
+                    },
+                )
+            )
 
 
 class Level1CProduct(ProductBase[MetadataLevel1C]):
@@ -170,9 +203,9 @@ class Level1CProduct(ProductBase[MetadataLevel1C]):
     ) -> None:
         super().__init__(image_path, metadata, target_ureg)
 
-        self.image = cast(
+        self._image = cast(
             rio.DatasetReader,
-            rio.open(self.image_path / "L1C.tif"),
+            rio.open(self.image_path / "L1C.tif", num_threads='16'),
         )
         self.data_tags = self.image.tags()
 
@@ -193,6 +226,12 @@ class Level1CProduct(ProductBase[MetadataLevel1C]):
         else:
             return f"{self.__class__.__name__} loaded from '{self.image_path}'"
 
+    @property
+    def image(self) -> rio.DatasetReader:
+        if self._image is None:
+            raise RuntimeError("Images has been released.")
+        return self._image
+
     def footprint(self, crs="") -> Polygon:
         """The product footprint as a Shapely polygon."""
         return image_footprint(self.image, crs)
@@ -231,9 +270,35 @@ class Level1CProduct(ProductBase[MetadataLevel1C]):
         """Explicitely closes the Rasterio DatasetReader and releases the memory of
         the `image` variable.
         """
-        self.image.close()
-        del self.image
-        self.image = None
+        if self._image is not None:
+            self._image.close()
+            del self._image
+            self._image = None
+
+    def generate_metadata_file(self) -> None:
+        """Write the sidecar files next to the product."""
+        metadata_file_name = self.image_path.name + ".json"
+
+        with rio.open(self.image_path / "L1C.tif") as src:
+            shape = (src.height, src.width)
+            crs_epsg = src.crs.to_epsg()
+            geotransform = src.transform
+            gsd_w, gsd_h = src.res
+
+
+        with (self.image_path / metadata_file_name).open("w") as fh:
+            fh.write(
+                self.metadata.model_dump_json(
+                    indent=2,
+                    context={
+                        "shape": shape,
+                        "epsg": crs_epsg,
+                        "transform": geotransform,
+                        "gsd_w": gsd_w,
+                        "gsd_h": gsd_h,
+                    },
+                )
+            )
 
 
 def generate_level_1_metafile():

kuva_reader/reader/level2.py

@@ -47,9 +47,9 @@ class Level2AProduct(ProductBase[MetadataLevel2A]):
     ) -> None:
         super().__init__(image_path, metadata, target_ureg)
 
-        self.image = cast(
+        self._image = cast(
             rio.DatasetReader,
-            rio.open(self.image_path / "L2A.tif"),
+            rio.open(self.image_path / "L2A.tif", num_threads='16'),
         )
         self.data_tags = self.image.tags()
 
@@ -70,6 +70,12 @@ class Level2AProduct(ProductBase[MetadataLevel2A]):
         else:
             return f"{self.__class__.__name__} loaded from '{self.image_path}'"
 
+    @property
+    def image(self) -> rio.DatasetReader:
+        if self._image is None:
+            raise RuntimeError("Images has been released.")
+        return self._image
+
     def footprint(self, crs="") -> Polygon:
         """The product footprint as a Shapely polygon."""
         return image_footprint(self.image, crs)
@@ -108,9 +114,35 @@ class Level2AProduct(ProductBase[MetadataLevel2A]):
         """Explicitely closes the Rasterio DatasetReader and releases the memory of
         the `image` variable.
         """
-        self.image.close()
-        del self.image
-        self.image = None
+        if self._image is not None:
+            self._image.close()
+            del self._image
+            self._image = None
+
+    def generate_metadata_file(self) -> None:
+        """Write the sidecar files next to the product."""
+        metadata_file_name = self.image_path.name + ".json"
+
+        with rio.open(self.image_path / "L2A.tif") as src:
+            shape = (src.height, src.width)
+            crs_epsg = src.crs.to_epsg()
+            geotransform = src.transform
+            gsd_w, gsd_h = src.res
+
+
+        with (self.image_path / metadata_file_name).open("w") as fh:
+            fh.write(
+                self.metadata.model_dump_json(
+                    indent=2,
+                    context={
+                        "shape": shape,
+                        "epsg": crs_epsg,
+                        "transform": geotransform,
+                        "gsd_w": gsd_w,
+                        "gsd_h": gsd_h,
+                    },
+                )
+            )
 
 
 def generate_level_2_metafile():

kuva_reader/reader/product_base.py

@@ -35,7 +35,7 @@ class ProductBase(Generic[TMetadata], metaclass=ABCMeta):
     def __init__(
         self,
         image_path: Path,
-        metadata: MetadataBase | None = None,
+        metadata: TMetadata | None = None,
         target_ureg: UnitRegistry | None = None,
     ):
         self.image_path = Path(image_path)

{kuva_reader-1.1.1.dist-info → kuva_reader-1.1.3.dist-info}/METADATA

@@ -1,22 +1,16 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: kuva-reader
-Version: 1.1.1
+Version: 1.1.3
 Summary: Manipulate the Kuva Space image and metadata formats
-License: MIT
-Author: Guillem Ballesteros
-Author-email: guillem@kuvaspace.com
-Requires-Python: >=3.10,<=3.13
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: kuva-geometry (>=1.0.1,<2.0.0)
-Requires-Dist: kuva-metadata (>=1.1.1,<2.0.0)
-Requires-Dist: numpy (>=1.26.4,<2.0.0)
-Requires-Dist: numpy-quaternion (>=2022.4.4,<2023.0.0)
-Requires-Dist: pint (>=0.22,<0.23)
-Requires-Dist: rasterio (>=1.4.1,<2.0.0)
+Author-email: Guillem Ballesteros <guillem@kuvaspace.com>, Lennert Antson <lennert.antson@kuvaspace.com>, Arthur Vandenhoeke <arthur.vandenhoeke@kuvaspace.com>, Olli Eloranta <olli.eloranta@kuvaspace.com>
+License-Expression: MIT
+Requires-Python: <=3.13,>=3.10
+Requires-Dist: kuva-geometry<2.0.0,>=1.0.1
+Requires-Dist: kuva-metadata<2.0.0,>=1.1.1
+Requires-Dist: numpy-quaternion>=2023.4.4
+Requires-Dist: numpy>=1.26.4
+Requires-Dist: pint<1.0.0,>=0.22
+Requires-Dist: rasterio<2,>=1.4.3
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -128,4 +122,3 @@ The `kuva-reader` project software is under the [MIT license](https://github.com
 # Status of unit tests
 
 [![Unit tests for kuva-reader](https://github.com/KuvaSpace/kuva-data-processing/actions/workflows/test-kuva-reader.yml/badge.svg)](https://github.com/KuvaSpace/kuva-data-processing/actions/workflows/test-kuva-reader.yml)
-

kuva_reader-1.1.3.dist-info/RECORD

@@ -0,0 +1,15 @@
+kuva_reader/__init__.py,sha256=7OSXrTPW3RTjctJZqyGTekuHRb_5aQNJaG_c-KFac_s,1389
+kuva_reader/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kuva_reader/reader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kuva_reader/reader/image.py,sha256=rqzs5C7nzRArFfVepydRUECd3u55McTZlKINg76nMX4,785
+kuva_reader/reader/level0.py,sha256=Dvf3I8Vv4UIFMd5DgR8JZx51zhjx4aKbt64TGYqXxk4,9761
+kuva_reader/reader/level1.py,sha256=tTQBqzXK1Wna2s866gjHzsYKwJ0evAhaoKFXk0jHPBo,10584
+kuva_reader/reader/level2.py,sha256=3SiD1lEPNi4mG7yR2omY9YkrPFbv-nxGFViuN2pig6U,5207
+kuva_reader/reader/product_base.py,sha256=OuV69tm53j6terSWb7ViXDYkGeK5mWwcbfwIVuZqVg4,4296
+kuva_reader/reader/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kuva_reader/reader/read.py,sha256=b5hGFbeJ-7sg-iilSJn29Hcqj4e7kiWT2S0BH2JwcDE,1761
+kuva_reader/reader/utils.py,sha256=oZ1G43nm8lCxzfbdqBs0KhKaXWe_uf-78uBXWvmZigs,1566
+kuva_reader-1.1.3.dist-info/METADATA,sha256=pDBBjplO1sTSq-_HOs-WQwjmPyUgb5Z-bX47XmbU-kk,4902
+kuva_reader-1.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kuva_reader-1.1.3.dist-info/entry_points.txt,sha256=KB_U0ziK5k7o8en_7lMtx3Jb1LRHhp39Wpt5MWbZuB0,152
+kuva_reader-1.1.3.dist-info/RECORD,,

{kuva_reader-1.1.1.dist-info → kuva_reader-1.1.3.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any

kuva_reader-1.1.3.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+make-l0-meta = kuva_reader.reader.level0:generate_level_0_metafile
+make-l1-meta = kuva_reader.reader.level1:generate_level_1_metafile

kuva_reader-1.1.1.dist-info/RECORD

@@ -1,15 +0,0 @@
-kuva_reader/__init__.py,sha256=y0ViXqCoRSWvGPXEUFP9TOohcJIS24-WIarkZrDgcKs,1389
-kuva_reader/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kuva_reader/reader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kuva_reader/reader/image.py,sha256=rqzs5C7nzRArFfVepydRUECd3u55McTZlKINg76nMX4,785
-kuva_reader/reader/level0.py,sha256=u61sKRXmx4uljlnqPfuLdtdEGfkw5NdtiR9N6XfUsrA,9352
-kuva_reader/reader/level1.py,sha256=HRvhSasDEzB03qJi6wrQiv-r9kSdWITuWIf-2IbYqX4,8355
-kuva_reader/reader/level2.py,sha256=kGd-6qTZXiNWuYwwLf_-UWS0Oeng8iTpeQFXtAfyPXA,4106
-kuva_reader/reader/product_base.py,sha256=4NpZQk3FvSPsN2kKde8lHJA_0Wwy6DQBMfOh4ItlO00,4299
-kuva_reader/reader/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kuva_reader/reader/read.py,sha256=b5hGFbeJ-7sg-iilSJn29Hcqj4e7kiWT2S0BH2JwcDE,1761
-kuva_reader/reader/utils.py,sha256=oZ1G43nm8lCxzfbdqBs0KhKaXWe_uf-78uBXWvmZigs,1566
-kuva_reader-1.1.1.dist-info/METADATA,sha256=RPbtVfnAl3B3uZ2gKpMJUj40CSAQuLKkqvWQ0vEnCZ0,5041
-kuva_reader-1.1.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-kuva_reader-1.1.1.dist-info/entry_points.txt,sha256=YJysY9EChfOb1W_IEht2oE5Z8Y9jA0J6-kofaPv-NdI,149
-kuva_reader-1.1.1.dist-info/RECORD,,

kuva_reader-1.1.1.dist-info/entry_points.txt

@@ -1,4 +0,0 @@
-[console_scripts]
-make-l0-meta=kuva_reader.reader.level0:generate_level_0_metafile
-make-l1-meta=kuva_reader.reader.level1:generate_level_1_metafile
-