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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#include <cvd/image_io.h>
#include <cvd/draw.h>
#include <cvd/morphology.h>
#include <cvd/byte.h>
#include <gvars3/instances.h>
#include <tag/printf.h>
#include <climits>
using namespace CVD;
using namespace std;
using namespace tag;
using namespace GVars3;
int main(int argc, char** argv)
{
GUI.parseArguments(argc, argv);
float dilate = 0;
int size = GV3::get<int>("size", 0, -1);
Image<CVD::byte> im = img_load(GV3::get<string>("image"));
int d = ceil(dilate);
vector<ImageRef> disc = getDisc(dilate);
Image<CVD::byte> mask(im.size()), filter(im.size()), final(im.size());
final.zero();
int n=0;
string mask_str = GV3::get<string>("mask", "", 1);
if(mask_str != "")
{
dilate = GV3::get<float>("radius", 0, -1);
}
string filt_str = GV3::get<string>("filter", "", -1);
int best_cells_filled = INT_MAX, bestxoff=0, bestyoff = 0;
int cells_00=-1;
//Find the optiml offset
for(int yoff=0; yoff < size; yoff++)
for(int xoff=0; xoff < size; xoff++)
{
int cells_filled=0;
for(int y=yoff; y < im.size().y-size; y += size)
for(int x=xoff; x < im.size().x-size; x += size)
{
if(x-d >= 0 && x+d < im.size().x && y-d >= 0 && y+d < im.size().y)
{
for(int i=0; i < size; i++)
for(int j=0; j < size; j++)
{
if(im[y+j][x+i])
{
cells_filled++;
goto cont;
}
}
}
cont:;
}
//clog << "Testing " << xoff << ", " << yoff << " " << cells_filled << "\n";
if(cells_00 == -1)
cells_00 = cells_filled;
if(cells_filled < best_cells_filled)
{
best_cells_filled = cells_filled;
bestxoff = xoff;
bestyoff = yoff;
}
}
clog << "Using " << best_cells_filled << ", " << bestyoff << " Cells= " << best_cells_filled << " ( default=" << cells_00 << ") " << endl;
for(int y=bestyoff; y < im.size().y-size; y += size)
for(int x=bestxoff; x < im.size().x-size; x += size)
if(x-d >= 0 && x+d < im.size().x && y-d >= 0 && y+d < im.size().y)
{
filter.zero();
mask.zero();
bool use=0;
for(int i=0; i < size; i++)
for(int j=0; j < size; j++)
{
if(im[y+j][x+i])
use=1;
filter[y+j][x+i] = 255;
}
if(use)
{
for(int i=0; i < size; i++)
for(int j=0; j < size; j++)
final[y+j][x+i] = 255;
img_save(filter, sPrintf(filt_str, n));
if(mask_str != "")
{
morphology(filter, disc, Morphology::Dilate<byte>(), mask);
img_save(mask, sPrintf(mask_str, n));
}
n++;
}
}
img_save(final, "final_xxxxx.png");
}
|
