这个算法是均匀采样算法,继承于Sampler类。
类声明:
#pragma once
#ifndef __REGULAR_HEADER__
#define __REGULAR_HEADER__
#include "sampler.h"
class Regular :public Sampler {
public:
Regular();
~Regular();
Regular(const integer samps);
Regular(const integer samps, const integer sets);
Regular(const Regular& reg);
Regular& operator=(const Regular& reg);
virtual Sampler* clone() const;
virtual void generate_samples();
};
#endif
类实现:
#include "pch.h"
#include "regular.h"
Regular::Regular() :Sampler() {
generate_samples();
}
Regular::~Regular() {}
Regular::Regular(const integer samps) : Sampler(samps) {
generate_samples();
}
Regular::Regular(const integer samps, const integer sets) : Sampler(samps, sets) {
generate_samples();
}
Regular::Regular(const Regular& reg) : Sampler(reg) {
generate_samples();
}
Regular& Regular::operator=(const Regular& reg) {
if (this == ®)
return *this;
Sampler::operator=(reg);
return *this;
}
Sampler* Regular::clone() const {
return new Regular(*this);
}
void Regular::generate_samples() {
integer n = (integer)std::sqrt((ldouble)nsamples);
for (integer p = 0; p < nsets; p++) {
for (integer i=0;i<n;i++)
for (integer j = 0; j < n; j++) {
Point2 sp((j + 0.5) / n, (i + 0.5) / n); //均匀采样
samples.push_back(sp);
}
}
}
测试采样算法:
添加Sampler类和采样数到ViewPlane类中,增加函数(若无特殊声明,不再重复修改的部分)
void set_samples(const integer n);
void set_sampler(Sampler* const sp);
ViewPlane类具体声明如下:
#pragma once
#ifndef __VIEWPLANE_HEADER__
#define __VIEWPLANE_HEADER__
#include "../../Types.h"
class Sampler;
class ViewPlane {
public:
ViewPlane();
ViewPlane(const ViewPlane& vp);
void set_hres(const integer hr);
void set_vres(const integer vr);
void set_pixelsize(const ldouble ps);
void set_gamma(const ldouble ga);
void set_samples(const integer n);//新增
void set_sampler(Sampler* const sp);//新增
integer hres;
integer vres;
integer nsamples;//新增
ldouble s;
ldouble g;
Sampler* sampler;//新增
};
#endif
需要修改的类成员:
#include "pch.h"
#include "viewplane.h"
#include "../samplers/jittered.h"
ViewPlane::ViewPlane() :hres(200), vres(100), s(0.02), g(1), nsamples(16), sampler(nullptr) {}
ViewPlane::ViewPlane(const ViewPlane& vp)
: hres(vp.hres), vres(vp.vres), s(vp.s), g(vp.g), nsamples(vp.nsamples), sampler(vp.sampler) {}
...
void ViewPlane::set_samples(const integer n) {//和书上不同,剔除特殊性,增加兼容性
nsamples = n;
sampler->clear();
sampler->set_num_samples(nsamples);
sampler->setup_shuffled_indices();
sampler->generate_samples();
}
void ViewPlane::set_sampler(Sampler* const sp) {
if (sampler != nullptr) {
delete sampler;
sampler = nullptr;
}
nsamples = sp->get_num_samples();
sampler = sp;
}
World类只修改build和render部分:
void World::build() {
vp.set_hres(200);
vp.set_vres(100);
vp.set_sampler(new Regular());//所有采样修改都在这里测试
tracer_ptr = new MultiSphere(this);
Geometrics* obj = new Sphere(0, 0.5);
obj->set_color(RGBColor(1, 0, 0));
add_object(obj);
obj = new Sphere(Point3(0, -100.5, 0), 100);
obj->set_color(RGBColor(0, 0, 1));
add_object(obj);
}
render部分若无特殊声明,之后不再重复修改部分。
void World::render() {
Ray ray;
ldouble x, y;
open_window(vp.hres, vp.vres);
Point3 sp;
ray.o = Point3(0, 0, 1);
for (integer r = vp.vres - 1; r >= 0; r--)//render from left-corner to right-corner
for (integer c = 0; c < vp.hres; c++) {
RGBColor color;
for (integer p = 0; p < vp.nsamples; p++) {//增加样本
sp = vp.sampler->sample_unit_square();
x = vp.s * (c - 0.5 * vp.hres + sp.x);
y = vp.s * (r - 0.5 * vp.vres + sp.y);
ray.d = Point3(x, y, -1);
color += tracer_ptr->trace_ray(ray);
}
color /= vp.nsamples;
display_pixel(r, c, color);
}
}
修改完后,运行程序,测试效果如下(放大后,发现边角圆润了不少,接下来我们测试别的采样算法):
