Noise.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\noise;
 
use function array_fill;
use function assert;
 
abstract class Noise{
 
    public static function linearLerp($x, $x1, $x2, $q0, $q1){
        return (($x2 - $x) / ($x2 - $x1)) * $q0 + (($x - $x1) / ($x2 - $x1)) * $q1;
    }
 
    public static function bilinearLerp($x, $y, $q00, $q01, $q10, $q11, $x1, $x2, $y1, $y2){
        $dx1 = (($x2 - $x) / ($x2 - $x1));
        $dx2 = (($x - $x1) / ($x2 - $x1));
 
        return (($y2 - $y) / ($y2 - $y1)) * (
            $dx1 * $q00 + $dx2 * $q10
        ) + (($y - $y1) / ($y2 - $y1)) * (
            $dx1 * $q01 + $dx2 * $q11
        );
    }
 
    public static function trilinearLerp($x, $y, $z, $q000, $q001, $q010, $q011, $q100, $q101, $q110, $q111, $x1, $x2, $y1, $y2, $z1, $z2){
        $dx1 = (($x2 - $x) / ($x2 - $x1));
        $dx2 = (($x - $x1) / ($x2 - $x1));
        $dy1 = (($y2 - $y) / ($y2 - $y1));
        $dy2 = (($y - $y1) / ($y2 - $y1));
 
        return (($z2 - $z) / ($z2 - $z1)) * (
            $dy1 * (
                $dx1 * $q000 + $dx2 * $q100
            ) + $dy2 * (
                $dx1 * $q001 + $dx2 * $q101
            )
        ) + (($z - $z1) / ($z2 - $z1)) * (
            $dy1 * (
                $dx1 * $q010 + $dx2 * $q110
            ) + $dy2 * (
                $dx1 * $q011 + $dx2 * $q111
            )
        );
    }
 
    public function __construct(
        protected int $octaves,
        protected float $persistence,
        protected float $expansion
    ){}
 
    abstract public function getNoise2D($x, $z);
 
    abstract public function getNoise3D($x, $y, $z);
 
    public function noise2D($x, $z, $normalized = false){
        $result = 0;
        $amp = 1;
        $freq = 1;
        $max = 0;
 
        $x *= $this->expansion;
        $z *= $this->expansion;
 
        for($i = 0; $i < $this->octaves; ++$i){
            $result += $this->getNoise2D($x * $freq, $z * $freq) * $amp;
            $max += $amp;
            $freq *= 2;
            $amp *= $this->persistence;
        }
 
        if($normalized === true){
            $result /= $max;
        }
 
        return $result;
    }
 
    public function noise3D($x, $y, $z, $normalized = false){
        $result = 0;
        $amp = 1;
        $freq = 1;
        $max = 0;
 
        $x *= $this->expansion;
        $y *= $this->expansion;
        $z *= $this->expansion;
 
        for($i = 0; $i < $this->octaves; ++$i){
            $result += $this->getNoise3D($x * $freq, $y * $freq, $z * $freq) * $amp;
            $max += $amp;
            $freq *= 2;
            $amp *= $this->persistence;
        }
 
        if($normalized === true){
            $result /= $max;
        }
 
        return $result;
    }
 
    public function getFastNoise1D(int $xSize, int $samplingRate, int $x, int $y, int $z) : \SplFixedArray{
        if($samplingRate === 0){
            throw new \InvalidArgumentException("samplingRate cannot be 0");
        }
        if($xSize % $samplingRate !== 0){
            throw new \InvalidArgumentException("xSize % samplingRate must return 0");
        }
 
        $noiseArray = new \SplFixedArray($xSize + 1);
 
        for($xx = 0; $xx <= $xSize; $xx += $samplingRate){
            $noiseArray[$xx] = $this->noise3D($xx + $x, $y, $z);
        }
 
        for($xx = 0; $xx < $xSize; ++$xx){
            if($xx % $samplingRate !== 0){
                $nx = (int) ($xx / $samplingRate) * $samplingRate;
                $noiseArray[$xx] = self::linearLerp(
                    x: $xx,
                    x1: $nx,
                    x2: $nx + $samplingRate,
                    q0: $noiseArray[$nx],
                    q1: $noiseArray[$nx + $samplingRate]
                );
            }
        }
 
        return $noiseArray;
    }
 
    public function getFastNoise2D(int $xSize, int $zSize, int $samplingRate, int $x, int $y, int $z) : \SplFixedArray{
        assert($samplingRate !== 0, new \InvalidArgumentException("samplingRate cannot be 0"));
 
        assert($xSize % $samplingRate === 0, new \InvalidArgumentException("xSize % samplingRate must return 0"));
        assert($zSize % $samplingRate === 0, new \InvalidArgumentException("zSize % samplingRate must return 0"));
 
        $noiseArray = new \SplFixedArray($xSize + 1);
 
        for($xx = 0; $xx <= $xSize; $xx += $samplingRate){
            $noiseArray[$xx] = new \SplFixedArray($zSize + 1);
            for($zz = 0; $zz <= $zSize; $zz += $samplingRate){
                $noiseArray[$xx][$zz] = $this->noise3D($x + $xx, $y, $z + $zz);
            }
        }
 
        for($xx = 0; $xx < $xSize; ++$xx){
            if($xx % $samplingRate !== 0){
                $noiseArray[$xx] = new \SplFixedArray($zSize + 1);
            }
 
            for($zz = 0; $zz < $zSize; ++$zz){
                if($xx % $samplingRate !== 0 || $zz % $samplingRate !== 0){
                    $nx = (int) ($xx / $samplingRate) * $samplingRate;
                    $nz = (int) ($zz / $samplingRate) * $samplingRate;
                    $noiseArray[$xx][$zz] = Noise::bilinearLerp(
                        x: $xx,
                        y: $zz,
                        q00: $noiseArray[$nx][$nz],
                        q01: $noiseArray[$nx][$nz + $samplingRate],
                        q10: $noiseArray[$nx + $samplingRate][$nz],
                        q11: $noiseArray[$nx + $samplingRate][$nz + $samplingRate],
                        x1: $nx,
                        x2: $nx + $samplingRate,
                        y1: $nz,
                        y2: $nz + $samplingRate
                    );
                }
            }
        }
 
        return $noiseArray;
    }
 
    public function getFastNoise3D(int $xSize, int $ySize, int $zSize, int $xSamplingRate, int $ySamplingRate, int $zSamplingRate, int $x, int $y, int $z) : array{
 
        assert($xSamplingRate !== 0, new \InvalidArgumentException("xSamplingRate cannot be 0"));
        assert($zSamplingRate !== 0, new \InvalidArgumentException("zSamplingRate cannot be 0"));
        assert($ySamplingRate !== 0, new \InvalidArgumentException("ySamplingRate cannot be 0"));
 
        assert($xSize % $xSamplingRate === 0, new \InvalidArgumentException("xSize % xSamplingRate must return 0"));
        assert($zSize % $zSamplingRate === 0, new \InvalidArgumentException("zSize % zSamplingRate must return 0"));
        assert($ySize % $ySamplingRate === 0, new \InvalidArgumentException("ySize % ySamplingRate must return 0"));
 
        $noiseArray = array_fill(0, $xSize + 1, array_fill(0, $zSize + 1, []));
 
        for($xx = 0; $xx <= $xSize; $xx += $xSamplingRate){
            for($zz = 0; $zz <= $zSize; $zz += $zSamplingRate){
                for($yy = 0; $yy <= $ySize; $yy += $ySamplingRate){
                    $noiseArray[$xx][$zz][$yy] = $this->noise3D($x + $xx, $y + $yy, $z + $zz, true);
                }
            }
        }
 
        for($xx = 0; $xx < $xSize; ++$xx){
            $nx = (int) ($xx / $xSamplingRate) * $xSamplingRate;
            $nnx = $nx + $xSamplingRate;
 
            $dx1 = (($nnx - $xx) / ($nnx - $nx));
            $dx2 = (($xx - $nx) / ($nnx - $nx));
 
            for($zz = 0; $zz < $zSize; ++$zz){
                $nz = (int) ($zz / $zSamplingRate) * $zSamplingRate;
                $nnz = $nz + $zSamplingRate;
 
                $dz1 = ($nnz - $zz) / ($nnz - $nz);
                $dz2 = ($zz - $nz) / ($nnz - $nz);
 
                for($yy = 0; $yy < $ySize; ++$yy){
                    if($xx % $xSamplingRate !== 0 || $zz % $zSamplingRate !== 0 || $yy % $ySamplingRate !== 0){
                        $ny = (int) ($yy / $ySamplingRate) * $ySamplingRate;
                        $nny = $ny + $ySamplingRate;
 
                        $dy1 = (($nny - $yy) / ($nny - $ny));
                        $dy2 = (($yy - $ny) / ($nny - $ny));
 
                        $noiseArray[$xx][$zz][$yy] = $dz1 * (
                                $dy1 * (
                                    $dx1 * $noiseArray[$nx][$nz][$ny] + $dx2 * $noiseArray[$nnx][$nz][$ny]
                                ) + $dy2 * (
                                    $dx1 * $noiseArray[$nx][$nz][$nny] + $dx2 * $noiseArray[$nnx][$nz][$nny]
                                )
                            ) + $dz2 * (
                                $dy1 * (
                                    $dx1 * $noiseArray[$nx][$nnz][$ny] + $dx2 * $noiseArray[$nnx][$nnz][$ny]
                                ) + $dy2 * (
                                    $dx1 * $noiseArray[$nx][$nnz][$nny] + $dx2 * $noiseArray[$nnx][$nnz][$nny]
                                )
                            );
                    }
                }
            }
        }
 
        return $noiseArray;
    }
}