LoadCompressedTextFile

Author: KaleiAlma

LuaSTG/CMakeLists.txt

@@ -146,6 +146,7 @@ set(LUASTG_ENGINE_SOURCES
     LuaSTG/AppFrameLua.cpp
     LuaSTG/AppFrameRender.cpp
     LuaSTG/AppFrameRenderEx.cpp
+    LuaSTG/AppFrameFileEx.cpp
     LuaSTG/LConfig.h
     LuaSTG/LMathConstant.hpp
     LuaSTG/Main.cpp

LuaSTG/LuaSTG/AppFrame.h

@@ -195,6 +195,7 @@ namespace LuaSTGPlus
         // Read a text file from a resource package.  
         // It is possible to read other files, but you may get meaningless results.
         int LoadTextFile(lua_State* L, const char* path, const char *packname) noexcept;
+        int LoadCompressedTextFile(lua_State* L, const char* path, const char *packname) noexcept;
 
         // TODO: LoadBinaryFile?
 

LuaSTG/LuaSTG/AppFrameFileEx.cpp

@@ -0,0 +1,46 @@
+#include "stb_image.h"
+
+#include "AppFrame.h"
+#include "Core/FileManager.hpp"
+
+using namespace LuaSTGPlus;
+
+int AppFrame::LoadCompressedTextFile(lua_State* L_, const char* path, const char *packname)noexcept
+{
+    if (ResourceMgr::GetResourceLoadingLog()) {
+        if (packname)
+            spdlog::info("[luastg] Reading compressed text file '{}' in package '{}'", packname, path);
+        else
+            spdlog::info("[luastg] Reading compressed text file '{}'", path);
+    }
+    bool loaded = false;
+    std::vector<uint8_t> src, decompressed;
+    if (packname)
+    {
+        auto& arc = GFileManager().getFileArchive(packname);
+        if (!arc.empty())
+        {
+            loaded = arc.load(path, src);
+        }
+    }
+    else
+    {
+        loaded = GFileManager().loadEx(path, src);
+    }
+    if (!loaded) {
+        spdlog::error("[luastg] Unable to load file '{}'", path);
+        return 0;
+    }
+    int decompressed_len;
+    char* decompressed_buf = stbi_zlib_decode_malloc((char*)src.data(), src.size(), &decompressed_len);
+    if (!decompressed_buf) {
+        spdlog::error("[luastg] Unable to allocate buffer for compressed text file '{}'", path);
+        return 0;
+    }
+    if (stbi_zlib_decode_buffer(decompressed_buf, decompressed_len, (char*)src.data(), src.size()) == -1) {
+        spdlog::error("[luastg] Unable to decompress text file '{}'", path);
+        return 0;
+    }
+    lua_pushlstring(L_, decompressed_buf, decompressed_len);
+    return 1;
+}

LuaSTG/LuaSTG/LuaBinding/LW_LuaSTG.cpp

@@ -97,6 +97,11 @@ void LuaSTGPlus::BuiltInFunctionWrapper::Register(lua_State* L)noexcept
 		static int LoadTextFile(lua_State* L)noexcept
 		{
 			return LAPP.LoadTextFile(L, luaL_checkstring(L, 1), luaL_optstring(L, 2, NULL));
+	
+		}
+		static int LoadCompressedTextFile(lua_State* L)noexcept
+		{
+			return LAPP.LoadCompressedTextFile(L, luaL_checkstring(L, 1), luaL_optstring(L, 2, NULL));
 		}
 		#pragma endregion
 
@@ -256,6 +261,7 @@ void LuaSTGPlus::BuiltInFunctionWrapper::Register(lua_State* L)noexcept
 		{ "Log", &WrapperImplement::Log },
 		{ "DoFile", &WrapperImplement::DoFile },
 		{ "LoadTextFile", &WrapperImplement::LoadTextFile },
+		{ "LoadCompressedTextFile", &WrapperImplement::LoadCompressedTextFile },
 		#pragma endregion
 
 		#pragma region 窗口与交换链控制函数