11 Mar
2013
11 Mar
'13
11:16 a.m.
---
scribo/sandbox/icdar_13_table/src/main.cc | 122 +++++++++++++++++++++++++++++
1 files changed, 122 insertions(+), 0 deletions(-)
diff --git a/scribo/sandbox/icdar_13_table/src/main.cc
b/scribo/sandbox/icdar_13_table/src/main.cc
index 2ed125d..e4dbb11 100644
--- a/scribo/sandbox/icdar_13_table/src/main.cc
+++ b/scribo/sandbox/icdar_13_table/src/main.cc
@@ -88,6 +88,94 @@ void get_vertical_lines(scribo::component_set<L>& vlines,
}
}
+unsigned find_left(image2d<bool>& ima)
+{
+ bool found = false;
+
+ def::coord col = geom::min_col(ima);
+ while (col < geom::max_col(ima) && !found)
+ {
+ def::coord row = geom::min_row(ima);
+ while (row < geom::max_row(ima) && !found)
+ {
+ found = opt::at(ima, row, col);
+ ++row;
+ }
+ ++col;
+ }
+
+ return (col - 1);
+}
+
+unsigned find_right(image2d<bool>& ima)
+{
+ bool found = false;
+
+ def::coord col = geom::max_col(ima) - 1;
+ while (col >= geom::min_col(ima) && !found)
+ {
+ def::coord row = geom::min_row(ima);
+ while (row < geom::max_row(ima) && !found)
+ {
+ found = opt::at(ima, row, col);
+ ++row;
+ }
+ --col;
+ }
+
+ return (col + 1);
+
+}
+
+unsigned find_top(image2d<bool>& ima)
+{
+ bool found = false;
+
+ def::coord row = geom::min_row(ima);
+ while (row < geom::max_row(ima) && !found)
+ {
+ def::coord col = geom::min_col(ima);
+ while (col < geom::max_col(ima) && !found)
+ {
+ found = opt::at(ima, row, col);
+ ++col;
+ }
+ ++row;
+ }
+
+ return (row - 1);
+
+}
+
+unsigned find_bottom(image2d<bool>& ima)
+{
+ bool found = false;
+
+ def::coord row = geom::max_row(ima) - 1;
+ while (row >= geom::min_row(ima) && !found)
+ {
+ def::coord col = geom::min_col(ima);
+ while (col < geom::max_col(ima) && !found)
+ {
+ found = opt::at(ima, row, col);
+ ++col;
+ }
+ --row;
+ }
+
+ return (row + 1);
+}
+
+void find_borders(image2d<bool>& ima,
+ unsigned& left, unsigned& right,
+ unsigned& top, unsigned& bottom)
+{
+ left = find_left(ima);
+ right = find_right(ima);
+ top = find_top(ima);
+ bottom = find_bottom(ima);
+}
+
int main(int argc, char** argv)
{
typedef value::label_16 V;
@@ -149,6 +237,39 @@ int main(int argc, char** argv)
// Compose table zones with bin_without_lines_lines
ima_texts = logical::and_(bin_without_lines_denoised, mask);
+ // Isolate texts between tables
+ for (unsigned i = 1; i <= masks.nelements(); ++i)
+ {
+ image2d<bool> table_mask, isolated_text;
+ std::ostringstream path;
+ bool empty = true;
+
+ initialize(table_mask, bin);
+ data::fill(table_mask, false);
+ data::fill((table_mask | masks(i).bbox()).rw(), true);
+
+ isolated_text = logical::and_(bin_without_lines_denoised, table_mask);
+
+ mln_piter_(image2d<bool>) p(isolated_text.domain());
+
+ for_all(p)
+ empty = empty && !(isolated_text(p));
+
+ if (!empty)
+ {
+ path << "output/8_" << i << "_isolated.pbm";
+ io::pbm::save(isolated_text, path.str());
+
+ // Find coordinated
+ unsigned left, right, top, bottom;
+ find_borders(isolated_text, left, right, top, bottom);
+
+ std::cout << "(" << left << "," << top
<< ") ->"
+ << "(" << right << "," <<
bottom << ")" << std::endl;
+ }
+ }
+
+
// Get lines images
hlines_ima = hlines.labeled_image();
vlines_ima = vlines.labeled_image();
@@ -164,6 +285,7 @@ int main(int argc, char** argv)
io::pbm::save(ima_tables, "output/5_tables.pbm");
io::pbm::save(mask, "output/6_mask.pbm");
io::pbm::save(ima_texts, "output/7_texts.pbm");
+ /* Save 8_i_isolated */
end_xml(xml);
--
1.7.2.5