eccodes/tools/grib_tools.c

1042 lines
30 KiB
C
Raw Normal View History

2013-03-25 12:04:10 +00:00
/*
2015-01-05 15:45:46 +00:00
* Copyright 2005-2015 ECMWF.
2013-03-25 12:04:10 +00:00
*
* This software is licensed under the terms of the Apache Licence Version 2.0
* which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
*
* In applying this licence, ECMWF does not waive the privileges and immunities granted to it by
* virtue of its status as an intergovernmental organisation nor does it submit to any jurisdiction.
*/
/*
* C Implementation: grib_tools
*
*/
#include "grib_tools.h"
#if HAVE_LIBJASPER
/* Remove compiler warnings re macros being redefined */
#undef PACKAGE_BUGREPORT
#undef PACKAGE_NAME
#undef PACKAGE_STRING
#undef PACKAGE_TARNAME
#undef PACKAGE_VERSION
2013-03-25 12:04:10 +00:00
#include "jasper/jasper.h"
#endif
#ifdef ENABLE_FLOATING_POINT_EXCEPTIONS
#define _GNU_SOURCE
#include <fenv.h>
int feenableexcept(int excepts);
#endif
2013-03-25 12:04:10 +00:00
GRIB_INLINE static int grib_inline_strcmp(const char* a,const char* b) {
if (*a != *b) return 1;
while((*a!=0 && *b!=0) && *(a) == *(b) ) {a++;b++;}
return (*a==0 && *b==0) ? 0 : 1;
2013-03-25 12:04:10 +00:00
}
static void grib_print_header(grib_runtime_options* options,grib_handle* h);
static void grib_tools_set_print_keys(grib_runtime_options* options,grib_handle* h,const char* ns);
static int grib_tool_with_orderby(grib_runtime_options* options);
static int grib_tool_without_orderby(grib_runtime_options* options);
static int grib_tool_onlyfiles(grib_runtime_options* options);
static int grib_tool_index(grib_runtime_options* options);
static int process(grib_context* c,grib_runtime_options* options,const char* path);
static int scan(grib_context* c,grib_runtime_options* options,const char* dir);
FILE* dump_file;
grib_runtime_options options={
0, /* verbose */
0, /* fail */
0, /* skip */
12, /* default_print_width */
0, /* print_header */
0, /* name_space */
0, /* print_number */
1, /* print_statistics */
{{0,},}, /* grib_values requested_print_keys[MAX_KEYS] */
0, /* requested_print_keys_count */
{{0,},}, /* grib_values print_keys[MAX_KEYS] */
0, /* print_keys_count */
0, /* strict */
0, /* multi_support */
0, /* set_values_count */
{{0,},}, /* grib_values set_values[MAX_KEYS] */
{{0,},}, /* grib_values constraints[MAX_KEYS] */
0, /* constraints_count */
{{0,},}, /* grib_values compare[MAX_KEYS] */
0, /* compare_count */
0, /* handle_count */
0, /* filter_handle_count */
0, /* file_count */
0, /* grib_tools_file infile_extra */
0, /* grib_tools_file current_infile */
0, /* grib_tools_file infile */
0, /*grib_tools_file outfile */
0, /* grib_action action */
0, /* grib_rule rules */
0, /* int dump_flags; */
0, /* char* dump_mode; */
0, /* repack */
0, /* error */
0, /* gts */
0, /* orderby */
0, /* latlon */
{0,},
{0,},
{0,},
{0,},
{0,},
4,
0,
-1,
{0,},
0, /* index */
0, /* index_on */
0, /* constant */
0, /* dump_filename*/
0, /* index */
0, /* random */
0, /* format */
0, /* onlyfiles */
0, /* tolerance_count */
0, /* through_index */
0, /* index1 */
0, /* index2 */
0, /* context */
0, /* stop */
0, /* mode */
2013-03-25 14:23:07 +00:00
0, /* headers_only */
2013-03-25 12:04:10 +00:00
0, /* skip_all */
{{0,},}, /* grib_values tolerance[MAX_KEYS] */
0 /* infile_offset */
2013-03-25 12:04:10 +00:00
};
static grib_handle* grib_handle_new_from_file_x(grib_context* c,FILE* f,int mode,int headers_only,int *err)
{
if (mode==MODE_ANY)
return any_new_from_file(c,f,err);
if (mode==MODE_GTS)
return gts_new_from_file(c,f,err);
2013-03-25 14:23:07 +00:00
if (mode==MODE_BUFR)
return bufr_new_from_file(c,f,err);
2013-03-25 12:04:10 +00:00
2014-07-07 20:55:19 +00:00
if (mode==MODE_METAR)
return metar_new_from_file(c,f,err);
2014-07-07 20:55:19 +00:00
if (mode==MODE_TAF)
return taf_new_from_file(c,f,err);
2014-07-07 20:55:19 +00:00
if (mode==MODE_GRIB)
return grib_new_from_file(c,f,headers_only,err);
Assert(!"grib_handle_new_from_file_x: unknown mode");
return NULL;
2013-03-25 12:04:10 +00:00
}
int grib_tool(int argc, char **argv)
{
int ret=0;
grib_context* c=grib_context_get_default();
options.context=c;
2013-03-25 12:04:10 +00:00
#ifdef ENABLE_FLOATING_POINT_EXCEPTIONS
feenableexcept(FE_ALL_EXCEPT & ~FE_INEXACT);
#endif
if (getenv("DOXYGEN_USAGE") && argc==1 ) usage_doxygen();
2013-03-25 12:04:10 +00:00
grib_get_runtime_options(argc,argv,&options);
2013-03-25 12:04:10 +00:00
grib_tool_before_getopt(&options);
grib_process_runtime_options(c,argc,argv,&options);
grib_tool_init(&options);
if (options.dump_filename) {
dump_file= fopen(options.dump_filename,"w");
if(!dump_file) {
perror(options.dump_filename);
exit(1);
}
} else {
dump_file=stdout;
}
if (is_index_file(options.infile->name) &&
( options.infile_extra && is_index_file(options.infile_extra->name))) {
options.through_index=1;
return grib_tool_index(&options);
}
2013-03-25 12:04:10 +00:00
if (options.onlyfiles)
ret=grib_tool_onlyfiles(&options);
else {
if (options.orderby)
ret=grib_tool_with_orderby(&options);
else
ret=grib_tool_without_orderby(&options);
}
if (options.dump_filename) fclose(dump_file);
return ret;
2013-03-25 12:04:10 +00:00
}
2013-03-25 12:04:10 +00:00
static int grib_tool_with_orderby(grib_runtime_options* options)
{
int err=0;
grib_failed *failed=NULL,*p=NULL;
grib_handle* h=NULL;
grib_context* c=NULL;
grib_tools_file* infile=NULL;
char** filenames;
size_t files_count=0;
grib_fieldset* set=NULL;
int i=0;
c=grib_context_get_default();
2013-03-25 12:04:10 +00:00
infile=options->infile;
if(infile) infile->failed=NULL;
2013-03-25 12:04:10 +00:00
files_count=0;
while(infile) {files_count++;infile=infile->next;}
2013-03-25 12:04:10 +00:00
filenames=(char**)grib_context_malloc_clear(c,files_count*sizeof(char*));
2013-03-25 12:04:10 +00:00
infile=options->infile;
for (i=0;i<files_count;i++) {filenames[i]=infile->name;infile=infile->next;}
2013-03-25 12:04:10 +00:00
if (grib_options_on("7")) c->no_fail_on_wrong_length=1;
2013-03-25 12:04:10 +00:00
set=grib_fieldset_new_from_files(0,filenames,files_count,0,0,0,options->orderby,&err);
if (err) {
grib_context_log(c,GRIB_LOG_FATAL,"unable to create index for input file %s (%s)",
filenames[0],grib_get_error_message(err));
exit(err);
}
2013-03-25 12:04:10 +00:00
options->handle_count=0;
while(!options->skip_all && ((h = grib_fieldset_next_handle(set,&err))
!= NULL || err != GRIB_SUCCESS )) {
options->handle_count++;
options->error=err;
if (!h) {
grib_no_handle_action(err);
failed=(grib_failed*)grib_context_malloc_clear(c,sizeof(grib_failed));
failed->count=infile->handle_count;
failed->error=err;
failed->next=NULL;
if (!infile->failed) {
infile->failed=failed;
} else {
p=infile->failed;
while (p->next) p=p->next;
p->next=failed;
}
2013-03-25 12:04:10 +00:00
continue;
}
2013-03-25 12:04:10 +00:00
grib_print_header(options,h);
2013-03-25 12:04:10 +00:00
grib_skip_check(options,h);
2013-03-25 12:04:10 +00:00
if (options->skip && options->strict) {
grib_tool_skip_handle(options,h);
continue;
}
2013-03-25 12:04:10 +00:00
grib_tool_new_handle_action(options,h);
2013-03-25 12:04:10 +00:00
grib_tool_print_key_values(options,h);
2013-03-25 12:04:10 +00:00
grib_handle_delete(h);
}
2013-03-25 12:04:10 +00:00
grib_tool_finalise_action(options);
2013-03-25 12:04:10 +00:00
return 0;
2013-03-25 12:04:10 +00:00
}
char iobuf[1024*1024];
static int grib_tool_without_orderby(grib_runtime_options* options)
{
int err=0;
int nofail=0;
grib_failed *failed=NULL,*p=NULL;
grib_handle* h=NULL;
grib_context* c=NULL;
grib_tools_file* infile=NULL;
c=grib_context_get_default();
options->file_count=0;
options->handle_count=0;
options->filter_handle_count=0;
options->current_infile=options->infile;
infile=options->infile;
2013-03-25 12:04:10 +00:00
infile->failed=NULL;
if (grib_options_on("7")) c->no_fail_on_wrong_length=1;
2013-03-25 12:04:10 +00:00
while (infile!=NULL && infile->name!=NULL) {
2013-03-25 12:04:10 +00:00
if (options->print_statistics && options->verbose) fprintf(dump_file,"%s\n",infile->name);
infile->file = fopen(infile->name,"r");
if(!infile->file) {
perror(infile->name);
exit(1);
}
if (options->infile_offset) {
2015-02-18 17:42:02 +00:00
#ifndef ECCODES_ON_WINDOWS
/* Check at compile time to ensure our file offset is at least 64 bits */
COMPILE_TIME_ASSERT( sizeof(options->infile_offset) >= 8 );
#endif
err=fseeko(infile->file, options->infile_offset, SEEK_SET);
if (err) {
/*fprintf(stderr, "Invalid file offset: %ld\n", options->infile_offset);*/
perror("Invalid file offset");
exit(1);
}
}
2013-03-25 12:04:10 +00:00
setvbuf(infile->file,iobuf,_IOFBF,sizeof(iobuf));
2013-03-25 12:04:10 +00:00
options->file_count++;
infile->handle_count=0;
infile->filter_handle_count=0;
2013-03-25 12:04:10 +00:00
grib_tool_new_file_action(options,infile);
nofail=grib_options_on("f");
2013-03-25 12:04:10 +00:00
while(!options->skip_all && ((h = grib_handle_new_from_file_x(c,infile->file,options->mode,
options->headers_only,&err))
!= NULL || err != GRIB_SUCCESS ) ) {
infile->handle_count++;
options->handle_count++;
options->error=err;
2013-03-25 12:04:10 +00:00
if (!h) {
/* fprintf(dump_file,"\t\t\"ERROR: unreadable message\"\n"); */
grib_no_handle_action(err);
2013-03-25 12:04:10 +00:00
failed=(grib_failed*)grib_context_malloc_clear(c,sizeof(grib_failed));
failed->count=infile->handle_count;
failed->error=err;
failed->next=NULL;
2013-03-25 12:04:10 +00:00
if (!infile->failed) {
infile->failed=failed;
} else {
p=infile->failed;
while (p->next) p=p->next;
p->next=failed;
}
2013-03-25 12:04:10 +00:00
continue;
}
2013-03-25 12:04:10 +00:00
grib_print_header(options,h);
2013-03-25 12:04:10 +00:00
grib_skip_check(options,h);
2013-03-25 12:04:10 +00:00
if (options->skip && options->strict) {
grib_tool_skip_handle(options,h);
continue;
}
2013-03-25 12:04:10 +00:00
grib_tool_new_handle_action(options,h);
2013-03-25 12:04:10 +00:00
grib_print_key_values(options,h);
2013-03-25 12:04:10 +00:00
grib_handle_delete(h);
}
2013-03-25 12:04:10 +00:00
grib_print_file_statistics(options,infile);
2013-03-25 12:04:10 +00:00
if (infile->file) fclose(infile->file);
2013-03-25 12:04:10 +00:00
if (infile->handle_count==0) {
2015-01-30 18:30:48 +00:00
fprintf(dump_file,"no messages found in %s\n", infile->name);
if (options->fail) exit(1);
}
2013-03-25 12:04:10 +00:00
infile=infile->next;
options->current_infile=infile;
}
2013-03-25 12:04:10 +00:00
grib_print_full_statistics(options);
2013-03-25 12:04:10 +00:00
grib_tool_finalise_action(options);
2013-03-25 12:04:10 +00:00
return options->error;
}
2013-03-25 12:04:10 +00:00
static int navigate(grib_field_tree* fields,grib_runtime_options* options)
{
int err=0;
2014-07-07 20:55:19 +00:00
int message_type=0;
if (!fields || options->stop) return 0;
2014-07-07 20:55:19 +00:00
switch (options->mode) {
case MODE_GRIB:
message_type=CODES_GRIB;
break;
case MODE_BUFR:
message_type=CODES_BUFR;
break;
default :
Assert(0);
}
if (fields->field) {
2014-07-07 20:55:19 +00:00
grib_handle* h=codes_index_get_handle(fields->field,message_type,&err);
if (!options->index2->current)
options->index2->current=(grib_field_list*)grib_context_malloc_clear(options->context,sizeof(grib_field_list));
options->index2->current->field=fields->field;
if (!h) return err;
grib_skip_check(options,h);
if (options->skip && options->strict) {
grib_tool_skip_handle(options,h);
} else {
grib_tool_new_handle_action(options,h);
grib_handle_delete(h);
}
}
2013-03-25 12:04:10 +00:00
err=navigate(fields->next_level,options);
if (err) return err;
2013-03-25 12:04:10 +00:00
err=navigate(fields->next,options);
2013-03-25 12:04:10 +00:00
return err;
2013-03-25 12:04:10 +00:00
}
static int grib_tool_index(grib_runtime_options* options)
{
int err=0;
grib_context* c=NULL;
char* f1=options->infile->name;
char* f2=options->infile_extra->name;
grib_index_key *k1,*k2;
int found=0;
2013-03-25 12:04:10 +00:00
c=grib_context_get_default();
2013-03-25 12:04:10 +00:00
options->index1=grib_index_read(c,f1,&err);
if (err)
grib_context_log(c,(GRIB_LOG_FATAL) | (GRIB_LOG_PERROR) ,
"unable to read index from %s",f1);
options->index2=grib_index_read(c,f2,&err);
if (err)
grib_context_log(c,(GRIB_LOG_FATAL) | (GRIB_LOG_PERROR) ,
"unable to read index from %s",f2);
k1=options->index1->keys;
while ( k1 ) {
k2=options->index2->keys;
found=0;
while (k2) {
if ( !strcmp(k1->name,k2->name) ) {
found=1;
break;
}
k2=k2->next;
}
if (!found) {
printf("Indexes contained in the input files have different keys\n");
printf("keys in file %s:\n",f1);
k1=options->index1->keys;
while (k1) {
printf("\t%s\n",k1->name);
k1=k1->next;
}
printf("keys in file %s:\n",f2);
k2=options->index2->keys;
while (k2) {
printf("\t%s\n",k2->name);
k2=k2->next;
}
exit(1);
}
k1->value[0]=0;
k1=k1->next;
}
k2=options->index2->keys;
while ( k2 ) {
k1=options->index1->keys;
found=0;
while (k1) {
if ( !strcmp(k1->name,k2->name) ) {
found=1;
break;
}
k1=k1->next;
}
if (!found) {
printf("Indexes contained in the input files have different keys\n");
printf("keys in file %s:\n",f2);
k2=options->index2->keys;
while (k2) {
printf("\t%s\n",k2->name);
k2=k2->next;
}
printf("keys in file %s:\n",f1);
k1=options->index1->keys;
while (k1) {
printf("\t%s\n",k1->name);
k1=k1->next;
}
exit(1);
}
k2=k2->next;
}
navigate(options->index2->fields,options);
grib_tool_finalise_action(options);
return 0;
2013-03-25 12:04:10 +00:00
}
2015-02-18 17:42:02 +00:00
#ifndef ECCODES_ON_WINDOWS
static int scan(grib_context* c,grib_runtime_options* options,const char* dir)
{
struct dirent *s;
DIR *d;
int err=0;
d= opendir(dir);
if (!d) {
grib_context_log(c,(GRIB_LOG_ERROR) | (GRIB_LOG_PERROR) , "opendir %s",dir);
return GRIB_IO_PROBLEM;
}
while ((s=readdir(d)) && (err==0)) {
if(strcmp(s->d_name,".") != 0 && strcmp(s->d_name,"..") != 0) {
char buf[1024];
sprintf(buf,"%s/%s",dir,s->d_name);
process(c,options,buf);
}
}
return 0;
2013-03-25 12:04:10 +00:00
}
2013-03-25 14:23:07 +00:00
#else
static int isWinDir(const struct _finddata_t *fileinfo)
{
if((fileinfo->attrib & 16) == 16)
return 1;
return 0;
2013-03-25 14:23:07 +00:00
}
static void doProcessing(grib_context* c,grib_runtime_options* options,const char* dir, const struct _finddata_t *fileinfo)
{
if(isWinDir(fileinfo))
{
if(strcmp(fileinfo->name, ".") != 0 && strcmp(fileinfo->name,"..") != 0) {
char buf[1024];
sprintf(buf,"%s/%s",dir,fileinfo->name);
process(c,options,buf);
}
}
2013-03-25 14:23:07 +00:00
}
static int scan(grib_context* c,grib_runtime_options* options,const char* dir) {
struct _finddata_t fileinfo;
intptr_t handle;
if((handle = _findfirst(dir, &fileinfo)) != -1)
2013-03-25 14:23:07 +00:00
doProcessing(c, options, dir, &fileinfo);
else
2013-03-25 14:23:07 +00:00
{
grib_context_log(c,(GRIB_LOG_ERROR) | (GRIB_LOG_PERROR) , "opendir %s",dir);
return GRIB_IO_PROBLEM;
}
while(!_findnext(handle, &fileinfo))
2013-03-25 14:23:07 +00:00
doProcessing(c, options, dir, &fileinfo);
return 0;
}
#endif
2013-03-25 12:04:10 +00:00
static int process(grib_context* c,grib_runtime_options* options,const char* path) {
struct stat s;
int stat_val=0;
int ioerr=0;
2013-03-25 14:23:07 +00:00
2015-02-18 17:42:02 +00:00
#ifndef ECCODES_ON_WINDOWS
stat_val = lstat(path,&s);
2013-03-25 14:23:07 +00:00
#else
stat_val = stat(path,&s);
2013-03-25 14:23:07 +00:00
#endif
if ( stat_val != 0 ) {
ioerr=errno;
grib_context_log(c,(GRIB_LOG_ERROR) | (GRIB_LOG_PERROR),"Cannot stat %s",path);
return GRIB_IO_PROBLEM;
}
2013-03-25 12:04:10 +00:00
if (S_ISDIR(s.st_mode) && !S_ISLNK(s.st_mode)) {
scan(c,options,path);
} else {
grib_tool_new_filename_action(options,path);
}
return 0;
2013-03-25 12:04:10 +00:00
}
static int grib_tool_onlyfiles(grib_runtime_options* options)
{
grib_context* c=NULL;
grib_tools_file* infile=NULL;
2013-03-25 12:04:10 +00:00
c=grib_context_get_default();
infile=options->infile;
2013-03-25 12:04:10 +00:00
while (infile!=NULL && infile->name!=NULL) {
2013-03-25 12:04:10 +00:00
process(c,options,infile->name);
2013-03-25 12:04:10 +00:00
infile=infile->next;
}
2013-03-25 12:04:10 +00:00
grib_tool_finalise_action(options);
2013-03-25 12:04:10 +00:00
return 0;
2013-03-25 12:04:10 +00:00
}
static void grib_print_header(grib_runtime_options* options,grib_handle* h)
{
size_t strlenkey=0;
int width;
int written_to_dump = 0; /* boolean */
if (!options->print_keys || options->handle_count!=1)
return;
2013-03-25 12:04:10 +00:00
grib_tools_set_print_keys(options,h,options->name_space);
if (options->print_keys
&& options->verbose
&& options->print_header) {
int j=0;
for (j=0;j<options->print_keys_count;j++) {
strlenkey=strlen(options->print_keys[j].name);
width= strlenkey < options->default_print_width ?
options->default_print_width+2 : strlenkey+2;
2013-03-25 12:04:10 +00:00
if (options->default_print_width < 0)
width=strlenkey+1;
fprintf(dump_file,"%-*s",(int)width,options->print_keys[j].name);
written_to_dump = 1;
}
if (options->latlon) {
if (options->latlon_mode==4) {
fprintf(dump_file," value1 ");
fprintf(dump_file," value2 ");
fprintf(dump_file," value3 ");
fprintf(dump_file," value4 ");
} else {
fprintf(dump_file," value ");
}
written_to_dump = 1;
}
if (options->index_on) {
fprintf(dump_file," value(%d) ",(int)options->index);
written_to_dump = 1;
}
if (written_to_dump) {
fprintf(dump_file,"\n");
}
}
2013-03-25 12:04:10 +00:00
}
static void grib_tools_set_print_keys(grib_runtime_options* options, grib_handle* h, const char* ns)
{
int i=0;
grib_keys_iterator* kiter=NULL;
2013-03-25 12:04:10 +00:00
options->print_keys_count=0;
2013-03-25 12:04:10 +00:00
for (i=0;i<options->requested_print_keys_count;i++) {
options->print_keys[options->print_keys_count].name=options->requested_print_keys[i].name;
if (strlen(options->requested_print_keys[i].name)>options->default_print_width)
options->default_print_width=strlen(options->requested_print_keys[i].name);
options->print_keys[options->print_keys_count].type=options->requested_print_keys[i].type;
options->print_keys_count++;
2013-03-25 12:04:10 +00:00
}
2013-03-25 12:04:10 +00:00
if (ns) {
kiter=grib_keys_iterator_new(h,0,(char*)ns);
if (!kiter) {
fprintf(dump_file,"ERROR: Unable to create keys iterator\n");
exit(1);
}
2013-03-25 12:04:10 +00:00
while(grib_keys_iterator_next(kiter))
{
const char* name = grib_keys_iterator_get_name(kiter);
2013-03-25 12:04:10 +00:00
if (options->print_keys_count >= MAX_KEYS ) {
fprintf(stderr,"ERROR: keys list too long (more than %d keys)\n",
options->print_keys_count);
exit(1);
}
options->print_keys[options->print_keys_count].name=strdup(name);
if (strlen(name)>options->default_print_width)
options->default_print_width=strlen(name);
options->print_keys[options->print_keys_count].type=GRIB_TYPE_STRING;
options->print_keys_count++;
2013-03-25 12:04:10 +00:00
}
grib_keys_iterator_delete(kiter);
if (options->print_keys_count==0 && options->latlon == 0 ) {
printf("ERROR: namespace \"%s\" does not contain any key\n",ns);
exit(1);
}
2013-03-25 12:04:10 +00:00
}
}
static int to_skip(grib_handle* h,grib_values* v,int *err)
{
double dvalue=0;
int ret=0;
long lvalue=0;
char value[MAX_STRING_LEN]={0,};
size_t len=MAX_STRING_LEN;
*err=0;
switch (v->type) {
2013-03-25 14:23:07 +00:00
case GRIB_TYPE_STRING:
*err=grib_get_string( h,v->name,value,&len);
ret = v->equal ? grib_inline_strcmp(value,v->string_value) : !grib_inline_strcmp(value,v->string_value);
break;
2013-03-25 14:23:07 +00:00
case GRIB_TYPE_DOUBLE:
*err=grib_get_double( h,v->name,&dvalue);
ret = v->equal ? (dvalue != v->double_value) : (dvalue == v->double_value);
break;
2013-03-25 14:23:07 +00:00
case GRIB_TYPE_LONG:
*err=grib_get_long( h,v->name,&lvalue);
ret = v->equal ? (lvalue != v->long_value) : (lvalue == v->long_value);
break;
2013-03-25 14:23:07 +00:00
case GRIB_TYPE_MISSING:
lvalue=grib_is_missing( h,v->name,err);
ret = (lvalue == v->equal) ? 0 : 1;
break;
default:
fprintf(dump_file,"invalid type for %s\n",v->name);
exit(1);
}
2013-03-25 12:04:10 +00:00
return ret;
2013-03-25 12:04:10 +00:00
}
void grib_skip_check(grib_runtime_options* options,grib_handle* h)
{
int i,ret=0;
2013-03-25 14:23:07 +00:00
grib_values* v=NULL;
for (i=0;i < options->constraints_count ;i++) {
v=&(options->constraints[i]);
if (v->equal) {
options->skip=1;
while (v) {
if (!to_skip(h,v,&ret)) {
if (!strcmp(v->name,"count") && !v->next)
options->skip_all=1;
options->skip=0;
break;
}
if (ret != GRIB_SUCCESS && options->fail) {
grib_context_log(h->context,GRIB_LOG_ERROR,"unable to get \"%s\" (%s)",
v->name,grib_get_error_message(ret));
exit(ret);
}
v=v->next;
2013-03-25 14:23:07 +00:00
}
} else {
options->skip=0;
while (v) {
if (to_skip(h,v,&ret)) {
options->skip=1;
break;
}
if (ret != GRIB_SUCCESS && options->fail) {
grib_context_log(h->context,GRIB_LOG_ERROR,"unable to get \"%s\" (%s)",
v->name,grib_get_error_message(ret));
exit(ret);
}
v=v->next;
2013-03-25 14:23:07 +00:00
}
}
if (options->skip==1)
break;
}
if (!options->skip) {
options->filter_handle_count++;
if (options->current_infile)
options->current_infile->filter_handle_count++;
}
2013-03-25 12:04:10 +00:00
}
void grib_print_key_values(grib_runtime_options* options,grib_handle* h)
{
int i=0;
int ret=0,width=0;
int strlenvalue=0;
double dvalue=0;
long lvalue=0;
char value[MAX_STRING_LEN];
char* notfound="not_found";
int written_to_dump = 0; /* boolean */
if (!options->verbose) return;
for (i=0;i<options->print_keys_count;i++) {
size_t len=MAX_STRING_LEN;
ret=GRIB_SUCCESS;
if (grib_is_missing(h,options->print_keys[i].name,&ret) && ret==GRIB_SUCCESS)
sprintf(value,"MISSING");
else if ( ret == GRIB_SUCCESS ) {
if (options->print_keys[i].type == GRIB_TYPE_UNDEFINED)
grib_get_native_type(h,options->print_keys[i].name,&(options->print_keys[i].type));
switch (options->print_keys[i].type) {
case GRIB_TYPE_STRING:
ret=grib_get_string( h,options->print_keys[i].name,value,&len);
break;
case GRIB_TYPE_DOUBLE:
ret=grib_get_double( h,options->print_keys[i].name,&dvalue);
sprintf(value,options->format,dvalue);
break;
case GRIB_TYPE_LONG:
ret=grib_get_long( h,options->print_keys[i].name,&lvalue);
sprintf(value,"%ld", lvalue);
break;
case GRIB_TYPE_BYTES:
ret=grib_get_string( h,options->print_keys[i].name,value,&len);
break;
default:
fprintf(dump_file,"invalid format option for %s\n",options->print_keys[i].name);
exit(1);
}
}
if (ret != GRIB_SUCCESS) {
if (options->fail) GRIB_CHECK_NOLINE(ret,options->print_keys[i].name);
if (ret == GRIB_NOT_FOUND) strcpy(value,notfound);
else {
fprintf(dump_file,"%s %s\n",grib_get_error_message(ret),options->print_keys[i].name);
exit(ret);
}
}
strlenvalue = strlen(value);
width = strlenvalue < options->default_print_width ?
options->default_print_width + 2 :
strlenvalue + 2;
if (options->default_print_width < 0) width = strlenvalue + 1;
if (options->print_keys_count==i+1 && options->latlon==0) width--;
fprintf(dump_file,"%-*s",(int)width,value);
written_to_dump = 1;
}
if (options->latlon) {
2013-03-25 12:04:10 +00:00
if (options->latlon_mode==4){
int i=0;
for (i=0;i<4;i++) {
fprintf(dump_file,options->format,options->values[i]);
fprintf(dump_file," ");
}
written_to_dump=1;
} else if (options->latlon_mode==1) {
sprintf(value,options->format,options->values[options->latlon_idx]);
strlenvalue = strlen(value);
width = strlenvalue < options->default_print_width ?
options->default_print_width + 2 :
strlenvalue + 2;
fprintf(dump_file,"%-*s",(int)width,value);
written_to_dump=1;
}
}
if (options->index_on) {
double v=0;
/*if (grib_get_double_element(h,"values",options->index,&v) != GRIB_SUCCESS) {*/
if (1) {
size_t size, the_index = 0;
double* values;
int err=0;
err=grib_get_size(h,"values",&size);
if (err) {
sprintf(value,"unknown");
if (!options->fail) exit(err);
return;
}
values=(double*)grib_context_malloc_clear(h->context,size*sizeof(double));
grib_get_double_array(h,"values",values,&size);
the_index = options->index;
if (the_index >= size) {
fprintf(dump_file,"\n");
fprintf(stderr, "invalid index value %d (should be between 0 and %ld)\n", options->index, size-1);
exit(1);
}
v=values[options->index];
grib_context_free(h->context,values);
}
sprintf(value,options->format,v);
strlenvalue = strlen(value);
width = strlenvalue < options->default_print_width ?
options->default_print_width + 2 :
strlenvalue + 2;
fprintf(dump_file,"%-*s",(int)width,value);
written_to_dump=1;
}
if (written_to_dump) {
fprintf(dump_file,"\n");
}
2013-03-25 12:04:10 +00:00
}
void grib_print_file_statistics(grib_runtime_options* options,grib_tools_file* file)
{
grib_failed* failed=NULL;
2013-03-25 12:04:10 +00:00
Assert(file);
2013-03-25 12:04:10 +00:00
failed=file->failed;
2013-03-25 12:04:10 +00:00
if (!options->print_statistics || !options->verbose) return;
2013-03-25 12:04:10 +00:00
2015-01-30 18:30:48 +00:00
fprintf(dump_file,"%d of %d messages in %s\n\n",
file->filter_handle_count,
file->handle_count,
file->name);
if (!failed) return;
/*
2015-01-30 18:30:48 +00:00
fprintf(dump_file,"Following bad messages found in %s\n",
2013-03-25 12:04:10 +00:00
file->name);
fprintf(dump_file,"N Error\n");
while (failed){
fprintf(dump_file,"%-*d %s\n",
7,failed->count,
grib_get_error_message(failed->error));
failed=failed->next;
}
fprintf(dump_file,"\n");
*/
}
void grib_print_full_statistics(grib_runtime_options* options)
{
if (options->print_statistics && options->verbose)
2015-01-30 18:30:48 +00:00
fprintf(dump_file,"%d of %d total messages in %d files\n",
options->filter_handle_count,options->handle_count,options->file_count);
2013-03-25 12:04:10 +00:00
}
void grib_tools_write_message(grib_runtime_options* options, grib_handle* h)
{
const void *buffer;
size_t size;
grib_file* of = NULL;
int err = 0;
int ioerr = 0;
char filename[1024] = { 0, };
Assert(options->outfile!=NULL && options->outfile->name!=NULL);
if (options->error == GRIB_WRONG_LENGTH)
return;
if ((err = grib_get_message(h, &buffer, &size)) != GRIB_SUCCESS) {
grib_context_log(h->context, GRIB_LOG_ERROR,"unable to get binary message\n");
exit(err);
}
err = grib_recompose_name(h, NULL, options->outfile->name, filename, 0);
of = grib_file_open(filename, "w", &err);
if (!of || !of->handle) {
ioerr = errno;
grib_context_log(h->context, (GRIB_LOG_ERROR) | (GRIB_LOG_PERROR),
"unable to open file %s\n", filename);
exit(GRIB_IO_PROBLEM);
}
if (options->gts && h->gts_header) {
if (fwrite(h->gts_header, 1, h->gts_header_len, of->handle) != h->gts_header_len) {
ioerr = errno;
grib_context_log(h->context, (GRIB_LOG_ERROR) | (GRIB_LOG_PERROR),
"Error writing GTS header to %s", filename);
exit(GRIB_IO_PROBLEM);
}
}
if (fwrite(buffer, 1, size, of->handle) != size) {
ioerr = errno;
grib_context_log(h->context, (GRIB_LOG_ERROR) | (GRIB_LOG_PERROR),
"Error writing to %s", filename);
exit(GRIB_IO_PROBLEM);
}
if (options->gts && h->gts_header) {
char gts_trailer[4] = { '\x0D', '\x0D', '\x0A', '\x03' };
if (fwrite(gts_trailer, 1, 4, of->handle) != 4) {
ioerr = errno;
grib_context_log(h->context, (GRIB_LOG_ERROR) | (GRIB_LOG_PERROR),
"Error writing GTS trailer to %s", filename);
exit(GRIB_IO_PROBLEM);
}
}
grib_file_close(filename, &err);
if (err != GRIB_SUCCESS) {
grib_context_log(h->context, GRIB_LOG_ERROR,"unable to write message\n");
exit(err);
}
options->outfile->file = NULL;
2013-03-25 12:04:10 +00:00
#if 0
if (!options->outfile->file) {
options->outfile->file = fopen(options->outfile->name,"w");
if(!options->outfile->file) {
perror(options->outfile->name);
exit(1);
}
}
GRIB_CHECK_NOLINE(grib_get_message(h,&buffer,&size),0);
if (options->gts && h->gts_header)
fwrite(h->gts_header,1,h->gts_header_len,options->outfile->file);
2013-03-25 12:04:10 +00:00
if(fwrite(buffer,1,size,options->outfile->file) != size)
{
perror(options->outfile->name);
exit(1);
}
2013-03-25 12:04:10 +00:00
if (options->gts && h->gts_header) {
char gts_trailer[4]={'\x0D','\x0D','\x0A','\x03'};
fwrite(gts_trailer,1,4,options->outfile->file);
}
#endif
2013-03-25 12:04:10 +00:00
}