BiomeDefinitionListPacket.php

<?php
 
/*
 * This file is part of BedrockProtocol.
 * Copyright (C) 2014-2022 PocketMine Team <https://github.com/pmmp/BedrockProtocol>
 *
 * BedrockProtocol 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.
 */
 
declare(strict_types=1);
 
namespace pocketmine\network\mcpe\protocol;
 
use pmmp\encoding\ByteBufferReader;
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\VarInt;
use pocketmine\network\mcpe\protocol\serializer\CommonTypes;
use pocketmine\network\mcpe\protocol\types\biome\BiomeDefinitionData;
use pocketmine\network\mcpe\protocol\types\biome\BiomeDefinitionEntry;
use function array_map;
use function count;
 
class BiomeDefinitionListPacket extends DataPacket implements ClientboundPacket{
    public const NETWORK_ID = ProtocolInfo::BIOME_DEFINITION_LIST_PACKET;
 
    private array $definitionData = [];
    private array $strings = [];
 
    public static function create(array $definitionData, array $strings) : self{
        $result = new self;
        $result->definitionData = $definitionData;
        $result->strings = $strings;
        return $result;
    }
 
    public static function fromDefinitions(array $definitions) : self{
        $stringIndexLookup = [];
        $strings = [];
        $addString = function(string $string) use (&$stringIndexLookup, &$strings) : int{
            if(isset($stringIndexLookup[$string])){
                return $stringIndexLookup[$string];
            }
 
            $stringIndexLookup[$string] = count($stringIndexLookup);
            $strings[] = $string;
            return $stringIndexLookup[$string];
        };
 
        $definitionData = array_map(fn(BiomeDefinitionEntry $entry) => new BiomeDefinitionData(
            $addString($entry->getBiomeName()),
            $entry->getId(),
            $entry->getTemperature(),
            $entry->getDownfall(),
            $entry->getFoliageSnow(),
            $entry->getDepth(),
            $entry->getScale(),
            $entry->getMapWaterColor(),
            $entry->hasRain(),
            $entry->getTags() === null ? null : array_map($addString, $entry->getTags()),
            $entry->getChunkGenData(),
        ), $definitions);
 
        return self::create($definitionData, $strings);
    }
 
    private function locateString(int $index) : string{
        return $this->strings[$index] ?? throw new PacketDecodeException("Unknown string index $index");
    }
 
    public function buildDefinitionsFromData() : array{
        return array_map(fn(BiomeDefinitionData $data) => new BiomeDefinitionEntry(
            $this->locateString($data->getNameIndex()),
            $data->getId(),
            $data->getTemperature(),
            $data->getDownfall(),
            $data->getFoliageSnow(),
            $data->getDepth(),
            $data->getScale(),
            $data->getMapWaterColor(),
            $data->hasRain(),
            ($tagIndexes = $data->getTagIndexes()) === null ? null : array_map($this->locateString(...), $tagIndexes),
            $data->getChunkGenData(),
        ), $this->definitionData);
    }
 
    protected function decodePayload(ByteBufferReader $in) : void{
        for($i = 0, $count = VarInt::readUnsignedInt($in); $i < $count; ++$i){
            $this->definitionData[] = BiomeDefinitionData::read($in);
        }
 
        for($i = 0, $count = VarInt::readUnsignedInt($in); $i < $count; ++$i){
            $this->strings[] = CommonTypes::getString($in);
        }
    }
 
    protected function encodePayload(ByteBufferWriter $out) : void{
        VarInt::writeUnsignedInt($out, count($this->definitionData));
        foreach($this->definitionData as $data){
            $data->write($out);
        }
 
        VarInt::writeUnsignedInt($out, count($this->strings));
        foreach($this->strings as $string){
            CommonTypes::putString($out, $string);
        }
    }
 
    public function handle(PacketHandlerInterface $handler) : bool{
        return $handler->handleBiomeDefinitionList($this);
    }
}