489 lines
18 KiB
Rust
489 lines
18 KiB
Rust
use crate::{actor, audio, hud, nature, shading, var};
|
|
use bevy::prelude::*;
|
|
use bevy::math::{DVec3, I64Vec3};
|
|
use bevy::scene::{InstanceId, SceneInstance};
|
|
use bevy::render::mesh::Indices;
|
|
use bevy_xpbd_3d::prelude::*;
|
|
use bevy_xpbd_3d::plugins::sync;
|
|
use std::collections::HashMap;
|
|
use std::f32::consts::PI;
|
|
use fastrand;
|
|
|
|
const ASTEROID_UPDATE_INTERVAL: f32 = 0.1; // seconds
|
|
const ASTEROID_SIZE_FACTOR: f32 = 10.0;
|
|
const RING_THICKNESS: f64 = 8.0e6;
|
|
const STARS_MAX_MAGNITUDE: f32 = 5.5; // max 7.0, see generate_starchart.py
|
|
|
|
const CENTER_WORLD_ON_PLAYER: bool = true;
|
|
const SKYBOX: bool = false;
|
|
|
|
const ASTEROID_SPAWN_STEP: f64 = 500.0;
|
|
const ASTEROID_VIEW_RADIUS: f64 = 3000.0;
|
|
|
|
const ASSET_ASTEROID1: &str = "models/asteroid.glb#Scene0";
|
|
const ASSET_ASTEROID2: &str = "models/asteroid2.glb#Scene0";
|
|
pub fn asset_name_to_path(name: &str) -> &'static str {
|
|
match name {
|
|
"suit" => "models/suit_with_collider.glb#Scene0",
|
|
"suit_ar_chefhat" => "models/suit_ar_chefhat.glb#Scene0",
|
|
"asteroid1" => ASSET_ASTEROID1,
|
|
"asteroid2" => ASSET_ASTEROID2,
|
|
"asteroid_lum" => "models/asteroid_lum.glb#Scene0",
|
|
"moonlet" => "models/moonlet.glb#Scene0",
|
|
"monolith" => "models/monolith_neon.glb#Scene0",
|
|
"lightorb" => "models/lightorb.glb#Scene0",
|
|
"orb_busstop" => "models/orb_busstop.glb#Scene0",
|
|
"orb_busstop_dim" => "models/orb_busstop_dim.glb#Scene0",
|
|
"MeteorAceGT" => "models/MeteorAceGT.glb#Scene0",
|
|
"satellite" => "models/satellite.glb#Scene0",
|
|
"pizzeria" => "models/pizzeria2.glb#Scene0",
|
|
"pizzasign" => "models/pizzasign.glb#Scene0",
|
|
"selectagon" => "models/selectagon.glb#Scene0",
|
|
"clippy" => "models/clippy.glb#Scene0",
|
|
"clippy_ar" => "models/clippy_ar.glb#Scene0",
|
|
"whale" => "models/whale.glb#Scene0",
|
|
_ => "models/error.glb#Scene0",
|
|
}
|
|
}
|
|
|
|
pub struct WorldPlugin;
|
|
impl Plugin for WorldPlugin {
|
|
fn build(&self, app: &mut App) {
|
|
app.add_systems(Startup, setup);
|
|
app.add_systems(Update, handle_cheats);
|
|
app.add_systems(PostUpdate, handle_despawn);
|
|
app.add_systems(Update, spawn_despawn_asteroids);
|
|
app.add_plugins(PhysicsPlugins::default());
|
|
//app.add_plugins(PhysicsDebugPlugin::default());
|
|
app.insert_resource(Gravity(DVec3::splat(0.0)));
|
|
app.insert_resource(AsteroidUpdateTimer(
|
|
Timer::from_seconds(ASTEROID_UPDATE_INTERVAL, TimerMode::Repeating)));
|
|
app.insert_resource(AsteroidDatabase(Vec::new()));
|
|
app.insert_resource(ActiveAsteroids(HashMap::new()));
|
|
app.add_event::<DespawnEvent>();
|
|
|
|
if CENTER_WORLD_ON_PLAYER {
|
|
// Disable bevy_xpbd's position->transform sync function
|
|
app.insert_resource(sync::SyncConfig {
|
|
position_to_transform: true,
|
|
transform_to_position: false,
|
|
});
|
|
// Add own position->transform sync function
|
|
app.add_systems(PostUpdate, position_to_transform
|
|
.after(sync::position_to_transform)
|
|
.in_set(sync::SyncSet::PositionToTransform));
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Resource)] struct AsteroidUpdateTimer(Timer);
|
|
#[derive(Resource)] struct AsteroidDatabase(Vec<AsteroidData>);
|
|
#[derive(Resource)] struct ActiveAsteroids(HashMap<I64Vec3, AsteroidData>);
|
|
#[derive(Resource)] struct AsteroidModel1(Handle<Scene>);
|
|
#[derive(Resource)] struct AsteroidModel2(Handle<Scene>);
|
|
|
|
#[derive(Component)] struct Asteroid;
|
|
#[derive(Component)] pub struct DespawnOnPlayerDeath;
|
|
|
|
struct AsteroidData {
|
|
entity: Entity,
|
|
//viewdistance: f64,
|
|
}
|
|
|
|
#[derive(Event)]
|
|
pub struct DespawnEvent {
|
|
entity: Entity,
|
|
sceneinstance: InstanceId,
|
|
origin: I64Vec3,
|
|
}
|
|
|
|
#[derive(Component)]
|
|
pub struct Star;
|
|
|
|
pub fn setup(
|
|
mut commands: Commands,
|
|
mut meshes: ResMut<Assets<Mesh>>,
|
|
mut materials: ResMut<Assets<StandardMaterial>>,
|
|
mut materials_jupiter: ResMut<Assets<shading::JupitersRing>>,
|
|
mut materials_skybox: ResMut<Assets<shading::SkyBox>>,
|
|
asset_server: Res<AssetServer>,
|
|
) {
|
|
// Load assets
|
|
commands.insert_resource(AsteroidModel1(asset_server.load(ASSET_ASTEROID1)));
|
|
commands.insert_resource(AsteroidModel2(asset_server.load(ASSET_ASTEROID2)));
|
|
|
|
// Generate starmap
|
|
let sphere_handle = meshes.add(Sphere::new(1.0).mesh().uv(16, 16));
|
|
let mut starcount = 0;
|
|
for (index, star) in nature::STARS.iter().enumerate() {
|
|
let (x, y, z, mag, absmag, color_index, radius, name) = *star;
|
|
if mag > STARS_MAX_MAGNITUDE {
|
|
continue;
|
|
}
|
|
|
|
let (r, g, b) = nature::star_color_index_to_rgb(color_index);
|
|
let mut pos = DVec3::new(x as f64, z as f64, -y as f64) * nature::PARSEC2METER;
|
|
if pos.length() > 1e21 {
|
|
pos *= 0.002;
|
|
}
|
|
let pos_render = pos * 1.0;
|
|
let scale_factor = 1e-4 * pos_render.length() as f32; // from experimentation
|
|
|
|
let mag = mag.min(6.0);
|
|
let scale_size = {|mag: f32|
|
|
scale_factor * (0.230299 * mag * mag - 3.09013 * mag + 15.1782)
|
|
};
|
|
let scale = scale_size(mag);
|
|
|
|
let scale_color = {|color: f32|
|
|
1.2 * color * (0.0659663 * mag * mag - 1.09862 * mag + 4.3)
|
|
};
|
|
//let scale = translation.length().powf(0.84);
|
|
//pos_render.length().powf(0.64)
|
|
//(radius as f64 * nature::SOL_RADIUS).powf(0.02) as f32 *
|
|
|
|
let star_color_handle = materials.add(StandardMaterial {
|
|
base_color: Color::rgb(scale_color(r), scale_color(g), scale_color(b)),
|
|
unlit: true,
|
|
..default()
|
|
});
|
|
let name = if name.is_empty() {
|
|
format!("Uncharted Star #{index}")
|
|
} else {
|
|
name.to_string()
|
|
};
|
|
let distance = if pos.length() > 1e21 {
|
|
None
|
|
} else {
|
|
Some(pos.length())
|
|
};
|
|
|
|
commands.spawn((
|
|
Star,
|
|
hud::IsClickable {
|
|
name: Some(name),
|
|
distance,
|
|
},
|
|
PbrBundle {
|
|
mesh: sphere_handle.clone(),
|
|
material: star_color_handle,
|
|
transform: Transform {
|
|
translation: pos_render.as_vec3(),
|
|
scale: Vec3::splat(scale as f32),
|
|
..default()
|
|
},
|
|
..default()
|
|
},
|
|
));
|
|
starcount += 1;
|
|
}
|
|
info!("Generated {starcount} stars");
|
|
|
|
// Add shaded skybox
|
|
if SKYBOX {
|
|
let mut mesh = Mesh::from(Sphere::new(1e10).mesh().uv(5, 5));
|
|
//let mut mesh = Mesh::from(Cuboid::from_size(Vec3::splat(2e10)));
|
|
if let Some(Indices::U32(indices)) = mesh.indices_mut() {
|
|
// Reverse the order of each triangle to avoid backface culling
|
|
for slice in indices.chunks_mut(3) {
|
|
slice.reverse();
|
|
}
|
|
}
|
|
commands.spawn(MaterialMeshBundle {
|
|
mesh: meshes.add(mesh),
|
|
material: materials_skybox.add(shading::SkyBox {}),
|
|
transform: Transform::from_translation(Vec3::new(0.0, 0.0, 0.0)),
|
|
..default()
|
|
});
|
|
}
|
|
|
|
// Add shaded ring
|
|
let ring_radius = 229_000_000.0;
|
|
let jupiter_radius = 71_492_000.0;
|
|
commands.spawn((
|
|
MaterialMeshBundle {
|
|
mesh: meshes.add(Mesh::from(Cylinder::new(ring_radius, 1.0))),
|
|
material: materials_jupiter.add(shading::JupitersRing {
|
|
alpha_mode: AlphaMode::Blend,
|
|
ring_radius: ring_radius,
|
|
jupiter_radius: jupiter_radius,
|
|
}),
|
|
..default()
|
|
},
|
|
Position::from_xyz(0.0, 0.0, 0.0),
|
|
Rotation::from(Quat::IDENTITY),
|
|
//Rotation::from(Quat::from_rotation_x(-0.3f32.to_radians())),
|
|
));
|
|
}
|
|
|
|
fn spawn_despawn_asteroids(
|
|
time: Res<Time>,
|
|
mut timer: ResMut<AsteroidUpdateTimer>,
|
|
mut commands: Commands,
|
|
q_player: Query<&Position, With<actor::PlayerCamera>>,
|
|
mut ew_despawn: EventWriter<DespawnEvent>,
|
|
mut db: ResMut<ActiveAsteroids>,
|
|
asteroid1_handle: Res<AsteroidModel1>,
|
|
asteroid2_handle: Res<AsteroidModel2>,
|
|
mut q_asteroid: Query<(Entity, &SceneInstance), With<Asteroid>>,
|
|
mut last_player_cell: Local<I64Vec3>,
|
|
) {
|
|
if !timer.0.tick(time.delta()).just_finished() || q_player.is_empty() {
|
|
//if q_player.is_empty() {
|
|
return;
|
|
}
|
|
let player = q_player.get_single().unwrap();
|
|
let player_cell = I64Vec3::new(
|
|
(player.x / ASTEROID_SPAWN_STEP).round() as i64,
|
|
(player.y / ASTEROID_SPAWN_STEP).round() as i64,
|
|
(player.z / ASTEROID_SPAWN_STEP).round() as i64,
|
|
);
|
|
if *last_player_cell == player_cell {
|
|
return;
|
|
}
|
|
*last_player_cell = player_cell;
|
|
|
|
// Parameters
|
|
let view_radius: f64 = ASTEROID_VIEW_RADIUS;
|
|
let step: f64 = ASTEROID_SPAWN_STEP;
|
|
let stepmax: i64 = (view_radius / step) as i64;
|
|
|
|
// Despawn far asteroids
|
|
let x_min = player_cell.x - stepmax;
|
|
let x_max = player_cell.x + stepmax;
|
|
let y_min = player_cell.y - stepmax;
|
|
let y_max = player_cell.y + stepmax;
|
|
let z_min = player_cell.z - stepmax;
|
|
let z_max = player_cell.z + stepmax;
|
|
for (origin, asteroid) in db.0.iter() {
|
|
if origin.x < x_min || origin.x > x_max
|
|
|| origin.y < y_min || origin.y > y_max
|
|
|| origin.z < z_min || origin.z > z_max
|
|
{
|
|
let mut despawning_worked = false;
|
|
for (ent, sceneinstance) in &mut q_asteroid {
|
|
if ent == asteroid.entity {
|
|
ew_despawn.send(DespawnEvent {
|
|
entity: asteroid.entity,
|
|
sceneinstance: **sceneinstance,
|
|
origin: origin.clone(),
|
|
});
|
|
despawning_worked = true;
|
|
break;
|
|
}
|
|
}
|
|
if !despawning_worked {
|
|
error!("Couldn't despawn asteroid:");
|
|
dbg!(origin);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Density based on the radius alone
|
|
let radius_plane = (player.x * player.x + player.z * player.z).sqrt();
|
|
let density_r = nature::ring_density((radius_plane / 1e6) as f32);
|
|
if density_r < 0.001 {
|
|
return;
|
|
}
|
|
// Density based on radius and the vertical distance to the ring
|
|
let normalized_distance = player.y / (RING_THICKNESS / 2.0);
|
|
let density = density_r * (-4.0 * normalized_distance.powf(2.0)).exp() as f32;
|
|
if density < 0.001 {
|
|
return;
|
|
}
|
|
|
|
let mut rng = fastrand::Rng::new();
|
|
for x in -stepmax..=stepmax {
|
|
for y in -stepmax..=stepmax {
|
|
for z in -stepmax..=stepmax {
|
|
let origin = I64Vec3::new(
|
|
player_cell.x + x as i64,
|
|
player_cell.y + y as i64,
|
|
player_cell.z + z as i64,
|
|
);
|
|
if db.0.contains_key(&origin) {
|
|
// already in db, nothing to do
|
|
continue;
|
|
}
|
|
|
|
// Get a seed based on all of the origin axes
|
|
// Probably there's a faster way
|
|
rng.seed(origin.x as u64);
|
|
let seed = rng.u64(0..u64::MAX).wrapping_add(origin.y as u64);
|
|
rng.seed(seed);
|
|
let seed = rng.u64(0..u64::MAX).wrapping_add(origin.z as u64);
|
|
rng.seed(seed);
|
|
|
|
let rand_s = rng.f32();
|
|
let size_raw: f32 = (rand_s + 1e-4).powf(-0.5) as f32; // -> between ~1 and 100
|
|
let size_density = size_raw * density;
|
|
|
|
if size_density < 0.3 {
|
|
continue;
|
|
}
|
|
if size_raw < 20.0 {
|
|
let dist = player_cell.as_dvec3().distance(origin.as_dvec3());
|
|
if dist > 6.0 {
|
|
continue;
|
|
}
|
|
}
|
|
let size: f32 = ASTEROID_SIZE_FACTOR * size_density;
|
|
|
|
let rand_x = rng.f64();
|
|
let rand_y = rng.f64();
|
|
let rand_z = rng.f64();
|
|
let rand_c = rng.bool();
|
|
let class = if rand_c { 0 } else { 1 };
|
|
|
|
//let max_viewdist = ASTEROID_VIEW_RADIUS / ASTEROID_SPAWN_STEP;
|
|
let wobble = ASTEROID_SPAWN_STEP * 0.5;
|
|
let pos = DVec3::new(
|
|
origin.x as f64 * ASTEROID_SPAWN_STEP + wobble * rand_x * 2.0 - 1.0,
|
|
origin.y as f64 * ASTEROID_SPAWN_STEP + wobble * rand_y * 2.0 - 1.0,
|
|
origin.z as f64 * ASTEROID_SPAWN_STEP + wobble * rand_z * 2.0 - 1.0,
|
|
);
|
|
|
|
// Spawn
|
|
let mut entity_commands = commands.spawn((
|
|
actor::Actor::default(),
|
|
RigidBody::Dynamic,
|
|
AngularVelocity(DVec3::new(0.1, 0.1, 0.03)),
|
|
LinearVelocity(DVec3::new(0.0, 0.0, 0.35)),
|
|
Collider::sphere(1.0),
|
|
Rotation::from(Quat::from_rotation_y(-PI / 3.)),
|
|
Position::new(pos),
|
|
Asteroid,
|
|
));
|
|
let model = match class {
|
|
0 => asteroid1_handle.0.clone(),
|
|
_ => asteroid2_handle.0.clone(),
|
|
};
|
|
entity_commands.insert(SceneBundle {
|
|
scene: model,
|
|
transform: Transform {
|
|
scale: Vec3::splat(size),
|
|
..default()
|
|
},
|
|
..default()
|
|
});
|
|
db.0.insert(origin, AsteroidData {
|
|
entity: entity_commands.id(),
|
|
//viewdistance: 99999999.0,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn handle_despawn(
|
|
mut commands: Commands,
|
|
mut er_despawn: EventReader<DespawnEvent>,
|
|
mut db: ResMut<ActiveAsteroids>,
|
|
mut scene_spawner: ResMut<SceneSpawner>,
|
|
) {
|
|
for despawn in er_despawn.read() {
|
|
commands.entity(despawn.entity).despawn();
|
|
scene_spawner.despawn_instance(despawn.sceneinstance);
|
|
db.0.remove(&despawn.origin);
|
|
}
|
|
}
|
|
|
|
fn handle_cheats(
|
|
key_input: Res<ButtonInput<KeyCode>>,
|
|
mut q_player: Query<(&Transform, &mut Position, &mut LinearVelocity), With<actor::PlayerCamera>>,
|
|
mut q_life: Query<(&mut actor::LifeForm, &mut actor::ExperiencesGForce), With<actor::Player>>,
|
|
q_target: Query<(&Transform, &Position, &LinearVelocity), (With<hud::IsTargeted>, Without<actor::PlayerCamera>)>,
|
|
mut ew_playerdies: EventWriter<actor::PlayerDiesEvent>,
|
|
mut settings: ResMut<var::Settings>,
|
|
mut ew_sfx: EventWriter<audio::PlaySfxEvent>,
|
|
) {
|
|
if q_player.is_empty() || q_life.is_empty() {
|
|
return;
|
|
}
|
|
let (trans, mut pos, mut v) = q_player.get_single_mut().unwrap();
|
|
let (mut lifeform, mut gforce) = q_life.get_single_mut().unwrap();
|
|
let boost = if key_input.pressed(KeyCode::ShiftLeft) {
|
|
1e6
|
|
} else {
|
|
1e3
|
|
};
|
|
if key_input.just_pressed(settings.key_cheat_god_mode) {
|
|
settings.god_mode ^= true;
|
|
if settings.god_mode {
|
|
ew_sfx.send(audio::PlaySfxEvent(audio::Sfx::EnterVehicle));
|
|
}
|
|
}
|
|
if !settings.god_mode && !settings.dev_mode {
|
|
return;
|
|
}
|
|
|
|
if key_input.just_pressed(settings.key_cheat_stop) {
|
|
gforce.ignore_gforce_seconds = 1.0;
|
|
v.0 = DVec3::ZERO;
|
|
}
|
|
if key_input.pressed(settings.key_cheat_speed) {
|
|
gforce.ignore_gforce_seconds = 1.0;
|
|
v.0 += DVec3::from(trans.rotation * Vec3::new(0.0, 0.0, boost));
|
|
}
|
|
if key_input.pressed(settings.key_cheat_speed_backward) {
|
|
gforce.ignore_gforce_seconds = 1.0;
|
|
v.0 += DVec3::from(trans.rotation * Vec3::new(0.0, 0.0, -boost));
|
|
}
|
|
if key_input.just_pressed(settings.key_cheat_teleport) {
|
|
if let Ok((transform, target_pos, target_v)) = q_target.get_single() {
|
|
let offset: DVec3 = 4.0 * (**pos - **target_pos).normalize() * transform.scale.as_dvec3();
|
|
pos.0 = **target_pos + offset;
|
|
*v = target_v.clone();
|
|
}
|
|
}
|
|
|
|
if !settings.dev_mode {
|
|
return;
|
|
}
|
|
|
|
if key_input.just_pressed(settings.key_cheat_pizza) {
|
|
pos.0 = DVec3::new(-121100218.0, 593057.0, -190818113.0);
|
|
gforce.ignore_gforce_seconds = 1.0;
|
|
}
|
|
if key_input.just_pressed(settings.key_cheat_farview1) {
|
|
pos.0 = DVec3::new(27643e3, -47e3, -124434e3);
|
|
gforce.ignore_gforce_seconds = 1.0;
|
|
}
|
|
if key_input.just_pressed(settings.key_cheat_farview2) {
|
|
pos.0 = DVec3::new(-184968e3, 149410e3, -134273e3);
|
|
gforce.ignore_gforce_seconds = 1.0;
|
|
}
|
|
if key_input.pressed(settings.key_cheat_adrenaline_zero) {
|
|
lifeform.adrenaline = 0.0;
|
|
}
|
|
if key_input.pressed(settings.key_cheat_adrenaline_mid) {
|
|
lifeform.adrenaline = 0.5;
|
|
}
|
|
if key_input.pressed(settings.key_cheat_adrenaline_max) {
|
|
lifeform.adrenaline = 1.0;
|
|
}
|
|
if key_input.just_pressed(settings.key_cheat_die) {
|
|
settings.god_mode = false;
|
|
ew_playerdies.send(actor::PlayerDiesEvent(actor::DamageType::Trauma));
|
|
}
|
|
}
|
|
|
|
// An extension of bevy_xpbd_3d::plugins::position_to_transform that adjusts
|
|
// the rendering position to center entities at the player camera.
|
|
// This avoids rendering glitches when very far away from the origin.
|
|
pub fn position_to_transform(
|
|
q_player: Query<&Position, With<actor::PlayerCamera>>,
|
|
mut q_trans: Query<(&'static mut Transform, &'static Position, &'static Rotation), Without<Parent>>,
|
|
) {
|
|
if let Ok(player_pos) = q_player.get_single() {
|
|
for (mut transform, pos, rot) in &mut q_trans {
|
|
transform.translation = Vec3::new(
|
|
(pos.x - player_pos.x) as f32,
|
|
(pos.y - player_pos.y) as f32,
|
|
(pos.z - player_pos.z) as f32,
|
|
);
|
|
transform.rotation = rot.as_quat();
|
|
}
|
|
}
|
|
}
|