toytracer/wtracer.cpp

124 lines
4.3 KiB
C++
Raw Normal View History

2020-06-02 22:06:53 +00:00
#include <iostream>
2020-06-06 14:31:51 +00:00
#include <memory>
2020-06-02 22:06:53 +00:00
#include "color.h"
#include "vec3.h"
2020-06-03 20:39:40 +00:00
#include "ray.h"
2020-06-03 22:15:32 +00:00
#include "util.h"
2020-06-03 20:39:40 +00:00
2020-06-03 22:15:32 +00:00
#include "hittable_list.h"
#include "sphere.h"
2020-06-03 22:39:14 +00:00
#include "camera.h"
2020-06-03 22:15:32 +00:00
2020-06-06 12:47:39 +00:00
Color ray_color(const Ray& r, const Hittable& world, int depth) {
2020-06-03 22:15:32 +00:00
hit_record rec;
2020-06-06 12:47:39 +00:00
if (depth <= 0)
return Color(0, 0, 0);
2020-06-06 14:16:31 +00:00
if (world.hit(r, 0.001, infinity, rec)) {
Ray scattered;
Color attenuation;
if (rec.mat_ptr->scatter(r, rec, attenuation, scattered))
return attenuation * ray_color(scattered, world, depth - 1);
return Color(0, 0, 0);
2020-06-03 21:25:47 +00:00
}
2020-06-03 20:39:40 +00:00
Vec3 unit_direction = unit_vector(r.direction());
2020-06-03 22:15:32 +00:00
auto t = 0.5 * (unit_direction.y() + 1.0);
2020-06-03 20:39:40 +00:00
return (1.0 - t) * Color(1.0, 1.0, 1.0) + t * Color(0.5, 0.7, 1.0);
}
2020-06-02 22:06:53 +00:00
2020-06-06 23:02:11 +00:00
Hittable_list setup_random_scene() {
Hittable_list world;
const int sph_i = 3;
auto ground_material = std::make_shared<Lambertian>(Color(0.5, 0.5, 0.5));
world.add(std::make_shared<Sphere>(Point3(0, -1000, 0), 1000, ground_material));
for (int a = -sph_i; a<sph_i; a++) {
for (int b = -sph_i; b<sph_i; b++) {
auto choose_mat = random_double();
Point3 center(a + 0.9 * (11/sph_i) * random_double(), 0.2, b + 0.9 * (11/sph_i) * random_double());
if ((center - Point3(4, 0.2, 0)).length() > 0.9) {
std::shared_ptr<Material> sphere_material;
if (choose_mat < 0.8) {
// diffuse
auto albedo = Color::random() * Color::random();
sphere_material = std::make_shared<Lambertian>(albedo);
world.add(std::make_shared<Sphere>(center, 0.2, sphere_material));
} else if (choose_mat < 0.95) {
// metal
auto albedo = Color::random(0.5, 1);
auto fuzz = random_double(0, 0.5);
sphere_material = std::make_shared<Metal>(albedo, fuzz);
world.add(std::make_shared<Sphere>(center, 0.2, sphere_material));
} else {
// glass
sphere_material = std::make_shared<Dielectric>(1.45);
world.add(std::make_shared<Sphere>(center, 0.2, sphere_material));
}
}
}
}
auto material1 = std::make_shared<Dielectric>(1.45);
world.add(std::make_shared<Sphere>(Point3(0, 1, 0), 1.0, material1));
auto material2 = std::make_shared<Lambertian>(Color(0.4, 0.2, 0.1));
world.add(std::make_shared<Sphere>(Point3(-4, 1, 0), 1.0, material2));
auto material3 = std::make_shared<Metal>(Color(0.7, 0.6, 0.5), 0.0);
world.add(std::make_shared<Sphere>(Point3(4, 1, 0), 1.0, material3));
return world;
}
2020-06-02 22:06:53 +00:00
int main() {
2020-06-03 20:39:40 +00:00
const auto aspect_ratio = 16.0 / 9.0;
2020-06-06 12:47:39 +00:00
//const int image_width = 1280;
const int image_width = 768;
//const int image_width = 384;
2020-06-03 20:39:40 +00:00
const int image_height = static_cast<int>(image_width / aspect_ratio);
//const int samples_per_pixel = 1000;
2020-06-06 14:16:31 +00:00
const int samples_per_pixel = 400;
2020-06-06 12:47:39 +00:00
const int max_depth = 50;
2020-06-02 22:06:53 +00:00
2020-06-06 23:02:11 +00:00
auto world = setup_random_scene();
2020-06-06 23:02:11 +00:00
Point3 lookfrom(13, 2, 3);
Point3 lookat(0, 0, 0);
2020-06-06 22:32:31 +00:00
Vec3 vup(0, 1, 0);
auto dist_to_focus = (lookfrom - lookat).length();
2020-06-06 23:02:11 +00:00
auto aperture = 0.1;
2020-06-06 22:32:31 +00:00
Camera cam(lookfrom, lookat, vup, 20, aspect_ratio, aperture, dist_to_focus);
2020-06-03 22:39:14 +00:00
2020-06-06 14:31:51 +00:00
auto image = std::make_unique<Color[]>(image_height*image_width);
2020-06-02 22:06:53 +00:00
for (int j = image_height - 1; j >= 0; --j) {
std::cerr << "\rScanlines remaining: " << j << " " << std::flush;
#pragma omp parallel for
2020-06-02 22:06:53 +00:00
for (int i = 0; i < image_width; ++i) {
2020-06-03 22:39:14 +00:00
Color pixel_color(0, 0, 0);
for (int s = 0; s<samples_per_pixel; ++s) {
2020-06-06 12:47:39 +00:00
auto u = double(i + random_double(-0.5, 0.5)) / (image_width - 1);
auto v = double(j + random_double(-0.5, 0.5)) / (image_height - 1);
2020-06-03 22:39:14 +00:00
Ray r = cam.get_ray(u, v);
2020-06-06 12:47:39 +00:00
pixel_color += ray_color(r, world, max_depth);
2020-06-03 22:39:14 +00:00
}
image[i*image_height+j] = pixel_color;
}
}
std::cerr << "\nWriting file.\n";
2020-06-06 23:02:11 +00:00
std::cout << "P3\n" << image_width << " " << image_height << "\n255\n";
for (int j = image_height - 1; j >= 0; --j) {
for (int i = 0; i < image_width; ++i) {
write_color(std::cout, image[i*image_height+j], samples_per_pixel);
2020-06-02 22:06:53 +00:00
}
}
std::cerr << "Done.\n";
2020-06-02 22:06:53 +00:00
}