Skip to content

Commit

Permalink
feat: add planet generation
Browse files Browse the repository at this point in the history
  • Loading branch information
@manankarnik
manankarnik committed Dec 24, 2023
1 parent 0979da0 commit 6444ac1
Show file tree
Hide file tree
Showing 5 changed files with 493 additions and 55 deletions.
276 changes: 276 additions & 0 deletions src/bin/terrain.rs
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,276 @@
use bevy::{pbr::wireframe::WireframePlugin, prelude::*};
use bevy_egui::{
egui::{self, RichText},
EguiContexts, EguiPlugin,
};
use bevy_generative::{
noise::{FunctionName, Method, Noise, Region},
terrain::{Terrain, TerrainBundle, TerrainPlugin},
};
use bevy_panorbit_camera::{PanOrbitCamera, PanOrbitCameraPlugin};
use egui::{ComboBox, DragValue, Slider};

#[cfg(target_arch = "wasm32")]
fn main() {
App::new()
.add_plugins(DefaultPlugins.set(WindowPlugin {
primary_window: Some(Window {
canvas: Some("#bevy-canvas".into()),
..default()
}),
..default()
}))
.add_plugins(EguiPlugin)
.add_plugins(WireframePlugin)
.add_plugins(PanOrbitCameraPlugin)
.add_plugins(TerrainPlugin)
.add_systems(Startup, setup)
.add_systems(Update, gui)
.run();
}
#[cfg(not(target_arch = "wasm32"))]
fn main() {
App::new()
.add_plugins(DefaultPlugins)
.add_plugins(EguiPlugin)
.add_plugins(WireframePlugin)
.add_plugins(PanOrbitCameraPlugin)
.add_plugins(TerrainPlugin)
.add_systems(Startup, setup)
.add_systems(Update, gui)
.run();
}

fn setup(mut commands: Commands) {
commands.spawn(DirectionalLightBundle {
directional_light: DirectionalLight {
color: Color::WHITE,
illuminance: 10000.0,
..default()
},
transform: Transform::from_xyz(-2.0, 2.5, 5.0).looking_at(Vec3::ZERO, Vec3::Y),
..default()
});
commands.spawn((
Camera3dBundle {
transform: Transform::from_xyz(-2.0, 2.5, 5.0).looking_at(Vec3::ZERO, Vec3::Y),
camera_3d: Camera3d {
clear_color: bevy::core_pipeline::clear_color::ClearColorConfig::Custom(
Color::BLACK,
),
..default()
},
..default()
},
PanOrbitCamera::default(),
));
commands.spawn(TerrainBundle {
terrain: Terrain {
noise: Noise {
size: [100; 2],
..default()
},
..default()
},
..default()
});
}

fn gui(mut contexts: EguiContexts, mut query: Query<&mut Terrain>) {
let mut terrain = query.single_mut();

let texture_id = contexts.add_image(terrain.noise.gradient.image.clone_weak());
let mut min_pos = 0.0;

egui::SidePanel::left("Config").show(contexts.ctx_mut(), |ui| {
ui.set_style(egui::style::Style {
spacing: egui::style::Spacing {
text_edit_width: 150.0,
..default()
},
..default()
});
ui.heading("Config");
ui.separator();

if ui.button("Export").clicked() {
terrain.export = true
}

ComboBox::from_label("Method")
.selected_text(terrain.noise.method.to_string())
.show_ui(ui, |ui| {
ui.selectable_value(
&mut terrain.noise.method,
Method::OpenSimplex,
Method::OpenSimplex.to_string(),
);
ui.selectable_value(
&mut terrain.noise.method,
Method::Perlin,
Method::Perlin.to_string(),
);
ui.selectable_value(
&mut terrain.noise.method,
Method::PerlinSurflet,
Method::PerlinSurflet.to_string(),
);
ui.selectable_value(
&mut terrain.noise.method,
Method::Simplex,
Method::Simplex.to_string(),
);
ui.selectable_value(
&mut terrain.noise.method,
Method::SuperSimplex,
Method::SuperSimplex.to_string(),
);
ui.selectable_value(
&mut terrain.noise.method,
Method::Value,
Method::Value.to_string(),
);
ui.selectable_value(
&mut terrain.noise.method,
Method::Worley,
Method::Worley.to_string(),
);
});
ui.horizontal(|ui| {
ui.label("Seed");
ui.add(DragValue::new(&mut terrain.noise.seed));
});
ui.horizontal(|ui| {
ui.label("X");
ui.add(DragValue::new(&mut terrain.noise.offset[0]));
});
ui.horizontal(|ui| {
ui.label("Y");
ui.add(DragValue::new(&mut terrain.noise.offset[1]));
});
ui.horizontal(|ui| {
ui.label("Width");
ui.add(DragValue::new(&mut terrain.noise.size[0]).clamp_range(1..=10000));
});
ui.horizontal(|ui| {
ui.label("Height");
ui.add(DragValue::new(&mut terrain.noise.size[1]).clamp_range(1..=10000));
});
ui.checkbox(&mut terrain.wireframe, "Wireframe");
ui.add(Slider::new(&mut terrain.noise.scale, 1.0..=100.0).text("Scale"));

ComboBox::from_label("Function")
.selected_text(if let Some(function_name) = &terrain.noise.function.name {
function_name.to_string()
} else {
"None".to_string()
})
.show_ui(ui, |ui| {
ui.selectable_value(&mut terrain.noise.function.name, None, "None");
ui.selectable_value(
&mut terrain.noise.function.name,
Some(FunctionName::BasicMulti),
FunctionName::BasicMulti.to_string(),
);
ui.selectable_value(
&mut terrain.noise.function.name,
Some(FunctionName::Billow),
FunctionName::Billow.to_string(),
);
ui.selectable_value(
&mut terrain.noise.function.name,
Some(FunctionName::Fbm),
FunctionName::Fbm.to_string(),
);
ui.selectable_value(
&mut terrain.noise.function.name,
Some(FunctionName::HybridMulti),
FunctionName::HybridMulti.to_string(),
);
ui.selectable_value(
&mut terrain.noise.function.name,
Some(FunctionName::RidgedMulti),
FunctionName::RidgedMulti.to_string(),
);
});
if let Some(_function_name) = &terrain.noise.function.name {
ui.add(Slider::new(&mut terrain.noise.function.octaves, 0..=10).text("Octaves"));
ui.add(
Slider::new(&mut terrain.noise.function.frequency, 0.0..=10.0).text("Frequency"),
);
ui.add(
Slider::new(&mut terrain.noise.function.lacunarity, 0.0..=30.0).text("Lacunarity"),
);
ui.add(
Slider::new(&mut terrain.noise.function.persistence, 0.01..=1.0)
.text("Persistence"),
);
}
ui.group(|ui| {
ui.add(egui::widgets::Image::new(egui::load::SizedTexture::new(
texture_id,
[
terrain.noise.gradient.size[0] as f32,
terrain.noise.gradient.size[1] as f32,
],
)));
ui.add(
Slider::new(&mut terrain.noise.gradient.smoothness, 0.0..=1.0).text("Smoothness"),
);
ui.horizontal(|ui| {
ui.label("Segments");
ui.add(DragValue::new(&mut terrain.noise.gradient.segments).clamp_range(0..=100));
});
ui.horizontal(|ui| {
ui.label("Base Color");
ui.color_edit_button_srgba_unmultiplied(&mut terrain.noise.base_color);
});
ui.add(Slider::new(&mut terrain.height_exponent, 0.1..=2.5).text("Height Exponent"));
ui.add(Slider::new(&mut terrain.sea_level, 0.0..=100.0).text("Sea Level"));
ui.separator();
if ui.button("Add Region").clicked() {
let index = terrain.noise.regions.len() + 1;
terrain.noise.regions.push(Region {
label: format!("Region #{index}"),
position: 0.0,
color: [0, 0, 0, 255],
..default()
});
}
ui.separator();
let regions_len = terrain.noise.regions.len();
let mut regions_to_remove: Vec<usize> = Vec::with_capacity(regions_len);
egui::ScrollArea::vertical().show(ui, |ui| {
for (i, region) in terrain.noise.regions.iter_mut().enumerate() {
ui.horizontal(|ui| {
ui.label(RichText::new(&format!("Region #{}", i + 1)).size(16.0));
if ui.button("Remove").clicked() {
regions_to_remove.push(i);
}
});
ui.horizontal(|ui| {
ui.label("Label");
ui.text_edit_singleline(&mut region.label);
});

ui.horizontal(|ui| {
ui.label("Position");
ui.add(DragValue::new(&mut region.position).clamp_range(min_pos..=100.0));
});
min_pos = region.position;

ui.horizontal(|ui| {
ui.label("Color");
ui.color_edit_button_srgba_unmultiplied(&mut region.color);
});
if i != regions_len - 1 {
ui.separator();
}
}
});
for i in regions_to_remove {
terrain.noise.regions.remove(i);
}
});
});
}
1 change: 1 addition & 0 deletions src/lib.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,4 +16,5 @@ mod util;

pub mod noise;
pub mod noise_map;
pub mod planet;
pub mod terrain;
Loading

0 comments on commit 6444ac1

Please sign in to comment.