IntSaveIdMapTrait.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\bedrock;
 
use function array_key_exists;
use function spl_object_id;
 
trait IntSaveIdMapTrait{
 
    private array $idToEnum = [];
 
    private array $enumToId = [];
 
    protected function getRuntimeId(object $enum) : int{
        //this is fine for enums and non-cloning object registries
        return spl_object_id($enum);
    }
 
    public function register(int $saveId, object $enum) : void{
        $this->idToEnum[$saveId] = $enum;
        $this->enumToId[$this->getRuntimeId($enum)] = $saveId;
    }
 
    public function fromId(int $id) : ?object{
        //we might not have all the effect IDs registered
        return $this->idToEnum[$id] ?? null;
    }
 
    public function toId(object $enum) : int{
        $runtimeId = $this->getRuntimeId($enum);
        if(!array_key_exists($runtimeId, $this->enumToId)){
            //this should never happen, so we treat it as an exceptional condition
            throw new \InvalidArgumentException("Object does not have a mapped save ID");
        }
        return $this->enumToId[$runtimeId];
    }
}