1 files changed, 65 insertions, 0 deletions

diff --git a/Source/Asura.Engine/graphics/stb_decoder.cpp b/Source/Asura.Engine/graphics/stb_decoder.cpp
new file mode 100644
index 0000000..b14d0f3
--- /dev/null
+++ b/Source/Asura.Engine/graphics/stb_decoder.cpp
@@ -0,0 +1,65 @@
+#include "STBDecoder.h"
+
+#include "Exceptions/Exception.h"
+#include "stb/stb_image.h"
+
+namespace AsuraEngine
+{
+	namespace Graphics
+	{
+
+		bool STBDecoder::CanDecode(const Filesystem::DataBuffer& buffer)
+		{
+			int w = 0;
+			int h = 0;
+			int comp = 0;
+
+			int status = stbi_info_from_memory((const stbi_uc*)buffer.data, buffer.size, &w, &h, &comp);
+
+			return status == 1 && w > 0 && h > 0;
+		}
+
+		void STBDecoder::Decode(const Filesystem::DataBuffer& db, ImageData& imageData)
+		{
+			const stbi_uc *buffer = (const stbi_uc *)db.data;
+			int bufferlen = db.size;
+			int width, height;
+			int comp = 0;
+			byte* data = nullptr;
+			PixelFormat format = PIXELFORMAT_UNKNOWN;
+			std::size_t size = 0;
+
+			if (stbi_is_hdr_from_memory(buffer, bufferlen))
+			{
+				// 4��channel����float
+				data = (byte*)stbi_loadf_from_memory(buffer, bufferlen, &width, &height, &comp, STBI_rgb_alpha);
+				format = PIXELFORMAT_RGBA32F;
+				size = width * height * 4 * sizeof(float);
+			}
+			else
+			{
+				data = (byte*)stbi_load_from_memory(buffer, bufferlen, &width, &height, &comp, STBI_rgb_alpha);
+				format = PIXELFORMAT_ASTC_8x5;
+				size = width * height * 4;
+			}
+			if (data)
+			{
+				// ��֤�ڴ汻�ͷţ���һ�������������ϲ������û���ͷŵ��������Ϊimage data��һ���Եģ�image���Ƕ�εġ�
+				if (imageData.pixels)
+					delete[] imageData.pixels;
+				imageData.pixels = (byte*)data;
+				imageData.format = format;
+				imageData.width = width;
+				imageData.height = height;
+			}
+			else
+			{
+				const char *err = stbi_failure_reason();
+				if (err == nullptr)
+					err = "unknown error";
+				throw Exception("Could not decode image with stb_image (%s).", err);
+			}
+		}
+
+	}
+}
\ No newline at end of file