Back to shtanton's homepage
summaryrefslogtreecommitdiff
path: root/src/img.c
blob: 09f000ba83884b6c79289e6b94be50936d0650dd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#define QOI_IMPLEMENTATION
#include "../lib/qoi/qoi.h"

#include <stdio.h>

typedef struct {
	unsigned char r, g, b, a;
} Color;

int pixels(char *index, char *filename) {
	qoi_desc desc;
	Color *pixels = qoi_read(filename, &desc, 4);

	Color colors[16];
	int colorsLen = 0;
	char *compressed = malloc(desc.width * desc.height);

	for (int i = 0; i < (int) (desc.width * desc.height); i++) {
		char c = -1;
		for (int j = 0; j < colorsLen; j++) {
			if (*((int *) &colors[j]) == *((int *) pixels + i)) {
				c = j;
				break;
			}
		}
		if (c == -1) {
			c = colorsLen;
			if (colorsLen >= 16) {
				return 1;
			}
			colors[colorsLen++] = pixels[i];
		}
		compressed[i] = c;
	}

	printf("const char imagePixels%s[%d * %d] = {\n", index, desc.width, desc.height);
	for (int y = 0; y < (int) desc.height; y++) {
		for (int x = 0; x < (int) desc.width; x++) {
			printf("%d,", compressed[x + y * desc.width]);
		}
		printf("\n");
	}
	printf("};\n");

	return 0;
}

int image(char *index, char *filename) {
	qoi_desc desc;
	Color *pixels = qoi_read(filename, &desc, 4);

	Color colors[16];
	int colorsLen = 0;
	char *compressed = malloc(desc.width * desc.height);

	for (int i = 0; i < (int) (desc.width * desc.height); i++) {
		char c = -1;
		for (int j = 0; j < colorsLen; j++) {
			if (*((int *) &colors[j]) == *((int *) pixels + i)) {
				c = j;
				break;
			}
		}
		if (c == -1) {
			c = colorsLen;
			if (colorsLen >= 16) {
				return 1;
			}
			colors[colorsLen++] = pixels[i];
		}
		compressed[i] = c;
	}

	printf("{\n.width = %d,\n.height = %d,\n.colors = {\n", desc.width, desc.height);
	for (int i = 0; i < colorsLen; i++) {
		printf("\t{%u, %u, %u, %u},\n", colors[i].r, colors[i].g, colors[i].b, colors[i].a);
	}
	printf("},\n.pixels = imagePixels%s,\n},\n", index);

	return 0;
}

int main(int argc, char **argv) {
	if (argc != 4) {
		return 1;
	}

	if (!strcmp(argv[1], "pixels")) {
		return pixels(argv[3], argv[2]);
	} else if (!strcmp(argv[1], "image")) {
		return image(argv[3], argv[2]);
	} else {
		return 1;
	}
}