hdf5/hl/test/test_image.c

968 lines
28 KiB
C
Raw Normal View History

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
2020-09-30 22:27:10 +08:00
* Copyright by The HDF Group. *
* Copyright by the Board of Trustees of the University of Illinois. *
* All rights reserved. *
* *
* This file is part of HDF5. The full HDF5 copyright notice, including *
* terms governing use, modification, and redistribution, is contained in *
* the COPYING file, which can be found at the root of the source code *
* distribution tree, or in https://www.hdfgroup.org/licenses. *
* If you do not have access to either file, you may request a copy from *
* help@hdfgroup.org. *
2020-09-30 22:27:10 +08:00
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
#include "h5hltest.h"
#include "H5srcdir.h"
#include "H5LTpublic.h"
#include "H5IMpublic.h"
#include "pal_rgb.h"
#define FILE_NAME "test_image1.h5"
#define FILE2 "test_image2.h5"
#define FILE3 "test_image3.h5"
#define DATA_FILE1 "image8.txt"
#define DATA_FILE2 "image24pixel.txt"
#define DATA_FILE3 "image24plane.txt"
#define DATA_FILE4 "usa.wri"
#define PAL2_FILE "sepia.pal"
#define PAL3_FILE "earth.pal"
#define IMAGE1_NAME "image8bit"
#define IMAGE2_NAME "image24bitpixel"
#define IMAGE3_NAME "image24bitplane"
#define PAL_NAME "palette"
#define PAL1_NAME "rainbow"
#define PAL2_NAME "sepia"
#define PAL3_NAME "earth"
#define PAL4_NAME "blue-red"
2020-09-30 22:27:10 +08:00
#define WIDTH 400
#define HEIGHT 200
#define PAL_ENTRIES 256
/* struct to store RGB values read from a .pal file */
typedef struct rgb_t {
unsigned char r;
unsigned char g;
unsigned char b;
} rgb_t;
/* prototypes */
static int test_simple(void);
static int test_data(void);
static int test_generate(void);
2020-09-30 22:27:10 +08:00
static int read_data(const char *file_name, hsize_t *width, hsize_t *height);
static int read_palette(const char *file_name, rgb_t *palette, size_t palette_size);
/* globals */
unsigned char *image_data = NULL;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* the main program
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
int
main(void)
{
2020-09-30 22:27:10 +08:00
int nerrors = 0;
2020-09-30 22:27:10 +08:00
nerrors += test_simple() < 0 ? 1 : 0;
nerrors += test_data() < 0 ? 1 : 0;
nerrors += test_generate() < 0 ? 1 : 0;
2020-09-30 22:27:10 +08:00
if (nerrors)
goto error;
2019-08-25 03:07:19 +08:00
HDprintf("All image tests passed.\n");
return 0;
error:
2020-09-30 22:27:10 +08:00
HDprintf("***** %d IMAGE TEST%s FAILED! *****\n", nerrors, 1 == nerrors ? "" : "S");
return 1;
}
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* a simple test that generates images and palettes
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
static int
test_simple(void)
{
2020-09-30 22:27:10 +08:00
hsize_t width = WIDTH;
hsize_t height = HEIGHT;
hsize_t planes;
hid_t fid;
int i, j, n, space;
hsize_t u;
char interlace[20];
hssize_t npals;
/* 8-bit image */
unsigned char *buf1 = NULL;
2020-09-30 22:27:10 +08:00
unsigned char pal[PAL_ENTRIES * 3]; /* palette array */
hsize_t pal_dims[2] = {PAL_ENTRIES, 3}; /* palette dimensions */
/* 24-bit image */
unsigned char *buf2 = NULL;
/* read data */
unsigned char *buf1_out = NULL;
unsigned char *buf2_out = NULL;
2020-09-30 22:27:10 +08:00
unsigned char pal_out[PAL_ENTRIES * 3]; /* palette array */
hsize_t pal_dims_out[2]; /* palette dimensions */
/* Allocate image buffers */
buf1 = (unsigned char *)HDmalloc(WIDTH * HEIGHT);
HDassert(buf1);
2020-09-30 22:27:10 +08:00
buf2 = (unsigned char *)HDmalloc(WIDTH * HEIGHT * 3);
HDassert(buf2);
buf1_out = (unsigned char *)HDmalloc(WIDTH * HEIGHT);
HDassert(buf1_out);
2020-09-30 22:27:10 +08:00
buf2_out = (unsigned char *)HDmalloc(WIDTH * HEIGHT * 3);
HDassert(buf2_out);
/* create an image */
2020-09-30 22:27:10 +08:00
space = WIDTH * HEIGHT / PAL_ENTRIES;
for (i = 0, j = 0, n = 0; i < WIDTH * HEIGHT; i++, j++) {
buf1[i] = (unsigned char)n;
2020-09-30 22:27:10 +08:00
if (j > space) {
n++;
2020-09-30 22:27:10 +08:00
j = 0;
}
}
/* create an image */
2020-09-30 22:27:10 +08:00
space = WIDTH * HEIGHT / 256;
for (i = 0, j = 0, n = 0; i < WIDTH * HEIGHT * 3; i += 3, j++) {
buf2[i] = (unsigned char)n;
buf2[i + 1] = 0;
buf2[i + 2] = (unsigned char)(255 - n);
if (j > space) {
n++;
2020-09-30 22:27:10 +08:00
j = 0;
}
}
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* define a palette, blue to red tones
*-------------------------------------------------------------------------
*/
for (i = 0, n = 0; i < PAL_ENTRIES * 3; i += 3, n++) {
pal[i] = (unsigned char)n; /* red */
pal[i + 1] = 0; /* green */
pal[i + 2] = (unsigned char)(255 - n); /* blue */
}
/* Create a new HDF5 file using default properties. */
2020-09-30 22:27:10 +08:00
fid = H5Fcreate(FILE_NAME, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* Indexed image test
*-------------------------------------------------------------------------
*/
HL_TESTING2("indexed image");
/* Write image */
2020-09-30 22:27:10 +08:00
if (H5IMmake_image_8bit(fid, IMAGE1_NAME, width, height, buf1) < 0)
goto out;
/* Make a palette */
2020-09-30 22:27:10 +08:00
if (H5IMmake_palette(fid, PAL_NAME, pal_dims, pal) < 0)
goto out;
/* Attach a palette to the image dataset */
2020-09-30 22:27:10 +08:00
if (H5IMlink_palette(fid, IMAGE1_NAME, PAL_NAME) < 0)
goto out;
/* Read image */
2020-09-30 22:27:10 +08:00
if (H5IMget_image_info(fid, IMAGE1_NAME, &width, &height, &planes, interlace, &npals) < 0)
goto out;
2020-09-30 22:27:10 +08:00
if (H5IMread_image(fid, IMAGE1_NAME, buf1_out) < 0)
goto out;
2020-09-30 22:27:10 +08:00
for (u = 0; u < height * width * planes; u++) {
if (buf1[u] != buf1_out[u])
goto out;
}
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* True color image test
*-------------------------------------------------------------------------
*/
HL_TESTING2("true color image");
/* Write image */
2020-09-30 22:27:10 +08:00
if (H5IMmake_image_24bit(fid, IMAGE2_NAME, width, height, "INTERLACE_PIXEL", buf2))
goto out;
/* Read image */
2020-09-30 22:27:10 +08:00
if (H5IMget_image_info(fid, IMAGE2_NAME, &width, &height, &planes, interlace, &npals) < 0)
goto out;
2020-09-30 22:27:10 +08:00
if (H5IMread_image(fid, IMAGE2_NAME, buf2_out) < 0)
goto out;
2020-09-30 22:27:10 +08:00
for (u = 0; u < height * width * planes; u++) {
if (buf2[u] != buf2_out[u])
goto out;
}
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* H5IMget_npalettes test
*-------------------------------------------------------------------------
*/
HL_TESTING2("pallete functions");
2020-09-30 22:27:10 +08:00
if (H5IMget_npalettes(fid, IMAGE1_NAME, &npals) < 0)
goto out;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* H5IMget_palette_info test
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
if (H5IMget_palette_info(fid, IMAGE1_NAME, 0, pal_dims_out) < 0)
goto out;
2020-09-30 22:27:10 +08:00
for (i = 0; i < 2; i++) {
if (pal_dims[i] != pal_dims_out[i])
goto out;
}
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* H5IMget_palette test
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
if (H5IMget_palette(fid, IMAGE1_NAME, 0, pal_out) < 0)
goto out;
2020-09-30 22:27:10 +08:00
for (i = 0; i < PAL_ENTRIES * 3; i++) {
if (pal[i] != pal_out[i])
goto out;
}
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* H5IMis_image test
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
if (H5IMis_image(fid, IMAGE1_NAME) < 0)
goto out;
2020-09-30 22:27:10 +08:00
if (H5IMis_image(fid, IMAGE2_NAME) < 0)
goto out;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* H5IMis_palette test
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
if (H5IMis_palette(fid, PAL_NAME) < 0)
goto out;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* end tests
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
if (buf1)
HDfree(buf1);
2020-09-30 22:27:10 +08:00
if (buf2)
HDfree(buf2);
2020-09-30 22:27:10 +08:00
if (buf1_out)
HDfree(buf1_out);
2020-09-30 22:27:10 +08:00
if (buf2_out)
HDfree(buf2_out);
/* Close the file. */
2020-09-30 22:27:10 +08:00
if (H5Fclose(fid) < 0)
goto out;
PASSED();
return 0;
/* error zone, gracefully close */
out:
2020-09-30 22:27:10 +08:00
if (buf1)
HDfree(buf1);
2020-09-30 22:27:10 +08:00
if (buf2)
HDfree(buf2);
2020-09-30 22:27:10 +08:00
if (buf1_out)
HDfree(buf1_out);
2020-09-30 22:27:10 +08:00
if (buf2_out)
HDfree(buf2_out);
H5E_BEGIN_TRY
{
H5Fclose(fid);
}
2020-09-30 22:27:10 +08:00
H5E_END_TRY;
H5_FAILED();
return FAIL;
}
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* read sample realistic image data from ASCII files
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
static int
test_data(void)
{
2020-09-30 22:27:10 +08:00
hid_t fid;
hsize_t pal_dims[2];
hsize_t width;
hsize_t height;
unsigned char pal[256 * 3]; /* buffer to hold an HDF5 palette */
rgb_t rgb[256]; /* buffer to hold a .pal file palette */
int i, n;
/* create a file using default properties */
2020-09-30 22:27:10 +08:00
if ((fid = H5Fcreate(FILE2, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT)) < 0)
goto out;
2019-08-25 03:07:19 +08:00
HDprintf("Testing read ascii image data and generate images\n");
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* read 8bit image data
*-------------------------------------------------------------------------
*/
HL_TESTING2("make indexed image");
/* read first data file */
2020-09-30 22:27:10 +08:00
if (read_data(DATA_FILE1, &width, &height) < 0)
goto out;
/* make an image */
2020-09-30 22:27:10 +08:00
if (H5IMmake_image_8bit(fid, IMAGE1_NAME, width, height, image_data) < 0)
goto out;
PASSED();
HL_TESTING2("attaching palettes");
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* palette #1. rainbow palette. data is contained in "pal_rgb.h"
*-------------------------------------------------------------------------
*/
/* initialize the palette data */
pal_dims[0] = 256;
pal_dims[1] = 3;
/* make a palette */
2020-09-30 22:27:10 +08:00
if (H5IMmake_palette(fid, PAL1_NAME, pal_dims, pal_rgb) < 0)
goto out;
/* attach a palette to the image dataset */
2020-09-30 22:27:10 +08:00
if (H5IMlink_palette(fid, IMAGE1_NAME, PAL1_NAME) < 0)
goto out;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* palette #2. sepia palette.
* read a PAL file and attach the palette to the HDF5 file
*-------------------------------------------------------------------------
*/
/* read a PAL file */
if (read_palette(PAL2_FILE, rgb, (sizeof(rgb) / sizeof(rgb[0]))) < 0)
goto out;
/* transfer to the HDF5 buffer */
2020-09-30 22:27:10 +08:00
for (i = 0, n = 0; i < 256 * 3; i += 3, n++) {
pal[i] = rgb[n].r;
pal[i + 1] = rgb[n].g;
pal[i + 2] = rgb[n].b;
}
/* make a palette */
2020-09-30 22:27:10 +08:00
if (H5IMmake_palette(fid, PAL2_NAME, pal_dims, pal) < 0)
goto out;
/* attach the palette to the image dataset */
2020-09-30 22:27:10 +08:00
if (H5IMlink_palette(fid, IMAGE1_NAME, PAL2_NAME) < 0)
goto out;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* palette #3. earth palette.
* read a PAL file and attach the palette to the HDF5 file
*-------------------------------------------------------------------------
*/
/* read a PAL file */
if (read_palette(PAL3_FILE, rgb, (sizeof(rgb) / sizeof(rgb[0]))) < 0)
goto out;
/* transfer to the HDF5 buffer */
2020-09-30 22:27:10 +08:00
for (i = 0, n = 0; i < 256 * 3; i += 3, n++) {
pal[i] = rgb[n].r;
pal[i + 1] = rgb[n].g;
pal[i + 2] = rgb[n].b;
}
/* make a palette */
2020-09-30 22:27:10 +08:00
if (H5IMmake_palette(fid, PAL3_NAME, pal_dims, pal) < 0)
goto out;
/* attach the palette to the image dataset */
2020-09-30 22:27:10 +08:00
if (H5IMlink_palette(fid, IMAGE1_NAME, PAL3_NAME) < 0)
goto out;
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* palette #4. blue-red
* make a palette whith blue to red colors
*-------------------------------------------------------------------------
*/
for (i = 0, n = 0; i < 256 * 3; i += 3, n++) {
pal[i] = (unsigned char)n;
pal[i + 1] = 0;
pal[i + 2] = (unsigned char)(255 - n);
}
/* make a palette */
2020-09-30 22:27:10 +08:00
if (H5IMmake_palette(fid, PAL4_NAME, pal_dims, pal) < 0)
goto out;
/* attach the palette to the image dataset */
2020-09-30 22:27:10 +08:00
if (H5IMlink_palette(fid, IMAGE1_NAME, PAL4_NAME) < 0)
goto out;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* true color image example with pixel interlace
*-------------------------------------------------------------------------
*/
HL_TESTING2("make true color image with pixel interlace");
/* read second data file */
2020-09-30 22:27:10 +08:00
if ((read_data(DATA_FILE2, &width, &height)) < 0)
goto out;
/* make image */
2020-09-30 22:27:10 +08:00
if ((H5IMmake_image_24bit(fid, IMAGE2_NAME, width, height, "INTERLACE_PIXEL", image_data)) < 0)
goto out;
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* True color image example with plane interlace
*-------------------------------------------------------------------------
*/
HL_TESTING2("make true color image with plane interlace");
/* read third data file */
2020-09-30 22:27:10 +08:00
if ((read_data(DATA_FILE3, &width, &height)) < 0)
goto out;
/* make image */
2020-09-30 22:27:10 +08:00
if ((H5IMmake_image_24bit(fid, IMAGE3_NAME, width, height, "INTERLACE_PLANE", image_data)) < 0)
goto out;
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* close
*-------------------------------------------------------------------------
*/
if (H5Fclose(fid) < 0)
goto out;
/* Release memory buffer */
HDfree(image_data);
return 0;
/* error zone, gracefully close */
out:
/* Release memory buffer */
2020-09-30 22:27:10 +08:00
if (image_data)
HDfree(image_data);
H5E_BEGIN_TRY
{
H5Fclose(fid);
}
2020-09-30 22:27:10 +08:00
H5E_END_TRY;
H5_FAILED();
return FAIL;
}
/*
The following test provides an examples of how to generate HDF5 image data from
floating point data. In the example we use real life topographic data
(from the North American hemisphere). In the dataset sea values are represented
as negative numbers and land values are represented as positive numbers.
The example generates 3 HDF5 images, one that generates an image from all the values,
another that generates an image from the land values and another that generates an
image from the sea values.
For the example we used data from MODB, the Mediterranean Oceanic Data Base
http://modb.oce.ulg.ac.be/
*/
2020-09-30 22:27:10 +08:00
static int
test_generate(void)
{
2020-09-30 22:27:10 +08:00
hid_t fid;
hsize_t pal_dims[2] = {256, 3};
float * data = NULL;
int imax, jmax, kmax;
int n_elements;
float valex, xmin, xmax, value;
FILE * f = NULL;
const char *data_file = H5_get_srcdir_filename(DATA_FILE4);
2020-09-30 22:27:10 +08:00
int i;
int retval = FAIL;
/* create a file using default properties */
2020-09-30 22:27:10 +08:00
if ((fid = H5Fcreate(FILE3, H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT)) < 0)
goto out;
2019-08-25 03:07:19 +08:00
HDprintf("Testing read and process data and make indexed images\n");
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* read data; the file data format is described below
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
f = HDfopen(data_file, "r");
if (f == NULL) {
HDprintf("Could not find file %s. Try set $srcdir \n", data_file);
goto out;
}
/*
!The first line of the ASCII file contains the dimension of the array :
! IMAX, JMAX, KMAX. The integers are stored with the FORTRAN format I5.
!The second line contains the exclusion value, the minimum and the maximum value of this
! file. These numbers are stored with the FORTRAN format E12.5.
! The remaining lines contains the data of the array, with 5 numbers per line
! (except the last line for each I-line).
! The array is stored in horizontal slices from sea surface to sea bottom and from
! north to south. So the indexes go from :
!
! DO K = KMAX to 1
! DO J = JMAX to 1
! DO I = 1 to IMAX
! read
! OD
! OD
! OD
!
! ____________________________
! / /| (imax,jmax,kmax)
! / sea surface / |
! / / |
! /__________________________ / |
! | | |
! | | | (imax,jmax,1) n
! | | / /
! | | / /
! ^ j | | / w <-----o-----> e
! k | / |__________________________|/ /
! | / (imax,1,1) /
! |----------> s
! i
!
*/
if (HDfscanf(f, "%d %d %d", &imax, &jmax, &kmax) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
if (HDfscanf(f, "%f %f %f", &valex, &xmin, &xmax) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
/* Sanity check on scanned-in values */
2020-09-30 22:27:10 +08:00
if (imax < 1 || jmax < 1 || kmax < 1)
goto out;
/* Test product for integer overflow */
2020-09-30 22:27:10 +08:00
if (imax > INT_MAX / jmax)
goto out;
2020-09-30 22:27:10 +08:00
if (imax * jmax > INT_MAX / kmax)
goto out;
n_elements = imax * jmax * kmax;
/* Test buffer sizes for overflow */
2020-09-30 22:27:10 +08:00
if (n_elements > INT_MAX / (int)sizeof(unsigned char))
goto out;
2020-09-30 22:27:10 +08:00
if (n_elements > INT_MAX / (int)sizeof(float))
goto out;
data = (float *)HDmalloc((size_t)n_elements * sizeof(float));
2020-09-30 22:27:10 +08:00
if (NULL == data)
goto out;
image_data = (unsigned char *)HDmalloc((size_t)n_elements * sizeof(unsigned char));
2020-09-30 22:27:10 +08:00
if (NULL == image_data)
goto out;
2020-09-30 22:27:10 +08:00
for (i = 0; i < n_elements; i++) {
if (HDfscanf(f, "%f ", &value) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
data[i] = value;
}
HDfclose(f);
f = NULL;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* transform the data from floating point to unsigned char
* we are processing all the data here
*-------------------------------------------------------------------------
*/
HL_TESTING2("make indexed image from all the data");
2020-09-30 22:27:10 +08:00
for (i = 0; i < n_elements; i++)
image_data[i] = (unsigned char)((255 * (data[i] - xmin)) / (xmax - xmin));
/* Make the image */
2020-09-30 22:27:10 +08:00
if ((H5IMmake_image_8bit(fid, "All data", (hsize_t)imax, (hsize_t)jmax, image_data)) < 0)
goto out;
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* transform the data from floating point to unsigned char
* here we just process the land data
*-------------------------------------------------------------------------
*/
HL_TESTING2("make indexed image from land data");
2020-09-30 22:27:10 +08:00
for (i = 0; i < n_elements; i++) {
if (data[i] < 0.0f)
image_data[i] = 0;
else
image_data[i] = (unsigned char)((255 * data[i]) / xmax);
}
/* make the image */
2020-09-30 22:27:10 +08:00
if ((H5IMmake_image_8bit(fid, "Land data", (hsize_t)imax, (hsize_t)jmax, image_data)) < 0)
goto out;
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* transform the data from floating point to unsigned char
* here we just process the sea data
*-------------------------------------------------------------------------
*/
HL_TESTING2("make indexed image from sea data");
2020-09-30 22:27:10 +08:00
for (i = 0; i < n_elements; i++) {
if (data[i] > 0.0f)
image_data[i] = 0;
else {
image_data[i] = (unsigned char)((255.0f * (data[i] - xmin)) / (xmax - xmin));
}
}
/* make the image */
2020-09-30 22:27:10 +08:00
if ((H5IMmake_image_8bit(fid, "Sea data", (hsize_t)imax, (hsize_t)jmax, image_data)) < 0)
goto out;
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* make a palette and attach it to the datasets
*-------------------------------------------------------------------------
*/
HL_TESTING2("attaching palettes");
/* make a palette */
2020-09-30 22:27:10 +08:00
if ((H5IMmake_palette(fid, PAL1_NAME, pal_dims, pal_rgb)) < 0)
goto out;
/* attach the palette to the image datasets */
2020-09-30 22:27:10 +08:00
if ((H5IMlink_palette(fid, "All data", PAL1_NAME)) < 0)
goto out;
2020-09-30 22:27:10 +08:00
if ((H5IMlink_palette(fid, "Land data", PAL1_NAME)) < 0)
goto out;
2020-09-30 22:27:10 +08:00
if ((H5IMlink_palette(fid, "Sea data", PAL1_NAME)) < 0)
goto out;
PASSED();
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* close
*-------------------------------------------------------------------------
*/
if (H5Fclose(fid) < 0)
goto out;
/* Release memory buffers */
HDfree(data);
HDfree(image_data);
/* Indicate success */
return 0;
/* error zone, gracefully close */
out:
/* Release memory buffers */
2020-09-30 22:27:10 +08:00
if (data)
HDfree(data);
2020-09-30 22:27:10 +08:00
if (image_data)
HDfree(image_data);
H5E_BEGIN_TRY
{
H5Fclose(fid);
}
2020-09-30 22:27:10 +08:00
H5E_END_TRY;
if (f)
HDfclose(f);
H5_FAILED();
return retval;
}
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* read_data
* utility function to read ASCII image data
* the files have a header of the type
*
* components
* n
* height
* n
* width
* n
*
* followed by the image data
*
*-------------------------------------------------------------------------
*/
static int
read_data(const char *fname, /*IN*/
hsize_t * width, /*OUT*/
hsize_t * height /*OUT*/)
{
2020-09-30 22:27:10 +08:00
int i, n;
int color_planes;
char str[20];
FILE * f = NULL;
int w, h;
int n_elements;
const char *data_file = H5_get_srcdir_filename(fname);
2020-09-30 22:27:10 +08:00
int ret_val = -1;
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* read
*-------------------------------------------------------------------------
*/
2020-09-30 22:27:10 +08:00
if (NULL == (f = HDfopen(data_file, "r"))) {
HDprintf("Could not open file %s. Try set $srcdir \n", data_file);
goto out;
}
if (HDfscanf(f, "%s", str) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
if (HDfscanf(f, "%d", &color_planes) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
if (HDfscanf(f, "%s", str) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
if (HDfscanf(f, "%d", &h) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
if (HDfscanf(f, "%s", str) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
if (HDfscanf(f, "%d", &w) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
/* Check product for overflow */
2020-09-30 22:27:10 +08:00
if (w < 1 || h < 1 || color_planes < 1)
goto out;
2020-09-30 22:27:10 +08:00
if (w > INT_MAX / h)
goto out;
2020-09-30 22:27:10 +08:00
if (w * h > INT_MAX / color_planes)
goto out;
/* Compute buffer size */
n_elements = w * h * color_planes;
/* Check buffer size for overflow */
2020-09-30 22:27:10 +08:00
if (n_elements > INT_MAX / (int)sizeof(unsigned char))
goto out;
/* Release the buffer, if it was previously allocated */
2020-09-30 22:27:10 +08:00
if (image_data)
HDfree(image_data);
/* Allocate the image data buffer */
image_data = (unsigned char *)HDmalloc((size_t)n_elements * sizeof(unsigned char));
2020-09-30 22:27:10 +08:00
if (NULL == image_data)
goto out;
2020-09-30 22:27:10 +08:00
*width = (hsize_t)w;
*height = (hsize_t)h;
/* Read data elements */
2020-09-30 22:27:10 +08:00
for (i = 0; i < n_elements; i++) {
if (HDfscanf(f, "%d", &n) < 0 && HDferror(f)) {
2020-09-30 22:27:10 +08:00
HDprintf("fscanf error in file %s.\n", data_file);
goto out;
} /* end if */
image_data[i] = (unsigned char)n;
} /* end for */
/* Indicate success */
ret_val = 1;
out:
2020-09-30 22:27:10 +08:00
if (f)
HDfclose(f);
return ret_val;
} /* end read_data() */
/*-------------------------------------------------------------------------
2020-09-30 22:27:10 +08:00
* read_palette
* Read an ASCII palette file .PAL into an array
* the files have a header of the type
*
* Parameters:
* fname - name of file to read.
* palette - array of rgb_t to store the read palette.
* palette_size - number of elements in 'palette' array
*
*-------------------------------------------------------------------------
*/
#define STRING_JASC "JASC-PAL"
#define VERSION_JASC "0100"
#define STRING_CWPAL "CWPAL"
#define VERSION_CWPAL "100"
2020-09-30 22:27:10 +08:00
static int
read_palette(const char *fname, rgb_t *palette, size_t palette_size)
{
2020-09-30 22:27:10 +08:00
FILE * file;
char buffer[80];
unsigned u;
unsigned int red;
unsigned int green;
unsigned int blue;
unsigned nentries;
const char * data_file = H5_get_srcdir_filename(fname);
/* ensure the given palette is valid */
if (!palette)
return -1;
/* open the input file */
2020-09-30 22:27:10 +08:00
if (!(file = HDfopen(data_file, "r"))) {
HDprintf("Could not open file %s. Try set $srcdir \n", data_file);
return -1;
}
/* read the file ident string */
2020-09-30 22:27:10 +08:00
if (HDfgets(buffer, sizeof(buffer), file) == NULL) {
HDfclose(file);
return -1;
}
/* ensure it matches the palette file ident string */
2020-09-30 22:27:10 +08:00
if (HDstrncmp(buffer, STRING_JASC, sizeof(STRING_JASC) - 1) != 0 &&
HDstrncmp(buffer, STRING_CWPAL, sizeof(STRING_CWPAL) - 1) != 0) {
HDfclose(file);
return -1;
}
/* read the version string */
2020-09-30 22:27:10 +08:00
if (HDfgets(buffer, sizeof(buffer), file) == NULL) {
HDfclose(file);
return -1;
}
/* ensure it matches the palette file version string */
2020-09-30 22:27:10 +08:00
if (HDstrncmp(buffer, VERSION_JASC, sizeof(VERSION_JASC) - 1) != 0 &&
HDstrncmp(buffer, VERSION_CWPAL, sizeof(VERSION_CWPAL) - 1) != 0) {
HDfclose(file);
return -1;
}
/* read the number of colors */
2020-09-30 22:27:10 +08:00
if (HDfgets(buffer, sizeof(buffer), file) == NULL) {
HDfclose(file);
return -1;
}
/* extract the number of colors.
check for missing version or number of colors
in this case it reads the first entry
*/
2020-09-30 22:27:10 +08:00
if (HDstrlen(buffer) > 4) {
HDfclose(file);
return -1;
}
if (HDsscanf(buffer, "%u", &nentries) != 1) {
HDfclose(file);
return -1;
}
/* ensure there are a sensible number of colors in the palette */
2020-09-30 22:27:10 +08:00
if ((nentries > 256) || (nentries > palette_size)) {
HDfclose(file);
2020-09-30 22:27:10 +08:00
return (-1);
}
/* read the palette entries */
2020-09-30 22:27:10 +08:00
for (u = 0; u < nentries; u++) {
/* extract the red, green and blue color components. */
if (HDfscanf(file, "%u %u %u", &red, &green, &blue) != 3) {
HDfclose(file);
return -1;
}
/* store this palette entry */
palette[u].r = (unsigned char)red;
palette[u].g = (unsigned char)green;
palette[u].b = (unsigned char)blue;
}
/* close file */
HDfclose(file);
return (int)nentries;
}