eccodes/src/action_class_if.cc

308 lines
9.5 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.
*/
/***************************************************************************
* Jean Baptiste Filippi - 01.11.2005 *
* Enrico Fucile *
***************************************************************************/
#include "grib_api_internal.h"
/*
This is used by make_class.pl
START_CLASS_DEF
CLASS = action
SUPER = action_class_section
IMPLEMENTS = create_accessor
IMPLEMENTS = dump
IMPLEMENTS = destroy
IMPLEMENTS = reparse;execute
MEMBERS = grib_expression *expression
MEMBERS = grib_action *block_true
MEMBERS = grib_action *block_false
MEMBERS = int transient
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 "action.class" and rerun ./make_class.pl
*/
static void init_class (grib_action_class*);
static void dump (grib_action* d, FILE*,int);
static void destroy (grib_context*,grib_action*);
static int create_accessor(grib_section*,grib_action*,grib_loader*);
static grib_action* reparse(grib_action* a,grib_accessor* acc,int *doit);
static int execute(grib_action* a,grib_handle* h);
typedef struct grib_action_if {
grib_action act;
/* Members defined in section */
/* Members defined in if */
grib_expression *expression;
grib_action *block_true;
grib_action *block_false;
int transient;
} grib_action_if;
extern grib_action_class* grib_action_class_section;
static grib_action_class _grib_action_class_if = {
&grib_action_class_section, /* super */
"action_class_if", /* name */
sizeof(grib_action_if), /* size */
0, /* inited */
&init_class, /* init_class */
0, /* init */
&destroy, /* destroy */
&dump, /* dump */
0, /* xref */
&create_accessor, /* create_accessor*/
0, /* notify_change */
&reparse, /* reparse */
&execute, /* execute */
};
grib_action_class* grib_action_class_if = &_grib_action_class_if;
static void init_class(grib_action_class* c)
{
c->xref = (*(c->super))->xref;
c->notify_change = (*(c->super))->notify_change;
}
/* END_CLASS_IMP */
grib_action* grib_action_create_if(grib_context* context,
grib_expression* expression,
grib_action* block_true, grib_action* block_false, int transient,
int lineno, const char* file_being_parsed)
{
char name[1024];
const size_t nameLen = sizeof(name);
grib_action_if* a;
grib_action_class* c = grib_action_class_if;
grib_action* act = (grib_action*)grib_context_malloc_clear_persistent(context, c->size);
act->op = grib_context_strdup_persistent(context, "section");
act->cclass = c;
a = (grib_action_if*)act;
act->context = context;
a->expression = expression;
a->block_true = block_true;
a->block_false = block_false;
a->transient = transient;
if (transient)
snprintf(name, nameLen, "__if%p", (void*)a);
else
snprintf(name, nameLen, "_if%p", (void*)a);
act->name = grib_context_strdup_persistent(context, name);
act->debug_info = NULL;
if (context->debug > 0 && file_being_parsed) {
/* Construct debug information showing definition file and line */
/* number of IF statement */
char debug_info[1024];
const size_t infoLen = sizeof(debug_info);
snprintf(debug_info, infoLen, "File=%s line=%d", file_being_parsed, lineno);
act->debug_info = grib_context_strdup_persistent(context, debug_info);
}
return act;
}
static int create_accessor(grib_section* p, grib_action* act, grib_loader* h)
{
grib_action_if* a = (grib_action_if*)act;
grib_action* next = NULL;
int ret = 0;
long lres = 0;
grib_accessor* as = NULL;
grib_section* gs = NULL;
as = grib_accessor_factory(p, act, 0, NULL);
if (!as)
return GRIB_INTERNAL_ERROR;
gs = as->sub_section;
grib_push_accessor(as, p->block);
if ((ret = grib_expression_evaluate_long(p->h, a->expression, &lres)) != GRIB_SUCCESS)
return ret;
if (lres)
next = a->block_true;
else
next = a->block_false;
if (p->h->context->debug > 1) {
printf("EVALUATE create_accessor_handle ");
grib_expression_print(p->h->context, a->expression, p->h);
printf(" [%s][_if%p]\n", (next == a->block_true ? "true" : "false"), (void*)a);
/*grib_dump_action_branch(stdout,next,5);*/
}
gs->branch = next;
grib_dependency_observe_expression(as, a->expression);
while (next) {
ret = grib_create_accessor(gs, next, h);
if (ret != GRIB_SUCCESS)
return ret;
next = next->next;
}
return GRIB_SUCCESS;
}
static void print_expression_debug_info(grib_context* ctx, grib_expression* exp, grib_handle* h)
{
grib_expression_print(ctx, exp, h); /* writes to stdout without a newline */
printf("\n");
}
static int execute(grib_action* act, grib_handle* h)
{
grib_action_if* a = (grib_action_if*)act;
grib_action* next = NULL;
grib_context* ctx = h->context;
int ret = 0;
long lres = 0;
/* See GRIB-394 */
int type = grib_expression_native_type(h, a->expression);
if (type != GRIB_TYPE_DOUBLE) {
if ((ret = grib_expression_evaluate_long(h, a->expression, &lres)) != GRIB_SUCCESS) {
if (ret == GRIB_NOT_FOUND)
lres = 0;
else {
if (ctx->debug)
print_expression_debug_info(ctx, a->expression, h);
return ret;
}
}
}
else {
double dres = 0.0;
ret = grib_expression_evaluate_double(h, a->expression, &dres);
lres = (long)dres;
if (ret != GRIB_SUCCESS) {
if (ret == GRIB_NOT_FOUND)
lres = 0;
else {
if (ctx->debug)
print_expression_debug_info(ctx, a->expression, h);
return ret;
}
}
}
if (lres)
next = a->block_true;
else
next = a->block_false;
while (next) {
ret = grib_action_execute(next, h);
if (ret != GRIB_SUCCESS)
return ret;
next = next->next;
}
return GRIB_SUCCESS;
}
static void dump(grib_action* act, FILE* f, int lvl)
{
grib_action_if* a = (grib_action_if*)act;
int i = 0;
for (i = 0; i < lvl; i++)
grib_context_print(act->context, f, " ");
printf("if(%s) { ", act->name);
grib_expression_print(act->context, a->expression, 0);
printf("\n");
if (a->block_true) {
/* grib_context_print(act->context,f,"IF \t TODO \n"); TODO */
grib_dump_action_branch(f, a->block_true, lvl + 1);
}
if (a->block_false) {
printf("}\n");
for (i = 0; i < lvl; i++)
grib_context_print(act->context, f, " ");
printf("else(%s) { ", act->name);
grib_expression_print(act->context, a->expression, 0);
/* grib_context_print(act->context,f,"ELSE \n" );*/
grib_dump_action_branch(f, a->block_false, lvl + 1);
}
for (i = 0; i < lvl; i++)
grib_context_print(act->context, f, " ");
printf("}\n");
}
static grib_action* reparse(grib_action* a, grib_accessor* acc, int* doit)
{
int ret = 0;
long lres = 0;
grib_action_if* self = (grib_action_if*)a;
/* printf("reparse %s %s\n",a->name,acc->name); */
if ((ret = grib_expression_evaluate_long(grib_handle_of_accessor(acc), self->expression, &lres)) != GRIB_SUCCESS)
grib_context_log(acc->context,
GRIB_LOG_ERROR, "action_class_if::reparse: grib_expression_evaluate_long failed: %s",
grib_get_error_message(ret));
if (lres)
return self->block_true;
else
return self->block_false;
}
static void destroy(grib_context* context, grib_action* act)
{
grib_action_if* a = (grib_action_if*)act;
grib_action* t = a->block_true;
grib_action* f = a->block_false;
while (t) {
grib_action* nt = t->next;
grib_action_delete(context, t);
t = nt;
}
while (f) {
grib_action* nf = f->next;
grib_action_delete(context, f);
f = nf;
}
grib_expression_free(context, a->expression);
grib_context_free_persistent(context, act->name);
grib_context_free_persistent(context, act->debug_info);
grib_context_free_persistent(context, act->op);
}