mirror of https://github.com/ecmwf/eccodes.git
956 lines
30 KiB
C++
956 lines
30 KiB
C++
/*
|
|
* (C) Copyright 2005- 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 <cctype>
|
|
|
|
#if GRIB_PTHREADS
|
|
static pthread_once_t once = PTHREAD_ONCE_INIT;
|
|
static pthread_mutex_t mutex1 = PTHREAD_MUTEX_INITIALIZER;
|
|
|
|
static void thread_init()
|
|
{
|
|
pthread_mutexattr_t attr;
|
|
pthread_mutexattr_init(&attr);
|
|
pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
|
|
pthread_mutex_init(&mutex1, &attr);
|
|
pthread_mutexattr_destroy(&attr);
|
|
}
|
|
#elif GRIB_OMP_THREADS
|
|
static int once = 0;
|
|
static omp_nest_lock_t mutex1;
|
|
|
|
static void thread_init()
|
|
{
|
|
GRIB_OMP_CRITICAL(lock_grib_accessor_class_codetable_c)
|
|
{
|
|
if (once == 0) {
|
|
omp_init_nest_lock(&mutex1);
|
|
once = 1;
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
/*
|
|
This is used by make_class.pl
|
|
|
|
START_CLASS_DEF
|
|
CLASS = accessor
|
|
SUPER = grib_accessor_class_unsigned
|
|
IMPLEMENTS = init;dump;unpack_string;pack_expression;unpack_long
|
|
IMPLEMENTS = value_count;pack_string; destroy; get_native_type;pack_missing
|
|
MEMBERS = const char* tablename
|
|
MEMBERS = const char* masterDir
|
|
MEMBERS = const char* localDir
|
|
MEMBERS = grib_codetable* table
|
|
MEMBERS = int table_loaded
|
|
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 pack_missing(grib_accessor*);
|
|
static int pack_string(grib_accessor*, const char*, size_t* len);
|
|
static int pack_expression(grib_accessor*, grib_expression*);
|
|
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 destroy(grib_context*, grib_accessor*);
|
|
static void dump(grib_accessor*, grib_dumper*);
|
|
static void init(grib_accessor*, const long, grib_arguments*);
|
|
|
|
typedef struct grib_accessor_codetable
|
|
{
|
|
grib_accessor att;
|
|
/* Members defined in gen */
|
|
/* Members defined in long */
|
|
/* Members defined in unsigned */
|
|
long nbytes;
|
|
grib_arguments* arg;
|
|
/* Members defined in codetable */
|
|
const char* tablename;
|
|
const char* masterDir;
|
|
const char* localDir;
|
|
grib_codetable* table;
|
|
int table_loaded;
|
|
} grib_accessor_codetable;
|
|
|
|
extern grib_accessor_class* grib_accessor_class_unsigned;
|
|
|
|
static grib_accessor_class _grib_accessor_class_codetable = {
|
|
&grib_accessor_class_unsigned, /* super */
|
|
"codetable", /* name */
|
|
sizeof(grib_accessor_codetable), /* size */
|
|
0, /* inited */
|
|
0, /* init_class */
|
|
&init, /* init */
|
|
0, /* post_init */
|
|
&destroy, /* destroy */
|
|
&dump, /* dump */
|
|
0, /* next_offset */
|
|
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 */
|
|
&pack_missing, /* pack_missing */
|
|
0, /* is_missing */
|
|
0, /* pack_long */
|
|
&unpack_long, /* unpack_long */
|
|
0, /* pack_double */
|
|
0, /* pack_float */
|
|
0, /* unpack_double */
|
|
0, /* unpack_float */
|
|
&pack_string, /* pack_string */
|
|
&unpack_string, /* unpack_string */
|
|
0, /* pack_string_array */
|
|
0, /* unpack_string_array */
|
|
0, /* pack_bytes */
|
|
0, /* unpack_bytes */
|
|
&pack_expression, /* 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 (double) */
|
|
0, /* unpack only ith value (float) */
|
|
0, /* unpack a given set of elements (double) */
|
|
0, /* unpack a given set of elements (float) */
|
|
0, /* unpack a subarray */
|
|
0, /* clear */
|
|
0, /* clone accessor */
|
|
};
|
|
|
|
|
|
grib_accessor_class* grib_accessor_class_codetable = &_grib_accessor_class_codetable;
|
|
|
|
/* END_CLASS_IMP */
|
|
|
|
static int grib_load_codetable(grib_context* c, const char* filename, const char* recomposed_name, size_t size, grib_codetable* t);
|
|
|
|
static void init(grib_accessor* a, const long len, grib_arguments* params)
|
|
{
|
|
int n = 0;
|
|
long new_len = len;
|
|
grib_handle* hand = grib_handle_of_accessor(a);
|
|
grib_accessor_codetable* self = (grib_accessor_codetable*)a;
|
|
grib_action* act = (grib_action*)(a->creator);
|
|
DEBUG_ASSERT(len == self->nbytes);
|
|
|
|
if (new_len == 0) {
|
|
/* ECC-485: When the codetable length is 0, it means we are passing
|
|
* its length as an identifier not an integer. This identifier is
|
|
* added to the argument list (at the beginning)
|
|
*/
|
|
new_len = grib_arguments_get_long(hand, params, n++);
|
|
if (new_len <= 0) {
|
|
grib_context_log(a->context, GRIB_LOG_FATAL, "%s: codetable length must be a positive integer", a->name);
|
|
}
|
|
self->nbytes = new_len;
|
|
}
|
|
|
|
self->tablename = grib_arguments_get_string(hand, params, n++);
|
|
if (self->tablename == NULL) {
|
|
grib_context_log(a->context, GRIB_LOG_FATAL, "%s: codetable table is invalid", a->name);
|
|
}
|
|
self->masterDir = grib_arguments_get_name(hand, params, n++); /* can be NULL */
|
|
self->localDir = grib_arguments_get_name(hand, params, n++); /* can be NULL */
|
|
|
|
/*if (a->flags & GRIB_ACCESSOR_FLAG_STRING_TYPE)
|
|
printf("-------- %s type string (%ld)\n",a->name,a->flags);*/
|
|
#ifdef DEBUG
|
|
if (a->flags & GRIB_ACCESSOR_FLAG_CAN_BE_MISSING) {
|
|
grib_context_log(a->context, GRIB_LOG_FATAL, "codetable '%s' has flag can_be_missing!", a->name);
|
|
Assert(!"codetable with can_be_missing?");
|
|
}
|
|
#endif
|
|
|
|
if (a->flags & GRIB_ACCESSOR_FLAG_TRANSIENT) {
|
|
a->length = 0;
|
|
if (!a->vvalue)
|
|
a->vvalue = (grib_virtual_value*)grib_context_malloc_clear(a->context, sizeof(grib_virtual_value));
|
|
a->vvalue->type = grib_accessor_get_native_type(a);
|
|
a->vvalue->length = new_len;
|
|
if (act->default_value != NULL) {
|
|
const char* p = 0;
|
|
size_t s_len = 1;
|
|
long l;
|
|
int ret = 0;
|
|
double d;
|
|
char tmp[1024];
|
|
grib_expression* expression = grib_arguments_get_expression(hand, act->default_value, 0);
|
|
int type = grib_expression_native_type(hand, expression);
|
|
switch (type) {
|
|
case GRIB_TYPE_DOUBLE:
|
|
grib_expression_evaluate_double(hand, expression, &d);
|
|
grib_pack_double(a, &d, &s_len);
|
|
break;
|
|
|
|
case GRIB_TYPE_LONG:
|
|
grib_expression_evaluate_long(grib_handle_of_accessor(a), expression, &l);
|
|
grib_pack_long(a, &l, &s_len);
|
|
break;
|
|
|
|
default:
|
|
s_len = sizeof(tmp);
|
|
p = grib_expression_evaluate_string(grib_handle_of_accessor(a), expression, tmp, &s_len, &ret);
|
|
if (ret != GRIB_SUCCESS) {
|
|
grib_context_log(a->context, GRIB_LOG_FATAL,
|
|
"Unable to evaluate %s as string", a->name);
|
|
}
|
|
s_len = strlen(p) + 1;
|
|
pack_string(a, p, &s_len);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
a->length = new_len;
|
|
}
|
|
}
|
|
|
|
/* Note: A fast cut-down version of strcmp which does NOT return -1 */
|
|
/* 0 means input strings are equal and 1 means not equal */
|
|
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;
|
|
}
|
|
|
|
static int str_eq(const char* a, const char* b)
|
|
{
|
|
if (a && b && (grib_inline_strcmp(a, b) == 0))
|
|
return 1;
|
|
return 0;
|
|
}
|
|
|
|
#ifdef DEBUGGING
|
|
static void dump_codetable(grib_codetable* atable)
|
|
{
|
|
grib_codetable* next = NULL;
|
|
int count = 0;
|
|
|
|
next = atable;
|
|
while (next) {
|
|
printf("[%.2d] CodeTable Dump: f0=%s f1=%s\n", count, next->filename[0], next->filename[1]);
|
|
count++;
|
|
next = next->next;
|
|
}
|
|
}
|
|
#endif
|
|
static grib_codetable* load_table(grib_accessor* a)
|
|
{
|
|
grib_accessor_codetable* self = (grib_accessor_codetable*)a;
|
|
size_t size = 0;
|
|
grib_handle* h = ((grib_accessor*)self)->parent->h;
|
|
grib_context* c = h->context;
|
|
grib_codetable* t = NULL;
|
|
grib_codetable* next = NULL;
|
|
char* filename = 0;
|
|
char recomposed[1024] = {0,};
|
|
char localRecomposed[1024] = {0,};
|
|
char* localFilename = 0;
|
|
char masterDir[1024] = {0,};
|
|
char localDir[1024] = {0,};
|
|
size_t 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) {
|
|
char name[2048] = {0,};
|
|
snprintf(name, sizeof(name), "%s/%s", masterDir, self->tablename);
|
|
grib_recompose_name(h, NULL, name, recomposed, 0);
|
|
filename = grib_context_full_defs_path(c, recomposed);
|
|
}
|
|
else {
|
|
grib_recompose_name(h, NULL, self->tablename, recomposed, 0);
|
|
filename = grib_context_full_defs_path(c, recomposed);
|
|
}
|
|
|
|
if (*localDir != 0) {
|
|
char localName[2048] = {0,};
|
|
snprintf(localName, sizeof(localName), "%s/%s", localDir, self->tablename);
|
|
grib_recompose_name(h, NULL, localName, localRecomposed, 0);
|
|
localFilename = grib_context_full_defs_path(c, localRecomposed);
|
|
}
|
|
|
|
GRIB_MUTEX_INIT_ONCE(&once, &thread_init);
|
|
GRIB_MUTEX_LOCK(&mutex1); /* GRIB-930 */
|
|
|
|
/*printf("DBG %s: Look in cache: f=%s lf=%s (recomposed=%s)\n", self->att.name, filename, localFilename,recomposed);*/
|
|
if (filename == NULL && localFilename == NULL) {
|
|
t = NULL;
|
|
goto the_end;
|
|
}
|
|
next = c->codetable;
|
|
while (next) {
|
|
if ((filename && next->filename[0] && grib_inline_strcmp(filename, next->filename[0]) == 0) &&
|
|
((localFilename == 0 && next->filename[1] == NULL) ||
|
|
((localFilename != 0 && next->filename[1] != NULL) && grib_inline_strcmp(localFilename, next->filename[1]) == 0))) {
|
|
t = next;
|
|
goto the_end;
|
|
}
|
|
/* Special case: see GRIB-735 */
|
|
if (filename == NULL && localFilename != NULL) {
|
|
if (str_eq(localFilename, next->filename[0]) ||
|
|
str_eq(localFilename, next->filename[1])) {
|
|
t = next;
|
|
goto the_end;
|
|
}
|
|
}
|
|
next = next->next;
|
|
}
|
|
|
|
if (a->flags & GRIB_ACCESSOR_FLAG_TRANSIENT) {
|
|
Assert(a->vvalue != NULL);
|
|
size = a->vvalue->length * 8;
|
|
}
|
|
else {
|
|
size = grib_byte_count((grib_accessor*)self) * 8;
|
|
}
|
|
|
|
size = (1ULL << size); /* 2^size - 64bits */
|
|
|
|
t = (grib_codetable*)grib_context_malloc_clear_persistent(c, sizeof(grib_codetable) +
|
|
(size - 1) * sizeof(code_table_entry));
|
|
|
|
if (filename != 0)
|
|
grib_load_codetable(c, filename, recomposed, size, t);
|
|
|
|
if (localFilename != 0)
|
|
grib_load_codetable(c, localFilename, localRecomposed, size, t);
|
|
|
|
/*dump_codetable(c->codetable);*/
|
|
|
|
if (t->filename[0] == NULL && t->filename[1] == NULL) {
|
|
grib_context_free_persistent(c, t);
|
|
t = NULL;
|
|
goto the_end;
|
|
}
|
|
|
|
the_end:
|
|
GRIB_MUTEX_UNLOCK(&mutex1);
|
|
|
|
return t;
|
|
}
|
|
|
|
static int grib_load_codetable(grib_context* c, const char* filename,
|
|
const char* recomposed_name, size_t size, grib_codetable* t)
|
|
{
|
|
char line[1024];
|
|
FILE* f = NULL;
|
|
int lineNumber = 0;
|
|
grib_context_log(c, GRIB_LOG_DEBUG, "Loading code table from %s", filename);
|
|
|
|
f = codes_fopen(filename, "r");
|
|
if (!f)
|
|
return GRIB_IO_PROBLEM;
|
|
|
|
Assert(t != NULL);
|
|
|
|
if (t->filename[0] == NULL) {
|
|
t->filename[0] = grib_context_strdup_persistent(c, filename);
|
|
t->recomposed_name[0] = grib_context_strdup_persistent(c, recomposed_name);
|
|
t->next = c->codetable;
|
|
t->size = size;
|
|
c->codetable = t;
|
|
}
|
|
else {
|
|
t->filename[1] = grib_context_strdup_persistent(c, filename);
|
|
t->recomposed_name[1] = grib_context_strdup_persistent(c, recomposed_name);
|
|
}
|
|
|
|
while (fgets(line, sizeof(line) - 1, f)) {
|
|
char* p = line;
|
|
int code = 0;
|
|
char abbreviation[1024] = {0,};
|
|
char title[1024] = {0,};
|
|
char* pAbbrev = abbreviation;
|
|
char* pTitle = title;
|
|
char* units = 0;
|
|
char unknown[] = "unknown";
|
|
char* last_open_paren = NULL;
|
|
char* last_clos_paren = NULL;
|
|
|
|
++lineNumber;
|
|
|
|
line[strlen(line) - 1] = 0;
|
|
|
|
while (*p != '\0' && isspace(*p))
|
|
p++;
|
|
|
|
if (*p == '#')
|
|
continue;
|
|
|
|
last_open_paren = strrchr(line, '(');
|
|
|
|
while (*p != '\0' && isspace(*p))
|
|
p++;
|
|
|
|
if (*p == '\0')
|
|
continue;
|
|
|
|
if (!isdigit(*p)) {
|
|
grib_context_log(c, GRIB_LOG_ERROR, "Invalid entry in file %s: line %d", filename, lineNumber);
|
|
continue; /* skip this line */
|
|
}
|
|
Assert(isdigit(*p));
|
|
|
|
while (*p != '\0') {
|
|
if (isspace(*p))
|
|
break;
|
|
code *= 10;
|
|
code += *p - '0';
|
|
p++;
|
|
}
|
|
|
|
if (code < 0 || code >= size) {
|
|
grib_context_log(c, GRIB_LOG_WARNING, "code_table_entry: invalid code in %s: %d (table size=%ld)", filename, code, size);
|
|
continue;
|
|
}
|
|
|
|
while (*p != '\0' && isspace(*p))
|
|
p++;
|
|
|
|
while (*p != '\0') {
|
|
if (isspace(*p))
|
|
break;
|
|
*pAbbrev++ = *p++;
|
|
}
|
|
*pAbbrev = 0;
|
|
while (*p != '\0' && isspace(*p))
|
|
p++;
|
|
|
|
/* The title goes as far as the last open paren */
|
|
while (*p != '\0') {
|
|
if (last_open_paren && p >= last_open_paren && *p == '(')
|
|
break;
|
|
*pTitle++ = *p++;
|
|
}
|
|
*pTitle = 0;
|
|
|
|
/* units at the end */
|
|
if (last_open_paren) {
|
|
last_clos_paren = strrchr(line, ')');
|
|
if (last_clos_paren && last_open_paren != last_clos_paren) {
|
|
units = last_open_paren + 1;
|
|
p = units;
|
|
p += (last_clos_paren - last_open_paren - 1);
|
|
*p = '\0';
|
|
}
|
|
}
|
|
if (!units)
|
|
units = unknown;
|
|
|
|
Assert(*abbreviation);
|
|
Assert(*title);
|
|
string_rtrim(title); /* ECC-1315 */
|
|
|
|
if (t->entries[code].abbreviation != NULL) {
|
|
grib_context_log(c, GRIB_LOG_WARNING, "code_table_entry: duplicate code in %s: %d (table size=%ld)", filename, code, size);
|
|
continue;
|
|
}
|
|
|
|
Assert(t->entries[code].abbreviation == NULL);
|
|
Assert(t->entries[code].title == NULL);
|
|
|
|
t->entries[code].abbreviation = grib_context_strdup_persistent(c, abbreviation);
|
|
t->entries[code].title = grib_context_strdup_persistent(c, title);
|
|
t->entries[code].units = grib_context_strdup_persistent(c, units);
|
|
}
|
|
|
|
fclose(f);
|
|
|
|
return 0;
|
|
}
|
|
|
|
void grib_codetable_delete(grib_context* c)
|
|
{
|
|
grib_codetable* t = c->codetable;
|
|
|
|
while (t) {
|
|
grib_codetable* s = t->next;
|
|
int i;
|
|
|
|
for (i = 0; i < t->size; i++) {
|
|
grib_context_free_persistent(c, t->entries[i].abbreviation);
|
|
grib_context_free_persistent(c, t->entries[i].title);
|
|
grib_context_free_persistent(c, t->entries[i].units);
|
|
}
|
|
grib_context_free_persistent(c, t->filename[0]);
|
|
if (t->filename[1])
|
|
grib_context_free_persistent(c, t->filename[1]);
|
|
grib_context_free_persistent(c, t->recomposed_name[0]);
|
|
if (t->recomposed_name[1])
|
|
grib_context_free_persistent(c, t->recomposed_name[1]);
|
|
grib_context_free_persistent(c, t);
|
|
t = s;
|
|
}
|
|
}
|
|
|
|
int codes_codetable_get_contents_malloc(const grib_handle* h, const char* key, code_table_entry** entries, size_t* num_entries)
|
|
{
|
|
long lvalue = 0;
|
|
size_t size = 1;
|
|
int err = 0;
|
|
grib_context* c = h->context;
|
|
|
|
grib_accessor* aa = grib_find_accessor(h, key);
|
|
if (!aa) return GRIB_NOT_FOUND;
|
|
|
|
if (!STR_EQUAL(aa->cclass->name, "codetable")) {
|
|
return GRIB_INVALID_ARGUMENT; // key is not a codetable
|
|
}
|
|
|
|
const grib_accessor_codetable* ca = (const grib_accessor_codetable*)aa; // could be dynamic_cast
|
|
|
|
// Decode the key itself. This will either fetch it from the cache or place it there
|
|
if ((err = grib_unpack_long(aa, &lvalue, &size)) != GRIB_SUCCESS) {
|
|
return err;
|
|
}
|
|
|
|
const grib_codetable* table = ca->table;
|
|
if (!table) return GRIB_INTERNAL_ERROR;
|
|
|
|
grib_codetable* cached_table = c->codetable; // Access the codetable cache
|
|
while (cached_table) {
|
|
if (STR_EQUAL(table->recomposed_name[0], cached_table->recomposed_name[0])) {
|
|
// Found a cache entry that matches the recomposed name of ours
|
|
*num_entries = cached_table->size;
|
|
*entries = (code_table_entry*)calloc(cached_table->size, sizeof(code_table_entry));
|
|
if (!*entries) {
|
|
return GRIB_OUT_OF_MEMORY;
|
|
}
|
|
for (size_t i = 0; i < cached_table->size; i++) {
|
|
(*entries)[i] = cached_table->entries[i];
|
|
}
|
|
return GRIB_SUCCESS;
|
|
}
|
|
cached_table = cached_table->next;
|
|
}
|
|
|
|
return GRIB_CODE_NOT_FOUND_IN_TABLE;
|
|
}
|
|
|
|
int codes_codetable_check_code_figure(const grib_handle* h, const char* key, long code_figure)
|
|
{
|
|
code_table_entry* entries = NULL;
|
|
size_t num_entries = 0;
|
|
int err = 0;
|
|
err = codes_codetable_get_contents_malloc(h, key, &entries, &num_entries);
|
|
if (err) return err;
|
|
|
|
if (code_figure < 0 || (size_t)code_figure >= num_entries) {
|
|
err = GRIB_OUT_OF_RANGE;
|
|
goto cleanup;
|
|
}
|
|
|
|
if (entries[code_figure].abbreviation == NULL) {
|
|
err = GRIB_INVALID_KEY_VALUE;
|
|
goto cleanup;
|
|
}
|
|
cleanup:
|
|
free(entries);
|
|
return err;
|
|
}
|
|
|
|
int codes_codetable_check_abbreviation(const grib_handle* h, const char* key, const char* abbreviation)
|
|
{
|
|
code_table_entry* entries = NULL;
|
|
size_t num_entries = 0;
|
|
int err = 0;
|
|
err = codes_codetable_get_contents_malloc(h, key, &entries, &num_entries);
|
|
if (err) return err;
|
|
|
|
bool found = false;
|
|
for (size_t i=0; i<num_entries; ++i) {
|
|
const char* abbrev = entries[i].abbreviation;
|
|
if (abbrev && STR_EQUAL(abbrev, abbreviation)) {
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
if (!found) err = GRIB_INVALID_KEY_VALUE;
|
|
|
|
free(entries);
|
|
return err;
|
|
}
|
|
|
|
static void dump(grib_accessor* a, grib_dumper* dumper)
|
|
{
|
|
grib_accessor_codetable* self = (grib_accessor_codetable*)a;
|
|
char comment[2048];
|
|
grib_codetable* table;
|
|
|
|
size_t llen = 1;
|
|
long value;
|
|
|
|
if (!self->table_loaded) {
|
|
self->table = load_table(a); /* may return NULL */
|
|
self->table_loaded = 1;
|
|
}
|
|
table = self->table;
|
|
|
|
grib_unpack_long(a, &value, &llen);
|
|
|
|
if (value == GRIB_MISSING_LONG) {
|
|
if (a->length < 4) {
|
|
value = (1L << a->length) - 1;
|
|
}
|
|
}
|
|
|
|
if (table && value >= 0 && value < table->size) {
|
|
if (table->entries[value].abbreviation) {
|
|
long b = atol(table->entries[value].abbreviation);
|
|
if (b == value)
|
|
strcpy(comment, table->entries[value].title);
|
|
else
|
|
snprintf(comment, sizeof(comment), "%s", table->entries[value].title);
|
|
|
|
if (table->entries[value].units != NULL && grib_inline_strcmp(table->entries[value].units, "unknown")) {
|
|
strcat(comment, " (");
|
|
strcat(comment, table->entries[value].units);
|
|
strcat(comment, ") ");
|
|
}
|
|
}
|
|
else {
|
|
strcpy(comment, "Unknown code table entry");
|
|
}
|
|
}
|
|
else {
|
|
strcpy(comment, "Unknown code table entry");
|
|
}
|
|
|
|
strcat(comment, " (");
|
|
if (table) {
|
|
strcat(comment, table->recomposed_name[0]);
|
|
if (table->recomposed_name[1] != NULL) {
|
|
strcat(comment, " , ");
|
|
strcat(comment, table->recomposed_name[1]);
|
|
}
|
|
}
|
|
strcat(comment, ") ");
|
|
|
|
grib_dump_long(dumper, a, comment);
|
|
}
|
|
|
|
static int unpack_string(grib_accessor* a, char* buffer, size_t* len)
|
|
{
|
|
grib_accessor_codetable* self = (grib_accessor_codetable*)a;
|
|
grib_codetable* table = NULL;
|
|
|
|
size_t size = 1;
|
|
long value;
|
|
int err = GRIB_SUCCESS;
|
|
char tmp[1024];
|
|
size_t l = 0;
|
|
|
|
if ((err = grib_unpack_long(a, &value, &size)) != GRIB_SUCCESS)
|
|
return err;
|
|
|
|
if (!self->table_loaded) {
|
|
self->table = load_table(a); /* may return NULL */
|
|
self->table_loaded = 1;
|
|
}
|
|
table = self->table;
|
|
|
|
if (table && (value >= 0) && (value < table->size) && table->entries[value].abbreviation) {
|
|
strcpy(tmp, table->entries[value].abbreviation);
|
|
}
|
|
else {
|
|
snprintf(tmp, sizeof(tmp), "%d", (int)value);
|
|
}
|
|
|
|
l = strlen(tmp) + 1;
|
|
|
|
if (*len < l) {
|
|
*len = l;
|
|
return GRIB_BUFFER_TOO_SMALL;
|
|
}
|
|
|
|
strcpy(buffer, tmp);
|
|
*len = l;
|
|
|
|
return GRIB_SUCCESS;
|
|
}
|
|
|
|
static int value_count(grib_accessor* a, long* count)
|
|
{
|
|
*count = 1;
|
|
return 0;
|
|
}
|
|
|
|
// Return true if the input is an integer (non-negative)
|
|
static bool is_number(const char* s)
|
|
{
|
|
while (*s) {
|
|
if (!isdigit(*s))
|
|
return false;
|
|
s++;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool strings_equal(const char* s1, const char* s2, bool case_sensitive)
|
|
{
|
|
if (case_sensitive) return (strcmp(s1, s2) == 0);
|
|
return (strcmp_nocase(s1, s2) == 0);
|
|
}
|
|
|
|
static int pack_string(grib_accessor* a, const char* buffer, size_t* len)
|
|
{
|
|
long lValue = 0;
|
|
Assert(buffer);
|
|
if (is_number(buffer) && string_to_long(buffer, &lValue, 1) == GRIB_SUCCESS) {
|
|
// ECC-1654: If value is a pure number, just pack as long
|
|
size_t l = 1;
|
|
return grib_pack_long(a, &lValue, &l);
|
|
}
|
|
|
|
if (STR_EQUAL_NOCASE(buffer, "missing")) {
|
|
return pack_missing(a);
|
|
}
|
|
|
|
grib_accessor_codetable* self = (grib_accessor_codetable*)a;
|
|
grib_codetable* table = NULL;
|
|
long i = 0;
|
|
size_t size = 1;
|
|
|
|
if (!self->table_loaded) {
|
|
self->table = load_table(a); /* may return NULL */
|
|
self->table_loaded = 1;
|
|
}
|
|
table = self->table;
|
|
|
|
if (!table)
|
|
return GRIB_ENCODING_ERROR;
|
|
|
|
if (a->set) {
|
|
int err = grib_set_string(grib_handle_of_accessor(a), a->set, buffer, len);
|
|
if (err != 0)
|
|
return err;
|
|
}
|
|
|
|
// If the key has the "lowercase" flag set, then the string comparison
|
|
// should ignore the case
|
|
bool case_sensitive = true;
|
|
if (a->flags & GRIB_ACCESSOR_FLAG_LOWERCASE) case_sensitive = false;
|
|
|
|
for (i = 0; i < table->size; i++) {
|
|
if (table->entries[i].abbreviation) {
|
|
if (strings_equal(table->entries[i].abbreviation, buffer, case_sensitive)) {
|
|
return grib_pack_long(a, &i, &size);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (a->flags & GRIB_ACCESSOR_FLAG_NO_FAIL) {
|
|
grib_action* act = (grib_action*)(a->creator);
|
|
if (act->default_value != NULL) {
|
|
const char* p = 0;
|
|
size_t s_len = 1;
|
|
long l = 0;
|
|
int ret = 0;
|
|
double d = 0;
|
|
char tmp[1024] = {0,};
|
|
grib_expression* expression = grib_arguments_get_expression(grib_handle_of_accessor(a), act->default_value, 0);
|
|
int type = grib_expression_native_type(grib_handle_of_accessor(a), expression);
|
|
switch (type) {
|
|
case GRIB_TYPE_DOUBLE:
|
|
grib_expression_evaluate_double(grib_handle_of_accessor(a), expression, &d);
|
|
grib_pack_double(a, &d, &s_len);
|
|
break;
|
|
|
|
case GRIB_TYPE_LONG:
|
|
grib_expression_evaluate_long(grib_handle_of_accessor(a), expression, &l);
|
|
grib_pack_long(a, &l, &s_len);
|
|
break;
|
|
|
|
default:
|
|
s_len = sizeof(tmp);
|
|
p = grib_expression_evaluate_string(grib_handle_of_accessor(a), expression, tmp, &s_len, &ret);
|
|
if (ret != GRIB_SUCCESS) {
|
|
grib_context_log(a->context, GRIB_LOG_ERROR,
|
|
"%s: Unable to evaluate default value of %s as string expression", __func__, a->name);
|
|
return ret;
|
|
}
|
|
s_len = strlen(p) + 1;
|
|
pack_string(a, p, &s_len);
|
|
break;
|
|
}
|
|
return GRIB_SUCCESS;
|
|
}
|
|
}
|
|
|
|
// ECC-1652: Failed. Now do a case-insensitive compare to give the user a hint
|
|
for (i = 0; i < table->size; i++) {
|
|
if (table->entries[i].abbreviation) {
|
|
if (strcmp_nocase(table->entries[i].abbreviation, buffer) == 0) {
|
|
grib_context_log(a->context, GRIB_LOG_ERROR,
|
|
"%s: No such code table entry: '%s' "
|
|
"(Did you mean '%s'?)",
|
|
a->name, buffer, table->entries[i].abbreviation);
|
|
}
|
|
}
|
|
}
|
|
|
|
return GRIB_ENCODING_ERROR;
|
|
}
|
|
|
|
static int pack_expression(grib_accessor* a, grib_expression* e)
|
|
{
|
|
const char* cval = NULL;
|
|
int ret = 0;
|
|
long lval = 0;
|
|
size_t len = 1;
|
|
grib_handle* hand = grib_handle_of_accessor(a);
|
|
|
|
if (strcmp(e->cclass->name, "long") == 0) {
|
|
grib_expression_evaluate_long(hand, e, &lval); /* TODO: check return value */
|
|
//if (hand->context->debug) printf("ECCODES DEBUG grib_accessor_class_codetable::pack_expression %s %ld\n", a->name,lval);
|
|
ret = grib_pack_long(a, &lval, &len);
|
|
}
|
|
else {
|
|
char tmp[1024];
|
|
len = sizeof(tmp);
|
|
cval = grib_expression_evaluate_string(hand, e, tmp, &len, &ret);
|
|
if (ret != GRIB_SUCCESS) {
|
|
grib_context_log(a->context, GRIB_LOG_ERROR,
|
|
"grib_accessor_codetable.%s: Unable to evaluate string %s to be set in %s",
|
|
__func__, grib_expression_get_name(e), a->name);
|
|
return ret;
|
|
}
|
|
len = strlen(cval) + 1;
|
|
//if (hand->context->debug)
|
|
// printf("ECCODES DEBUG grib_accessor_class_codetable::pack_expression %s %s\n", a->name, cval);
|
|
ret = grib_pack_string(a, cval, &len);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static void destroy(grib_context* context, grib_accessor* a)
|
|
{
|
|
if (a->vvalue != NULL) {
|
|
grib_context_free(context, a->vvalue);
|
|
a->vvalue = NULL;
|
|
}
|
|
}
|
|
|
|
static int get_native_type(grib_accessor* a)
|
|
{
|
|
int type = GRIB_TYPE_LONG;
|
|
/*printf("---------- %s flags=%ld GRIB_ACCESSOR_FLAG_STRING_TYPE=%d\n",
|
|
a->name,a->flags,GRIB_ACCESSOR_FLAG_STRING_TYPE);*/
|
|
if (a->flags & GRIB_ACCESSOR_FLAG_STRING_TYPE)
|
|
type = GRIB_TYPE_STRING;
|
|
return type;
|
|
}
|
|
|
|
static int unpack_long(grib_accessor* a, long* val, size_t* len)
|
|
{
|
|
grib_accessor_codetable* self = (grib_accessor_codetable*)a;
|
|
long rlen = 0, i = 0;
|
|
long pos = a->offset * 8;
|
|
grib_handle* hand = NULL;
|
|
|
|
#ifdef DEBUG
|
|
{
|
|
int err = grib_value_count(a, &rlen);
|
|
Assert(!err);
|
|
Assert(rlen == 1);
|
|
}
|
|
#endif
|
|
rlen = 1; /* ECC-480 Performance: avoid func call overhead of grib_value_count */
|
|
|
|
if (!self->table_loaded) {
|
|
self->table = load_table(a); /* may return NULL */
|
|
self->table_loaded = 1;
|
|
}
|
|
|
|
if (*len < rlen) {
|
|
grib_context_log(a->context, GRIB_LOG_ERROR, "Wrong size (%lu) for %s, it contains %ld values",
|
|
*len, a->name, rlen);
|
|
*len = 0;
|
|
return GRIB_ARRAY_TOO_SMALL;
|
|
}
|
|
|
|
if (a->flags & GRIB_ACCESSOR_FLAG_TRANSIENT) {
|
|
*val = a->vvalue->lval;
|
|
*len = 1;
|
|
return GRIB_SUCCESS;
|
|
}
|
|
|
|
/* ECC-480 Performance: inline the grib_handle_of_accessor here to reduce func call overhead */
|
|
if (a->parent == NULL)
|
|
hand = a->h;
|
|
else
|
|
hand = a->parent->h;
|
|
|
|
for (i = 0; i < rlen; i++) {
|
|
val[i] = (long)grib_decode_unsigned_long(hand->buffer->data, &pos, self->nbytes * 8);
|
|
}
|
|
|
|
*len = rlen;
|
|
return GRIB_SUCCESS;
|
|
}
|
|
|
|
static int pack_missing(grib_accessor* a)
|
|
{
|
|
// Many of the code tables do have a 'Missing' entry (all bits = 1)
|
|
// So it is more user-friendly to allow setting codetable keys to
|
|
// missing. For tables that do not have such an entry, an error is issued
|
|
grib_accessor_codetable* self = (grib_accessor_codetable*)a;
|
|
grib_handle* h = grib_handle_of_accessor(a);
|
|
|
|
const long nbytes = a->length;
|
|
const long nbits = nbytes*8;
|
|
const long maxVal = (1<<nbits) - 1;
|
|
|
|
int err = codes_codetable_check_code_figure(h, a->name, maxVal);
|
|
if (!err) {
|
|
size_t l = 1;
|
|
return grib_pack_long(a, &maxVal, &l);
|
|
}
|
|
|
|
grib_context_log(a->context, GRIB_LOG_ERROR, "There is no 'missing' entry in Code Table %s (%s)",
|
|
self->tablename, grib_get_error_message(err));
|
|
|
|
return err;
|
|
}
|