Working Threaded Generation

2025-09-07 10:22:56 +00:00 · 2014-06-10 20:45:09 +02:00
parent 392eb74901
commit fa50cbf4b3
56 changed files with 1631 additions and 1733 deletions
--- a/src/pocketmine/level/generator/GenerationRequestManager.php
+++ b/src/pocketmine/level/generator/GenerationRequestManager.php
@ -0,0 +1,140 @@
+<?php
+
+/*
+ *
+ *  ____            _        _   __  __ _                  __  __ ____
+ * |  _ \ ___   ___| | _____| |_|  \/  (_)_ __   ___      |  \/  |  _ \
+ * | |_) / _ \ / __| |/ / _ \ __| |\/| | | '_ \ / _ \_____| |\/| | |_) |
+ * |  __/ (_) | (__|   <  __/ |_| |  | | | | | |  __/_____| |  | |  __/
+ * |_|   \___/ \___|_|\_\___|\__|_|  |_|_|_| |_|\___|     |_|  |_|_|
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * @author PocketMine Team
+ * @link http://www.pocketmine.net/
+ *
+ *
+*/
+
+namespace pocketmine\level\generator;
+
+use pocketmine\level\format\SimpleChunk;
+use pocketmine\level\Level;
+use pocketmine\Server;
+use pocketmine\utils\Binary;
+
+class GenerationRequestManager{
+
+	protected $socket;
+	/** @var Server */
+	protected $server;
+	/** @var GenerationThread */
+	protected $generationThread;
+
+	/**
+	 * @param Server $server
+	 */
+	public function __construct(Server $server){
+		$this->server = $server;
+		$this->generationThread = new GenerationThread($server->getLogger(), $server->getLoader());
+		$this->socket = $this->generationThread->getExternalSocket();
+	}
+
+	/**
+	 * @param Level  $level
+	 * @param string $generator
+	 * @param array  $options
+	 */
+	public function openLevel(Level $level, $generator, array $options = []){
+		$buffer = chr(GenerationManager::PACKET_OPEN_LEVEL) . Binary::writeInt($level->getID()) . Binary::writeInt($level->getSeed()) .
+			Binary::writeShort(strlen($generator)) . $generator . serialize($options);
+
+		@socket_write($this->socket, Binary::writeInt(strlen($buffer)) . $buffer);
+	}
+
+	/**
+	 * @param Level $level
+	 */
+	public function closeLevel(Level $level){
+		$buffer = chr(GenerationManager::PACKET_CLOSE_LEVEL) . Binary::writeInt($level->getID());
+		@socket_write($this->socket, Binary::writeInt(strlen($buffer)) . $buffer);
+	}
+
+	public function addNamespace($namespace, $path){
+		$buffer = chr(GenerationManager::PACKET_ADD_NAMESPACE) . Binary::writeShort(strlen($namespace)) . $namespace . $path;
+		@socket_write($this->socket, Binary::writeInt(strlen($buffer)) . $buffer);
+	}
+
+	protected function socketRead($len){
+		$buffer = "";
+		while(strlen($buffer) < $len){
+			$buffer .= @socket_read($this->socket, $len - strlen($buffer));
+		}
+
+		return $buffer;
+	}
+
+	protected function sendChunk($levelID, SimpleChunk $chunk){
+		$binary = chr(GenerationManager::PACKET_SEND_CHUNK) . Binary::writeInt($levelID) . $chunk->toBinary();
+		@socket_write($this->socket, Binary::writeInt(strlen($binary)) . $binary);
+	}
+
+	public function requestChunk(Level $level, $chunkX, $chunkZ){
+		$buffer = chr(GenerationManager::PACKET_REQUEST_CHUNK) . Binary::writeInt($level->getID()) . Binary::writeInt($chunkX) . Binary::writeInt($chunkZ);
+		@socket_write($this->socket, Binary::writeInt(strlen($buffer)) . $buffer);
+	}
+
+	protected function handleRequest($levelID, $chunkX, $chunkZ){
+		if(($level = $this->server->getLevel($levelID)) instanceof Level){
+			$this->sendChunk($levelID, $level->getChunk($chunkX, $chunkZ, false));
+		}else{
+			$buffer = chr(GenerationManager::PACKET_CLOSE_LEVEL) . Binary::writeInt($levelID);
+			@socket_write($this->socket, Binary::writeInt(strlen($buffer)) . $buffer);
+		}
+	}
+
+	protected function receiveChunk($levelID, SimpleChunk $chunk){
+		if(($level = $this->server->getLevel($levelID)) instanceof Level){
+			$level->setChunk($chunk->getX(), $chunk->getZ(), $chunk);
+		}else{
+			$buffer = chr(GenerationManager::PACKET_CLOSE_LEVEL) . Binary::writeInt($levelID);
+			@socket_write($this->socket, Binary::writeInt(strlen($buffer)) . $buffer);
+		}
+	}
+
+	public function handlePackets(){
+		if(($len = @socket_read($this->socket, 4)) !== false){
+			if(strlen($len) < 4){
+				$len .= $this->socketRead(4 - strlen($len));
+			}
+			$packet = $this->socketRead(Binary::readInt($len));
+			$pid = ord($packet{0});
+			$offset = 1;
+
+			if($pid === GenerationManager::PACKET_REQUEST_CHUNK){
+				$levelID = Binary::readInt(substr($packet, $offset, 4));
+				$offset += 4;
+				$chunkX = Binary::readInt(substr($packet, $offset, 4));
+				$offset += 4;
+				$chunkZ = Binary::readInt(substr($packet, $offset, 4));
+				$this->handleRequest($levelID, $chunkX, $chunkZ);
+			}elseif($pid === GenerationManager::PACKET_SEND_CHUNK){
+				$levelID = Binary::readInt(substr($packet, $offset, 4));
+				$offset += 4;
+				$chunk = SimpleChunk::fromBinary(substr($packet, $offset));
+				$this->receiveChunk($levelID, $chunk);
+			}
+		}
+	}
+
+	public function shutdown(){
+		$buffer = chr(GenerationManager::PACKET_SHUTDOWN);
+		@socket_write($this->socket, Binary::writeInt(strlen($buffer)) . $buffer);
+		$this->generationThread->join();
+	}
+
+
+}