test_gdal_to_raster()

static void test_gdal_to_raster ( )

static

Definition at line 353 of file cu_gdal.c.

                                  {
        rt_pixtype pixtype = PT_64BF;
        rt_band band = NULL;
 
        rt_raster raster;
        rt_raster rast;
        const uint32_t width = 100;
        const uint32_t height = 100;
        uint32_t x;
        uint32_t y;
        int v;
        double values[width][height];
        int rtn = 0;
        double value;
 
        GDALDriverH gddrv = NULL;
        int destroy = 0;
        GDALDatasetH gdds = NULL;
 
        raster = rt_raster_new(width, height);
        CU_ASSERT(raster != NULL); /* or we're out of virtual memory */
 
        band = cu_add_band(raster, pixtype, 1, 0);
        CU_ASSERT(band != NULL);
 
        for (x = 0; x < width; x++) {
                for (y = 0; y < height; y++) {
                        values[x][y] = (((double) x * y) + (x + y) + (x + y * x)) / (x + y + 1);
                        rt_band_set_pixel(band, x, y, values[x][y], NULL);
                }
        }
 
        gdds = rt_raster_to_gdal_mem(raster, NULL, NULL, NULL, 0, &gddrv, &destroy);
        CU_ASSERT(gddrv != NULL);
        CU_ASSERT(gdds != NULL);
        CU_ASSERT_EQUAL((uint32_t)GDALGetRasterXSize(gdds), width);
        CU_ASSERT_EQUAL((uint32_t)GDALGetRasterYSize(gdds), height);
 
        rast = rt_raster_from_gdal_dataset(gdds);
        CU_ASSERT(rast != NULL);
        CU_ASSERT_EQUAL(rt_raster_get_num_bands(rast), 1);
 
        band = rt_raster_get_band(rast, 0);
        CU_ASSERT(band != NULL);
 
        for (x = 0; x < width; x++) {
                for (y = 0; y < height; y++) {
                        rtn = rt_band_get_pixel(band, x, y, &value, NULL);
                        CU_ASSERT_EQUAL(rtn, ES_NONE);
                        CU_ASSERT_DOUBLE_EQUAL(value, values[x][y], DBL_EPSILON);
                }
        }
 
        if (destroy && gddrv) {
                GDALDeregisterDriver(gddrv);
                GDALDestroyDriver(gddrv);
        }
        GDALClose(gdds);
        gdds = NULL;
        gddrv = NULL;
 
        cu_free_raster(rast);
        cu_free_raster(raster);
 
        raster = rt_raster_new(width, height);
        CU_ASSERT(raster != NULL); /* or we're out of virtual memory */
 
        pixtype = PT_8BSI;
        band = cu_add_band(raster, pixtype, 1, 0);
        CU_ASSERT(band != NULL);
 
        v = -127;
        for (x = 0; x < width; x++) {
                for (y = 0; y < height; y++) {
                        values[x][y] = v++;
                        rt_band_set_pixel(band, x, y, values[x][y], NULL);
                        if (v == 128)
                                v = -127;
                }
        }
 
        gdds = rt_raster_to_gdal_mem(raster, NULL, NULL, NULL, 0, &gddrv, &destroy);
        CU_ASSERT(gddrv != NULL);
        CU_ASSERT(gdds != NULL);
        CU_ASSERT_EQUAL((uint32_t)GDALGetRasterXSize(gdds), width);
        CU_ASSERT_EQUAL((uint32_t)GDALGetRasterYSize(gdds), height);
 
        rast = rt_raster_from_gdal_dataset(gdds);
        CU_ASSERT(rast != NULL);
        CU_ASSERT_EQUAL(rt_raster_get_num_bands(rast), 1);
 
        band = rt_raster_get_band(rast, 0);
        CU_ASSERT(band != NULL);
#if POSTGIS_GDAL_VERSION < 30700
        CU_ASSERT_EQUAL(rt_band_get_pixtype(band), PT_16BSI);
#else
        CU_ASSERT_EQUAL(rt_band_get_pixtype(band), PT_8BSI);
#endif
        for (x = 0; x < width; x++) {
                for (y = 0; y < height; y++) {
                        rtn = rt_band_get_pixel(band, x, y, &value, NULL);
#if POSTGIS_GDAL_VERSION < 30700
                        CU_ASSERT_DOUBLE_EQUAL(value, values[x][y], 1.);
                        CU_ASSERT_DOUBLE_EQUAL(value, values[x][y], DBL_EPSILON);
#else
                        CU_ASSERT_EQUAL(rtn, ES_NONE);
                        CU_ASSERT_DOUBLE_EQUAL(value, values[x][y], DBL_EPSILON);
#endif
                }
        }
        if (destroy && gddrv) {
                GDALDeregisterDriver(gddrv);
                GDALDestroyDriver(gddrv);
        }
 
        GDALClose(gdds);
        gdds = NULL;
        gddrv = NULL;
 
        cu_free_raster(rast);
        cu_free_raster(raster);
}

References cu_add_band(), cu_free_raster(), ES_NONE, PT_16BSI, PT_64BF, PT_8BSI, rt_band_get_pixel(), rt_band_get_pixtype(), rt_band_set_pixel(), rt_raster_from_gdal_dataset(), rt_raster_get_band(), rt_raster_get_num_bands(), rt_raster_new(), and rt_raster_to_gdal_mem().

Referenced by gdal_suite_setup().

Here is the call graph for this function:

Here is the caller graph for this function: