X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fshoppinglist%2Fmain.cpp;h=449ef8be4d1f6042fec6a4be625d97ca651193a9;hb=e9653dffd0026fb3c02c91a4c0feca688a86c721;hp=a87e2f8cab7a817db5d78ed36b6ae0bade96fe00;hpb=6c61179fe09af2f5366d50f10aadbf5f83438087;p=r2c2.git

diff --git a/source/shoppinglist/main.cpp b/source/shoppinglist/main.cpp
index a87e2f8..449ef8b 100644
--- a/source/shoppinglist/main.cpp
+++ b/source/shoppinglist/main.cpp
@@ -1,9 +1,15 @@
-#include <fstream>
-#include <iostream>
+/* $Id$
+
+This file is part of the MSP Märklin suite
+Copyright © 2006-2008 Mikkosoft Productions, Mikko Rasa
+Distributed under the GPL
+*/
+
 #include <map>
 #include <msp/core/getopt.h>
-#include <msp/parser/loader.h>
-#include <msp/parser/parser.h>
+#include <msp/datafile/loader.h>
+#include <msp/datafile/parser.h>
+#include <msp/io/print.h>
 
 using namespace std;
 using namespace Msp;
@@ -12,9 +18,9 @@ class ShoppingList
 {
 public:
 	ShoppingList(int, char **);
-	void print(ostream &);
+	void print(IO::Base &);
 private:
-	class InventoryLoader: public Parser::Loader
+	class InventoryLoader: public DataFile::Loader
 	{
 	public:
 		InventoryLoader(ShoppingList &);
@@ -24,7 +30,7 @@ private:
 		void track(unsigned, unsigned);
 	};
 
-	class LayoutLoader: public Parser::Loader
+	class LayoutLoader: public DataFile::Loader
 	{
 	public:
 		LayoutLoader(ShoppingList &);
@@ -34,10 +40,8 @@ private:
 		void track(unsigned);
 	};
 
-	typedef map<unsigned, unsigned> TrackMap;
-
-	TrackMap inventory;
-	TrackMap layout;
+	map<unsigned, unsigned> inventory;
+	map<unsigned, unsigned> layout;
 
 	void load_inventory(const string &);
 	void load_layout(const string &);
@@ -46,7 +50,7 @@ private:
 int main(int argc, char **argv)
 {
 	ShoppingList sl(argc, argv);
-	sl.print(cout);
+	sl.print(IO::cout);
 	return 0;
 }
 
@@ -63,54 +67,54 @@ ShoppingList::ShoppingList(int argc, char **argv)
 
 void ShoppingList::load_inventory(const string &fn)
 {
-	ifstream in(fn.c_str());
-	Parser::Parser parser(in, fn);
+	IO::File in(fn);
+	DataFile::Parser parser(in, fn);
 	InventoryLoader il(*this);
 	il.load(parser);
 }
 
 void ShoppingList::load_layout(const string &fn)
 {
-	ifstream in(fn.c_str());
-	Parser::Parser parser(in, fn);
+	IO::File in(fn);
+	DataFile::Parser parser(in, fn);
 	LayoutLoader ll(*this);
 	ll.load(parser);
 }
 
-void ShoppingList::print(ostream &out)
+void ShoppingList::print(IO::Base &out)
 {
-	out<<"// Need to get:\n";
-	for(TrackMap::iterator i=layout.begin(); i!=layout.end(); ++i)
+	IO::print(out, "// Need to get:\n");
+	for(map<unsigned, unsigned>::iterator i=layout.begin(); i!=layout.end(); ++i)
 	{
-		TrackMap::iterator j=inventory.find(i->first);
+		map<unsigned, unsigned>::iterator j=inventory.find(i->first);
 		if(j!=inventory.end())
 		{
 			if(j->second<i->second)
-				out<<"track "<<i->first<<' '<<i->second-j->second<<";\n";
+				IO::print(out, "track %d %d;\n", i->first, i->second-j->second);
 		}
 		else
-			out<<"track "<<i->first<<' '<<i->second<<";\n";
+			IO::print(out, "track %d %d;\n", i->first, i->second);
 	}
 
-	out<<"// Pre-existing:\n";
-	for(TrackMap::iterator i=layout.begin(); i!=layout.end(); ++i)
+	IO::print(out, "// Pre-existing:\n");
+	for(map<unsigned, unsigned>::iterator i=layout.begin(); i!=layout.end(); ++i)
 	{
-		TrackMap::iterator j=inventory.find(i->first);
+		map<unsigned, unsigned>::iterator j=inventory.find(i->first);
 		if(j!=inventory.end())
-			out<<"track "<<i->first<<' '<<min(i->second,j->second)<<";\n";
+			IO::print(out, "track %d %d;\n", i->first, min(i->second, j->second));
 	}
 
-	out<<"// Unused:\n";
-	for(TrackMap::iterator i=inventory.begin(); i!=inventory.end(); ++i)
+	IO::print(out, "// Unused:\n");
+	for(map<unsigned, unsigned>::iterator i=inventory.begin(); i!=inventory.end(); ++i)
 	{
-		TrackMap::iterator j=layout.find(i->first);
+		map<unsigned, unsigned>::iterator j=layout.find(i->first);
 		if(j!=layout.end())
 		{
 			if(j->second<i->second)
-				out<<"track "<<i->first<<' '<<i->second-j->second<<";\n";
+				IO::print(out, "track %d %d;\n", i->first, i->second-j->second);
 		}
 		else
-			out<<"track "<<i->first<<' '<<i->second<<";\n";
+			IO::print(out, "track %d %d;\n", i->first, i->second);
 	}
 }
 
@@ -130,6 +134,7 @@ ShoppingList::LayoutLoader::LayoutLoader(ShoppingList &s):
 {
 	add("track", &LayoutLoader::track);
 	add("base");
+	add("route");
 }
 
 void ShoppingList::LayoutLoader::track(unsigned part)