80 lines
2.0 KiB
Rust
80 lines
2.0 KiB
Rust
use super::Command;
|
|
|
|
use crate::spatial::{Vec3, Area, area_rel_block_overlap, area_contains_block};
|
|
use crate::instance::{ArgType, InstBundle};
|
|
use crate::map_block::{MapBlock};
|
|
use crate::block_utils::clean_name_id_map;
|
|
use crate::utils::{query_keys, fmt_big_num};
|
|
|
|
|
|
fn fill_area(block: &mut MapBlock, area: Area, id: u16) {
|
|
let nd = block.node_data.get_mut();
|
|
for z in area.min.z ..= area.max.z {
|
|
let z_start = z * 256;
|
|
for y in area.min.y ..= area.max.y {
|
|
let zy_start = z_start + y * 16;
|
|
for x in area.min.x ..= area.max.x {
|
|
nd.nodes[(zy_start + x) as usize] = id;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
fn fill(inst: &mut InstBundle) {
|
|
let area = inst.args.area.unwrap();
|
|
let node = inst.args.new_node.as_ref().unwrap().as_bytes().to_owned();
|
|
|
|
let keys = query_keys(&mut inst.db, &mut inst.status,
|
|
None, Some(area), false, true);
|
|
|
|
inst.status.begin_editing();
|
|
|
|
let mut count: u64 = 0;
|
|
for key in keys {
|
|
let pos = Vec3::from_block_key(key);
|
|
let data = inst.db.get_block(key).unwrap();
|
|
let mut block = MapBlock::deserialize(&data).unwrap();
|
|
|
|
if area_contains_block(&area, pos) {
|
|
let nd = block.node_data.get_mut();
|
|
for x in &mut nd.nodes {
|
|
*x = 0;
|
|
}
|
|
block.nimap.map.clear();
|
|
block.nimap.insert(0, &node);
|
|
count += nd.nodes.len() as u64;
|
|
} else {
|
|
let slice = area_rel_block_overlap(&area, pos).unwrap();
|
|
let fill_id = block.nimap.get_id(&node).unwrap_or_else(|| {
|
|
let next = block.nimap.get_max_id().unwrap() + 1;
|
|
block.nimap.insert(next, &node);
|
|
next
|
|
});
|
|
fill_area(&mut block, slice, fill_id);
|
|
clean_name_id_map(&mut block);
|
|
count += slice.volume();
|
|
}
|
|
|
|
inst.db.set_block(key, &block.serialize()).unwrap();
|
|
inst.status.inc_done();
|
|
}
|
|
|
|
inst.status.end_editing();
|
|
inst.status.log_info(
|
|
format!("{} nodes filled.", fmt_big_num(count)).as_str());
|
|
}
|
|
|
|
|
|
pub fn get_command() -> Command {
|
|
Command {
|
|
func: fill,
|
|
verify_args: None,
|
|
args: vec![
|
|
(ArgType::Area(true), "Area to fill"),
|
|
(ArgType::NewNode(true), "Node to fill area with")
|
|
],
|
|
help: "Fill the entire area with one node."
|
|
}
|
|
}
|