X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=laserbrain_demo;a=blobdiff_plain;f=src%2Fdataset.inl;h=954081e58125e459ca515a072e626f5c9954117a;hp=ca01ea2d3336d450e74f7e960bc749e9ebb464b6;hb=8512530474f0e0827447b6c81302120592f0e297;hpb=59af92c432928d1a325393472cb8ec7a4a1594ad

diff --git a/src/dataset.inl b/src/dataset.inl
index ca01ea2..954081e 100644
--- a/src/dataset.inl
+++ b/src/dataset.inl
@@ -51,6 +51,28 @@ T DataSet<T>::get(const char *name) const
 	return res;
 }
 
+template <typename T>
+void DataSet<T>::add_loaded_callback(LoadedCallbackFunc func, void *cls)
+{
+	std::pair<LoadedCallbackFunc, void*> cb;
+	cb.first = func;
+	cb.second = cls;
+	loaded_cb.push_back(cb);
+}
+
+template <typename T>
+bool DataSet<T>::remove_loaded_callback(LoadedCallbackFunc func)
+{
+	int numcb = (int)loaded_cb.size();
+	for(int i=0; i<numcb; i++) {
+		if(loaded_cb[i].first == func) {
+			loaded_cb.erase(loaded_cb.begin() + i);
+			return true;
+		}
+	}
+	return false;
+}
+
 
 // --- static functions to pass as callback to resman ---
 
@@ -84,6 +106,11 @@ int DataSet<T>::dataset_done_func(int id, void *cls)
 	if(dset->done) {
 		dset->done(data);
 	}
+
+	int numcb = (int)dset->loaded_cb.size();
+	for(int i=0; i<numcb; i++) {
+		dset->loaded_cb[i].first(data, dset->loaded_cb[i].second);
+	}
 	return 0;
 }