X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=tools%2Fsetupgen%2Fsetupgen.cpp;h=571396ff0cfad6dfe80a26a6e760327aa8a7acf6;hb=84516f86642ec6fae7f908b29649c75d2606e221;hp=083fc8a96be3e6ec158d18e10de7ca993b2f0bb0;hpb=f4ca190e869b9d03bf4f8b7b002c65af9f56c0ad;p=libs%2Fgame.git

diff --git a/tools/setupgen/setupgen.cpp b/tools/setupgen/setupgen.cpp
index 083fc8a..571396f 100644
--- a/tools/setupgen/setupgen.cpp
+++ b/tools/setupgen/setupgen.cpp
@@ -1,6 +1,8 @@
 #include "setupgen.h"
+#include <algorithm>
 #include <msp/core/getopt.h>
 #include <msp/core/maputils.h>
+#include <msp/fs/stat.h>
 #include <msp/fs/utils.h>
 #include <msp/io/console.h>
 #include <msp/io/print.h>
@@ -11,10 +13,15 @@ using namespace Msp;
 
 SetupGen::SetupGen(int argc, char **argv)
 {
+	vector<string> import_dirs;
+
 	GetOpt getopt;
+	getopt.add_option('I', "importdir", import_dirs, GetOpt::REQUIRED_ARG);
 	getopt.add_option('o', "output", out_fn, GetOpt::REQUIRED_ARG);
 	getopt.add_argument("input_file", in_fn, GetOpt::REQUIRED_ARG);
 	getopt(argc, argv);
+
+	import_path.insert(import_path.end(), import_dirs.begin(), import_dirs.end());
 }
 
 int SetupGen::main()
@@ -192,6 +199,7 @@ void SetupGen::Loader::init_actions()
 	add("component", &Loader::struct_def, Struct::COMPONENT);
 	add("entity", &Loader::struct_def, Struct::ENTITY);
 	add("enum", &Loader::enum_def);
+	add("import", &Loader::import);
 	add("namespace", &Loader::name_space);
 }
 
@@ -203,6 +211,20 @@ void SetupGen::Loader::enum_def(const DataFile::Symbol &n)
 	type.set_enum(*mod.enums.emplace_back(make_unique<Enum>(move(en))));
 }
 
+void SetupGen::Loader::import(const string &n)
+{
+	string fn = n+".mgs";
+	FS::Path full_path;
+	if(!ranges::any_of(obj.import_path, [&full_path, &fn](const FS::Path &p){
+			full_path = p/fn;
+			return FS::exists(full_path);
+		}))
+		throw IO::file_not_found(fn);
+
+	obj.load(full_path);
+	obj.headers.insert(n+".h");
+}
+
 void SetupGen::Loader::name_space(const string &ns)
 {
 	mod.name_space = ns;