a

1 files changed, 149 insertions, 0 deletions

diff --git a/loader/loader-impl.cpp b/loader/loader-impl.cpp
new file mode 100644
index 00000000..def67c1b
--- /dev/null
+++ b/loader/loader-impl.cpp
@@ -0,0 +1,149 @@
+#include "src/loader.hpp"
+#include "src/tile-atlas.hpp"
+#include "compat/assert.hpp"
+#include "compat/alloca.hpp"
+#include <filesystem>
+#include <unordered_map>
+#include <utility>
+#include <optional>
+#include <Corrade/Containers/Optional.h>
+#include <Corrade/Containers/Pair.h>
+#include <Corrade/Containers/StringStlView.h>
+#include <Corrade/PluginManager/PluginManager.h>
+#include <Corrade/Utility/Resource.h>
+#include <Corrade/Utility/Path.h>
+#include <Magnum/ImageView.h>
+#include <Magnum/Trade/AbstractImporter.h>
+#include <Magnum/Trade/ImageData.h>
+#include <Magnum/Trade/AbstractImageConverter.h>
+
+#ifdef __GNUG__
+#pragma GCC diagnostic ignored "-Walloca"
+#endif
+
+namespace floormat {
+
+struct loader_impl final : loader_
+{
+    std::optional<Utility::Resource> shader_res;
+    PluginManager::Manager<Trade::AbstractImporter> importer_plugins;
+    Containers::Pointer<Trade::AbstractImporter> tga_importer =
+        importer_plugins.loadAndInstantiate("AnyImageImporter");
+
+    PluginManager::Manager<Trade::AbstractImageConverter> image_converter_plugins;
+    Containers::Pointer<Trade::AbstractImageConverter> tga_converter =
+        image_converter_plugins.loadAndInstantiate("AnyImageConverter");
+
+    std::unordered_map<std::string, std::shared_ptr<struct tile_atlas>> atlas_map;
+
+    std::string shader(Containers::StringView filename) override;
+    Trade::ImageData2D tile_texture(Containers::StringView filename) override;
+    std::shared_ptr<struct tile_atlas> tile_atlas(Containers::StringView filename, Vector2ub size) override;
+
+    static void set_application_working_directory();
+
+    explicit loader_impl();
+    ~loader_impl() override;
+};
+
+std::string loader_impl::shader(Containers::StringView filename)
+{
+    if (!shader_res)
+        shader_res = std::make_optional<Utility::Resource>("floormat/shaders");
+    auto ret = shader_res->getString(filename);
+    if (ret.isEmpty())
+        fm_abort("can't find shader resource '%s'", filename.cbegin());
+    return ret;
+}
+
+std::shared_ptr<tile_atlas> loader_impl::tile_atlas(Containers::StringView name, Vector2ub size)
+{
+    auto it = std::find_if(atlas_map.begin(), atlas_map.end(), [&](const auto& x) {
+        const auto& [k, v] = x;
+        return Containers::StringView{k} == name;
+    });
+    if (it != atlas_map.end())
+        return it->second;
+    auto image = tile_texture(name);
+    auto atlas = std::make_shared<struct tile_atlas>(name, image, size);
+    atlas_map[name] = atlas;
+    return atlas;
+}
+
+Trade::ImageData2D loader_impl::tile_texture(Containers::StringView filename_)
+{
+    static_assert(IMAGE_PATH[sizeof(IMAGE_PATH)-2] == '/');
+    fm_assert(filename_.size() < 4096);
+    fm_assert(filename_.find('\\') == filename_.end());
+    fm_assert(tga_importer);
+    constexpr std::size_t max_extension_length = 16;
+
+    char* const filename = (char*)alloca(filename_.size() + std::size(IMAGE_PATH) + max_extension_length);
+    const std::size_t len = fm_begin(
+        std::size_t off = std::size(IMAGE_PATH)-1;
+        std::memcpy(filename, IMAGE_PATH, off);
+        std::memcpy(filename + off, filename_.cbegin(), filename_.size());
+        return off + filename_.size();
+    );
+
+    for (const auto& extension : std::initializer_list<Containers::StringView>{ ".tga", ".png", ".webp", })
+    {
+        std::memcpy(filename + len, extension.data(), extension.size());
+        filename[len + extension.size()] = '\0';
+        if (tga_importer->openFile(filename))
+        {
+            auto img = tga_importer->image2D(0);
+            if (!img)
+                fm_abort("can't allocate tile image for '%s'", filename);
+            auto ret = std::move(*img);
+            return ret;
+        }
+        Debug{} << "failed to open" << filename << extension;
+    }
+    const auto path = Utility::Path::currentDirectory();
+    fm_log("fatal: can't open tile image '%s' (cwd '%s')", filename, path ? path->data() : "(null)");
+    std::abort();
+}
+
+void loader_::destroy()
+{
+    loader.~loader_();
+    new (&loader) loader_impl();
+}
+
+void loader_impl::set_application_working_directory()
+{
+    static bool once = false;
+    if (once)
+        return;
+    once = true;
+    const auto location = Utility::Path::executableLocation();
+    if (!location)
+        return;
+    std::filesystem::path path((std::string)*location);
+    path.replace_filename("..");
+    std::error_code error;
+    std::filesystem::current_path(path, error);
+    if (error.value()) {
+        fm_warn("failed to change working directory to '%s' (%s)",
+             path.string().data(), error.message().data());
+    }
+}
+
+loader_impl::loader_impl()
+{
+    set_application_working_directory();
+}
+
+loader_impl::~loader_impl() = default;
+
+static loader_& make_default_loader()
+{
+    static loader_impl loader_singleton{};
+    return loader_singleton;
+}
+
+// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
+loader_& loader = make_default_loader();
+
+} // namespace floormat