ginac/function.pl

   1 #!/usr/bin/perl -w
   2
   3 $maxargs=13;
   4
   5 sub generate_seq {
   6     my ($seq_template,$n)=@_;
   7     my ($res,$N);
   8
   9     $res='';
  10     for ($N=1; $N<=$n; $N++) {
  11         $res .= eval('"' . $seq_template . '"');
  12         if ($N!=$n) {
  13             $res .= ', ';
  14         }
  15     }
  16     return $res;
  17 }
  18
  19 sub generate_from_to {
  20     my ($template,$seq_template1,$seq_template2,$from,$to)=@_;
  21     my ($res,$N,$SEQ);
  22
  23     $res='';
  24     for ($N=$from; $N<=$to; $N++) {
  25         $SEQ1=generate_seq($seq_template1,$N);
  26         $SEQ2=generate_seq($seq_template2,$N);
  27         $res .= eval('"' . $template . '"');
  28         $SEQ1=''; # to avoid main::SEQ1 used only once warning
  29         $SEQ2=''; # same as above
  30     }
  31     return $res;
  32 }
  33
  34 sub generate {
  35     my ($template,$seq_template1,$seq_template2)=@_;
  36     return generate_from_to($template,$seq_template1,$seq_template2,1,$maxargs);
  37 }
  38
  39 $declare_function_macro_namespace = <<'END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NAMESPACE';
  40 #ifdef CINT_CONVERSION_WORKAROUND
  41
  42 #define DECLARE_FUNCTION_1P(NAME) \
  43 extern const unsigned function_index_##NAME; \
  44 inline GiNaC::function NAME(const GiNaC::ex & p1) { \
  45     return GiNaC::function(function_index_##NAME, p1); \
  46 } \
  47 inline GiNaC::function NAME(const GiNaC::basic & p1) { \
  48     return GiNaC::function(function_index_##NAME, GiNaC::ex(p1)); \
  49 }
  50 #define DECLARE_FUNCTION_2P(NAME) \
  51 extern const unsigned function_index_##NAME; \
  52 inline GiNaC::function NAME(const GiNaC::ex & p1, const GiNaC::ex & p2) { \
  53     return GiNaC::function(function_index_##NAME, p1, p2); \
  54 } \
  55 inline GiNaC::function NAME(const GiNaC::basic & p1, const GiNaC::ex & p2) { \
  56     return GiNaC::function(function_index_##NAME, GiNaC::ex(p1), p2); \
  57 } \
  58 inline GiNaC::function NAME(const GiNaC::ex & p1, const GiNaC::basic & p2) { \
  59     return GiNaC::function(function_index_##NAME, p1, GiNaC::ex(p2)); \
  60 } \
  61 inline GiNaC::function NAME(const GiNaC::basic & p1, const GiNaC::basic & p2) { \
  62     return GiNaC::function(function_index_##NAME, GiNaC::ex(p1), GiNaC::ex(p2)); \
  63 }
  64
  65 #else // def CINT_CONVERSION_WORKAROUND
  66
  67 #define DECLARE_FUNCTION_1P(NAME) \
  68 extern const unsigned function_index_##NAME; \
  69 inline GiNaC::function NAME(const GiNaC::ex & p1) { \
  70     return GiNaC::function(function_index_##NAME, p1); \
  71 }
  72 #define DECLARE_FUNCTION_2P(NAME) \
  73 extern const unsigned function_index_##NAME; \
  74 inline GiNaC::function NAME(const GiNaC::ex & p1, const GiNaC::ex & p2) { \
  75     return GiNaC::function(function_index_##NAME, p1, p2); \
  76 }
  77
  78 #endif // def CINT_CONVERSION_WORKAROUND
  79
  80 END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NAMESPACE
  81
  82 $declare_function_macro_namespace .= generate_from_to(
  83     <<'END_OF_DECLARE_FUNCTION_MACRO_NAMESPACE','const GiNaC::ex & p${N}','p${N}',3,$maxargs);
  84 #define DECLARE_FUNCTION_${N}P(NAME) \\
  85 extern const unsigned function_index_##NAME; \\
  86 inline GiNaC::function NAME(${SEQ1}) { \\
  87     return GiNaC::function(function_index_##NAME, ${SEQ2}); \\
  88 }
  89
  90 END_OF_DECLARE_FUNCTION_MACRO_NAMESPACE
  91
  92 $declare_function_macro_no_namespace = <<'END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NO_NAMESPACE';
  93 #ifdef CINT_CONVERSION_WORKAROUND
  94
  95 #define DECLARE_FUNCTION_1P(NAME) \
  96 extern const unsigned function_index_##NAME; \
  97 inline function NAME(const ex & p1) { \
  98     return function(function_index_##NAME, p1); \
  99 } \
 100 inline function NAME(const basic & p1) { \
 101     return function(function_index_##NAME, ex(p1)); \
 102 }
 103 #define DECLARE_FUNCTION_2P(NAME) \
 104 extern const unsigned function_index_##NAME; \
 105 inline function NAME(const ex & p1, const ex & p2) { \
 106     return function(function_index_##NAME, p1, p2); \
 107 } \
 108 inline function NAME(const basic & p1, const ex & p2) { \
 109     return function(function_index_##NAME, ex(p1), p2); \
 110 } \
 111 inline function NAME(const ex & p1, const basic & p2) { \
 112     return function(function_index_##NAME, p1, ex(p2)); \
 113 } \
 114 inline function NAME(const basic & p1, const basic & p2) { \
 115     return function(function_index_##NAME, ex(p1), ex(p2)); \
 116 }
 117
 118 #else // def CINT_CONVERSION_WORKAROUND
 119
 120 #define DECLARE_FUNCTION_1P(NAME) \
 121 extern const unsigned function_index_##NAME; \
 122 inline function NAME(const ex & p1) { \
 123     return function(function_index_##NAME, p1); \
 124 }
 125 #define DECLARE_FUNCTION_2P(NAME) \
 126 extern const unsigned function_index_##NAME; \
 127 inline function NAME(const ex & p1, const ex & p2) { \
 128     return function(function_index_##NAME, p1, p2); \
 129 }
 130
 131 #endif // def CINT_CONVERSION_WORKAROUND
 132
 133 END_OF_DECLARE_FUNCTION_1_AND_2P_MACRO_NO_NAMESPACE
 134
 135 $declare_function_macro_no_namespace .= generate_from_to(
 136     <<'END_OF_DECLARE_FUNCTION_MACRO_NO_NAMESPACE','const ex & p${N}','p${N}',3,$maxargs);
 137 #define DECLARE_FUNCTION_${N}P(NAME) \\
 138 extern const unsigned function_index_##NAME; \\
 139 inline function NAME(${SEQ1}) { \\
 140     return function(function_index_##NAME, ${SEQ2}); \\
 141 }
 142
 143 END_OF_DECLARE_FUNCTION_MACRO_NO_NAMESPACE
 144
 145 $typedef_eval_funcp=generate(
 146 'typedef ex (* eval_funcp_${N})(${SEQ1});'."\n",
 147 'const ex &','');
 148
 149 $typedef_evalf_funcp=generate(
 150 'typedef ex (* evalf_funcp_${N})(${SEQ1});'."\n",
 151 'const ex &','');
 152
 153 $typedef_derivative_funcp=generate(
 154 'typedef ex (* derivative_funcp_${N})(${SEQ1}, unsigned);'."\n",
 155 'const ex &','');
 156
 157 $typedef_series_funcp=generate(
 158 'typedef ex (* series_funcp_${N})(${SEQ1}, const symbol &, const ex &, int);'."\n",
 159 'const ex &','');
 160
 161 $eval_func_interface=generate('    function_options & eval_func(eval_funcp_${N} e);'."\n",'','');
 162
 163 $evalf_func_interface=generate('    function_options & evalf_func(evalf_funcp_${N} ef);'."\n",'','');
 164
 165 $derivative_func_interface=generate('    function_options & derivative_func(derivative_funcp_${N} d);'."\n",'','');
 166
 167 $series_func_interface=generate('    function_options & series_func(series_funcp_${N} s);'."\n",'','');
 168
 169 $constructors_interface=generate(
 170 '    function(unsigned ser, ${SEQ1});'."\n",
 171 'const ex & param${N}','');
 172
 173 $constructors_implementation=generate(
 174     <<'END_OF_CONSTRUCTORS_IMPLEMENTATION','const ex & param${N}','param${N}');
 175 function::function(unsigned ser, ${SEQ1})
 176     : exprseq(${SEQ2}), serial(ser)
 177 {
 178     debugmsg(\"function constructor from unsigned,${N}*ex\",LOGLEVEL_CONSTRUCT);
 179     tinfo_key = TINFO_function;
 180 }
 181 END_OF_CONSTRUCTORS_IMPLEMENTATION
 182
 183 $eval_switch_statement=generate(
 184     <<'END_OF_EVAL_SWITCH_STATEMENT','seq[${N}-1]','');
 185     case ${N}:
 186         eval_result=((eval_funcp_${N})(registered_functions()[serial].eval_f))(${SEQ1});
 187         break;
 188 END_OF_EVAL_SWITCH_STATEMENT
 189
 190 $evalf_switch_statement=generate(
 191     <<'END_OF_EVALF_SWITCH_STATEMENT','eseq[${N}-1]','');
 192     case ${N}:
 193         return ((evalf_funcp_${N})(registered_functions()[serial].evalf_f))(${SEQ1});
 194         break;
 195 END_OF_EVALF_SWITCH_STATEMENT
 196
 197 $diff_switch_statement=generate(
 198     <<'END_OF_DIFF_SWITCH_STATEMENT','seq[${N}-1]','');
 199     case ${N}:
 200         return ((derivative_funcp_${N})(registered_functions()[serial].derivative_f))(${SEQ1},diff_param);
 201         break;
 202 END_OF_DIFF_SWITCH_STATEMENT
 203
 204 $series_switch_statement=generate(
 205     <<'END_OF_SERIES_SWITCH_STATEMENT','seq[${N}-1]','');
 206     case ${N}:
 207         try {
 208             res = ((series_funcp_${N})(registered_functions()[serial].series_f))(${SEQ1},s,point,order);
 209         } catch (do_taylor) {
 210             res = basic::series(s, point, order);
 211         }
 212         return res;
 213         break;
 214 END_OF_SERIES_SWITCH_STATEMENT
 215
 216 $eval_func_implementation=generate(
 217     <<'END_OF_EVAL_FUNC_IMPLEMENTATION','','');
 218 function_options & function_options::eval_func(eval_funcp_${N} e)
 219 {
 220     test_and_set_nparams(${N});
 221     eval_f=eval_funcp(e);
 222     return *this;
 223 }
 224 END_OF_EVAL_FUNC_IMPLEMENTATION
 225
 226 $evalf_func_implementation=generate(
 227     <<'END_OF_EVALF_FUNC_IMPLEMENTATION','','');
 228 function_options & function_options::evalf_func(evalf_funcp_${N} ef)
 229 {
 230     test_and_set_nparams(${N});
 231     evalf_f=evalf_funcp(ef);
 232     return *this;
 233 }
 234 END_OF_EVALF_FUNC_IMPLEMENTATION
 235
 236 $derivative_func_implementation=generate(
 237     <<'END_OF_DERIVATIVE_FUNC_IMPLEMENTATION','','');
 238 function_options & function_options::derivative_func(derivative_funcp_${N} d)
 239 {
 240     test_and_set_nparams(${N});
 241     derivative_f=derivative_funcp(d);
 242     return *this;
 243 }
 244 END_OF_DERIVATIVE_FUNC_IMPLEMENTATION
 245
 246 $series_func_implementation=generate(
 247     <<'END_OF_SERIES_FUNC_IMPLEMENTATION','','');
 248 function_options & function_options::series_func(series_funcp_${N} s)
 249 {
 250     test_and_set_nparams(${N});
 251     series_f=series_funcp(s);
 252     return *this;
 253 }
 254 END_OF_SERIES_FUNC_IMPLEMENTATION
 255
 256 $interface=<<END_OF_INTERFACE;
 257 /** \@file function.h
 258  *
 259  *  Interface to abstract class function (new function concept). */
 260
 261 /*
 262  *  This file was generated automatically by function.pl.
 263  *  Please do not modify it directly, edit the perl script instead!
 264  *  function.pl options: \$maxargs=${maxargs}
 265  *
 266  *  GiNaC Copyright (C) 1999-2000 Johannes Gutenberg University Mainz, Germany
 267  *
 268  *  This program is free software; you can redistribute it and/or modify
 269  *  it under the terms of the GNU General Public License as published by
 270  *  the Free Software Foundation; either version 2 of the License, or
 271  *  (at your option) any later version.
 272  *
 273  *  This program is distributed in the hope that it will be useful,
 274  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 275  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 276  *  GNU General Public License for more details.
 277  *
 278  *  You should have received a copy of the GNU General Public License
 279  *  along with this program; if not, write to the Free Software
 280  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 281  */
 282
 283 #ifndef __GINAC_FUNCTION_H__
 284 #define __GINAC_FUNCTION_H__
 285
 286 #include <string>
 287 #include <vector>
 288
 289 #ifdef __CINT__
 290 // CINT needs <algorithm> to work properly with <vector>
 291 #include <algorithm>
 292 #endif // def __CINT__
 293
 294 #include "exprseq.h"
 295
 296 #ifndef NO_NAMESPACE_GINAC
 297
 298 // the following lines have been generated for max. ${maxargs} parameters
 299 $declare_function_macro_namespace
 300 // end of generated lines
 301
 302 #else // ndef NO_NAMESPACE_GINAC
 303
 304 // the following lines have been generated for max. ${maxargs} parameters
 305 $declare_function_macro_no_namespace
 306 // end of generated lines
 307
 308 #endif // ndef NO_NAMESPACE_GINAC
 309
 310 #ifndef NO_NAMESPACE_GINAC
 311
 312 #define REGISTER_FUNCTION(NAME,OPT) \\
 313 const unsigned function_index_##NAME= \\
 314     GiNaC::function::register_new(GiNaC::function_options(#NAME).OPT);
 315
 316 #define REGISTER_FUNCTION_OLD(NAME,E,EF,D,S) \\
 317 const unsigned function_index_##NAME= \\
 318     GiNaC::function::register_new(GiNaC::function_options(#NAME). \\
 319                                   eval_func(E). \\
 320                                   evalf_func(EF). \\
 321                                   derivative_func(D). \\
 322                                   series_func(S));
 323
 324 #else // ndef NO_NAMESPACE_GINAC
 325
 326 #define REGISTER_FUNCTION(NAME,OPT) \\
 327 const unsigned function_index_##NAME= \\
 328     function::register_new(function_options(#NAME).OPT);
 329
 330 #define REGISTER_FUNCTION_OLD(NAME,E,EF,D,S) \\
 331 const unsigned function_index_##NAME= \\
 332     function::register_new(function_options(#NAME). \\
 333                            eval_func(E). \\
 334                            evalf_func(EF). \\
 335                            derivative_func(D). \\
 336                            series_func(S));
 337
 338 #endif // ndef NO_NAMESPACE_GINAC
 339
 340 #define BEGIN_TYPECHECK \\
 341 bool automatic_typecheck=true;
 342
 343 #define TYPECHECK(VAR,TYPE) \\
 344 if (!is_ex_exactly_of_type(VAR,TYPE)) { \\
 345     automatic_typecheck=false; \\
 346 } else
 347
 348 #ifndef NO_NAMESPACE_GINAC
 349
 350 #define TYPECHECK_INTEGER(VAR) \\
 351 if (!(VAR).info(GiNaC::info_flags::integer)) { \\
 352     automatic_typecheck=false; \\
 353 } else
 354
 355 #else // ndef NO_NAMESPACE_GINAC
 356
 357 #define TYPECHECK_INTEGER(VAR) \\
 358 if (!(VAR).info(info_flags::integer)) { \\
 359     automatic_typecheck=false; \\
 360 } else
 361
 362 #endif // ndef NO_NAMESPACE_GINAC
 363
 364 #define END_TYPECHECK(RV) \\
 365 {} \\
 366 if (!automatic_typecheck) { \\
 367     return RV.hold(); \\
 368 }
 369
 370 #ifndef NO_NAMESPACE_GINAC
 371 namespace GiNaC {
 372 #endif // ndef NO_NAMESPACE_GINAC
 373
 374 class function;
 375
 376 typedef ex (* eval_funcp)();
 377 typedef ex (* evalf_funcp)();
 378 typedef ex (* derivative_funcp)();
 379 typedef ex (* series_funcp)();
 380
 381 // the following lines have been generated for max. ${maxargs} parameters
 382 $typedef_eval_funcp
 383 $typedef_evalf_funcp
 384 $typedef_derivative_funcp
 385 $typedef_series_funcp
 386 // end of generated lines
 387
 388 class function_options
 389 {
 390     friend class function;
 391 public:
 392     function_options();
 393     function_options(string const & n, string const & tn=string());
 394     ~function_options();
 395     void initialize(void);
 396     function_options & set_name(string const & n, string const & tn=string());
 397 // the following lines have been generated for max. ${maxargs} parameters
 398 $eval_func_interface
 399 $evalf_func_interface
 400 $derivative_func_interface
 401 $series_func_interface
 402 // end of generated lines
 403     function_options & set_return_type(unsigned rt, unsigned rtt=0);
 404     function_options & do_not_evalf_params(void);
 405     function_options & remember(unsigned size, unsigned assoc_size=0,
 406                                 unsigned strategy=remember_strategies::delete_never);
 407     function_options & overloaded(unsigned o);
 408     void test_and_set_nparams(unsigned n);
 409     string get_name(void) const { return name; }
 410     unsigned get_nparams(void) const { return nparams; }
 411
 412 protected:
 413     string name;
 414     string TeX_name;
 415
 416     unsigned nparams;
 417
 418     eval_funcp eval_f;
 419     evalf_funcp evalf_f;
 420     derivative_funcp derivative_f;
 421     series_funcp series_f;
 422
 423     bool evalf_params_first;
 424
 425     bool use_return_type;
 426     unsigned return_type;
 427     unsigned return_type_tinfo;
 428
 429     bool use_remember;
 430     unsigned remember_size;
 431     unsigned remember_assoc_size;
 432     unsigned remember_strategy;
 433
 434     unsigned functions_with_same_name;
 435 };
 436
 437 /** The class function is used to implement builtin functions like sin, cos...
 438     and user defined functions */
 439 class function : public exprseq
 440 {
 441     GINAC_DECLARE_REGISTERED_CLASS(function, exprseq)
 442
 443     // CINT has a linking problem
 444     friend void ginsh_get_ginac_functions(void);
 445
 446     friend class remember_table_entry;
 447     // friend class remember_table_list;
 448     // friend class remember_table;
 449
 450 // member functions
 451
 452     // default constructor, destructor, copy constructor assignment operator and helpers
 453 public:
 454     function();
 455     ~function();
 456     function(const function & other);
 457     const function & operator=(const function & other);
 458 protected:
 459     void copy(const function & other);
 460     void destroy(bool call_parent);
 461
 462     // other constructors
 463 public:
 464     function(unsigned ser);
 465     // the following lines have been generated for max. ${maxargs} parameters
 466 $constructors_interface
 467     // end of generated lines
 468     function(unsigned ser, const exprseq & es);
 469     function(unsigned ser, const exvector & v, bool discardable=0);
 470     function(unsigned ser, exvector * vp); // vp will be deleted
 471
 472     // functions overriding virtual functions from bases classes
 473 public:
 474     basic * duplicate() const;
 475     void printraw(ostream & os) const;
 476     void print(ostream & os, unsigned upper_precedence=0) const;
 477     void printtree(ostream & os, unsigned indent) const;
 478     void printcsrc(ostream & os, unsigned type, unsigned upper_precedence=0) const;
 479     ex expand(unsigned options=0) const;
 480     ex eval(int level=0) const;
 481     ex evalf(int level=0) const;
 482     ex series(const symbol & s, const ex & point, int order) const;
 483     ex thisexprseq(const exvector & v) const;
 484     ex thisexprseq(exvector * vp) const;
 485 protected:
 486     ex derivative(const symbol & s) const;
 487     int compare_same_type(const basic & other) const;
 488     bool is_equal_same_type(const basic & other) const;
 489     unsigned return_type(void) const;
 490     unsigned return_type_tinfo(void) const;
 491
 492     // new virtual functions which can be overridden by derived classes
 493     // none
 494
 495     // non-virtual functions in this class
 496 protected:
 497     ex pderivative(unsigned diff_param) const; // partial differentiation
 498     static vector<function_options> & registered_functions(void);
 499     bool lookup_remember_table(ex & result) const;
 500     void store_remember_table(ex const & result) const;
 501 public:
 502     static unsigned register_new(function_options const & opt);
 503     unsigned getserial(void) const {return serial;}
 504
 505 // member variables
 506
 507 protected:
 508     unsigned serial;
 509 };
 510
 511 // utility macros
 512
 513 #ifndef NO_NAMESPACE_GINAC
 514
 515 #define is_ex_the_function(OBJ, FUNCNAME) \\
 516     (is_ex_exactly_of_type(OBJ, function) && static_cast<GiNaC::function *>(OBJ.bp)->getserial() == function_index_##FUNCNAME)
 517
 518 #else // ndef NO_NAMESPACE_GINAC
 519
 520 #define is_ex_the_function(OBJ, FUNCNAME) \\
 521     (is_ex_exactly_of_type(OBJ, function) && static_cast<function *>(OBJ.bp)->getserial() == function_index_##FUNCNAME)
 522
 523 #endif // ndef NO_NAMESPACE_GINAC
 524
 525 // global constants
 526
 527 extern const function some_function;
 528 extern const type_info & typeid_function;
 529
 530 #ifndef NO_NAMESPACE_GINAC
 531 } // namespace GiNaC
 532 #endif // ndef NO_NAMESPACE_GINAC
 533
 534 #endif // ndef __GINAC_FUNCTION_H__
 535
 536 END_OF_INTERFACE
 537
 538 $implementation=<<END_OF_IMPLEMENTATION;
 539 /** \@file function.cpp
 540  *
 541  *  Implementation of class function. */
 542
 543 /*
 544  *  This file was generated automatically by function.pl.
 545  *  Please do not modify it directly, edit the perl script instead!
 546  *  function.pl options: \$maxargs=${maxargs}
 547  *
 548  *  GiNaC Copyright (C) 1999-2000 Johannes Gutenberg University Mainz, Germany
 549  *
 550  *  This program is free software; you can redistribute it and/or modify
 551  *  it under the terms of the GNU General Public License as published by
 552  *  the Free Software Foundation; either version 2 of the License, or
 553  *  (at your option) any later version.
 554  *
 555  *  This program is distributed in the hope that it will be useful,
 556  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 557  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 558  *  GNU General Public License for more details.
 559  *
 560  *  You should have received a copy of the GNU General Public License
 561  *  along with this program; if not, write to the Free Software
 562  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 563  */
 564
 565 #include <string>
 566 #include <stdexcept>
 567 #include <list>
 568
 569 #include "function.h"
 570 #include "ex.h"
 571 #include "archive.h"
 572 #include "inifcns.h"
 573 #include "utils.h"
 574 #include "debugmsg.h"
 575 #include "remember.h"
 576
 577 #ifndef NO_NAMESPACE_GINAC
 578 namespace GiNaC {
 579 #endif // ndef NO_NAMESPACE_GINAC
 580
 581 //////////
 582 // helper class function_options
 583 //////////
 584
 585 function_options::function_options()
 586 {
 587     initialize();
 588 }
 589
 590 function_options::function_options(string const & n, string const & tn)
 591 {
 592     initialize();
 593     set_name(n,tn);
 594 }
 595
 596 function_options::~function_options()
 597 {
 598     // nothing to clean up at the moment
 599 }
 600
 601 void function_options::initialize(void)
 602 {
 603     set_name("unnamed_function","\\\\operatorname{unnamed}");
 604     nparams=0;
 605     eval_f=evalf_f=derivative_f=series_f=0;
 606     evalf_params_first=true;
 607     use_return_type=false;
 608     use_remember=false;
 609     functions_with_same_name=1;
 610 }
 611
 612 function_options & function_options::set_name(string const & n,
 613                                               string const & tn)
 614 {
 615     name=n;
 616     if (tn==string()) {
 617         TeX_name="\\\\operatorname{"+name+"}";
 618     } else {
 619         TeX_name=tn;
 620     }
 621     return *this;
 622 }
 623
 624 // the following lines have been generated for max. ${maxargs} parameters
 625 $eval_func_implementation
 626 $evalf_func_implementation
 627 $derivative_func_implementation
 628 $series_func_implementation
 629 // end of generated lines
 630
 631 function_options & function_options::set_return_type(unsigned rt, unsigned rtt)
 632 {
 633     use_return_type=true;
 634     return_type=rt;
 635     return_type_tinfo=rtt;
 636     return *this;
 637 }
 638
 639 function_options & function_options::do_not_evalf_params(void)
 640 {
 641     evalf_params_first=false;
 642     return *this;
 643 }
 644
 645 function_options & function_options::remember(unsigned size,
 646                                               unsigned assoc_size,
 647                                               unsigned strategy)
 648 {
 649     use_remember=true;
 650     remember_size=size;
 651     remember_assoc_size=assoc_size;
 652     remember_strategy=strategy;
 653     return *this;
 654 }
 655
 656 function_options & function_options::overloaded(unsigned o)
 657 {
 658     functions_with_same_name=o;
 659     return *this;
 660 }
 661
 662 void function_options::test_and_set_nparams(unsigned n)
 663 {
 664     if (nparams==0) {
 665         nparams=n;
 666     } else if (nparams!=n) {
 667         // we do not throw an exception here because this code is
 668         // usually executed before main(), so the exception could not
 669         // caught anyhow
 670         cerr << "WARNING: number of parameters ("
 671              << n << ") differs from number set before ("
 672              << nparams << ")" << endl;
 673     }
 674 }
 675
 676 GINAC_IMPLEMENT_REGISTERED_CLASS(function, exprseq)
 677
 678 //////////
 679 // default constructor, destructor, copy constructor assignment operator and helpers
 680 //////////
 681
 682 // public
 683
 684 function::function() : serial(0)
 685 {
 686     debugmsg("function default constructor",LOGLEVEL_CONSTRUCT);
 687     tinfo_key = TINFO_function;
 688 }
 689
 690 function::~function()
 691 {
 692     debugmsg("function destructor",LOGLEVEL_DESTRUCT);
 693     destroy(0);
 694 }
 695
 696 function::function(const function & other)
 697 {
 698     debugmsg("function copy constructor",LOGLEVEL_CONSTRUCT);
 699     copy(other);
 700 }
 701
 702 const function & function::operator=(const function & other)
 703 {
 704     debugmsg("function operator=",LOGLEVEL_ASSIGNMENT);
 705     if (this != &other) {
 706         destroy(1);
 707         copy(other);
 708     }
 709     return *this;
 710 }
 711
 712 // protected
 713
 714 void function::copy(const function & other)
 715 {
 716     exprseq::copy(other);
 717     serial=other.serial;
 718 }
 719
 720 void function::destroy(bool call_parent)
 721 {
 722     if (call_parent) exprseq::destroy(call_parent);
 723 }
 724
 725 //////////
 726 // other constructors
 727 //////////
 728
 729 // public
 730
 731 function::function(unsigned ser) : serial(ser)
 732 {
 733     debugmsg("function constructor from unsigned",LOGLEVEL_CONSTRUCT);
 734     tinfo_key = TINFO_function;
 735 }
 736
 737 // the following lines have been generated for max. ${maxargs} parameters
 738 $constructors_implementation
 739 // end of generated lines
 740
 741 function::function(unsigned ser, const exprseq & es) : exprseq(es), serial(ser)
 742 {
 743     debugmsg("function constructor from unsigned,exprseq",LOGLEVEL_CONSTRUCT);
 744     tinfo_key = TINFO_function;
 745 }
 746
 747 function::function(unsigned ser, const exvector & v, bool discardable)
 748     : exprseq(v,discardable), serial(ser)
 749 {
 750     debugmsg("function constructor from string,exvector,bool",LOGLEVEL_CONSTRUCT);
 751     tinfo_key = TINFO_function;
 752 }
 753
 754 function::function(unsigned ser, exvector * vp)
 755     : exprseq(vp), serial(ser)
 756 {
 757     debugmsg("function constructor from unsigned,exvector *",LOGLEVEL_CONSTRUCT);
 758     tinfo_key = TINFO_function;
 759 }
 760
 761 //////////
 762 // archiving
 763 //////////
 764
 765 /** Construct object from archive_node. */
 766 function::function(const archive_node &n, const lst &sym_lst) : inherited(n, sym_lst)
 767 {
 768     debugmsg("function constructor from archive_node", LOGLEVEL_CONSTRUCT);
 769
 770     // Find serial number by function name
 771     string s;
 772     if (n.find_string("name", s)) {
 773         unsigned int ser = 0;
 774         vector<function_options>::const_iterator i = registered_functions().begin(), iend = registered_functions().end();
 775         while (i != iend) {
 776             if (s == i->name) {
 777                 serial = ser;
 778                 return;
 779             }
 780             i++; ser++;
 781         }
 782         throw (std::runtime_error("unknown function '" + s + "' in archive"));
 783     } else
 784         throw (std::runtime_error("unnamed function in archive"));
 785 }
 786
 787 /** Unarchive the object. */
 788 ex function::unarchive(const archive_node &n, const lst &sym_lst)
 789 {
 790     return (new function(n, sym_lst))->setflag(status_flags::dynallocated);
 791 }
 792
 793 /** Archive the object. */
 794 void function::archive(archive_node &n) const
 795 {
 796     inherited::archive(n);
 797     GINAC_ASSERT(serial < registered_functions().size());
 798     n.add_string("name", registered_functions()[serial].name);
 799 }
 800
 801 //////////
 802 // functions overriding virtual functions from bases classes
 803 //////////
 804
 805 // public
 806
 807 basic * function::duplicate() const
 808 {
 809     debugmsg("function duplicate",LOGLEVEL_DUPLICATE);
 810     return new function(*this);
 811 }
 812
 813 void function::printraw(ostream & os) const
 814 {
 815     debugmsg("function printraw",LOGLEVEL_PRINT);
 816
 817     GINAC_ASSERT(serial<registered_functions().size());
 818
 819     os << "function(name=" << registered_functions()[serial].name;
 820     for (exvector::const_iterator it=seq.begin(); it!=seq.end(); ++it) {
 821         os << ",";
 822         (*it).bp->print(os);
 823     }
 824     os << ")";
 825 }
 826
 827 void function::print(ostream & os, unsigned upper_precedence) const
 828 {
 829     debugmsg("function print",LOGLEVEL_PRINT);
 830
 831     GINAC_ASSERT(serial<registered_functions().size());
 832
 833     os << registered_functions()[serial].name;
 834     printseq(os,'(',',',')',exprseq::precedence,function::precedence);
 835 }
 836
 837 void function::printtree(ostream & os, unsigned indent) const
 838 {
 839     debugmsg("function printtree",LOGLEVEL_PRINT);
 840
 841     GINAC_ASSERT(serial<registered_functions().size());
 842
 843     os << string(indent,' ') << "function "
 844        << registered_functions()[serial].name
 845        << ", hash=" << hashvalue << " (0x" << hex << hashvalue << dec << ")"
 846        << ", flags=" << flags
 847        << ", nops=" << nops() << endl;
 848     for (unsigned i=0; i<nops(); ++i) {
 849         seq[i].printtree(os,indent+delta_indent);
 850     }
 851     os << string(indent+delta_indent,' ') << "=====" << endl;
 852 }
 853
 854 void function::printcsrc(ostream & os, unsigned type, unsigned upper_precedence) const
 855 {
 856     debugmsg("function print csrc",LOGLEVEL_PRINT);
 857
 858     GINAC_ASSERT(serial<registered_functions().size());
 859
 860         // Print function name in lowercase
 861     string lname;
 862     lname=registered_functions()[serial].name;
 863     for (unsigned i=0; i<lname.size(); i++)
 864         lname[i] = tolower(lname[i]);
 865     os << lname << "(";
 866
 867         // Print arguments, separated by commas
 868     exvector::const_iterator it = seq.begin();
 869     exvector::const_iterator itend = seq.end();
 870     while (it != itend) {
 871         it->bp->printcsrc(os, type, 0);
 872         it++;
 873         if (it != itend)
 874             os << ",";
 875     }
 876     os << ")";
 877 }
 878
 879 ex function::expand(unsigned options) const
 880 {
 881     return this->setflag(status_flags::expanded);
 882 }
 883
 884 ex function::eval(int level) const
 885 {
 886     GINAC_ASSERT(serial<registered_functions().size());
 887
 888     if (level>1) {
 889         // first evaluate children, then we will end up here again
 890         return function(serial,evalchildren(level));
 891     }
 892
 893     if (registered_functions()[serial].eval_f==0) {
 894         return this->hold();
 895     }
 896
 897     bool use_remember=registered_functions()[serial].use_remember;
 898     ex eval_result;
 899     if (use_remember && lookup_remember_table(eval_result)) {
 900         return eval_result;
 901     }
 902
 903     switch (registered_functions()[serial].nparams) {
 904         // the following lines have been generated for max. ${maxargs} parameters
 905 ${eval_switch_statement}
 906         // end of generated lines
 907     default:
 908         throw(std::logic_error("function::eval(): invalid nparams"));
 909     }
 910     if (use_remember) {
 911         store_remember_table(eval_result);
 912     }
 913     return eval_result;
 914 }
 915
 916 ex function::evalf(int level) const
 917 {
 918     GINAC_ASSERT(serial<registered_functions().size());
 919
 920     exvector eseq=evalfchildren(level);
 921
 922     if (registered_functions()[serial].evalf_f==0) {
 923         return function(serial,eseq).hold();
 924     }
 925     switch (registered_functions()[serial].nparams) {
 926         // the following lines have been generated for max. ${maxargs} parameters
 927 ${evalf_switch_statement}
 928         // end of generated lines
 929     }
 930     throw(std::logic_error("function::evalf(): invalid nparams"));
 931 }
 932
 933 ex function::thisexprseq(const exvector & v) const
 934 {
 935     return function(serial,v);
 936 }
 937
 938 ex function::thisexprseq(exvector * vp) const
 939 {
 940     return function(serial,vp);
 941 }
 942
 943 /** Implementation of ex::series for functions.
 944  *  \@see ex::series */
 945 ex function::series(const symbol & s, const ex & point, int order) const
 946 {
 947     GINAC_ASSERT(serial<registered_functions().size());
 948
 949     if (registered_functions()[serial].series_f==0) {
 950         return basic::series(s, point, order);
 951     }
 952     ex res;
 953     switch (registered_functions()[serial].nparams) {
 954         // the following lines have been generated for max. ${maxargs} parameters
 955 ${series_switch_statement}
 956         // end of generated lines
 957     }
 958     throw(std::logic_error("function::series(): invalid nparams"));
 959 }
 960
 961 // protected
 962
 963
 964 /** Implementation of ex::diff() for functions. It applies the chain rule,
 965  *  except for the Order term function.
 966  *  \@see ex::diff */
 967 ex function::derivative(const symbol & s) const
 968 {
 969     ex result;
 970
 971     if (serial==function_index_Order) {
 972         // Order Term function only differentiates the argument
 973         return Order(seq[0].diff(s));
 974     } else {
 975         // Chain rule
 976         ex arg_diff;
 977         for (unsigned i=0; i!=seq.size(); i++) {
 978             arg_diff = seq[i].diff(s);
 979             // We apply the chain rule only when it makes sense.  This is not
 980             // just for performance reasons but also to allow functions to
 981             // throw when differentiated with respect to one of its arguments
 982             // without running into trouble with our automatic full
 983             // differentiation:
 984             if (!arg_diff.is_zero())
 985                 result += pderivative(i)*arg_diff;
 986         }
 987     }
 988     return result;
 989 }
 990
 991 int function::compare_same_type(const basic & other) const
 992 {
 993     GINAC_ASSERT(is_of_type(other, function));
 994     const function & o=static_cast<function &>(const_cast<basic &>(other));
 995
 996     if (serial!=o.serial) {
 997         return serial < o.serial ? -1 : 1;
 998     }
 999     return exprseq::compare_same_type(o);
1000 }
1001
1002 bool function::is_equal_same_type(const basic & other) const
1003 {
1004     GINAC_ASSERT(is_of_type(other, function));
1005     const function & o=static_cast<function &>(const_cast<basic &>(other));
1006
1007     if (serial!=o.serial) return false;
1008     return exprseq::is_equal_same_type(o);
1009 }
1010
1011 unsigned function::return_type(void) const
1012 {
1013     if (seq.size()==0) {
1014         return return_types::commutative;
1015     }
1016     return (*seq.begin()).return_type();
1017 }
1018
1019 unsigned function::return_type_tinfo(void) const
1020 {
1021     if (seq.size()==0) {
1022         return tinfo_key;
1023     }
1024     return (*seq.begin()).return_type_tinfo();
1025 }
1026
1027 //////////
1028 // new virtual functions which can be overridden by derived classes
1029 //////////
1030
1031 // none
1032
1033 //////////
1034 // non-virtual functions in this class
1035 //////////
1036
1037 // protected
1038
1039 ex function::pderivative(unsigned diff_param) const // partial differentiation
1040 {
1041     GINAC_ASSERT(serial<registered_functions().size());
1042
1043     if (registered_functions()[serial].derivative_f==0) {
1044         throw(std::logic_error(string("function::pderivative(") + registered_functions()[serial].name + "): no diff function defined"));
1045     }
1046     switch (registered_functions()[serial].nparams) {
1047         // the following lines have been generated for max. ${maxargs} parameters
1048 ${diff_switch_statement}
1049         // end of generated lines
1050     }
1051     throw(std::logic_error("function::pderivative(): no diff function defined"));
1052 }
1053
1054 vector<function_options> & function::registered_functions(void)
1055 {
1056     static vector<function_options> * rf=new vector<function_options>;
1057     return *rf;
1058 }
1059
1060 bool function::lookup_remember_table(ex & result) const
1061 {
1062     return remember_table::remember_tables()[serial].lookup_entry(*this,result);
1063 }
1064
1065 void function::store_remember_table(ex const & result) const
1066 {
1067     remember_table::remember_tables()[serial].add_entry(*this,result);
1068 }
1069
1070 // public
1071
1072 unsigned function::register_new(function_options const & opt)
1073 {
1074     unsigned same_name=0;
1075     for (unsigned i=0; i<registered_functions().size(); ++i) {
1076         if (registered_functions()[i].name==opt.name) {
1077             same_name++;
1078         }
1079     }
1080     if (same_name>=opt.functions_with_same_name) {
1081         // we do not throw an exception here because this code is
1082         // usually executed before main(), so the exception could not
1083         // caught anyhow
1084         cerr << "WARNING: function name " << opt.name
1085              << " already in use!" << endl;
1086     }
1087     registered_functions().push_back(opt);
1088     if (opt.use_remember) {
1089         remember_table::remember_tables().
1090             push_back(remember_table(opt.remember_size,
1091                                      opt.remember_assoc_size,
1092                                      opt.remember_strategy));
1093     } else {
1094         remember_table::remember_tables().push_back(remember_table());
1095     }
1096     return registered_functions().size()-1;
1097 }
1098
1099 //////////
1100 // static member variables
1101 //////////
1102
1103 // none
1104
1105 //////////
1106 // global constants
1107 //////////
1108
1109 const function some_function;
1110 const type_info & typeid_function=typeid(some_function);
1111
1112 #ifndef NO_NAMESPACE_GINAC
1113 } // namespace GiNaC
1114 #endif // ndef NO_NAMESPACE_GINAC
1115
1116 END_OF_IMPLEMENTATION
1117
1118 print "Creating interface file function.h...";
1119 open OUT,">function.h" or die "cannot open function.h";
1120 print OUT $interface;
1121 close OUT;
1122 print "ok.\n";
1123
1124 print "Creating implementation file function.cpp...";
1125 open OUT,">function.cpp" or die "cannot open function.cpp";
1126 print OUT $implementation;
1127 close OUT;
1128 print "ok.\n";
1129
1130 print "done.\n";