demo7and8.rs

// Copyright 2022 Lukasz Janyst <lukasz@jany.st>
// Licensed under the MIT license, see the LICENSE file for details.

use ray_tracer::sphere;
use ray_tracer::{color, point, scaling, vector, view_transform};
use ray_tracer::{point_light, rotation_x, rotation_y, translation};
use ray_tracer::{Camera, Material, World};

use std::f64::consts::PI;
use std::io::{self, Write};
use std::time::Instant;

pub fn demo7and8() {
    print!("Rendering demo7... ");
    io::stdout().flush().unwrap();
    let now = Instant::now();

    let mut world = World::empty();

    let mut floor_m = Material::new();
    floor_m.color = color(1.0, 0.9, 0.9);
    floor_m.specular = 0.0;

    let mut floor = sphere(scaling(10.0, 0.01, 10.0));
    floor.set_material(&floor_m);
    world.shapes.push(floor);

    let mut left_wall = sphere(
        translation(0.0, 0.0, 5.0)
            * rotation_y(-PI / 4.0)
            * rotation_x(PI / 2.0)
            * scaling(10.0, 0.01, 10.0),
    );
    left_wall.set_material(&floor_m);
    world.shapes.push(left_wall);

    let mut right_wall = sphere(
        translation(0.0, 0.0, 5.0)
            * rotation_y(PI / 4.0)
            * rotation_x(PI / 2.0)
            * scaling(10.0, 0.01, 10.0),
    );
    right_wall.set_material(&floor_m);
    world.shapes.push(right_wall);

    let mut s1 = sphere(translation(-0.5, 1.0, 0.5));
    let mut s1m = Material::new();
    s1m.color = color(0.1, 1.0, 0.5);
    s1m.specular = 0.3;
    s1m.diffuse = 0.7;
    s1.set_material(&s1m);
    world.shapes.push(s1);

    let mut s2 = sphere(translation(1.5, 0.5, -0.5) * scaling(0.5, 0.5, 0.5));
    let mut s2m = Material::new();
    s2m.color = color(0.5, 1.0, 0.1);
    s2m.specular = 0.3;
    s2m.diffuse = 0.7;
    s2.set_material(&s2m);
    world.shapes.push(s2);

    let mut s3 = sphere(translation(-1.5, 0.33, -0.75) * scaling(0.33, 0.33, 0.33));
    let mut s3m = Material::new();
    s3m.color = color(1.0, 0.8, 0.1);
    s3m.specular = 0.3;
    s3m.diffuse = 0.7;
    s3.set_material(&s3m);
    world.shapes.push(s3);

    let l = point_light(point(-10.0, 10.0, -10.0), color(1.0, 1.0, 1.0));
    world.lights.push(l);

    let mut camera = Camera::new(1280, 620, PI / 3.0);
    camera.set_transform(view_transform(
        point(0.0, 1.5, -5.0),
        point(0.0, 1.0, 0.0),
        vector(0.0, 1.0, 0.0),
    ));

    world.shadows = false;
    let canvas = camera.render(&world);

    let elapsed = now.elapsed();
    let fname = "demo7.ppm";
    canvas.save(fname).unwrap();
    println!("done. Elapsed {:.2?}. Saved {}.", elapsed, fname);

    // Shadows
    print!("Rendering demo8... ");
    io::stdout().flush().unwrap();
    let now = Instant::now();

    world.shadows = true;
    let canvas = camera.render(&world);

    let elapsed = now.elapsed();
    let fname = "demo8.ppm";
    canvas.save(fname).unwrap();
    println!("done. Elapsed {:.2?}. Saved {}.", elapsed, fname);
}