Stem.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\BlockEventHelper;
use pocketmine\block\utils\CropGrowthHelper;
use pocketmine\block\utils\FortuneDropHelper;
use pocketmine\data\runtime\RuntimeDataDescriber;
use pocketmine\item\Item;
use pocketmine\item\VanillaItems;
use pocketmine\math\Facing;
use function array_rand;
 
abstract class Stem extends Crops{
    protected int $facing = Facing::UP;
 
    protected function describeBlockOnlyState(RuntimeDataDescriber $w) : void{
        parent::describeBlockOnlyState($w);
        $w->facingExcept($this->facing, Facing::DOWN);
    }
 
    public function getFacing() : int{ return $this->facing; }
 
    public function setFacing(int $facing) : self{
        if($facing === Facing::DOWN){
            throw new \InvalidArgumentException("DOWN is not a valid facing for this block");
        }
        $this->facing = $facing;
        return $this;
    }
 
    abstract protected function getPlant() : Block;
 
    public function onNearbyBlockChange() : void{
        if($this->facing !== Facing::UP && !$this->getSide($this->facing)->hasSameTypeId($this->getPlant())){
            $this->position->getWorld()->setBlock($this->position, $this->setFacing(Facing::UP));
        }
        parent::onNearbyBlockChange();
    }
 
    public function ticksRandomly() : bool{
        return $this->age < self::MAX_AGE || $this->facing === Facing::UP;
    }
 
    public function onRandomTick() : void{
        if($this->facing === Facing::UP && CropGrowthHelper::canGrow($this)){
            $world = $this->position->getWorld();
            if($this->age < self::MAX_AGE){
                $block = clone $this;
                ++$block->age;
                BlockEventHelper::grow($this, $block, null);
            }else{
                $grow = $this->getPlant();
                foreach(Facing::HORIZONTAL as $side){
                    if($this->getSide($side)->hasSameTypeId($grow)){
                        return;
                    }
                }
 
                $facing = Facing::HORIZONTAL[array_rand(Facing::HORIZONTAL)];
                $side = $this->getSide($facing);
                if($side->getTypeId() === BlockTypeIds::AIR && $side->getSide(Facing::DOWN)->hasTypeTag(BlockTypeTags::DIRT)){
                    if(BlockEventHelper::grow($side, $grow, null)){
                        $this->position->getWorld()->setBlock($this->position, $this->setFacing($facing));
                    }
                }
            }
        }
    }
 
    public function getDropsForCompatibleTool(Item $item) : array{
        //TODO: bit annoying we have to pass an Item instance here
        //this should not be affected by Fortune, but still follows a binomial distribution
        return [
            $this->asItem()->setCount(FortuneDropHelper::binomial(VanillaItems::AIR(), 0, chance: ($this->age + 1) / 15))
        ];
    }
}