src/rcrt/gi/PhotonMap.h

Go to the documentation of this file.

#ifndef PHOTONMAP_H_
#define PHOTONMAP_H_

#include "Photon.h"
#include "PhotonKDTree.h"
#include "../lights/Light.h"
#include "../Scene.h"
#include <vector>
#include "../materials/Material.h"

namespace rcrt
{

class PhotonMap
{
private:
        Scene* scene;
        std::vector<Photon*> photons;
        std::vector<Photon*> caustics;
        PhotonKDTree globalTree;
        PhotonKDTree causticTree;
        float maxRadius;
        int maxBounces;
        float currentRadius;
        float flat;
        
        void generateJob(Light* light, std::vector<Photon*>* photons, std::vector<Photon*>* caustics1,
                        const int& noPhotons, int& storedTotal);
        
public:
        PhotonMap(Scene* s, const float& maxRadius, const int& maxBounces, const float& flatten = 0);
        virtual ~PhotonMap();
        
        void generate(const int& noPhotons);
        
        RGBColor getRadiance(const Vec3D& wOut,
                        const int& noGPhotons, const int& noCPhotons, Intersection& is);
        
        const float& getMaxRadius() const;
};

}

#endif /*PHOTONMAP_H_*/

---