Skip to content

Latest commit

 

History

History
248 lines (206 loc) · 10.8 KB

ProgrammingGuideNRD.md

File metadata and controls

248 lines (206 loc) · 10.8 KB

Streamline - NRD

The focus of this guide is on using Streamline to integrate NVIDIA Real-Time Denoisers (NRD) into an application. For more information about NRD itself, please visit the NVIDIA Developer NRD Page

Version 2.0

1.0 INITIALIZE AND SHUTDOWN

Call slInit as early as possible (before any d3d11/d3d12/vk APIs are invoked)

#include <sl.h>
#include <sl_consts.h>
#include <sl_nrd.h>

sl::Preferences pref;
pref.showConsole = true;                        // for debugging, set to false in production
pref.logLevel = sl::eLogLevelDefault;
pref.pathsToPlugins = {}; // change this if Streamline plugins are not located next to the executable
pref.numPathsToPlugins = 0; // change this if Streamline plugins are not located next to the executable
pref.pathToLogsAndData = {};                    // change this to enable logging to a file
pref.logMessageCallback = myLogMessageCallback; // highly recommended to track warning/error messages in your callback
if(!slInit(pref, myApplicationId))
{
    // Handle error, check the logs
}

For more details please see preferences

Call slShutdown() before destroying dxgi/d3d11/d3d12/vk instances, devices and other components in your engine.

if(!slShutdown())
{
    // Handle error, check the logs
}

2.0 CHECK IF NRD IS SUPPORTED

As soon as SL is initialized, you can check if NRD is available for the specific adapter you want to use:

Microsoft::WRL::ComPtr<IDXGIFactory> factory;
if (SUCCEEDED(CreateDXGIFactory(__uuidof(IDXGIFactory), (void**)&factory)))
{
    Microsoft::WRL::ComPtr<IDXGIAdapter> adapter{};
    uint32_t i = 0;
    while (factory->EnumAdapters(i, &adapter) != DXGI_ERROR_NOT_FOUND)
    {
        DXGI_ADAPTER_DESC desc{};
        if (SUCCEEDED(adapter->GetDesc(&desc)))
        {
            sl::AdapterInfo adapterInfo{};
            adapterInfo.deviceLUID = (uint8_t*)&desc.AdapterLuid;
            adapterInfo.deviceLUIDSizeInBytes = sizeof(LUID);
            if (SL_FAILED(result, slIsFeatureSupported(sl::kFeatureNRD, adapterInfo)))
            {
                // Requested feature is not supported on the system, fallback to the default method
                switch (result)
                {
                    case sl::Result::eErrorOSOutOfDate:         // inform user to update OS
                    case sl::Result::eErrorDriverOutOfDate:     // inform user to update driver
                    case sl::Result::eErrorNoSupportedAdapter:  // cannot use this adapter (older or non-NVDA GPU etc)
                    // and so on ...
                };
            }
            else
            {
                // Feature is supported on this adapter!
            }
        }
        i++;
    }
}

3.0 PREPARE RESOURCES FOR DENOISING

NRD expects inputs in a specific format, so make sure to run a pre-processing pass (simple compute shader) to prepare them:

  • eNormalRoughness (RGB normal, A roughness)
  • eSpecularHitNoisy (RGB color, A reflection ray hitT)
  • eDiffuseHitNoisy (RGB color, A primary ray hitT)
  • eAmbientOcclusionNoisy (R ambient)

4.0 PROVIDE NRD CONSTANTS

NRD constants must be set so that the NRD plugin can track any changes made by the user:

sl::NRDConstants nrdConsts = {};
// Depending on what type of denoising we are doing, set the method mask
if (enableAo)
{
    nrdConsts.methodMask |= (1 << sl::NRDMethods::eNRDMethodReblurDiffuseOcclusion);
}
else if (enableDiffuse && enableSpecular)
{
    nrdConsts.methodMask |= (1 << sl::NRDMethods::eNRDMethodReblurDiffuseSpecular);
}
else if (enableDiffuse)
{
    nrdConsts.methodMask |= (1 << sl::NRDMethods::eNRDMethodReblurDiffuse);
}
else if (enableSpecular)
{
    nrdConsts.methodMask |= (1 << sl::NRDMethods::eNRDMethodReblurSpecular);
}

// Various camera matrices must be provided
nrdConsts.clipToWorld = myMatrix;
nrdConsts.clipToWorldPrev = myMatrix; // clipToWorld from the previous frame
nrdConsts.common = {};
nrdConsts.common.worldToViewMatrix = myworldToViewMatrix;
nrdConsts.common.motionVectorScale[0] = myMVecScaleX; // Normally 1
nrdConsts.common.motionVectorScale[1] = myMVecScaleY; // Normally 1
nrdConsts.common.resolutionScale[0] = myScaleX; // Are we rendering within a larger render target? If so set scale, otherwise set to 1.0f
nrdConsts.common.resolutionScale[1] = myScaleY; // Are we rendering within a larger render target? If so set scale, otherwise set to 1.0f
nrdConsts.common.denoisingRange = myDenoisingRange;
nrdConsts.common.splitScreen = mySplitScreen;
nrdConsts.common.accumulationMode = myAccumMode; // Reset or keep history etc

// Now set the specific settings for the method(s) chosen above (defaults are OK for 99% of the applications)
sl::NRDReblurSettings& reblurSettings = nrdConsts.reblurSettings;
reblurSettings = {};

// Note that we use method mask as a unique id
// This allows us to evaluate different NRD denoisers in the same or different viewports but at the different stages with the same frame
if(!slSetFeatureConstants(sl::kFeatureNRD, &nrdConsts, myFrameIndex, nrdConsts.methodMask))
{
    // Handle error here, check the logs
}

You may need to set additional values in the NRDConstants structure, and the NRD method specific constants structures contained within it, depending on the NRD method in use.

NOTE: To disable NRD, set sl::NRDConstants.methodMask to sl::NRDMode::eNRDMethodOff or simply stop calling slEvaluate

5.0 PROVIDE COMMON CONSTANTS

Various per frame camera related constants are required by all Streamline features and must be provided if any SL feature is active and as early in the frame as possible. Please keep in mind the following:

  • All SL matrices are row-major and should not contain any jitter offsets
  • If motion vector values in your buffer are in {-1,1} range then motion vector scale factor in common constants should be {1,1}
  • If motion vector values in your buffer are NOT in {-1,1} range then motion vector scale factor in common constants must be adjusted so that values end up in {-1,1} range
sl::Constants consts = {};
// Set motion vector scaling based on your setup
consts.mvecScale = {1,1}; // Values in eMotionVectors are in [-1,1] range
consts.mvecScale = {1.0f / renderWidth,1.0f / renderHeight}; // Values in eMotionVectors are in pixel space
consts.mvecScale = myCustomScaling; // Custom scaling to ensure values end up in [-1,1] range
// Set all other constants here
if(!setConstants(consts, myFrameIndex)) // constants are changing per frame so frame index is required
{
    // Handle error, check logs
}

For more details please see common constants

6.0 TAG ALL REQUIRED RESOURCES

NRD requires depth, motion vectors, noisy inputs and outputs depending on which type of denoising is selected:

// These are required for any denoising method

// Prepare resources (assuming d3d11/d3d12 integration so leaving Vulkan view and device memory as null pointers)
sl::Resource depth = {sl::ResourceType::Tex2d, myDepthBuffer, nullptr, nullptr, nullptr};
sl::Resource mvec = {sl::ResourceType::Tex2d, myMotionVectorsBuffer, nullptr, nullptr, nullptr};
// Note that you can also pass unique id (if using multiple viewports) and the extent of the resource if dynamic resolution is active
setTag(&depth, sl::kBufferTypeDepth);
setTag(&mvec, sl::kBufferTypeMotionVectors);

// Depending on what type of denoising is done tag the remaining resources.
// Note that we are passing the method mask to map each resource to the appropriate NRD instance
sl::Resource nroughness = {sl::ResourceType::Tex2d, myNormalRoughness, nullptr, nullptr, nullptr};
sl::Resource specularIn = {sl::ResourceType::Tex2d, mySpecularInput, nullptr, nullptr, nullptr};
sl::Resource specularOut = {sl::ResourceType::Tex2d, mySpecularOutput, nullptr, nullptr, nullptr};
setTag(&nroughness, sl::kBufferTypeNormalRoughness, nrdConsts.methodMask);
setTag(&specularIn, sl::kBufferTypeSpecularHitNoisy, nrdConsts.methodMask);
setTag(&specularOut, sl::kBufferTypeSpecularHitDenoised, nrdConsts.methodMask);
// and so on ...

NOTE: If dynamic resolution is used then please specify the extent for each tagged resource. Please note that SL manages resource states so there is no need to transition tagged resources.

7.0 PROVIDE COMMON CONSTANTS

Various per frame camera related constants are required by all Streamline features and must be provided if any SL feature is active as early in the frame as possible. Please keep in mind the following:

  • All SL matrices are row-major and should not contain any jitter offsets
  • If motion vector values in your buffer are in {-1,1} range then motion vector scale factor in common constants should be {1,1}
  • If motion vector values in your buffer are NOT in {-1,1} range then motion vector scale factor in common constants must be adjusted so that values end up in {-1,1} range
sl::Constants consts = {};
// Set motion vector scaling based on your setup
consts.mvecScale = {1,1}; // Values in eMotionVectors are in [-1,1] range
consts.mvecScale = {1.0f / renderWidth,1.0f / renderHeight}; // Values in eMotionVectors are in pixel space
consts.mvecScale = myCustomScaling; // Custom scaling to ensure values end up in [-1,1] range
sl::Constants consts = {};
// Set all constants here
if(!setConstants(consts, myFrameIndex)) // constants are changing per frame so frame index is required
{
    // Handle error, check logs
}

For more details please see common constants

8.0 ADD NRD TO THE RENDERING PIPELINE

On your rendering thread, call slEvaluate at the appropriate location where denoising should happen. Please note that myFrameIndex and myId used in slEvaluate must match the one used when settings constants.

// Make sure NRD is available and user selected this option in the UI
bool useNRD = slIsFeatureSupported(sl::kFeatureNRD) && userSelectedNRDInUI;
if(useNRD) 
{
    // Inform SL that NRD should be injected at this point.
    // Note that we are passing the specific id which needs to match the id used when setting constants.
    // This can be method mask or something different (like for example viewport id | method mask etc.)
    auto myId = nrdConsts.methodMask;
    if(!slEvaluate(myCmdList, sl::kFeatureNRD, myFrameIndex, myId)) 
    {
        // Handle error
    }
    else
    {
        // IMPORTANT: Host is responsible for restoring state on the command list used
        restoreState(myCmdList);
    }
}
else
{
    // Default denoising method
}

IMPORTANT: Plase note that host is responsible for restoring the command buffer(list) state after calling slEvaluate. For more details on which states are affected please see restore pipeline section