mirror of
https://github.com/open-goal/jak-project
synced 2026-05-23 15:02:01 -04:00
Tyler Wilding
6446389263
* extractor: refactor and cleanup for multi-game support * deps: switch to `ghc::filesystem` as it is utf-8 everywhere by default * extractor: finally working with unicode * unicode: fix unicode cli args on windows in all `main` functions
67 lines
2.1 KiB
C++
67 lines
2.1 KiB
C++
#include "TextureDB.h"
|
|
|
|
#include "common/util/Assert.h"
|
|
|
|
#include "third-party/fmt/core.h"
|
|
#include "third-party/stb_image/stb_image.h"
|
|
|
|
namespace decompiler {
|
|
|
|
void TextureDB::add_texture(u32 tpage,
|
|
u32 texid,
|
|
const std::vector<u32>& data,
|
|
u16 w,
|
|
u16 h,
|
|
const std::string& tex_name,
|
|
const std::string& tpage_name,
|
|
const std::vector<std::string>& level_names) {
|
|
auto existing_tpage_name = tpage_names.find(tpage);
|
|
if (existing_tpage_name == tpage_names.end()) {
|
|
tpage_names[tpage] = tpage_name;
|
|
} else {
|
|
ASSERT(existing_tpage_name->second == tpage_name);
|
|
}
|
|
|
|
u32 combo_id = (tpage << 16) | texid;
|
|
auto existing_tex = textures.find(combo_id);
|
|
if (existing_tex != textures.end()) {
|
|
ASSERT(existing_tex->second.name == tex_name);
|
|
ASSERT(existing_tex->second.w == w);
|
|
ASSERT(existing_tex->second.h == h);
|
|
ASSERT(existing_tex->second.rgba_bytes == data);
|
|
ASSERT(existing_tex->second.page == tpage);
|
|
} else {
|
|
auto& new_tex = textures[combo_id];
|
|
new_tex.rgba_bytes = data;
|
|
new_tex.name = tex_name;
|
|
new_tex.w = w;
|
|
new_tex.h = h;
|
|
new_tex.page = tpage;
|
|
}
|
|
for (const auto& level_name : level_names) {
|
|
texture_ids_per_level[level_name].insert(combo_id);
|
|
}
|
|
}
|
|
|
|
void TextureDB::replace_textures(const fs::path& path) {
|
|
fs::path base_path(path);
|
|
for (auto& tex : textures) {
|
|
fs::path full_path = base_path / tpage_names.at(tex.second.page) / (tex.second.name + ".png");
|
|
if (fs::exists(full_path)) {
|
|
fmt::print("Replacing {}\n", full_path.string().c_str());
|
|
int w, h;
|
|
auto data = stbi_load(full_path.string().c_str(), &w, &h, 0, 4); // rgba channels
|
|
if (!data) {
|
|
fmt::print("failed to load PNG file: {}\n", full_path.string().c_str());
|
|
continue;
|
|
}
|
|
tex.second.rgba_bytes.resize(w * h);
|
|
memcpy(tex.second.rgba_bytes.data(), data, w * h * 4);
|
|
tex.second.w = w;
|
|
tex.second.h = h;
|
|
stbi_image_free(data);
|
|
}
|
|
}
|
|
}
|
|
} // namespace decompiler
|