PocketMine-MP/src/data/runtime/RuntimeDataWriter.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\data\runtime;

use pocketmine\block\utils\BrewingStandSlot;
use pocketmine\block\utils\WallConnectionType;
use pocketmine\math\Axis;
use pocketmine\math\Facing;
use pocketmine\utils\AssumptionFailedError;
use function array_flip;

final class RuntimeDataWriter implements RuntimeDataDescriber{
	use RuntimeEnumSerializerTrait;

	private int $value = 0;
	private int $offset = 0;

	public function __construct(
		private int $maxBits
	){}

	public function writeInt(int $bits, int $value) : void{
		if($this->offset + $bits > $this->maxBits){
			throw new \InvalidArgumentException("Bit buffer cannot be larger than $this->maxBits bits (already have $this->offset bits)");
		}
		if(($value & (~0 << $bits)) !== 0){
			throw new \InvalidArgumentException("Value $value does not fit into $bits bits");
		}

		$this->value |= ($value << $this->offset);
		$this->offset += $bits;
	}

	public function int(int $bits, int &$value) : void{
		$this->writeInt($bits, $value);
	}

	protected function writeBoundedInt(int $bits, int $min, int $max, int $value) : void{
		if($value < $min || $value > $max){
			throw new \InvalidArgumentException("Value $value is outside the range $min - $max");
		}
		$this->writeInt($bits, $value - $min);
	}

	public function boundedInt(int $bits, int $min, int $max, int &$value) : void{
		$this->writeBoundedInt($bits, $min, $max, $value);
	}

	protected function writeBool(bool $value) : void{
		$this->writeInt(1, $value ? 1 : 0);
	}

	public function bool(bool &$value) : void{
		$this->writeBool($value);
	}

	public function horizontalFacing(int &$facing) : void{
		$this->writeInt(2, match($facing){
			Facing::NORTH => 0,
			Facing::EAST => 1,
			Facing::SOUTH => 2,
			Facing::WEST => 3,
			default => throw new \InvalidArgumentException("Invalid horizontal facing $facing")
		});
	}

	/**
	 * @param int[] $faces
	 */
	public function facingFlags(array &$faces) : void{
		$uniqueFaces = array_flip($faces);
		foreach(Facing::ALL as $facing){
			$this->writeBool(isset($uniqueFaces[$facing]));
		}
	}

	/**
	 * @param int[] $faces
	 */
	public function horizontalFacingFlags(array &$faces) : void{
		$uniqueFaces = array_flip($faces);
		foreach(Facing::HORIZONTAL as $facing){
			$this->writeBool(isset($uniqueFaces[$facing]));
		}
	}

	public function facing(int &$facing) : void{
		$this->writeInt(3, match($facing){
			0 => Facing::DOWN,
			1 => Facing::UP,
			2 => Facing::NORTH,
			3 => Facing::SOUTH,
			4 => Facing::WEST,
			5 => Facing::EAST,
			default => throw new \InvalidArgumentException("Invalid facing $facing")
		});
	}

	public function facingExcept(int &$facing, int $except) : void{
		$this->facing($facing);
	}

	public function axis(int &$axis) : void{
		$this->writeInt(2, match($axis){
			Axis::X => 0,
			Axis::Z => 1,
			Axis::Y => 2,
			default => throw new \InvalidArgumentException("Invalid axis $axis")
		});
	}

	public function horizontalAxis(int &$axis) : void{
		$this->writeInt(1, match($axis){
			Axis::X => 0,
			Axis::Z => 1,
			default => throw new \InvalidArgumentException("Invalid horizontal axis $axis")
		});
	}

	/**
	 * @param WallConnectionType[] $connections
	 * @phpstan-param array<Facing::NORTH|Facing::EAST|Facing::SOUTH|Facing::WEST, WallConnectionType> $connections
	 */
	public function wallConnections(array &$connections) : void{
		$packed = 0;
		$offset = 0;
		foreach(Facing::HORIZONTAL as $facing){
			$packed += match($connections[$facing] ?? null){
				null => 0,
				WallConnectionType::SHORT() => 1,
				WallConnectionType::TALL() => 2,
				default => throw new AssumptionFailedError("Unreachable")
			} * (3 ** $offset);
			$offset++;
		}
		$this->writeBoundedInt(7, 0, (3 ** 4) - 1, $packed);
	}

	/**
	 * @param BrewingStandSlot[] $slots
	 * @phpstan-param array<int, BrewingStandSlot> $slots
	 */
	public function brewingStandSlots(array &$slots) : void{
		foreach([
			BrewingStandSlot::EAST(),
			BrewingStandSlot::NORTHWEST(),
			BrewingStandSlot::SOUTHWEST(),
		] as $member){
			$this->writeBool(isset($slots[$member->id()]));
		}
	}

	public function railShape(int &$railShape) : void{
		$this->int(4, $railShape);
	}

	public function straightOnlyRailShape(int &$railShape) : void{
		$this->int(3, $railShape);
	}

	public function getValue() : int{ return $this->value; }

	public function getOffset() : int{ return $this->offset; }
}