// SPDX-License-Identifier: GPL-2.0+

#include <stdarg.h>
#include <stdio.h>
#include <math.h>

#include "utils.h"
#include "json_print.h"

char *sprint_size(__u32 sz, char *buf)
{
	long kilo = 1024;
	long mega = kilo * kilo;
	size_t len = SPRINT_BSIZE - 1;
	double tmp = sz;

	if (sz >= mega && fabs(mega * rint(tmp / mega) - sz) < 1024)
		snprintf(buf, len, "%gMb", rint(tmp / mega));
	else if (sz >= kilo && fabs(kilo * rint(tmp / kilo) - sz) < 16)
		snprintf(buf, len, "%gKb", rint(tmp / kilo));
	else
		snprintf(buf, len, "%ub", sz);

	return buf;
}

int print_color_size(enum output_type type, enum color_attr color,
		     const char *key, const char *fmt, __u32 sz)
{
	SPRINT_BUF(buf);

	if (_IS_JSON_CONTEXT(type))
		return print_color_uint(type, color, key, "%u", sz);

	sprint_size(sz, buf);
	return print_color_string(type, color, key, fmt, buf);
}