$extrastylesheet
Olena  User documentation 2.1
An Image Processing Platform
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Groups Pages
show_text_lines.cc
1 // Copyright (C) 2011 EPITA Research and Development Laboratory (LRDE)
2 //
3 // This file is part of Olena.
4 //
5 // Olena is free software: you can redistribute it and/or modify it under
6 // the terms of the GNU General Public License as published by the Free
7 // Software Foundation, version 2 of the License.
8 //
9 // Olena is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 // General Public License for more details.
13 //
14 // You should have received a copy of the GNU General Public License
15 // along with Olena. If not, see <http://www.gnu.org/licenses/>.
16 //
17 // As a special exception, you may use this file as part of a free
18 // software project without restriction. Specifically, if other files
19 // instantiate templates or use macros or inline functions from this
20 // file, or you compile this file and link it with other files to produce
21 // an executable, this file does not by itself cause the resulting
22 // executable to be covered by the GNU General Public License. This
23 // exception does not however invalidate any other reasons why the
24 // executable file might be covered by the GNU General Public License.
25 
26 #include <iostream>
27 
28 #include <mln/core/image/image2d.hh>
29 #include <mln/io/pbm/load.hh>
30 #include <mln/value/int_u16.hh>
31 #include <mln/draw/box_plain.hh>
32 #include <mln/pw/all.hh>
33 #include <mln/core/image/dmorph/image_if.hh>
34 #include <mln/data/convert.hh>
35 #include <mln/literal/colors.hh>
36 
37 #include <scribo/debug/usage.hh>
38 
39 #include <scribo/core/document.hh>
40 #include <scribo/core/component_set.hh>
41 #include <scribo/core/object_links.hh>
42 #include <scribo/core/object_groups.hh>
43 
44 #include <scribo/text/extract_lines.hh>
45 
46 #include <scribo/io/xml/save.hh>
47 
48 
49 const char *args_desc[][2] =
50 {
51  {0, 0}
52 };
53 
54 
55 
56 int main(int argc, char* argv[])
57 {
58  using namespace scribo;
59  using namespace mln;
60 
61  if (argc != 7)
62  return scribo::debug::usage(argv,
63  "Show text lines",
64  "input.pbm input_seps.pbm out_text_boxes.ppm out_text_boxes.pbm out_text_comps.pbm out_lines.xml",
65  args_desc);
66 
67  mln_trace("main");
68 
69  image2d<bool> input;
70  mln::io::pbm::load(input, argv[1]);
71 
72  image2d<bool> input_seps;
73  mln::io::pbm::load(input_seps, argv[2]);
74 
76  t.start();
77 
79  line_set<L> lines = text::extract_lines(input, c8(), input_seps);
80 
81  t.stop();
82  std::cout << t << "s" << std::endl;
83 
84  // Color image with line bboxes.
85  {
86  image2d<value::rgb8> output = data::convert(value::rgb8(), input);
87  for_all_lines(l, lines)
88  if (lines(l).is_textline())
89  mln::draw::box(output, lines(l).bbox(), literal::red);
90  mln::io::ppm::save(output, argv[3]);
91  }
92 
93  // Binary image with line bboxes.
94  {
95  image2d<bool> output;
96  initialize(output, input);
97  data::fill(output, false);
98  for_all_lines(l, lines)
99  if (lines(l).is_textline())
100  mln::draw::box_plain(output, lines(l).bbox(), true);
101  mln::io::pbm::save(output, argv[4]);
102  }
103 
104  {
105  image2d<bool> output;
106  initialize(output, input);
107  data::fill(output, false);
108  for_all_lines(l, lines)
109  if (lines(l).is_textline())
110  {
111  const mln::util::array<component_id_t>& comps = lines(l).component_ids();
112  const L& lbl = lines.components().labeled_image();
113  for_all_elements(c, comps)
114  data::fill((output | lines.components()(comps(c)).bbox()).rw(),
115  ((input | lines.components()(comps(c)).bbox()) | (pw::value(lbl) == comps(c))));
116  }
117  mln::io::pbm::save(output, argv[5]);
118  }
119 
120  // Saving stoppers data to XML
121  document<L> doc(argv[1]);
122  doc.open();
123  doc.set_paragraphs(scribo::make::paragraph(lines));
124  scribo::io::xml::save(doc, argv[6], scribo::io::xml::Full);
125 
126 
127 }