230 lines
7.3 KiB
Rust
230 lines
7.3 KiB
Rust
#![allow(dead_code)]
|
|
#![allow(unused_imports)]
|
|
#![allow(unused_variables)]
|
|
//Use linear algebra module
|
|
|
|
//Cameras
|
|
|
|
use crate::{camera::Camera, gui::Gui, light::Light, primitive::*, ray::Ray, scene::Scene};
|
|
use log::error;
|
|
|
|
use error_iter::ErrorIter as _;
|
|
use nalgebra::{Point3, Vector3};
|
|
use pixels::{Error, Pixels, SurfaceTexture};
|
|
use std::env;
|
|
use std::rc::Rc;
|
|
use std::sync::Arc;
|
|
use winit::dpi::{LogicalSize, PhysicalSize};
|
|
use winit::event::{Event, KeyboardInput, MouseButton, VirtualKeyCode, WindowEvent};
|
|
use winit::event_loop::{ControlFlow, EventLoop};
|
|
use winit::window::WindowBuilder;
|
|
use winit_input_helper::WinitInputHelper;
|
|
|
|
mod camera;
|
|
mod gui;
|
|
mod light;
|
|
mod primitive;
|
|
mod ray;
|
|
mod raytracer;
|
|
mod scene;
|
|
|
|
const START_WIDTH: i32 = 600;
|
|
const START_HEIGHT: i32 = 555;
|
|
const BOX_SIZE: i16 = 64;
|
|
|
|
const EPSILON: f32 = 1e-6;
|
|
const INFINITY: f32 = f32::MAX;
|
|
const EPSILON_VECTOR: Vector3<f32> = Vector3::new(EPSILON, EPSILON, EPSILON);
|
|
const INFINITY_VECTOR: Vector3<f32> = Vector3::new(INFINITY, INFINITY, INFINITY);
|
|
|
|
struct State {
|
|
scene: Scene,
|
|
camera: Camera,
|
|
rays: Vec<Ray>,
|
|
|
|
index: usize,
|
|
|
|
width: i32,
|
|
height: i32,
|
|
gui: Gui,
|
|
pixels: Pixels,
|
|
}
|
|
|
|
impl State {
|
|
/// Create a new `World` instance that can draw a moving box.
|
|
fn new(
|
|
width: i32,
|
|
height: i32,
|
|
scene: Scene,
|
|
camera: Camera,
|
|
pixels: Pixels,
|
|
gui: Gui,
|
|
) -> Self {
|
|
let rays = camera.cast_rays(width, height);
|
|
Self {
|
|
width,
|
|
height,
|
|
index: 0,
|
|
rays,
|
|
scene,
|
|
camera,
|
|
pixels,
|
|
gui,
|
|
}
|
|
}
|
|
|
|
/// Update the `World` internal state; bounce the box around the screen.
|
|
fn update(&mut self) {
|
|
let gui = &self.gui;
|
|
if gui.reset {
|
|
self.index = 0;
|
|
}
|
|
}
|
|
|
|
/// Resize the world
|
|
fn resize(&mut self, size: &PhysicalSize<u32>) -> bool {
|
|
if let Err(err) = self.pixels.resize_surface(size.width, size.height) {
|
|
log_error("pixels.resize_surface", err);
|
|
return false;
|
|
}
|
|
self.width = size.width as i32;
|
|
self.height = size.height as i32;
|
|
true
|
|
}
|
|
|
|
fn keyboard_input(&mut self, key: &KeyboardInput) {
|
|
match key.virtual_keycode {
|
|
Some(key) => match key {
|
|
VirtualKeyCode::A => {}
|
|
_ => {}
|
|
},
|
|
None => {}
|
|
}
|
|
}
|
|
fn mouse_input(&mut self, button: &MouseButton) {}
|
|
|
|
/// Draw the `World` state to the frame buffer.
|
|
///
|
|
/// Assumes the default texture format: `wgpu::TextureFormat::Rgba8UnormSrgb`
|
|
fn draw(&mut self) {
|
|
let frame = self.pixels.frame_mut();
|
|
for i in 0..self.gui.num_rays {
|
|
let ray = &self.rays[self.index];
|
|
let colour = raytracer::shade_ray(&self.scene, &ray);
|
|
let rgba = match colour {
|
|
Some(colour) => [colour.x, colour.y, colour.z, 255],
|
|
None => [122, 122, 122, 100],
|
|
};
|
|
frame[self.index * 4..(self.index + 1) * 4].copy_from_slice(&rgba);
|
|
self.index = self.index + 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn main() -> Result<(), Error> {
|
|
env_logger::init();
|
|
//Window
|
|
let event_loop = EventLoop::new();
|
|
let window = {
|
|
let size = LogicalSize::new(START_WIDTH as f64, START_HEIGHT as f64);
|
|
WindowBuilder::new()
|
|
.with_title("Hello Pixels + Dear ImGui")
|
|
.with_inner_size(size)
|
|
.with_min_inner_size(size)
|
|
.build(&event_loop)
|
|
.unwrap()
|
|
};
|
|
//Pixel surface
|
|
let pixels = {
|
|
let window_size = window.inner_size();
|
|
let surface_texture = SurfaceTexture::new(window_size.width, window_size.height, &window);
|
|
Pixels::new(START_WIDTH as u32, START_HEIGHT as u32, surface_texture)?
|
|
};
|
|
//Camera
|
|
let eye = Point3::new(10.0, 10.0, 10.0);
|
|
let target = Point3::new(0.0, 0.0, 0.0);
|
|
let up = Vector3::new(0.0, 1.0, 0.0);
|
|
let camera = Camera::new(
|
|
eye,
|
|
target,
|
|
up,
|
|
120.0,
|
|
(START_WIDTH as f32 / START_HEIGHT as f32) as f32,
|
|
);
|
|
let cameras: Vec<Arc<Camera>> = Vec::new();
|
|
// SETUP PRIMITIVES
|
|
let mut primitives: Vec<Arc<dyn Primitive>> = Vec::new();
|
|
let magenta = Arc::new(Material::magenta());
|
|
let blue = Arc::new(Material::blue());
|
|
let turquoise = Arc::new(Material::turquoise());
|
|
// let sphere = Arc::new(Sphere::unit(magenta.clone()));
|
|
// primitives.push(sphere.clone());
|
|
// let cone = Arc::new(Cone::new(0.25, 1.0, -0.5, turquoise.clone()));
|
|
// primitives.push(cone.clone());
|
|
let cube = Arc::new(Box::unit(blue.clone()));
|
|
primitives.push(cube.clone());
|
|
//Lights
|
|
let mut lights: Vec<Arc<Light>> = Vec::new();
|
|
let light_pos = Point3::new(10.0, 12.0, 10.0);
|
|
let light_colour = Vector3::new(1.0, 0.0, 1.0);
|
|
let light_falloff = [1.0, 0.00, 0.00];
|
|
let light = Arc::new(Light::new(light_colour, light_pos, light_falloff));
|
|
lights.push(light.clone());
|
|
let ambient_light = Arc::new(Vector3::new(0.0, 0.0, 0.2));
|
|
//State
|
|
// Set up Dear ImGui
|
|
let gui = Gui::new(&window, &pixels);
|
|
let scene = Scene::new(primitives, lights, cameras, ambient_light);
|
|
let pixels = {
|
|
let window_size = window.inner_size();
|
|
let surface_texture = SurfaceTexture::new(window_size.width, window_size.height, &window);
|
|
Pixels::new(START_WIDTH as u32, START_HEIGHT as u32, surface_texture)?
|
|
};
|
|
let mut state = State::new(START_WIDTH, START_HEIGHT, scene, camera, pixels, gui);
|
|
|
|
event_loop.run(move |event, _, control_flow| {
|
|
// Draw the current frame
|
|
state.gui.handle_event(&window, &event); //Let gui handle its events
|
|
match event {
|
|
Event::WindowEvent { event, .. } => match event {
|
|
WindowEvent::CloseRequested => {
|
|
*control_flow = ControlFlow::Exit;
|
|
}
|
|
WindowEvent::Resized(size) => {
|
|
state.resize(&size);
|
|
}
|
|
WindowEvent::KeyboardInput { input, .. } => {
|
|
state.keyboard_input(&input);
|
|
}
|
|
WindowEvent::MouseInput { button, .. } => {
|
|
state.mouse_input(&button);
|
|
}
|
|
_ => {}
|
|
},
|
|
Event::RedrawRequested(_) => {
|
|
state.draw(); //Draw to pixels
|
|
state.gui.prepare(&window).expect("gui.prepare() failed"); //Prepare imgui
|
|
let render_result = state.pixels.render_with(|encoder, render_target, context| {
|
|
context.scaling_renderer.render(encoder, render_target); // Render pixels
|
|
state.gui.render(&window, encoder, render_target, context)?;
|
|
Ok(())
|
|
});
|
|
if let Err(err) = render_result {
|
|
log_error("pixels.render", err);
|
|
*control_flow = ControlFlow::Exit;
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
state.update(); //Update state
|
|
window.request_redraw(); //Redraw window
|
|
});
|
|
}
|
|
|
|
fn log_error<E: std::error::Error + 'static>(method_name: &str, err: E) {
|
|
error!("{method_name}() failed: {err}");
|
|
for source in err.sources().skip(1) {
|
|
error!(" Caused by: {source}");
|
|
}
|
|
}
|