mirror of
https://github.com/PancakeTAS/lsfg-vk.git
synced 2025-10-30 07:01:10 +00:00
114 lines
3.4 KiB
C++
114 lines
3.4 KiB
C++
#include "extract/extract.hpp"
|
|
|
|
#include <cstdlib>
|
|
#include <pe-parse/parse.h>
|
|
|
|
#include <algorithm>
|
|
#include <cstdint>
|
|
#include <stdexcept>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
using namespace Extract;
|
|
|
|
const std::unordered_map<std::string, uint32_t> nameHashTable = {{
|
|
{ "mipmaps", 0xe365474d },
|
|
{ "alpha[0]", 0xf37c8aa8 },
|
|
{ "alpha[1]", 0xeb7a52d4 },
|
|
{ "alpha[2]", 0x8901788e },
|
|
{ "alpha[3]", 0xa06a5e36 },
|
|
{ "beta[0]", 0x63c16b89 },
|
|
{ "beta[1]", 0xe3967ed5 },
|
|
{ "beta[2]", 0x570085ee },
|
|
{ "beta[3]", 0x4f4530db },
|
|
{ "beta[4]", 0x39727389 },
|
|
{ "gamma[0]", 0x94c4edf6 },
|
|
{ "gamma[1]", 0xf4e32702 },
|
|
{ "gamma[2]", 0xa3dc56fc },
|
|
{ "gamma[3]", 0x8b5ed8f6 },
|
|
{ "gamma[4]", 0x1cbf3c4d },
|
|
{ "delta[0]", 0x94c4edf6 },
|
|
{ "delta[1]", 0xacfd805b },
|
|
{ "delta[2]", 0x891dc48b },
|
|
{ "delta[3]", 0x98536d9d },
|
|
{ "delta[4]", 0x8e3f2155 },
|
|
{ "delta[5]", 0x8f0e70a1 },
|
|
{ "delta[6]", 0xd5eca8f1 },
|
|
{ "delta[7]", 0xa9e54e37 },
|
|
{ "delta[8]", 0x1dee8b84 },
|
|
{ "delta[9]", 0x1576c3f5 },
|
|
{ "generate", 0x5c040bd5 }
|
|
}};
|
|
|
|
namespace {
|
|
std::unordered_map<uint32_t, std::vector<uint8_t>> shaderData;
|
|
|
|
uint32_t fnv1a_hash(const std::vector<uint8_t>& data) {
|
|
uint32_t hash = 0x811C9DC5;
|
|
for (auto byte : data) {
|
|
hash ^= byte;
|
|
hash *= 0x01000193;
|
|
}
|
|
return hash;
|
|
}
|
|
|
|
int on_resource(void* data, const peparse::resource& res) { // NOLINT
|
|
if (res.type != peparse::RT_RCDATA || res.buf == nullptr || res.buf->bufLen <= 0)
|
|
return 0;
|
|
std::vector<uint8_t> resource_data(res.buf->bufLen);
|
|
std::copy_n(res.buf->buf, res.buf->bufLen, resource_data.data());
|
|
|
|
const uint32_t hash = fnv1a_hash(resource_data);
|
|
shaderData[hash] = resource_data;
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
void Extract::extractShaders() {
|
|
if (shaderData.size() > 0)
|
|
return;
|
|
|
|
// find path to dll (absolutely beautiful code)
|
|
char* dllPath = getenv("LSFG_DLL_PATH");
|
|
std::string dllPathStr;
|
|
if (dllPath && *dllPath != '\0') {
|
|
dllPathStr = std::string(dllPath);
|
|
} else {
|
|
const char* dataDir = getenv("XDG_DATA_HOME");
|
|
if (dataDir && *dataDir != '\0') {
|
|
dllPathStr = std::string(dataDir) +
|
|
"/Steam/steamapps/common/Lossless Scaling/Lossless.dll";
|
|
} else {
|
|
const char* homeDir = getenv("HOME");
|
|
if (homeDir && *homeDir != '\0') {
|
|
dllPathStr = std::string(homeDir) +
|
|
"/.local/share/Steam/steamapps/common/Lossless Scaling/Lossless.dll";
|
|
} else {
|
|
dllPathStr = "Lossless.dll";
|
|
}
|
|
}
|
|
}
|
|
|
|
// parse the dll
|
|
peparse::parsed_pe* dll = peparse::ParsePEFromFile(dllPathStr.c_str());
|
|
if (!dll)
|
|
throw std::runtime_error("Unable to read Lossless.dll, is it installed?");
|
|
peparse::IterRsrc(dll, on_resource, nullptr);
|
|
peparse::DestructParsedPE(dll);
|
|
}
|
|
|
|
std::vector<uint8_t> Extract::getShader(const std::string& name) {
|
|
if (shaderData.empty())
|
|
throw std::runtime_error("Shaders are not loaded.");
|
|
|
|
auto hit = nameHashTable.find(name);
|
|
if (hit == nameHashTable.end())
|
|
throw std::runtime_error("Shader not found: " + name);
|
|
|
|
auto sit = shaderData.find(hit->second);
|
|
if (sit == shaderData.end())
|
|
throw std::runtime_error("Shader not found: " + name);
|
|
|
|
return sit->second;
|
|
}
|