]> www.ginac.de Git - ginac.git/blobdiff - ginac/archive.cpp
Faster archiving by adding a map from strings to idices in the atoms vector.
[ginac.git] / ginac / archive.cpp
index 2dc14f672c2f8252b6e62f5ab50ca4c39bd2f3d2..e08aaafb0713cf634106aab0acd48acf5e45d7a3 100644 (file)
@@ -3,7 +3,7 @@
  *  Archiving of GiNaC expressions. */
 
 /*
- *  GiNaC Copyright (C) 1999-2002 Johannes Gutenberg University Mainz, Germany
+ *  GiNaC Copyright (C) 1999-2007 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
@@ -17,7 +17,7 @@
  *
  *  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>
@@ -26,6 +26,7 @@
 #include "archive.h"
 #include "registrar.h"
 #include "ex.h"
+#include "lst.h"
 #include "config.h"
 #include "tostring.h"
 
@@ -49,18 +50,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;
 }
 
 
@@ -89,7 +91,8 @@ ex archive::unarchive_ex(const lst &sym_lst, const char *name) const
 
 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);
 }
 
 ex archive::unarchive_ex(const lst &sym_lst, unsigned index) const
@@ -98,7 +101,8 @@ ex archive::unarchive_ex(const lst &sym_lst, unsigned index) const
                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);
 }
 
 ex archive::unarchive_ex(const lst &sym_lst, std::string &name, unsigned index) const
@@ -110,10 +114,11 @@ ex archive::unarchive_ex(const lst &sym_lst, std::string &name, unsigned index)
        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);
 }
 
-unsigned archive::num_expressions(void) const
+unsigned archive::num_expressions() const
 {
        return exprs.size();
 }
@@ -183,9 +188,9 @@ static unsigned read_unsigned(std::istream &is)
        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);
@@ -267,8 +272,10 @@ std::istream &operator>>(std::istream &is, archive &ar)
        // Read atoms
        unsigned num_atoms = read_unsigned(is);
        ar.atoms.resize(num_atoms);
-       for (unsigned 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 num_exprs = read_unsigned(is);
@@ -292,17 +299,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;
 }
 
@@ -316,18 +321,11 @@ const std::string &archive::unatomize(archive_atom id) const
 }
 
 
-/** 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;
@@ -354,6 +352,27 @@ 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();
+}
 
 void archive_node::add_bool(const std::string &name, bool value)
 {
@@ -381,7 +400,7 @@ void archive_node::add_ex(const std::string &name, const ex &value)
 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) {
@@ -399,7 +418,7 @@ bool archive_node::find_bool(const std::string &name, bool &ret, unsigned index)
 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) {
@@ -417,7 +436,7 @@ bool archive_node::find_unsigned(const std::string &name, unsigned &ret, unsigne
 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) {
@@ -432,10 +451,16 @@ bool archive_node::find_string(const std::string &name, std::string &ret, unsign
        return false;
 }
 
-bool archive_node::find_ex(const std::string &name, ex &ret, const lst &sym_lst, unsigned 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();
+       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) {
@@ -453,7 +478,7 @@ bool archive_node::find_ex(const std::string &name, ex &ret, const lst &sym_lst,
 const archive_node &archive_node::find_ex_node(const std::string &name, 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_NODE && i->name == name_atom) {
@@ -470,7 +495,7 @@ const archive_node &archive_node::find_ex_node(const std::string &name, unsigned
 void archive_node::get_properties(propinfovector &v) const
 {
        v.clear();
-       std::vector<property>::const_iterator i = props.begin(), iend = props.end();
+       archive_node_cit i = props.begin(), iend = props.end();
        while (i != iend) {
                property_type type = i->type;
                std::string name = a.unatomize(i->name);
@@ -493,7 +518,7 @@ void archive_node::get_properties(propinfovector &v) const
 
 
 /** 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)
@@ -512,45 +537,24 @@ ex archive_node::unarchive(const lst &sym_lst) const
 }
 
 
-/** Assignment operator of property_info. */
-const archive_node::property_info &archive_node::property_info::operator=(const property_info &other)
-{
-       if (this != &other) {
-               type = other.type;
-               name = other.name;
-               count = other.count;
-       }
-       return *this;
-}
-
-/** Assignment operator of property. */
-const archive_node::property &archive_node::property::operator=(const property &other)
-{
-       if (this != &other) {
-               type = other.type;
-               name = other.name;
-               value = other.value;
-       }
-       return *this;
-}
-
-
-void archive::clear(void)
+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()
 {
        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;
@@ -607,7 +611,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) {
@@ -624,7 +628,7 @@ 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;