object/Ore.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\world\generator\object;
 
use pocketmine\math\VectorMath;
use pocketmine\utils\Random;
use pocketmine\world\ChunkManager;
use pocketmine\world\format\SubChunk;
use pocketmine\world\utils\SubChunkExplorer;
use pocketmine\world\utils\SubChunkExplorerStatus;
use pocketmine\world\World;
use function sin;
use const M_PI;
 
class Ore{
    public function __construct(
        private Random $random,
        public OreType $type
    ){}
 
    public function getType() : OreType{
        return $this->type;
    }
 
    public function canPlaceObject(ChunkManager $world, int $x, int $y, int $z) : bool{
        return $world->getBlockAt($x, $y, $z)->hasSameTypeId($this->type->replaces);
    }
 
    public function placeObject(ChunkManager $world, int $x, int $y, int $z) : void{
        $clusterSize = $this->type->clusterSize;
        $angle = $this->random->nextFloat() * M_PI;
        $offset = VectorMath::getDirection2D($angle)->multiply($clusterSize / 8);
        $x1 = $x + 8 + $offset->x;
        $x2 = $x + 8 - $offset->x;
        $z1 = $z + 8 + $offset->y;
        $z2 = $z + 8 - $offset->y;
        $y1 = $y + $this->random->nextBoundedInt(3) + 2;
        $y2 = $y + $this->random->nextBoundedInt(3) + 2;
 
        $explorer = new SubChunkExplorer($world);
 
        $tried = [];
        $replaceableStateIds = [];
        $materialStateId = $this->type->material->getStateId();
        for($count = 0; $count <= $clusterSize; ++$count){
            $seedX = $x1 + ($x2 - $x1) * $count / $clusterSize;
            $seedY = $y1 + ($y2 - $y1) * $count / $clusterSize;
            $seedZ = $z1 + ($z2 - $z1) * $count / $clusterSize;
            $size = ((sin($count * (M_PI / $clusterSize)) + 1) * $this->random->nextFloat() * $clusterSize / 16 + 1) / 2;
 
            $startX = (int) ($seedX - $size);
            $startY = (int) ($seedY - $size);
            $startZ = (int) ($seedZ - $size);
            $endX = (int) ($seedX + $size);
            $endY = (int) ($seedY + $size);
            $endZ = (int) ($seedZ + $size);
 
            for($xx = $startX; $xx <= $endX; ++$xx){
                $sizeX = ($xx + 0.5 - $seedX) / $size;
                $sizeX *= $sizeX;
 
                if($sizeX < 1){
                    for($yy = $startY; $yy <= $endY; ++$yy){
                        $sizeY = ($yy + 0.5 - $seedY) / $size;
                        $sizeY *= $sizeY;
 
                        if($yy > 0 && ($sizeX + $sizeY) < 1){
                            for($zz = $startZ; $zz <= $endZ; ++$zz){
                                $sizeZ = ($zz + 0.5 - $seedZ) / $size;
                                $sizeZ *= $sizeZ;
 
                                if(($sizeX + $sizeY + $sizeZ) < 1){
                                    $hash = World::blockHash($xx, $yy, $zz);
                                    if(isset($tried[$hash])){
                                        continue;
                                    }
                                    $tried[$hash] = true;
                                    if($explorer->moveTo($xx, $yy, $zz) === SubChunkExplorerStatus::INVALID || $explorer->currentSubChunk === null){
                                        throw new \LogicException("Unavailable chunk at block x=$xx, y=$yy, z=$zz");
                                    }
                                    $stateId = $explorer->currentSubChunk->getBlockStateId($xx & SubChunk::COORD_MASK, $yy & SubChunk::COORD_MASK, $zz & SubChunk::COORD_MASK);
                                    $replaceable = $replaceableStateIds[$stateId] ??= $world->getBlockAt($xx, $yy, $zz)->hasSameTypeId($this->type->replaces);
                                    if($replaceable){
                                        $explorer->currentSubChunk->setBlockStateId($xx & SubChunk::COORD_MASK, $yy & SubChunk::COORD_MASK, $zz & SubChunk::COORD_MASK, $materialStateId);
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
    }
}