doc/examples/mystring.cpp

   1 /**
   2  * @file mystring.cpp Example of extending GiNaC: writing new classes
   3  */
   4 #include <iostream>
   5 #include <string>
   6 #include <stdexcept>
   7 #ifdef IN_GINAC
   8 #include "ginac.h"
   9 #else
  10 #include <ginac/ginac.h>
  11 #endif
  12 using namespace std;
  13 using namespace GiNaC;
  14
  15 class mystring : public basic
  16 {
  17         GINAC_DECLARE_REGISTERED_CLASS(mystring, basic)
  18 public:
  19         mystring(const string &s);
  20         ex eval() const override;
  21 private:
  22         string str;
  23
  24 protected:
  25         void do_print(const print_context &c, unsigned level = 0) const;
  26
  27 };
  28
  29 GINAC_IMPLEMENT_REGISTERED_CLASS_OPT(mystring, basic,
  30   print_func<print_context>(&mystring::do_print))
  31
  32 // ctors
  33 mystring::mystring() { }
  34 mystring::mystring(const string &s) :  str(s) { }
  35
  36 // comparison
  37 int mystring::compare_same_type(const basic &other) const
  38 {
  39         const mystring &o = static_cast<const mystring &>(other);
  40         int cmpval = str.compare(o.str);
  41         if (cmpval == 0)
  42                 return 0;
  43         else if (cmpval < 0)
  44                 return -1;
  45         else
  46                 return 1;
  47 }
  48
  49 // printing
  50 void mystring::do_print(const print_context &c, unsigned level) const
  51 {
  52         // print_context::s is a reference to an ostream
  53         c.s << '\"' << str << '\"';
  54 }
  55
  56 /**
  57  * evaluation: all strings automatically converted to lowercase with
  58  * non-alphabetic characters stripped, and empty strings removed
  59  */
  60 ex mystring::eval() const
  61 {
  62         string new_str;
  63         for (size_t i=0; i<str.length(); i++) {
  64                 char c = str[i];
  65                 if (c >= 'A' && c <= 'Z')
  66                         new_str += tolower(c);
  67                 else if (c >= 'a' && c <= 'z')
  68                         new_str += c;
  69         }
  70
  71         if (new_str.length() == 0)
  72                 return 0;
  73         else
  74                 return mystring(new_str).hold();
  75 }
  76
  77 int main(int argc, char** argv)
  78 {
  79         ex e = mystring("Hello, world!");
  80         cout << is_a<mystring>(e) << endl;
  81         cout << ex_to<basic>(e).class_name() << endl;
  82         cout << e << endl;
  83         ex another = pow(mystring("One string"), 2*sin(Pi-mystring("Another string")));
  84         cout << another << endl;
  85         return 0;
  86 }