src/StratifiedSampleGenerator.h

Go to the documentation of this file.

#ifndef STRATIFIEDSAMPLEGENERATOR_H
#define STRATIFIEDSAMPLEGENERATOR_H

#include "SampleGenerator.h"

class StratifiedSampleGenerator : public SampleGenerator
{
public:
    void samples(int n, float u[], float v[], float weight[]) const
    {
        int x = static_cast<int>(floorf(sqrtf(n)));
        assert(x*x == n);

        // create a grid of jittered sample coordinates
        float size = 1.0/n;
        for (int i=0; i < x; i++)
        {
            for (int j=0; j < x; j++)
            {
                if (i*x+j >= n)
                    return;
                u[i*x+j] = static_cast<float>(i)/x + frand()*size;
                v[i*x+j] = static_cast<float>(j)/x + frand()*size;
                weight[i*x+j] = 1.0 / float(n);
            }
        }
    }
};

#endif

---