]> www.ginac.de Git - ginac.git/blobdiff - ginac/archive.cpp
Fix some apparent typos and misindentations.
[ginac.git] / ginac / archive.cpp
index ff8e755fd5094a54d08772ebd8f63f3ce306801d..fc9ec2c4c64d042756d71350fd087900c0bd3a1e 100644 (file)
@@ -3,7 +3,7 @@
  *  Archiving of GiNaC expressions. */
 
 /*
- *  GiNaC Copyright (C) 1999-2000 Johannes Gutenberg University Mainz, Germany
+ *  GiNaC Copyright (C) 1999-2015 Johannes Gutenberg University Mainz, Germany
  *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
  *
  *  You should have received a copy of the GNU General Public License
  *  along with this program; if not, write to the Free Software
- *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  */
 
-#include <iostream>
-#include <stdexcept>
-
 #include "archive.h"
 #include "registrar.h"
 #include "ex.h"
+#include "lst.h"
+#ifdef HAVE_CONFIG_H
 #include "config.h"
-#include "utils.h"
+#endif
+#include "tostring.h"
+#include "version.h"
+
+#include <iostream>
+#include <stdexcept>
 
-#ifndef NO_NAMESPACE_GINAC
 namespace GiNaC {
-#endif // ndef NO_NAMESPACE_GINAC
 
 
-/** Archive an expression.
- *  @param ex  the expression to be archived
- *  @param name  name under which the expression is stored */
 void archive::archive_ex(const ex &e, const char *name)
 {
        // Create root node (which recursively archives the whole expression tree)
@@ -54,18 +53,19 @@ void archive::archive_ex(const ex &e, const char *name)
  *  @return ID of archived node */
 archive_node_id archive::add_node(const archive_node &n)
 {
-       // Search for node in nodes vector
-       std::vector<archive_node>::const_iterator i = nodes.begin(), iend = nodes.end();
-       archive_node_id id = 0;
-       while (i != iend) {
-               if (i->has_same_ex_as(n))
-                       return id;
-               i++; id++;
+       // Look if expression is known to be in some node already.
+       if (n.has_ex()) {
+               mapit i = exprtable.find(n.get_ex());
+               if (i != exprtable.end())
+                       return i->second;
+               nodes.push_back(n);
+               exprtable[n.get_ex()] = nodes.size() - 1;
+               return nodes.size() - 1;
        }
 
        // Not found, add archive_node to nodes vector
        nodes.push_back(n);
-       return id;
+       return nodes.size()-1;
 }
 
 
@@ -79,8 +79,6 @@ archive_node &archive::get_node(archive_node_id id)
 }
 
 
-/** Retrieve expression from archive by name.
- *  @param sym_lst  list of pre-defined symbols */
 ex archive::unarchive_ex(const lst &sym_lst, const char *name) const
 {
        // Find root node
@@ -92,27 +90,25 @@ ex archive::unarchive_ex(const lst &sym_lst, const char *name) const
                        goto found;
                i++;
        }
-       throw (std::logic_error("expression with name '" + name_string + "' not found in archive"));
+       throw (std::runtime_error("expression with name '" + name_string + "' not found in archive"));
 
 found:
        // Recursively unarchive all nodes, starting at the root node
-       return nodes[i->root].unarchive(sym_lst);
+       lst sym_lst_copy = sym_lst;
+       return nodes[i->root].unarchive(sym_lst_copy);
 }
 
-/** Retrieve expression from archive by index.
- *  @param sym_lst  list of pre-defined symbols */
-ex archive::unarchive_ex(const lst &sym_lst, unsigned int index) const
+ex archive::unarchive_ex(const lst &sym_lst, unsigned index) const
 {
        if (index >= exprs.size())
                throw (std::range_error("index of archived expression out of range"));
 
        // Recursively unarchive all nodes, starting at the root node
-       return nodes[exprs[index].root].unarchive(sym_lst);
+       lst sym_lst_copy = sym_lst;
+       return nodes[exprs[index].root].unarchive(sym_lst_copy);
 }
 
-/** Retrieve expression and its name from archive by index.
- *  @param sym_lst  list of pre-defined symbols */
-ex archive::unarchive_ex(const lst &sym_lst, std::string &name, unsigned int index) const
+ex archive::unarchive_ex(const lst &sym_lst, std::string &name, unsigned index) const
 {
        if (index >= exprs.size())
                throw (std::range_error("index of archived expression out of range"));
@@ -121,16 +117,23 @@ ex archive::unarchive_ex(const lst &sym_lst, std::string &name, unsigned int ind
        name = unatomize(exprs[index].name);
 
        // Recursively unarchive all nodes, starting at the root node
-       return nodes[exprs[index].root].unarchive(sym_lst);
+       lst sym_lst_copy = sym_lst;
+       return nodes[exprs[index].root].unarchive(sym_lst_copy);
 }
 
-
-/** Return number of archived expressions. */
-unsigned int archive::num_expressions(void) const
+unsigned archive::num_expressions() const
 {
        return exprs.size();
 }
 
+const archive_node &archive::get_top_node(unsigned index) const
+{
+       if (index >= exprs.size())
+               throw (std::range_error("index of archived expression out of range"));
+
+       return nodes[exprs[index].root];
+}
+
 
 /*
  *  Archive file format
@@ -172,9 +175,9 @@ unsigned int archive::num_expressions(void) const
  */
 
 /** Write unsigned integer quantity to stream. */
-static void write_unsigned(std::ostream &os, unsigned int val)
+static void write_unsigned(std::ostream &os, unsigned val)
 {
-       while (val > 0x80) {
+       while (val >= 0x80) {
                os.put((val & 0x7f) | 0x80);
                val >>= 7;
        }
@@ -182,15 +185,15 @@ static void write_unsigned(std::ostream &os, unsigned int val)
 }
 
 /** Read unsigned integer quantity from stream. */
-static unsigned int read_unsigned(std::istream &is)
+static unsigned read_unsigned(std::istream &is)
 {
        unsigned char b;
-       unsigned int ret = 0;
-       unsigned int shift = 0;
+       unsigned ret = 0;
+       unsigned shift = 0;
        do {
-        char b2;
+               char b2;
                is.get(b2);
-        b = b2;
+               b = b2;
                ret |= (b & 0x7f) << shift;
                shift += 7;
        } while (b & 0x80);
@@ -201,9 +204,9 @@ static unsigned int read_unsigned(std::istream &is)
 std::ostream &operator<<(std::ostream &os, const archive_node &n)
 {
        // Write properties
-       unsigned int num_props = n.props.size();
+       unsigned num_props = n.props.size();
        write_unsigned(os, num_props);
-       for (unsigned int i=0; i<num_props; i++) {
+       for (unsigned i=0; i<num_props; i++) {
                write_unsigned(os, n.props[i].type | (n.props[i].name << 3));
                write_unsigned(os, n.props[i].value);
        }
@@ -218,26 +221,26 @@ std::ostream &operator<<(std::ostream &os, const archive &ar)
        os.put('A');
        os.put('R');
        os.put('C');
-       write_unsigned(os, ARCHIVE_VERSION);
+       write_unsigned(os, GINACLIB_ARCHIVE_VERSION);
 
        // Write atoms
-       unsigned int num_atoms = ar.atoms.size();
+       unsigned num_atoms = ar.atoms.size();
        write_unsigned(os, num_atoms);
-       for (unsigned int i=0; i<num_atoms; i++)
+       for (unsigned i=0; i<num_atoms; i++)
                os << ar.atoms[i] << std::ends;
 
        // Write expressions
-       unsigned int num_exprs = ar.exprs.size();
+       unsigned num_exprs = ar.exprs.size();
        write_unsigned(os, num_exprs);
-       for (unsigned int i=0; i<num_exprs; i++) {
+       for (unsigned i=0; i<num_exprs; i++) {
                write_unsigned(os, ar.exprs[i].name);
                write_unsigned(os, ar.exprs[i].root);
        }
 
        // Write nodes
-       unsigned int num_nodes = ar.nodes.size();
+       unsigned num_nodes = ar.nodes.size();
        write_unsigned(os, num_nodes);
-       for (unsigned int i=0; i<num_nodes; i++)
+       for (unsigned i=0; i<num_nodes; i++)
                os << ar.nodes[i];
        return os;
 }
@@ -246,10 +249,10 @@ std::ostream &operator<<(std::ostream &os, const archive &ar)
 std::istream &operator>>(std::istream &is, archive_node &n)
 {
        // Read properties
-       unsigned int num_props = read_unsigned(is);
+       unsigned num_props = read_unsigned(is);
        n.props.resize(num_props);
-       for (unsigned int i=0; i<num_props; i++) {
-               unsigned int name_type = read_unsigned(is);
+       for (unsigned i=0; i<num_props; i++) {
+               unsigned name_type = read_unsigned(is);
                n.props[i].type = (archive_node::property_type)(name_type & 7);
                n.props[i].name = name_type >> 3;
                n.props[i].value = read_unsigned(is);
@@ -265,29 +268,33 @@ std::istream &operator>>(std::istream &is, archive &ar)
        is.get(c1); is.get(c2); is.get(c3); is.get(c4);
        if (c1 != 'G' || c2 != 'A' || c3 != 'R' || c4 != 'C')
                throw (std::runtime_error("not a GiNaC archive (signature not found)"));
-       unsigned int version = read_unsigned(is);
-       if (version > ARCHIVE_VERSION || version < ARCHIVE_VERSION - ARCHIVE_AGE)
-               throw (std::runtime_error("archive version " + ToString(version) + " cannot be read by this GiNaC library (which supports versions " + ToString(ARCHIVE_VERSION-ARCHIVE_AGE) + " thru " + ToString(ARCHIVE_VERSION)));
+       static const unsigned max_version = GINACLIB_ARCHIVE_VERSION;
+       static const unsigned min_version = GINACLIB_ARCHIVE_VERSION - GINACLIB_ARCHIVE_AGE;
+       unsigned version = read_unsigned(is);
+       if ((version > max_version) || (version < min_version))
+               throw (std::runtime_error("archive version " + ToString(version) + " cannot be read by this GiNaC library (which supports versions " + ToString(min_version) + " thru " + ToString(max_version)));
 
        // Read atoms
-       unsigned int num_atoms = read_unsigned(is);
+       unsigned num_atoms = read_unsigned(is);
        ar.atoms.resize(num_atoms);
-       for (unsigned int i=0; i<num_atoms; i++)
+       for (unsigned i=0; i<num_atoms; i++) {
                getline(is, ar.atoms[i], '\0');
+               ar.inverse_atoms[ar.atoms[i]] = i;
+       }
 
        // Read expressions
-       unsigned int num_exprs = read_unsigned(is);
+       unsigned num_exprs = read_unsigned(is);
        ar.exprs.resize(num_exprs);
-       for (unsigned int i=0; i<num_exprs; i++) {
+       for (unsigned i=0; i<num_exprs; i++) {
                archive_atom name = read_unsigned(is);
                archive_node_id root = read_unsigned(is);
                ar.exprs[i] = archive::archived_ex(name, root);
        }
 
        // Read nodes
-       unsigned int num_nodes = read_unsigned(is);
+       unsigned num_nodes = read_unsigned(is);
        ar.nodes.resize(num_nodes, ar);
-       for (unsigned int i=0; i<num_nodes; i++)
+       for (unsigned i=0; i<num_nodes; i++)
                is >> ar.nodes[i];
        return is;
 }
@@ -297,17 +304,15 @@ std::istream &operator>>(std::istream &is, archive &ar)
  *  represents the string). */
 archive_atom archive::atomize(const std::string &s) const
 {
-       // Search for string in atoms vector
-       std::vector<std::string>::const_iterator i = atoms.begin(), iend = atoms.end();
-       archive_atom id = 0;
-       while (i != iend) {
-               if (*i == s)
-                       return id;
-               i++; id++;
-       }
+       // Search for string in inverse_atoms map.
+       inv_at_cit i = inverse_atoms.find(s);
+       if (i!=inverse_atoms.end())
+               return i->second;
 
        // Not found, add to atoms vector
+       archive_atom id = atoms.size();
        atoms.push_back(s);
+       inverse_atoms[s] = id;
        return id;
 }
 
@@ -315,24 +320,17 @@ archive_atom archive::atomize(const std::string &s) const
 const std::string &archive::unatomize(archive_atom id) const
 {
        if (id >= atoms.size())
-               throw (std::range_error("archive::unatomizee(): atom ID out of range"));
+               throw (std::range_error("archive::unatomize(): atom ID out of range"));
 
        return atoms[id];
 }
 
 
-/** Copy constructor of archive_node. */
-archive_node::archive_node(const archive_node &other)
-  : a(other.a), props(other.props), has_expression(other.has_expression), e(other.e)
-{
-}
-
-
 /** Assignment operator of archive_node. */
 const archive_node &archive_node::operator=(const archive_node &other)
 {
        if (this != &other) {
-               a = other.a;
+               // archive &a member doesn't get copied
                props = other.props;
                has_expression = other.has_expression;
                e = other.e;
@@ -359,26 +357,43 @@ bool archive_node::has_same_ex_as(const archive_node &other) const
        return e.bp == other.e.bp;
 }
 
+archive_node::archive_node_cit
+               archive_node::find_first(const std::string &name) const
+{      
+       archive_atom name_atom = a.atomize(name);
+       for (archive_node_cit i=props.begin(); i!=props.end(); ++i)
+               if (i->name == name_atom)
+                       return i;
+       return props.end();;
+}
+
+archive_node::archive_node_cit
+               archive_node::find_last(const std::string &name) const
+{
+       archive_atom name_atom = a.atomize(name);
+       for (archive_node_cit i=props.end(); i!=props.begin();) {
+               --i;
+               if (i->name == name_atom)
+                       return i;
+       }
+       return props.end();
+}
 
-/** Add property of type "bool" to node. */
 void archive_node::add_bool(const std::string &name, bool value)
 {
        props.push_back(property(a.atomize(name), PTYPE_BOOL, value));
 }
 
-/** Add property of type "unsigned int" to node. */
-void archive_node::add_unsigned(const std::string &name, unsigned int value)
+void archive_node::add_unsigned(const std::string &name, unsigned value)
 {
        props.push_back(property(a.atomize(name), PTYPE_UNSIGNED, value));
 }
 
-/** Add property of type "string" to node. */
 void archive_node::add_string(const std::string &name, const std::string &value)
 {
        props.push_back(property(a.atomize(name), PTYPE_STRING, a.atomize(value)));
 }
 
-/** Add property of type "ex" to node. */
 void archive_node::add_ex(const std::string &name, const ex &value)
 {
        // Recursively create an archive_node and add its ID to the properties of this node
@@ -387,79 +402,134 @@ void archive_node::add_ex(const std::string &name, const ex &value)
 }
 
 
-/** Retrieve property of type "bool" from node.
- *  @return "true" if property was found, "false" otherwise */
-bool archive_node::find_bool(const std::string &name, bool &ret) const
+bool archive_node::find_bool(const std::string &name, bool &ret, unsigned index) const
 {
        archive_atom name_atom = a.atomize(name);
-       std::vector<property>::const_iterator i = props.begin(), iend = props.end();
+       archive_node_cit i = props.begin(), iend = props.end();
+       unsigned found_index = 0;
        while (i != iend) {
                if (i->type == PTYPE_BOOL && i->name == name_atom) {
-                       ret = i->value;
-                       return true;
+                       if (found_index == index) {
+                               ret = i->value;
+                               return true;
+                       }
+                       found_index++;
                }
                i++;
        }
        return false;
 }
 
-/** Retrieve property of type "unsigned" from node.
- *  @return "true" if property was found, "false" otherwise */
-bool archive_node::find_unsigned(const std::string &name, unsigned int &ret) const
+bool archive_node::find_unsigned(const std::string &name, unsigned &ret, unsigned index) const
 {
        archive_atom name_atom = a.atomize(name);
-       std::vector<property>::const_iterator i = props.begin(), iend = props.end();
+       archive_node_cit i = props.begin(), iend = props.end();
+       unsigned found_index = 0;
        while (i != iend) {
                if (i->type == PTYPE_UNSIGNED && i->name == name_atom) {
-                       ret = i->value;
-                       return true;
+                       if (found_index == index) {
+                               ret = i->value;
+                               return true;
+                       }
+                       found_index++;
                }
                i++;
        }
        return false;
 }
 
-/** Retrieve property of type "string" from node.
- *  @return "true" if property was found, "false" otherwise */
-bool archive_node::find_string(const std::string &name, std::string &ret) const
+bool archive_node::find_string(const std::string &name, std::string &ret, unsigned index) const
 {
        archive_atom name_atom = a.atomize(name);
-       std::vector<property>::const_iterator i = props.begin(), iend = props.end();
+       archive_node_cit i = props.begin(), iend = props.end();
+       unsigned found_index = 0;
        while (i != iend) {
                if (i->type == PTYPE_STRING && i->name == name_atom) {
-                       ret = a.unatomize(i->value);
-                       return true;
+                       if (found_index == index) {
+                               ret = a.unatomize(i->value);
+                               return true;
+                       }
+                       found_index++;
                }
                i++;
        }
        return false;
 }
 
-/** Retrieve property of type "ex" from node.
- *  @return "true" if property was found, "false" otherwise */
-bool archive_node::find_ex(const std::string &name, ex &ret, const lst &sym_lst, unsigned int index) const
+void archive_node::find_ex_by_loc(archive_node_cit loc, ex &ret, lst &sym_lst)
+               const
+{
+       ret = a.get_node(loc->value).unarchive(sym_lst);
+}
+
+bool archive_node::find_ex(const std::string &name, ex &ret, lst &sym_lst, unsigned index) const
 {
        archive_atom name_atom = a.atomize(name);
-       std::vector<property>::const_iterator i = props.begin(), iend = props.end();
-       unsigned int found_index = 0;
+       archive_node_cit i = props.begin(), iend = props.end();
+       unsigned found_index = 0;
        while (i != iend) {
                if (i->type == PTYPE_NODE && i->name == name_atom) {
-                       if (found_index == index)
-                               goto found;
+                       if (found_index == index) {
+                               ret = a.get_node(i->value).unarchive(sym_lst);
+                               return true;
+                       }
                        found_index++;
                }
                i++;
        }
        return false;
+}
 
-found:
-       ret = a.get_node(i->value).unarchive(sym_lst);
-       return true;
+const archive_node &archive_node::find_ex_node(const std::string &name, unsigned index) const
+{
+       archive_atom name_atom = a.atomize(name);
+       archive_node_cit i = props.begin(), iend = props.end();
+       unsigned found_index = 0;
+       while (i != iend) {
+               if (i->type == PTYPE_NODE && i->name == name_atom) {
+                       if (found_index == index)
+                               return a.get_node(i->value);
+                       found_index++;
+               }
+               i++;
+       }
+       throw (std::runtime_error("property with name '" + name + "' not found in archive node"));
+}
+
+
+void archive_node::get_properties(propinfovector &v) const
+{
+       v.clear();
+       archive_node_cit i = props.begin(), iend = props.end();
+       while (i != iend) {
+               property_type type = i->type;
+               std::string name = a.unatomize(i->name);
+
+               propinfovector::iterator a = v.begin(), aend = v.end();
+               bool found = false;
+               while (a != aend) {
+                       if (a->type == type && a->name == name) {
+                               a->count++;
+                               found = true;
+                               break;
+                       }
+                       ++a;
+               }
+               if (!found)
+                       v.push_back(property_info(type, name));
+               i++;
+       }       
 }
 
+static synthesize_func find_factory_fcn(const std::string& name)
+{
+       static unarchive_table_t the_table;
+       synthesize_func ret = the_table.find(name);
+       return ret;
+}
 
 /** Convert archive node to GiNaC expression. */
-ex archive_node::unarchive(const lst &sym_lst) const
+ex archive_node::unarchive(lst &sym_lst) const
 {
        // Already unarchived? Then return cached unarchived expression.
        if (has_expression)
@@ -469,48 +539,69 @@ ex archive_node::unarchive(const lst &sym_lst) const
        std::string class_name;
        if (!find_string("class", class_name))
                throw (std::runtime_error("archive node contains no class name"));
-       unarch_func f = find_unarch_func(class_name);
 
        // Call instantiation function
-       e = f(*this, sym_lst);
+       synthesize_func factory_fcn = find_factory_fcn(class_name);
+       ptr<basic> obj(factory_fcn());
+       obj->setflag(status_flags::dynallocated);
+       obj->read_archive(*this, sym_lst);
+       e = ex(*obj);
        has_expression = true;
        return e;
 }
 
+int unarchive_table_t::usecount = 0;
+unarchive_map_t* unarchive_table_t::unarch_map = 0;
 
-/** Assignment operator of property. */
-const archive_node::property &archive_node::property::operator=(const property &other)
+unarchive_table_t::unarchive_table_t()
 {
-       if (this != &other) {
-               type = other.type;
-               name = other.name;
-               value = other.value;
-       }
-       return *this;
+       if (usecount == 0)
+               unarch_map = new unarchive_map_t();
+       ++usecount;
+}
+
+synthesize_func unarchive_table_t::find(const std::string& classname) const
+{
+       unarchive_map_t::const_iterator i = unarch_map->find(classname);
+       if (i != unarch_map->end())
+               return i->second;
+       throw std::runtime_error(std::string("no unarchiving function for \"")
+                       + classname + "\" class");
 }
 
+void unarchive_table_t::insert(const std::string& classname, synthesize_func f)
+{
+       if (unarch_map->find(classname) != unarch_map->end())
+               throw std::runtime_error(std::string("Class \"" + classname
+                                       + "\" is already registered"));
+       unarch_map->operator[](classname) = f;
+}
 
-/** Clear all archived expressions. */
-void archive::clear(void)
+unarchive_table_t::~unarchive_table_t()
+{
+       if (--usecount == 0)
+               delete unarch_map;
+}
+
+
+void archive::clear()
 {
        atoms.clear();
+       inverse_atoms.clear();
        exprs.clear();
        nodes.clear();
+       exprtable.clear();
 }
 
 
 /** Delete cached unarchived expressions in all archive_nodes (mainly for debugging). */
-void archive::forget(void)
+void archive::forget()
 {
-       std::vector<archive_node>::iterator i = nodes.begin(), iend = nodes.end();
-       while (i != iend) {
-               i->forget();
-               i++;
-       }
+       for_each(nodes.begin(), nodes.end(), std::mem_fun_ref(&archive_node::forget));
 }
 
 /** Delete cached unarchived expressions from node (for debugging). */
-void archive_node::forget(void)
+void archive_node::forget()
 {
        has_expression = false;
        e = 0;
@@ -536,7 +627,7 @@ void archive::printraw(std::ostream &os) const
        os << "Expressions:\n";
        {
                std::vector<archived_ex>::const_iterator i = exprs.begin(), iend = exprs.end();
-               unsigned int index = 0;
+               unsigned index = 0;
                while (i != iend) {
                        os << " " << index << " \"" << unatomize(i->name) << "\" root node " << i->root << std::endl;
                        i++; index++;
@@ -567,7 +658,7 @@ void archive_node::printraw(std::ostream &os) const
                os << "\n";
 
        // Dump properties
-       std::vector<property>::const_iterator i = props.begin(), iend = props.end();
+       archive_node_cit i = props.begin(), iend = props.end();
        while (i != iend) {
                os << "  ";
                switch (i->type) {
@@ -584,13 +675,11 @@ void archive_node::printraw(std::ostream &os) const
 
 /** Create a dummy archive.  The intention is to fill archive_node's default
  *  ctor, which is currently a Cint-requirement. */
-archive* archive_node::dummy_ar_creator(void)
+archive* archive_node::dummy_ar_creator()
 {
        static archive* some_ar = new archive;
        return some_ar;
 }
 
 
-#ifndef NO_NAMESPACE_GINAC
 } // namespace GiNaC
-#endif // ndef NO_NAMESPACE_GINAC