Rasterio source

.gitignore

@@ -41,6 +41,13 @@ coverage.xml
 # pyenv
 .python-version
 
+# IPython
+profile_default/
+ipython_config.py
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
 # dotenv
 .env
 

large_image/tilesource/base.py

@@ -22,7 +22,7 @@
 from .utilities import (_encodeImage, _encodeImageBinary,  # noqa: F401
                         _gdalParameters, _imageToNumpy, _imageToPIL,
                         _letterboxImage, _makeSameChannelDepth, _vipsCast,
-                        _vipsParameters, dictToEtree, etreeToDict,
+                        _vipsParameters, dictToEtree, etreeToDict, _rasterioParameters,
                         getPaletteColors, histogramThreshold, nearPowerOfTwo)
 
 

large_image/tilesource/utilities.py

@@ -275,6 +275,43 @@ def _vipsCast(image, mustBe8Bit=False, originalScale=None):
     image = ((image.cast(pyvips.BandFormat.DOUBLE) + offset) * multiplier).cast(target)
     return image
 
+def _rasterioParameters(defaultCompression=None, eightbit=None, **kwargs):
+    """
+    return a dictionnary of creation option for the rasterio driver
+
+    :param defaultCompression: if not specified, use this value.
+    :param eightbit: True or False to indicate that the bit depth per sample is known.  None for unknown.
+
+    Optional parameters that can be specified in kwargs:
+
+    :param tileSize: the horizontal and vertical tile size.
+    :param compression: one of 'jpeg', 'deflate' (zip), 'lzw', 'packbits', 'zstd', or 'none'.
+    :param quality: a jpeg quality passed to gdal.  0 is small, 100 is high quality.  90 or above is recommended.
+    :param level: compression level for zstd, 1-22 (default is 10).
+    :param predictor: one of 'none', 'horizontal', or 'float' used for lzw and deflate.
+
+    :returns: a dictionary of parameters.
+    """
+
+    # some default option and parameters
+    options = {"blocksize":256, "compress": "lzw", "quality": 90}
+
+    # the name of the predictor need to be strings so we convert here from set values to actual
+    # required values (https://rasterio.readthedocs.io/en/latest/topics/image_options.html)
+    predictor = {"none": "NO", "horizontal": "STANDARD", "float": "FLOATING_POINT", "yes": "YES"}
+
+    if eightbit is not None:
+        options["predictor"] = "yes" if eightbit else "none"
+
+    # add the values from kwargs to the options. Remove anything that isnot set.
+    options.update({k: v for k, v in kwargs.items() if v not in (None, "")})
+
+    # add the remaining options
+    options.update(bigtiff="IF_SAFER")
+    "predictor" not in options or options.update(predictor=predictor[options["predictor"]])
+
+    return options
+
 
 def _gdalParameters(defaultCompression=None, eightbit=None, **kwargs):
     """
@@ -319,7 +356,53 @@ def _gdalParameters(defaultCompression=None, eightbit=None, **kwargs):
     if 'level' in options:
         cmdopt += ['-co', 'LEVEL=%s' % options['level']]
     return cmdopt
+
+def _rasterioParameters(defaultCompression=None, eightbit=None, **kwargs):
+    """
+    return a dictionnary of creation option for the rasterio driver
+
+    :param defaultCompression: if not specified, use this value.
+    :param eightbit: True or False to indicate that the bit depth per sample is
+        known.  None for unknown.
+    Optional parameters that can be specified in kwargs:
+    :param tileSize: the horizontal and vertical tile size.
+    :param compression: one of 'jpeg', 'deflate' (zip), 'lzw', 'packbits',
+        'zstd', or 'none'.
+    :param quality: a jpeg quality passed to gdal.  0 is small, 100 is high
+        quality.  90 or above is recommended.
+    :param level: compression level for zstd, 1-22 (default is 10).
+    :param predictor: one of 'none', 'horizontal', or 'float' used for lzw and
+        deflate.
+    :returns: a dictionary of parameters.
+    """
+
+    # some default option and parameters
+    options = {
+        "blocksize":256,
+        "compress": "lzw",
+        "quality": 90,
+    }
+
+    # the name of the predictor need to be strings so we convert here from set values to actual
+    # required values (https://rasterio.readthedocs.io/en/latest/topics/image_options.html)
+    predictor = {
+        'none': 'NO',
+        'horizontal': 'STANDARD',
+        'float': 'FLOATING_POINT',
+        'yes': 'YES',
+    }
+
+    if eightbit is not None:
+        options['predictor'] = 'yes' if eightbit else 'none'
+
+    # add the values from kwargs to the options. Remove anything that isnot set.
+    options.update({k: v for k, v in kwargs.items() if v not in (None, '')})
+
+    # add the remaining options
+    options.update(bigtiff="IF_SAFER")
+    "predictor" not in options or options.update(predictor=predictor[options["predictor"]])
 
+    return options
 
 def _vipsParameters(forTiled=True, defaultCompression=None, **kwargs):
     """

setup.py

@@ -58,6 +58,7 @@ def prerelease_local_scheme(version):
     'tiff': [f'large-image-source-tiff{limit_version}'],
     'tifffile': [f'large-image-source-tifffile{limit_version}'],
     'vips': [f'large-image-source-vips{limit_version}'],
+    'rasterio': [f'large-image-source-rasterio{limit_version}'],
 }
 if sys.version_info >= (3, 7):
     sources.update({

sources/gdal/large_image_source_gdal/__init__.py

@@ -483,7 +483,7 @@ def getBounds(self, srs=None):
         """
         Returns bounds of the image.
 
-        :param srs: the projection for the bounds.  None for the default.
+        :param srs: the projection for the bounds.  None for the default 4326.
         :returns: an object with the four corners and the projection that was
             used.  None if we don't know the original projection.
         """