Changeset 14413

Timestamp:

05/09/08 11:04:42 (3 months ago)

Author:

rouault

Message:

Add tests for color tables, 8 bit and 16 bit cases, for GTiff

Files:

• trunk/autotest/gcore/tiff_write.py (modified) (2 diffs)

trunk/autotest/gcore/tiff_write.py

@@ -788 +788 @@
-        return 'fail'
-    src_ds = None
+
+    return 'success'
+
+
+###############################################################################
+# Test color table in a 8 bit image
+
+def tiff_write_26():
+
+    ds = gdaltest.tiff_drv.Create( 'tmp/ct8.tif', 1, 1, 1, gdal.GDT_Byte)
+
+    ct = gdal.ColorTable()
+    ct.SetColorEntry( 0, (255,255,255,255) )
+    ct.SetColorEntry( 1, (255,255,0,255) )
+    ct.SetColorEntry( 2, (255,0,255,255) )
+    ct.SetColorEntry( 3, (0,255,255,255) )
+
+    ds.GetRasterBand( 1 ).SetRasterColorTable( ct )
+
+    ds = None
+
+    ds = gdal.Open( 'tmp/ct8.tif' )
+
+    ct = ds.GetRasterBand( 1 ).GetRasterColorTable()
+    if ct.GetCount() != 256 or \
+       ct.GetColorEntry(0) != (255,255,255,255) or \
+       ct.GetColorEntry(1) != (255,255,0,255) or \
+       ct.GetColorEntry(2) != (255,0,255,255) or \
+       ct.GetColorEntry(3) != (0,255,255,255):
+        gdaltest.post_reason( 'Wrong color table entry.' )
+        return 'fail'
+
+    gdaltest.tiff_drv.Delete( 'tmp/ct8.tif' )
+
+    return 'success'
+
+###############################################################################
+# Test color table in a 16 bit image
+
+def tiff_write_27():
+
+    ds = gdaltest.tiff_drv.Create( 'tmp/ct16.tif', 1, 1, 1, gdal.GDT_UInt16)
+
+    ct = gdal.ColorTable()
+    ct.SetColorEntry( 0, (255,255,255,255) )
+    ct.SetColorEntry( 1, (255,255,0,255) )
+    ct.SetColorEntry( 2, (255,0,255,255) )
+    ct.SetColorEntry( 3, (0,255,255,255) )
+
+    ds.GetRasterBand( 1 ).SetRasterColorTable( ct )
+
+    ds = None
+
+    ds = gdal.Open( 'tmp/ct16.tif' )
+    new_ds = gdaltest.tiff_drv.CreateCopy( 'tmp/ct16_copy.tif', ds )
+    new_ds = None
+    ds = None
+
+    ds = gdal.Open( 'tmp/ct16_copy.tif' )
+
+    ct = ds.GetRasterBand( 1 ).GetRasterColorTable()
+    if ct.GetCount() != 65536 or \
+       ct.GetColorEntry(0) != (255,255,255,255) or \
+       ct.GetColorEntry(1) != (255,255,0,255) or \
+       ct.GetColorEntry(2) != (255,0,255,255) or \
+       ct.GetColorEntry(3) != (0,255,255,255):
+        gdaltest.post_reason( 'Wrong color table entry.' )
+        return 'fail'
+
+    ds = None
+
+    gdaltest.tiff_drv.Delete( 'tmp/ct16.tif' )
+    gdaltest.tiff_drv.Delete( 'tmp/ct16_copy.tif' )
-
-    return 'success'
@@ -822 +895 @@
-    tiff_write_24,
-    tiff_write_25,
+    tiff_write_26,
+    tiff_write_27,
-    tiff_write_cleanup ]
-