include/library/memtracker.hpp

   1 #ifndef _library__memtracker__hpp__included__
   2 #define _library__memtracker__hpp__included__
   3
   4 #include "threads.hpp"
   5 #include <map>
   6
   7 class memtracker
   8 {
   9 public:
  10         memtracker();
  11         ~memtracker();
  12         void operator()(const char* category, ssize_t change);
  13         void reset(const char* category, size_t value);
  14         std::map<std::string, size_t> report();
  15         static memtracker& singleton();
  16         class autorelease
  17         {
  18         public:
  19                 autorelease(memtracker& _track, const char* cat, size_t amount)
  20                         : tracker(_track), category(cat), committed(amount)
  21                 {
  22                         tracker(category, committed);
  23                 }
  24                 ~autorelease()
  25                 {
  26                         tracker(category, -(ssize_t)committed);
  27                 }
  28                 void operator()(ssize_t delta)
  29                 {
  30                         if(delta < 0 && committed < (size_t)-delta) {
  31                                 tracker(category, -(ssize_t)committed);
  32                                 committed = 0;
  33                         } else {
  34                                 tracker(category, delta);
  35                                 committed = committed + delta;
  36                         }
  37                 }
  38         private:
  39                 memtracker& tracker;
  40                 const char* category;
  41                 size_t committed;
  42         };
  43 private:
  44         bool invalid;
  45         threads::lock mut;
  46         std::map<std::string, size_t> data;
  47         memtracker(const memtracker&);
  48         memtracker& operator=(const memtracker&);
  49 };
  50
  51
  52 #endif