FunDef.h

/*          
 *             Copyright (C) 2005 Maarten Keijzer
 *
 *          This program is free software; you can redistribute it and/or modify
 *          it under the terms of version 2 of the GNU General Public License as 
 *          published by the Free Software Foundation. 
 *
 *          This program 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
 *          GNU General Public License for more details.
 *
 *          You should have received a copy of the GNU General Public License
 *          along with this program; if not, write to the Free Software
 *          Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#ifndef FUNCTION_DEF_H_
#define FUNCTION_DEF_H_

#include <string>
#include <vector>
#include <memory>
#include <iostream>

#include "Sym.h"
#include "Interval.h"

class FunDef {
    public:
    
    virtual ~FunDef() {}
    
    // (possibly) lazy evaluation function, default implementation calls 'eager' eval
    virtual double eval(const SymVec&   args, const std::vector<double>& inputs) const;

    // eager evaluation function 
    virtual double eval(const std::vector<double>& args, const std::vector<double>& inputs) const = 0;
   
    // interval evaluation
    virtual Interval eval(const std::vector<Interval>& args, const std::vector<Interval>& inputs) const = 0; 
    
    // prints 'c' like code
    virtual std::string c_print(const std::vector<std::string>&   names, const std::vector<std::string>& names) const = 0;

    virtual unsigned min_arity() const = 0;
    virtual bool has_varargs() const { return false; } // sum, prod, min, max are variable arity

    virtual std::string name() const = 0;
    
    protected:
    
};

extern std::vector<const FunDef*> get_defined_functions();

extern const FunDef& get_element(token_t token);

extern double      eval(const Sym& sym, const std::vector<double>& inputs);

extern Interval eval(const Sym& sym, const std::vector<Interval>& inputs);

extern std::string c_print(const Sym& sym);

extern std::string c_print(const Sym& sym, const std::vector<std::string>& var_names);

inline std::ostream& operator<<(std::ostream& os, const Sym& sym) { return os << c_print(sym); }

/* Support for Ephemeral Random Constants (ERC) */

extern Sym SymConst(double value);
extern Sym SymVar(unsigned idx);

extern Sym SymLambda(Sym expression);

extern Sym SymUnlambda(Sym sym);

extern Sym expand_all(Sym sym);
extern Sym compress(Sym sym);

std::vector<double> get_constants(Sym sym);

Sym set_constants(Sym sym, const std::vector<double>& constants);

extern bool is_constant(token_t token);
extern double get_constant_value(token_t token);
extern bool is_variable(token_t token);
extern unsigned get_variable_index(token_t token);

extern bool is_lambda(token_t token);


extern Sym simplify(Sym sym);

extern Sym differentiate(Sym sym, token_t dx);
struct differentiation_error{}; // thrown in case of ifltz

/* Add function to the language table (and take a guess at the arity) */
class LanguageTable;
extern void add_function_to_table(LanguageTable& table, token_t token);

enum {
    sum_token,
    prod_token,
    inv_token,
    min_token,
    pow_token,
    ifltz_token,
    sin_token, cos_token, tan_token,
    asin_token, acos_token, atan_token,
    sinh_token, cosh_token, tanh_token,
    acosh_token, asinh_token, atanh_token,
    exp_token, log_token,
    sqr_token, sqrt_token
};

/* Defition of function overloads: for example, this define the function 'Sym sin(Sym)' */

#define HEADERFUNC(name) inline Sym name(Sym arg) { return Sym(name##_token, arg); }

/* This defines the tokens: sin_token, cos_token, etc. */
HEADERFUNC(inv);
HEADERFUNC(sin);
HEADERFUNC(cos);
HEADERFUNC(tan);
HEADERFUNC(asin);
HEADERFUNC(acos);
HEADERFUNC(atan);

HEADERFUNC(sinh);
HEADERFUNC(cosh);
HEADERFUNC(tanh);
HEADERFUNC(asinh);
HEADERFUNC(acosh);
HEADERFUNC(atanh);

HEADERFUNC(exp);
HEADERFUNC(log);

HEADERFUNC(sqr);
HEADERFUNC(sqrt);

/* Get the prototype functions out, this is needed for compilation */
extern std::string get_prototypes();

// reading and writing in internal format, no parser for symbolic functions implemented yet
extern std::string write_raw(const Sym& sym);
extern void write_raw(std::ostream& os, const Sym& sym);
extern Sym read_raw(std::string str);
extern Sym read_raw(std::istream& is); 
extern void read_raw(std::istream& is, Sym& sym); 

#include "SymOps.h"

#endif

---