mirror of
https://github.com/HDFGroup/hdf5.git
synced 2025-03-31 17:10:47 +08:00
[svn-r15364] Convert the HDF5 8bit images to jpeg 24bit truecolor. Add an option for conversion from HDF5 indexed image to jpeg true color and an option to use a specific palette in the conversion.
tested: windows, linux
This commit is contained in:
Pedro Vicente Nunes
parent
ea27003154
commit
bca3b0e1fc
@ -40,14 +40,15 @@ const char *progname = "h52jpeg";
|
||||
int d_status = EXIT_SUCCESS;
|
||||
|
||||
/* command-line options: The user can specify short or long-named parameters */
|
||||
static const char *s_opts = "hVvi:t:c:";
|
||||
static const char *s_opts = "hVvi:t:cp:";
|
||||
static struct long_options l_opts[] = {
|
||||
{ "help", no_arg, 'h' },
|
||||
{ "version", no_arg, 'V' },
|
||||
{ "verbose", no_arg, 'v' },
|
||||
{ "image", require_arg, 'i' },
|
||||
{ "type", require_arg, 't' },
|
||||
{ "convert", require_arg, 'c' },
|
||||
{ "convert", no_arg, 'c' },
|
||||
{ "palette", require_arg, 'p' },
|
||||
{ NULL, 0, '\0' }
|
||||
};
|
||||
|
||||
@ -59,7 +60,8 @@ typedef struct
|
||||
const char *template_name;
|
||||
const char *image_name;
|
||||
int image_type;
|
||||
int convert_type;
|
||||
int convert_true;
|
||||
int idx_palette;
|
||||
int verbose;
|
||||
} h52jpeg_opt_t;
|
||||
|
||||
@ -68,8 +70,10 @@ typedef struct
|
||||
static void usage(const char *prog);
|
||||
static int h52jpeg(h52jpeg_opt_t opt);
|
||||
static void make_jpeg_name( const char* template_name, const char* image_name, char* jpeg_name);
|
||||
static int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name);
|
||||
static int do_object(hid_t fid, h52jpeg_opt_t opt, const char* image_name);
|
||||
static int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name, char* jpeg_name);
|
||||
static void write_JPEG_file(char *filename, JSAMPLE *image_buffer, int image_height, int image_width, int planes);
|
||||
static void convert_to_true( hsize_t width, hsize_t height, unsigned char* ibuf, unsigned char* pbuf, unsigned char* tbuf);
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* Function: main
|
||||
@ -86,12 +90,11 @@ int main(int argc, const char *argv[])
|
||||
{
|
||||
h52jpeg_opt_t opt;
|
||||
const char *image_type = NULL;
|
||||
const char *convert_type = NULL;
|
||||
int op;
|
||||
|
||||
/* initialze options to 0 */
|
||||
memset(&opt,0,sizeof(h52jpeg_opt_t));
|
||||
|
||||
|
||||
/* parse command line options */
|
||||
while ((op = get_option(argc, argv, s_opts, l_opts)) != EOF)
|
||||
{
|
||||
@ -110,8 +113,7 @@ int main(int argc, const char *argv[])
|
||||
opt.image_name = opt_arg;
|
||||
break;
|
||||
case 't':
|
||||
image_type = opt_arg;
|
||||
|
||||
image_type = opt_arg;
|
||||
|
||||
if ( HDstrcmp( image_type, "grey" ) == 0 )
|
||||
{
|
||||
@ -129,24 +131,12 @@ int main(int argc, const char *argv[])
|
||||
|
||||
break;
|
||||
case 'c':
|
||||
convert_type = opt_arg;
|
||||
|
||||
|
||||
if ( HDstrcmp( convert_type, "grey" ) == 0 )
|
||||
{
|
||||
opt.convert_type = 0;
|
||||
}
|
||||
else if ( HDstrcmp( convert_type, "true" ) == 0 )
|
||||
{
|
||||
opt.convert_type = 1;
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("<%s> is an invalid image type\n", image_type);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
opt.convert_true = 1;
|
||||
|
||||
break;
|
||||
case 'p':
|
||||
opt.idx_palette = atoi(opt_arg);
|
||||
break;
|
||||
|
||||
} /* switch */
|
||||
|
||||
@ -194,12 +184,13 @@ static void usage(const char *prog)
|
||||
printf(" -v, --verbose Verbose mode, print object information\n");
|
||||
printf(" -V, --version Print HDF5 version number and exit\n");
|
||||
printf(" -i, --image Image name (full path in HDF5 file)\n");
|
||||
printf(" -t T, --type=T Type of image to read (graycolor or truecolor)\n");
|
||||
printf(" -c T, --convert=T Convert image to type T (graycolor or truecolor)\n");
|
||||
printf(" -c, --convert Convert image from graycolor to truecolor\n");
|
||||
printf(" -p P, --palette=P Use HDF5 palette index P in conversion -c\n");
|
||||
|
||||
printf("\n");
|
||||
|
||||
printf(" T - is a string, either <grey> or <true>\n");
|
||||
printf(" P - is an integer, the palette index in the HDF5 image. Default is 0\n");
|
||||
|
||||
}
|
||||
|
||||
@ -231,20 +222,19 @@ static int h52jpeg(h52jpeg_opt_t opt)
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* image/dataset name was specified at command line
|
||||
* object name was specified at command line
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
if ( opt.image_name )
|
||||
{
|
||||
|
||||
/* read HDF5 image/dataset, save jpeg image */
|
||||
do_image(fid, opt, opt.image_name);
|
||||
/* read object, save jpeg image */
|
||||
do_object(fid, opt, opt.image_name);
|
||||
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* image name was not specified; traverse the file
|
||||
* object name was not specified; traverse the file
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
@ -277,8 +267,8 @@ static int h52jpeg(h52jpeg_opt_t opt)
|
||||
|
||||
case H5TRAV_TYPE_DATASET:
|
||||
|
||||
/* read HDF5 image/dataset, save jpeg image */
|
||||
do_image(fid, opt, travt->objs[i].name);
|
||||
/* read object, save jpeg image */
|
||||
do_object(fid, opt, travt->objs[i].name);
|
||||
|
||||
break; /* H5TRAV_TYPE_DATASET */
|
||||
|
||||
@ -290,7 +280,7 @@ static int h52jpeg(h52jpeg_opt_t opt)
|
||||
/* free table */
|
||||
trav_table_free(travt);
|
||||
|
||||
} /* image_name */
|
||||
} /* opt.image_name */
|
||||
|
||||
|
||||
/* close */
|
||||
@ -311,40 +301,29 @@ out:
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* Function: do_image
|
||||
* Function: do_object
|
||||
*
|
||||
* Parameters: HDF5 file id, command line options, an image name
|
||||
* Parameters: HDF5 file id, command line options, an object name
|
||||
*
|
||||
* Purpose: read HDF5 image/dataset, save jpeg image
|
||||
* Purpose: read HDF5 object, save jpeg image
|
||||
*
|
||||
* Return: 0, all is fine, -1 not all is fine
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
static
|
||||
int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name)
|
||||
int do_object(hid_t fid, h52jpeg_opt_t opt, const char* object_name)
|
||||
{
|
||||
hsize_t width;
|
||||
hsize_t height;
|
||||
hsize_t planes;
|
||||
char interlace[20];
|
||||
hssize_t npals;
|
||||
const char* name;
|
||||
int done;
|
||||
char jpeg_name[1024];
|
||||
|
||||
name = image_name;
|
||||
|
||||
int done=0; /* return value from do_image */
|
||||
char jpeg_name[1024];
|
||||
|
||||
/* build the jpeg file name */
|
||||
make_jpeg_name( opt.template_name, image_name, jpeg_name);
|
||||
|
||||
done = 0;
|
||||
|
||||
make_jpeg_name( opt.template_name, object_name, jpeg_name);
|
||||
|
||||
if ( opt.verbose)
|
||||
{
|
||||
printf("%s ...", name );
|
||||
printf("%s ...", object_name );
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
@ -352,35 +331,10 @@ int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name)
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
if ( H5IMis_image( fid, name ) )
|
||||
if ( H5IMis_image( fid, object_name ) )
|
||||
{
|
||||
|
||||
unsigned char* buf=NULL;
|
||||
|
||||
if ( H5IMget_image_info( fid, name, &width, &height, &planes, interlace, &npals ) < 0 )
|
||||
goto out;
|
||||
|
||||
if (NULL == (buf = HDmalloc( (size_t)width * (size_t)height * (size_t)planes )))
|
||||
goto out;
|
||||
|
||||
if ( H5IMread_image( fid, name, buf ) < 0 )
|
||||
goto out;
|
||||
|
||||
/* write the jpeg file */
|
||||
write_JPEG_file (jpeg_name,
|
||||
buf,
|
||||
(int) height,
|
||||
(int) width,
|
||||
(int) planes);
|
||||
|
||||
|
||||
free( buf );
|
||||
buf = NULL;
|
||||
|
||||
done = 1;
|
||||
|
||||
|
||||
|
||||
/* read image, save jpeg image */
|
||||
done = do_image(fid, opt, object_name, jpeg_name);
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
@ -388,7 +342,7 @@ int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name)
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
else if ( H5IMis_palette( fid, name ) )
|
||||
else if ( H5IMis_palette( fid, object_name ) )
|
||||
{
|
||||
|
||||
}
|
||||
@ -402,7 +356,6 @@ int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name)
|
||||
{
|
||||
|
||||
|
||||
|
||||
} /* else */
|
||||
|
||||
|
||||
@ -420,6 +373,163 @@ int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name)
|
||||
}
|
||||
|
||||
return 0;
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* Function: do_image
|
||||
*
|
||||
* Parameters: HDF5 file id, command line options, an image name
|
||||
*
|
||||
* Purpose: read HDF5 image, save jpeg image
|
||||
*
|
||||
* Return: 0, all is fine, -1 not all is fine
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
static
|
||||
int do_image(hid_t fid, h52jpeg_opt_t opt, const char* image_name, char* jpeg_name)
|
||||
{
|
||||
hsize_t width;
|
||||
hsize_t height;
|
||||
hsize_t planes;
|
||||
char interlace[20];
|
||||
hssize_t npals;
|
||||
const char* name;
|
||||
int done;
|
||||
unsigned char* ibuf=NULL;
|
||||
|
||||
name = image_name;
|
||||
|
||||
done = 0;
|
||||
|
||||
if ( H5IMget_image_info( fid, name, &width, &height, &planes, interlace, &npals ) < 0 )
|
||||
goto out;
|
||||
|
||||
if (NULL == (ibuf = HDmalloc( (size_t)width * (size_t)height * (size_t)planes )))
|
||||
goto out;
|
||||
|
||||
if ( H5IMread_image( fid, name, ibuf ) < 0 )
|
||||
{
|
||||
goto out;
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* no conversion to true color requested or true color image, just save what we found
|
||||
* this will result either in
|
||||
*
|
||||
* 24bit HDF5 ---> true color jpeg
|
||||
* 8bit HDF5 ---> grey color jpeg
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
if ( planes == 3 || !opt.convert_true )
|
||||
{
|
||||
/* write the jpeg file */
|
||||
write_JPEG_file (jpeg_name,
|
||||
ibuf,
|
||||
(int) height,
|
||||
(int) width,
|
||||
(int) planes);
|
||||
|
||||
done = 1;
|
||||
}
|
||||
/*-------------------------------------------------------------------------
|
||||
* conversion to truecolor
|
||||
* this will result in
|
||||
*
|
||||
* 8bit HDF5 ---> true color jpeg
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
else if (opt.convert_true && planes == 1 )
|
||||
{
|
||||
hssize_t npals; /* number of palettes */
|
||||
hsize_t pdims[2]; /* palette dimensions */
|
||||
unsigned char *pbuf=NULL;/* palette array */
|
||||
unsigned char *tbuf=NULL;/* true color array */
|
||||
int ipal; /* palette to use */
|
||||
|
||||
if ( H5IMget_npalettes( fid, name, &npals ) < 0 )
|
||||
{
|
||||
goto out;
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* there are palettes
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
if ( npals > 0 )
|
||||
{
|
||||
/* use either the default (0) palette or a requested one */
|
||||
ipal = ( opt.idx_palette > 0 ) ? opt.idx_palette : 0;
|
||||
|
||||
/* the requested palette may not exist . use the default */
|
||||
if ( opt.idx_palette >= npals )
|
||||
{
|
||||
ipal = 0;
|
||||
|
||||
if ( opt.verbose )
|
||||
{
|
||||
printf("palette index <%d> does not exist. Using default...",
|
||||
ipal );
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
if ( H5IMget_palette_info( fid, name, ipal, pdims ) < 0 )
|
||||
goto out;
|
||||
|
||||
if (NULL == (pbuf = HDmalloc( (size_t)pdims[0] * (size_t)pdims[1] )))
|
||||
goto out;
|
||||
|
||||
if (NULL == (tbuf = HDmalloc( (size_t)width * (size_t)height * 3 )))
|
||||
goto out;
|
||||
|
||||
if ( H5IMget_palette( fid, name, ipal, pbuf ) < 0 )
|
||||
goto out;
|
||||
|
||||
/* convert indexed image to true color image */
|
||||
convert_to_true(width, height, ibuf, pbuf, tbuf);
|
||||
|
||||
/* write the jpeg file */
|
||||
write_JPEG_file (jpeg_name,
|
||||
tbuf,
|
||||
(int) height,
|
||||
(int) width,
|
||||
3);
|
||||
|
||||
done = 1;
|
||||
|
||||
free( pbuf );
|
||||
free( tbuf );
|
||||
pbuf = NULL;
|
||||
tbuf = NULL;
|
||||
}
|
||||
/*-------------------------------------------------------------------------
|
||||
* there are no palettes
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
else
|
||||
{
|
||||
done = 0;
|
||||
if ( opt.verbose )
|
||||
{
|
||||
printf("image <%s> has no palette...", name );
|
||||
}
|
||||
} /* no palettes */
|
||||
|
||||
} /* conversion to truecolor */
|
||||
|
||||
free( ibuf );
|
||||
ibuf = NULL;
|
||||
|
||||
|
||||
|
||||
|
||||
return done;
|
||||
|
||||
out:
|
||||
|
||||
@ -470,6 +580,46 @@ void make_jpeg_name( const char* template_name, const char* image_name, char* jp
|
||||
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* Function: convert_to_true
|
||||
*
|
||||
* Parameters:
|
||||
*
|
||||
* Purpose: convert a greycolor buffer to a true color using a palette buffer
|
||||
*
|
||||
* Return:
|
||||
*
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
static
|
||||
void convert_to_true( hsize_t width, hsize_t height, unsigned char* ibuf, unsigned char* pbuf, unsigned char* tbuf)
|
||||
{
|
||||
hsize_t i, j;
|
||||
|
||||
for ( i = 0, j = 0; i < width * height; i++, j += 3)
|
||||
{
|
||||
unsigned char idx;
|
||||
unsigned char r;
|
||||
unsigned char g;
|
||||
unsigned char b;
|
||||
|
||||
/* get the index from the grey image */
|
||||
idx = ibuf[i];
|
||||
|
||||
/* get the RGB color */
|
||||
r = pbuf[3*idx];
|
||||
g = pbuf[3*idx+1];
|
||||
b = pbuf[3*idx+2];
|
||||
|
||||
/* define the color buffer */
|
||||
tbuf[j] = r;
|
||||
tbuf[j+1] = g;
|
||||
tbuf[j+2] = b;
|
||||
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
|
||||
@ -69,13 +69,22 @@ TOOLTEST()
|
||||
##############################################################################
|
||||
|
||||
|
||||
# Test for traversing the file and export all images/datasets to jpeg
|
||||
|
||||
# Test for traversing the file and export all images using defaults
|
||||
TESTING $TOOL $SRCFILE myjpeg1
|
||||
TOOLTEST $TESTFILE myjpeg1
|
||||
|
||||
# Test for reading an image path
|
||||
TESTING $TOOL -i image8bit $SRCFILE myjpeg2
|
||||
TOOLTEST -i image8bit $TESTFILE myjpeg2
|
||||
# Test for reading an image path
|
||||
TESTING $TOOL -i im8_1pal $SRCFILE myjpeg2
|
||||
TOOLTEST -i im8_1pal $TESTFILE myjpeg2
|
||||
|
||||
# Test for conversion HDF5 8bit ---> jpeg true, use the default palette
|
||||
TESTING $TOOL -c $SRCFILE myjpeg3
|
||||
TOOLTEST -c $TESTFILE myjpeg3
|
||||
|
||||
# Test for conversion HDF5 8bit ---> jpeg true, use a palette
|
||||
TESTING $TOOL -c -p 1 $SRCFILE myjpeg4
|
||||
TOOLTEST -c -p 1 $TESTFILE myjpeg4
|
||||
|
||||
|
||||
|
||||
|
||||
@ -16,11 +16,14 @@
|
||||
#include "hdf5.h"
|
||||
#include "hdf5_hl.h"
|
||||
#include "H5private.h"
|
||||
#include "hdf5_hl.h"
|
||||
#include "h52jpegtst.h"
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#define IM_1PAL "im8_1pal"
|
||||
#define IM_2PAL "im8_2pal"
|
||||
#define PAL1 "pal1"
|
||||
#define PAL2 "pal2"
|
||||
|
||||
static int make_images( hid_t fid );
|
||||
static int read_data(const char* fname, hsize_t *width, hsize_t *height );
|
||||
@ -72,6 +75,7 @@ out:
|
||||
|
||||
static int make_images( hid_t fid )
|
||||
{
|
||||
hid_t gid;
|
||||
hsize_t width; /* width of image */
|
||||
hsize_t height; /* height of image */
|
||||
hsize_t pal_dims[2] = {PAL_ENTRIES,3}; /* palette dimensions */
|
||||
@ -79,7 +83,7 @@ static int make_images( hid_t fid )
|
||||
int i, n;
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* indexed image
|
||||
* indexed image with 1 palette
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
@ -88,15 +92,30 @@ static int make_images( hid_t fid )
|
||||
goto out;
|
||||
|
||||
/* make the image */
|
||||
if ( H5IMmake_image_8bit( fid, "img8", width, height, gbuf ) < 0 )
|
||||
if ( H5IMmake_image_8bit( fid, IM_1PAL, width, height, gbuf ) < 0 )
|
||||
goto out;
|
||||
|
||||
/* make a palette */
|
||||
if ( H5IMmake_palette( fid, "pal1", pal_dims, pal_rgb ) < 0 )
|
||||
if ( H5IMmake_palette( fid, PAL1, pal_dims, pal_rgb ) < 0 )
|
||||
goto out;
|
||||
|
||||
/* attach the palette to the image */
|
||||
if ( H5IMlink_palette( fid, "img8", "pal1" ) < 0 )
|
||||
/* attach the 1st palette to the image */
|
||||
if ( H5IMlink_palette( fid, IM_1PAL, PAL1 ) < 0 )
|
||||
{
|
||||
goto out;
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* indexed image with 2 palettes
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
/* make the image */
|
||||
if ( H5IMmake_image_8bit( fid, IM_2PAL, width, height, gbuf ) < 0 )
|
||||
goto out;
|
||||
|
||||
/* attach the 1st palette to the image */
|
||||
if ( H5IMlink_palette( fid, IM_2PAL, PAL1 ) < 0 )
|
||||
{
|
||||
goto out;
|
||||
}
|
||||
@ -113,14 +132,29 @@ static int make_images( hid_t fid )
|
||||
}
|
||||
|
||||
/* save the palette */
|
||||
if ( H5IMmake_palette( fid, "pal2", pal_dims, pal ) < 0 )
|
||||
if ( H5IMmake_palette( fid, PAL2, pal_dims, pal ) < 0 )
|
||||
goto out;
|
||||
|
||||
/* attach the palette to the image */
|
||||
if ( H5IMlink_palette( fid, "img8", "pal2" ) < 0 )
|
||||
if ( H5IMlink_palette( fid, IM_2PAL, PAL2 ) < 0 )
|
||||
{
|
||||
goto out;
|
||||
}
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* make another image, in a group and no palette
|
||||
*-------------------------------------------------------------------------
|
||||
*/
|
||||
|
||||
if (( gid = H5Gcreate2(fid, "g1", H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT)) < 0 )
|
||||
goto out;
|
||||
|
||||
/* make the image with a name with "%", "@", "$", "/", ":", "&", and "*" */
|
||||
if ( H5IMmake_image_8bit( gid, "1%2@3$45:6&7*", width, height, gbuf ) < 0 )
|
||||
goto out;
|
||||
|
||||
H5Gclose(gid);
|
||||
|
||||
|
||||
/*-------------------------------------------------------------------------
|
||||
* true color image with pixel interlace in RGB type
|
||||
|
||||
Binary file not shown.
Loading…
x
Reference in New Issue
Block a user