2020-06-06 19:28:30 +02:00
|
|
|
#include "rtweekend.h"
|
|
|
|
|
|
|
|
#include "hittable_list.h"
|
|
|
|
#include "sphere.h"
|
2020-06-01 02:49:50 +02:00
|
|
|
#include "colour.h"
|
2020-06-06 22:11:09 +02:00
|
|
|
#include "camera.h"
|
2020-06-07 00:59:30 +02:00
|
|
|
#include "material.h"
|
2020-06-01 02:49:50 +02:00
|
|
|
|
2020-06-01 00:52:34 +02:00
|
|
|
#include <iostream>
|
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
const double ASPECT_RATIO = 16.0 / 9.0;
|
|
|
|
const int WIDTH = 1920;
|
2020-06-03 02:56:59 +02:00
|
|
|
const int HEIGHT = static_cast<int>(WIDTH / ASPECT_RATIO);
|
2020-06-07 05:13:39 +02:00
|
|
|
const int SAMPLES_PER_PIXEL = 32;
|
2022-06-03 00:09:50 +02:00
|
|
|
const int MAX_DEPTH = 5;
|
|
|
|
|
|
|
|
const colour pink(0.973, 0.784, 0.863);
|
|
|
|
const colour grey(0.133, 0.133, 0.133);
|
2020-06-03 02:56:59 +02:00
|
|
|
|
2020-06-07 00:01:46 +02:00
|
|
|
colour ray_colour(const ray& r, const hittable& world, int depth)
|
2020-06-03 02:56:59 +02:00
|
|
|
{
|
2020-06-06 19:28:30 +02:00
|
|
|
hit_record rec;
|
2020-06-07 00:01:46 +02:00
|
|
|
if (depth <= 0)
|
|
|
|
{
|
2022-06-03 00:09:50 +02:00
|
|
|
return grey;
|
2020-06-07 00:01:46 +02:00
|
|
|
}
|
|
|
|
|
2020-06-07 00:25:36 +02:00
|
|
|
if (world.hit(r, 0.001, infinity, rec))
|
2020-06-06 04:54:05 +02:00
|
|
|
{
|
2020-06-07 00:59:30 +02:00
|
|
|
ray scattered;
|
|
|
|
colour attenuation;
|
|
|
|
|
|
|
|
if (rec.mat_ptr->scatter(r, rec, attenuation, scattered))
|
|
|
|
{
|
|
|
|
return attenuation * ray_colour(scattered, world, depth-1);
|
|
|
|
}
|
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
return grey;
|
2020-06-06 04:54:05 +02:00
|
|
|
}
|
|
|
|
|
2020-06-06 05:09:13 +02:00
|
|
|
vec3 unit_direction = unit_vector(r.direction());
|
2022-06-03 00:09:50 +02:00
|
|
|
auto t = 0.5 * (unit_direction.y() + 1.0) + 0.5;
|
2020-06-04 01:47:33 +02:00
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
return lerp(grey, pink, t);
|
2020-06-03 02:56:59 +02:00
|
|
|
}
|
2020-06-01 00:52:34 +02:00
|
|
|
|
2020-06-07 05:13:39 +02:00
|
|
|
hittable_list random_scene()
|
|
|
|
{
|
|
|
|
hittable_list world;
|
|
|
|
|
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
//auto ground_material = make_shared<lambertian>(pink);
|
|
|
|
//world.add(make_shared<sphere>(point3(0,-1000,0), 1000, ground_material));
|
|
|
|
|
|
|
|
//for (int a = -11; a < 11; a++)
|
|
|
|
//{
|
|
|
|
// for (int b = -11; b < 11; b++)
|
|
|
|
// {
|
|
|
|
// auto choose_mat = random_double();
|
|
|
|
// point3 centre(a + 0.9*random_double(), 0.2, b + 0.9*random_double());
|
|
|
|
|
|
|
|
// if ((centre - point3(4, 0.2, 0)).length() > 0.9)
|
|
|
|
// {
|
|
|
|
// shared_ptr<material> sphere_material;
|
|
|
|
|
|
|
|
// if (choose_mat < 0.8)
|
|
|
|
// {
|
|
|
|
// // diffuse
|
|
|
|
// //auto albedo = colour::random() * colour::random();
|
|
|
|
// sphere_material = make_shared<lambertian>(pink);
|
|
|
|
// world.add(make_shared<sphere>(centre, 0.2, sphere_material));
|
|
|
|
// }
|
|
|
|
// else if (choose_mat < 0.95)
|
|
|
|
// {
|
|
|
|
// // metal
|
|
|
|
// auto fuzz = random_double(0, 0.5);
|
|
|
|
// sphere_material = make_shared<metal>(pink, fuzz);
|
|
|
|
// world.add(make_shared<sphere>(centre, 0.2, sphere_material));
|
|
|
|
// }
|
|
|
|
// else
|
|
|
|
// {
|
|
|
|
// // glass
|
|
|
|
// sphere_material = make_shared<dielectric>(1.5);
|
|
|
|
// world.add(make_shared<sphere>(centre,0.2, sphere_material));
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
//}
|
2020-06-07 05:13:39 +02:00
|
|
|
|
|
|
|
auto material1 = make_shared<dielectric>(1.5);
|
2022-06-03 00:09:50 +02:00
|
|
|
world.add(make_shared<sphere>(point3(0, 0, 0), 3.0, material1));
|
2020-06-07 05:13:39 +02:00
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
//auto material2 = make_shared<lambertian>(pink);
|
|
|
|
//world.add(make_shared<sphere>(point3(-4, 1, 0), 1.0, material2));
|
2020-06-07 05:13:39 +02:00
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
auto material3 = make_shared<metal>(pink, 0.5);
|
|
|
|
int sphere_count = 3;
|
|
|
|
for (int i = 0; i < sphere_count; i++)
|
|
|
|
{
|
|
|
|
float a = 6.28 * (float)i/sphere_count - 100.0;
|
|
|
|
float r = 8.0;
|
|
|
|
float x = r*sin(a);
|
|
|
|
float y = 2.0*cos(a);
|
|
|
|
float z = r*cos(a);
|
|
|
|
point3 pos(x,y,z);
|
|
|
|
world.add(make_shared<sphere>(pos, 4.0, material3));
|
|
|
|
}
|
|
|
|
|
2020-06-07 05:13:39 +02:00
|
|
|
|
|
|
|
return world;
|
|
|
|
}
|
|
|
|
|
2020-06-01 00:52:34 +02:00
|
|
|
int main()
|
|
|
|
{
|
2020-06-06 19:28:30 +02:00
|
|
|
std::cout << "P3\n" << WIDTH << ' ' << HEIGHT << "\n255\n";
|
|
|
|
|
2020-06-07 05:13:39 +02:00
|
|
|
hittable_list world = random_scene();
|
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
auto dist_to_target = 10.0;
|
|
|
|
auto dist_to_focus = dist_to_target + 1.0;
|
|
|
|
auto cam_y = 1.0;
|
|
|
|
point3 lookfrom(0,cam_y,-dist_to_target);
|
2020-06-07 05:13:39 +02:00
|
|
|
point3 lookat(0,0,0);
|
|
|
|
vec3 vup(0,1,0);
|
2022-06-03 00:09:50 +02:00
|
|
|
auto aperture = 0.5;
|
2020-06-07 05:13:39 +02:00
|
|
|
|
2022-06-03 00:09:50 +02:00
|
|
|
camera cam(lookfrom, lookat, vup, 47, ASPECT_RATIO, aperture, dist_to_focus);
|
2020-06-06 22:11:09 +02:00
|
|
|
|
|
|
|
for (int j = HEIGHT - 1; j >= 0; --j)
|
2020-06-01 00:52:34 +02:00
|
|
|
{
|
2020-06-06 22:11:09 +02:00
|
|
|
std::cerr << "\rScanlines remaining: " << j << ' ' << std::flush;
|
|
|
|
for (int i = 0; i < WIDTH; ++i)
|
2020-06-01 00:52:34 +02:00
|
|
|
{
|
2020-06-06 22:11:09 +02:00
|
|
|
colour pixel_colour(0,0,0);
|
|
|
|
|
|
|
|
for (int s = 0; s < SAMPLES_PER_PIXEL; ++s)
|
|
|
|
{
|
|
|
|
auto u = (i + random_double()) / (WIDTH-1);
|
|
|
|
auto v = (j + random_double()) / (HEIGHT-1);
|
|
|
|
ray r = cam.get_ray(u, v);
|
2020-06-07 00:01:46 +02:00
|
|
|
pixel_colour += ray_colour(r, world, MAX_DEPTH);
|
2020-06-06 22:11:09 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
write_colour(std::cout, pixel_colour, SAMPLES_PER_PIXEL);
|
2020-06-01 00:52:34 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
std::cerr << "\nDone." << std::endl;
|
|
|
|
}
|
|
|
|
|