RuntimeBlockStateRegistry.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\block;
 
use pocketmine\block\BlockBreakInfo as BreakInfo;
use pocketmine\block\BlockIdentifier as BID;
use pocketmine\utils\AssumptionFailedError;
use pocketmine\utils\SingletonTrait;
use pocketmine\world\light\LightUpdate;
use function min;
 
class RuntimeBlockStateRegistry{
    use SingletonTrait;
 
    private array $fullList = [];
 
    private array $typeIndex = [];
 
    public array $light = [];
    public array $lightFilter = [];
    public array $blocksDirectSkyLight = [];
    public array $blastResistance = [];
 
    public function __construct(){
        foreach(VanillaBlocks::getAll() as $block){
            $this->register($block);
        }
    }
 
    public function register(Block $block) : void{
        $typeId = $block->getTypeId();
 
        if(isset($this->typeIndex[$typeId])){
            throw new \InvalidArgumentException("Block ID $typeId is already used by another block");
        }
 
        $this->typeIndex[$typeId] = clone $block;
 
        foreach($block->generateStatePermutations() as $v){
            $this->fillStaticArrays($v->getStateId(), $v);
        }
    }
 
    private function fillStaticArrays(int $index, Block $block) : void{
        $fullId = $block->getStateId();
        if($index !== $fullId){
            throw new AssumptionFailedError("Cannot fill static arrays for an invalid blockstate");
        }else{
            $this->fullList[$index] = $block;
            $this->blastResistance[$index] = $block->getBreakInfo()->getBlastResistance();
            $this->light[$index] = $block->getLightLevel();
            $this->lightFilter[$index] = min(15, $block->getLightFilter() + LightUpdate::BASE_LIGHT_FILTER);
            if($block->blocksDirectSkyLight()){
                $this->blocksDirectSkyLight[$index] = true;
            }
        }
    }
 
    public function fromStateId(int $stateId) : Block{
        if($stateId < 0){
            throw new \InvalidArgumentException("Block state ID cannot be negative");
        }
        if(isset($this->fullList[$stateId])) { //hot
            $block = clone $this->fullList[$stateId];
        }else{
            $typeId = $stateId >> Block::INTERNAL_STATE_DATA_BITS;
            $stateData = ($stateId ^ $typeId) & Block::INTERNAL_STATE_DATA_MASK;
            $block = new UnknownBlock(new BID($typeId), new BlockTypeInfo(BreakInfo::instant()), $stateData);
        }
 
        return $block;
    }
 
    public function getAllKnownStates() : array{
        return $this->fullList;
    }
}