psfpt.h

/*
 * Fermat
 *
 * Copyright (c) 2016-2019, NVIDIA CORPORATION. All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *    * Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *    * Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *    * Neither the name of the NVIDIA CORPORATION nor the
 *      names of its contributors may be used to endorse or promote products
 *      derived from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#pragma once

#include <pathtracer.h>
#include <types.h>
#include <buffers.h>
#include <ray.h>
#include <tiled_sequence.h>
#include <cugar/sampling/lfsr.h>
#include <cugar/linalg/bbox.h>
#include <renderer_interface.h>
#include <hashmap.h>

struct RenderingContext;
struct MeshVTLStorage;
struct ClusteredRLStorage;
struct AdaptiveClusteredRLStorage;






struct PSFPTOptions : PTOptions
{
    uint32  psf_depth;
    float   psf_width;
    float   psf_min_dist;
    float   psf_max_prob;
    uint32  psf_temporal_reuse;
    float   firefly_filter;

    PSFPTOptions() :
        psf_depth(1),
        psf_width(3.0f),
        psf_min_dist(0.1f),
        psf_max_prob(32.0f),
        psf_temporal_reuse(64),
        firefly_filter(100.0f) {}

    void parse(const int argc, char** argv)
    {
        PTOptions::parse(argc, argv);

        for (int i = 0; i < argc; ++i)
        {
            if (strcmp(argv[i], "-filter-depth") == 0)
                psf_depth = atoi(argv[++i]);
            else if (strcmp(argv[i], "-filter-width") == 0)
                psf_width = (float)atof(argv[++i]);
            else if (strcmp(argv[i], "-filter-min-dist") == 0)
                psf_min_dist = (float)atof(argv[++i]);
            else if (strcmp(argv[i], "-filter-max-prob") == 0)
                psf_max_prob = (float)atof(argv[++i]);
            else if (strcmp(argv[i], "-temporal-reuse") == 0)
                psf_temporal_reuse = atoi(argv[++i]);
            else if (strcmp(argv[i], "-firefly-filter") == 0 ||
                     strcmp(argv[i], "-ff") == 0)
                firefly_filter = (float)atof(argv[++i]);
        }
    }
};

struct PSFPT : RendererInterface
{
    typedef AdaptiveClusteredRLStorage  VTLRLStorage;

    enum PassType {
        kPresamplePass  = 0,
        kFinalPass      = 1
    };

    PSFPT();

    void init(int argc, char** argv, RenderingContext& renderer);

    void render_pass(const uint32 instance, RenderingContext& renderer, const PassType pass_type);

    void render(const uint32 instance, RenderingContext& renderer);

    void keyboard(unsigned char character, int x, int y, bool& invalidate);

    void destroy() { delete this;  }

    static RendererInterface* factory() { return new PSFPT(); }

    // general path tracing members
    DomainBuffer<CUDA_BUFFER, uint8>    m_memory_pool;

    PSFPTOptions                m_options;
    TiledSequence               m_sequence;

    cugar::LFSRGeneratorMatrix  m_generator;
    cugar::LFSRRandomStream     m_random;

    MeshVTLStorage*             m_mesh_vtls;
    VTLRLStorage*               m_vtls_rl;
    cugar::Bbox3f               m_bbox;

    float                       m_time;
    PTStats                     m_stats;

    // PSFPT specific members
    DeviceHashTable                     m_psf_hash;
    DomainBuffer<CUDA_BUFFER, float4>   m_psf_values;
    //DomainBuffer<CUDA_BUFFER, uint32> m_psf_timestamps;
};