From 68d41d3b214856f8d321b4dbdffc0cd4c7333c36 Mon Sep 17 00:00:00 2001
From: Krzysztof Dermont <erihel@gmail.com>
Date: Mon, 23 Jun 2014 23:19:55 +0200
Subject: More work on streams

---
 src/common/resources/outputstreambuffer.cpp | 85 +++++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 src/common/resources/outputstreambuffer.cpp

(limited to 'src/common/resources/outputstreambuffer.cpp')

diff --git a/src/common/resources/outputstreambuffer.cpp b/src/common/resources/outputstreambuffer.cpp
new file mode 100644
index 0000000..e611d13
--- /dev/null
+++ b/src/common/resources/outputstreambuffer.cpp
@@ -0,0 +1,85 @@
+// * This file is part of the COLOBOT source code
+// * Copyright (C) 2014 Polish Portal of Colobot (PPC)
+// *
+// * This program is free software: you can redistribute it and/or modify
+// * it under the terms of the GNU General Public License as published by
+// * the Free Software Foundation, either version 3 of the License, or
+// * (at your option) any later version.
+// *
+// * This program is distributed in the hope that it will be useful,
+// * but WITHOUT ANY WARRANTY; without even the implied warranty of
+// * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// * GNU General Public License for more details.
+// *
+// * You should have received a copy of the GNU General Public License
+// * along with this program. If not, see  http://www.gnu.org/licenses/.
+
+#include "common/resources/outputstreambuffer.h"
+
+#include <stdexcept>
+#include <sstream>
+
+COutputStreamBuffer::COutputStreamBuffer(size_t buffer_size) : m_buffer_size(buffer_size)
+{
+    m_file = nullptr;
+    m_buffer = new char[buffer_size];
+    setp(m_buffer, m_buffer + buffer_size);
+}
+
+
+COutputStreamBuffer::~COutputStreamBuffer()
+{
+    close();
+    delete m_buffer;
+}
+
+
+void COutputStreamBuffer::open(const std::string &filename)
+{
+    if (PHYSFS_isInit())
+        m_file = PHYSFS_openWrite(filename.c_str());
+}
+
+
+void COutputStreamBuffer::close()
+{
+    if (is_open())
+        PHYSFS_close(m_file);
+}
+
+
+bool COutputStreamBuffer::is_open()
+{
+    return m_file;
+}
+
+
+std::streambuf::int_type COutputStreamBuffer::overflow(std::streambuf::int_type ch)
+{
+    /* This function should be called when pptr() == epptr(). We use it also in sync()
+       so we also have to write data if buffer is not full. */
+    
+    if (pbase() == pptr()) // no data to write, sync() called with empty buffer
+        return 0;
+
+    // save buffer
+    PHYSFS_sint64 bytes_written = PHYSFS_write(m_file, pbase(), 1, pptr() - pbase());
+    if (bytes_written <= 0)
+        return traits_type::eof();
+
+    pbump(-bytes_written);    
+    // write final char
+    if (ch != traits_type::eof()) {
+        bytes_written = PHYSFS_write(m_file, &ch, 1, 1);
+        if (bytes_written <= 0)
+            return traits_type::eof();
+    }
+
+    return ch;
+}
+
+
+int COutputStreamBuffer::sync()
+{
+    return overflow(traits_type::eof());
+}
-- 
cgit v1.2.3-1-g7c22