ProtoStream added

git-svn-id: https://www4.informatik.uni-erlangen.de/i4svn/danceos/trunk/devel/fail@1240 8c4709b5-6ec9-48aa-a5cd-a96041d1645a

core/util/ProtoStream.cc

@@ -0,0 +1,77 @@
+#include "ProtoStream.hpp"
+
+using namespace std;
+
+ProtoStream::ProtoStream(ostream* outfile){
+	
+	m_write_mode = true;
+	m_outfile = outfile;
+	m_log.setDescription("ProtoStream");
+	m_log.showTime(false);
+}
+
+ProtoStream::ProtoStream(istream* infile){
+	
+	m_write_mode = false;
+	m_infile = infile;
+	m_log.setDescription("ProtoStream");
+	m_log.showTime(false);
+}
+
+bool ProtoStream::writeMessage(google::protobuf::Message* m){
+	
+	if(!m_write_mode){
+		m_log << "To attach a message, please call the constructor of the ProtoStream-Object with a Pointer of an ostream." << endl;
+		return false;
+	}
+	
+	m_size = htonl(m->ByteSize());
+	m_outfile->write(reinterpret_cast<char*>(&m_size), sizeof(m_size));
+	
+	if(m_outfile->bad()){
+		m_log << "Could not write to file!" << endl;
+	}
+	
+	m->SerializeToOstream(m_outfile);
+	
+    return true;
+}
+
+
+bool ProtoStream:: reset(){
+
+	if(m_write_mode){
+		m_log << "To reset, please call the constructor of the ProtoStream-Object with a Pointer of an istream." << endl;
+		return false;
+	}
+	
+	m_infile->seekg(0,ios::beg);
+	return true;
+}
+
+bool ProtoStream::getNext(google::protobuf::Message* m){
+	
+	if(m_write_mode){
+		m_log << "To get a message, please call the constructor of the ProtoStream-Object with a Pointer of an istream." << endl;
+		return false;
+	}
+	
+	m_infile->read(reinterpret_cast<char*>(&m_size), sizeof(m_size));
+	
+	if(m_infile->bad()){
+		m_log << "Could not write to file!" << endl;
+	}
+	
+	m_size = ntohl(m_size);
+	
+	char *buf = new char[m_size];
+	
+	m_infile->read(buf, m_size);
+	
+	std::string st(buf, m_size);
+	
+	m->ParseFromString(st);
+	
+	
+	return true;
+}