---

func_evaluator.h

Go to the documentation of this file.

// Function Evaluator implementation -*- C++ -*-

// Copyright (C) 2001-2003 Hermann Schichl
//
// This file is part of the COCONUT API.  This library
// is free software; you can redistribute it and/or modify it under the
// terms of the Library GNU General Public License as published by the
// Free Software Foundation; either version 2, or (at your option)
// any later version.

// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// Library GNU General Public License for more details.

// As a special exception, you may use this file as part of a free software
// library without restriction.  Specifically, if other files instantiate
// templates or use macros or inline functions from this file, or you compile
// this file and link it with other files to produce an executable, this
// file does not by itself cause the resulting executable to be covered by
// the Library GNU General Public License.  This exception does not however
// invalidate any other reasons why the executable file might be covered by
// the Library GNU General Public License.

#ifndef _FUNC_EVALUATOR_H_
#define _FUNC_EVALUATOR_H_

#include <coconut_config.h>
#include <evaluator.h>
#include <expression.h>
#include <model.h>
#include <eval_main.h>
#include <linalg.h>
#include <math.h>

using namespace vgtl;

typedef double (*func_evaluator)(const std::vector<double>* __x,
                                 const variable_indicator& __v);

struct func_eval_type
{
  const std::vector<double>* x;
  std::vector<double>* cache;
  const model* mod;
  union { void *p; double d; } u;
  double r;
  unsigned int n;
};

class func_eval :
  public cached_forward_evaluator_base<func_eval_type,expression_node,double,
                                       model::walker>
{
private:
  typedef cached_forward_evaluator_base<func_eval_type,expression_node,
                                        double, model::walker> _Base;

protected:
  bool is_cached(const node_data_type& __data)
  {
    if(__data.operator_type == EXPRINFO_LIN ||
       __data.operator_type == EXPRINFO_QUAD)
      return true;
    if(eval_data.cache && __data.n_parents > 1 && __data.n_children > 0 &&
       v_ind->match(__data.var_indicator()))
      return true;
    else
      return false;
  }

private:
  double __power(double __coeff, double __x, int __exp)
  {
    if(__exp == 0)
      return 1.;
    else
    {
      double k = __coeff*__x;
      switch(__exp)
      {
        case 1:
          return k;
          break;
        case 2:
          return k*k;
          break;
        case -1:
          return 1./k;
          break;
        case -2:
          return 1./(k*k);
          break;
        default:
          if(__exp & 1) // is odd
          {
            if(k < 0)
              return -pow(-k, __exp);
            else
              return pow(k, __exp);
          }
          else // even
            return pow(fabs(k), __exp);
          break;
      }
    }
  }

public:
  func_eval(const std::vector<double>& __x, const variable_indicator& __v,
            const model& __m, std::vector<double>* __c) : _Base()
  {
    eval_data.x = &__x;
    eval_data.cache = __c;
    eval_data.mod = &__m;
    eval_data.u.d = 0.;
    v_ind = &__v;
    eval_data.n = 0;
  }

  func_eval(const func_eval& __v) : _Base(__v) {}
  
  ~func_eval() {}

  model::walker short_cut_to(const expression_node& __data)
  { return eval_data.mod->node(0); }    // return anything
  

  void new_point(const std::vector<double>& __x, const variable_indicator& __v)
  {
    eval_data.x = &__x;
    v_ind = &__v;
  }

  void initialize() { return; }
  
  int initialize(const expression_node& __data)
  {
    eval_data.n = 0;
    if(__data.ev != NULL && (*__data.ev)[FUNC_EVALUATOR] != NULL)
    // is there a short-cut evaluator defined?
    {
      eval_data.r = (*(func_evaluator)(*__data.ev)[FUNC_EVALUATOR])(eval_data.x,
                                                                 *v_ind);
      return 0;     // don't perform the remaining graph walk
    }
    else
    {
      switch(__data.operator_type)
      {
        case EXPRINFO_VARIABLE:
          eval_data.r = (*eval_data.x)[__data.params.nn()];
          break;
        case EXPRINFO_SUM:
        case EXPRINFO_PROD:
        case EXPRINFO_MAX:
        case EXPRINFO_MIN:
        case EXPRINFO_INVERT:
        case EXPRINFO_DIV:
        case EXPRINFO_CONSTANT:
          eval_data.r = __data.params.nd();
          break;
        case EXPRINFO_MONOME:
        case EXPRINFO_IN:
        case EXPRINFO_AND:
        case EXPRINFO_NOGOOD:
          eval_data.r = 1.;
          break;
        case EXPRINFO_ALLDIFF:
          eval_data.u.p = (void*) new std::vector<double>;
          ((std::vector<double>*)eval_data.u.p)->reserve(__data.n_children);
          // no break!
        case EXPRINFO_MEAN:
        case EXPRINFO_IF:
        case EXPRINFO_OR:
        case EXPRINFO_NOT:
        case EXPRINFO_COUNT:
        case EXPRINFO_SCPROD:
        case EXPRINFO_NORM:
        case EXPRINFO_LEVEL:
          eval_data.r = 0.;
          break;
        case EXPRINFO_DET:
        case EXPRINFO_PSD:
          // here construct square double matrix
          break;
        case EXPRINFO_COND:
        case EXPRINFO_FEM:
        case EXPRINFO_MPROD:
          // here construct rectangular double matrix
          break;
      }
      return 1;
    }
  }

  void calculate(const expression_node& __data)
  {
    if(__data.operator_type > 0)
    {
      eval_data.r = __data.f_evaluate(-1, __data.params.nn(), *eval_data.x,
                                      *v_ind, eval_data.r, 0, NULL);
    }
  }

  void retrieve_from_cache(const expression_node& __data)
  {
    if(__data.operator_type == EXPRINFO_LIN)
      eval_data.r = linalg_dot(eval_data.mod->lin[__data.params.nn()],
                               *eval_data.x,0.);
    else if(__data.operator_type == EXPRINFO_QUAD)
    {
      std::vector<double> irslt;

      linalg_matvec(__data.params.m(), *eval_data.x, irslt);
      eval_data.r = irslt.back();
      irslt.pop_back();
      eval_data.r += linalg_dot(irslt,*eval_data.x,0.);
    }
    else
      eval_data.r = (*eval_data.cache)[__data.node_num];
  }

  int update(const double& __rval)
  {
    eval_data.r = __rval;
    return 0;
  }

  int update(const expression_node& __data, const double& __rval)
  {
    double __x;
    int ret = 0;
    if(__data.operator_type < 0)
    {
      switch(__data.operator_type)
      {
        case EXPRINFO_CONSTANT:
        case EXPRINFO_VARIABLE:
          break;
        case EXPRINFO_SUM:
        case EXPRINFO_MEAN:
          eval_data.r += __data.coeffs[eval_data.n++]*__rval;
          break;
        case EXPRINFO_PROD:
          // the coefficients MUST be collected in __data.params.nd()
          eval_data.r *= __rval;
          if(eval_data.r == 0) ret=-1;
          break;
        case EXPRINFO_MONOME:
          eval_data.r *= __power(__data.coeffs[eval_data.n],
                        __rval, __data.params.n()[eval_data.n]);
          if(eval_data.r == 0) ret=-1;
          eval_data.n++;
          break;
        case EXPRINFO_MAX:
          __x = __rval * __data.coeffs[eval_data.n++];
          if(__x > eval_data.r)
            eval_data.r = __x;
          break;
        case EXPRINFO_MIN:
          __x = __rval * __data.coeffs[eval_data.n++];
          if(__x < eval_data.r)
            eval_data.r = __x;
          break;
        case EXPRINFO_SCPROD:
          { double h = __data.coeffs[eval_data.n]*__rval;
            // if the number of children is odd, the list is implicitly
            // lengthened by one null element (i.e. implicitly shortened)
            if(eval_data.n & 1) // y_n
              eval_data.r += __data.coeffs[eval_data.n]*__rval*h;
            else // x_n
              eval_data.u.d = h;
          }
          eval_data.n++;
          break;
        case EXPRINFO_NORM:
          if(__data.params.nd() == INFINITY)
          {
            __x = fabs(__rval*__data.coeffs[eval_data.n]);
            if(__x > eval_data.r)
              eval_data.r = __x;
          }
          else
          { __x = fabs(__data.coeffs[eval_data.n]*__rval);
            eval_data.r += pow(__x, __data.params.nd());
          }
          eval_data.n++;
          if(eval_data.n == __data.n_children && __data.params.nd() != INFINITY)
            eval_data.r = pow(eval_data.r, 1./__data.params.nd());
          break;
        case EXPRINFO_INVERT:
          eval_data.r /= __rval;
          break;
        case EXPRINFO_DIV:
          // this evaluator requires, that the second coefficient
          // is put into the first.
          if(eval_data.n == 0)
          {
            eval_data.r *= __rval;
            if(eval_data.r == 0) ret=-1;
          }
          else
            eval_data.r /= __rval;
          ++eval_data.n;
          break;
        case EXPRINFO_SQUARE:
          __x = __data.coeffs[0]*__rval+__data.params.nd();
          eval_data.r = __x*__x;
          break;
        case EXPRINFO_INTPOWER:
          eval_data.r = __power(__data.coeffs[0], __rval, __data.params.nn());
          break;
        case EXPRINFO_SQROOT:
          eval_data.r = sqrt(__data.coeffs[0]*__rval+__data.params.nd());
          break;
        case EXPRINFO_ABS:
          eval_data.r = fabs(__data.coeffs[0]*__rval+__data.params.nd());
          break;
        case EXPRINFO_POW:
          __x = __rval * __data.coeffs[eval_data.n];
          if(eval_data.n == 0)
            eval_data.r = __x+__data.params.nd();
          else
            eval_data.r = pow(eval_data.r, __x);
          ++eval_data.n;
          break;
        case EXPRINFO_EXP:
          eval_data.r = exp(__rval*__data.coeffs[0]+__data.params.nd());
          break;
        case EXPRINFO_LOG:
          eval_data.r = log(__rval*__data.coeffs[0]+__data.params.nd());
          break;
        case EXPRINFO_SIN:
          eval_data.r = sin(__rval*__data.coeffs[0]+__data.params.nd());
          break;
        case EXPRINFO_COS:
          eval_data.r = cos(__rval*__data.coeffs[0]+__data.params.nd());
          break;
        case EXPRINFO_ATAN2:
          __x = __rval * __data.coeffs[eval_data.n];
          if(eval_data.n == 0)
            eval_data.r = __x;
          else
            eval_data.r = atan2(eval_data.r, __x);
          ++eval_data.n;
          break;
        case EXPRINFO_GAUSS:
          __x = (__data.coeffs[0]*__rval-__data.params.d()[0])/
                                __data.params.d()[1];
          eval_data.r = exp(-__x*__x);
          break;
        case EXPRINFO_POLY:
          std::cerr << "Polynomes NYI" << std::endl;
          throw "NYI";
          break;
        case EXPRINFO_LIN:
        case EXPRINFO_QUAD:
          /* we will never come here! */
          break;
        case EXPRINFO_IN:
          {
            __x = __data.coeffs[eval_data.n]*__rval;
            const interval& i(__data.params.i()[eval_data.n]);
            if(eval_data.r != -1 && i.contains(__x))
            {
              if(eval_data.r == 1 && (__x == i.inf() || __x == i.sup()))
                eval_data.r = 0;
            }
            else
            {
              eval_data.r = -1;
              ret = -1;
            }
          }
          eval_data.n++;
          break;
        case EXPRINFO_IF:
          __x = __rval * __data.coeffs[eval_data.n];
          if(eval_data.n == 0) // this is the condition
          {
            const interval& i(__data.params.ni());
            if(!i.contains(__x)) // go to else part
              ret = 1;           // don't walk the next child
          }
          else // this is either the if or the else part
          {
            eval_data.r = __x;
            ret = -1;            // don't continue walking
          }
          eval_data.n += ret+1;
          break;
        case EXPRINFO_AND:
          { __x = __data.coeffs[eval_data.n]*__rval;
            const interval& i(__data.params.i()[eval_data.n]);
            if(eval_data.r == 1 && !i.contains(__x))
            {
              eval_data.r = 0;
              ret = -1;         // result cannot become more false
            }
          }
          eval_data.n++;
          break;
        case EXPRINFO_OR:
          { __x = __data.coeffs[eval_data.n]*__rval;
            const interval& i(__data.params.i()[eval_data.n]);
            if(eval_data.r == 0 && i.contains(__x))
            {
              eval_data.r = 1;
              ret = -1;         // result cannot become more true
            }
          }
          eval_data.n++;
          break;
        case EXPRINFO_NOT:
          { __x = __data.coeffs[0]*__rval;
            const interval& i(__data.params.ni());
            if(i.contains(__x))
              eval_data.r = 0;
            else
              eval_data.r = 1;
          }
          break;
        case EXPRINFO_IMPLIES:
          { const interval& i(__data.params.i()[eval_data.n]);
            __x = __rval * __data.coeffs[eval_data.n];
            if(eval_data.n == 0)
            {
              if(!i.contains(__x))
              {
                eval_data.r = 1;
                ret = -1;        // ex falso quodlibet
              }
            }
            else // we only get here if first clause is true
              eval_data.r = i.contains(__x) ? 1 : 0;
            ++eval_data.n;
          }
          break;
        case EXPRINFO_COUNT:
          { __x = __data.coeffs[eval_data.n]*__rval;
            const interval& i(__data.params.i()[eval_data.n]);
            if(i.contains(__x))
              eval_data.r += 1;
          }
          eval_data.n++;
          break;
        case EXPRINFO_ALLDIFF:
          { __x = __data.coeffs[eval_data.n]*__rval;
            for(std::vector<double>::const_iterator _b =
                        ((std::vector<double>*)eval_data.u.p)->begin();
                _b != ((std::vector<double>*)eval_data.u.p)->end(); ++_b)
            {
              if(fabs(__x-*_b) <= __data.params.nd())
              {
                eval_data.r = 0;
                ret = -1;
                break;
              }
            }
            if(ret != -1)
              ((std::vector<double>*) eval_data.u.p)->push_back(__x);
          }
          eval_data.n++;
          if(eval_data.n == __data.n_children || ret == -1)
            delete (std::vector<double>*) eval_data.u.p;
          break;
        case EXPRINFO_HISTOGRAM:
          std::cerr << "func_evaluator: histogram NYI" << std::endl;
          throw "NYI";
          break;
        case EXPRINFO_LEVEL:
          { int h = (int)eval_data.r;
            __x = __data.coeffs[eval_data.n]*__rval;
            interval _h;

            if(h != INT_MAX)
            {
              while(h < __data.params.im().nrows())
              { 
                _h = __data.params.im()[h][eval_data.n];
                if(_h.contains(__x))
                  break;
                h++;
              }
              if(h == __data.params.im().nrows())
              {
                ret = -1;
                eval_data.r = INT_MAX;
              }
              else
                eval_data.r = h;
            }
          }
          eval_data.n++;
          break;
        case EXPRINFO_NEIGHBOR:
          if(eval_data.n == 0)
            eval_data.r = __data.coeffs[0]*__rval;
          else
          {
            double h = eval_data.r;
            eval_data.r = 0;
            __x = __data.coeffs[1]*__rval;
            for(unsigned int i = 0; i < __data.params.n().size(); i+=2)
            {
              if(h == __data.params.n()[i] && __x == __data.params.n()[i+1])
              {
                eval_data.r = 1;
                break;
              }
            }
          }
          eval_data.n++;
          break;
        case EXPRINFO_NOGOOD:
          {
            __x = __data.coeffs[eval_data.n]*__rval;
            if(eval_data.r == 0 || __data.params.n()[eval_data.n] != __x)
            {
              eval_data.r = 0;
              ret = -1;
            }
          }
          eval_data.n++;
          break;
        case EXPRINFO_EXPECTATION:
          std::cerr << "func_evaluator: E NYI" << std::endl;
          throw "NYI";
          break;
        case EXPRINFO_INTEGRAL:
          std::cerr << "func_evaluator: INT NYI" << std::endl;
          throw "NYI";
          break;
        case EXPRINFO_LOOKUP:
        case EXPRINFO_PWLIN:
        case EXPRINFO_SPLINE:
        case EXPRINFO_PWCONSTLC:
        case EXPRINFO_PWCONSTRC:
          std::cerr << "func_evaluator: Table Operations NYI" << std::endl;
          throw "NYI";
          break;
        case EXPRINFO_DET:
        case EXPRINFO_COND:
        case EXPRINFO_PSD:
        case EXPRINFO_MPROD:
        case EXPRINFO_FEM:
          std::cerr << "func_evaluator: Matrix Operations NYI" << std::endl;
          throw "NYI";
          break;
        case EXPRINFO_RE:
        case EXPRINFO_IM:
        case EXPRINFO_ARG:
        case EXPRINFO_CPLXCONJ:
          std::cerr << "func_evaluator: Complex Numbers NYI" << std::endl;
          throw "NYI";
          break;
        case EXPRINFO_CMPROD:
        case EXPRINFO_CGFEM:
          std::cerr << "func_evaluator: Const Matrix Operations NYI" << std::endl;
          throw "NYI";
          break;
        default:
          std::cerr << "func_evaluator: unknown function type " <<
            __data.operator_type << std::endl;
          throw "Programming error";
          break;
      }
    }
    else if(__data.operator_type > 0)
      // update the function arguments
      eval_data.r = __data.f_evaluate(eval_data.n++, __data.params.nn(),
                                      *eval_data.x, *v_ind, eval_data.r,
                                      __rval, NULL);
    // use the cache only if it is worthwhile
    if(eval_data.cache && __data.n_parents > 1 && __data.n_children > 0)
      (*eval_data.cache)[__data.node_num] = eval_data.r;
    return ret;
  }

  double calculate_value(bool eval_all)
  {
    return eval_data.r;
  }
};

#endif /* _FUNC_EVALUATOR_H_ */

---