40 lines
1.1 KiB
C++
40 lines
1.1 KiB
C++
|
|
#include "Editor.hpp"
|
|
|
|
#include "OpenImageDenoise/oidn.hpp"
|
|
#include <iostream>
|
|
|
|
using namespace tp;
|
|
|
|
void Editor::denoise() {
|
|
auto& out = mPathTracerBuffers;
|
|
|
|
// Initialize OIDN device
|
|
oidn::DeviceRef device = oidn::newDevice(oidn::DeviceType::CPU);
|
|
device.commit();
|
|
|
|
// Define buffer size
|
|
const int channels = 4; // ARGB
|
|
const auto size = out.color.size();
|
|
// Create the denoising filter
|
|
oidn::FilterRef filter = device.newFilter("RT"); // Use the 'RT' filter for path tracing
|
|
|
|
// Set the input and output buffer (same buffer for in-place denoising)
|
|
filter.setImage("color", out.color.getBuff(), oidn::Format::Float3, size.x, size.y, 0, sizeof(float) * channels); // ARGB
|
|
filter.setImage("output", out.color.getBuff(), oidn::Format::Float3, size.x, size.y, 0, sizeof(float) * channels); // ARGB
|
|
|
|
// Set additional parameters if needed
|
|
filter.set("hdr", false); // Assuming the input image is not HDR
|
|
|
|
// Commit the filter
|
|
filter.commit();
|
|
|
|
// Execute the filter
|
|
filter.execute();
|
|
|
|
// Check for errors
|
|
const char* errorMessage;
|
|
if (device.getError(errorMessage) != oidn::Error::None) {
|
|
std::cerr << "Error: " << errorMessage << std::endl;
|
|
}
|
|
}
|