git.mutantstargoat.com Git - laserbrain_demo/blob - src/dataset.inl

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include "logger.h"
   4
   5 template <typename T>
   6 DataSet<T>::DataSet(T (*create_func)(), bool (*load_func)(T, const char*), bool (*done_func)(T), void (*destr_func)(T))
   7 {
   8         create = create_func;
   9         load = load_func;
  10         done = done_func;
  11         destroy = destr_func;
  12
  13         rman = resman_create();
  14         resman_set_load_func(rman, dataset_load_func, this);
  15         resman_set_done_func(rman, dataset_done_func, this);
  16         resman_set_destroy_func(rman, dataset_destroy_func, this);
  17 }
  18
  19 template <typename T>
  20 DataSet<T>::~DataSet()
  21 {
  22         resman_free(rman);
  23 }
  24
  25 template <typename T>
  26 void DataSet<T>::clear()
  27 {
  28         resman_free(rman);
  29         data.clear();
  30
  31         rman = resman_create();
  32 }
  33
  34 template <typename T>
  35 void DataSet<T>::update()
  36 {
  37         resman_poll(rman);
  38 }
  39
  40 template <typename T>
  41 T DataSet<T>::get(const char *name) const
  42 {
  43         typename std::map<std::string, T>::const_iterator iter = data.find(name);
  44         if(iter != data.end()) {
  45                 return iter->second;
  46         }
  47
  48         T res = create();
  49         data[name] = res;
  50         resman_add(rman, name, res);
  51         return res;
  52 }
  53
  54 template <typename T>
  55 int DataSet<T>::pending() const
  56 {
  57         return resman_pending(rman);
  58 }
  59
  60 template <typename T>
  61 void DataSet<T>::wait() const
  62 {
  63         resman_waitall(rman);
  64 }
  65
  66 template <typename T>
  67 void DataSet<T>::add_loaded_callback(LoadedCallbackFunc func, void *cls)
  68 {
  69         std::pair<LoadedCallbackFunc, void*> cb;
  70         cb.first = func;
  71         cb.second = cls;
  72         loaded_cb.push_back(cb);
  73 }
  74
  75 template <typename T>
  76 bool DataSet<T>::remove_loaded_callback(LoadedCallbackFunc func)
  77 {
  78         int numcb = (int)loaded_cb.size();
  79         for(int i=0; i<numcb; i++) {
  80                 if(loaded_cb[i].first == func) {
  81                         loaded_cb.erase(loaded_cb.begin() + i);
  82                         return true;
  83                 }
  84         }
  85         return false;
  86 }
  87
  88
  89 // --- static functions to pass as callback to resman ---
  90
  91 template <typename T>
  92 int DataSet<T>::dataset_load_func(const char *fname, int id, void *cls)
  93 {
  94         DataSet<T> *dset = (DataSet<T>*)cls;
  95         T data = (T)resman_get_res_data(dset->rman, id);
  96         if(!data) return -1;
  97
  98         if(!dset->load(data, fname)) {
  99                 return -1;
 100         }
 101         return 0;
 102 }
 103
 104 template <typename T>
 105 int DataSet<T>::dataset_done_func(int id, void *cls)
 106 {
 107         DataSet<T> *dset = (DataSet<T>*)cls;
 108
 109         T data = (T)resman_get_res_data(dset->rman, id);
 110         int load_res = resman_get_res_result(dset->rman, id);
 111
 112         if(load_res != 0) {
 113                 error_log("failed to load resource %d (%s)\n", id, resman_get_res_name(dset->rman, id));
 114         } else {
 115                 info_log("done loading resource %d (%s)\n", id, resman_get_res_name(dset->rman, id));
 116         }
 117
 118         if(dset->done) {
 119                 dset->done(data);
 120         }
 121
 122         int numcb = (int)dset->loaded_cb.size();
 123         for(int i=0; i<numcb; i++) {
 124                 dset->loaded_cb[i].first(data, dset->loaded_cb[i].second);
 125         }
 126         return 0;
 127 }
 128
 129 template <typename T>
 130 void DataSet<T>::dataset_destroy_func(int id, void *cls)
 131 {
 132         DataSet<T> *dset = (DataSet<T>*)cls;
 133         T data = (T)resman_get_res_data(dset->rman, id);
 134
 135         if(dset->destroy) {
 136                 dset->destroy(data);
 137         }
 138 }