aboutsummaryrefslogtreecommitdiff
path: root/include/cru/common
diff options
context:
space:
mode:
Diffstat (limited to 'include/cru/common')
-rw-r--r--include/cru/common/Buffer.h50
-rw-r--r--include/cru/common/io/AutoReadStream.h88
-rw-r--r--include/cru/common/platform/unix/PosixSpawnSubProcess.h5
3 files changed, 143 insertions, 0 deletions
diff --git a/include/cru/common/Buffer.h b/include/cru/common/Buffer.h
new file mode 100644
index 00000000..5390ebb4
--- /dev/null
+++ b/include/cru/common/Buffer.h
@@ -0,0 +1,50 @@
+#pragma once
+
+#include "Base.h"
+
+namespace cru {
+class Buffer {
+ public:
+ explicit Buffer(Index size);
+
+ Buffer(const Buffer& other);
+ Buffer(Buffer&& other) noexcept;
+
+ Buffer& operator=(const Buffer& other);
+ Buffer& operator=(Buffer&& other) noexcept;
+
+ ~Buffer();
+
+ private:
+ Index GetSize() const;
+ Index GetUsedSize() const;
+ bool IsFull() const { return GetSize() == GetUsedSize(); }
+ bool IsNotFull() const { return !IsFull(); }
+
+ std::byte& GetAt(Index index);
+ std::byte GetAt(Index index) const;
+
+ Index PushEnd(std::byte* other, Index other_size);
+ Index PopEnd(Index size);
+
+ std::byte* Data();
+ const std::byte* Data() const;
+
+ operator std::byte*() { return Data(); }
+ operator const std::byte*() const { return Data(); }
+
+ private:
+ std::byte* ptr_;
+ Index size_;
+ Index used_size_;
+};
+
+void swap(Buffer& left, Buffer& right);
+
+class BufferList {
+ public:
+ explicit BufferList(Index buffer_size);
+ private:
+
+};
+} // namespace cru
diff --git a/include/cru/common/io/AutoReadStream.h b/include/cru/common/io/AutoReadStream.h
new file mode 100644
index 00000000..7857e8b9
--- /dev/null
+++ b/include/cru/common/io/AutoReadStream.h
@@ -0,0 +1,88 @@
+#pragma once
+
+#include "Stream.h"
+#include "../Buffer.h"
+
+#include <condition_variable>
+#include <list>
+#include <mutex>
+#include <thread>
+#include <vector>
+
+namespace cru::io {
+struct AutoReadStreamOptions {
+ /**
+ * @brief The size of a single buffer allocated each time new space is needed.
+ * Use default value if <= 0.
+ *
+ * When current buffer is full and there is no space for following data, a new
+ * buffer will be allocated and appended to the buffer list. Note if sum size
+ * of all buffers reaches the total_buffer_size, no more buffer will be
+ * allocated but wait.
+ */
+ int buffer_block_size = 0;
+
+ /**
+ * @brief Total size limit of saved data in buffer. Use default value if < 0.
+ * No limit if == 0.
+ *
+ * When the buffer is filled, it will block and wait for user to read to get
+ * free space of buffer to continue read.
+ */
+ int total_buffer_size = 0;
+};
+
+/**
+ * @brief A stream that wraps another stream and auto read it into a buffer in a
+ * background thread.
+ */
+class CRU_BASE_API AutoReadStream : public Stream {
+ private:
+ class BufferBlock {
+
+ };
+
+ public:
+ /**
+ * @brief Wrap a stream and auto read it in background.
+ * @param stream The stream to auto read.
+ * @param auto_delete Whether to delete the stream object in destructor.
+ * @param options Options to modify the behavior.
+ */
+ AutoReadStream(
+ Stream* stream, bool auto_delete,
+ const AutoReadStreamOptions& options = AutoReadStreamOptions());
+
+ ~AutoReadStream() override;
+
+ public:
+ bool CanSeek() override;
+ Index Seek(Index offset, SeekOrigin origin = SeekOrigin::Current) override;
+
+ bool CanRead() = 0;
+ virtual Index Read(std::byte* buffer, Index offset, Index size) = 0;
+
+ bool CanWrite() override;
+ Index Write(const std::byte* buffer, Index offset, Index size) override;
+
+ void Flush() override;
+
+ void Close() = 0;
+
+ private:
+ void BackgroundThreadRun();
+
+ private:
+ Stream* stream_;
+ bool auto_delete_;
+
+ int buffer_block_size_;
+ int total_buffer_size_;
+
+ std::mutex buffer_mutex_;
+ std::condition_variable buffer_condition_variable_;
+ std::list<Buffer> buffer_list_;
+
+ std::thread background_thread_;
+};
+} // namespace cru::io
diff --git a/include/cru/common/platform/unix/PosixSpawnSubProcess.h b/include/cru/common/platform/unix/PosixSpawnSubProcess.h
index a55c2c1c..0a0f0ad4 100644
--- a/include/cru/common/platform/unix/PosixSpawnSubProcess.h
+++ b/include/cru/common/platform/unix/PosixSpawnSubProcess.h
@@ -17,6 +17,11 @@ class PosixSpawnSubProcess : public PlatformSubProcessBase {
explicit PosixSpawnSubProcess(const PlatformSubProcessStartInfo& start_info);
~PosixSpawnSubProcess();
+ protected:
+ void PlatformCreateProcess() override;
+ PlatformSubProcessExitResult PlatformWaitForProcess() override;
+ void PlatformKillProcess() override;
+
private:
pid_t pid_;
int exit_code_;