ChunkSerializer.php

<?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/
 *
 *
 */
 
declare(strict_types=1);
 
namespace pocketmine\network\mcpe\serializer;
 
use pmmp\encoding\Byte;
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\VarInt;
use pocketmine\block\tile\Spawnable;
use pocketmine\data\bedrock\BiomeIds;
use pocketmine\data\bedrock\LegacyBiomeIdToStringIdMap;
use pocketmine\nbt\TreeRoot;
use pocketmine\network\mcpe\convert\BlockTranslator;
use pocketmine\network\mcpe\protocol\serializer\NetworkNbtSerializer;
use pocketmine\network\mcpe\protocol\types\DimensionIds;
use pocketmine\world\format\Chunk;
use pocketmine\world\format\PalettedBlockArray;
use pocketmine\world\format\SubChunk;
use function count;
 
final class ChunkSerializer{
    private function __construct(){
        //NOOP
    }
 
    public static function getDimensionChunkBounds(int $dimensionId) : array{
        return match($dimensionId){
            DimensionIds::OVERWORLD => [-4, 19],
            DimensionIds::NETHER => [0, 7],
            DimensionIds::THE_END => [0, 15],
            default => throw new \InvalidArgumentException("Unknown dimension ID $dimensionId"),
        };
    }
 
    public static function getSubChunkCount(Chunk $chunk, int $dimensionId) : int{
        //if the protocol world bounds ever exceed the PM supported bounds again in the future, we might need to
        //polyfill some stuff here
        [$minSubChunkIndex, $maxSubChunkIndex] = self::getDimensionChunkBounds($dimensionId);
        for($y = $maxSubChunkIndex, $count = $maxSubChunkIndex - $minSubChunkIndex + 1; $y >= $minSubChunkIndex; --$y, --$count){
            if($chunk->getSubChunk($y)->isEmptyFast()){
                continue;
            }
            return $count;
        }
 
        return 0;
    }
 
    public static function serializeFullChunk(Chunk $chunk, int $dimensionId, BlockTranslator $blockTranslator, ?string $tiles = null) : string{
        $stream = new ByteBufferWriter();
 
        $subChunkCount = self::getSubChunkCount($chunk, $dimensionId);
        $writtenCount = 0;
 
        [$minSubChunkIndex, $maxSubChunkIndex] = self::getDimensionChunkBounds($dimensionId);
        for($y = $minSubChunkIndex; $writtenCount < $subChunkCount; ++$y, ++$writtenCount){
            self::serializeSubChunk($chunk->getSubChunk($y), $blockTranslator, $stream, false);
        }
 
        $biomeIdMap = LegacyBiomeIdToStringIdMap::getInstance();
        //all biomes must always be written :(
        for($y = $minSubChunkIndex; $y <= $maxSubChunkIndex; ++$y){
            self::serializeBiomePalette($chunk->getSubChunk($y)->getBiomeArray(), $biomeIdMap, $stream);
        }
 
        Byte::writeUnsigned($stream, 0); //border block array count
        //Border block entry format: 1 byte (4 bits X, 4 bits Z). These are however useless since they crash the regular client.
 
        if($tiles !== null){
            $stream->writeByteArray($tiles);
        }else{
            $stream->writeByteArray(self::serializeTiles($chunk));
        }
        return $stream->getData();
    }
 
    public static function serializeSubChunk(SubChunk $subChunk, BlockTranslator $blockTranslator, ByteBufferWriter $stream, bool $persistentBlockStates) : void{
        $layers = $subChunk->getBlockLayers();
        Byte::writeUnsigned($stream, 8); //version
 
        Byte::writeUnsigned($stream, count($layers));
 
        $blockStateDictionary = $blockTranslator->getBlockStateDictionary();
 
        foreach($layers as $blocks){
            $bitsPerBlock = $blocks->getBitsPerBlock();
            $words = $blocks->getWordArray();
            Byte::writeUnsigned($stream, ($bitsPerBlock << 1) | ($persistentBlockStates ? 0 : 1));
            $stream->writeByteArray($words);
            $palette = $blocks->getPalette();
 
            if($bitsPerBlock !== 0){
                VarInt::writeSignedInt($stream, count($palette)); //yes, this is intentionally zigzag
            }
            if($persistentBlockStates){
                $nbtSerializer = new NetworkNbtSerializer();
                foreach($palette as $p){
                    //TODO: introduce a binary cache for this
                    $state = $blockStateDictionary->generateDataFromStateId($blockTranslator->internalIdToNetworkId($p));
                    if($state === null){
                        $state = $blockTranslator->getFallbackStateData();
                    }
 
                    $stream->writeByteArray($nbtSerializer->write(new TreeRoot($state->toNbt())));
                }
            }else{
                //we would use writeSignedIntArray() here, but the gains of writing in batch are negated by the cost of
                //allocating a temporary array for the mapped palette IDs, especially for small palettes
                foreach($palette as $p){
                    VarInt::writeSignedInt($stream, $blockTranslator->internalIdToNetworkId($p));
                }
            }
        }
    }
 
    private static function serializeBiomePalette(PalettedBlockArray $biomePalette, LegacyBiomeIdToStringIdMap $biomeIdMap, ByteBufferWriter $stream) : void{
        $biomePaletteBitsPerBlock = $biomePalette->getBitsPerBlock();
        Byte::writeUnsigned($stream, ($biomePaletteBitsPerBlock << 1) | 1); //the last bit is non-persistence (like for blocks), though it has no effect on biomes since they always use integer IDs
        $stream->writeByteArray($biomePalette->getWordArray());
 
        $biomePaletteArray = $biomePalette->getPalette();
        if($biomePaletteBitsPerBlock !== 0){
            VarInt::writeSignedInt($stream, count($biomePaletteArray));
        }
 
        foreach($biomePaletteArray as $p){
            //we would use writeSignedIntArray() here, but the gains of writing in batch are negated by the cost of
            //allocating a temporary array for the mapped palette IDs, especially for small palettes
            VarInt::writeSignedInt($stream, $biomeIdMap->legacyToString($p) !== null ? $p : BiomeIds::OCEAN);
        }
    }
 
    public static function serializeTiles(Chunk $chunk) : string{
        $stream = new ByteBufferWriter();
        foreach($chunk->getTiles() as $tile){
            if($tile instanceof Spawnable){
                $stream->writeByteArray($tile->getSerializedSpawnCompound()->getEncodedNbt());
            }
        }
 
        return $stream->getData();
    }
}