def test_applyverticalshiftgrid_3():
src_ds = gdal.Open('../gcore/data/byte.tif')
grid_ds = gdal.Warp('',
src_ds,
format='MEM',
dstSRS='EPSG:4326',
width=40,
height=40)
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
grid_ds,
options=['RESAMPLING=NEAREST'])
assert out_ds.RasterXSize == src_ds.RasterXSize
assert out_ds.RasterYSize == src_ds.RasterYSize
assert out_ds.GetGeoTransform() == src_ds.GetGeoTransform()
assert out_ds.GetProjectionRef() == src_ds.GetProjectionRef()
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 5112
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
grid_ds,
options=['RESAMPLING=BILINEAR'])
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 4867 or cs == 4868
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
grid_ds,
options=['RESAMPLING=CUBIC'])
cs = out_ds.GetRasterBand(1).Checksum()
assert cs in (4841, 4854, 4842) # 4842 on Mac / Conda
def test_applyverticalshiftgrid_5():
src_ds = gdal.Open('../gcore/data/byte.tif')
grid_ds = gdal.Translate('', src_ds, format='MEM')
grid_ds.GetRasterBand(1).Fill(0)
src_ds = gdal.Translate('',
src_ds,
format='MEM',
outputType=gdal.GDT_Float32,
scaleParams=[[0, 1, 0, 0.5]])
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds, srcUnitToMeter=2)
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 4672
src_ds = gdal.Open('../gcore/data/byte.tif')
grid_ds = gdal.Translate('', src_ds, format='MEM')
grid_ds.GetRasterBand(1).Fill(0)
src_ds = gdal.Translate('',
src_ds,
format='MEM',
outputType=gdal.GDT_Float32,
scaleParams=[[0, 1, 0, 0.5]])
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds, dstUnitToMeter=0.5)
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 4672
def applyverticalshiftgrid_5():
src_ds = gdal.Open('../gcore/data/byte.tif')
grid_ds = gdal.Translate('', src_ds, format='MEM')
grid_ds.GetRasterBand(1).Fill(0)
src_ds = gdal.Translate('',
src_ds,
format='MEM',
outputType=gdal.GDT_Float32,
scaleParams=[[0, 1, 0, 0.5]])
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds, srcUnitToMeter=2)
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 4672:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
src_ds = gdal.Open('../gcore/data/byte.tif')
grid_ds = gdal.Translate('', src_ds, format='MEM')
grid_ds.GetRasterBand(1).Fill(0)
src_ds = gdal.Translate('',
src_ds,
format='MEM',
outputType=gdal.GDT_Float32,
scaleParams=[[0, 1, 0, 0.5]])
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds, dstUnitToMeter=0.5)
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 4672:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
return 'success'
def applyverticalshiftgrid_3():
src_ds = gdal.Open('../gcore/data/byte.tif')
grid_ds = gdal.Warp('',
src_ds,
format='MEM',
dstSRS='EPSG:4326',
width=40,
height=40)
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
grid_ds,
options=['RESAMPLING=NEAREST'])
if out_ds.RasterXSize != src_ds.RasterXSize:
gdaltest.post_reason('fail')
print(out_ds.RasterXSize)
return 'fail'
if out_ds.RasterYSize != src_ds.RasterYSize:
gdaltest.post_reason('fail')
print(out_ds.RasterYSize)
return 'fail'
if out_ds.GetGeoTransform() != src_ds.GetGeoTransform():
gdaltest.post_reason('fail')
print(out_ds.GetGeoTransform())
return 'fail'
if out_ds.GetProjectionRef() != src_ds.GetProjectionRef():
gdaltest.post_reason('fail')
print(out_ds.GetProjectionRef())
return 'fail'
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 5112:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
grid_ds,
options=['RESAMPLING=BILINEAR'])
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 4867 and cs != 4868:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
grid_ds,
options=['RESAMPLING=CUBIC'])
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 4841 and cs != 4854:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
return 'success'
def test_applyverticalshiftgrid_1():
src_ds = gdal.Open('../gcore/data/byte.tif')
src_ds = gdal.Translate('', src_ds, format='MEM', width=20, height=40)
grid_ds = gdal.Translate('', src_ds, format='MEM')
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
assert out_ds.GetRasterBand(1).DataType == gdal.GDT_Byte
assert out_ds.RasterXSize == src_ds.RasterXSize
assert out_ds.RasterYSize == src_ds.RasterYSize
assert out_ds.GetGeoTransform() == src_ds.GetGeoTransform()
assert out_ds.GetProjectionRef() == src_ds.GetProjectionRef()
# Check that we can drop the reference to the sources
src_ds = None
grid_ds = None
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 10038
src_ds = gdal.Open('../gcore/data/byte.tif')
src_ds = gdal.Translate('', src_ds, format='MEM', width=20, height=40)
# Test block size
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
src_ds,
options=['BLOCKSIZE=15'])
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 10038
# Inverse transformer
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
src_ds,
True,
options=['DATATYPE=Float32'])
assert out_ds.GetRasterBand(1).DataType == gdal.GDT_Float32
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 0
def test_applyverticalshiftgrid_2():
sr = osr.SpatialReference()
sr.SetFromUserInput("WGS84")
for i in range(6):
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
if i != 0:
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
if i != 1:
src_ds.SetProjection(sr.ExportToWkt())
if i == 2:
src_ds.AddBand(gdal.GDT_Byte)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
if i != 3:
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
if i != 4:
grid_ds.SetProjection(sr.ExportToWkt())
if i == 5:
grid_ds.AddBand(gdal.GDT_Byte)
with gdaltest.error_handler():
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
assert out_ds is None, i
# Non invertable source geotransform
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 0, 0, 0, 0, 0])
src_ds.SetProjection(sr.ExportToWkt())
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
with gdaltest.error_handler():
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
assert out_ds is None
# Non invertable grid geotransform
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 0, 0, 0, 0, 0])
grid_ds.SetProjection(sr.ExportToWkt())
with gdaltest.error_handler():
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
assert out_ds is None
# No PROJ.4 translation for source SRS, coordinate transformation
# initialization has failed
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection('LOCAL_CS["foo"]')
with gdaltest.error_handler():
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
assert out_ds is None
# Out of memory
if gdal.GetConfigOption('SKIP_MEM_INTENSIVE_TEST') is None:
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
with gdaltest.error_handler():
out_ds = gdal.ApplyVerticalShiftGrid(
src_ds, grid_ds, options=['BLOCKSIZE=2000000000'])
assert out_ds is None
# Wrong DATATYPE
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
with gdaltest.error_handler():
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
grid_ds,
options=['DATATYPE=x'])
assert out_ds is None
def test_applyverticalshiftgrid_4():
sr = osr.SpatialReference()
sr.SetFromUserInput("WGS84")
# Nodata on source
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(1)
src_ds.GetRasterBand(1).SetNoDataValue(1)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
grid_ds.GetRasterBand(1).Fill(30)
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
assert out_ds.GetRasterBand(1).GetNoDataValue() == 1
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 1
# Nodata on grid
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(1)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
grid_ds.GetRasterBand(1).Fill(30)
grid_ds.GetRasterBand(1).SetNoDataValue(30)
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
assert out_ds.GetRasterBand(1).GetNoDataValue() is None
cs = out_ds.GetRasterBand(1).Checksum()
assert cs == 1
# ERROR_ON_MISSING_VERT_SHIFT due to non compatible extents
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(255)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([10, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
out_ds = gdal.ApplyVerticalShiftGrid(
src_ds, grid_ds, options=['ERROR_ON_MISSING_VERT_SHIFT=YES'])
with gdaltest.error_handler():
data = out_ds.GetRasterBand(1).ReadRaster()
assert data is None
# ERROR_ON_MISSING_VERT_SHIFT due to nodata in grid
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(255)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
grid_ds.GetRasterBand(1).SetNoDataValue(0)
out_ds = gdal.ApplyVerticalShiftGrid(
src_ds, grid_ds, options=['ERROR_ON_MISSING_VERT_SHIFT=YES'])
with gdaltest.error_handler():
data = out_ds.GetRasterBand(1).ReadRaster()
assert data is None
def applyverticalshiftgrid_1():
src_ds = gdal.Open('../gcore/data/byte.tif')
src_ds = gdal.Translate('', src_ds, format='MEM', width=20, height=40)
grid_ds = gdal.Translate('', src_ds, format='MEM')
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
if out_ds.GetRasterBand(1).DataType != gdal.GDT_Byte:
gdaltest.post_reason('fail')
print(out_ds.GetRasterBand(1).DataType)
return 'fail'
if out_ds.RasterXSize != src_ds.RasterXSize:
gdaltest.post_reason('fail')
print(out_ds.RasterXSize)
return 'fail'
if out_ds.RasterYSize != src_ds.RasterYSize:
gdaltest.post_reason('fail')
print(out_ds.RasterYSize)
return 'fail'
if out_ds.GetGeoTransform() != src_ds.GetGeoTransform():
gdaltest.post_reason('fail')
print(out_ds.GetGeoTransform())
return 'fail'
if out_ds.GetProjectionRef() != src_ds.GetProjectionRef():
gdaltest.post_reason('fail')
print(out_ds.GetProjectionRef())
return 'fail'
# Check that we can drop the reference to the sources
src_ds = None
grid_ds = None
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 10038:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
src_ds = gdal.Open('../gcore/data/byte.tif')
src_ds = gdal.Translate('', src_ds, format='MEM', width=20, height=40)
# Test block size
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
src_ds,
options=['BLOCKSIZE=15'])
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 10038:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
# Inverse transformer
out_ds = gdal.ApplyVerticalShiftGrid(src_ds,
src_ds,
True,
options=['DATATYPE=Float32'])
if out_ds.GetRasterBand(1).DataType != gdal.GDT_Float32:
gdaltest.post_reason('fail')
print(out_ds.GetRasterBand(1).DataType)
return 'fail'
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 0:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
return 'success'
def applyverticalshiftgrid_4():
sr = osr.SpatialReference()
sr.SetFromUserInput("WGS84")
# Nodata on source
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(1)
src_ds.GetRasterBand(1).SetNoDataValue(1)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
grid_ds.GetRasterBand(1).Fill(30)
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
if out_ds.GetRasterBand(1).GetNoDataValue() != 1:
gdaltest.post_reason('fail')
print(out_ds.GetRasterBand(1).GetNoDataValue())
return 'fail'
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 1:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
# Nodata on grid
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(1)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
grid_ds.GetRasterBand(1).Fill(30)
grid_ds.GetRasterBand(1).SetNoDataValue(30)
out_ds = gdal.ApplyVerticalShiftGrid(src_ds, grid_ds)
if out_ds.GetRasterBand(1).GetNoDataValue() is not None:
gdaltest.post_reason('fail')
print(out_ds.GetRasterBand(1).GetNoDataValue())
return 'fail'
cs = out_ds.GetRasterBand(1).Checksum()
if cs != 1:
gdaltest.post_reason('fail')
print(cs)
return 'fail'
# ERROR_ON_MISSING_VERT_SHIFT due to non compatible extents
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(255)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([10, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
out_ds = gdal.ApplyVerticalShiftGrid(
src_ds, grid_ds, options=['ERROR_ON_MISSING_VERT_SHIFT=YES'])
with gdaltest.error_handler():
data = out_ds.GetRasterBand(1).ReadRaster()
if data is not None:
gdaltest.post_reason('fail')
return 'fail'
# ERROR_ON_MISSING_VERT_SHIFT due to nodata in grid
src_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
src_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
src_ds.SetProjection(sr.ExportToWkt())
src_ds.GetRasterBand(1).Fill(255)
grid_ds = gdal.GetDriverByName('MEM').Create('', 1, 1)
grid_ds.SetGeoTransform([0, 1, 0, 0, 0, -1])
grid_ds.SetProjection(sr.ExportToWkt())
grid_ds.GetRasterBand(1).SetNoDataValue(0)
out_ds = gdal.ApplyVerticalShiftGrid(
src_ds, grid_ds, options=['ERROR_ON_MISSING_VERT_SHIFT=YES'])
with gdaltest.error_handler():
data = out_ds.GetRasterBand(1).ReadRaster()
if data is not None:
gdaltest.post_reason('fail')
return 'fail'
return 'success'
