From 250e30d73f09e9da2b5a81d0fbae63744ae12a73 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Tue, 2 Apr 2019 08:47:15 +0800
Subject: *misc

---
 .../libs/asura-lib-core/graphics/stb_decoder.cpp   | 65 ++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 source/libs/asura-lib-core/graphics/stb_decoder.cpp

(limited to 'source/libs/asura-lib-core/graphics/stb_decoder.cpp')

diff --git a/source/libs/asura-lib-core/graphics/stb_decoder.cpp b/source/libs/asura-lib-core/graphics/stb_decoder.cpp
new file mode 100644
index 0000000..a13d6b8
--- /dev/null
+++ b/source/libs/asura-lib-core/graphics/stb_decoder.cpp
@@ -0,0 +1,65 @@
+#include <asura-lib-utils/exceptions/exception.h>
+
+#include "stb_decoder.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
-- 
cgit v1.1-26-g67d0