X-Git-Url: http://git.tdb.fi/?p=libs%2Fdatafile.git;a=blobdiff_plain;f=tool%2Ftool.cpp;h=d58853eef5cbf519228e1fa7a0401811173db6df;hp=d0499cb663970c818f3f2d8dfeced6bdf31d48e0;hb=0bf1684c1ae0f5ecc23b7c64f0a0fec79a8c2082;hpb=8955db30f9cd1c1566b349da29e669f065f84e36

diff --git a/tool/tool.cpp b/tool/tool.cpp
index d0499cb..d58853e 100644
--- a/tool/tool.cpp
+++ b/tool/tool.cpp
@@ -4,8 +4,8 @@
 #include <msp/io/file.h>
 #include <msp/datafile/parser.h>
 #include <msp/datafile/statement.h>
-#include <msp/datafile/writer.h>
 #include "compiler.h"
+#include "packer.h"
 #include "tool.h"
 
 using namespace std;
@@ -16,24 +16,33 @@ DataTool::DataTool(int argc, char **argv):
 	binary(false),
 	compile(false),
 	float_size(0),
-	compress(false)
+	compress(false),
+	pack(false),
+	debug(false)
 {
 	GetOpt getopt;
-	getopt.add_option('b', "binary", binary, GetOpt::NO_ARG);
-	getopt.add_option('c', "compile", compile, GetOpt::NO_ARG);
-	getopt.add_option('f', "float-size", float_size, GetOpt::REQUIRED_ARG);
-	getopt.add_option('o', "output", out_fn, GetOpt::REQUIRED_ARG);
-	getopt.add_option('z', "compress", compress, GetOpt::NO_ARG);
+	getopt.add_option('b', "binary", binary, GetOpt::NO_ARG).set_help("Produce a binary datafile");
+	getopt.add_option('c', "compile", compile, GetOpt::NO_ARG).set_help("Create a collection based on a template file");
+	getopt.add_option('f', "float-size", float_size, GetOpt::REQUIRED_ARG).set_help("Floating-point precision", "BITS");
+	getopt.add_option('g', "debug", debug, GetOpt::NO_ARG).set_help("Display control statements");
+	getopt.add_option('o', "output", out_fn, GetOpt::REQUIRED_ARG).set_help("Output to a file instead of stdout", "FILE");
+	getopt.add_option('p', "pack", pack, GetOpt::NO_ARG).set_help("Create a pack from multiple files");
+	getopt.add_option('z', "compress", compress, GetOpt::NO_ARG).set_help("Produce a compressed datafile");
+	getopt.add_argument("infile", in_fns, GetOpt::OPTIONAL_ARG).set_help("Files to process");
 	getopt(argc, argv);
 
-	in_fns = getopt.get_args();
 	if(in_fns.empty())
 		in_fns.push_back("-");
+
+	if(pack && out_fn=="-")
+		throw usage_error("Can't write pack to stdout");
 }
 
 int DataTool::main()
 {
-	if(compile)
+	if(pack)
+		do_pack();
+	else if(compile)
 		do_compile();
 	else
 		do_transfer();
@@ -44,35 +53,34 @@ int DataTool::main()
 void DataTool::do_transfer()
 {
 	IO::Base *out = open_output(out_fn);
-	DataFile::Writer writer(*out);
-	set_writer_parameters(writer);
+	DataFile::Writer *writer = create_writer(*out);
 
-	for(vector<string>::const_iterator i=in_fns.begin(); i!=in_fns.end(); ++i)
+	for(list<string>::const_iterator i=in_fns.begin(); i!=in_fns.end(); ++i)
 	{
 		IO::Base *in = open_input(*i);
 		DataFile::Parser parser(*in, *i);
 
 		while(parser)
 		{
-			DataFile::Statement st = parser.parse();
-			if(st.valid)
-				writer.write(st);
+			DataFile::Statement st = parser.parse(true);
+			if(st.valid && (!st.control || st.keyword=="__src" || debug))
+				writer->write(st);
 		}
 
 		delete in;
 	}
 
+	delete writer;
 	delete out;
 }
 
 void DataTool::do_compile()
 {
 	IO::Base *out = open_output(out_fn);
-	DataFile::Writer writer(*out);
-	set_writer_parameters(writer);
+	DataFile::Writer *writer = create_writer(*out);
 
-	Compiler compiler(writer);
-	for(vector<string>::const_iterator i=in_fns.begin(); i!=in_fns.end(); ++i)
+	Compiler compiler(*writer);
+	for(list<string>::const_iterator i=in_fns.begin(); i!=in_fns.end(); ++i)
 	{
 		IO::Base *in = open_input(*i);
 		DataFile::Parser parser(*in, *i);
@@ -80,9 +88,18 @@ void DataTool::do_compile()
 		delete in;
 	}
 
+	delete writer;
 	delete out;
 }
 
+void DataTool::do_pack()
+{
+	Packer packer(*this);
+	for(list<string>::const_iterator i=in_fns.begin(); i!=in_fns.end(); ++i)
+		packer.pack_file(*i);
+	packer.create_pack(out_fn);
+}
+
 IO::Base *DataTool::open_output(const string &fn)
 {
 	if(fn=="-")
@@ -99,12 +116,14 @@ IO::Base *DataTool::open_input(const string &fn)
 		return new IO::BufferedFile(fn, IO::M_READ);
 }
 
-void DataTool::set_writer_parameters(DataFile::Writer &writer)
+DataFile::Writer *DataTool::create_writer(IO::Base &out)
 {
+	DataFile::Writer *writer = new DataFile::Writer(out);
 	if(compress)
-		writer.set_compressed();
+		writer->set_compressed();
 	if(binary)
-		writer.set_binary(true);
+		writer->set_binary(true);
 	if(float_size)
-		writer.set_float_precision(float_size);
+		writer->set_float_precision(float_size);
+	return writer;
 }