Merge branch 'newtree'
[Faustine.git] / architecture / morph.lib
1 import("fft.lib");
2 import("fft2d.lib");
3
4 DILATION_MIN = 0;
5 EROSION_MAX = 255;
6
7 strel_shift_dilation = DILATION_MIN : vectorize(1);
8 strel_shift_erosion = EROSION_MAX : vectorize(1);
9
10 spray_by_three(n) = _ <: par(i, n, ([i], [i+1], [i+2]));
11
12 tri_max = max, _ : max;
13 tri_maxs(n) = par(i, n, tri_max);
14 tri_min = min, _ : min;
15 tri_mins(n) = par(i, n, tri_min);
16
17 dilating(n) = strel_shift_dilation, _, strel_shift_dilation : #, _ : # : spray_by_three(n) : tri_maxs(n) : nconcat(n);
18 eroding(n) = strel_shift_erosion, _, strel_shift_erosion : #, _ : # : spray_by_three(n) : tri_mins(n) : nconcat(n);
19
20 //dilation(size, iter) = seq(i, iter, dilating(size));
21 //erosion(size, iter) = seq(i, iter, eroding(size));
22
23 dilation_square(x, y) = _ : vectorize(y) : serialize : dilating(x) : vectorize(y) : matrix_transpose(y, x) : serialize : dilating(y) : vectorize(x) : matrix_transpose(x, y) : serialize;
24
25 erosion_square(x, y) = _ : vectorize(y) : serialize : eroding(x) : vectorize(y) : matrix_transpose(y, x) : serialize : eroding(y) : vectorize(x) : matrix_transpose(x, y) : serialize;
26
27 dilations(x, y, iter) = seq(i, iter, dilation_square(x,y));
28 erosions(x, y, iter) = seq(i, iter, erosion_square(x,y));
29
30 open(x, y, iter) = erosions(x, y, iter) : dilations(x, y, iter);
31 close(x, y, iter) = dilations(x, y, iter) : erosions(x, y, iter);
32
33 //process = dilations(112, 3);
34 //process = erosions(119, 10);
35 //process = open(119, 123, 10);
36 //process = close(119, 123, 1);
37
38