test_gdal_to_raster()

static void test_gdal_to_raster ( )

static

Definition at line 419 of file cu_gdal.c.

References ovdump::band, cu_add_band(), cu_free_raster(), ES_NONE, PT_16BSI, PT_64BF, PT_8BSI, rtpixdump::rast, rtrowdump::raster, 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(), rt_raster_to_gdal_mem(), genraster::value, pixval::x, and pixval::y.

Referenced by gdal_suite_setup().

                                  {
        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_EQUAL(destroy, 0);
        CU_ASSERT(gdds != NULL);
        CU_ASSERT_EQUAL(GDALGetRasterXSize(gdds), width);
        CU_ASSERT_EQUAL(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);
                }
        }

        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_EQUAL(destroy, 0);
        CU_ASSERT(gdds != NULL);
        CU_ASSERT_EQUAL(GDALGetRasterXSize(gdds), width);
        CU_ASSERT_EQUAL(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);
        CU_ASSERT_EQUAL(rt_band_get_pixtype(band), PT_16BSI);

        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], 1.);
                }
        }

        GDALClose(gdds);
        gdds = NULL;
        gddrv = NULL;

        cu_free_raster(rast);
        cu_free_raster(raster);
}

Here is the call graph for this function:

Here is the caller graph for this function: