missing source file from previous commit

This commit is contained in:
Enrico Fucile 2014-11-04 15:08:57 +00:00
parent f22153bd1a
commit 70f24489a4
3 changed files with 410 additions and 2 deletions

View File

@ -0,0 +1,408 @@
/*
* Copyright 2005-2014 ECMWF.
*
* 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.
*/
/*****************************************
* Enrico Fucile
****************************************/
#include "grib_api_internal.h"
#include <ctype.h>
/*
This is used by make_class.pl
START_CLASS_DEF
CLASS = accessor
SUPER = grib_accessor_class_gen
IMPLEMENTS = init;unpack_string;unpack_long;unpack_double
IMPLEMENTS = value_count; get_native_type;
MEMBERS = const char* dictionary
MEMBERS = const char* masterDir
MEMBERS = const char* localDir
END_CLASS_DEF
*/
/* START_CLASS_IMP */
/*
Don't edit anything between START_CLASS_IMP and END_CLASS_IMP
Instead edit values between START_CLASS_DEF and END_CLASS_DEF
or edit "accessor.class" and rerun ./make_class.pl
*/
static int get_native_type(grib_accessor*);
static int unpack_double(grib_accessor*, double* val,size_t *len);
static int unpack_long(grib_accessor*, long* val,size_t *len);
static int unpack_string (grib_accessor*, char*, size_t *len);
static int value_count(grib_accessor*,long*);
static void init(grib_accessor*,const long, grib_arguments* );
static void init_class(grib_accessor_class*);
typedef struct grib_accessor_bufr_elements_table {
grib_accessor att;
/* Members defined in gen */
/* Members defined in bufr_elements_table */
const char* dictionary;
const char* masterDir;
const char* localDir;
} grib_accessor_bufr_elements_table;
extern grib_accessor_class* grib_accessor_class_gen;
static grib_accessor_class _grib_accessor_class_bufr_elements_table = {
&grib_accessor_class_gen, /* super */
"bufr_elements_table", /* name */
sizeof(grib_accessor_bufr_elements_table), /* size */
0, /* inited */
&init_class, /* init_class */
&init, /* init */
0, /* post_init */
0, /* free mem */
0, /* describes himself */
0, /* get length of section */
0, /* get length of string */
&value_count, /* get number of values */
0, /* get number of bytes */
0, /* get offset to bytes */
&get_native_type, /* get native type */
0, /* get sub_section */
0, /* grib_pack procedures long */
0, /* grib_pack procedures long */
0, /* grib_pack procedures long */
&unpack_long, /* grib_unpack procedures long */
0, /* grib_pack procedures double */
&unpack_double, /* grib_unpack procedures double */
0, /* grib_pack procedures string */
&unpack_string, /* grib_unpack procedures string */
0, /* grib_pack array procedures string */
0, /* grib_unpack array procedures string */
0, /* grib_pack procedures bytes */
0, /* grib_unpack procedures bytes */
0, /* pack_expression */
0, /* notify_change */
0, /* update_size */
0, /* preferred_size */
0, /* resize */
0, /* nearest_smaller_value */
0, /* next accessor */
0, /* compare vs. another accessor */
0, /* unpack only ith value */
0, /* unpack a subarray */
0, /* clear */
};
grib_accessor_class* grib_accessor_class_bufr_elements_table = &_grib_accessor_class_bufr_elements_table;
static void init_class(grib_accessor_class* c)
{
c->dump = (*(c->super))->dump;
c->next_offset = (*(c->super))->next_offset;
c->string_length = (*(c->super))->string_length;
c->byte_count = (*(c->super))->byte_count;
c->byte_offset = (*(c->super))->byte_offset;
c->sub_section = (*(c->super))->sub_section;
c->pack_missing = (*(c->super))->pack_missing;
c->is_missing = (*(c->super))->is_missing;
c->pack_long = (*(c->super))->pack_long;
c->pack_double = (*(c->super))->pack_double;
c->pack_string = (*(c->super))->pack_string;
c->pack_string_array = (*(c->super))->pack_string_array;
c->unpack_string_array = (*(c->super))->unpack_string_array;
c->pack_bytes = (*(c->super))->pack_bytes;
c->unpack_bytes = (*(c->super))->unpack_bytes;
c->pack_expression = (*(c->super))->pack_expression;
c->notify_change = (*(c->super))->notify_change;
c->update_size = (*(c->super))->update_size;
c->preferred_size = (*(c->super))->preferred_size;
c->resize = (*(c->super))->resize;
c->nearest_smaller_value = (*(c->super))->nearest_smaller_value;
c->next = (*(c->super))->next;
c->compare = (*(c->super))->compare;
c->unpack_double_element = (*(c->super))->unpack_double_element;
c->unpack_double_subarray = (*(c->super))->unpack_double_subarray;
c->clear = (*(c->super))->clear;
}
/* END_CLASS_IMP */
static void init(grib_accessor* a, const long len, grib_arguments* params) {
int n=0;
grib_accessor_bufr_elements_table* self = (grib_accessor_bufr_elements_table*)a;
self->dictionary = grib_arguments_get_string(a->parent->h,params,n++);
self->masterDir = grib_arguments_get_name(a->parent->h,params,n++);
self->localDir = grib_arguments_get_name(a->parent->h,params,n++);
a->length = 0;
a->flags |= GRIB_ACCESSOR_FLAG_READ_ONLY;
}
char** str_split(char* a_str, const char a_delim)
{
char** result = 0;
size_t count = 0;
char* tmp = a_str;
char* last_comma = 0;
char delim[2];
delim[0] = a_delim;
delim[1] = 0;
/* Count how many elements will be extracted. */
while (*tmp)
{
if (a_delim == *tmp)
{
count++;
last_comma = tmp;
}
tmp++;
}
/* Add space for trailing token. */
count += last_comma < (a_str + strlen(a_str) - 1);
/* Add space for terminating null string so caller
knows where the list of returned strings ends. */
count++;
result = malloc(sizeof(char*) * count);
if (result)
{
size_t idx = 0;
char* token = strtok(a_str, delim);
while (token)
{
Assert(idx < count);
*(result + idx++) = strdup(token);
token = strtok(0, delim);
}
Assert(idx == count - 1);
*(result + idx) = 0;
}
return result;
}
static grib_trie* load_bufr_elements_table(grib_accessor* a, int* err) {
grib_accessor_bufr_elements_table* self = (grib_accessor_bufr_elements_table*)a;
char* filename=NULL;
char line[1024]={0,};
char masterDir[1024]={0,};
char localDir[1024]={0,};
char name[1024]={0,};
char localName[1024]={0,};
char recomposed[1024]={0,};
char localRecomposed[1024]={0,};
char dictName[1024]={0,};
char *localFilename=0;
char** list=0;
size_t len=1024;
grib_trie* dictionary=NULL;
FILE* f=NULL;
grib_handle* h=a->parent->h;
grib_context* c=a->parent->h->context;
*err=GRIB_SUCCESS;
len=1024;
if (self->masterDir != NULL) grib_get_string(h,self->masterDir,masterDir,&len);
len=1024;
if (self->localDir != NULL) grib_get_string(h,self->localDir,localDir,&len);
if (*masterDir!=0) {
sprintf(name,"%s/%s",masterDir,self->dictionary);
grib_recompose_name(h, NULL,name, recomposed,0);
filename=grib_context_full_defs_path(c,recomposed);
} else {
filename=grib_context_full_defs_path(c,self->dictionary);
}
if (*localDir!=0) {
sprintf(localName,"%s/%s",localDir,self->dictionary);
grib_recompose_name(h, NULL,localName, localRecomposed,0);
localFilename=grib_context_full_defs_path(c,localRecomposed);
sprintf(dictName,"%s:%s",localFilename,filename);
} else {
sprintf(dictName,"%s",filename);
}
if (!filename) {
grib_context_log(c,GRIB_LOG_ERROR,"unable to find def file %s",self->dictionary);
*err=GRIB_FILE_NOT_FOUND;
return NULL;
} else {
grib_context_log(c,GRIB_LOG_DEBUG,"found def file %s",filename);
}
dictionary=(grib_trie*)grib_trie_get(c->lists,dictName);
if (dictionary) {
grib_context_log(c,GRIB_LOG_DEBUG,"using dictionary %s from cache",self->dictionary);
return dictionary;
} else {
grib_context_log(c,GRIB_LOG_DEBUG,"using dictionary %s from file %s",self->dictionary,filename);
}
f=fopen(filename,"r");
if (!f) {*err=GRIB_IO_PROBLEM; return NULL;}
dictionary=grib_trie_new(c);
while(fgets(line,sizeof(line)-1,f)) {
list=str_split(line,'|');
grib_trie_insert(dictionary,list[0],list);
}
fclose(f);
if (localFilename!=0) {
f=fopen(localFilename,"r");
if (!f) {*err=GRIB_IO_PROBLEM; return NULL;}
while(fgets(line,sizeof(line)-1,f)) {
list=str_split(line,'|');
grib_trie_insert(dictionary,list[0],list);
}
fclose(f);
}
grib_trie_insert(c->lists,filename,dictionary);
return dictionary;
}
static int convert_type(char* stype) {
int ret=BUFR_DESCRIPTOR_TYPE_UNKNOWN;
switch (stype[0]) {
case 's':
if (!strcmp(stype,"string") ) ret=BUFR_DESCRIPTOR_TYPE_STRING;
break;
case 'l':
if (!strcmp(stype,"long") ) ret=BUFR_DESCRIPTOR_TYPE_LONG;
break;
case 'd':
if (!strcmp(stype,"double") ) ret=BUFR_DESCRIPTOR_TYPE_DOUBLE;
break;
case 't':
if (!strcmp(stype,"table") ) ret=BUFR_DESCRIPTOR_TYPE_TABLE;
break;
case 'f':
if (!strcmp(stype,"flag") ) ret=BUFR_DESCRIPTOR_TYPE_FLAG;
break;
default:
ret=BUFR_DESCRIPTOR_TYPE_UNKNOWN;
}
return ret;
}
static int bufr_get_from_table(grib_accessor* a,bufr_descriptor* v) {
int ret=0;
char** list=0;
char code[7]={0};
grib_context* c;
grib_trie* table;
table=load_bufr_elements_table(a,&ret);
if (ret) return ret;
c=a->parent->h->context;
sprintf(code,"%06ld",v->code);
list=(char**)grib_trie_get(table,code);
if (!list) return GRIB_NOT_FOUND;
v->shortName=grib_context_strdup(c,list[1]);
v->type=convert_type(list[2]);
v->name=grib_context_strdup(c,list[3]);
v->units=grib_context_strdup(c,list[4]);
v->scale=atol(grib_context_strdup(c,list[5]));
v->factor=grib_power(-v->scale,10);
v->reference=atol(grib_context_strdup(c,list[6]));
v->width=atol(grib_context_strdup(c,list[7]));
return ret;
}
bufr_descriptor* accessor_bufr_elements_table_get_descriptor(grib_accessor* a,int code,int *err) {
grib_context* c;
bufr_descriptor* v=NULL;
if (!a) return NULL;
c=a->parent->h->context;
v=(bufr_descriptor*)grib_context_malloc_clear(c,sizeof(bufr_descriptor));
if (!v) {
grib_context_log(c,GRIB_LOG_ERROR,
"grib_bufr_descriptor_new unable to allocate %d bytes\n",sizeof(bufr_descriptor));
*err=GRIB_OUT_OF_MEMORY;
return NULL;
}
v->context=c;
v->code=code;
v->F=code/100000;
v->X=(code-v->F*100000)/1000;
v->Y=(code-v->F*100000)%1000;
switch (v->F) {
case 0:
*err=bufr_get_from_table(a,v);
break;
case 1:
v->type=BUFR_DESCRIPTOR_TYPE_REPLICATION;
break;
case 2:
v->type=BUFR_DESCRIPTOR_TYPE_OPERATOR;
break;
case 3:
v->type=BUFR_DESCRIPTOR_TYPE_SEQUENCE;
break;
}
return v;
}
static int unpack_string (grib_accessor* a, char* buffer, size_t *len)
{
return GRIB_NOT_IMPLEMENTED;
}
static int value_count(grib_accessor* a,long* count)
{
*count=1;
return 0;
}
static int get_native_type(grib_accessor* a){
return GRIB_TYPE_STRING;
}
static int unpack_long (grib_accessor* a, long* val, size_t *len)
{
return GRIB_NOT_IMPLEMENTED;
}
static int unpack_double (grib_accessor* a, double* val, size_t *len)
{
return GRIB_NOT_IMPLEMENTED;
}

View File

@ -8,7 +8,7 @@
# virtue of its status as an intergovernmental organisation nor does it submit to any jurisdiction.
#
set -ex
set -e
. ./include.sh
REDIRECT=/dev/null

View File

@ -8,7 +8,7 @@
# virtue of its status as an intergovernmental organisation nor does it submit to any jurisdiction.
#
set -ex
set -e
. ./include.sh