From bad78945ceba425f6a80e3b8dca2414d592970eb Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Fri, 2 Aug 2019 20:51:00 +0800
Subject: =?UTF-8?q?*=E4=BF=AE=E6=94=B9=E6=96=87=E4=BB=B6=E5=90=8D=E6=A0=BC?=
 =?UTF-8?q?=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 source/modules/asura-utils/IO/DataBuffer.cpp | 156 +++++++++++++++++++++++++++
 1 file changed, 156 insertions(+)
 create mode 100644 source/modules/asura-utils/IO/DataBuffer.cpp

(limited to 'source/modules/asura-utils/IO/DataBuffer.cpp')

diff --git a/source/modules/asura-utils/IO/DataBuffer.cpp b/source/modules/asura-utils/IO/DataBuffer.cpp
new file mode 100644
index 0000000..6660a94
--- /dev/null
+++ b/source/modules/asura-utils/IO/DataBuffer.cpp
@@ -0,0 +1,156 @@
+#include <cstdlib>
+#include <cstring>
+#include "DataBuffer.h"
+
+using namespace AEThreading;
+
+namespace AsuraEngine
+{
+	namespace IO
+	{
+
+		DataBuffer::DataBuffer(DataBuffer& src)
+			: m_Size(0)
+			, m_Capacity(0)
+			, m_Bytes(nullptr)
+		{
+			// 初始化容量
+			lock(m_Mutex)
+			{
+			 m_Capacity = src.m_Size;
+			 m_Bytes = new byte[m_Capacity];
+				Clear();
+
+				Load(src);
+			}
+		}
+
+		DataBuffer::DataBuffer(std::size_t capacity)
+			: m_Size(0)
+			, m_Capacity(0)
+			, m_Bytes(nullptr)
+		{
+			lock(m_Mutex)
+			{
+			 m_Capacity = capacity;
+			 m_Bytes = new byte[m_Capacity];
+				Clear();
+			}
+		}
+
+		DataBuffer::DataBuffer(const void* data, std::size_t size)
+			: m_Capacity(0)
+			, m_Size(0)
+			, m_Bytes(nullptr)
+		{
+			lock(m_Mutex)
+			{
+			 m_Capacity = size;
+			 m_Bytes = new byte[m_Capacity];
+				Clear();
+
+				Load(data, size);
+			}
+		}
+
+		DataBuffer::~DataBuffer()
+		{
+			lock(m_Mutex)
+			{
+				delete[] m_Bytes;
+			}
+		}
+
+		void DataBuffer::Refactor(size_t capacity)
+		{
+			lock(m_Mutex)
+			{
+				if (!m_Bytes || m_Capacity != capacity)
+				{
+					if(m_Bytes)
+						delete[] m_Bytes;
+				 m_Capacity = capacity;
+				 m_Bytes = new byte[m_Capacity];
+				 m_Size = 0;
+				}
+				Clear();
+			}
+		}
+
+		void DataBuffer::Load(DataBuffer& db)
+		{
+			lock(m_Mutex)
+			{
+				Load(db.GetData(), db.GetSize());
+			}
+		}
+
+		void DataBuffer::Load(const void* data, std::size_t size)
+		{
+			lock(m_Mutex)
+			{
+				ASSERT(m_Capacity >= size);
+				memcpy(m_Bytes, data, size);
+			 m_Size = size;
+			}
+		}
+
+		void DataBuffer::Move(void* bytes, std::size_t size)
+		{
+			lock(m_Mutex)
+			{
+				if (m_Bytes == bytes)
+				{
+					// 如果是自身，代表更新size值，在读文件时会这样
+				 m_Size = size;
+				}
+				else
+				{
+					if (m_Bytes)
+						delete[] m_Bytes;
+				 m_Bytes = (byte*)bytes;
+				 m_Size = size;
+				 m_Capacity = size;
+				}
+			}
+		}
+
+		byte* DataBuffer::GetData()
+		{
+			return m_Bytes;
+		}
+
+		void DataBuffer::Clear()
+		{
+			lock(m_Mutex)
+			{
+				if (m_Bytes)
+				{
+					memset(m_Bytes, 0, m_Size);
+				 m_Size = 0;
+				}
+			}
+		}
+		
+		std::size_t DataBuffer::GetSize()
+		{
+			return m_Size;
+		}
+
+		std::size_t DataBuffer::GetCapacity()
+		{
+			return m_Capacity;
+		}
+
+		void DataBuffer::Lock()
+		{
+		 m_Mutex.Lock();
+		}
+
+		void DataBuffer::Unlock()
+		{
+		 m_Mutex.Unlock();
+		}
+
+	}
+}
\ No newline at end of file
-- 
cgit v1.1-26-g67d0