kuva-reader 1.1.1__tar.gz → 1.1.3__tar.gz

kuva_reader-1.1.3/.gitignore

@@ -0,0 +1,134 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+*.ipynb
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Torch stuff
+lightning_logs/
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# VSCode
+.vscode
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+
+# Direnv stuff
+.direnv/
+.envrc
+
+# Poetry
+# poetry.lock
+
+# Supervisord
+supervisord.log
+supervisord.pid
+
+# Debug folders
+_debug/
+
+# Kuva data
+*.tif
+*.tiff
+*.npy
+hyperfield*.json
+
+# Do not ignore Kuva files in the test_data directory
+!kuva-reader/tests/test_data/**

{kuva_reader-1.1.1 → kuva_reader-1.1.3}/PKG-INFO

@@ -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.1 → kuva_reader-1.1.3}/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-1.1.1 → kuva_reader-1.1.3}/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-1.1.1 → kuva_reader-1.1.3}/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-1.1.1 → kuva_reader-1.1.3}/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-1.1.1 → kuva_reader-1.1.3}/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)