DataSize.php

Go to the documentation of this file.

<?php

declare(strict_types=1);

namespace ILIAS\Data;

final class DataSize implements \Stringable
{
    private const SIZE_FACTOR = 1000;
    private const PRECISION = 2;

    public const Byte = 1;

    // binary
    public const KiB = 1024;
    public const MiB = 1_048_576; // pow(1024, 2)
    public const GiB = 1_073_741_824;
    public const TiB = 1_099_511_627_776;

    // decimal
    public const KB = 1000; // kilobyte
    public const MB = 1_000_000; // megabyte
    public const GB = 1_000_000_000; // gigabyte
    public const TB = 1_000_000_000_000; // terabyte

    public static array $abbreviations = [
        'B' => self::Byte,

        'KB' => self::KB,
        'K' => self::KiB,
        'k' => self::KiB,
        'KiB' => self::KiB,

        'MB' => self::MB,
        'M' => self::MiB,
        'm' => self::MiB,
        'MiB' => self::MiB,

        'GB' => self::GB,
        'G' => self::GiB,
        'g' => self::GiB,
        'GiB' => self::GiB,

        'TB' => self::TB,
        'TiB' => self::TiB,
    ];

    private float $size;
    private int $unit;
    private string $suffix;

    public function __construct(int $size, int $unit)
    {
        $this->suffix = $this->mapUnitToSuffix($unit);
        $this->size = (float) $size / (float) $unit; // the div operation can return int and float
        $this->unit = $unit;
    }

    private function mapUnitToSuffix(int $unit): string
    {
        return match ($unit) {
            self::Byte => 'B',
            self::KiB => 'KiB',
            self::MiB => 'MiB',
            self::GiB => 'GiB',
            self::TiB => 'TiB',
            self::KB => 'KB',
            self::MB => 'MB',
            self::GB => 'GB',
            self::TB => 'TB',
            default => throw new \InvalidArgumentException('The given data size unit is not valid, please check the provided class constants of the DataSize class.')
        };
    }

    public function getSize(): float
    {
        return $this->size;
    }

    public function getUnit(): int
    {
        return $this->unit;
    }

    public function inBytes(): float
    {
        return $this->size * $this->unit;
    }

    public function __toString(): string
    {
        $size = $this->inBytes();
        $unit = match (true) {
            $size > self::SIZE_FACTOR * self::SIZE_FACTOR * self::SIZE_FACTOR * self::SIZE_FACTOR => DataSize::TB,
            $size > self::SIZE_FACTOR * self::SIZE_FACTOR * self::SIZE_FACTOR => DataSize::GB,
            $size > self::SIZE_FACTOR * self::SIZE_FACTOR => DataSize::MB,
            $size > self::SIZE_FACTOR => DataSize::KB,
            default => DataSize::Byte,
        };

        $size = round($size / (float) $unit, self::PRECISION);

        return "$size " . $this->mapUnitToSuffix($unit);
    }
}