#!/usr/bin/env pytest ############################################################################### # $Id$ # # Project: GDAL/OGR Test Suite # Purpose: gdal_retile.py testing # Author: Even Rouault # ############################################################################### # Copyright (c) 2010, Even Rouault # # Permission is hereby granted, free of charge, to any person obtaining a # copy of this software and associated documentation files (the "Software"), # to deal in the Software without restriction, including without limitation # the rights to use, copy, modify, merge, publish, distribute, sublicense, # and/or sell copies of the Software, and to permit persons to whom the # Software is furnished to do so, subject to the following conditions: # # The above copyright notice and this permission notice shall be included # in all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL # THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER # DEALINGS IN THE SOFTWARE. ############################################################################### import os import shutil import pytest import test_py_scripts from osgeo import gdal, osr pytestmark = pytest.mark.skipif( test_py_scripts.get_py_script("gdal_retile") is None, reason="gdal_retile not available", ) @pytest.fixture() def script_path(): return test_py_scripts.get_py_script("gdal_retile") ############################################################################### # Test gdal_retile.py def test_gdal_retile_1(script_path): try: os.mkdir("tmp/outretile") except OSError: pass test_py_scripts.run_py_script( script_path, "gdal_retile", "-v -levels 2 -r bilinear -targetDir tmp/outretile " + test_py_scripts.get_data_path("gcore") + "byte.tif", ) ds = gdal.Open("tmp/outretile/byte_1_1.tif") assert ds.GetRasterBand(1).Checksum() == 4672 ds = None ds = gdal.Open("tmp/outretile/1/byte_1_1.tif") assert ds.RasterXSize == 10 # if ds.GetRasterBand(1).Checksum() != 1152: # print(ds.GetRasterBand(1).Checksum()) # return 'fail' ds = None ds = gdal.Open("tmp/outretile/2/byte_1_1.tif") assert ds.RasterXSize == 5 # if ds.GetRasterBand(1).Checksum() != 215: # print(ds.GetRasterBand(1).Checksum()) # return 'fail' ds = None ############################################################################### # Test gdal_retile.py with RGBA dataset def test_gdal_retile_2(script_path): try: os.mkdir("tmp/outretile2") except OSError: pass test_py_scripts.run_py_script( script_path, "gdal_retile", "-v -levels 2 -r bilinear -targetDir tmp/outretile2 " + test_py_scripts.get_data_path("gcore") + "rgba.tif", ) ds = gdal.Open("tmp/outretile2/2/rgba_1_1.tif") assert ds.GetRasterBand(1).Checksum() == 35, "wrong checksum for band 1" assert ds.GetRasterBand(4).Checksum() == 35, "wrong checksum for band 4" ds = None ############################################################################### # Test gdal_retile.py with input images of different pixel sizes def test_gdal_retile_3(script_path): drv = gdal.GetDriverByName("GTiff") srs = osr.SpatialReference() srs.SetWellKnownGeogCS("WGS84") wkt = srs.ExportToWkt() # Create two images to tile together. The images will cover the geographic # range 0E-30E and 0-60N, split horizontally at 30N. The pixel size in the # second image will be twice that of the first time. If the make the first # image black and the second gray, then the result of tiling these two # together should be gray square stacked on top of a black square. # # 60 N --------------- # | | \ # | 50x50 | \ Image 2 # | | / # | | / # 30 N --------------- # | | \ # | 100x100 | \ Image 1 # | | / # | | / # 0 N --------------- # 0 E 30 E ds = drv.Create("tmp/in1.tif", 100, 100, 1) px1_x = 30.0 / ds.RasterXSize px1_y = 30.0 / ds.RasterYSize ds.SetProjection(wkt) ds.SetGeoTransform([0, px1_x, 0, 30, 0, -px1_y]) ds.GetRasterBand(1).Fill(0) ds = None ds = drv.Create("tmp/in2.tif", 50, 50, 1) px2_x = 30.0 / ds.RasterXSize px2_y = 30.0 / ds.RasterYSize ds.SetProjection(wkt) ds.SetGeoTransform([0, px2_x, 0, 60, 0, -px2_y]) ds.GetRasterBand(1).Fill(42) ds = None try: os.mkdir("tmp/outretile3") except OSError: pass test_py_scripts.run_py_script( script_path, "gdal_retile", "-v -levels 2 -r bilinear -targetDir tmp/outretile3 tmp/in1.tif tmp/in2.tif", ) ds = gdal.Open("tmp/outretile3/in1_1_1.tif") assert ds.GetProjectionRef().find("WGS 84") != -1, "Expected WGS 84\nGot : %s" % ( ds.GetProjectionRef() ) gt = ds.GetGeoTransform() expected_gt = [0, px1_x, 0, 60, 0, -px1_y] for i in range(6): assert not abs(gt[i] - expected_gt[i] > 1e-5), "Expected : %s\nGot : %s" % ( expected_gt, gt, ) assert ( ds.RasterXSize == 100 and ds.RasterYSize == 200 ), "Wrong raster dimensions : %d x %d" % (ds.RasterXSize, ds.RasterYSize) assert ds.RasterCount == 1, "Wrong raster count : %d " % (ds.RasterCount) assert ds.GetRasterBand(1).Checksum() == 38999, "Wrong checksum" ############################################################################### # Test gdal_retile.py -overlap def test_gdal_retile_4(script_path): try: os.mkdir("tmp/outretile4") except OSError: pass test_py_scripts.run_py_script( script_path, "gdal_retile", "-v -ps 8 7 -overlap 3 -targetDir tmp/outretile4 " + test_py_scripts.get_data_path("gcore") + "byte.tif", ) expected_results = [ ["tmp/outretile4/byte_1_1.tif", 8, 7], ["tmp/outretile4/byte_1_2.tif", 8, 7], ["tmp/outretile4/byte_1_3.tif", 8, 7], ["tmp/outretile4/byte_1_4.tif", 5, 7], ["tmp/outretile4/byte_2_1.tif", 8, 7], ["tmp/outretile4/byte_2_2.tif", 8, 7], ["tmp/outretile4/byte_2_3.tif", 8, 7], ["tmp/outretile4/byte_2_4.tif", 5, 7], ["tmp/outretile4/byte_3_1.tif", 8, 7], ["tmp/outretile4/byte_3_2.tif", 8, 7], ["tmp/outretile4/byte_3_3.tif", 8, 7], ["tmp/outretile4/byte_3_4.tif", 5, 7], ["tmp/outretile4/byte_4_1.tif", 8, 7], ["tmp/outretile4/byte_4_2.tif", 8, 7], ["tmp/outretile4/byte_4_3.tif", 8, 7], ["tmp/outretile4/byte_4_4.tif", 5, 7], ["tmp/outretile4/byte_5_1.tif", 8, 4], ["tmp/outretile4/byte_5_2.tif", 8, 4], ["tmp/outretile4/byte_5_3.tif", 8, 4], ["tmp/outretile4/byte_5_4.tif", 5, 4], ] for (filename, width, height) in expected_results: ds = gdal.Open(filename) assert ds.RasterXSize == width, filename assert ds.RasterYSize == height, filename ds = None test_py_scripts.run_py_script( script_path, "gdal_retile", "-v -levels 1 -ps 8 8 -overlap 4 -targetDir tmp/outretile4 " + test_py_scripts.get_data_path("gcore") + "byte.tif", ) expected_results = [ ["tmp/outretile4/byte_1_1.tif", 8, 8], ["tmp/outretile4/byte_1_2.tif", 8, 8], ["tmp/outretile4/byte_1_3.tif", 8, 8], ["tmp/outretile4/byte_1_4.tif", 8, 8], ["tmp/outretile4/byte_2_1.tif", 8, 8], ["tmp/outretile4/byte_2_2.tif", 8, 8], ["tmp/outretile4/byte_2_3.tif", 8, 8], ["tmp/outretile4/byte_2_4.tif", 8, 8], ["tmp/outretile4/byte_3_1.tif", 8, 8], ["tmp/outretile4/byte_3_2.tif", 8, 8], ["tmp/outretile4/byte_3_3.tif", 8, 8], ["tmp/outretile4/byte_3_4.tif", 8, 8], ["tmp/outretile4/byte_4_1.tif", 8, 8], ["tmp/outretile4/byte_4_2.tif", 8, 8], ["tmp/outretile4/byte_4_3.tif", 8, 8], ["tmp/outretile4/byte_4_4.tif", 8, 8], ["tmp/outretile4/1/byte_1_1.tif", 8, 8], ["tmp/outretile4/1/byte_1_2.tif", 6, 8], ["tmp/outretile4/1/byte_2_1.tif", 8, 6], ["tmp/outretile4/1/byte_2_2.tif", 6, 6], ] for (filename, width, height) in expected_results: ds = gdal.Open(filename) assert ds.RasterXSize == width, filename assert ds.RasterYSize == height, filename ds = None ############################################################################### # Test gdal_retile.py with input having a NoData value def test_gdal_retile_5(script_path): np = pytest.importorskip("numpy") nodata_value = -3.4028234663852886e38 raster_array = np.array(([0.0, 2.0], [-1.0, nodata_value])) drv = gdal.GetDriverByName("GTiff") srs = osr.SpatialReference() srs.SetWellKnownGeogCS("WGS84") wkt = srs.ExportToWkt() ds = drv.Create("tmp/in5.tif", 2, 2, 1, gdal.GDT_Float32) px1_x = 0.1 / ds.RasterXSize px1_y = 0.1 / ds.RasterYSize ds.SetProjection(wkt) ds.SetGeoTransform([0, px1_x, 0, 30, 0, -px1_y]) raster_band = ds.GetRasterBand(1) raster_band.SetNoDataValue(nodata_value) raster_band.WriteArray(raster_array) raster_band = None ds = None try: os.mkdir("tmp/outretile5") except OSError: pass test_py_scripts.run_py_script( script_path, "gdal_retile", "-v -targetDir tmp/outretile5 tmp/in5.tif" ) ds = gdal.Open("tmp/outretile5/in5_1_1.tif") raster_band = ds.GetRasterBand(1) assert ( raster_band.GetNoDataValue() == nodata_value ), "Wrong nodata value.\nExpected %f, Got: %f" % ( nodata_value, raster_band.GetNoDataValue(), ) min_val, max_val = raster_band.ComputeRasterMinMax() assert max_val, "Wrong maximum value.\nExpected 2.0, Got: %f" % max_val assert min_val == -1.0, "Wrong minimum value.\nExpected -1.0, Got: %f" % min_val ds = None ############################################################################### # Test gdal_retile.py @pytest.mark.require_driver("PNG") def test_gdal_retile_png(script_path): out_dirname = os.path.join("tmp", "outretile_png") os.mkdir(out_dirname) try: test_py_scripts.run_py_script( script_path, "gdal_retile", "-v -levels 2 -r bilinear -of PNG -targetDir " + '"' + out_dirname + '"' + " " + test_py_scripts.get_data_path("gcore") + "byte.tif", ) ds = gdal.Open(out_dirname + "/byte_1_1.png") assert ds.GetRasterBand(1).Checksum() == 4672 ds = None assert os.path.exists(out_dirname + "/byte_1_1.png.aux.xml") finally: shutil.rmtree(out_dirname) ############################################################################### # Cleanup def test_gdal_retile_cleanup(): lst = [ "tmp/outretile/1/byte_1_1.tif", "tmp/outretile/2/byte_1_1.tif", "tmp/outretile/byte_1_1.tif", "tmp/outretile/1", "tmp/outretile/2", "tmp/outretile", "tmp/outretile2/1/rgba_1_1.tif", "tmp/outretile2/2/rgba_1_1.tif", "tmp/outretile2/1", "tmp/outretile2/2", "tmp/outretile2/rgba_1_1.tif", "tmp/outretile2", "tmp/in1.tif", "tmp/in2.tif", "tmp/outretile3/1/in1_1_1.tif", "tmp/outretile3/2/in1_1_1.tif", "tmp/outretile3/1", "tmp/outretile3/2", "tmp/outretile3/in1_1_1.tif", "tmp/outretile3", "tmp/in5.tif", ] for filename in lst: try: os.remove(filename) except OSError: try: os.rmdir(filename) except OSError: pass shutil.rmtree("tmp/outretile4") if os.path.exists("tmp/outretile5"): shutil.rmtree("tmp/outretile5")