X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Floader.h;h=0c8118625918823ee2b5ebf6640bfd171a9d85fb;hb=256f7238bc60d6dcc31a564988f5cc02a60c4537;hp=da1da3a23f462f4ed4ec77efd4b8e4c7977336e2;hpb=1d9c21a8a301007fb242e05b69cc6390ec566273;p=libs%2Fdatafile.git

diff --git a/source/loader.h b/source/loader.h
index da1da3a..0c81186 100644
--- a/source/loader.h
+++ b/source/loader.h
@@ -1,15 +1,16 @@
 /* $Id$
 
 This file is part of libmspdatafile
-Copyright © 2006  Mikko Rasa, Mikkosoft Productions
+Copyright © 2006-2008  Mikko Rasa, Mikkosoft Productions
 Distributed under the LGPL
 */
 
 #ifndef MSP_DATAFILE_LOADER_H_
 #define MSP_DATAFILE_LOADER_H_
 
-#include <fstream>
 #include <map>
+#include <msp/io/buffered.h>
+#include <msp/io/file.h>
 #include "except.h"
 #include "parser.h"
 #include "statement.h"
@@ -49,7 +50,7 @@ public:
 	LoaderFunc0(FuncType f): func(f) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=0) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=0) throw TypeError("Wrong number of arguments");
 		(dynamic_cast<L &>(l).*func)();
 	};
 private:
@@ -69,7 +70,7 @@ public:
 	LoaderFunc1(FuncType f): func(f) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=1) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=1) throw TypeError("Wrong number of arguments");
 		(dynamic_cast<L &>(l).*func)(st.args[0].get<A0>());
 	}
 private:
@@ -109,7 +110,7 @@ public:
 	LoaderFunc2(FuncType f): func(f) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=2) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=2) throw TypeError("Wrong number of arguments");
 		(dynamic_cast<L &>(l).*func)(st.args[0].get<A0>(), st.args[1].get<A1>());
 	}
 private:
@@ -126,7 +127,7 @@ public:
 	LoaderFunc3(FuncType f): func(f) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=3) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=3) throw TypeError("Wrong number of arguments");
 		(dynamic_cast<L &>(l).*func)(st.args[0].get<A0>(), st.args[1].get<A1>(), st.args[2].get<A2>());
 	}
 private:
@@ -143,7 +144,7 @@ public:
 	LoaderFunc4(FuncType f): func(f) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=4) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=4) throw TypeError("Wrong number of arguments");
 		(dynamic_cast<L &>(l).*func)(st.args[0].get<A0>(), st.args[1].get<A1>(), st.args[2].get<A2>(), st.args[3].get<A3>());
 	}
 private:
@@ -151,6 +152,23 @@ private:
 };
 
 
+template<typename L, typename A0, typename A1, typename A2, typename A3, typename A4>
+class LoaderFunc5: public LoaderAction
+{
+public:
+	typedef void (L::*FuncType)(A0, A1, A2, A3, A4);
+
+	LoaderFunc5(FuncType f): func(f) { }
+	void execute(Loader &l, const Statement &st) const
+	{
+		if(st.args.size()!=5) throw TypeError("Wrong number of arguments");
+		(dynamic_cast<L &>(l).*func)(st.args[0].get<A0>(), st.args[1].get<A1>(), st.args[2].get<A2>(), st.args[3].get<A3>(), st.args[4].get<A4>());
+	}
+private:
+	FuncType func;
+};
+
+
 template<typename L, typename T0>
 class LoadValue1: public LoaderAction
 {
@@ -160,7 +178,7 @@ public:
 	LoadValue1(Pointer0Type p0): ptr0(p0) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=1) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=1) throw TypeError("Wrong number of arguments");
 		dynamic_cast<typename L::Loader &>(l).get_object().*ptr0=st.args[0].get<T0>();
 	}
 private:
@@ -177,9 +195,9 @@ public:
 	LoadValue1(Pointer0Type p0): ptr0(p0) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=1) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=1) throw TypeError("Wrong number of arguments");
 		typename L::Loader &ldr=dynamic_cast<typename L::Loader &>(l);
-		ldr.get_object().*ptr0=&ldr.get_collection().template get<T0>(st.args[0].get<std::string>());
+		ldr.get_object().*ptr0=ldr.get_collection().template get<T0>(st.args[0].get<std::string>());
 	}
 private:
 	Pointer0Type ptr0;
@@ -196,7 +214,7 @@ public:
 	LoadValue2(Pointer0Type p0, Pointer1Type p1): ptr0(p0), ptr1(p1) { }
 	void execute(Loader &l, const Statement &st) const
 	{
-		if(st.args.size()!=2) throw TypeError(st.get_location()+": Wrong number of arguments");
+		if(st.args.size()!=2) throw TypeError("Wrong number of arguments");
 		dynamic_cast<typename L::Loader &>(l).get_object().*ptr0=st.args[0].get<T0>();
 		dynamic_cast<typename L::Loader &>(l).get_object().*ptr1=st.args[1].get<T1>();
 	}
@@ -239,40 +257,44 @@ protected:
 	*/
 	template<typename L>
 	void add(const std::string &k, void (L::*func)())
-	{ actions.insert(typename ActionMap::value_type(k, new LoaderFunc0<L>(func))); }
+	{ add(k, new LoaderFunc0<L>(func)); }
 
 	template<typename L, typename A0>
 	void add(const std::string &k, void (L::*func)(A0))
-	{ actions.insert(typename ActionMap::value_type(k, new LoaderFunc1<L, A0>(func))); }
+	{ add(k, new LoaderFunc1<L, A0>(func)); }
 
 	template<typename L, typename A0, typename A1>
 	void add(const std::string &k, void (L::*func)(A0, A1))
-	{ actions.insert(typename ActionMap::value_type(k, new LoaderFunc2<L, A0, A1>(func))); }
+	{ add(k, new LoaderFunc2<L, A0, A1>(func)); }
 
 	template<typename L, typename A0, typename A1, typename A2>
 	void add(const std::string &k, void (L::*func)(A0, A1, A2))
-	{ actions.insert(typename ActionMap::value_type(k, new LoaderFunc3<L, A0, A1, A2>(func))); }
+	{ add(k, new LoaderFunc3<L, A0, A1, A2>(func)); }
 
 	template<typename L, typename A0, typename A1, typename A2, typename A3>
 	void add(const std::string &k, void (L::*func)(A0, A1, A2, A3))
-	{ actions.insert(typename ActionMap::value_type(k, new LoaderFunc4<L, A0, A1, A2, A3>(func))); }
+	{ add(k, new LoaderFunc4<L, A0, A1, A2, A3>(func)); }
+
+	template<typename L, typename A0, typename A1, typename A2, typename A3, typename A4>
+	void add(const std::string &k, void (L::*func)(A0, A1, A2, A3, A4))
+	{ add(k, new LoaderFunc5<L, A0, A1, A2, A3, A4>(func)); }
 
 	/**
 	Adds a keyword that is loaded into a variable of the loaded object.
 	*/
 	template<typename L, typename T0>
 	void add(const std::string &k, T0 L::*p0)
-	{ actions.insert(typename ActionMap::value_type(k, new LoadValue1<L, T0>(p0))); }
+	{ add(k, new LoadValue1<L, T0>(p0)); }
 
 	template<typename L, typename T0, typename T1>
 	void add(const std::string &k, T0 L::*p0, T1 L::*p1)
-	{ actions.insert(typename ActionMap::value_type(k, new LoadValue2<L, T0, T1>(p0, p1))); }
+	{ add(k, new LoadValue2<L, T0, T1>(p0, p1)); }
 
 	/**
 	Adds a keyword that is recognized but ignored.
 	*/
 	void add(const std::string &k)
-	{ actions.insert(ActionMap::value_type(k, 0)); }
+	{ add(k, 0); }
 
 	/**
 	Loads a sub-object from the statement being processed.  The Loader class of
@@ -280,37 +302,27 @@ protected:
 	*/
 	template<typename S>
 	void load_sub(S &s)
-	{ load_sub<typename S::Loader, S>(s); }
-
-	/**
-	Loads a sub-object with a custom Loader class.
-	*/
-	template<typename L, typename S>
-	void load_sub(S &s)
 	{
-		if(!cur_st)
-			throw InvalidState("load_sub called without current statement");
-		L loader(s);
-		loader.load(*cur_st);
+		typename S::Loader ldr(s);
+		load_sub_with(ldr);
 	}
 
-	template<typename S, typename T>
-	void load_sub(S &s, T &p)
-	{ load_sub<typename S::Loader, S, T>(s, p); }
-
 	/**
-	Loads a sub-object with a custom Loader class that takes one argument in
-	addition to to object to be loaded.
+	Loads a sub-object from the statement being processed with an extra parameter
+	for the Loader.  The Loader class of the sub-object is automatically used.
 	*/
-	template<typename L, typename S, typename T>
+	template<typename S, typename T>
 	void load_sub(S &s, T &p)
 	{
-		if(!cur_st)
-			throw InvalidState("load_sub called without current statement");
-		L loader(s, p);
-		loader.load(*cur_st);
+		typename S::Loader ldr(s, p);
+		load_sub_with(ldr);
 	}
 
+	/**
+	Processes the current statement's substatements with another Loader.
+	*/
+	void load_sub_with(Loader &);
+
 	/**
 	Returns the source of the statement being processed.  This can be used to
 	implement relative paths in include-like statements.  Note that the source
@@ -322,12 +334,15 @@ protected:
 			throw InvalidState("get_source called without current statement");
 		return cur_st->source;
 	}
+
+	virtual void finish() { }
 private:
 	typedef std::map<std::string, LoaderAction *> ActionMap;
 
 	ActionMap       actions;
 	const Statement *cur_st;
 
+	void add(const std::string &, LoaderAction *);
 	void load_statement(const Statement &st);
 };
 
@@ -337,15 +352,25 @@ Loads an object from a file.  The object must have a public Loader class.
 template<typename T>
 void load(T &obj, const std::string &fn)
 {
-	std::ifstream in(fn.c_str());
-	if(!in)
-		throw Exception("Couldn't open "+fn);
+	IO::File in(fn);
+	IO::Buffered buf(in);
 
-	Parser parser(in, fn);
+	Parser parser(buf, fn);
 	typename T::Loader loader(obj);
 	loader.load(parser);
 }
 
+template<typename T, typename U>
+void load(T &obj, const std::string &fn, U &arg)
+{
+	IO::File in(fn);
+	IO::Buffered buf(in);
+
+	Parser parser(in, fn);
+	typename T::Loader loader(obj, arg);
+	loader.load(parser);
+}
+
 } // namespace DataFile
 } // namespace Msp