ワールド生成時にどのくらいチャンクを生成するかを指定できるように

Author: Nerahikada

plugin.yml

@@ -1,5 +1,5 @@
 name: MiniatureWorld
-version: 0.0.11
+version: 0.0.12
 main: NeiroNetwork\MiniatureWorld\Main
 api: 4.0.0
 depend:

src/NeiroNetwork/MiniatureWorld/ui/form/CreateWorldForm.php

@@ -5,7 +5,9 @@
 namespace NeiroNetwork\MiniatureWorld\ui\form;
 
 use dktapps\pmforms\CustomFormResponse;
+use dktapps\pmforms\element\Slider;
 use NeiroNetwork\MiniatureWorld\ui\invmenu\WorldListViewer;
+use NeiroNetwork\MiniatureWorld\world\WorldGenerator;
 use pocketmine\nbt\tag\CompoundTag;
 use pocketmine\player\Player;
 use pocketmine\world\generator\FlatGeneratorOptions;
@@ -17,12 +19,19 @@ class CreateWorldForm extends BaseWorldForm{
 
 	public function __construct(){
 		parent::__construct("新規ワールド作成");
+
+		// HACK & FIXME
+		$property = (new \ReflectionClass($this))->getParentClass()->getParentClass()->getProperty("elements");
+		$property->setAccessible(true);
+		$property->setValue($this, array_merge($this->getAllElements(), [
+			new Slider("chunk_generation", "生成するチャンク数", 2, 64, 1, 8)
+		]));
 	}
 
 	protected function onSubmit(Player $player, CustomFormResponse $response) : void{
 		$validatedResponse = $this->validateResponse($player, $response);
 		if($validatedResponse === false) return;
-		[$name, $type, $presets] = $validatedResponse;
+		[$name, $type, $presets, $radius] = $validatedResponse;
 
 		$options = WorldCreationOptions::create()
 			->setGeneratorClass(GeneratorManager::getInstance()->getGenerator($type)->getGeneratorClass())
@@ -34,12 +43,12 @@ protected function onSubmit(Player $player, CustomFormResponse $response) : void
 			$options->setSpawnPosition($spawn);
 		}
 
-		if(!$player->getServer()->getWorldManager()->generateWorld($name, $options)){
+		if(!WorldGenerator::generate($name, $options, $radius, fn() => $player->sendMessage("ワールド「{$name}」のチャンク生成が完了しました"))){
 			$this->resendForm($player, $response, "ワールド「{$name}」はすでに存在しています");
 			return;
 		}
 
-		$player->sendMessage("ワールド「{$name}」を作成しました");
+		$player->sendMessage("ワールド「{$name}」を作成します");
 
 		$world = $player->getServer()->getWorldManager()->getWorldByName($name);
 		$world->setTime(World::TIME_NOON);
@@ -53,4 +62,10 @@ protected function onSubmit(Player $player, CustomFormResponse $response) : void
 	protected function onClose(Player $player) : void{
 		(new WorldListViewer())->sendTo($player);
 	}
+
+	protected function validateResponse(Player $player, CustomFormResponse $response) : bool|array{
+		if(is_bool($result = parent::validateResponse($player, $response))) return $result;
+		$result[] = (int) $response->getFloat("chunk_generation");
+		return $result;
+	}
 }

src/NeiroNetwork/MiniatureWorld/world/WorldGenerator.php

@@ -0,0 +1,81 @@
+<?php
+
+namespace NeiroNetwork\MiniatureWorld\world;
+
+use pocketmine\event\world\WorldInitEvent;
+use pocketmine\event\world\WorldLoadEvent;
+use pocketmine\player\ChunkSelector;
+use pocketmine\Server;
+use pocketmine\world\format\Chunk;
+use pocketmine\world\World;
+use pocketmine\world\WorldCreationOptions;
+use pocketmine\world\WorldManager;
+
+final class WorldGenerator{
+
+	/**
+	 * @see WorldManager::generateWorld()
+	 */
+	public static function generate(string $name, WorldCreationOptions $options, int $chunkRadius = 8, ?\Closure $onCompletion = null) : bool{
+		$manager = ($server = Server::getInstance())->getWorldManager();
+
+		if(trim($name) === "" || $manager->isWorldGenerated($name)){
+			return false;
+		}
+
+		$providerEntry = $manager->getProviderManager()->getDefault();
+
+		$path = self::getWorldPath($name);
+		$providerEntry->generate($path, $name, $options);
+
+		$world = new World($server, $name, $providerEntry->fromPath($path), $server->getAsyncPool());
+		self::addWorldToManager($world);
+
+		$world->setAutoSave($manager->getAutoSave());
+
+		(new WorldInitEvent($world))->call();
+
+		(new WorldLoadEvent($world))->call();
+
+		$spawnLocation = $world->getSpawnLocation();
+		$centerX = $spawnLocation->getFloorX() >> Chunk::COORD_BIT_SIZE;
+		$centerZ = $spawnLocation->getFloorZ() >> Chunk::COORD_BIT_SIZE;
+
+		$selected = iterator_to_array((new ChunkSelector())->selectChunks($chunkRadius - 1, $centerX, $centerZ));
+		$done = 0;
+		$total = count($selected);
+		foreach($selected as $index){
+			World::getXZ($index, $chunkX, $chunkZ);
+			$world->orderChunkPopulation($chunkX, $chunkZ, null)->onCompletion(
+				static function() use (&$done, $total, $onCompletion) : void{
+					if(++$done === $total) $onCompletion();
+				},
+				static function() : void{}
+			);
+		}
+
+		return true;
+	}
+
+	/**
+	 * @see WorldManager::getWorldPath()
+	 */
+	private static function getWorldPath(string $name) : string{
+		$manager = Server::getInstance()->getWorldManager();
+		$method = (new \ReflectionClass($manager))->getMethod("getWorldPath");
+		$method->setAccessible(true);
+		return $method->invokeArgs($manager, [$name]);
+	}
+
+	private static function addWorldToManager(World $world) : void{
+		$manager = Server::getInstance()->getWorldManager();
+
+		$property = (new \ReflectionClass($manager))->getProperty("worlds");
+		$property->setAccessible(true);
+
+		$worlds = $property->getValue($manager);
+		$worlds[$world->getId()] = $world;
+
+		$property->setValue($manager, $worlds);
+	}
+}