about summary refs log tree commit diff
path: root/src/tools/dilate.c
blob: f53e76560e2b5c6dd110fb34c3b1a03064de4248 (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
96
97
98

#include "../engine/client/pnglite/pnglite.c"

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

static int clamp(int v, int min, int max)
{
	if(v > max)
		return max;
	if(v < min)
		return min;
	return v;
}

static void dilate(int w, int h, pixel *src, pixel *dst)
{
	int x, y, k, m, c;
	int ix, iy;

	m = 0;
	for(y = 0; y < h; y++)
	{
		for(x = 0; x < w; x++, m++)
		{
			dst[m] = src[m];
			if(src[m].a)
				continue;
			
			const int xo[] = {0, -1, 1, 0};
			const int yo[] = {-1, 0, 0, 1};
			for(c = 0; c < 4; c++)
			{
				ix = clamp(x + xo[c], 0, w-1);
				iy = clamp(y + yo[c], 0, h-1);
				k = iy*w+ix;
				if(src[k].a)
				{
					dst[m] = src[k];
					dst[m].a = 255;
					break;
				}
			}
		}
	}
}

static void copy_alpha(int w, int h, pixel *src, pixel *dst)
{
	int x, y, m;

	for(y = 0; y < h; y++)
		for(x = 0; x < w; x++, m++)
			dst[m].a = src[m].a;
}

int main(int argc, char **argv)
{
	png_t png;
	pixel *buffer[3] = {0,0,0};
	int i, w, h;
	
	png_init(0,0);
	png_open_file(&png, argv[1]);
	
	if(png.color_type != PNG_TRUECOLOR_ALPHA)
	{
		printf("not an RGBA image\n");
		return -1;
	}
	
	buffer[0] = (pixel*)malloc(png.width*png.height*sizeof(pixel));
	buffer[1] = (pixel*)malloc(png.width*png.height*sizeof(pixel));
	buffer[2] = (pixel*)malloc(png.width*png.height*sizeof(pixel));
	png_get_data(&png, (unsigned char *)buffer[0]);
	png_close_file(&png);
	
	w = png.width;
	h = png.height;
	
	dilate(w, h, buffer[0], buffer[1]);
	for(i = 0; i < 5; i++)
	{
		dilate(w, h, buffer[1], buffer[2]);
		dilate(w, h, buffer[2], buffer[1]);
	}
	
	copy_alpha(w, h, buffer[0], buffer[1]);
	
	// save here
	png_open_file_write(&png, argv[1]);
	png_set_data(&png, w, h, 8, PNG_TRUECOLOR_ALPHA, (unsigned char *)buffer[1]);
	png_close_file(&png);
	
	return 0;
}