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_lookup(rman, name, res);
  51         return res;
  52 }
  53
  54 template <typename T>
  55 void DataSet<T>::add_loaded_callback(LoadedCallbackFunc func, void *cls)
  56 {
  57         std::pair<LoadedCallbackFunc, void*> cb;
  58         cb.first = func;
  59         cb.second = cls;
  60         loaded_cb.push_back(cb);
  61 }
  62
  63 template <typename T>
  64 bool DataSet<T>::remove_loaded_callback(LoadedCallbackFunc func)
  65 {
  66         int numcb = (int)loaded_cb.size();
  67         for(int i=0; i<numcb; i++) {
  68                 if(loaded_cb[i].first == func) {
  69                         loaded_cb.erase(loaded_cb.begin() + i);
  70                         return true;
  71                 }
  72         }
  73         return false;
  74 }
  75
  76
  77 // --- static functions to pass as callback to resman ---
  78
  79 template <typename T>
  80 int DataSet<T>::dataset_load_func(const char *fname, int id, void *cls)
  81 {
  82         DataSet<T> *dset = (DataSet<T>*)cls;
  83         T data = (T)resman_get_res_data(dset->rman, id);
  84         if(!data) return -1;
  85
  86         if(!dset->load(data, fname)) {
  87                 return -1;
  88         }
  89         return 0;
  90 }
  91
  92 template <typename T>
  93 int DataSet<T>::dataset_done_func(int id, void *cls)
  94 {
  95         DataSet<T> *dset = (DataSet<T>*)cls;
  96
  97         T data = (T)resman_get_res_data(dset->rman, id);
  98         int load_res = resman_get_res_result(dset->rman, id);
  99
 100         if(load_res != 0) {
 101                 error_log("failed to load resource %d (%s)\n", id, resman_get_res_name(dset->rman, id));
 102         } else {
 103                 info_log("done loading resource %d (%s)\n", id, resman_get_res_name(dset->rman, id));
 104         }
 105
 106         if(dset->done) {
 107                 dset->done(data);
 108         }
 109
 110         int numcb = (int)dset->loaded_cb.size();
 111         for(int i=0; i<numcb; i++) {
 112                 dset->loaded_cb[i].first(data, dset->loaded_cb[i].second);
 113         }
 114         return 0;
 115 }
 116
 117 template <typename T>
 118 void DataSet<T>::dataset_destroy_func(int id, void *cls)
 119 {
 120         DataSet<T> *dset = (DataSet<T>*)cls;
 121         T data = (T)resman_get_res_data(dset->rman, id);
 122
 123         if(dset->destroy) {
 124                 dset->destroy(data);
 125         }
 126 }