minedmap: move inputs to processing steps from run() to new()

This commit is contained in:
Matthias Schiffer 2023-08-15 18:45:38 +02:00
parent 6a82fcc9b4
commit 722fe00d77
Signed by: neocturne
GPG key ID: 16EF3F64CB201D9C
4 changed files with 26 additions and 18 deletions

View file

@ -48,9 +48,9 @@ fn main() -> Result<()> {
.unwrap();
let regions = RegionProcessor::new(&config).run()?;
TileRenderer::new(&config, &rt).run(&regions)?;
let tiles = TileMipmapper::new(&config).run(&regions)?;
MetadataWriter::new(&config).run(tiles)?;
TileRenderer::new(&config, &rt, &regions).run()?;
let tiles = TileMipmapper::new(&config, &regions).run()?;
MetadataWriter::new(&config, &tiles).run()?;
Ok(())
}

View file

@ -4,10 +4,6 @@ use serde::Serialize;
use super::common::*;
pub struct MetadataWriter<'a> {
config: &'a Config,
}
#[derive(Debug, Serialize)]
#[serde(rename_all = "camelCase")]
struct Bounds {
@ -35,9 +31,14 @@ struct Metadata<'t> {
spawn: Spawn,
}
pub struct MetadataWriter<'a> {
config: &'a Config,
tiles: &'a [TileCoordMap],
}
impl<'a> MetadataWriter<'a> {
pub fn new(config: &'a Config) -> Self {
MetadataWriter { config }
pub fn new(config: &'a Config, tiles: &'a [TileCoordMap]) -> Self {
MetadataWriter { config, tiles }
}
fn mipmap_entry(regions: &TileCoordMap) -> Mipmap {
@ -87,7 +88,7 @@ impl<'a> MetadataWriter<'a> {
}
}
pub fn run(&self, tiles: Vec<TileCoordMap>) -> Result<()> {
pub fn run(self) -> Result<()> {
let level_dat = self.read_level_dat()?;
let mut metadata = Metadata {
@ -95,7 +96,7 @@ impl<'a> MetadataWriter<'a> {
spawn: Self::spawn(&level_dat),
};
for tile_map in tiles.iter() {
for tile_map in self.tiles.iter() {
metadata.mipmaps.push(Self::mipmap_entry(tile_map));
}

View file

@ -7,11 +7,12 @@ use super::common::*;
pub struct TileMipmapper<'a> {
config: &'a Config,
regions: &'a [TileCoords],
}
impl<'a> TileMipmapper<'a> {
pub fn new(config: &'a Config) -> Self {
TileMipmapper { config }
pub fn new(config: &'a Config, regions: &'a [TileCoords]) -> Self {
TileMipmapper { config, regions }
}
fn done(tiles: &TileCoordMap) -> bool {
@ -129,11 +130,11 @@ impl<'a> TileMipmapper<'a> {
})
}
pub fn run(self, tiles: &[TileCoords]) -> Result<Vec<TileCoordMap>> {
pub fn run(self) -> Result<Vec<TileCoordMap>> {
let mut tile_stack = {
let mut tile_map = TileCoordMap::default();
for &TileCoords { x, z } in tiles {
for &TileCoords { x, z } in self.regions {
tile_map.0.entry(z).or_default().insert(x);
}

View file

@ -50,17 +50,23 @@ fn biome_at(
pub struct TileRenderer<'a> {
config: &'a Config,
rt: &'a tokio::runtime::Runtime,
regions: &'a [TileCoords],
region_cache: Mutex<LruCache<PathBuf, Arc<OnceCell<RegionRef>>>>,
}
impl<'a> TileRenderer<'a> {
pub fn new(config: &'a Config, rt: &'a tokio::runtime::Runtime) -> Self {
pub fn new(
config: &'a Config,
rt: &'a tokio::runtime::Runtime,
regions: &'a [TileCoords],
) -> Self {
let region_cache = Mutex::new(LruCache::new(
NonZeroUsize::new(6 + 6 * config.num_threads).unwrap(),
));
TileRenderer {
config,
rt,
regions,
region_cache,
}
}
@ -261,11 +267,11 @@ impl<'a> TileRenderer<'a> {
)
}
pub fn run(self, regions: &[TileCoords]) -> Result<()> {
pub fn run(self) -> Result<()> {
fs::create_dir_all(&self.config.tile_dir(TileKind::Map, 0))?;
// Use par_bridge to process items in order (for better use of region cache)
regions.iter().par_bridge().try_for_each(|&coords| {
self.regions.iter().par_bridge().try_for_each(|&coords| {
self.render_tile(coords)
.with_context(|| format!("Failed to render tile {:?}", coords))
})?;