mirror of
https://gitlab.com/veloren/veloren.git
synced 2024-08-30 18:12:32 +00:00
446 lines
15 KiB
Rust
446 lines
15 KiB
Rust
#![deny(unsafe_code)]
|
|
#![allow(incomplete_features)]
|
|
#![allow(
|
|
clippy::option_map_unit_fn,
|
|
clippy::blocks_in_if_conditions,
|
|
clippy::too_many_arguments
|
|
)]
|
|
#![allow(clippy::branches_sharing_code)] // TODO: evaluate
|
|
#![deny(clippy::clone_on_ref_ptr)]
|
|
#![feature(
|
|
arbitrary_enum_discriminant,
|
|
bool_to_option,
|
|
const_generics,
|
|
const_panic,
|
|
label_break_value,
|
|
array_map
|
|
)]
|
|
|
|
mod all;
|
|
mod block;
|
|
pub mod canvas;
|
|
pub mod civ;
|
|
mod column;
|
|
pub mod config;
|
|
pub mod index;
|
|
pub mod land;
|
|
pub mod layer;
|
|
pub mod pathfinding;
|
|
pub mod sim;
|
|
pub mod sim2;
|
|
pub mod site;
|
|
pub mod site2;
|
|
pub mod util;
|
|
|
|
// Reexports
|
|
pub use crate::{
|
|
canvas::{Canvas, CanvasInfo},
|
|
config::CONFIG,
|
|
land::Land,
|
|
};
|
|
pub use block::BlockGen;
|
|
pub use column::ColumnSample;
|
|
pub use index::{IndexOwned, IndexRef};
|
|
|
|
use crate::{
|
|
column::ColumnGen,
|
|
index::Index,
|
|
site::SiteKind,
|
|
util::{Grid, Sampler},
|
|
};
|
|
use common::{
|
|
assets,
|
|
generation::{ChunkSupplement, EntityInfo},
|
|
resources::TimeOfDay,
|
|
terrain::{Block, BlockKind, SpriteKind, TerrainChunk, TerrainChunkMeta, TerrainChunkSize},
|
|
vol::{ReadVol, RectVolSize, WriteVol},
|
|
};
|
|
use common_net::msg::{world_msg, WorldMapMsg};
|
|
use rand::Rng;
|
|
use serde::Deserialize;
|
|
use std::time::Duration;
|
|
use vek::*;
|
|
|
|
#[derive(Debug)]
|
|
pub enum Error {
|
|
Other(String),
|
|
}
|
|
|
|
pub struct World {
|
|
sim: sim::WorldSim,
|
|
civs: civ::Civs,
|
|
}
|
|
|
|
#[derive(Deserialize)]
|
|
pub struct Colors {
|
|
pub deep_stone_color: (u8, u8, u8),
|
|
pub block: block::Colors,
|
|
pub column: column::Colors,
|
|
pub layer: layer::Colors,
|
|
pub site: site::Colors,
|
|
}
|
|
|
|
impl assets::Asset for Colors {
|
|
type Loader = assets::RonLoader;
|
|
|
|
const EXTENSION: &'static str = "ron";
|
|
}
|
|
|
|
impl World {
|
|
pub fn generate(
|
|
seed: u32,
|
|
opts: sim::WorldOpts,
|
|
threadpool: &rayon::ThreadPool,
|
|
) -> (Self, IndexOwned) {
|
|
// NOTE: Generating index first in order to quickly fail if the color manifest
|
|
// is broken.
|
|
threadpool.install(|| {
|
|
let mut index = Index::new(seed);
|
|
|
|
let mut sim = sim::WorldSim::generate(seed, opts, threadpool);
|
|
|
|
let civs = civ::Civs::generate(seed, &mut sim, &mut index);
|
|
|
|
sim2::simulate(&mut index, &mut sim);
|
|
|
|
(Self { sim, civs }, IndexOwned::new(index))
|
|
})
|
|
}
|
|
|
|
pub fn sim(&self) -> &sim::WorldSim { &self.sim }
|
|
|
|
pub fn civs(&self) -> &civ::Civs { &self.civs }
|
|
|
|
pub fn tick(&self, _dt: Duration) {
|
|
// TODO
|
|
}
|
|
|
|
pub fn get_map_data(&self, index: IndexRef, threadpool: &rayon::ThreadPool) -> WorldMapMsg {
|
|
threadpool.install(|| {
|
|
// we need these numbers to create unique ids for cave ends
|
|
let num_sites = self.civs().sites().count() as u64;
|
|
let num_caves = self.civs().caves.values().count() as u64;
|
|
WorldMapMsg {
|
|
pois: self.civs().pois.iter().map(|(_, poi)| {
|
|
world_msg::PoiInfo {
|
|
name: poi.name.clone(),
|
|
kind: match &poi.kind {
|
|
civ::PoiKind::Peak(alt) => world_msg::PoiKind::Peak(*alt),
|
|
civ::PoiKind::Lake(size) => world_msg::PoiKind::Lake(*size),
|
|
},
|
|
wpos: poi.loc * TerrainChunkSize::RECT_SIZE.map(|e| e as i32),
|
|
}
|
|
}).collect(),
|
|
sites: self
|
|
.civs()
|
|
.sites
|
|
.iter()
|
|
.map(|(_, site)| {
|
|
world_msg::SiteInfo {
|
|
id: site.site_tmp.map(|i| i.id()).unwrap_or_default(),
|
|
name: site.site_tmp.map(|id| index.sites[id].name().to_string()),
|
|
// TODO: Probably unify these, at some point
|
|
kind: match &site.kind {
|
|
civ::SiteKind::Settlement => world_msg::SiteKind::Town,
|
|
civ::SiteKind::Dungeon => world_msg::SiteKind::Dungeon {
|
|
difficulty: match site.site_tmp.map(|id| &index.sites[id].kind) {
|
|
Some(site::SiteKind::Dungeon(d)) => d.difficulty(),
|
|
_ => 0,
|
|
},
|
|
},
|
|
civ::SiteKind::Castle => world_msg::SiteKind::Castle,
|
|
civ::SiteKind::Refactor => world_msg::SiteKind::Town,
|
|
civ::SiteKind::Tree => world_msg::SiteKind::Tree,
|
|
},
|
|
wpos: site.center * TerrainChunkSize::RECT_SIZE.map(|e| e as i32),
|
|
}
|
|
})
|
|
.chain(
|
|
self.civs()
|
|
.caves
|
|
.iter()
|
|
.map(|(id, info)| {
|
|
// separate the two locations, combine with name
|
|
std::iter::once((id.id() + num_sites, info.name.clone(), info.location.0))
|
|
// unfortunately we have to introduce a fake id (as it gets stored in a map in the client)
|
|
.chain(std::iter::once((id.id() + num_sites + num_caves, info.name.clone(), info.location.1)))
|
|
})
|
|
.flatten() // unwrap inner iteration
|
|
.map(|(id, name, pos)| world_msg::SiteInfo {
|
|
id,
|
|
name: Some(name),
|
|
kind: world_msg::SiteKind::Cave,
|
|
wpos: pos,
|
|
}),
|
|
)
|
|
.collect(),
|
|
..self.sim.get_map(index)
|
|
}
|
|
})
|
|
}
|
|
|
|
pub fn sample_columns(
|
|
&self,
|
|
) -> impl Sampler<Index = (Vec2<i32>, IndexRef), Sample = Option<ColumnSample>> + '_ {
|
|
ColumnGen::new(&self.sim)
|
|
}
|
|
|
|
pub fn sample_blocks(&self) -> BlockGen { BlockGen::new(ColumnGen::new(&self.sim)) }
|
|
|
|
pub fn find_lowest_accessible_pos(&self, index: IndexRef, chunk_pos: Vec2<i32>) -> Vec3<f32> {
|
|
// Calculate the middle of the chunk in the world
|
|
let spawn_wpos = TerrainChunkSize::center_wpos(chunk_pos);
|
|
|
|
// Unwrapping because generate_chunk only returns err when should_continue evals
|
|
// to true
|
|
let (tc, _cs) = self
|
|
.generate_chunk(index, chunk_pos, || false, None)
|
|
.unwrap();
|
|
let min_z = tc.get_min_z();
|
|
let max_z = tc.get_max_z();
|
|
|
|
let pos = Vec3::new(spawn_wpos.x, spawn_wpos.y, min_z);
|
|
(0..(max_z - min_z))
|
|
.map(|z_diff| pos + Vec3::unit_z() * z_diff)
|
|
.find(|test_pos| {
|
|
let chunk_relative_xy = test_pos
|
|
.xy()
|
|
.map2(TerrainChunkSize::RECT_SIZE, |e, sz| e.rem_euclid(sz as i32));
|
|
tc.get(
|
|
Vec3::new(chunk_relative_xy.x, chunk_relative_xy.y, test_pos.z)
|
|
- Vec3::unit_z(),
|
|
)
|
|
.map_or(false, |b| b.is_filled())
|
|
&& (0..3).all(|z| {
|
|
tc.get(
|
|
Vec3::new(chunk_relative_xy.x, chunk_relative_xy.y, test_pos.z)
|
|
+ Vec3::unit_z() * z,
|
|
)
|
|
.map_or(true, |b| !b.is_solid())
|
|
})
|
|
})
|
|
.unwrap_or(pos)
|
|
.map(|e| e as f32)
|
|
+ 0.5
|
|
}
|
|
|
|
#[allow(clippy::or_fun_call)] // TODO: Pending review in #587
|
|
#[allow(clippy::eval_order_dependence)]
|
|
#[allow(clippy::result_unit_err)]
|
|
pub fn generate_chunk(
|
|
&self,
|
|
index: IndexRef,
|
|
chunk_pos: Vec2<i32>,
|
|
// TODO: misleading name
|
|
mut should_continue: impl FnMut() -> bool,
|
|
time: Option<TimeOfDay>,
|
|
) -> Result<(TerrainChunk, ChunkSupplement), ()> {
|
|
let mut sampler = self.sample_blocks();
|
|
|
|
let chunk_wpos2d = chunk_pos * TerrainChunkSize::RECT_SIZE.map(|e| e as i32);
|
|
let chunk_center_wpos2d = chunk_wpos2d + TerrainChunkSize::RECT_SIZE.map(|e| e as i32 / 2);
|
|
let grid_border = 4;
|
|
let zcache_grid = Grid::populate_from(
|
|
TerrainChunkSize::RECT_SIZE.map(|e| e as i32) + grid_border * 2,
|
|
|offs| sampler.get_z_cache(chunk_wpos2d - grid_border + offs, index),
|
|
);
|
|
|
|
let air = Block::air(SpriteKind::Empty);
|
|
let stone = Block::new(
|
|
BlockKind::Rock,
|
|
zcache_grid
|
|
.get(grid_border + TerrainChunkSize::RECT_SIZE.map(|e| e as i32) / 2)
|
|
.and_then(|zcache| zcache.as_ref())
|
|
.map(|zcache| zcache.sample.stone_col)
|
|
.unwrap_or(index.colors.deep_stone_color.into()),
|
|
);
|
|
let water = Block::new(BlockKind::Water, Rgb::zero());
|
|
|
|
let (base_z, sim_chunk) = match self
|
|
.sim
|
|
/*.get_interpolated(
|
|
chunk_pos.map2(chunk_size2d, |e, sz: u32| e * sz as i32 + sz as i32 / 2),
|
|
|chunk| chunk.get_base_z(),
|
|
)
|
|
.and_then(|base_z| self.sim.get(chunk_pos).map(|sim_chunk| (base_z, sim_chunk))) */
|
|
.get_base_z(chunk_pos)
|
|
{
|
|
Some(base_z) => (base_z as i32, self.sim.get(chunk_pos).unwrap()),
|
|
// Some((base_z, sim_chunk)) => (base_z as i32, sim_chunk),
|
|
None => {
|
|
return Ok((
|
|
TerrainChunk::new(
|
|
CONFIG.sea_level as i32,
|
|
water,
|
|
air,
|
|
TerrainChunkMeta::void(),
|
|
),
|
|
ChunkSupplement::default(),
|
|
));
|
|
},
|
|
};
|
|
|
|
let meta = TerrainChunkMeta::new(
|
|
sim_chunk
|
|
.sites
|
|
.iter()
|
|
.filter(|id| {
|
|
index.sites[**id]
|
|
.get_origin()
|
|
.distance_squared(chunk_center_wpos2d) as f32
|
|
<= index.sites[**id].radius().powi(2)
|
|
})
|
|
.min_by_key(|id| {
|
|
index.sites[**id]
|
|
.get_origin()
|
|
.distance_squared(chunk_center_wpos2d)
|
|
})
|
|
.map(|id| index.sites[*id].name().to_string()),
|
|
sim_chunk.get_biome(),
|
|
sim_chunk.alt,
|
|
sim_chunk.tree_density,
|
|
sim_chunk.cave.1.alt != 0.0,
|
|
sim_chunk.river.is_river(),
|
|
sim_chunk.temp,
|
|
sim_chunk
|
|
.sites
|
|
.iter()
|
|
.any(|site| matches!(index.sites[*site].kind, SiteKind::Settlement(_))),
|
|
sim_chunk
|
|
.sites
|
|
.iter()
|
|
.any(|site| matches!(index.sites[*site].kind, SiteKind::Dungeon(_))),
|
|
);
|
|
|
|
let mut chunk = TerrainChunk::new(base_z, stone, air, meta);
|
|
|
|
for y in 0..TerrainChunkSize::RECT_SIZE.y as i32 {
|
|
for x in 0..TerrainChunkSize::RECT_SIZE.x as i32 {
|
|
if should_continue() {
|
|
return Err(());
|
|
};
|
|
|
|
let offs = Vec2::new(x, y);
|
|
|
|
let z_cache = match zcache_grid.get(grid_border + offs) {
|
|
Some(Some(z_cache)) => z_cache,
|
|
_ => continue,
|
|
};
|
|
|
|
let (min_z, max_z) = z_cache.get_z_limits();
|
|
|
|
(base_z..min_z as i32).for_each(|z| {
|
|
let _ = chunk.set(Vec3::new(x, y, z), stone);
|
|
});
|
|
|
|
(min_z as i32..max_z as i32).for_each(|z| {
|
|
let lpos = Vec3::new(x, y, z);
|
|
let wpos = Vec3::from(chunk_wpos2d) + lpos;
|
|
|
|
if let Some(block) = sampler.get_with_z_cache(wpos, Some(&z_cache)) {
|
|
let _ = chunk.set(lpos, block);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
let sample_get = |offs| {
|
|
zcache_grid
|
|
.get(grid_border + offs)
|
|
.map(Option::as_ref)
|
|
.flatten()
|
|
.map(|zc| &zc.sample)
|
|
};
|
|
|
|
// Only use for rng affecting dynamic elements like chests and entities!
|
|
let mut dynamic_rng = rand::thread_rng();
|
|
|
|
// Apply layers (paths, caves, etc.)
|
|
let mut canvas = Canvas {
|
|
info: CanvasInfo {
|
|
wpos: chunk_pos * TerrainChunkSize::RECT_SIZE.map(|e| e as i32),
|
|
column_grid: &zcache_grid,
|
|
column_grid_border: grid_border,
|
|
chunks: &self.sim,
|
|
index,
|
|
chunk: sim_chunk,
|
|
},
|
|
chunk: &mut chunk,
|
|
};
|
|
|
|
layer::apply_caves_to(&mut canvas, &mut dynamic_rng);
|
|
layer::apply_trees_to(&mut canvas, &mut dynamic_rng);
|
|
layer::apply_scatter_to(&mut canvas, &mut dynamic_rng);
|
|
layer::apply_paths_to(&mut canvas);
|
|
// layer::apply_coral_to(&mut canvas);
|
|
|
|
// Apply site generation
|
|
sim_chunk
|
|
.sites
|
|
.iter()
|
|
.for_each(|site| index.sites[*site].apply_to(&mut canvas, &mut dynamic_rng));
|
|
|
|
let gen_entity_pos = |dynamic_rng: &mut rand::rngs::ThreadRng| {
|
|
let lpos2d = TerrainChunkSize::RECT_SIZE
|
|
.map(|sz| dynamic_rng.gen::<u32>().rem_euclid(sz) as i32);
|
|
let mut lpos = Vec3::new(
|
|
lpos2d.x,
|
|
lpos2d.y,
|
|
sample_get(lpos2d).map(|s| s.alt as i32 - 32).unwrap_or(0),
|
|
);
|
|
|
|
while let Some(block) = chunk.get(lpos).ok().copied().filter(Block::is_solid) {
|
|
lpos.z += block.solid_height().ceil() as i32;
|
|
}
|
|
|
|
(Vec3::from(chunk_wpos2d) + lpos).map(|e: i32| e as f32) + 0.5
|
|
};
|
|
|
|
let mut supplement = ChunkSupplement {
|
|
entities: Vec::new(),
|
|
};
|
|
|
|
if sim_chunk.contains_waypoint {
|
|
supplement.add_entity(EntityInfo::at(gen_entity_pos(&mut dynamic_rng)).into_waypoint());
|
|
}
|
|
|
|
// Apply layer supplement
|
|
layer::apply_caves_supplement(
|
|
&mut dynamic_rng,
|
|
chunk_wpos2d,
|
|
sample_get,
|
|
&chunk,
|
|
index,
|
|
&mut supplement,
|
|
);
|
|
|
|
// Apply layer supplement
|
|
layer::wildlife::apply_wildlife_supplement(
|
|
&mut dynamic_rng,
|
|
chunk_wpos2d,
|
|
sample_get,
|
|
&chunk,
|
|
index,
|
|
sim_chunk,
|
|
&mut supplement,
|
|
time,
|
|
);
|
|
|
|
// Apply site supplementary information
|
|
sim_chunk.sites.iter().for_each(|site| {
|
|
index.sites[*site].apply_supplement(
|
|
&mut dynamic_rng,
|
|
chunk_wpos2d,
|
|
sample_get,
|
|
&mut supplement,
|
|
site.id(),
|
|
)
|
|
});
|
|
|
|
// Finally, defragment to minimize space consumption.
|
|
chunk.defragment();
|
|
|
|
Ok((chunk, supplement))
|
|
}
|
|
}
|