PocketMine-MP/src/block/Candle.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\utils\CandleTrait;
use pocketmine\block\utils\SupportType;
use pocketmine\data\runtime\RuntimeDataDescriber;
use pocketmine\item\Item;
use pocketmine\math\Axis;
use pocketmine\math\AxisAlignedBB;
use pocketmine\math\Facing;
use pocketmine\math\Vector3;
use pocketmine\player\Player;
use pocketmine\utils\AssumptionFailedError;
use pocketmine\world\BlockTransaction;

class Candle extends Transparent{
	use CandleTrait {
		describeBlockOnlyState as encodeLitState;
		getLightLevel as getBaseLightLevel;
	}

	public const MIN_COUNT = 1;
	public const MAX_COUNT = 4;

	private int $count = self::MIN_COUNT;

	protected function describeBlockOnlyState(RuntimeDataDescriber $w) : void{
		$this->encodeLitState($w);
		$w->boundedIntAuto(self::MIN_COUNT, self::MAX_COUNT, $this->count);
	}

	public function getCount() : int{ return $this->count; }

	/** @return $this */
	public function setCount(int $count) : self{
		if($count < self::MIN_COUNT || $count > self::MAX_COUNT){
			throw new \InvalidArgumentException("Count must be in range " . self::MIN_COUNT . " ... " . self::MAX_COUNT);
		}
		$this->count = $count;
		return $this;
	}

	public function getLightLevel() : int{
		return $this->getBaseLightLevel() * $this->count;
	}

	protected function recalculateCollisionBoxes() : array{
		return [
			(match($this->count){
				1 => AxisAlignedBB::one()
					->squash(Axis::X, 7 / 16)
					->squash(Axis::Z, 7 / 16),
				2 => AxisAlignedBB::one()
					->squash(Axis::X, 5 / 16)
					->trim(Facing::NORTH, 7 / 16) //0.3 thick on the Z axis
					->trim(Facing::SOUTH, 6 / 16),
				3 => AxisAlignedBB::one()
					->trim(Facing::WEST, 5 / 16)
					->trim(Facing::EAST, 6 / 16)
					->trim(Facing::NORTH, 6 / 16)
					->trim(Facing::SOUTH, 5 / 16),
				4 => AxisAlignedBB::one()
					->squash(Axis::X, 5 / 16)
					->trim(Facing::NORTH, 5 / 16)
					->trim(Facing::SOUTH, 6 / 16),
				default => throw new AssumptionFailedError("Unreachable")
			})->trim(Facing::UP, 10 / 16)
		];
	}

	public function getSupportType(int $facing) : SupportType{
		return SupportType::NONE;
	}

	protected function getCandleIfCompatibleType(Block $block) : ?Candle{
		return $block instanceof Candle && $block->hasSameTypeId($this) ? $block : null;
	}

	public function canBePlacedAt(Block $blockReplace, Vector3 $clickVector, int $face, bool $isClickedBlock) : bool{
		$candle = $this->getCandleIfCompatibleType($blockReplace);
		return $candle !== null ? $candle->count < self::MAX_COUNT : parent::canBePlacedAt($blockReplace, $clickVector, $face, $isClickedBlock);
	}

	public function place(BlockTransaction $tx, Item $item, Block $blockReplace, Block $blockClicked, int $face, Vector3 $clickVector, ?Player $player = null) : bool{
		if(!$blockReplace->getAdjacentSupportType(Facing::DOWN)->hasCenterSupport()){
			return false;
		}
		$existing = $this->getCandleIfCompatibleType($blockReplace);
		if($existing !== null){
			if($existing->count >= self::MAX_COUNT){
				return false;
			}

			$this->count = $existing->count + 1;
			$this->lit = $existing->lit;
		}
		return parent::place($tx, $item, $blockReplace, $blockClicked, $face, $clickVector, $player);
	}

	public function getDropsForCompatibleTool(Item $item) : array{
		return [$this->asItem()->setCount($this->count)];
	}
}