master ohmu/fastfetch-openbsd / src / common / impl / size.c
ohmu  ·  2026-10-15
 1#include "common/size.h"
 2
 3#include <inttypes.h>
 4
 5static void appendNum(FFstrbuf* result, uint64_t bytes, uint32_t base, const char** prefixes) {
 6    const FFOptionsDisplay* options = &instance.config.display;
 7    double size = (double) bytes;
 8    uint8_t counter = 0;
 9
10    while (size >= base && counter < options->sizeMaxPrefix && prefixes[counter + 1]) {
11        size /= base;
12        counter++;
13    }
14
15    if (counter == 0) {
16        ffStrbufAppendUInt(result, bytes);
17    } else {
18        ffStrbufAppendDouble(result, size, (int8_t) options->sizeNdigits, true);
19    }
20    if (options->sizeSpaceBeforeUnit != FF_SPACE_BEFORE_UNIT_NEVER) {
21        ffStrbufAppendC(result, ' ');
22    }
23    ffStrbufAppendS(result, prefixes[counter]);
24}
25
26void ffSizeAppendNum(uint64_t bytes, FFstrbuf* result) {
27    const FFOptionsDisplay* options = &instance.config.display;
28    switch (options->sizeBinaryPrefix) {
29        case FF_SIZE_BINARY_PREFIX_TYPE_IEC:
30            appendNum(result, bytes, 1024, (const char*[]) { "B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB", "YiB", nullptr });
31            break;
32        case FF_SIZE_BINARY_PREFIX_TYPE_SI:
33            appendNum(result, bytes, 1000, (const char*[]) { "B", "kB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB", nullptr });
34            break;
35        case FF_SIZE_BINARY_PREFIX_TYPE_JEDEC:
36            appendNum(result, bytes, 1024, (const char*[]) { "B", "KB", "MB", "GB", "TB", nullptr });
37            break;
38        default:
39            appendNum(result, bytes, 1024, (const char*[]) { "B", nullptr });
40            break;
41    }
42}