1 files changed, 74 insertions, 0 deletions

diff --git a/lib/pointer_map.cc b/lib/pointer_map.cc
new file mode 100644
index 0000000..44f7ebc
--- a/dev/null
+++ b/lib/pointer_map.cc
@@ -0,0 +1,74 @@
+#include <cassert>
+#include <map>
+#include <string>
+using namespace std;
+#include "konforka/exception.h"
+#include "konforka/pointer_map.h"
+
+namespace konforka {
+
+    /*
+     * to_t->from_t->from->to
+     */
+    class mpointer_t {
+        public:
+            void *ptr;
+            int ref;
+
+            mpointer_t() : ptr(0), ref(0) { }
+
+            void _ref() { ref++; }
+            void _unref() { ref--; }
+    };
+    typedef map<void*,mpointer_t> map_ptr_1_t;
+    typedef map<string,map_ptr_1_t> map_ptr_2_t;
+    typedef map<string,map_ptr_2_t> map_ptr_t;
+
+    static map_ptr_t pointer_map;
+    
+    void _map_pointer(const type_info& tf,void *pf,const type_info& tt,void *pt) {
+        mpointer_t& mpt = pointer_map[tf.name()][tt.name()][pf];
+        if(mpt.ref) {
+            if(mpt.ptr != pt)
+                throw exception(CODEPOINT,"trying to remap existing pointer with different value");
+        }else{
+            mpt.ptr = pt;
+        }
+        mpt.ref++;
+    }
+    void _unmap_pointer(const type_info& tf,void *pf,const type_info& tt,void *pt) {
+        map_ptr_t::iterator i1 = pointer_map.find(tf.name());
+        if(i1==pointer_map.end())
+            throw exception(CODEPOINT,"no pointer of such type has ever been mapped");
+        map_ptr_2_t::iterator i2 = i1->second.find(tt.name());
+        if(i2==i1->second.end())
+            throw exception(CODEPOINT,"no pointer has ever been mapped to the object of this type");
+        map_ptr_1_t::iterator i3 = i2->second.find(pf);
+        if(i3==i2->second.end())
+            throw exception(CODEPOINT,"this pointer has never been mapped");
+        assert(i3->second.ref>0);
+        if(--(i3->second.ref))
+            return;
+        i2->second.erase(i3);
+        if(!i2->second.empty())
+            return;
+        i1->second.erase(i2);
+        if(!i1->second.empty())
+            return;
+        pointer_map.erase(i1);
+    }
+    void *_mapped_pointer(const type_info& tf,void *pf,const type_info& tt) {
+        map_ptr_t::iterator i1 = pointer_map.find(tf.name());
+        if(i1==pointer_map.end())
+            throw exception(CODEPOINT,"no pointer of such type has ever been mapped");
+        map_ptr_2_t::iterator i2 = i1->second.find(tt.name());
+        if(i2==i1->second.end())
+            throw exception(CODEPOINT,"no pointer has ever been mapped to the object of this type");
+        map_ptr_1_t::iterator i3 = i2->second.find(pf);
+        if(i3==i2->second.end())
+            throw exception(CODEPOINT,"this pointer has never been mapped");
+        assert(i3->second.ref>0);
+        return i3->second.ptr;
+    }
+
+}