]> www.ginac.de Git - ginac.git/blob - doc/examples/mystring.cpp
[TESTS] Fix node::operator< in time_antipode.cpp.
[ginac.git] / 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 }