JezK
Edit File: gdal_edit.py
#!/usr/bin/env python3 # -*- coding: utf-8 -*- ############################################################################### # $Id$ # # Project: GDAL samples # Purpose: Edit in place various information of an existing GDAL dataset # Author: Even Rouault <even dot rouault at spatialys.com> # ############################################################################### # Copyright (c) 2011-2013, Even Rouault <even dot rouault at spatialys.com> # # 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 sys from osgeo import gdal, osr def Usage(): print("Usage: gdal_edit [--help-general] [-ro] [-a_srs srs_def]") print( " [-a_ullr ulx uly lrx lry] [-a_ulurll ulx uly urx ury llx lly]" ) print( " [-tr xres yres] [-unsetgt] [-unsetrpc] [-a_nodata value] [-unsetnodata]" ) print(" [-offset value] [-scale value] [-units value]") print(" [-colorinterp_X red|green|blue|alpha|gray|undefined]*") print(" [-unsetstats] [-stats] [-approx_stats]") print(" [-setstats min max mean stddev]") print(" [-gcp pixel line easting northing [elevation]]*") print( ' [-unsetmd] [-oo NAME=VALUE]* [-mo "META-TAG=VALUE"]* datasetname' ) print("") print("Edit in place various information of an existing GDAL dataset.") return 2 def ArgIsNumeric(s): i = 0 while i < len(s): if ( (s[i] < "0" or s[i] > "9") and s[i] != "." and s[i] != "e" and s[i] != "+" and s[i] != "-" ): return False i = i + 1 return True def gdal_edit(argv): argv = gdal.GeneralCmdLineProcessor(argv) if argv is None: return 2 datasetname = None srs = None ulx = None uly = None urx = None ury = None llx = None lly = None lrx = None lry = None nodata = None unsetnodata = False units = None xres = None yres = None unsetgt = False unsetstats = False stats = False setstats = False approx_stats = False unsetmd = False ro = False molist = [] gcp_list = [] open_options = [] offset = [] scale = [] colorinterp = {} unsetrpc = False i = 1 argc = len(argv) while i < argc: if argv[i] == "-ro": ro = True elif argv[i] == "-a_srs" and i < len(argv) - 1: srs = argv[i + 1] i = i + 1 elif argv[i] == "-a_ullr" and i < len(argv) - 4: ulx = float(argv[i + 1]) i = i + 1 uly = float(argv[i + 1]) i = i + 1 lrx = float(argv[i + 1]) i = i + 1 lry = float(argv[i + 1]) i = i + 1 elif argv[i] == "-a_ulurll" and i < len(argv) - 6: ulx = float(argv[i + 1]) i = i + 1 uly = float(argv[i + 1]) i = i + 1 urx = float(argv[i + 1]) i = i + 1 ury = float(argv[i + 1]) i = i + 1 llx = float(argv[i + 1]) i = i + 1 lly = float(argv[i + 1]) i = i + 1 elif argv[i] == "-tr" and i < len(argv) - 2: xres = float(argv[i + 1]) i = i + 1 yres = float(argv[i + 1]) i = i + 1 elif argv[i] == "-a_nodata" and i < len(argv) - 1: try: nodata = int(argv[i + 1]) except Exception: nodata = float(argv[i + 1]) i = i + 1 elif argv[i] == "-scale" and i < len(argv) - 1: scale.append(float(argv[i + 1])) i = i + 1 while i < len(argv) - 1 and ArgIsNumeric(argv[i + 1]): scale.append(float(argv[i + 1])) i = i + 1 elif argv[i] == "-offset" and i < len(argv) - 1: offset.append(float(argv[i + 1])) i = i + 1 while i < len(argv) - 1 and ArgIsNumeric(argv[i + 1]): offset.append(float(argv[i + 1])) i = i + 1 elif argv[i] == "-mo" and i < len(argv) - 1: molist.append(argv[i + 1]) i = i + 1 elif argv[i] == "-gcp" and i + 4 < len(argv): pixel = float(argv[i + 1]) i = i + 1 line = float(argv[i + 1]) i = i + 1 x = float(argv[i + 1]) i = i + 1 y = float(argv[i + 1]) i = i + 1 if i + 1 < len(argv) and ArgIsNumeric(argv[i + 1]): z = float(argv[i + 1]) i = i + 1 else: z = 0 gcp = gdal.GCP(x, y, z, pixel, line) gcp_list.append(gcp) elif argv[i] == "-unsetgt": unsetgt = True elif argv[i] == "-unsetrpc": unsetrpc = True elif argv[i] == "-unsetstats": unsetstats = True elif argv[i] == "-approx_stats": stats = True approx_stats = True elif argv[i] == "-stats": stats = True elif argv[i] == "-setstats" and i < len(argv) - 4: stats = True setstats = True if argv[i + 1] != "None": statsmin = float(argv[i + 1]) else: statsmin = None i = i + 1 if argv[i + 1] != "None": statsmax = float(argv[i + 1]) else: statsmax = None i = i + 1 if argv[i + 1] != "None": statsmean = float(argv[i + 1]) else: statsmean = None i = i + 1 if argv[i + 1] != "None": statsdev = float(argv[i + 1]) else: statsdev = None i = i + 1 elif argv[i] == "-units" and i < len(argv) - 1: units = argv[i + 1] i = i + 1 elif argv[i] == "-unsetmd": unsetmd = True elif argv[i] == "-unsetnodata": unsetnodata = True elif argv[i] == "-oo" and i < len(argv) - 1: open_options.append(argv[i + 1]) i = i + 1 elif argv[i].startswith("-colorinterp_") and i < len(argv) - 1: band = int(argv[i][len("-colorinterp_") :]) val = argv[i + 1] if val.lower() == "red": val = gdal.GCI_RedBand elif val.lower() == "green": val = gdal.GCI_GreenBand elif val.lower() == "blue": val = gdal.GCI_BlueBand elif val.lower() == "alpha": val = gdal.GCI_AlphaBand elif val.lower() == "gray" or val.lower() == "grey": val = gdal.GCI_GrayIndex elif val.lower() == "undefined": val = gdal.GCI_Undefined else: sys.stderr.write( "Unsupported color interpretation %s.\n" % val + "Only red, green, blue, alpha, gray, undefined are supported.\n" ) return Usage() colorinterp[band] = val i = i + 1 elif argv[i][0] == "-": sys.stderr.write("Unrecognized option : %s\n" % argv[i]) return Usage() elif datasetname is None: datasetname = argv[i] else: sys.stderr.write("Unexpected option : %s\n" % argv[i]) return Usage() i = i + 1 if datasetname is None: return Usage() if ( srs is None and lry is None and yres is None and not unsetgt and not unsetstats and not stats and not setstats and nodata is None and not units and not molist and not unsetmd and not gcp_list and not unsetnodata and not colorinterp and scale is None and offset is None and not unsetrpc ): print("No option specified") print("") return Usage() exclusive_option = 0 if lry is not None: exclusive_option = exclusive_option + 1 if lly is not None: # -a_ulurll exclusive_option = exclusive_option + 1 if yres is not None: exclusive_option = exclusive_option + 1 if unsetgt: exclusive_option = exclusive_option + 1 if exclusive_option > 1: print("-a_ullr, -a_ulurll, -tr and -unsetgt options are exclusive.") print("") return Usage() if unsetstats and stats: print("-unsetstats and either -stats or -approx_stats options are exclusive.") print("") return Usage() if unsetnodata and nodata: print("-unsetnodata and -nodata options are exclusive.") print("") return Usage() if open_options is not None: if ro: ds = gdal.OpenEx(datasetname, gdal.OF_RASTER, open_options=open_options) else: ds = gdal.OpenEx( datasetname, gdal.OF_RASTER | gdal.OF_UPDATE, open_options=open_options ) # GDAL 1.X compat elif ro: ds = gdal.Open(datasetname) else: ds = gdal.Open(datasetname, gdal.GA_Update) if ds is None: return -1 if scale: if len(scale) == 1: scale = scale * ds.RasterCount elif len(scale) != ds.RasterCount: print( "If more than one scale value is provided, their number must match the number of bands." ) print("") return Usage() if offset: if len(offset) == 1: offset = offset * ds.RasterCount elif len(offset) != ds.RasterCount: print( "If more than one offset value is provided, their number must match the number of bands." ) print("") return Usage() wkt = None if srs == "" or srs == "None": ds.SetProjection("") elif srs is not None: sr = osr.SpatialReference() if sr.SetFromUserInput(srs) != 0: print("Failed to process SRS definition: %s" % srs) return -1 wkt = sr.ExportToWkt() if not gcp_list: ds.SetProjection(wkt) if lry is not None: gt = [ ulx, (lrx - ulx) / ds.RasterXSize, 0, uly, 0, (lry - uly) / ds.RasterYSize, ] ds.SetGeoTransform(gt) elif lly is not None: # -a_ulurll gt = [ ulx, (urx - ulx) / ds.RasterXSize, (llx - ulx) / ds.RasterYSize, uly, (ury - uly) / ds.RasterXSize, (lly - uly) / ds.RasterYSize, ] ds.SetGeoTransform(gt) if yres is not None: gt = ds.GetGeoTransform() # Doh ! why is gt a tuple and not an array... gt = [gt[j] for j in range(6)] gt[1] = xres gt[5] = yres ds.SetGeoTransform(gt) if unsetgt: # For now only the GTiff drivers understands full-zero as a hint # to unset the geotransform if ds.GetDriver().ShortName == "GTiff": ds.SetGeoTransform([0, 0, 0, 0, 0, 0]) else: ds.SetGeoTransform([0, 1, 0, 0, 0, 1]) if gcp_list: if wkt is None: wkt = ds.GetGCPProjection() if wkt is None: wkt = "" ds.SetGCPs(gcp_list, wkt) if nodata is not None: for i in range(ds.RasterCount): ds.GetRasterBand(i + 1).SetNoDataValue(nodata) elif unsetnodata: for i in range(ds.RasterCount): ds.GetRasterBand(i + 1).DeleteNoDataValue() if scale: for i in range(ds.RasterCount): ds.GetRasterBand(i + 1).SetScale(scale[i]) if offset: for i in range(ds.RasterCount): ds.GetRasterBand(i + 1).SetOffset(offset[i]) if units: for i in range(ds.RasterCount): ds.GetRasterBand(i + 1).SetUnitType(units) if unsetstats: for i in range(ds.RasterCount): band = ds.GetRasterBand(i + 1) for key in band.GetMetadata().keys(): if key.startswith("STATISTICS_"): band.SetMetadataItem(key, None) if stats: for i in range(ds.RasterCount): ds.GetRasterBand(i + 1).ComputeStatistics(approx_stats) if setstats: for i in range(ds.RasterCount): if ( statsmin is None or statsmax is None or statsmean is None or statsdev is None ): ds.GetRasterBand(i + 1).ComputeStatistics(approx_stats) min, max, mean, stdev = ds.GetRasterBand(i + 1).GetStatistics( approx_stats, True ) if statsmin is None: statsmin = min if statsmax is None: statsmax = max if statsmean is None: statsmean = mean if statsdev is None: statsdev = stdev ds.GetRasterBand(i + 1).SetStatistics( statsmin, statsmax, statsmean, statsdev ) if molist: if unsetmd: md = {} else: md = ds.GetMetadata() for moitem in molist: equal_pos = moitem.find("=") if equal_pos > 0: md[moitem[0:equal_pos]] = moitem[equal_pos + 1 :] ds.SetMetadata(md) elif unsetmd: ds.SetMetadata({}) for band in colorinterp: ds.GetRasterBand(band).SetColorInterpretation(colorinterp[band]) if unsetrpc: ds.SetMetadata(None, "RPC") ds = band = None return 0 def main(argv=sys.argv): return gdal_edit(argv) if __name__ == "__main__": sys.exit(main(sys.argv))