1 files changed, 25 insertions, 1 deletions

diff --git a/src/cachecalc.h b/src/cachecalc.h
index 289c4ac..dd9712f 100644
--- a/src/cachecalc.h
+++ b/src/cachecalc.h
@@ -1,25 +1,49 @@
 #ifndef BU_CACHE_CALC_H
 #define BU_CACHE_CALC_H
 
+#include "bu/trace.h"
+
 namespace Bu
 {
+        template<class obtype, class keytype> class Cache;
+
         template<class obtype, class keytype>
         class CacheCalc
         {
+        friend class Cache<obtype, keytype>;
+        private:
+                typedef Cache<obtype, keytype> MyCache;
         public:
-                CacheCalc()
+                CacheCalc() :
+                        pCache( (MyCache *)0 )
                 {
+                        TRACE();
                 }
 
                 virtual ~CacheCalc()
                 {
+                        TRACE();
                 }
 
                 virtual void onLoad( obtype *pSrc, const keytype &key )=0;
                 virtual void onUnload( obtype *pSrc, const keytype &key )=0;
                 virtual void onTick() { };
 
+        protected:
+                MyCache *getCache()
+                {
+                        TRACE();
+                        return pCache;
+                }
+
         private:
+                void setCache( MyCache *pCache )
+                {
+                        TRACE();
+                        this->pCache = pCache;
+                }
+
+                MyCache *pCache;
         };
 };