#include <string>
#include <list>
+#include <vector>
+
+#include "class_info.h"
+#include "print.h"
namespace GiNaC {
-class registered_class_info;
class ex;
class archive_node;
typedef ex (*unarch_func)(const archive_node &n, lst &sym_lst);
-/** Head of list of all registered_class_info structures. */
-extern registered_class_info *first_registered_class;
+/** This class stores information about a registered GiNaC class. */
+class registered_class_options {
+public:
+ registered_class_options(const char *n, const char *p, unsigned ti, unarch_func f)
+ : name(n), parent_name(p), tinfo_key(ti), unarchive(f) {}
+
+ const char *get_name() const { return name; }
+ const char *get_parent_name() const { return parent_name; }
+ unsigned get_id() const { return tinfo_key; }
+ unarch_func get_unarch_func() const { return unarchive; }
+ const std::vector<print_functor> &get_print_dispatch_table() const { return print_dispatch_table; }
+
+ template <class Ctx, class T, class C>
+ registered_class_options & print_func(void f(const T &, const C & c, unsigned))
+ {
+ set_print_func(Ctx::get_class_info_static().options.get_id(), f);
+ return *this;
+ }
+
+ template <class Ctx, class T, class C>
+ registered_class_options & print_func(void (T::*f)(const C &, unsigned))
+ {
+ set_print_func(Ctx::get_class_info_static().options.get_id(), f);
+ return *this;
+ }
+ template <class Ctx>
+ registered_class_options & print_func(const print_functor & f)
+ {
+ set_print_func(Ctx::get_class_info_static().options.get_id(), f);
+ return *this;
+ }
-/** This structure stores information about a registered GiNaC class. */
-struct registered_class_info {
- registered_class_info(const char *n, const char *s, unsigned int k, unarch_func f)
- : name(n), super(s), tinfo_key(k), unarchive(f)
+ void set_print_func(unsigned id, const print_functor & f)
{
- // Add structure to list
- next = first_registered_class;
- first_registered_class = this;
+ if (id >= print_dispatch_table.size())
+ print_dispatch_table.resize(id + 1);
+ print_dispatch_table[id] = f;
}
- registered_class_info *next; /**< Pointer to next registered_class_info in list. */
- const char *name; /**< Class name. */
- const char *super; /**< Name of superclass. */
- unsigned int tinfo_key; /**< TINFO_* key. */
- unarch_func unarchive; /**< Pointer to unarchiving function. */
+private:
+ const char *name; /**< Class name. */
+ const char *parent_name; /**< Name of superclass. */
+ unsigned tinfo_key; /**< TINFO_* key. */
+ unarch_func unarchive; /**< Pointer to unarchiving function. */
+ std::vector<print_functor> print_dispatch_table; /**< Method table for print() dispatch */
};
+typedef class_info<registered_class_options> registered_class_info;
+
/** Primary macro for inclusion in the declaration of each registered class. */
#define GINAC_DECLARE_REGISTERED_CLASS_NO_CTORS(classname, supername) \
public: \
typedef supername inherited; \
- \
+private: \
static GiNaC::registered_class_info reg_info; \
- virtual const char *class_name() const; \
+public: \
+ static GiNaC::registered_class_info &get_class_info_static() { return reg_info; } \
+ virtual const GiNaC::registered_class_info &get_class_info() const { return classname::get_class_info_static(); } \
+ virtual GiNaC::registered_class_info &get_class_info() { return classname::get_class_info_static(); } \
+ virtual const char *class_name() const { return classname::get_class_info_static().options.get_name(); } \
\
classname(const GiNaC::archive_node &n, GiNaC::lst &sym_lst); \
virtual void archive(GiNaC::archive_node &n) const; \
GINAC_DECLARE_REGISTERED_CLASS_NO_CTORS(classname, supername) \
public: \
classname(); \
- classname * duplicate() const { return new classname(*this); } \
+ virtual classname * duplicate() const { return new classname(*this); } \
\
- void accept(GiNaC::visitor & v) const \
+ virtual void accept(GiNaC::visitor & v) const \
{ \
if (visitor *p = dynamic_cast<visitor *>(&v)) \
p->visit(*this); \
inherited::accept(v); \
} \
protected: \
- int compare_same_type(const GiNaC::basic & other) const; \
+ virtual int compare_same_type(const GiNaC::basic & other) const; \
private:
-/** Macro for inclusion in the implementation of each registered class.
- * It defines some members that are the same in all classes derived from
- * 'basic'. */
+/** Macro for inclusion in the implementation of each registered class. */
#define GINAC_IMPLEMENT_REGISTERED_CLASS(classname, supername) \
- GiNaC::registered_class_info classname::reg_info(#classname, #supername, TINFO_##classname, &classname::unarchive); \
- const char *classname::class_name() const {return reg_info.name;}
+ GiNaC::registered_class_info classname::reg_info = GiNaC::registered_class_info(GiNaC::registered_class_options(#classname, #supername, TINFO_##classname, &classname::unarchive));
+
+/** Macro for inclusion in the implementation of each registered class.
+ * Additional options can be specified. */
+#define GINAC_IMPLEMENT_REGISTERED_CLASS_OPT(classname, supername, options) \
+ GiNaC::registered_class_info classname::reg_info = GiNaC::registered_class_info(GiNaC::registered_class_options(#classname, #supername, TINFO_##classname, &classname::unarchive).options);
/** Find TINFO_* key by class name. */
-extern unsigned int find_tinfo_key(const std::string &class_name);
+extern unsigned find_tinfo_key(const std::string &class_name);
/** Find unarchiving function by class name. */
extern unarch_func find_unarch_func(const std::string &class_name);
+/** Add or replace a print method. */
+template <class Alg, class Ctx, class T, class C>
+extern void set_print_func(void f(const T &, const C & c, unsigned))
+{
+ Alg::get_class_info_static().options.set_print_func(Ctx::get_class_info_static().options.get_id(), f);
+}
+
+/** Add or replace a print method. */
+template <class Alg, class Ctx, class T, class C>
+extern void set_print_func(void (T::*f)(const C &, unsigned))
+{
+ Alg::get_class_info_static().options.set_print_func(Ctx::get_class_info_static().options.get_id(), f);
+}
+
+
} // namespace GiNaC
#endif // ndef __GINAC_REGISTRAR_H__