X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fdynamicobjectloader.h;h=ee1827bbd067f10e32123bde3e484743e64d0495;hb=daca21051927eabee098e422fe5a0990acacfb96;hp=c83d91407f4312520911807d67145cfb63a54282;hpb=794461b9f73a20c086f456459e39a98387060c3a;p=libs%2Fdatafile.git

diff --git a/source/dynamicobjectloader.h b/source/dynamicobjectloader.h
index c83d914..ee1827b 100644
--- a/source/dynamicobjectloader.h
+++ b/source/dynamicobjectloader.h
@@ -2,6 +2,7 @@
 #define MSP_DATAFILE_DYNAMICOBJECTLOADER_H_
 
 #include <msp/core/typeregistry.h>
+#include "collection.h"
 #include "except.h"
 #include "loader.h"
 
@@ -30,9 +31,10 @@ protected:
 	typedef Msp::TypeRegistry<CreateObject, DynamicObjectLoader &> TypeRegistry;
 
 	Collection *coll;
-private:
 	T *object;
+private:
 	Loader *obj_loader;
+	std::function<void(Collection &, const std::string &, T *)> store_func;
 
 	static ActionMap shared_actions;
 
@@ -41,12 +43,17 @@ protected:
 public:
 	~DynamicObjectLoader() { delete object; delete obj_loader; }
 
-	T *get_object() { T *o = object; object = 0; return o; }
 private:
 	virtual void init_actions();
 
-	void type(const Symbol &);
+public:
+	T *get_object() { T *o = object; object = 0; return o; }
+	T *store_object(Collection &, const std::string &);
+
+protected:
+	virtual void type(const Symbol &);
 
+private:
 	template<typename U>
 	typename std::enable_if<NeedsCollection<typename U::Loader>::value, typename U::Loader *>::type create_object_loader(U &obj) const;
 
@@ -64,8 +71,8 @@ Loader::ActionMap DynamicObjectLoader<T, C>::shared_actions;
 template<typename T, typename C>
 DynamicObjectLoader<T, C>::DynamicObjectLoader(Collection *c):
 	coll(c),
-	object(0),
-	obj_loader(0)
+	object(nullptr),
+	obj_loader(nullptr)
 {
 	set_actions(shared_actions);
 }
@@ -76,6 +83,18 @@ void DynamicObjectLoader<T, C>::init_actions()
 	add("type", &DynamicObjectLoader::type);
 }
 
+template<typename T, typename C>
+T *DynamicObjectLoader<T, C>::store_object(Collection &c, const std::string &name)
+{
+	if(!store_func)
+		throw std::logic_error("no store function");
+
+	T *o = object;
+	store_func(c, name, object);
+	object = 0;
+	return o;
+}
+
 template<typename T, typename C>
 void DynamicObjectLoader<T, C>::type(const Symbol &t)
 {
@@ -110,6 +129,7 @@ void DynamicObjectLoader<T, C>::CreateObject<U>::operator()(const std::string &,
 	ldr.object = obj;
 	ldr.obj_loader = ldr.create_object_loader<U>(*obj);
 	ldr.add_auxiliary_loader(*ldr.obj_loader);
+	ldr.store_func = [](Collection &c, const std::string &n, T *o){ c.add(n, static_cast<U *>(o)); };
 }
 
 } // namespace DataFile