src/rcrt/lights/sampling/AreaLightSampler.h

Go to the documentation of this file.

#ifndef LIGHTSAMPLETREE_H_
#define LIGHTSAMPLETREE_H_

#include "../../math/Point2D.h"
#include "../AreaLight.h"
#include "LightSampler.h"

namespace rcrt
{


enum ShadowType
{
        NONE,
        PENUMBRA,
        FULL,
};

class LSTreeNode;
class LSTreeLeaf;
class AreaLightSampler;

class LSTree
{       
        friend class AreaLightSampler;
        
protected:
        ShadowType type;
        int size;
        LSTreeNode* parent;
                
public:
        
        LSTree(LSTreeNode* p=0):type(PENUMBRA), size(-1),parent(p)
        {}
        
        LSTree(const ShadowType& t, const int& s, LSTreeNode* p=0)
                :type(t),size(s), parent(p)
        {}
        
        virtual ~LSTree()
        {}
        
        virtual const RGBColor& getMean() const =0;
        virtual const RGBColor& getVariance() const =0;
        virtual const float& getWeight() const =0;
        virtual void setWeight(const float& w) =0;
        virtual void getSamples(std::vector<LightSample>& samples) const =0;
        
        inline const int& getSize() const
        {
                return size;
        }
        
        inline const ShadowType& getType() const
        {
                return type;
        }
        
        LSTreeNode* getParent()
        {
                return parent;
        }
        
        virtual bool isLeaf()
        {
                return false;
        }
        
        virtual LSTreeLeaf* refine() =0;
};

class LSTreeLeaf : public LSTree
{       
        friend class AreaLightSampler;
        
protected:
        LightSample sample;
        Point2D uvMin;
        Point2D uvMax;
        Point2D uvs;
        
public:
        LSTreeLeaf(LightSample sa, const Point2D& uvMi, const Point2D& uvMa, const Point2D& uv
                        , LSTreeNode* p);
        
        ~LSTreeLeaf()
        {}
        
        const RGBColor& getMean() const;
        const RGBColor& getVariance() const;
        const float& getWeight() const;
        void setWeight(const float& w);
        virtual void getSamples(std::vector<LightSample>& samples) const;
        
        bool isLeaf();
        
        LSTreeLeaf* refine();
};

class LSTreeNode : public LSTree
{       
        friend class LSTreeLeaf;
        friend class AreaLightSampler;
        
protected:
        LSTree* lc;
        LSTree* rc;
        RGBColor mean;//weighted mean
        RGBColor variance;//unbiased weighted variance
        float weight;
        Point2D uvMin;
        Point2D uvMax;
        
public: 
        LSTreeNode(const Point2D& uvMi, const Point2D& uvMa, LSTreeNode* p=0);
        LSTreeNode(LSTree* l, LSTree* r, LSTreeNode* p, const Point2D& uvMi, const Point2D& uvMa);
        ~LSTreeNode()
        {
                //std::cout << "[LSTreeNode] destructor:" << std::endl;
                //std::cout << "[LSTreeNode]             left " << std::endl;
                if(lc != 0)delete lc;
                //std::cout << "[LSTreeNode]             right " << std::endl;
                if(rc)delete rc;
        }
        
        void setLeft(LSTree* l);
        
        void setRight(LSTree* r);

        void update();
        
        LSTreeLeaf* refine();   
        LSTree* getOtherChild(LSTree* c);       
        const RGBColor& getMean() const;        
        const float& getWeight() const; 
        void setWeight(const float& w);
        virtual void getSamples(std::vector<LightSample>& samples) const;       
        const RGBColor& getVariance() const;

};

class AreaLightSampler : public LightSampler
{
private:
        AreaLight* light;
        LSTreeNode* root;       
        //Point3D lastPos;
        //Vec3D lastNormal;
        
        void subdivide(LSTreeLeaf* leaf, std::vector<LightSample>& samples,
                        const Point3D& p, const Vec3D& no, Scene* scene);
        
public:
        AreaLightSampler(AreaLight* l);
        ~AreaLightSampler()
        {
                if(root != 0)delete root;
        }
        
        void sample(const Point3D& p, const Vec3D& no,
                        std::vector<LightSample>& samples, const int& maxSamples, Scene* scene);
};



} //Namespace

#endif /*LIGHTSAMPLETREE_H_*/

---