BlueMap/BlueMapCore/src/main/java/de/bluecolored/bluemap/core/render/hires/HiresModelManager.java

/*
 * This file is part of BlueMap, licensed under the MIT License (MIT).
 *
 * Copyright (c) Blue (Lukas Rieger) <https://bluecolored.de>
 * Copyright (c) contributors
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package de.bluecolored.bluemap.core.render.hires;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;
import java.util.concurrent.ExecutorService;
import java.util.zip.GZIPOutputStream;

import com.flowpowered.math.vector.Vector2i;
import com.flowpowered.math.vector.Vector3d;
import com.flowpowered.math.vector.Vector3i;

import de.bluecolored.bluemap.core.logger.Logger;
import de.bluecolored.bluemap.core.render.RenderSettings;
import de.bluecolored.bluemap.core.render.WorldTile;
import de.bluecolored.bluemap.core.resourcepack.ResourcePack;
import de.bluecolored.bluemap.core.util.AABB;
import de.bluecolored.bluemap.core.util.FileUtil;
import de.bluecolored.bluemap.core.world.ChunkNotGeneratedException;

public class HiresModelManager {

	private Path fileRoot;
	private HiresModelRenderer renderer;
	
	private Vector2i tileSize;
	private Vector2i gridOrigin;
	
	private ExecutorService savingExecutor;
	
	public HiresModelManager(Path fileRoot, ResourcePack resourcePack, Vector2i tileSize, ExecutorService savingExecutor) {
		this(fileRoot, new HiresModelRenderer(resourcePack), tileSize, new Vector2i(2, 2), savingExecutor);
	}
	
	public HiresModelManager(Path fileRoot, HiresModelRenderer renderer, Vector2i tileSize, Vector2i gridOrigin, ExecutorService savingExecutor) {
		this.fileRoot = fileRoot;
		this.renderer = renderer;
		
		this.tileSize = tileSize;
		this.gridOrigin = gridOrigin;
		
		this.savingExecutor = savingExecutor;
	}
	
	/**
	 * Renders the given world tile with the provided render-settings
	 * @throws ChunkNotGeneratedException if a minecraft-chunk needed for thies tile is not yet generated
	 */
	public HiresModel render(WorldTile tile, RenderSettings renderSettings) throws ChunkNotGeneratedException {
		HiresModel model = renderer.render(tile, getTileRegion(tile), renderSettings);
		save(model);
		return model;
	}
	
	private void save(final HiresModel model) {
		final String modelJson = model.toBufferGeometry().toJson();
		savingExecutor.submit(() -> save(model, modelJson));
	}
	
	private void save(HiresModel model, String modelJson){
		File file = getFile(model.getTile());
		
		try {
			if (!file.exists()){
				file.getParentFile().mkdirs();
				file.createNewFile();
			}
	
			FileOutputStream fos = new FileOutputStream(file);
			GZIPOutputStream zos = new GZIPOutputStream(fos); 
			OutputStreamWriter osw = new OutputStreamWriter(zos, StandardCharsets.UTF_8);
			try (
				PrintWriter pw = new PrintWriter(osw);
			){
				pw.print(modelJson);
			}
			
			//logger.logDebug("Saved hires model: " + model.getTile()); 
		} catch (IOException e){
			Logger.global.logError("Failed to save hires model: " + file, e);
		}
	}
	
	/**
	 * Returns the region of blocks that a tile includes
	 */
	public AABB getTileRegion(WorldTile tile) {
		Vector3i min = new Vector3i(
				tile.getTile().getX() * tileSize.getX() + gridOrigin.getX(), 
				tile.getWorld().getBoundaries().getMin().getFloorY(), 
				tile.getTile().getY() * tileSize.getY() + gridOrigin.getY()
			);
		Vector3i max = min.add(
				tileSize.getX() - 1,
				tile.getWorld().getBoundaries().getMax().getFloorY(),
				tileSize.getY() - 1
			);
		return new AABB(min, max);
	}
	
	/**
	 * Returns the tile-size
	 */
	public Vector2i getTileSize() {
		return tileSize;
	}

	/**
	 * Returns the grid-origin
	 */
	public Vector2i getGridOrigin() {
		return gridOrigin;
	}
	
	/**
	 * Converts a block-position to a map-tile-coordinate
	 */
	public Vector2i posToTile(Vector3i pos){
		return posToTile(pos.toDouble());
	}

	/**
	 * Converts a block-position to a map-tile-coordinate
	 */
	public Vector2i posToTile(Vector3d pos){
		pos = pos.sub(new Vector3d(gridOrigin.getX(), 0.0, gridOrigin.getY()));
		return Vector2i.from(
				(int) Math.floor(pos.getX() / (double) getTileSize().getX()),
				(int) Math.floor(pos.getZ() / (double) getTileSize().getY())  
			);
	}
	
	/**
	 * Returns the file for a tile
	 */
	public File getFile(Vector2i tilePos){
		return FileUtil.coordsToFile(fileRoot, tilePos, "json.gz");
	}
	
}