CraftingManager.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\crafting;
 
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\VarInt;
use pocketmine\item\Item;
use pocketmine\nbt\LittleEndianNbtSerializer;
use pocketmine\nbt\TreeRoot;
use pocketmine\utils\DestructorCallbackTrait;
use pocketmine\utils\ObjectSet;
use function array_shift;
use function count;
use function implode;
use function ksort;
use function spl_object_id;
use const SORT_STRING;
 
class CraftingManager{
    use DestructorCallbackTrait;
 
    protected array $shapedRecipes = [];
    protected array $shapelessRecipes = [];
 
    private array $craftingRecipeIndex = [];
 
    protected array $furnaceRecipeManagers = [];
 
    protected array $potionTypeRecipes = [];
 
    protected array $potionContainerChangeRecipes = [];
 
    private array $brewingRecipeCache = [];
 
    private ObjectSet $recipeRegisteredCallbacks;
 
    public function __construct(){
        $this->recipeRegisteredCallbacks = new ObjectSet();
        foreach(FurnaceType::cases() as $furnaceType){
            $this->furnaceRecipeManagers[spl_object_id($furnaceType)] = new FurnaceRecipeManager();
        }
 
        $recipeRegisteredCallbacks = $this->recipeRegisteredCallbacks;
        foreach($this->furnaceRecipeManagers as $furnaceRecipeManager){
            $furnaceRecipeManager->getRecipeRegisteredCallbacks()->add(static function(FurnaceRecipe $recipe) use ($recipeRegisteredCallbacks) : void{
                foreach($recipeRegisteredCallbacks as $callback){
                    $callback();
                }
            });
        }
    }
 
    public function getRecipeRegisteredCallbacks() : ObjectSet{ return $this->recipeRegisteredCallbacks; }
 
    private static function hashOutput(Item $output) : string{
        $write = new ByteBufferWriter();
        VarInt::writeSignedInt($write, $output->getStateId());
        //TODO: the NBT serializer allocates its own ByteBufferWriter, we should change the API in the future to
        //allow passing our own to avoid this extra allocation
        $write->writeByteArray((new LittleEndianNbtSerializer())->write(new TreeRoot($output->getNamedTag())));
 
        return $write->getData();
    }
 
    private static function hashOutputs(array $outputs) : string{
        if(count($outputs) === 1){
            return self::hashOutput(array_shift($outputs));
        }
        $unique = [];
        foreach($outputs as $o){
            //count is not written because the outputs might be from multiple repetitions of a single recipe
            //this reduces the accuracy of the hash, but it won't matter in most cases.
            $hash = self::hashOutput($o);
            $unique[$hash] = $hash;
        }
        ksort($unique, SORT_STRING);
        return implode("", $unique);
    }
 
    public function getShapelessRecipes() : array{
        return $this->shapelessRecipes;
    }
 
    public function getShapedRecipes() : array{
        return $this->shapedRecipes;
    }
 
    public function getCraftingRecipeIndex() : array{
        return $this->craftingRecipeIndex;
    }
 
    public function getCraftingRecipeFromIndex(int $index) : ?CraftingRecipe{
        return $this->craftingRecipeIndex[$index] ?? null;
    }
 
    public function getFurnaceRecipeManager(FurnaceType $furnaceType) : FurnaceRecipeManager{
        return $this->furnaceRecipeManagers[spl_object_id($furnaceType)];
    }
 
    public function getPotionTypeRecipes() : array{
        return $this->potionTypeRecipes;
    }
 
    public function getPotionContainerChangeRecipes() : array{
        return $this->potionContainerChangeRecipes;
    }
 
    public function registerShapedRecipe(ShapedRecipe $recipe) : void{
        $this->shapedRecipes[self::hashOutputs($recipe->getResults())][] = $recipe;
        $this->craftingRecipeIndex[] = $recipe;
 
        foreach($this->recipeRegisteredCallbacks as $callback){
            $callback();
        }
    }
 
    public function registerShapelessRecipe(ShapelessRecipe $recipe) : void{
        $this->shapelessRecipes[self::hashOutputs($recipe->getResults())][] = $recipe;
        $this->craftingRecipeIndex[] = $recipe;
 
        foreach($this->recipeRegisteredCallbacks as $callback){
            $callback();
        }
    }
 
    public function registerPotionTypeRecipe(PotionTypeRecipe $recipe) : void{
        $this->potionTypeRecipes[] = $recipe;
 
        foreach($this->recipeRegisteredCallbacks as $callback){
            $callback();
        }
    }
 
    public function registerPotionContainerChangeRecipe(PotionContainerChangeRecipe $recipe) : void{
        $this->potionContainerChangeRecipes[] = $recipe;
 
        foreach($this->recipeRegisteredCallbacks as $callback){
            $callback();
        }
    }
 
    public function matchRecipe(CraftingGrid $grid, array $outputs) : ?CraftingRecipe{
        //TODO: try to match special recipes before anything else (first they need to be implemented!)
 
        $outputHash = self::hashOutputs($outputs);
 
        if(isset($this->shapedRecipes[$outputHash])){
            foreach($this->shapedRecipes[$outputHash] as $recipe){
                if($recipe->matchesCraftingGrid($grid)){
                    return $recipe;
                }
            }
        }
 
        if(isset($this->shapelessRecipes[$outputHash])){
            foreach($this->shapelessRecipes[$outputHash] as $recipe){
                if($recipe->matchesCraftingGrid($grid)){
                    return $recipe;
                }
            }
        }
 
        return null;
    }
 
    public function matchRecipeByOutputs(array $outputs) : \Generator{
        //TODO: try to match special recipes before anything else (first they need to be implemented!)
 
        $outputHash = self::hashOutputs($outputs);
 
        if(isset($this->shapedRecipes[$outputHash])){
            foreach($this->shapedRecipes[$outputHash] as $recipe){
                yield $recipe;
            }
        }
 
        if(isset($this->shapelessRecipes[$outputHash])){
            foreach($this->shapelessRecipes[$outputHash] as $recipe){
                yield $recipe;
            }
        }
    }
 
    public function matchBrewingRecipe(Item $input, Item $ingredient) : ?BrewingRecipe{
        $inputHash = $input->getStateId();
        $ingredientHash = $ingredient->getStateId();
        $cached = $this->brewingRecipeCache[$inputHash][$ingredientHash] ?? null;
        if($cached !== null){
            return $cached;
        }
 
        foreach($this->potionContainerChangeRecipes as $recipe){
            if($recipe->getIngredient()->accepts($ingredient) && $recipe->getResultFor($input) !== null){
                return $this->brewingRecipeCache[$inputHash][$ingredientHash] = $recipe;
            }
        }
 
        foreach($this->potionTypeRecipes as $recipe){
            if($recipe->getIngredient()->accepts($ingredient) && $recipe->getResultFor($input) !== null){
                return $this->brewingRecipeCache[$inputHash][$ingredientHash] = $recipe;
            }
        }
 
        return null;
    }
}