Changeset 17656

Timestamp:

04/06/14 18:47:50 (11 years ago)

Author:

cborstad

Message:

CHG: check whether geotiff is a 3-band rgb raster; if so, convert to single band greyscale

File:

• issm/trunk-jpl/src/m/plot/plot_overlay.py (modified) (2 diffs)

issm/trunk-jpl/src/m/plot/plot_overlay.py

@@ -52 +52 @@
         gtif=gdal.Open(inputname)
         arr=gtif.ReadAsArray()
-        #os.system('rm -rf ./temp.tif')
+        os.system('rm -rf ./temp.tif')
         
+        if gtif.RasterCount>=3:  # RGB array
+                r=gtif.GetRasterBand(1).ReadAsArray()
+                g=gtif.GetRasterBand(2).ReadAsArray()
+                b=gtif.GetRasterBand(3).ReadAsArray()
+                arr=0.299*r+0.587*g+0.114*b
+
+        # normalize array
+        arr=arr/npy.max(arr.ravel())
+
         # get parameters from cropped geotiff
         trans=gtif.GetGeoTransform()
@@ -67 +76 @@
         xg,yg=npy.meshgrid(xarr,yarr)
         if options.exist('basemap'):
-                # get handle to basemap instance
+                # TODO get handle to basemap instance 
                 # handle= functiontogethandle()
                 # create coordinate grid in map projection units (for plotting)