X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?p=ginac.git;a=blobdiff_plain;f=ginac%2Ffunction.pl;h=68673ddc38e9fb0032eefa1fb733a17f6f623201;hp=988206e9265ebfb231c607b1d75eb3bd4a321525;hb=d03664462aedcf6f300f26c6492b3ca3327a640f;hpb=2db113296f65f8fd3a0d62781ddc9276ff07b487 diff --git a/ginac/function.pl b/ginac/function.pl index 988206e9..68673ddc 100755 --- a/ginac/function.pl +++ b/ginac/function.pl @@ -1,269 +1,171 @@ -#!/usr/bin/perl -w - $maxargs=13; sub generate_seq { - my ($seq_template,$n)=@_; - my ($res,$N); - - $res=''; - for ($N=1; $N<=$n; $N++) { - $res .= eval('"' . $seq_template . '"'); - if ($N!=$n) { - $res .= ', '; - } - } - return $res; + my ($seq_template,$n)=@_; + my ($res,$N); + + $res=''; + for ($N=1; $N<=$n; $N++) { + $res .= eval('"' . $seq_template . '"'); + if ($N!=$n) { + $res .= ', '; + } + } + return $res; } sub generate_from_to { - my ($template,$seq_template1,$seq_template2,$from,$to)=@_; - my ($res,$N,$SEQ); - - $res=''; - for ($N=$from; $N<=$to; $N++) { - $SEQ1=generate_seq($seq_template1,$N); - $SEQ2=generate_seq($seq_template2,$N); - $res .= eval('"' . $template . '"'); - $SEQ1=''; # to avoid main::SEQ1 used only once warning - $SEQ2=''; # same as above - } - return $res; + my ($template,$seq_template1,$seq_template2,$seq_template3,$from,$to)=@_; + my ($res,$N,$SEQ); + + $res=''; + for ($N=$from; $N<=$to; $N++) { + $SEQ1=generate_seq($seq_template1,$N); + $SEQ2=generate_seq($seq_template2,$N); + $SEQ3=generate_seq($seq_template3,$N); + $res .= eval('"' . $template . '"'); + $SEQ1=''; # to avoid main::SEQ1 used only once warning + $SEQ2=''; # same as above + $SEQ3=''; # same as above + } + return $res; } sub generate { - my ($template,$seq_template1,$seq_template2)=@_; - return generate_from_to($template,$seq_template1,$seq_template2,1,$maxargs); -} - -$declare_function_macro_namespace = <<'END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NAMESPACE'; -#ifdef CINT_CONVERSION_WORKAROUND - -#define DECLARE_FUNCTION_1P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline GiNaC::function NAME(const GiNaC::ex & p1) { \ - return GiNaC::function(function_index_##NAME, p1); \ -} \ -inline GiNaC::function NAME(const GiNaC::basic & p1) { \ - return GiNaC::function(function_index_##NAME, GiNaC::ex(p1)); \ -} -#define DECLARE_FUNCTION_2P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline GiNaC::function NAME(const GiNaC::ex & p1, const GiNaC::ex & p2) { \ - return GiNaC::function(function_index_##NAME, p1, p2); \ -} \ -inline GiNaC::function NAME(const GiNaC::basic & p1, const GiNaC::ex & p2) { \ - return GiNaC::function(function_index_##NAME, GiNaC::ex(p1), p2); \ -} \ -inline GiNaC::function NAME(const GiNaC::ex & p1, const GiNaC::basic & p2) { \ - return GiNaC::function(function_index_##NAME, p1, GiNaC::ex(p2)); \ -} \ -inline GiNaC::function NAME(const GiNaC::basic & p1, const GiNaC::basic & p2) { \ - return GiNaC::function(function_index_##NAME, GiNaC::ex(p1), GiNaC::ex(p2)); \ -} - -#else // def CINT_CONVERSION_WORKAROUND - -#define DECLARE_FUNCTION_1P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline GiNaC::function NAME(const GiNaC::ex & p1) { \ - return GiNaC::function(function_index_##NAME, p1); \ -} -#define DECLARE_FUNCTION_2P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline GiNaC::function NAME(const GiNaC::ex & p1, const GiNaC::ex & p2) { \ - return GiNaC::function(function_index_##NAME, p1, p2); \ -} - -#endif // def CINT_CONVERSION_WORKAROUND - -END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NAMESPACE - -$declare_function_macro_namespace .= generate_from_to( - <<'END_OF_DECLARE_FUNCTION_MACRO_NAMESPACE','const GiNaC::ex & p${N}','p${N}',3,$maxargs); -#define DECLARE_FUNCTION_${N}P(NAME) \\ -extern const unsigned function_index_##NAME; \\ -inline GiNaC::function NAME(${SEQ1}) { \\ - return GiNaC::function(function_index_##NAME, ${SEQ2}); \\ -} - -END_OF_DECLARE_FUNCTION_MACRO_NAMESPACE - -$declare_function_macro_no_namespace = <<'END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NO_NAMESPACE'; -#ifdef CINT_CONVERSION_WORKAROUND - -#define DECLARE_FUNCTION_1P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline function NAME(const ex & p1) { \ - return function(function_index_##NAME, p1); \ -} \ -inline function NAME(const basic & p1) { \ - return function(function_index_##NAME, ex(p1)); \ -} -#define DECLARE_FUNCTION_2P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline function NAME(const ex & p1, const ex & p2) { \ - return function(function_index_##NAME, p1, p2); \ -} \ -inline function NAME(const basic & p1, const ex & p2) { \ - return function(function_index_##NAME, ex(p1), p2); \ -} \ -inline function NAME(const ex & p1, const basic & p2) { \ - return function(function_index_##NAME, p1, ex(p2)); \ -} \ -inline function NAME(const basic & p1, const basic & p2) { \ - return function(function_index_##NAME, ex(p1), ex(p2)); \ -} - -#else // def CINT_CONVERSION_WORKAROUND - -#define DECLARE_FUNCTION_1P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline function NAME(const ex & p1) { \ - return function(function_index_##NAME, p1); \ -} -#define DECLARE_FUNCTION_2P(NAME) \ -extern const unsigned function_index_##NAME; \ -inline function NAME(const ex & p1, const ex & p2) { \ - return function(function_index_##NAME, p1, p2); \ + my ($template,$seq_template1,$seq_template2,$seq_template3)=@_; + return generate_from_to($template,$seq_template1,$seq_template2,$seq_template3,1,$maxargs); } -#endif // def CINT_CONVERSION_WORKAROUND - -END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NO_NAMESPACE - -$declare_function_macro_no_namespace .= generate_from_to( - <<'END_OF_DECLARE_FUNCTION_MACRO_NO_NAMESPACE','const ex & p${N}','p${N}',3,$maxargs); +$declare_function_macro = generate( + <<'END_OF_DECLARE_FUNCTION_MACRO','typename T${N}','const T${N} & p${N}','GiNaC::ex(p${N})'); #define DECLARE_FUNCTION_${N}P(NAME) \\ extern const unsigned function_index_##NAME; \\ -inline function NAME(${SEQ1}) { \\ - return function(function_index_##NAME, ${SEQ2}); \\ +template<${SEQ1}> \\ +inline const GiNaC::function NAME(${SEQ2}) { \\ + return GiNaC::function(function_index_##NAME, ${SEQ3}); \\ } -END_OF_DECLARE_FUNCTION_MACRO_NO_NAMESPACE +END_OF_DECLARE_FUNCTION_MACRO $typedef_eval_funcp=generate( 'typedef ex (* eval_funcp_${N})(${SEQ1});'."\n", -'const ex &',''); +'const ex &','',''); $typedef_evalf_funcp=generate( 'typedef ex (* evalf_funcp_${N})(${SEQ1});'."\n", -'const ex &',''); +'const ex &','',''); $typedef_derivative_funcp=generate( 'typedef ex (* derivative_funcp_${N})(${SEQ1}, unsigned);'."\n", -'const ex &',''); +'const ex &','',''); $typedef_series_funcp=generate( -'typedef ex (* series_funcp_${N})(${SEQ1}, const symbol &, const ex &, int);'."\n", -'const ex &',''); +'typedef ex (* series_funcp_${N})(${SEQ1}, const relational &, int, unsigned);'."\n", +'const ex &','',''); -$eval_func_interface=generate(' function_options & eval_func(eval_funcp_${N} e);'."\n",'',''); +$eval_func_interface=generate(' function_options & eval_func(eval_funcp_${N} e);'."\n",'','',''); -$evalf_func_interface=generate(' function_options & evalf_func(evalf_funcp_${N} ef);'."\n",'',''); +$evalf_func_interface=generate(' function_options & evalf_func(evalf_funcp_${N} ef);'."\n",'','',''); -$derivative_func_interface=generate(' function_options & derivative_func(derivative_funcp_${N} d);'."\n",'',''); +$derivative_func_interface=generate(' function_options & derivative_func(derivative_funcp_${N} d);'."\n",'','',''); -$series_func_interface=generate(' function_options & series_func(series_funcp_${N} s);'."\n",'',''); +$series_func_interface=generate(' function_options & series_func(series_funcp_${N} s);'."\n",'','',''); $constructors_interface=generate( ' function(unsigned ser, ${SEQ1});'."\n", -'const ex & param${N}',''); +'const ex & param${N}','',''); $constructors_implementation=generate( - <<'END_OF_CONSTRUCTORS_IMPLEMENTATION','const ex & param${N}','param${N}'); + <<'END_OF_CONSTRUCTORS_IMPLEMENTATION','const ex & param${N}','param${N}',''); function::function(unsigned ser, ${SEQ1}) - : exprseq(${SEQ2}), serial(ser) + : exprseq(${SEQ2}), serial(ser) { - debugmsg(\"function constructor from unsigned,${N}*ex\",LOGLEVEL_CONSTRUCT); - tinfo_key = TINFO_function; + debugmsg(\"function ctor from unsigned,${N}*ex\",LOGLEVEL_CONSTRUCT); + tinfo_key = TINFO_function; } END_OF_CONSTRUCTORS_IMPLEMENTATION $eval_switch_statement=generate( - <<'END_OF_EVAL_SWITCH_STATEMENT','seq[${N}-1]',''); - case ${N}: - eval_result=((eval_funcp_${N})(registered_functions()[serial].eval_f))(${SEQ1}); - break; + <<'END_OF_EVAL_SWITCH_STATEMENT','seq[${N}-1]','',''); + case ${N}: + eval_result = ((eval_funcp_${N})(registered_functions()[serial].eval_f))(${SEQ1}); + break; END_OF_EVAL_SWITCH_STATEMENT $evalf_switch_statement=generate( - <<'END_OF_EVALF_SWITCH_STATEMENT','eseq[${N}-1]',''); - case ${N}: - return ((evalf_funcp_${N})(registered_functions()[serial].evalf_f))(${SEQ1}); - break; + <<'END_OF_EVALF_SWITCH_STATEMENT','eseq[${N}-1]','',''); + case ${N}: + return ((evalf_funcp_${N})(registered_functions()[serial].evalf_f))(${SEQ1}); END_OF_EVALF_SWITCH_STATEMENT $diff_switch_statement=generate( - <<'END_OF_DIFF_SWITCH_STATEMENT','seq[${N}-1]',''); - case ${N}: - return ((derivative_funcp_${N})(registered_functions()[serial].derivative_f))(${SEQ1},diff_param); - break; + <<'END_OF_DIFF_SWITCH_STATEMENT','seq[${N}-1]','',''); + case ${N}: + return ((derivative_funcp_${N})(registered_functions()[serial].derivative_f))(${SEQ1},diff_param); END_OF_DIFF_SWITCH_STATEMENT $series_switch_statement=generate( - <<'END_OF_SERIES_SWITCH_STATEMENT','seq[${N}-1]',''); - case ${N}: - try { - res = ((series_funcp_${N})(registered_functions()[serial].series_f))(${SEQ1},s,point,order); - } catch (do_taylor) { - res = basic::series(s, point, order); - } - return res; - break; + <<'END_OF_SERIES_SWITCH_STATEMENT','seq[${N}-1]','',''); + case ${N}: + try { + res = ((series_funcp_${N})(registered_functions()[serial].series_f))(${SEQ1},r,order,options); + } catch (do_taylor) { + res = basic::series(r, order, options); + } + return res; END_OF_SERIES_SWITCH_STATEMENT $eval_func_implementation=generate( - <<'END_OF_EVAL_FUNC_IMPLEMENTATION','',''); + <<'END_OF_EVAL_FUNC_IMPLEMENTATION','','',''); function_options & function_options::eval_func(eval_funcp_${N} e) { - test_and_set_nparams(${N}); - eval_f=eval_funcp(e); - return *this; -} + test_and_set_nparams(${N}); + eval_f = eval_funcp(e); + return *this; +} END_OF_EVAL_FUNC_IMPLEMENTATION $evalf_func_implementation=generate( - <<'END_OF_EVALF_FUNC_IMPLEMENTATION','',''); + <<'END_OF_EVALF_FUNC_IMPLEMENTATION','','',''); function_options & function_options::evalf_func(evalf_funcp_${N} ef) { - test_and_set_nparams(${N}); - evalf_f=evalf_funcp(ef); - return *this; -} + test_and_set_nparams(${N}); + evalf_f = evalf_funcp(ef); + return *this; +} END_OF_EVALF_FUNC_IMPLEMENTATION $derivative_func_implementation=generate( - <<'END_OF_DERIVATIVE_FUNC_IMPLEMENTATION','',''); + <<'END_OF_DERIVATIVE_FUNC_IMPLEMENTATION','','',''); function_options & function_options::derivative_func(derivative_funcp_${N} d) { - test_and_set_nparams(${N}); - derivative_f=derivative_funcp(d); - return *this; -} + test_and_set_nparams(${N}); + derivative_f = derivative_funcp(d); + return *this; +} END_OF_DERIVATIVE_FUNC_IMPLEMENTATION $series_func_implementation=generate( - <<'END_OF_SERIES_FUNC_IMPLEMENTATION','',''); + <<'END_OF_SERIES_FUNC_IMPLEMENTATION','','',''); function_options & function_options::series_func(series_funcp_${N} s) { - test_and_set_nparams(${N}); - series_f=series_funcp(s); - return *this; -} + test_and_set_nparams(${N}); + series_f = series_funcp(s); + return *this; +} END_OF_SERIES_FUNC_IMPLEMENTATION $interface=< #include -#ifdef __CINT__ -// CINT needs to work properly with +// CINT needs to work properly with #include -#endif // def __CINT__ #include "exprseq.h" -#ifndef NO_NAMESPACE_GINAC - // the following lines have been generated for max. ${maxargs} parameters -$declare_function_macro_namespace +$declare_function_macro // end of generated lines -#else // ndef NO_NAMESPACE_GINAC - -// the following lines have been generated for max. ${maxargs} parameters -$declare_function_macro_no_namespace -// end of generated lines - -#endif // ndef NO_NAMESPACE_GINAC - -#ifndef NO_NAMESPACE_GINAC - -#define REGISTER_FUNCTION(NAME,OPT) \\ -const unsigned function_index_##NAME= \\ - GiNaC::function::register_new(GiNaC::function_options(#NAME).OPT); - -#define REGISTER_FUNCTION_OLD(NAME,E,EF,D,S) \\ -const unsigned function_index_##NAME= \\ - GiNaC::function::register_new(GiNaC::function_options(#NAME). \\ - eval_func(E). \\ - evalf_func(EF). \\ - derivative_func(D). \\ - series_func(S)); - -#else // ndef NO_NAMESPACE_GINAC - #define REGISTER_FUNCTION(NAME,OPT) \\ const unsigned function_index_##NAME= \\ - function::register_new(function_options(#NAME).OPT); - -#define REGISTER_FUNCTION_OLD(NAME,E,EF,D,S) \\ -const unsigned function_index_##NAME= \\ - function::register_new(function_options(#NAME). \\ - eval_func(E). \\ - evalf_func(EF). \\ - derivative_func(D). \\ - series_func(S)); + GiNaC::function::register_new(GiNaC::function_options(#NAME).OPT); -#endif // ndef NO_NAMESPACE_GINAC +// The TYPECHECK-macros were used inside the _evalf() functions. They are +// considered obsolete now: (FIXME: remove them) #define BEGIN_TYPECHECK \\ bool automatic_typecheck=true; #define TYPECHECK(VAR,TYPE) \\ -if (!is_ex_exactly_of_type(VAR,TYPE)) { \\ - automatic_typecheck=false; \\ +if (!is_exactly_a(VAR)) { \\ + automatic_typecheck=false; \\ } else -#ifndef NO_NAMESPACE_GINAC - #define TYPECHECK_INTEGER(VAR) \\ if (!(VAR).info(GiNaC::info_flags::integer)) { \\ - automatic_typecheck=false; \\ + automatic_typecheck=false; \\ } else -#else // ndef NO_NAMESPACE_GINAC - -#define TYPECHECK_INTEGER(VAR) \\ -if (!(VAR).info(info_flags::integer)) { \\ - automatic_typecheck=false; \\ -} else - -#endif // ndef NO_NAMESPACE_GINAC - #define END_TYPECHECK(RV) \\ {} \\ if (!automatic_typecheck) { \\ - return RV.hold(); \\ + return RV.hold(); \\ } -#ifndef NO_NAMESPACE_GINAC namespace GiNaC { -#endif // ndef NO_NAMESPACE_GINAC class function; +class symmetry; typedef ex (* eval_funcp)(); typedef ex (* evalf_funcp)(); @@ -385,155 +240,160 @@ $typedef_derivative_funcp $typedef_series_funcp // end of generated lines +// Alternatively, an exvector may be passed into the static function, instead +// of individual ex objects. Then, the number of arguments is not limited. +typedef ex (* eval_funcp_exvector)(const exvector &); +typedef ex (* evalf_funcp_exvector)(const exvector &); +typedef ex (* derivative_funcp_exvector)(const exvector &, unsigned); +typedef ex (* series_funcp_exvector)(const exvector &, const relational &, int, unsigned); + class function_options { - friend class function; + friend class function; + friend class fderivative; public: - function_options(); - function_options(string const & n, string const & tn=string()); - ~function_options(); - void initialize(void); - function_options & set_name(string const & n, string const & tn=string()); + function_options(); + function_options(std::string const & n, std::string const & tn=std::string()); + ~function_options(); + void initialize(void); + function_options & set_name(std::string const & n, std::string const & tn=std::string()); + function_options & latex_name(std::string const & tn); // the following lines have been generated for max. ${maxargs} parameters $eval_func_interface $evalf_func_interface $derivative_func_interface $series_func_interface // end of generated lines - function_options & set_return_type(unsigned rt, unsigned rtt=0); - function_options & do_not_evalf_params(void); - function_options & remember(unsigned size, unsigned assoc_size=0, - unsigned strategy=remember_strategies::delete_never); - function_options & overloaded(unsigned o); - void test_and_set_nparams(unsigned n); - string get_name(void) const { return name; } - unsigned get_nparams(void) const { return nparams; } + function_options & eval_func(eval_funcp_exvector e); + function_options & evalf_func(evalf_funcp_exvector ef); + function_options & derivative_func(derivative_funcp_exvector d); + function_options & series_func(series_funcp_exvector s); + + function_options & set_return_type(unsigned rt, unsigned rtt=0); + function_options & do_not_evalf_params(void); + function_options & remember(unsigned size, unsigned assoc_size=0, + unsigned strategy=remember_strategies::delete_never); + function_options & overloaded(unsigned o); + function_options & set_symmetry(const symmetry & s); + void test_and_set_nparams(unsigned n); + std::string get_name(void) const { return name; } + unsigned get_nparams(void) const { return nparams; } + bool has_derivative(void) const { return derivative_f != NULL; } protected: - string name; - string TeX_name; + std::string name; + std::string TeX_name; - unsigned nparams; + unsigned nparams; - eval_funcp eval_f; - evalf_funcp evalf_f; - derivative_funcp derivative_f; - series_funcp series_f; + eval_funcp eval_f; + evalf_funcp evalf_f; + derivative_funcp derivative_f; + series_funcp series_f; - bool evalf_params_first; + bool evalf_params_first; - bool use_return_type; - unsigned return_type; - unsigned return_type_tinfo; + bool use_return_type; + unsigned return_type; + unsigned return_type_tinfo; - bool use_remember; - unsigned remember_size; - unsigned remember_assoc_size; - unsigned remember_strategy; + bool use_remember; + unsigned remember_size; + unsigned remember_assoc_size; + unsigned remember_strategy; - unsigned functions_with_same_name; + bool eval_use_exvector_args; + bool evalf_use_exvector_args; + bool derivative_use_exvector_args; + bool series_use_exvector_args; + + unsigned functions_with_same_name; + + ex symtree; }; /** The class function is used to implement builtin functions like sin, cos... - and user defined functions */ + and user defined functions */ class function : public exprseq { - GINAC_DECLARE_REGISTERED_CLASS(function, exprseq) + GINAC_DECLARE_REGISTERED_CLASS(function, exprseq) - // CINT has a linking problem - friend void ginsh_get_ginac_functions(void); + // CINT has a linking problem +#ifndef __MAKECINT__ + friend void ginsh_get_ginac_functions(void); +#endif // def __MAKECINT__ - friend class remember_table_entry; - // friend class remember_table_list; - // friend class remember_table; + friend class remember_table_entry; + // friend class remember_table_list; + // friend class remember_table; // member functions - // default constructor, destructor, copy constructor assignment operator and helpers + // other ctors public: - function(); - ~function(); - function(const function & other); - const function & operator=(const function & other); -protected: - void copy(const function & other); - void destroy(bool call_parent); - - // other constructors -public: - function(unsigned ser); - // the following lines have been generated for max. ${maxargs} parameters + function(unsigned ser); + // the following lines have been generated for max. ${maxargs} parameters $constructors_interface - // end of generated lines - function(unsigned ser, const exprseq & es); - function(unsigned ser, const exvector & v, bool discardable=0); - function(unsigned ser, exvector * vp); // vp will be deleted + // end of generated lines + function(unsigned ser, const exprseq & es); + function(unsigned ser, const exvector & v, bool discardable = false); + function(unsigned ser, exvector * vp); // vp will be deleted - // functions overriding virtual functions from bases classes + // functions overriding virtual functions from base classes public: - basic * duplicate() const; - void printraw(ostream & os) const; - void print(ostream & os, unsigned upper_precedence=0) const; - void printtree(ostream & os, unsigned indent) const; - void printcsrc(ostream & os, unsigned type, unsigned upper_precedence=0) const; - ex expand(unsigned options=0) const; - ex eval(int level=0) const; - ex evalf(int level=0) const; - ex series(const symbol & s, const ex & point, int order) const; - ex thisexprseq(const exvector & v) const; - ex thisexprseq(exvector * vp) const; + void print(const print_context & c, unsigned level = 0) const; + unsigned precedence(void) const {return 70;} + int degree(const ex & s) const; + int ldegree(const ex & s) const; + ex coeff(const ex & s, int n = 1) const; + ex expand(unsigned options=0) const; + ex eval(int level=0) const; + ex evalf(int level=0) const; + unsigned calchash(void) const; + ex series(const relational & r, int order, unsigned options = 0) const; + ex thisexprseq(const exvector & v) const; + ex thisexprseq(exvector * vp) const; protected: - ex derivative(const symbol & s) const; - int compare_same_type(const basic & other) const; - bool is_equal_same_type(const basic & other) const; - unsigned return_type(void) const; - unsigned return_type_tinfo(void) const; - - // new virtual functions which can be overridden by derived classes - // none - - // non-virtual functions in this class + ex derivative(const symbol & s) const; + bool is_equal_same_type(const basic & other) const; + bool match_same_type(const basic & other) const; + unsigned return_type(void) const; + unsigned return_type_tinfo(void) const; + + // new virtual functions which can be overridden by derived classes + // none + + // non-virtual functions in this class protected: - ex pderivative(unsigned diff_param) const; // partial differentiation - static vector & registered_functions(void); - bool lookup_remember_table(ex & result) const; - void store_remember_table(ex const & result) const; + ex pderivative(unsigned diff_param) const; // partial differentiation + static std::vector & registered_functions(void); + bool lookup_remember_table(ex & result) const; + void store_remember_table(ex const & result) const; public: - static unsigned register_new(function_options const & opt); - unsigned getserial(void) const {return serial;} - + static unsigned register_new(function_options const & opt); + static unsigned current_serial; + static unsigned find_function(const std::string &name, unsigned nparams); + unsigned get_serial(void) const {return serial;} + std::string get_name(void) const; + // member variables protected: - unsigned serial; + unsigned serial; }; // utility functions/macros -inline const function &ex_to_function(const ex &e) + +/** Specialization of is_exactly_a(obj) for objects of type function. */ +template<> inline bool is_exactly_a(const basic & obj) { - return static_cast(*e.bp); + return obj.tinfo()==TINFO_function; } -#ifndef NO_NAMESPACE_GINAC - -#define is_ex_the_function(OBJ, FUNCNAME) \\ - (is_ex_exactly_of_type(OBJ, function) && static_cast(OBJ.bp)->getserial() == function_index_##FUNCNAME) - -#else // ndef NO_NAMESPACE_GINAC - #define is_ex_the_function(OBJ, FUNCNAME) \\ - (is_ex_exactly_of_type(OBJ, function) && static_cast(OBJ.bp)->getserial() == function_index_##FUNCNAME) + (is_exactly_a(OBJ) && ex_to(OBJ).get_serial() == function_index_##FUNCNAME) -#endif // ndef NO_NAMESPACE_GINAC - -// global constants - -extern const function some_function; -extern const type_info & typeid_function; - -#ifndef NO_NAMESPACE_GINAC } // namespace GiNaC -#endif // ndef NO_NAMESPACE_GINAC #endif // ndef __GINAC_FUNCTION_H__ @@ -542,14 +402,14 @@ END_OF_INTERFACE $implementation=< #include "function.h" +#include "fderivative.h" #include "ex.h" +#include "lst.h" +#include "symmetry.h" +#include "print.h" #include "archive.h" #include "inifcns.h" +#include "tostring.h" #include "utils.h" #include "debugmsg.h" #include "remember.h" -#ifndef NO_NAMESPACE_GINAC namespace GiNaC { -#endif // ndef NO_NAMESPACE_GINAC ////////// // helper class function_options @@ -588,41 +451,51 @@ namespace GiNaC { function_options::function_options() { - initialize(); + initialize(); } -function_options::function_options(string const & n, string const & tn) +function_options::function_options(std::string const & n, std::string const & tn) { - initialize(); - set_name(n,tn); + initialize(); + set_name(n,tn); } function_options::~function_options() { - // nothing to clean up at the moment + // nothing to clean up at the moment } void function_options::initialize(void) { - set_name("unnamed_function","\\\\operatorname{unnamed}"); - nparams=0; - eval_f=evalf_f=derivative_f=series_f=0; - evalf_params_first=true; - use_return_type=false; - use_remember=false; - functions_with_same_name=1; + set_name("unnamed_function","\\\\mbox{unnamed}"); + nparams = 0; + eval_f = evalf_f = derivative_f = series_f = 0; + evalf_params_first = true; + use_return_type = false; + eval_use_exvector_args = false; + evalf_use_exvector_args = false; + derivative_use_exvector_args = false; + series_use_exvector_args = false; + use_remember = false; + functions_with_same_name = 1; + symtree = 0; +} + +function_options & function_options::set_name(std::string const & n, + std::string const & tn) +{ + name = n; + if (tn==std::string()) + TeX_name = "\\\\mbox{"+name+"}"; + else + TeX_name = tn; + return *this; } -function_options & function_options::set_name(string const & n, - string const & tn) +function_options & function_options::latex_name(std::string const & tn) { - name=n; - if (tn==string()) { - TeX_name="\\\\operatorname{"+name+"}"; - } else { - TeX_name=tn; - } - return *this; + TeX_name = tn; + return *this; } // the following lines have been generated for max. ${maxargs} parameters @@ -632,110 +505,125 @@ $derivative_func_implementation $series_func_implementation // end of generated lines +function_options& function_options::eval_func(eval_funcp_exvector e) +{ + eval_use_exvector_args = true; + eval_f = eval_funcp(e); + return *this; +} +function_options& function_options::evalf_func(evalf_funcp_exvector ef) +{ + evalf_use_exvector_args = true; + evalf_f = evalf_funcp(ef); + return *this; +} +function_options& function_options::derivative_func(derivative_funcp_exvector d) +{ + derivative_use_exvector_args = true; + derivative_f = derivative_funcp(d); + return *this; +} +function_options& function_options::series_func(series_funcp_exvector s) +{ + series_use_exvector_args = true; + series_f = series_funcp(s); + return *this; +} + + function_options & function_options::set_return_type(unsigned rt, unsigned rtt) { - use_return_type=true; - return_type=rt; - return_type_tinfo=rtt; - return *this; + use_return_type = true; + return_type = rt; + return_type_tinfo = rtt; + return *this; } function_options & function_options::do_not_evalf_params(void) { - evalf_params_first=false; - return *this; + evalf_params_first = false; + return *this; } function_options & function_options::remember(unsigned size, unsigned assoc_size, unsigned strategy) { - use_remember=true; - remember_size=size; - remember_assoc_size=assoc_size; - remember_strategy=strategy; - return *this; + use_remember = true; + remember_size = size; + remember_assoc_size = assoc_size; + remember_strategy = strategy; + return *this; } function_options & function_options::overloaded(unsigned o) { - functions_with_same_name=o; - return *this; + functions_with_same_name = o; + return *this; } - + +function_options & function_options::set_symmetry(const symmetry & s) +{ + symtree = s; + return *this; +} + void function_options::test_and_set_nparams(unsigned n) { - if (nparams==0) { - nparams=n; - } else if (nparams!=n) { - // we do not throw an exception here because this code is - // usually executed before main(), so the exception could not - // caught anyhow - cerr << "WARNING: number of parameters (" - << n << ") differs from number set before (" - << nparams << ")" << endl; - } + if (nparams==0) { + nparams = n; + } else if (nparams!=n) { + // we do not throw an exception here because this code is + // usually executed before main(), so the exception could not + // caught anyhow + std::cerr << "WARNING: number of parameters (" + << n << ") differs from number set before (" + << nparams << ")" << std::endl; + } } +/** This can be used as a hook for external applications. */ +unsigned function::current_serial = 0; + + GINAC_IMPLEMENT_REGISTERED_CLASS(function, exprseq) ////////// -// default constructor, destructor, copy constructor assignment operator and helpers +// default ctor, dtor, copy ctor assignment operator and helpers ////////// // public function::function() : serial(0) { - debugmsg("function default constructor",LOGLEVEL_CONSTRUCT); - tinfo_key = TINFO_function; -} - -function::~function() -{ - debugmsg("function destructor",LOGLEVEL_DESTRUCT); - destroy(0); -} - -function::function(const function & other) -{ - debugmsg("function copy constructor",LOGLEVEL_CONSTRUCT); - copy(other); -} - -const function & function::operator=(const function & other) -{ - debugmsg("function operator=",LOGLEVEL_ASSIGNMENT); - if (this != &other) { - destroy(1); - copy(other); - } - return *this; + debugmsg("function default ctor",LOGLEVEL_CONSTRUCT); + tinfo_key = TINFO_function; } // protected void function::copy(const function & other) { - exprseq::copy(other); - serial=other.serial; + inherited::copy(other); + serial = other.serial; } void function::destroy(bool call_parent) { - if (call_parent) exprseq::destroy(call_parent); + if (call_parent) + inherited::destroy(call_parent); } ////////// -// other constructors +// other ctors ////////// // public function::function(unsigned ser) : serial(ser) { - debugmsg("function constructor from unsigned",LOGLEVEL_CONSTRUCT); - tinfo_key = TINFO_function; + debugmsg("function ctor from unsigned",LOGLEVEL_CONSTRUCT); + tinfo_key = TINFO_function; } // the following lines have been generated for max. ${maxargs} parameters @@ -744,22 +632,22 @@ $constructors_implementation function::function(unsigned ser, const exprseq & es) : exprseq(es), serial(ser) { - debugmsg("function constructor from unsigned,exprseq",LOGLEVEL_CONSTRUCT); - tinfo_key = TINFO_function; + debugmsg("function ctor from unsigned,exprseq",LOGLEVEL_CONSTRUCT); + tinfo_key = TINFO_function; } function::function(unsigned ser, const exvector & v, bool discardable) - : exprseq(v,discardable), serial(ser) + : exprseq(v,discardable), serial(ser) { - debugmsg("function constructor from string,exvector,bool",LOGLEVEL_CONSTRUCT); - tinfo_key = TINFO_function; + debugmsg("function ctor from string,exvector,bool",LOGLEVEL_CONSTRUCT); + tinfo_key = TINFO_function; } function::function(unsigned ser, exvector * vp) - : exprseq(vp), serial(ser) + : exprseq(vp), serial(ser) { - debugmsg("function constructor from unsigned,exvector *",LOGLEVEL_CONSTRUCT); - tinfo_key = TINFO_function; + debugmsg("function ctor from unsigned,exvector *",LOGLEVEL_CONSTRUCT); + tinfo_key = TINFO_function; } ////////// @@ -769,263 +657,327 @@ function::function(unsigned ser, exvector * vp) /** Construct object from archive_node. */ function::function(const archive_node &n, const lst &sym_lst) : inherited(n, sym_lst) { - debugmsg("function constructor from archive_node", LOGLEVEL_CONSTRUCT); - - // Find serial number by function name - string s; - if (n.find_string("name", s)) { - unsigned int ser = 0; - vector::const_iterator i = registered_functions().begin(), iend = registered_functions().end(); - while (i != iend) { - if (s == i->name) { - serial = ser; - return; - } - i++; ser++; - } - throw (std::runtime_error("unknown function '" + s + "' in archive")); - } else - throw (std::runtime_error("unnamed function in archive")); + debugmsg("function ctor from archive_node", LOGLEVEL_CONSTRUCT); + + // Find serial number by function name + std::string s; + if (n.find_string("name", s)) { + unsigned int ser = 0; + std::vector::const_iterator i = registered_functions().begin(), iend = registered_functions().end(); + while (i != iend) { + if (s == i->name) { + serial = ser; + return; + } + ++i; ++ser; + } + throw (std::runtime_error("unknown function '" + s + "' in archive")); + } else + throw (std::runtime_error("unnamed function in archive")); } /** Unarchive the object. */ ex function::unarchive(const archive_node &n, const lst &sym_lst) { - return (new function(n, sym_lst))->setflag(status_flags::dynallocated); + return (new function(n, sym_lst))->setflag(status_flags::dynallocated); } /** Archive the object. */ void function::archive(archive_node &n) const { - inherited::archive(n); - GINAC_ASSERT(serial < registered_functions().size()); - n.add_string("name", registered_functions()[serial].name); + inherited::archive(n); + GINAC_ASSERT(serial < registered_functions().size()); + n.add_string("name", registered_functions()[serial].name); } ////////// -// functions overriding virtual functions from bases classes +// functions overriding virtual functions from base classes ////////// // public -basic * function::duplicate() const +void function::print(const print_context & c, unsigned level) const { - debugmsg("function duplicate",LOGLEVEL_DUPLICATE); - return new function(*this); + debugmsg("function print", LOGLEVEL_PRINT); + + GINAC_ASSERT(serialprint(c); + ++it; + if (it != itend) + c.s << ","; + } + c.s << ")"; + + } else if (is_of_type(c, print_latex)) { + c.s << registered_functions()[serial].TeX_name; + printseq(c, '(', ',', ')', exprseq::precedence(), function::precedence()); + } else { + c.s << registered_functions()[serial].name; + printseq(c, '(', ',', ')', exprseq::precedence(), function::precedence()); + } } -void function::printraw(ostream & os) const -{ - debugmsg("function printraw",LOGLEVEL_PRINT); - - GINAC_ASSERT(serialprint(os); - } - os << ")"; -} - -void function::print(ostream & os, unsigned upper_precedence) const +ex function::expand(unsigned options) const { - debugmsg("function print",LOGLEVEL_PRINT); - - GINAC_ASSERT(serial(s)) ? 1 : 0; } -void function::printcsrc(ostream & os, unsigned type, unsigned upper_precedence) const +int function::ldegree(const ex & s) const { - debugmsg("function print csrc",LOGLEVEL_PRINT); - - GINAC_ASSERT(serialbp->printcsrc(os, type, 0); - it++; - if (it != itend) - os << ","; - } - os << ")"; + return is_equal(ex_to(s)) ? 1 : 0; } -ex function::expand(unsigned options) const +ex function::coeff(const ex & s, int n) const { - return this->setflag(status_flags::expanded); + if (is_equal(ex_to(s))) + return n==1 ? _ex1() : _ex0(); + else + return n==0 ? ex(*this) : _ex0(); } ex function::eval(int level) const { - GINAC_ASSERT(serial1) { - // first evaluate children, then we will end up here again - return function(serial,evalchildren(level)); - } - - if (registered_functions()[serial].eval_f==0) { - return this->hold(); - } - - bool use_remember=registered_functions()[serial].use_remember; - ex eval_result; - if (use_remember && lookup_remember_table(eval_result)) { - return eval_result; - } - - switch (registered_functions()[serial].nparams) { - // the following lines have been generated for max. ${maxargs} parameters + GINAC_ASSERT(serial1) { + // first evaluate children, then we will end up here again + return function(serial,evalchildren(level)); + } + + const function_options &opt = registered_functions()[serial]; + + // Canonicalize argument order according to the symmetry properties + if (seq.size() > 1 && !(opt.symtree.is_zero())) { + exvector v = seq; + GINAC_ASSERT(is_a(opt.symtree)); + int sig = canonicalize(v.begin(), ex_to(opt.symtree)); + if (sig != INT_MAX) { + // Something has changed while sorting arguments, more evaluations later + if (sig == 0) + return _ex0(); + return ex(sig) * thisexprseq(v); + } + } + + if (opt.eval_f==0) { + return this->hold(); + } + + bool use_remember = opt.use_remember; + ex eval_result; + if (use_remember && lookup_remember_table(eval_result)) { + return eval_result; + } + current_serial = serial; + if (registered_functions()[serial].eval_use_exvector_args) + eval_result = ((eval_funcp_exvector)(registered_functions()[serial].eval_f))(seq); + else + switch (opt.nparams) { + // the following lines have been generated for max. ${maxargs} parameters ${eval_switch_statement} - // end of generated lines - default: - throw(std::logic_error("function::eval(): invalid nparams")); - } - if (use_remember) { - store_remember_table(eval_result); - } - return eval_result; + // end of generated lines + default: + throw(std::logic_error("function::eval(): invalid nparams")); + } + if (use_remember) { + store_remember_table(eval_result); + } + return eval_result; } ex function::evalf(int level) const { - GINAC_ASSERT(serialevalf(level)); + ++it; + } + + if (registered_functions()[serial].evalf_f==0) { + return function(serial,eseq).hold(); + } + current_serial = serial; + if (registered_functions()[serial].evalf_use_exvector_args) + return ((evalf_funcp_exvector)(registered_functions()[serial].evalf_f))(seq); + switch (registered_functions()[serial].nparams) { + // the following lines have been generated for max. ${maxargs} parameters ${evalf_switch_statement} - // end of generated lines - } - throw(std::logic_error("function::evalf(): invalid nparams")); + // end of generated lines + } + throw(std::logic_error("function::evalf(): invalid nparams")); +} + +unsigned function::calchash(void) const +{ + unsigned v = golden_ratio_hash(golden_ratio_hash(tinfo()) ^ serial); + for (unsigned i=0; iop(i).gethash(); + } + v &= 0x7FFFFFFFU; + if (flags & status_flags::evaluated) { + setflag(status_flags::hash_calculated); + hashvalue = v; + } + return v; } ex function::thisexprseq(const exvector & v) const { - return function(serial,v); + return function(serial,v); } ex function::thisexprseq(exvector * vp) const { - return function(serial,vp); + return function(serial,vp); } /** Implementation of ex::series for functions. * \@see ex::series */ -ex function::series(const symbol & s, const ex & point, int order) const +ex function::series(const relational & r, int order, unsigned options) const { - GINAC_ASSERT(serial(const_cast(other)); + GINAC_ASSERT(is_of_type(other, function)); + const function & o = static_cast(other); - if (serial!=o.serial) { - return serial < o.serial ? -1 : 1; - } - return exprseq::compare_same_type(o); + if (serial != o.serial) + return serial < o.serial ? -1 : 1; + else + return exprseq::compare_same_type(o); } bool function::is_equal_same_type(const basic & other) const { - GINAC_ASSERT(is_of_type(other, function)); - const function & o=static_cast(const_cast(other)); + GINAC_ASSERT(is_of_type(other, function)); + const function & o = static_cast(other); - if (serial!=o.serial) return false; - return exprseq::is_equal_same_type(o); + if (serial != o.serial) + return false; + else + return exprseq::is_equal_same_type(o); +} + +bool function::match_same_type(const basic & other) const +{ + GINAC_ASSERT(is_of_type(other, function)); + const function & o = static_cast(other); + + return serial == o.serial; } unsigned function::return_type(void) const { - if (seq.size()==0) { - return return_types::commutative; - } - return (*seq.begin()).return_type(); + if (seq.empty()) + return return_types::commutative; + else + return seq.begin()->return_type(); } - + unsigned function::return_type_tinfo(void) const { - if (seq.size()==0) { - return tinfo_key; - } - return (*seq.begin()).return_type_tinfo(); + if (seq.empty()) + return tinfo_key; + else + return seq.begin()->return_type_tinfo(); } ////////// @@ -1042,80 +994,91 @@ unsigned function::return_type_tinfo(void) const ex function::pderivative(unsigned diff_param) const // partial differentiation { - GINAC_ASSERT(serial & function::registered_functions(void) +std::vector & function::registered_functions(void) { - static vector * rf=new vector; - return *rf; + static std::vector * rf = new std::vector; + return *rf; } bool function::lookup_remember_table(ex & result) const { - return remember_table::remember_tables()[serial].lookup_entry(*this,result); + return remember_table::remember_tables()[serial].lookup_entry(*this,result); } void function::store_remember_table(ex const & result) const { - remember_table::remember_tables()[serial].add_entry(*this,result); + remember_table::remember_tables()[serial].add_entry(*this,result); } // public unsigned function::register_new(function_options const & opt) { - unsigned same_name=0; - for (unsigned i=0; i=opt.functions_with_same_name) { - // we do not throw an exception here because this code is - // usually executed before main(), so the exception could not - // caught anyhow - cerr << "WARNING: function name " << opt.name - << " already in use!" << endl; - } - registered_functions().push_back(opt); - if (opt.use_remember) { - remember_table::remember_tables(). - push_back(remember_table(opt.remember_size, - opt.remember_assoc_size, - opt.remember_strategy)); - } else { - remember_table::remember_tables().push_back(remember_table()); - } - return registered_functions().size()-1; + unsigned same_name = 0; + for (unsigned i=0; i=opt.functions_with_same_name) { + // we do not throw an exception here because this code is + // usually executed before main(), so the exception could not + // caught anyhow + std::cerr << "WARNING: function name " << opt.name + << " already in use!" << std::endl; + } + registered_functions().push_back(opt); + if (opt.use_remember) { + remember_table::remember_tables(). + push_back(remember_table(opt.remember_size, + opt.remember_assoc_size, + opt.remember_strategy)); + } else { + remember_table::remember_tables().push_back(remember_table()); + } + return registered_functions().size()-1; } -////////// -// static member variables -////////// - -// none - -////////// -// global constants -////////// +/** Find serial number of function by name and number of parameters. + * Throws exception if function was not found. */ +unsigned function::find_function(const std::string &name, unsigned nparams) +{ + std::vector::const_iterator i = function::registered_functions().begin(), end = function::registered_functions().end(); + unsigned serial = 0; + while (i != end) { + if (i->get_name() == name && i->get_nparams() == nparams) + return serial; + ++i; + ++serial; + } + throw (std::runtime_error("no function '" + name + "' with " + ToString(nparams) + " parameters defined")); +} -const function some_function; -const type_info & typeid_function=typeid(some_function); +/** Return the print name of the function. */ +std::string function::get_name(void) const +{ + GINAC_ASSERT(serial