#include <stdio.h>
#include <string.h>
#include "utils.h"

//
// bintostr
//

int bintostr(char *buf, uint64_t n) { // Lazy code 2.0, sorry
	float f = n;
	char *fs;
	//TODO: base-10 {kB|MB|GB|TB|PB}
	if (f >= TiB) {
		fs = "%.2f TiB";
		f /= TiB;
	} else if (f >= GiB) {
		fs = "%.2f GiB";
		f /= GiB;
	} else if (f >= MiB) {
		fs = "%.1f MiB";
		f /= MiB;
	} else if (f >= KiB) {
		fs = "%.1f KiB";
		f /= KiB;
	} else
		fs = "%g B";
	return snprintf(buf, BINSTR_LENGTH, fs, f);
}

//
// strtobin
//

int strtobin(uint64_t *size, const oschar *input) {
	float f;
	char c;
#ifdef _WIN32
	if (swscanf(input, L"%f%c", &f, &c) != 2) {
		return 1;
	}
#else
	if (sscanf(input, "%f%c", &f, &c) != 2) {
		return 1;
	}
#endif
	if (f <= 0) return 2;
	uint64_t u;
	switch (c) {
	case 'T': case 't':
		u = f * 1099511627776; break;
	case 'G': case 'g':
		u = f * 1073741824; break;
	case 'M': case 'm':
		u = f * 1048576; break;
	case 'K': case 'k':
		u = f * 1024; break;
	case 'B': case 'b':
		u = f; break;
	default: return 3;
	}
	*size = u;
	return 0;
}

//
// bswap16
//

uint16_t bswap16(uint16_t v) {
	return v >> 8 | v << 8;
}

//
// bswap32
//

uint32_t bswap32(uint32_t v) {
	v = (v >> 16) | (v << 16);
	return ((v & 0xFF00FF00) >> 8) | ((v & 0x00FF00FF) << 8);
}

//
// bswap64
//

uint64_t bswap64(uint64_t v) {
	v = (v >> 32) | (v << 32);
	v = ((v & 0xFFFF0000FFFF0000) >> 16) | ((v & 0x0000FFFF0000FFFF) << 16);
	return ((v & 0xFF00FF00FF00FF00) >> 8) | ((v & 0x00FF00FF00FF00FF) << 8);
}

//
// print_array
//

void print_array(char *p, uint8_t *a, size_t s) {
	size_t i = 0;
	putout(p);
	while (--s)
		printf(" %02X", a[i++]);
	putchar('\n');
}

//
// extcmp
//

int extcmp(const oschar *s1, const oschar *s2) {
#ifdef _WIN32
	wchar_t *ext = wcsrchr(s1, '.');
	if (ext == NULL) // Not found
		return 0;
	if (*++ext == 0) // Return if NULL after '.'
		return 0;
	//TODO: Lowercase s1
	return wcscmp(ext, s2) == 0;
#else
	const char *ext = strrchr(s1, '.');
	if (ext == NULL) // Not found
		return 0;
	if (*++ext == 0) // Return if NULL after '.'
		return 0;
	//TODO: Lowercase s1
	return strcmp(ext, s2) == 0;
#endif
}

//
// putout
//

void putout(const char *s) {
	fputs(s, stdout);
}

//
// pow2
//

uint32_t pow2(uint32_t n) {
	return (n & (n - 1)) == 0;
}

//
// pow264
//

uint64_t pow264(uint64_t n) {
	return (n & (n - 1)) == 0;
}

//
// fpow2
//

uint32_t fpow2(uint32_t n) {
	// Adapted from VBox/Storage/VDI.cpp getPowerOfTwo
	if (n == 0)
		return 0;
	uint32_t p = 0;
	while ((n & 1) == 0) {
		n >>= 1;
		++p;
	}
	return n == 1 ? p : 0;
}

//
// fpow264
//

uint64_t fpow264(uint64_t n) {
	if (n == 0)
		return 0;
	uint64_t p = 0;
	while ((n & 1) == 0) {
		n >>= 1;
		++p;
	}
	return n == 1 ? p : 0;
}

//
// str_s
//

void str_s(char *str, int size) {
	size_t i = 0;
	while (--size > 0 && str[i] != 0) {
		if (str[i] < 0x20 || str[i] > 0x7e)
			str[i] = ' ';
	}
}

//
// wstra
//

int wstra(char *dest, char16 *src, int nchars) {
	char c = src[0];
	if (c == 0) {
		strcpy(dest, "<NO NAME>");
		return -1;
	}
	size_t bi = 0; // buffer index
	--nchars; // to include null byte later on
	while (bi < nchars && (c = src[bi])) {
		dest[bi] = c < 0x20 || c > 0x7E ? '?' : c;
		++bi;
	}
	dest[bi] = 0;
	return (int)bi;
}