FortuneDropHelper.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\utils;
 
use pocketmine\item\enchantment\VanillaEnchantments;
use pocketmine\item\Item;
use function max;
use function min;
use function mt_getrandmax;
use function mt_rand;
 
final class FortuneDropHelper{
    public static function weighted(Item $usedItem, int $min, int $maxBase) : int{
        if($maxBase < $min){
            throw new \InvalidArgumentException("Maximum drop amount must be greater than or equal to minimum drop amount");
        }
 
        $fortuneLevel = $usedItem->getEnchantmentLevel(VanillaEnchantments::FORTUNE());
 
        return mt_rand($min,
            $fortuneLevel > 0 && mt_rand() / mt_getrandmax() > 2 / ($fortuneLevel + 2) ?
                $maxBase * ($fortuneLevel + 1) :
                $maxBase
        );
    }
 
    public static function binomial(Item $usedItem, int $min, int $minRolls = 3, float $chance = 4 / 7) : int{
        $fortuneLevel = $usedItem->getEnchantmentLevel(VanillaEnchantments::FORTUNE());
 
        $count = $min;
        $rolls = $minRolls + $fortuneLevel;
        for($i = 0; $i < $rolls; ++$i){
            if(mt_rand() / mt_getrandmax() < $chance){
                ++$count;
            }
        }
 
        return $count;
    }
 
    public static function discrete(Item $usedItem, int $min, int $maxBase) : int{
        if($maxBase < $min){
            throw new \InvalidArgumentException("Minimum base drop amount must be less than or equal to maximum base drop amount");
        }
 
        $max = $maxBase + $usedItem->getEnchantmentLevel(VanillaEnchantments::FORTUNE());
        return mt_rand($min, $max);
    }
 
    public static function bonusChanceDivisor(Item $usedItem, int $divisorBase, int $divisorSubtractPerLevel) : bool{
        $fortuneLevel = $usedItem->getEnchantmentLevel(VanillaEnchantments::FORTUNE());
        return mt_rand(1, max(1, $divisorBase - ($fortuneLevel * $divisorSubtractPerLevel))) === 1;
    }
 
    public static function bonusChanceFixed(Item $usedItem, float $chanceBase, float $addedChancePerLevel) : bool{
        $fortuneLevel = $usedItem->getEnchantmentLevel(VanillaEnchantments::FORTUNE());
        $chance = min(1, $chanceBase + ($fortuneLevel * $addedChancePerLevel));
        return mt_rand() / mt_getrandmax() < $chance;
    }
}