diff --git a/src/main.rs b/src/main.rs index 93958e3..717dc60 100644 --- a/src/main.rs +++ b/src/main.rs @@ -7,6 +7,7 @@ use anyhow::{anyhow, Result}; use log::*; +use scene::generators; use winit::dpi::{LogicalSize, LogicalPosition}; use winit::event::{ElementState, Event, WindowEvent}; use winit::event_loop::EventLoop; @@ -222,6 +223,7 @@ impl App { image::create_texture_image_view(&device, &mut data)?; image::create_texture_sampler(&device, &mut data)?; + generators::generate_test_scene(&mut scene_handler, &mut data)?; scene_handler.prepare_data(&instance, &device, &mut data)?; buffer::create_uniform_buffers(&instance, &device, &mut data)?; @@ -237,7 +239,7 @@ impl App { cam_angle_x: 0.0, cam_angle_y: 0.0, last_pos: LogicalPosition::new(-1 as f32, -1 as f32), view_direction: vertex::Vec3::new(0.0, 0.0, 0.0), - cur_pos: cgmath::point3(0.0, 0.0, 0.0), + cur_pos: cgmath::point3(5.0, 5.0, 10.0), scene_handler, show_frame_rate: false, synchronized: 0 diff --git a/src/scene/empty_volume.rs b/src/scene/empty_volume.rs index 5277686..38b690a 100644 --- a/src/scene/empty_volume.rs +++ b/src/scene/empty_volume.rs @@ -1,6 +1,6 @@ -use cgmath::Vector3; +use cgmath::{ElementWise, Vector3}; -use std::cell::RefCell; +use std::cell::{RefCell, Ref}; use std::rc::Rc; use std::time::Instant; @@ -24,6 +24,8 @@ pub struct EmptyVolume { pub size_y: usize, pub size_z: usize, + pub tree_offset: Vector3<usize>, + pub tree_size: usize, pub position: Vector3<usize>, pub color_left: Vec<Vector3<u8>>, @@ -63,22 +65,22 @@ impl EmptyVolume { } // MARK: From Oct Tree - pub fn from_oct_tree(tree: &OctTree<Cube>) -> (Vec<Rc<RefCell<EmptyVolume>>>, OctTree<Rc<RefCell<EmptyVolume>>>) { + pub fn from_oct_tree(tree: &Rc<RefCell<OctTree<Cube>>>, tree_offset: Vector3<usize>) -> (Vec<Rc<RefCell<EmptyVolume>>>, OctTree<Rc<RefCell<EmptyVolume>>>) { // todo: ppotentially use a child exist check while going through the oct tree to find some obvios starting empty volumes. Will still need to check for possible expansions though let mut volumes: Vec<Rc<RefCell<EmptyVolume>>> = vec![]; - let mut neighbors: OctTree<Rc<RefCell<EmptyVolume>>> = OctTree::create(tree.size).unwrap(); + let mut neighbors: OctTree<Rc<RefCell<EmptyVolume>>> = OctTree::create(tree.borrow().size).unwrap(); let start_time = Instant::now(); // iterate over all block positions in the oct tree let mut check_its = 0; let mut x_index = 0; - while x_index < tree.size { + while x_index < tree.borrow().size { let mut y_index = 0; - while y_index < tree.size { + while y_index < tree.borrow().size { let mut z_index = 0; - while z_index < tree.size { + while z_index < tree.borrow().size { // check if there is a block at that position - let query_result = tree.test_element(x_index, y_index, z_index); + let query_result = tree.borrow().test_element(x_index, y_index, z_index); let mut transparent = false; let mut transparent_color = Vector3 {x: 0.0, y: 0.0, z: 0.0}; let mut tranparent_roughness = 0; @@ -116,12 +118,12 @@ impl EmptyVolume { } let mut grow = true; while grow { - grow &= (x_index + x_size + 1) < tree.size; + grow &= (x_index + x_size + 1) < tree.borrow().size; if grow { let mut z = 0; let mut y = 0; while z < z_size.max(1) && y < y_size.max(1) { - let query_result = tree.test_element(x_index + x_size + 1, y_index + y, z_index + z); + let query_result = tree.borrow().test_element(x_index + x_size + 1, y_index + y, z_index + z); check_its += 1; grow &= ((!query_result.0 && !transparent) || (transparent && EmptyVolume::check_transparent(query_result.3, &transparent_color, &tranparent_roughness))) && neighbors.get_element(x_index + x_size + 1, y_index + y, z_index + z).is_none(); @@ -163,12 +165,12 @@ impl EmptyVolume { println!("start growing volume y"); grow = true; while grow { - grow &= (y_index + y_size + 1) < tree.size; + grow &= (y_index + y_size + 1) < tree.borrow().size; if grow { let mut z = 0; let mut x = 0; while z < z_size.max(1) && x < x_size.max(1) { - let query_result = tree.test_element(x_index + x, y_index + y_size + 1, z_index + z); + let query_result = tree.borrow().test_element(x_index + x, y_index + y_size + 1, z_index + z); check_its += 1; grow &= ((!query_result.0 && !transparent) || (transparent && EmptyVolume::check_transparent(query_result.3, &transparent_color, &tranparent_roughness))) && neighbors.get_element(x_index + x, y_index + y_size + 1, z_index + z).is_none(); @@ -211,12 +213,12 @@ impl EmptyVolume { println!("start growing volume z"); grow = true; while grow { - grow &= (z_index + z_size + 1) < tree.size; + grow &= (z_index + z_size + 1) < tree.borrow().size; if grow { let mut y = 0; let mut x = 0; while y < y_size.max(1) && x < x_size.max(1) { - let query_result = tree.test_element(x_index + x, y_index + y, z_index + z_size + 1); + let query_result = tree.borrow().test_element(x_index + x, y_index + y, z_index + z_size + 1); check_its += 1; grow &= ((!query_result.0 && !transparent) || (transparent && EmptyVolume::check_transparent(query_result.3, &transparent_color, &tranparent_roughness))) && neighbors.get_element(x_index + x, y_index + y, z_index + z_size + 1).is_none(); @@ -262,6 +264,8 @@ impl EmptyVolume { size_x: x_size + 1, size_y: y_size + 1, size_z: z_size + 1, + tree_offset, + tree_size: tree.borrow().size, position: Vector3{x: x_index, y: y_index, z: z_index}, color_left: vec![], color_right: vec![], @@ -330,7 +334,7 @@ impl EmptyVolume { let mut bottom_elements_num = 0; for x in 0..x_size+1 { for y in 0..y_size+1 { - if let Some(c) = tree.get_element(reference.borrow().position.x + x, reference.borrow().position.y + y, z_min_pos) { + if let Some(c) = tree.borrow().get_element(reference.borrow().position.x + x, reference.borrow().position.y + y, z_min_pos) { bottom_elements_num += 1; let u8_color = Vector3 {x: (c.color * 255.0).x.min(255.0).max(0.0) as u8, y: (c.color * 255.0).y.min(255.0).max(0.0) as u8, z: (c.color * 255.0).z.min(255.0).max(0.0) as u8}; bottom_colors.push(u8_color); @@ -356,7 +360,7 @@ impl EmptyVolume { let mut top_elements_num = 0; for x in 0..x_size+1 { for y in 0..y_size+1 { - if let Some(c) = tree.get_element(reference.borrow().position.x + x, reference.borrow().position.y + y, z_max_pos) { + if let Some(c) = tree.borrow().get_element(reference.borrow().position.x + x, reference.borrow().position.y + y, z_max_pos) { top_elements_num += 1; let u8_color = Vector3 {x: (c.color * 255.0).x.min(255.0).max(0.0) as u8, y: (c.color * 255.0).y.min(255.0).max(0.0) as u8, z: (c.color * 255.0).z.min(255.0).max(0.0) as u8}; top_colors.push(u8_color); @@ -383,7 +387,7 @@ impl EmptyVolume { let mut back_elements_num = 0; for x in 0..x_size+1 { for z in 0..z_size+1 { - if let Some(c) = tree.get_element(reference.borrow().position.x + x, y_max_pos, reference.borrow().position.z + z) { + if let Some(c) = tree.borrow().get_element(reference.borrow().position.x + x, y_max_pos, reference.borrow().position.z + z) { back_elements_num += 1; let u8_color = Vector3 {x: (c.color * 255.0).x.min(255.0).max(0.0) as u8, y: (c.color * 255.0).y.min(255.0).max(0.0) as u8, z: (c.color * 255.0).z.min(255.0).max(0.0) as u8}; back_colors.push(u8_color); @@ -410,7 +414,7 @@ impl EmptyVolume { let mut front_elements_num = 0; for x in 0..x_size+1 { for z in 0..z_size+1 { - if let Some(c) = tree.get_element(reference.borrow().position.x + x, y_min_pos, reference.borrow().position.z + z) { + if let Some(c) = tree.borrow().get_element(reference.borrow().position.x + x, y_min_pos, reference.borrow().position.z + z) { front_elements_num += 1; let u8_color = Vector3 {x: (c.color * 255.0).x.min(255.0).max(0.0) as u8, y: (c.color * 255.0).y.min(255.0).max(0.0) as u8, z: (c.color * 255.0).z.min(255.0).max(0.0) as u8}; front_colors.push(u8_color); @@ -437,7 +441,7 @@ impl EmptyVolume { let mut left_elements_num = 0; for y in 0..y_size+1 { for z in 0..z_size+1 { - if let Some(c) = tree.get_element(x_min_pos, reference.borrow().position.y + y, reference.borrow().position.z + z) { + if let Some(c) = tree.borrow().get_element(x_min_pos, reference.borrow().position.y + y, reference.borrow().position.z + z) { left_elements_num += 1; let u8_color = Vector3 {x: (c.color * 255.0).x.min(255.0).max(0.0) as u8, y: (c.color * 255.0).y.min(255.0).max(0.0) as u8, z: (c.color * 255.0).z.min(255.0).max(0.0) as u8}; left_colors.push(u8_color); @@ -464,7 +468,7 @@ impl EmptyVolume { let mut right_elements_num = 0; for y in 0..y_size+1 { for z in 0..z_size+1 { - if let Some(c) = tree.get_element(x_max_pos, reference.borrow().position.y + y, reference.borrow().position.z + z) { + if let Some(c) = tree.borrow().get_element(x_max_pos, reference.borrow().position.y + y, reference.borrow().position.z + z) { right_elements_num += 1; let u8_color = Vector3 {x: (c.color * 255.0).x.min(255.0).max(0.0) as u8, y: (c.color * 255.0).y.min(255.0).max(0.0) as u8, z: (c.color * 255.0).z.min(255.0).max(0.0) as u8}; right_colors.push(u8_color); @@ -540,6 +544,7 @@ impl EmptyVolume { } else { bottom_neighbors.push(None); + all_same = all_same && (bottom_neighbors[0] == None); } } } @@ -568,6 +573,7 @@ impl EmptyVolume { } else { top_neighbors.push(None); + all_same = all_same && (top_neighbors[0] == None); } } } @@ -596,6 +602,7 @@ impl EmptyVolume { } else { back_neighbors.push(None); + all_same = all_same && (back_neighbors[0] == None); } } } @@ -625,6 +632,7 @@ impl EmptyVolume { } else { front_neighbors.push(None); + all_same = all_same && (front_neighbors[0] == None); } } } @@ -655,6 +663,7 @@ impl EmptyVolume { } else { left_neighbors.push(None); + all_same = all_same && (left_neighbors[0] == None); } } } @@ -684,6 +693,7 @@ impl EmptyVolume { } else { right_neighbors.push(None); + all_same = all_same && (right_neighbors[0] == None); } } } @@ -705,7 +715,7 @@ impl EmptyVolume { // MARK: To Quads pub fn to_quads(&self) -> Vec<Quad> { let mut quads = vec![]; - let float_pos = Vector3 {x: self.position.x as f32, y: self.position.y as f32, z: self.position.z as f32}; + let float_pos = Vector3 {x: (self.tree_offset.x * self.tree_size + self.position.x) as f32, y: (self.tree_offset.y * self.tree_size + self.position.y) as f32, z: (self.tree_offset.z * self.tree_size + self.position.z) as f32}; //bottom sides of the volumes, top side of the block for x in 0..self.size_x { for y in 0..self.size_y { @@ -890,6 +900,182 @@ impl EmptyVolume { } out_index } + + pub fn combine_results(first: &Rc<RefCell<OctTree<Cube>>>,first_neighbors: &Rc<OctTree<Rc<RefCell<EmptyVolume>>>>, second: &Rc<RefCell<OctTree<Cube>>>, second_neighbors: &Rc<OctTree<Rc<RefCell<EmptyVolume>>>>, facing: vertex::Facing) { + let mut first_start; + let mut second_start; + + let step_one; + let step_two; + + match facing { + vertex::Facing::Back => { + first_start = Vector3{x: 0, y: first.borrow().size - 1, z: 0}; + second_start = Vector3{x: 0, y: 0, z: 0}; + + step_one = Vector3{x: 1, y: 0, z: 0}; + step_two = Vector3{x: 0, y: 0, z: 1}; + }, + vertex::Facing::Front => { + first_start = Vector3{x: 0, y: 0, z: 0}; + second_start = Vector3{x: 0, y: first.borrow().size - 1, z: 0}; + + step_one = Vector3{x: 1, y: 0, z: 0}; + step_two = Vector3{x: 0, y: 0, z: 1}; + }, + vertex::Facing::Top => { + first_start = Vector3{x: 0, y: 0, z: first.borrow().size - 1}; + second_start = Vector3{x: 0, y: 0, z: 0}; + + step_one = Vector3{x: 1, y: 0, z: 0}; + step_two = Vector3{x: 0, y: 1, z: 0}; + }, + vertex::Facing::Bottom => { + first_start = Vector3{x: 0, y: 0, z: 0}; + second_start = Vector3{x: 0, y: 0, z: first.borrow().size - 1}; + + step_one = Vector3{x: 1, y: 0, z: 0}; + step_two = Vector3{x: 0, y: 1, z: 0}; + }, + vertex::Facing::Left => { + first_start = Vector3{x: 0, y: 0, z: 0}; + second_start = Vector3{x: first.borrow().size - 1, y: 0, z: 0}; + + step_one = Vector3{x: 0, y: 1, z: 0}; + step_two = Vector3{x: 0, y: 0, z: 1}; + }, + vertex::Facing::Right => { + first_start = Vector3{x: first.borrow().size - 1, y: 0, z: 0}; + second_start = Vector3{x: 0, y: 0, z: 0}; + + step_one = Vector3{x: 0, y: 1, z: 0}; + step_two = Vector3{x: 0, y: 0, z: 1}; + } + } + + let mut done_volumes = vec![]; + + for u in 0..first.borrow().size { + for v in 0..first.borrow().size { + let first_pos = first_start + v * step_two + u * step_one; + let second_pos = second_start + v * step_two + u * step_one; + let volume_option = first_neighbors.get_element(first_pos.x, first_pos.y, first_pos.z); + if let Some(volume) = volume_option { + if !done_volumes.contains(&volume) { + + let mask = Vector3 {x: 1, y: 1, z: 1} - (step_one + step_two); + let negated_mask = (step_one + step_two); + + let volume_start_first = negated_mask.mul_element_wise(volume.borrow().position) + first_pos.mul_element_wise(mask); + let volume_start_second = negated_mask.mul_element_wise(volume.borrow().position) + second_pos.mul_element_wise(mask); + + let size_u; + let size_v; + if negated_mask.x == 1 { + size_u = volume.borrow().size_x; + if negated_mask.y == 1 { + size_v = volume.borrow().size_y; + } else { + size_v = volume.borrow().size_z; + } + } else { + size_u = volume.borrow().size_y; + size_v = volume.borrow().size_z; + } + + let mut new_colors = vec![]; + let mut new_roughness = vec![]; + let mut new_neighbors = vec![]; + let mut color_elements = 0; + let mut neighbor_elements = 0; + let mut all_same = true; + for volume_u in 0..size_u { + for volume_v in 0..size_v { + let pos = second_pos + volume_u * step_one + volume_v * step_two; + let new_neighbor = second_neighbors.get_element(pos.x, pos.y, pos.z); + let new_cube = second.borrow().get_element(pos.x, pos.y, pos.z); + + if let Some(c) = new_cube { + let u8_color = Vector3 {x: (c.color * 255.0).x.min(255.0).max(0.0) as u8, y: (c.color * 255.0).y.min(255.0).max(0.0) as u8, z: (c.color * 255.0).z.min(255.0).max(0.0) as u8}; + new_colors.push(u8_color); + new_roughness.push(c.roughness); + color_elements += 1; + } else { + new_colors.push(Vector3 { x: 0, y: 0, z: 0 }); + new_roughness.push(0); + } + + if let Some(n) = new_neighbor { + neighbor_elements += 1; + new_neighbors.push(Some(n.clone())); + all_same = all_same && (new_neighbors[0] == Some(n)); + } else { + new_neighbors.push(None); + all_same = all_same && (new_neighbors[0] == None); + } + } + } + + if color_elements > 0 { + new_colors = new_colors; + new_roughness = new_roughness; + } + else { + new_colors= vec![]; + new_roughness = vec![]; + } + + if neighbor_elements > 0 { + if all_same { + new_neighbors = vec![new_neighbors[0].clone()]; + } + else { + new_neighbors = new_neighbors; + } + } + else { + new_neighbors = vec![None]; + } + + match facing { + vertex::Facing::Back => { + volume.borrow_mut().color_back = new_colors; + volume.borrow_mut().roughness_back = new_roughness; + volume.borrow_mut().neighbor_back = new_neighbors; + }, + vertex::Facing::Front => { + volume.borrow_mut().color_front = new_colors; + volume.borrow_mut().roughness_front = new_roughness; + volume.borrow_mut().neighbor_front = new_neighbors; + }, + vertex::Facing::Top => { + volume.borrow_mut().color_top = new_colors; + volume.borrow_mut().roughness_top = new_roughness; + volume.borrow_mut().neighbor_top = new_neighbors; + }, + vertex::Facing::Bottom => { + volume.borrow_mut().color_bottom = new_colors; + volume.borrow_mut().roughness_bottom = new_roughness; + volume.borrow_mut().neighbor_bottom = new_neighbors; + }, + vertex::Facing::Left => { + volume.borrow_mut().color_left = new_colors; + volume.borrow_mut().roughness_left = new_roughness; + volume.borrow_mut().neighbor_left = new_neighbors; + }, + vertex::Facing::Right => { + volume.borrow_mut().color_right = new_colors; + volume.borrow_mut().roughness_right = new_roughness; + volume.borrow_mut().neighbor_right = new_neighbors; + }, + } + + done_volumes.push(volume); + } + } + } + } + } } impl Memorizable for EmptyVolume { @@ -923,11 +1109,11 @@ impl Memorizable for EmptyVolume { fn insert_into_memory(&self, mut v: Vec<u32>, data: &AppData, scene: &Scene) -> Vec<u32> { let mut mem_index = self.memory_start; //pos - v[mem_index] = self.position.x as u32; + v[mem_index] = (self.tree_offset.x * self.tree_size + self.position.x) as u32; mem_index += 1; - v[mem_index] = self.position.y as u32; + v[mem_index] = (self.tree_offset.y * self.tree_size + self.position.y) as u32; mem_index += 1; - v[mem_index] = self.position.z as u32; + v[mem_index] = (self.tree_offset.z * self.tree_size + self.position.z) as u32; mem_index += 1; //max sizes v[mem_index] = self.size_x as u32; diff --git a/src/scene/generators.rs b/src/scene/generators.rs new file mode 100644 index 0000000..71dffba --- /dev/null +++ b/src/scene/generators.rs @@ -0,0 +1,119 @@ +use super::Scene; +use super::oct_tree::{CHUNK_SIZE, OctTree}; +use crate::primitives::cube::Cube; +use crate::primitives::rec_cuboid::Cuboid; +use crate::primitives::drawable::Drawable; +use crate::app_data::AppData; + +extern crate rand; +use rand::Rng; + +use anyhow::Result; + +use cgmath::{vec2, vec3, Vector3}; + +use std::cell::RefCell; +use std::rc::Rc; + +use super::light::{DirectionalLight, PointLight}; + + +pub fn generate_test_scene(scene: &mut Scene, data: &mut AppData) -> Result<()> { + let mut rng = rand::thread_rng(); + let grid_size = CHUNK_SIZE as i32; + + let mut oct_tree1: OctTree<Cube> = OctTree::create(CHUNK_SIZE)?; + let mut oct_tree2: OctTree<Cube> = OctTree::create(CHUNK_SIZE)?; + + for x_index in 0..grid_size { + for y_index in 0..grid_size { + let shade = (rng.gen_range(0..50) as f32) / 100.0; + let cube = Cube { + pos: vec3(x_index as f32, y_index as f32, 5.0), + color: vec3(shade, 1.0, shade), + tex_coord: vec2(0.0, 0.0), + transparent: false, + roughness: 255, + }; + + oct_tree1.set_cube(cube.clone()); + + let shade = (rng.gen_range(0..50) as f32) / 100.0; + let cube = Cube { + pos: vec3(x_index as f32, y_index as f32, 6.0), + color: vec3(shade, shade, 0.0), + tex_coord: vec2(0.0, 0.0), + transparent: false, + roughness: 255, + }; + + oct_tree2.set_cube(cube.clone()); + } + } + + let shade = (rng.gen_range(0..25) as f32) / 100.0; + let cube = Cube { + pos: vec3(10.0, 10.0, 10.0), + color: vec3(0.9, 0.9, 0.9), + tex_coord: vec2(0.0, 0.0), + transparent: true, + roughness: 32, + }; + oct_tree1.set_cube(cube.clone()); + + let cube = Cube { + pos: vec3(10.0, 10.0, 9.0), + color: vec3(0.9, 0.9, 0.9), + tex_coord: vec2(0.0, 0.0), + transparent: true, + roughness: 32, + }; + oct_tree1.set_cube(cube.clone()); + + + let cube = Cube { + pos: vec3(10.0, 10.0, 10.0), + color: vec3(0.9, 0.9, 0.9), + tex_coord: vec2(0.0, 0.0), + transparent: true, + roughness: 32, + }; + oct_tree2.set_cube(cube.clone()); + + let cube = Cube { + pos: vec3(10.0, 10.0, 9.0), + color: vec3(0.9, 0.9, 0.9), + tex_coord: vec2(0.0, 0.0), + transparent: true, + roughness: 32, + }; + oct_tree2.set_cube(cube.clone()); + + scene.point_lights.push(Rc::new(RefCell::new(PointLight { pos: vec3(11.0, 11.0, 11.0), color: vec3(1.0, 1.0, 1.0), memory_start: 0 }))); + scene.point_lights.push(Rc::new(RefCell::new(PointLight { pos: vec3(9.0, 9.0, 11.0), color: vec3(0.5, 0.5, 0.5), memory_start: 0 }))); + scene.directional_lights.push(Rc::new(RefCell::new(DirectionalLight { direction: vec3(1.0, 1.0, -1.0), color: vec3(0.1, 0.1, 0.1), memory_start: 0 }))); + + let cube = Cuboid { + pos: vec3(11.0, 11.0, 11.0), + color: vec3(1.0, 1.0, 1.0), + tex_coord: vec2(0.0, 0.0), + size: Vector3 {x: 0.5, y: 0.5, z: 0.5} + }; + let index = scene.sized_vertices.len(); + cube.draw(&data.topology, index, scene); + + let cube = Cuboid { + pos: vec3(9.0, 9.0, 11.0), + color: vec3(1.0, 1.0, 1.0), + tex_coord: vec2(0.0, 0.0), + size: Vector3 {x: 0.5, y: 0.5, z: 0.5} + }; + let index = scene.sized_vertices.len(); + cube.draw(&data.topology, index, scene); + + let tree_ref_one = Rc::new(RefCell::new(oct_tree1)); + let tree_ref_two = Rc::new(RefCell::new(oct_tree2)); + scene.oct_trees = vec![vec![vec![tree_ref_one.clone(), tree_ref_two.clone()]]]; + + Ok(()) +} \ No newline at end of file diff --git a/src/scene/mod.rs b/src/scene/mod.rs index e9d19e7..5ae9501 100644 --- a/src/scene/mod.rs +++ b/src/scene/mod.rs @@ -2,9 +2,10 @@ mod oct_tree; mod empty_volume; mod light; mod memorizable; +pub mod generators; use anyhow::Ok; -use light::{DirectionalLight, LightSource}; +use light::{DirectionalLight, LightSource, PointLight}; use vulkanalia::prelude::v1_0::*; use anyhow::Result; @@ -22,7 +23,6 @@ use crate::primitives::cube::Cube; use crate::primitives::drawable::Drawable; use crate::scene::oct_tree::{OctTree, OctTreeIter, CHUNK_SIZE}; use crate::scene::empty_volume::EmptyVolume; -use crate::scene::light::PointLight; extern crate rand; use rand::Rng; @@ -57,102 +57,60 @@ pub struct Scene { pub rt_memory: Vec<u32>, + pub oct_trees: Vec<Vec<Vec<Rc<RefCell<OctTree<Cube>>>>>>, + point_lights: Vec<Rc<RefCell<PointLight>>>, directional_lights: Vec<Rc<RefCell<DirectionalLight>>>, } impl Scene { pub unsafe fn prepare_data(&mut self, instance: &vulkanalia::Instance, device: &vulkanalia::Device, data: &mut AppData) -> Result<()> { - let mut rng = rand::thread_rng(); - let grid_size = CHUNK_SIZE as i32; + // todo store the chunks somewhere (or only use them as intermediary for neighbourhood calculation idc) - // todo store the chunks somewhere (or only use them as intermediary for neighbouthood calculation idc) - let mut oct_tree: OctTree<Cube> = OctTree::create(CHUNK_SIZE)?; + let mut empty_volumes: Vec<Rc<RefCell<EmptyVolume>>> = vec![]; - for x_index in 0..grid_size { - for y_index in 0..grid_size { - let shade = (rng.gen_range(0..50) as f32) / 100.0; - let cube = Cube { - pos: vec3(x_index as f32, y_index as f32, 5.0), - color: vec3(shade, 1.0, shade), - tex_coord: vec2(0.0, 0.0), - transparent: false, - roughness: 0, - }; + let mut neighbor_trees: Vec<Vec<Vec<Rc<OctTree<Rc<RefCell<EmptyVolume>>>>>>> = vec![]; - oct_tree.set_cube(cube.clone()); - } - } + let mut x_index = 0; + let mut y_index = 0; + let mut z_index = 0; + for oct_tree_plane_xy in &self.oct_trees { + neighbor_trees.push(vec![]); + for oct_tree_line_y in oct_tree_plane_xy { + neighbor_trees[z_index].push(vec![]); + for oct_tree in oct_tree_line_y { + let mut new_volumes: Vec<Rc<RefCell<EmptyVolume>>>; + let new_neighbors; + (new_volumes, new_neighbors) = EmptyVolume::from_oct_tree(oct_tree, Vector3 { x: x_index, y: y_index, z: z_index }); + empty_volumes.append(&mut new_volumes); - let shade = (rng.gen_range(0..25) as f32) / 100.0; - let cube = Cube { - pos: vec3(10.0, 10.0, 10.0), - color: vec3(1.0, 0.0, 0.0), - tex_coord: vec2(0.0, 0.0), - transparent: true, - roughness: 32, - }; - oct_tree.set_cube(cube.clone()); + neighbor_trees[z_index][y_index].push(Rc::new(new_neighbors)); - let cube = Cube { - pos: vec3(10.0, 10.0, 9.0), - color: vec3(1.0, 0.0, 0.0), - tex_coord: vec2(0.0, 0.0), - transparent: true, - roughness: 32, - }; - oct_tree.set_cube(cube.clone()); - - self.point_lights.push(Rc::new(RefCell::new(PointLight { pos: vec3(11.0, 11.0, 11.0), color: vec3(1.0, 1.0, 1.0), memory_start: 0 }))); - self.point_lights.push(Rc::new(RefCell::new(PointLight { pos: vec3(9.0, 9.0, 11.0), color: vec3(0.5, 0.5, 0.5), memory_start: 0 }))); - self.directional_lights.push(Rc::new(RefCell::new(DirectionalLight { direction: vec3(1.0, 1.0, -1.0), color: vec3(0.1, 0.1, 0.1), memory_start: 0 }))); - - let empty_volumes: Vec<Rc<RefCell<EmptyVolume>>>; - (empty_volumes, _) = EmptyVolume::from_oct_tree(&oct_tree); - println!("number of empty volumes is {}", empty_volumes.len()); - - let oct_tree_iter = OctTreeIter::create(&oct_tree)?; - for item in oct_tree_iter { - let sized_index = self.sized_vertices.len(); - let index = self.vertices.len(); - match item { - Some(cube) => { - /*if (cube.pos.x + cube.pos.y) as usize % 2 == 0{ - /**/ - let cuboid = Cuboid { - pos: cube.pos, - color: cube.color, - tex_coord: cube.tex_coord, - size: Vector3 {x: 1.0, y: 1.0, z: 1.0}, - }; - cuboid.draw(&data.topology, sized_index, self); - } - else { - cube.draw(&data.topology, index, self); - }*/ - //cube.draw(&data.topology, index, self); + x_index += 1; } - None => {} + y_index += 1; } + z_index += 1; } + let mut x_index = 0; + let mut y_index = 0; + let mut z_index = 0; + for oct_tree_plane_xy in &self.oct_trees { + for oct_tree_line_x in oct_tree_plane_xy { + for oct_tree in oct_tree_line_x { + + if oct_tree_line_x.len() > x_index + 1 { + EmptyVolume::combine_results(oct_tree, &neighbor_trees[z_index][y_index][x_index], &oct_tree_line_x[x_index + 1], &neighbor_trees[z_index][y_index][x_index + 1], vertex::Facing::Right); + EmptyVolume::combine_results(&oct_tree_line_x[x_index + 1], &neighbor_trees[z_index][y_index][x_index + 1], oct_tree, &neighbor_trees[z_index][y_index][x_index], vertex::Facing::Left); + } - let cube = Cuboid { - pos: vec3(11.0, 11.0, 11.0), - color: vec3(1.0, 1.0, 1.0), - tex_coord: vec2(0.0, 0.0), - size: Vector3 {x: 0.5, y: 0.5, z: 0.5} - }; - let index = self.sized_vertices.len(); - cube.draw(&data.topology, index, self); - - let cube = Cuboid { - pos: vec3(9.0, 9.0, 11.0), - color: vec3(1.0, 1.0, 1.0), - tex_coord: vec2(0.0, 0.0), - size: Vector3 {x: 0.5, y: 0.5, z: 0.5} - }; - let index = self.sized_vertices.len(); - cube.draw(&data.topology, index, self); + x_index += 1; + } + y_index += 1; + } + z_index += 1; + } + println!("number of empty volumes is {}", empty_volumes.len()); let mut memory_index = 6; // 0 - location for the maximum number of lights referenced per chunk (also will be the invalid memory allocation for pointing to a nonexistant neighbor) diff --git a/src/scene/oct_tree.rs b/src/scene/oct_tree.rs index c9f94fa..f0e5da6 100644 --- a/src/scene/oct_tree.rs +++ b/src/scene/oct_tree.rs @@ -7,7 +7,7 @@ use crate::primitives::cube::Cube; extern crate rand; -pub const CHUNK_SIZE_EXPONENT: u32 = 6; +pub const CHUNK_SIZE_EXPONENT: u32 = 4; pub const CHUNK_SIZE: usize = (2 as usize).pow(CHUNK_SIZE_EXPONENT); pub const MAX_TREE_DEPTH: usize = CHUNK_SIZE_EXPONENT as usize - 2; pub const MIN_CHUNK_SIZE: usize = CHUNK_SIZE / (2 as usize).pow(MAX_TREE_DEPTH as u32);