71 lines
2.3 KiB
C++
71 lines
2.3 KiB
C++
#include <iostream>
|
|
|
|
#include "color.h"
|
|
#include "vec3.h"
|
|
#include "ray.h"
|
|
#include "util.h"
|
|
|
|
#include "hittable_list.h"
|
|
#include "sphere.h"
|
|
#include "camera.h"
|
|
|
|
Color ray_color(const Ray& r, const Hittable& world, int depth) {
|
|
hit_record rec;
|
|
|
|
if (depth <= 0)
|
|
return Color(0, 0, 0);
|
|
|
|
if (world.hit(r, 0.001, infinity, rec)) {
|
|
//Point3 target = rec.p + rec.normal + random_in_unit_sphere();
|
|
//Point3 target = rec.p + rec.normal + random_unit_vector();
|
|
Point3 target = rec.p + random_in_hemisphere(rec.normal);
|
|
return 0.5 * ray_color(Ray(rec.p, target - rec.p), world, depth-1);
|
|
}
|
|
Vec3 unit_direction = unit_vector(r.direction());
|
|
auto t = 0.5 * (unit_direction.y() + 1.0);
|
|
return (1.0 - t) * Color(1.0, 1.0, 1.0) + t * Color(0.5, 0.7, 1.0);
|
|
}
|
|
|
|
int main() {
|
|
const auto aspect_ratio = 16.0 / 9.0;
|
|
//const int image_width = 1280;
|
|
const int image_width = 768;
|
|
//const int image_width = 384;
|
|
const int image_height = static_cast<int>(image_width / aspect_ratio);
|
|
const int samples_per_pixel = 400;
|
|
const int max_depth = 50;
|
|
|
|
std::cout << "P3\n" << image_width << " " << image_height << "\n255\n";
|
|
|
|
Hittable_list world;
|
|
world.add(std::make_shared<Sphere>(Point3(0, 0, -1), 0.5));
|
|
world.add(std::make_shared<Sphere>(Point3(0, -100.5, -1), 100));
|
|
|
|
Camera cam;
|
|
|
|
Color* image = new Color[image_height*image_width];
|
|
|
|
for (int j = image_height - 1; j >= 0; --j) {
|
|
std::cerr << "\rScanlines remaining: " << j << " " << std::flush;
|
|
#pragma omp parallel for
|
|
for (int i = 0; i < image_width; ++i) {
|
|
Color pixel_color(0, 0, 0);
|
|
for (int s = 0; s<samples_per_pixel; ++s) {
|
|
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);
|
|
Ray r = cam.get_ray(u, v);
|
|
pixel_color += ray_color(r, world, max_depth);
|
|
}
|
|
image[i*image_height+j] = pixel_color;
|
|
}
|
|
}
|
|
std::cerr << "\nWriting file.\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);
|
|
}
|
|
}
|
|
delete[] image;
|
|
std::cerr << "Done.\n";
|
|
}
|