Diamond75/Panda3DS
1
0
Fork 0
mirror of https://github.com/wheremyfoodat/Panda3DS.git synced 2025-04-06 06:05:40 +12:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor ShaderJITTest into a specialization of ShaderInterpreterTest

Browse source
This commit is contained in:
Wunkolo 2024-03-19 10:13:04 -07:00
parent 27ad7b01f3
commit dcd64802a3
1 changed files with 37 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

62
tests/shader.cpp
View file

@ -7,6 +7,7 @@
#include <catch2/catch_test_macros.hpp>
#include <initializer_list>
#include <memory>
#include <span>
using namespace Floats;
static const nihstro::SourceRegister input0 = nihstro::SourceRegister::MakeInput(0);
@ -28,48 +29,59 @@ static std::unique_ptr<PICAShader> assembleVertexShader(std::initializer_list<ni
return newShader;
}
class ShaderInterpreterTest final {
private:
class ShaderInterpreterTest {
protected:
std::unique_ptr<PICAShader> shader = {};
virtual void runShader() { shader->run(); }
public:
explicit ShaderInterpreterTest(std::initializer_list<nihstro::InlineAsm> code) : shader(assembleVertexShader(code)) {}
// Multiple inputs, singular scalar output
float runScalar(std::initializer_list<float> inputs) {
usize inputIndex = 0;
for (const float& input : inputs) {
const std::array<Floats::f24, 4> input_vec = std::array<Floats::f24, 4>{f24::fromFloat32(input), f24::zero(), f24::zero(), f24::zero()};
shader->inputs[inputIndex++] = input_vec;
}
shader->run();
return shader->outputs[0][0];
std::span<const std::array<Floats::f24, 4>> runTest(std::span<const std::array<Floats::f24, 4>> inputs) {
std::copy(inputs.begin(), inputs.end(), shader->inputs.begin());
runShader();
return shader->outputs;
}
// Each input is written to the x component of sequential input registers
// The first output vector is returned
const std::array<Floats::f24, 4>& runVector(std::initializer_list<float> inputs) {
std::vector<std::array<Floats::f24, 4>> inputsVec;
for (const float& input : inputs) {
const std::array<Floats::f24, 4> inputVec = {
f24::fromFloat32(input),
f24::zero(),
f24::zero(),
f24::zero(),
};
inputsVec.emplace_back(inputVec);
}
return runTest(inputsVec)[0];
}
// Each input is written to the x component of sequential input registers
// The x component of the first output
float runScalar(std::initializer_list<float> inputs) { return runVector(inputs)[0].toFloat32(); }
[[nodiscard]] std::array<std::array<Floats::f24, 4>, 96>& floatUniforms() const { return shader->floatUniforms; }
[[nodiscard]] std::array<std::array<u8, 4>, 4>& intUniforms() const { return shader->intUniforms; }
[[nodiscard]] u32& boolUniform() const { return shader->boolUniform; }
static std::unique_ptr<ShaderInterpreterTest> assembleTest(std::initializer_list<nihstro::InlineAsm> code) {
return std::make_unique<ShaderInterpreterTest>(code);
}
};
#if defined(PANDA3DS_SHADER_JIT_SUPPORTED)
class ShaderJITTest final {
class ShaderJITTest final : public ShaderInterpreterTest {
private:
std::unique_ptr<PICAShader> shader = {};
ShaderJIT shaderJit = {};
public:
explicit ShaderJITTest(std::initializer_list<nihstro::InlineAsm> code) : shader(assembleVertexShader(code)) { shaderJit.prepare(*shader.get()); }
void runShader() override { shaderJit.run(*shader); }
// Multiple inputs, singular scalar output
float runScalar(std::initializer_list<float> inputs) {
usize inputIndex = 0;
for (const float& input : inputs) {
const std::array<Floats::f24, 4> input_vec = std::array<Floats::f24, 4>{f24::fromFloat32(input), f24::zero(), f24::zero(), f24::zero()};
shader->inputs[inputIndex++] = input_vec;
}
shaderJit.run(*shader.get());
return shader->outputs[0][0];
}
public:
explicit ShaderJITTest(std::initializer_list<nihstro::InlineAsm> code) : ShaderInterpreterTest(code) { shaderJit.prepare(*shader); }
static std::unique_ptr<ShaderJITTest> assembleTest(std::initializer_list<nihstro::InlineAsm> code) {
return std::make_unique<ShaderJITTest>(code);