SLRaytracer.h

Go to the documentation of this file.

/**
 * \file      SLRaytracer.h
 * \date      July 2014
 * \authors   Marcus Hudritsch
 * \copyright http://opensource.org/licenses/GPL-3.0
 * \remarks   Please use clangformat to format the code. See more code style on
 *            https://github.com/cpvrlab/SLProject4/wiki/SLProject-Coding-Style
 */
 
#ifndef SLRAYTRACER_H
#define SLRAYTRACER_H
 
#include <SLEventHandler.h>
#include <SLGLTexture.h>
#include <SLVec4.h>
#include <SLLight.h>
#include <Averaged.h>
 
class SLScene;
class SLSceneView;
class SLRay;
class SLMaterial;
class SLCamera;
 
//-----------------------------------------------------------------------------
//! Ray tracing state
typedef enum
{
    rtReady,    // RT is ready to start
    rtBusy,     // RT is running
    rtFinished, // RT is finished
    rtMoveGL    // RT is finished and GL camera is moving
} SLRTState;
//-----------------------------------------------------------------------------
//! Pixel index struct used in anti aliasing in ray tracing
struct SLRTAAPixel
{
    explicit SLRTAAPixel(SLushort X = 0, SLushort Y = 0)
    {
        x = X;
        y = Y;
    }
    SLushort x; //!< Unsigned short x-pixel index
    SLushort y; //!< Unsigned short x-pixel index
};
typedef vector<SLRTAAPixel> SLVPixel;
//-----------------------------------------------------------------------------
//! SLRaytracer hold all the methods for Whitted style Ray Tracing.
/*!
SLRaytracer implements the methods render, eyeToPixel, trace and shade for
classic Whitted style Ray Tracing. This class is a friend class of SLScene and
can access via the pointer _s all members of SLScene. The scene traversal for
the ray intersection tests is done within the intersection method of all nodes.
The result is written into an image of an SLGLTexture and that then is rendered
with OpenGL in a orthographic projection.
*/
class SLRaytracer : public SLGLTexture
{
public:
    SLRaytracer();
    ~SLRaytracer() override;
 
    // ray tracer functions
    SLbool  renderClassic(SLSceneView* sv);
    SLbool  renderDistrib(SLSceneView* sv);
    void    renderSlices(bool isMainThread, SLuint threadNum);
    void    renderSlicesMS(bool isMainThread, SLuint threadNum);
    SLCol4f trace(SLRay* ray);
    SLCol4f shade(SLRay* ray);
    void    sampleAAPixels(bool isMainThread, SLuint threadNum);
    void    renderUIBeforeUpdate();
 
    // additional ray tracer functions
    void         setPrimaryRay(SLfloat x, SLfloat y, SLRay* primaryRay);
    void         getAAPixels();
    SLCol4f      fogBlend(SLfloat z, SLCol4f color);
    virtual void printStats(SLfloat sec);
    virtual void initStats(SLint depth);
 
    // Setters
    void state(SLRTState state)
    {
        if (_state != rtBusy) _state = state;
    }
    void maxDepth(SLint depth)
    {
        _maxDepth = depth;
        state(rtReady);
    }
    void resolutionFactor(SLfloat rf) { _resolutionFactor = rf; }
    void doDistributed(SLbool distrib) { _doDistributed = distrib; }
    void doContinuous(SLbool cont)
    {
        _doContinuous = cont;
        state(rtReady);
    }
    void doFresnel(SLbool fresnel)
    {
        _doFresnel = fresnel;
        state(rtReady);
    }
    void aaSamples(SLint samples)
    {
        _aaSamples = samples;
        state(rtReady);
    }
    void gamma(SLfloat g)
    {
        _gamma        = g;
        _oneOverGamma = 1.0f / g;
    }
 
    // Getters
    SLRTState     state() const { return _state; }
    SLint         maxDepth() const { return _maxDepth; }
    SLbool        doDistributed() const { return _doDistributed; }
    SLbool        doContinuous() const { return _doContinuous; }
    SLbool        doFresnel() const { return _doFresnel; }
    SLint         aaSamples() const { return _aaSamples; }
    static SLuint numThreads() { return Utils::maxThreads(); }
    SLint         progressPC() const { return _progressPC; }
    SLfloat       aaThreshold() const { return _aaThreshold; }
    SLfloat       renderSec() const { return _renderSec; }
    SLfloat       gamma() const { return _gamma; }
    SLfloat       oneOverGamma() const { return _oneOverGamma; }
    SLfloat       resolutionFactor() const { return _resolutionFactor; }
    SLint         resolutionFactorPC() const { return (SLint)(_resolutionFactor * 100.0f + 0.00001f); }
    SLfloat       raysPerMS() { return _raysPerMS.average(); }
 
    // Render target image
    virtual void prepareImage();
    virtual void renderImage(bool updateTextureGL);
    virtual void saveImage();
 
protected:
    function<void(bool, SLuint)> renderSlicesAsync;
    function<void(bool, SLuint)> sampleAAPixelsAsync;
 
    SLSceneView* _sv;               //!< Parent sceneview
    SLRTState    _state;            //!< RT state;
    SLCamera*    _cam;              //!< shortcut to the camera
    SLfloat      _resolutionFactor; //!< screen to RT image size factor (default 1.0)
    SLint        _maxDepth;         //!< Max. allowed recursion depth
    SLbool       _doContinuous;     //!< if true state goes into ready again
    SLbool       _doDistributed;    //!< Flag for parallel distributed RT
    SLbool       _doFresnel;        //!< Flag for Fresnel reflection
    SLint        _progressPC;       //!< progress in %
    SLfloat      _renderSec;        //!< Rendering time in seconds
    AvgFloat     _raysPerMS;        //!< Averaged rays per ms
 
    SLfloat  _pxSize;       //!< Pixel size
    SLVec3f  _eye;          //!< Camera position
    SLVec3f  _la, _lu, _lr; //!< Camera lookat, lookup, lookright
    SLVec3f  _bl;           //!< Bottom left vector
    SLint    _nextLine;     //!< next line index to render RT in a thread
    SLVPixel _aaPixels;     //!< Vector for antialiasing pixels
    SLfloat  _gamma;        //!< gamma correction value
    SLfloat  _oneOverGamma; //!< one over gamma correction value
 
    // variables for distributed ray tracing
    SLfloat _aaThreshold; //!< threshold for anti aliasing
    SLint   _aaSamples;   //!< SQRT of uneven num. of AA samples
};
//-----------------------------------------------------------------------------
#endif