X-Git-Url: http://git.tdb.fi/?p=libs%2Fdatafile.git;a=blobdiff_plain;f=source%2Fbinarywriter.cpp;h=2ff96a457524c7a1e28a13c6e098de535bf0e006;hp=9651609164ce4d5a5679928ba7e4f1b1b2947337;hb=35d0da96bad97214376f18d15078e6d731e6f219;hpb=3b0d6fb8462bf87956099365d02ebb28435509cd

diff --git a/source/binarywriter.cpp b/source/binarywriter.cpp
index 9651609..2ff96a4 100644
--- a/source/binarywriter.cpp
+++ b/source/binarywriter.cpp
@@ -1,3 +1,4 @@
+#include <msp/core/maputils.h>
 #include "binarywriter.h"
 #include "statement.h"
 
@@ -23,11 +24,9 @@ void BinaryWriter::write(const Statement &st)
 
 void BinaryWriter::write_(const Statement &st)
 {
-	Dictionary::iterator i = dict.find(DictEntry(st.keyword, st.get_signature()));
-	if(i==dict.end())
-		throw InvalidParameterValue("Unknown statement");
+	unsigned id = get_item(dict, DictEntry(st.keyword, st.get_signature()));
 
-	write_int(i->second);
+	write_int(id);
 	for(Statement::Arguments::const_iterator j = st.args.begin(); j!=st.args.end(); ++j)
 		switch(j->get_signature())
 		{
@@ -135,10 +134,7 @@ void BinaryWriter::write_float(float f)
 
 void BinaryWriter::write_enum(const string &e)
 {
-	StringMap::const_iterator i = strings.find(e);
-	if(i==strings.end())
-		throw InvalidParameterValue("Unknown enum");
-	write_int(i->second);
+	write_int(get_item(strings, e));
 }
 
 } // namespace DataFile