/**
* Copyright 2005-2007 ECMWF
*
* Licensed under the GNU Lesser General Public License which
* incorporates the terms and conditions of version 3 of the GNU
* General Public License.
* See LICENSE and gpl-3.0.txt for details.
*/

#include "grib_api_internal.h"

/*
   This is used by make_class.pl

   START_CLASS_DEF
   CLASS      = expression
   IMPLEMENTS = init_class
   IMPLEMENTS = destroy
   IMPLEMENTS = native_type
   IMPLEMENTS = evaluate_long
   IMPLEMENTS = evaluate_double
   IMPLEMENTS = print
   IMPLEMENTS = compile
   IMPLEMENTS = add_dependency
   MEMBERS    = grib_expression *left
   MEMBERS = grib_expression *right
   MEMBERS = grib_binop_long_proc    long_func
   MEMBERS = grib_binop_double_proc  double_func
   MEMBERS = grib_binop_string_proc  string_func
   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 "expression.class" and rerun ./make_class.pl

*/

typedef const char* string; /* to keep make_class.pl happy */


static void init_class              (grib_expression_class*);

static void        destroy(grib_context*,grib_expression* e);

static void        print(grib_context*,grib_expression*,grib_handle*);
static void        compile(grib_expression*,grib_compiler*);
static void        add_dependency(grib_expression* e, grib_accessor* observer);

static int        native_type(grib_expression*,grib_handle*);

static int        evaluate_long(grib_expression*,grib_handle*,long*);
static int      evaluate_double(grib_expression*,grib_handle*,double*);

typedef struct grib_expression_binop{
  grib_expression base;
/* Members defined in binop */
	grib_expression *left;
	grib_expression *right;
	grib_binop_long_proc    long_func;
	grib_binop_double_proc  double_func;
	grib_binop_string_proc  string_func;
} grib_expression_binop;


static grib_expression_class _grib_expression_class_binop = {
    0,                    /* super                     */
    "binop",                    /* name                      */
    sizeof(grib_expression_binop),/* size of instance          */
    0,                           /* inited */
    &init_class,                 /* init_class */
    0,                     /* constructor               */
    &destroy,                  /* destructor                */
    &print,                 
    &compile,                 
    &add_dependency,       

	&native_type,
	0,

	&evaluate_long,
	&evaluate_double,
	0,
};

grib_expression_class* grib_expression_class_binop = &_grib_expression_class_binop;


static void init_class(grib_expression_class* c)
{
}
/* END_CLASS_IMP */

static int evaluate_long(grib_expression *g,grib_handle* h,long* lres)
{
  long v1=0;
  long v2=0;
  int ret;
  grib_expression_binop* e = (grib_expression_binop*)g;

  ret = grib_expression_evaluate_long(h,e->left,&v1);
  if (ret != GRIB_SUCCESS) return ret;

  ret = grib_expression_evaluate_long(h,e->right,&v2);
  if (ret != GRIB_SUCCESS) return ret;

  *lres=e->long_func(v1,v2);
  return GRIB_SUCCESS;
}

static int evaluate_double(grib_expression *g,grib_handle* h,double* dres)
{
  double v1=0.0;
  double v2=0.0;
  int ret;

  grib_expression_binop* e = (grib_expression_binop*)g;

  ret = grib_expression_evaluate_double(h,e->left,&v1);
  if (ret != GRIB_SUCCESS) return ret;

  ret = grib_expression_evaluate_double(h,e->right,&v2);
  if (ret != GRIB_SUCCESS) return ret;

  *dres = e->double_func ? e->double_func(v1,v2) : e->long_func(v1,v2);

  return GRIB_SUCCESS;

}

static void print(grib_context* c,grib_expression* g,grib_handle* f)
{
  grib_expression_binop* e = (grib_expression_binop*)g;
  printf("binop(");
  grib_expression_print(c,e->left,f);
  printf(",");
  grib_expression_print(c,e->right,f);
  printf(")");
}

static void destroy(grib_context* c,grib_expression* g)
{
  grib_expression_binop* e = (grib_expression_binop*)g;
  grib_expression_free(c,e->left);
  grib_expression_free(c,e->right);
}


static void  add_dependency(grib_expression* g, grib_accessor* observer){
  grib_expression_binop* e = (grib_expression_binop*)g;
  grib_dependency_observe_expression(observer,e->left);
  grib_dependency_observe_expression(observer,e->right);
}

grib_expression* new_binop_expression(grib_context* c,
  grib_binop_long_proc  long_func,
  grib_binop_double_proc double_func,
  grib_expression* left,grib_expression* right)
{
  grib_expression_binop* e = grib_context_malloc_clear_persistent(c,sizeof(grib_expression_binop));
  e->base.cclass                 = grib_expression_class_binop;
  e->left                = left;
  e->right               = right;
  e->long_func            = long_func;
  e->double_func          = double_func;
  return (grib_expression*)e;
}

static void compile(grib_expression* g,grib_compiler* c)
{
	grib_expression_binop* e = (grib_expression_binop*)g;
    fprintf(c->out,"new_binop_expression(ctx,");
    fprintf(c->out,"%s,",grib_binop_long_proc_name(e->long_func));
    fprintf(c->out,"%s,",grib_binop_double_proc_name(e->double_func));
    grib_expression_compile(e->left,c);
    fprintf(c->out,",");
    grib_expression_compile(e->right,c);
    fprintf(c->out,")");
}

static int native_type(grib_expression* g,grib_handle *h)
{
  grib_expression_binop* e = (grib_expression_binop*)g;
  return e->double_func ? GRIB_TYPE_LONG : GRIB_TYPE_DOUBLE;
}