Committed current version of visitor refactoring

1 files changed, 33 insertions, 0 deletions

diff --git a/src/nodecache.h b/src/nodecache.h
new file mode 100644
index 0000000..c5a5524
--- /dev/null
+++ b/src/nodecache.h
@@ -0,0 +1,33 @@
+#ifndef NODECACHE_H_
+#define NODECACHE_H_
+
+#include <vector>
+#include "node.h"
+
+template <class T>
+class NodeCache
+{
+public:
+  NodeCache() { }
+  virtual ~NodeCache() { }
+
+  const T & operator[](const AbstractNode &node) const {
+    if (this->cache.size() > node.index()) return this->cache[node.index()];
+    else return nullvalue;
+  }
+
+  void insert(const class AbstractNode &node, const T & value) {
+    this->cache.resize(node.index() + 1);
+    this->cache[node.index()] = value;
+  }
+
+  void remove(const class AbstractNode &node) {
+    if (this->cache.size() > node.index()) this->cache[node.index()] = nullvalue;
+  }
+
+private:
+  std::vector<T> cache;
+  T nullvalue;
+};
+
+#endif