[Olena-patches] 3504: Split Scribo's code into several routines.

* scribo/Makefile: update include path. * scribo/core/central_sites.hh, * scribo/core/component_bboxes.hh, * scribo/core/erase_bboxes.hh, * scribo/core/macros.hh, * scribo/debug/save_label_image.hh, * scribo/debug/save_linked_textbboxes_image.hh, * scribo/debug/save_table_image.hh, * scribo/debug/save_textbboxes_image.hh, * scribo/draw/bounding_box_links.hh, * scribo/draw/bounding_boxes.hh, * scribo/make/debug_filename.hh, * scribo/make/influence_zone_graph.hh, * scribo/table/align_lines_horizontaly.hh, * scribo/table/align_lines_verticaly.hh, * scribo/table/connect_horizontal_lines.hh, * scribo/table/connect_vertical_lines.hh, * scribo/table/erase.hh, * scribo/table/extract_lines_with_rank.hh, * scribo/table/internal/align_lines.hh, * scribo/table/internal/connect_lines.hh, * scribo/table/internal/repair_lines.hh, * scribo/table/repair_horizontal_lines.hh, * scribo/table/repair_vertical_lines.hh, * scribo/text/extract_bboxes.hh, * scribo/text/grouping/group_from_double_link.hh, * scribo/text/grouping/group_from_multiple_links.hh, * scribo/text/grouping/group_from_single_link.hh, * scribo/text/grouping/group_with_multiple_links.hh, * scribo/text/grouping/group_with_single_left_link.hh, * scribo/text/grouping/group_with_single_right_link.hh, * scribo/text/grouping/internal/find_root.hh, * scribo/text/grouping/internal/init_link_array.hh, * scribo/text/grouping/internal/update_link_array.hh, * scribo/text/grouping/internal/update_link_graph.hh, * scribo/text/recognition.hh: new routines. * scribo/photo.cc, * scribo/table.cc: move... * scribo/src/photo.cc, * scribo/src/table.cc: ... here. * scribo/src/extract_text_double_link.cc, * scribo/src/extract_text_multiple_links.cc, * scribo/src/extract_text_single_link.cc: new small test related to text extraction. --- milena/sandbox/ChangeLog | 52 +++++ milena/sandbox/scribo/Makefile | 23 ++- milena/sandbox/scribo/core/central_sites.hh | 97 ++++++++++ milena/sandbox/scribo/core/component_bboxes.hh | 85 +++++++++ .../scribo/{photo.cc => core/erase_bboxes.hh} | 77 +++++--- milena/sandbox/scribo/{photo.cc => core/macros.hh} | 45 +---- .../scribo/{photo.cc => debug/save_label_image.hh} | 77 +++++---- .../scribo/debug/save_linked_textbboxes_image.hh | 170 +++++++++++++++++ milena/sandbox/scribo/debug/save_table_image.hh | 97 ++++++++++ .../sandbox/scribo/debug/save_textbboxes_image.hh | 90 +++++++++ milena/sandbox/scribo/draw/bounding_box_links.hh | 197 ++++++++++++++++++++ milena/sandbox/scribo/draw/bounding_boxes.hh | 93 +++++++++ .../scribo/{photo.cc => make/debug_filename.hh} | 89 ++++++---- milena/sandbox/scribo/make/influence_zone_graph.hh | 121 ++++++++++++ .../sandbox/scribo/src/extract_text_double_link.cc | 109 +++++++++++ .../scribo/src/extract_text_multiple_links.cc | 78 ++++++++ .../sandbox/scribo/src/extract_text_single_link.cc | 108 +++++++++++ milena/sandbox/scribo/{ => src}/photo.cc | 0 milena/sandbox/scribo/{ => src}/table.cc | 0 .../scribo/table/align_lines_horizontaly.hh | 97 ++++++++++ .../sandbox/scribo/table/align_lines_verticaly.hh | 98 ++++++++++ .../scribo/table/connect_horizontal_lines.hh | 94 ++++++++++ .../sandbox/scribo/table/connect_vertical_lines.hh | 95 ++++++++++ milena/sandbox/scribo/table/erase.hh | 97 ++++++++++ .../scribo/table/extract_lines_with_rank.hh | 127 +++++++++++++ .../sandbox/scribo/table/internal/align_lines.hh | 183 ++++++++++++++++++ .../sandbox/scribo/table/internal/connect_lines.hh | 118 ++++++++++++ .../sandbox/scribo/table/internal/repair_lines.hh | 132 +++++++++++++ .../scribo/table/repair_horizontal_lines.hh | 92 +++++++++ .../sandbox/scribo/table/repair_vertical_lines.hh | 98 ++++++++++ milena/sandbox/scribo/text/extract_bboxes.hh | 95 ++++++++++ .../scribo/text/grouping/group_from_double_link.hh | 108 +++++++++++ .../text/grouping/group_from_multiple_links.hh | 154 +++++++++++++++ .../scribo/text/grouping/group_from_single_link.hh | 101 ++++++++++ .../text/grouping/group_with_multiple_links.hh | 115 ++++++++++++ .../text/grouping/group_with_single_left_link.hh | 135 +++++++++++++ .../text/grouping/group_with_single_right_link.hh | 135 +++++++++++++ .../grouping/internal/find_root.hh} | 82 +++++---- .../grouping/internal/init_link_array.hh} | 79 +++++---- .../text/grouping/internal/update_link_array.hh | 93 +++++++++ .../text/grouping/internal/update_link_graph.hh | 92 +++++++++ milena/sandbox/scribo/text/recognition.hh | 142 ++++++++++++++ 42 files changed, 3970 insertions(+), 200 deletions(-) create mode 100644 milena/sandbox/scribo/core/central_sites.hh create mode 100644 milena/sandbox/scribo/core/component_bboxes.hh copy milena/sandbox/scribo/{photo.cc => core/erase_bboxes.hh} (51%) copy milena/sandbox/scribo/{photo.cc => core/macros.hh} (52%) copy milena/sandbox/scribo/{photo.cc => debug/save_label_image.hh} (52%) create mode 100644 milena/sandbox/scribo/debug/save_linked_textbboxes_image.hh create mode 100644 milena/sandbox/scribo/debug/save_table_image.hh create mode 100644 milena/sandbox/scribo/debug/save_textbboxes_image.hh create mode 100644 milena/sandbox/scribo/draw/bounding_box_links.hh create mode 100644 milena/sandbox/scribo/draw/bounding_boxes.hh copy milena/sandbox/scribo/{photo.cc => make/debug_filename.hh} (52%) create mode 100644 milena/sandbox/scribo/make/influence_zone_graph.hh create mode 100644 milena/sandbox/scribo/src/extract_text_double_link.cc create mode 100644 milena/sandbox/scribo/src/extract_text_multiple_links.cc create mode 100644 milena/sandbox/scribo/src/extract_text_single_link.cc copy milena/sandbox/scribo/{ => src}/photo.cc (100%) rename milena/sandbox/scribo/{ => src}/table.cc (100%) create mode 100644 milena/sandbox/scribo/table/align_lines_horizontaly.hh create mode 100644 milena/sandbox/scribo/table/align_lines_verticaly.hh create mode 100644 milena/sandbox/scribo/table/connect_horizontal_lines.hh create mode 100644 milena/sandbox/scribo/table/connect_vertical_lines.hh create mode 100644 milena/sandbox/scribo/table/erase.hh create mode 100644 milena/sandbox/scribo/table/extract_lines_with_rank.hh create mode 100644 milena/sandbox/scribo/table/internal/align_lines.hh create mode 100644 milena/sandbox/scribo/table/internal/connect_lines.hh create mode 100644 milena/sandbox/scribo/table/internal/repair_lines.hh create mode 100644 milena/sandbox/scribo/table/repair_horizontal_lines.hh create mode 100644 milena/sandbox/scribo/table/repair_vertical_lines.hh create mode 100644 milena/sandbox/scribo/text/extract_bboxes.hh create mode 100644 milena/sandbox/scribo/text/grouping/group_from_double_link.hh create mode 100644 milena/sandbox/scribo/text/grouping/group_from_multiple_links.hh create mode 100644 milena/sandbox/scribo/text/grouping/group_from_single_link.hh create mode 100644 milena/sandbox/scribo/text/grouping/group_with_multiple_links.hh create mode 100644 milena/sandbox/scribo/text/grouping/group_with_single_left_link.hh create mode 100644 milena/sandbox/scribo/text/grouping/group_with_single_right_link.hh copy milena/sandbox/scribo/{photo.cc => text/grouping/internal/find_root.hh} (52%) rename milena/sandbox/scribo/{photo.cc => text/grouping/internal/init_link_array.hh} (52%) create mode 100644 milena/sandbox/scribo/text/grouping/internal/update_link_array.hh create mode 100644 milena/sandbox/scribo/text/grouping/internal/update_link_graph.hh create mode 100644 milena/sandbox/scribo/text/recognition.hh diff --git a/milena/sandbox/ChangeLog b/milena/sandbox/ChangeLog index 4b4d75e..beea9cf 100644 --- a/milena/sandbox/ChangeLog +++ b/milena/sandbox/ChangeLog @@ -1,3 +1,55 @@ +2009-03-10 Guillaume Lazzara &lt;z(a)lrde.epita.fr&gt; + + Split Scribo's code into several routines. + + * scribo/Makefile: update include path. + + * scribo/core/central_sites.hh, + * scribo/core/component_bboxes.hh, + * scribo/core/erase_bboxes.hh, + * scribo/core/macros.hh, + * scribo/debug/save_label_image.hh, + * scribo/debug/save_linked_textbboxes_image.hh, + * scribo/debug/save_table_image.hh, + * scribo/debug/save_textbboxes_image.hh, + * scribo/draw/bounding_box_links.hh, + * scribo/draw/bounding_boxes.hh, + * scribo/make/debug_filename.hh, + * scribo/make/influence_zone_graph.hh, + * scribo/table/align_lines_horizontaly.hh, + * scribo/table/align_lines_verticaly.hh, + * scribo/table/connect_horizontal_lines.hh, + * scribo/table/connect_vertical_lines.hh, + * scribo/table/erase.hh, + * scribo/table/extract_lines_with_rank.hh, + * scribo/table/internal/align_lines.hh, + * scribo/table/internal/connect_lines.hh, + * scribo/table/internal/repair_lines.hh, + * scribo/table/repair_horizontal_lines.hh, + * scribo/table/repair_vertical_lines.hh, + * scribo/text/extract_bboxes.hh, + * scribo/text/grouping/group_from_double_link.hh, + * scribo/text/grouping/group_from_multiple_links.hh, + * scribo/text/grouping/group_from_single_link.hh, + * scribo/text/grouping/group_with_multiple_links.hh, + * scribo/text/grouping/group_with_single_left_link.hh, + * scribo/text/grouping/group_with_single_right_link.hh, + * scribo/text/grouping/internal/find_root.hh, + * scribo/text/grouping/internal/init_link_array.hh, + * scribo/text/grouping/internal/update_link_array.hh, + * scribo/text/grouping/internal/update_link_graph.hh, + * scribo/text/recognition.hh: new routines. + + * scribo/photo.cc, + * scribo/table.cc: move... + * scribo/src/photo.cc, + * scribo/src/table.cc: ... here. + + * scribo/src/extract_text_double_link.cc, + * scribo/src/extract_text_multiple_links.cc, + * scribo/src/extract_text_single_link.cc: new small test related to + text extraction. + 2009-03-10 Fabien Freling &lt;fabien.freling(a)lrde.epita.fr&gt; Implement 3D watershed and various tools. diff --git a/milena/sandbox/scribo/Makefile b/milena/sandbox/scribo/Makefile index 5d09e02..cbb462b 100644 --- a/milena/sandbox/scribo/Makefile +++ b/milena/sandbox/scribo/Makefile @@ -1,16 +1,31 @@ all: table photo tabledbg: demat.hh - g++ -Wextra -Wall -I../.. -I$(HOME)/local/include -O1 -g table.cc $(HOME)/local/lib/libtesseract_full.a -lpthread -o table + g++ -Wextra -Wall -I../.. -I../ -I$(HOME)/local/include -O1 -g src/table.cc $(HOME)/local/lib/libtesseract_full.a -lpthread -o bin/table.out table: demat.hh - g++ -Wextra -Wall -I../.. -I$(HOME)/local/include -O1 -DNDEBUG table.cc $(HOME)/local/lib/libtesseract_full.a -lpthread -o table + g++ -Wextra -Wall -I../.. -I. -I.. -I$(HOME)/local/include -O1 -DNDEBUG src/table.cc $(HOME)/local/lib/libtesseract_full.a -lpthread -o bin/table.out + +oldtable: demat.old.hh + g++ -Wextra -Wall -I../.. -I../ -I$(HOME)/local/include -O1 -DNDEBUG src/table_old.cc $(HOME)/local/lib/libtesseract_full.a -lpthread -o bin/table_old.out photo: demat.hh - g++ -Wextra -Wall -I../.. -I$(HOME)/local/include -O1 -DNDEBUG photo.cc $(HOME)/local/lib/libtesseract_full.a -lpthread -o photo + g++ -Wextra -Wall -I../.. -I. -I.. -I$(HOME)/local/include -O1 -DNDEBUG src/photo.cc $(HOME)/local/lib/libtesseract_full.a -lpthread -o bin/photo.out + + + +extract_text_single_link: + g++ -Wextra -Wall -I../.. -I../ -I$(HOME)/local/include -O1 -DNDEBUG src/extract_text_single_link.cc -o bin/extract_text_single_link.out + +extract_text_double_link: + g++ -Wextra -Wall -I../.. -I../ -I$(HOME)/local/include -O1 -DNDEBUG src/extract_text_double_link.cc -o bin/extract_text_double_link.out + +extract_text_multiple_links: + g++ -Wextra -Wall -I../.. -I../ -I$(HOME)/local/include -O1 -DNDEBUG src/extract_text_multiple_links.cc -o bin/extract_text_multiple_links.out + clean: rm *.ppm *.pgm *.pbm dist-clean: clean - rm -f table photo + rm -f bin/table.out bin/photo.out diff --git a/milena/sandbox/scribo/core/central_sites.hh b/milena/sandbox/scribo/core/central_sites.hh new file mode 100644 index 0000000..fe47625 --- /dev/null +++ b/milena/sandbox/scribo/core/central_sites.hh @@ -0,0 +1,97 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_CORE_CENTRAL_SITES_HH +# define SCRIBO_CORE_CENTRAL_SITES_HH + +/// \file scribo/core/central_sites.hh +/// +/// Returns the edge central sites of a box. + +# include <mln/util/couple.hh> +# include <mln/core/site_set/box.hh> + +namespace scribo +{ + using namespace mln; + + /// Returns the edge central sites of a box + /// + /// \param[in] b the bbbox + /// \param[in] dim the dimension used to compute the site. + /*! + ** + ** If dim == 0, returns the left and right central sites. + ** + ** |-------| \n + ** X X \n + ** |-------| \n + ** + ** If dim == 1, returns the top and bottom central sites. + ** + ** |---X---| \n + ** | | \n + ** |---X---| \n + ** + ** + ** ... And so on. + ** + */ + template <typename P> + util::couple<P,P> + central_sites(const box<P>& b, unsigned dim); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename P> + inline + util::couple<P,P> + central_sites(const box<P>& b, unsigned dim) + { + trace::entering("scribo::central_sites"); + mln_precondition(b.is_valid()); + + unsigned n = b.pmax()[dim] - b.pmin()[dim]; + + P p1 = b.center(); + p1[dim] -= n / 2; + P p2 = b.center(); + p2[dim] += n / 2; + + trace::exiting("scribo::central_sites"); + return make::couple(p1, p2); + } + +# endif // ! MLN_INCLUDE_ONLY + +} // end of namespace scribo + + +#endif // ! SCRIBO_CORE_CENTRAL_SITES_HH diff --git a/milena/sandbox/scribo/core/component_bboxes.hh b/milena/sandbox/scribo/core/component_bboxes.hh new file mode 100644 index 0000000..5af99fc --- /dev/null +++ b/milena/sandbox/scribo/core/component_bboxes.hh @@ -0,0 +1,85 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_CORE_COMPONENT_BBOXES_HH +# define SCRIBO_CORE_COMPONENT_BBOXES_HH + +/// \file scribo/core/component_bboxes.hh +/// +/// Extract the bounding boxes of image components. + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/neighborhood.hh> +# include <mln/core/site_set/box.hh> + +# include <mln/labeling/blobs.hh> +# include <mln/labeling/compute.hh> + +# include <mln/util/array.hh> + + +namespace scribo +{ + + /// Extract the components bboxes. + template <typename I, typename N, typename V> + util::array< box<mln_site(I)> > + component_bboxes(const Image<I>& input, + const Neighborhood<N>& nbh, const V& label_type); + +# ifndef MLN_INCLUDE_ONLY + + template <typename I, typename N, typename V> + inline + util::array< box<mln_site(I)> > + component_bboxes(const Image<I>& input, + const Neighborhood<N>& nbh, const V&) + { + trace::entering("scribo::component_bboxes"); + + mlc_equal(mln_value(I),bool)::check(); + mlc_is_a(V, mln::value::Symbolic)::check(); + mln_precondition(exact(input).is_valid()); + mln_precondition(exact(nbh).is_valid()); + + V nbboxes; + mln_ch_value(I,V) lbl = labeling::blobs(input, nbh, nbboxes); + typedef util::array< box<mln_site(I)> > bboxes_t; + bboxes_t bboxes = labeling::compute(accu::meta::bbox(), lbl, nbboxes); + + trace::exiting("scribo::component_bboxes"); + return bboxes; + } + +# endif // ! MLN_INCLUDE_ONLY + +} // end of namespace scribo + + +#endif // ! SCRIBO_CORE_COMPONENT_BBOXES_HH diff --git a/milena/sandbox/scribo/photo.cc b/milena/sandbox/scribo/core/erase_bboxes.hh similarity index 51% copy from milena/sandbox/scribo/photo.cc copy to milena/sandbox/scribo/core/erase_bboxes.hh index a173750..cea998b 100644 --- a/milena/sandbox/scribo/photo.cc +++ b/milena/sandbox/scribo/core/erase_bboxes.hh @@ -1,4 +1,5 @@ -// Copyright (C) 2008 EPITA Research and Development Laboratory +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) // // This file is part of the Olena Library. This library is free // software; you can redistribute it and/or modify it under the terms @@ -25,40 +26,56 @@ // reasons why the executable file might be covered by the GNU General // Public License. +#ifndef SCRIBO_CORE_ERASE_BBOXES_HH +# define SCRIBO_CORE_ERASE_BBOXES_HH -#include "demat.hh" +/// \file scribo/core/erase_bboxes.hh +/// +/// Remove the content of bounding boxes from an image. -int main(int argc, char*argv[]) +# include <mln/core/concept/image.hh> +# include <mln/core/site_set/box.hh> +# include <mln/data/paste.hh> +# include <mln/pw/all.hh> +# include <mln/util/array.hh> + +# include <scribo/core/macros.hh> + +namespace scribo { - using namespace mln; - using value::int_u8; - if (argc < 2) + /// Remove the content of bounding boxes from an image. + template <typename I> + void + erase_bboxes(Image<I>& input_, + const util::array< box<mln_site(I)> >& bboxes); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + void + erase_bboxes(Image<I>& input_, + const util::array< box<mln_site(I)> >& bboxes) { - std::cout << argv[0] << " <in.pbm> <out.pgm> <l> <bbox_larger> <bbox_distance> <min_comp_nsites>" << std::endl - << std::endl << std::endl - << std::endl - << "==========" - << std::endl << std::endl - << "<in.pbm> B/W inverted input image." - << std::endl << std::endl -/* << "<out.ppm> RGB8 output image." - << std::endl << std::endl - << "<bbox_distance> Maximum distance between character bounding boxes. Used for bbox grouping." - << std::endl << std::endl - << "<min_comp_nsites> Minimum site count of a character/text component." - << std::endl - << " If a component have a site count lesser than this value, it is erased." - << std::endl << std::endl - << std::endl*/ - << "==========" - << std::endl << std::endl - << "HINT: compile with -DNOUT to avoid debug images." - << std::endl << std::endl; - return 1; + trace::entering("scribo::erase_bboxes"); + + mlc_equal(mln_value(I),bool)::check(); + + I& input = exact(input_); + mln_precondition(input.is_valid()); + + for_all_components(i, bboxes) + data::paste((pw::cst(false) | bboxes[i] | + (pw::value(input) == true)), input); + + trace::exiting("scribo::erase_bboxes"); } - scribo::demat_photo(argv); - return 0; -} +# endif // ! MLN_INCLUDE_ONLY + +} // end of namespace scribo + +#endif // ! SCRIBO_CORE_ERASE_BBOXES_HH diff --git a/milena/sandbox/scribo/photo.cc b/milena/sandbox/scribo/core/macros.hh similarity index 52% copy from milena/sandbox/scribo/photo.cc copy to milena/sandbox/scribo/core/macros.hh index a173750..91a335c 100644 --- a/milena/sandbox/scribo/photo.cc +++ b/milena/sandbox/scribo/core/macros.hh @@ -1,4 +1,5 @@ -// Copyright (C) 2008 EPITA Research and Development Laboratory +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) // // This file is part of the Olena Library. This library is free // software; you can redistribute it and/or modify it under the terms @@ -25,40 +26,16 @@ // reasons why the executable file might be covered by the GNU General // Public License. +#ifndef SCRIBO_CORE_MACROS_HH +# define SCRIBO_CORE_MACROS_HH -#include "demat.hh" +# define for_all_ncomponents(C, NCOMP) \ + for (unsigned C = 1; C <= NCOMP; ++C) -int main(int argc, char*argv[]) -{ - using namespace mln; - using value::int_u8; +# define for_all_components(C, S) \ + for (unsigned C = 1; C < S.nelements(); ++C) - if (argc < 2) - { - std::cout << argv[0] << " <in.pbm> <out.pgm> <l> <bbox_larger> <bbox_distance> <min_comp_nsites>" << std::endl - << std::endl << std::endl - << std::endl - << "==========" - << std::endl << std::endl - << "<in.pbm> B/W inverted input image." - << std::endl << std::endl -/* << "<out.ppm> RGB8 output image." - << std::endl << std::endl - << "<bbox_distance> Maximum distance between character bounding boxes. Used for bbox grouping." - << std::endl << std::endl - << "<min_comp_nsites> Minimum site count of a character/text component." - << std::endl - << " If a component have a site count lesser than this value, it is erased." - << std::endl << std::endl - << std::endl*/ - << "==========" - << std::endl << std::endl - << "HINT: compile with -DNOUT to avoid debug images." - << std::endl << std::endl; - return 1; - } +# define for_all_elements(E, S) \ + for (unsigned E = 0; E < S.nelements(); ++E) - scribo::demat_photo(argv); - - return 0; -} +#endif // ! SCRIBO_CORE_MACROS_HH diff --git a/milena/sandbox/scribo/photo.cc b/milena/sandbox/scribo/debug/save_label_image.hh similarity index 52% copy from milena/sandbox/scribo/photo.cc copy to milena/sandbox/scribo/debug/save_label_image.hh index a173750..e9b01f1 100644 --- a/milena/sandbox/scribo/photo.cc +++ b/milena/sandbox/scribo/debug/save_label_image.hh @@ -1,4 +1,5 @@ -// Copyright (C) 2008 EPITA Research and Development Laboratory +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) // // This file is part of the Olena Library. This library is free // software; you can redistribute it and/or modify it under the terms @@ -26,39 +27,49 @@ // Public License. -#include "demat.hh" +#ifndef SCRIBO_MAKE_DEBUG_SAVE_LABEL_IMAGE_HH +# define SCRIBO_MAKE_DEBUG_SAVE_LABEL_IMAGE_HH -int main(int argc, char*argv[]) +/// \file scribo/debug/save_label_image.hh +/// +/// Save a labeled image in a color image. + +namespace scribo { - using namespace mln; - using value::int_u8; - if (argc < 2) + namespace debug { - std::cout << argv[0] << " <in.pbm> <out.pgm> <l> <bbox_larger> <bbox_distance> <min_comp_nsites>" << std::endl - << std::endl << std::endl - << std::endl - << "==========" - << std::endl << std::endl - << "<in.pbm> B/W inverted input image." - << std::endl << std::endl -/* << "<out.ppm> RGB8 output image." - << std::endl << std::endl - << "<bbox_distance> Maximum distance between character bounding boxes. Used for bbox grouping." - << std::endl << std::endl - << "<min_comp_nsites> Minimum site count of a character/text component." - << std::endl - << " If a component have a site count lesser than this value, it is erased." - << std::endl << std::endl - << std::endl*/ - << "==========" - << std::endl << std::endl - << "HINT: compile with -DNOUT to avoid debug images." - << std::endl << std::endl; - return 1; - } - - scribo::demat_photo(argv); - - return 0; -} + + /// Save a labeled image in a color image. + template <typename I> + void + save_label_image(const Image<I>& lbl, const mln_value(I)& nlabels, + const char *filename); + + +# ifndef MLN_INCLUDE_ONLY + + template <typename I> + inline + void + save_label_image(const Image<I>& lbl, const mln_value(I)& nlabels, + const char *filename) + { + trace::entering("scribo::debug::save_label_image"); + mlc_is_a(mln_value(I), mln::value::Symbolic)::check(); + mln_precondition(exact(lbl).is_valid()); + + io::ppm::save(debug::colorize(rgb8(), lbl, nlabels), filename); + + trace::exiting("scribo::debug::save_label_image"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::debug + +} // end of namespace scribo + + +#endif // ! SCRIBO_MAKE_DEBUG_SAVE_LABEL_IMAGE_HH diff --git a/milena/sandbox/scribo/debug/save_linked_textbboxes_image.hh b/milena/sandbox/scribo/debug/save_linked_textbboxes_image.hh new file mode 100644 index 0000000..a7f8927 --- /dev/null +++ b/milena/sandbox/scribo/debug/save_linked_textbboxes_image.hh @@ -0,0 +1,170 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_DEBUG_SAVE_LINKED_TEXTBBOXES_IMAGE_HH +# define SCRIBO_DEBUG_SAVE_LINKED_TEXTBBOXES_IMAGE_HH + +/// \file scribo/draw/bounding_boxes.hh +/// +/// Save the bounding box links image. + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/graph.hh> +# include <mln/level/convert.hh> +# include <mln/value/rgb8.hh> +# include <mln/util/array.hh> +# include <mln/io/ppm/save.hh> + +# include <scribo/draw/bounding_boxes.hh> +# include <scribo/draw/bounding_box_links.hh> + + +namespace scribo +{ + + namespace debug + { + + using namespace mln; + + + /// Save the bounding box links image. + template <typename I> + void + save_linked_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const util::array<unsigned>& link_array, + const value::rgb8& box_value, + const value::rgb8& link_value, + const std::string& filename); + + + template <typename I> + void + save_linked_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const util::array<unsigned>& left_link, + const util::array<unsigned>& right_array, + const value::rgb8& box_value, + const value::rgb8& left_link_value, + const value::rgb8& right_link_value, + const std::string& filename); + + + template <typename I, typename G> + void + save_linked_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const Graph<G>& g, + const value::rgb8& box_value, + const value::rgb8& link_value, + const std::string& filename); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + inline + void + save_linked_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const util::array<unsigned>& link_array, + const value::rgb8& box_value, + const value::rgb8& link_value, + const std::string& filename) + { + trace::entering("scribo::debug::save_linked_textbboxes_image"); + mln_precondition(exact(input).is_valid()); + + mln_ch_value(I,value::rgb8) tmp = level::convert(value::rgb8(), input); + draw::bounding_boxes(tmp, textbboxes, box_value); + draw::bounding_box_links(tmp, textbboxes, link_array, link_value); + io::ppm::save(tmp, filename); + + trace::exiting("scribo::debug::save_linked_textbboxes_image"); + } + + + template <typename I> + inline + void + save_linked_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const util::array<unsigned>& left_link, + const util::array<unsigned>& right_link, + const value::rgb8& box_value, + const value::rgb8& left_value, + const value::rgb8& right_value, + const std::string& filename) + { + trace::entering("scribo::debug::save_linked_textbboxes_image"); + mln_precondition(exact(input).is_valid()); + + mln_ch_value(I,value::rgb8) tmp = level::convert(value::rgb8(), input); + draw::bounding_boxes(tmp, textbboxes, box_value); + draw::bounding_box_links(tmp, textbboxes, + left_link, right_link, + left_value, right_value); + io::ppm::save(tmp, filename); + + trace::exiting("scribo::debug::save_linked_textbboxes_image"); + } + + + template <typename I, typename G> + inline + void + save_linked_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const Graph<G>& g, + const value::rgb8& box_value, + const value::rgb8& link_value, + const std::string& filename) + { + trace::entering("scribo::debug::save_linked_textbboxes_image"); + mln_precondition(exact(g).is_valid()); + mln_precondition(exact(input).is_valid()); + + mln_ch_value(I,value::rgb8) tmp = level::convert(value::rgb8(), input); + draw::bounding_boxes(tmp, textbboxes, box_value); + draw::bounding_box_links(tmp, textbboxes, g, link_value); + io::ppm::save(tmp, filename); + + trace::exiting("scribo::debug::save_linked_textbboxes_image"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::debug + +} // end of namespace scribo + + +#endif // ! SCRIBO_DEBUG_SAVE_LINKED_TEXTBBOXES_IMAGE_HH diff --git a/milena/sandbox/scribo/debug/save_table_image.hh b/milena/sandbox/scribo/debug/save_table_image.hh new file mode 100644 index 0000000..b170420 --- /dev/null +++ b/milena/sandbox/scribo/debug/save_table_image.hh @@ -0,0 +1,97 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_DEBUG_SAVE_TABLE_IMAGE_HH +# define SCRIBO_DEBUG_SAVE_TABLE_IMAGE_HH + +/// \file scribo/debug/save_table_image.hh +/// +/// Save table line bounding boxes in an image. + +# include <string> + +# include <mln/core/concept/image.hh> +# include <mln/data/fill.hh> +# include <mln/util/array.hh> +# include <mln/util/couple.hh> +# include <mln/value/rgb8.hh> +# include <mln/io/ppm/save.hh> + +# include <scribo/draw/bounding_boxes.hh> + +namespace scribo +{ + + namespace debug + { + + /// Save lines bounding boxes in an image filled with \p bg_color. + /// Bounding boxes are displayed with \p bbox_color. + template <typename I> + void + save_table(const Image<I>& input_, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > tableboxes, + const value::rgb8& bg_color, + const value::rgb8& bbox_color, + const std::string& filename); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + void + save_table(const Image<I>& input_, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > tableboxes, + const value::rgb8& bg_color, + const value::rgb8& bbox_color, + const std::string& filename) + { + trace::entering("scribo::debug::save_table_image"); + const I& input = exact(input_); + mln_precondition(input.is_valid()); + + mln_ch_value(I,value::rgb8) out2(exact(input).domain()); + data::fill(out2, bg_color); + draw::bounding_boxes(out2, tableboxes.first(), bbox_color); + draw::bounding_boxes(out2, tableboxes.second(), bbox_color); + io::ppm::save(out2, filename); + + trace::exiting("scribo::internal::save_table"); + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::debug + +} // end of namespace scribo + + +#endif // ! SCRIBO_DEBUG_SAVE_TABLE_IMAGE_HH diff --git a/milena/sandbox/scribo/debug/save_textbboxes_image.hh b/milena/sandbox/scribo/debug/save_textbboxes_image.hh new file mode 100644 index 0000000..b210b39 --- /dev/null +++ b/milena/sandbox/scribo/debug/save_textbboxes_image.hh @@ -0,0 +1,90 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_DEBUG_SAVE_TEXTBBOXES_IMAGE_HH +# define SCRIBO_DEBUG_SAVE_TEXTBBOXES_IMAGE_HH + +/// \file scribo/draw/bounding_boxes.hh +/// +/// Draw a list of bounding boxes and their associated mass center. + +# include <mln/core/concept/image.hh> +# include <mln/level/convert.hh> +# include <mln/value/rgb8.hh> +# include <mln/util/array.hh> +# include <mln/io/ppm/save.hh> + +# include <scribo/draw/bounding_boxes.hh> + + +namespace scribo +{ + + namespace debug + { + + using namespace mln; + + + /// Draw a list of bounding boxes and their associated mass center. + template <typename I> + void + save_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const value::rgb8& value, + const std::string& filename); + + +# ifndef MLN_INCLUDE_ONLY + + template <typename I> + inline + void + save_textbboxes_image(const Image<I>& input, + const util::array< box<mln_site(I)> >& textbboxes, + const value::rgb8& value, + const std::string& filename) + { + trace::entering("scribo::debug::save_textbboxes_image"); + mln_precondition(exact(input).is_valid()); + + mln_ch_value(I,value::rgb8) tmp = level::convert(value::rgb8(), input); + draw::bounding_boxes(tmp, textbboxes, value); + io::ppm::save(tmp, filename); + + trace::exiting("scribo::debug::save_textbboxes_image"); + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::debug + +} // end of namespace scribo + + +#endif // ! SCRIBO_DEBUG_SAVE_TEXTBBOXES_IMAGE_HH diff --git a/milena/sandbox/scribo/draw/bounding_box_links.hh b/milena/sandbox/scribo/draw/bounding_box_links.hh new file mode 100644 index 0000000..7cf57ac --- /dev/null +++ b/milena/sandbox/scribo/draw/bounding_box_links.hh @@ -0,0 +1,197 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_DRAW_BOUNDING_BOX_LINKS_HH +# define SCRIBO_DRAW_BOUNDING_BOX_LINKS_HH + +/// \file scribo/draw/bounding_boxes.hh +/// +/// Draw a list of bounding box links. + +# include <mln/core/concept/image.hh> +# include <mln/draw/line.hh> +# include <mln/util/array.hh> +# include <mln/canvas/browsing/depth_first_search.hh> + +# include <scribo/core/macros.hh> + +namespace scribo +{ + + namespace draw + { + + using namespace mln; + + /// Draw a list of bounding box links. + template <typename I> + void + bounding_box_links(Image<I>& input_, + const util::array< box<mln_site(I)> >& bboxes, + const util::array<unsigned>& link_array, + const mln_value(I)& value); + + + template <typename I> + void + bounding_box_links(Image<I>& input, + const util::array< box<mln_site(I)> >& bboxes, + const util::array<unsigned>& left_link, + const util::array<unsigned>& right_link, + const mln_value(I)& left_link_value, + const mln_value(I)& right_link_value); + + +# ifndef MLN_INCLUDE_ONLY + + + namespace internal + { + /// Functor to be passed to depth_first_search. + template <typename I> + struct draw_graph_edges_functor + { + + draw_graph_edges_functor(I& ima, + const util::array<box<mln_site(I)> >& textbboxes, + const mln_value(I)& value) + : ima_(ima), textbboxes_(textbboxes), value_(value) + {} + + template <typename G> + void + init(const Graph<G>& g) + { deja_vu.resize(exact(g).v_nmax(), false); } + + void final() + {} + + void next_component() + {} + + void new_component_from_vertex(unsigned) + {} + + void added_to_queue(unsigned id) + { + deja_vu[id] = true; + mln::draw::line(ima_, textbboxes_[current_vertex].center(), + textbboxes_[id].center(), value_); + } + + void process_vertex(unsigned id) + { current_vertex = id; } + + bool to_be_treated(unsigned id) + { return ! deja_vu[id]; } + + bool to_be_queued(unsigned id) + { return to_be_treated(id); } + + I& ima_; + const util::array<box<mln_site(I)> >& textbboxes_; + mln_value(I) value_; + unsigned current_vertex; + std::vector<bool> deja_vu; + }; + + } // end of namespace scribo::text::grouping::internal + + + template <typename I> + inline + void + bounding_box_links(Image<I>& input_, + const util::array< box<mln_site(I)> >& bboxes, + const util::array<unsigned>& link_array, + const mln_value(I)& value) + { + trace::entering("scribo::draw::bounding_box_links"); + + I& input = exact(input_); + + mln_precondition(input.is_valid()); + + for_all_components(i, link_array) + if (link_array[i] != i) + mln::draw::line(input, + bboxes[i].center(), + bboxes[link_array[i]].center(), + value); + + trace::exiting("scribo::draw::bounding_box_links"); + } + + + template <typename I> + inline + void + bounding_box_links(Image<I>& input, + const util::array< box<mln_site(I)> >& bboxes, + const util::array<unsigned>& left_link, + const util::array<unsigned>& right_link, + const mln_value(I)& left_link_value, + const mln_value(I)& right_link_value) + { + trace::entering("scribo::draw::bounding_box_links"); + mln_precondition(exact(input).is_valid()); + + bounding_box_links(input, bboxes, left_link, left_link_value); + bounding_box_links(input, bboxes, right_link, right_link_value); + + trace::exiting("scribo::draw::bounding_box_links"); + } + + + template <typename I, typename G> + inline + void + bounding_box_links(Image<I>& input, + const util::array< box<mln_site(I)> >& bboxes, + const Graph<G>& g, + const mln_value(I)& link_value) + { + trace::entering("scribo::draw::bounding_box_links"); + mln_precondition(exact(input).is_valid()); + mln_precondition(exact(g).v_nmax() == bboxes.nelements()); + + internal::draw_graph_edges_functor<I> f(exact(input), bboxes, link_value); + canvas::browsing::depth_first_search(g, f); + + trace::exiting("scribo::draw::bounding_box_links"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::draw + +} // end of namespace scribo + + +#endif // ! SCRIBO_DRAW_BOUNDING_BOX_LINKS_HH diff --git a/milena/sandbox/scribo/draw/bounding_boxes.hh b/milena/sandbox/scribo/draw/bounding_boxes.hh new file mode 100644 index 0000000..0db60a4 --- /dev/null +++ b/milena/sandbox/scribo/draw/bounding_boxes.hh @@ -0,0 +1,93 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_DRAW_BOUNDING_BOXES_HH +# define SCRIBO_DRAW_BOUNDING_BOXES_HH + +/// \file scribo/draw/bounding_boxes.hh +/// +/// Draw a list of bounding boxes and their associated mass center. + +# include <mln/core/concept/image.hh> +# include <mln/draw/box.hh> +# include <mln/util/array.hh> + +# include <scribo/core/macros.hh> + +namespace scribo +{ + + namespace draw + { + + using namespace mln; + + /// Draw a list of bounding boxes and their associated mass center. + template <typename I> + void + bounding_boxes(Image<I>& input_, + const util::array< box<mln_site(I)> >& boxes, + const mln_value(I)& value); + + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + inline + void + bounding_boxes(Image<I>& input_, + const util::array< box<mln_site(I)> >& boxes, + const mln_value(I)& value) + { + trace::entering("scribo::draw::bounding_boxes"); + + I& input = exact(input_); + + mln_precondition(input.is_valid()); + + for_all_components(i, boxes) + if (boxes[i].is_valid()) + { + input(boxes[i].center()) = value; + mln::draw::box(input, boxes[i], value); + } + + trace::exiting("scribo::draw::bounding_boxes"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::draw + +} // end of namespace scribo + + +#endif // ! SCRIBO_DRAW_BOUNDING_BOXES_HH diff --git a/milena/sandbox/scribo/photo.cc b/milena/sandbox/scribo/make/debug_filename.hh similarity index 52% copy from milena/sandbox/scribo/photo.cc copy to milena/sandbox/scribo/make/debug_filename.hh index a173750..b159328 100644 --- a/milena/sandbox/scribo/photo.cc +++ b/milena/sandbox/scribo/make/debug_filename.hh @@ -1,4 +1,5 @@ -// Copyright (C) 2008 EPITA Research and Development Laboratory +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) // // This file is part of the Olena Library. This library is free // software; you can redistribute it and/or modify it under the terms @@ -26,39 +27,61 @@ // Public License. -#include "demat.hh" +#ifndef SCRIBO_MAKE_DEBUG_FILENAME_HH +# define SCRIBO_MAKE_DEBUG_FILENAME_HH -int main(int argc, char*argv[]) +/// \file scribo/make/debug_filename.hh +/// +/// Construct and returns a formated output file name. + +# include <sstream> + + +namespace scribo { - using namespace mln; - using value::int_u8; - if (argc < 2) + namespace make { - std::cout << argv[0] << " <in.pbm> <out.pgm> <l> <bbox_larger> <bbox_distance> <min_comp_nsites>" << std::endl - << std::endl << std::endl - << std::endl - << "==========" - << std::endl << std::endl - << "<in.pbm> B/W inverted input image." - << std::endl << std::endl -/* << "<out.ppm> RGB8 output image." - << std::endl << std::endl - << "<bbox_distance> Maximum distance between character bounding boxes. Used for bbox grouping." - << std::endl << std::endl - << "<min_comp_nsites> Minimum site count of a character/text component." - << std::endl - << " If a component have a site count lesser than this value, it is erased." - << std::endl << std::endl - << std::endl*/ - << "==========" - << std::endl << std::endl - << "HINT: compile with -DNOUT to avoid debug images." - << std::endl << std::endl; - return 1; - } - - scribo::demat_photo(argv); - - return 0; -} + + /// Construct and returns a formated output file name: + /// + /// `input_filename`_`id`_`name` + std::string + debug_filename(const char *input_filename, + const char *name); + + +# ifndef MLN_INCLUDE_ONLY + + + inline + std::string + debug_filename(const char *input_filename, + const char *name) + { + static int file_id = 1; + + std::ostringstream os; + os << "./" + << input_filename + << "_"; + + if (file_id < 10) + os << "0"; + + os << file_id++ + << "_" + << name; + return os.str(); + } + + +# endif // ! MLN_INCLUDE_ONLY + + + } // end of namespace scribo::make + +} // end of namespace scribo + + +#endif // ! SCRIBO_MAKE_DEBUG_FILENAME_HH diff --git a/milena/sandbox/scribo/make/influence_zone_graph.hh b/milena/sandbox/scribo/make/influence_zone_graph.hh new file mode 100644 index 0000000..c3ad15e --- /dev/null +++ b/milena/sandbox/scribo/make/influence_zone_graph.hh @@ -0,0 +1,121 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_MAKE_INFLUENCE_ZONE_GRAPH_HH +# define SCRIBO_MAKE_INFLUENCE_ZONE_GRAPH_HH + +/// \file scribo/make/influence_zone_graph.hh +/// +/// Create a region adjacency graph from the influence zone image +/// of binary image. + + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/neighborhood.hh> + +# include <mln/labeling/blobs.hh> + +# include <mln/transform/influence_zone_geodesic.hh> + +# include <mln/util/graph.hh> + + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + using namespace mln; + + + /// Compute a labeled image of input, then compute an influence + /// zone image and make a graph from it. + /// + /// \param[in] input_ a binary image. + /// \param[in] nbh_ a neighbordhood. + /// \param[in] label_type The type of this argument is used + /// as label type while labeling the image. + /// \param[in] iz_dmax Max distance of the influence zone. + /// + /// \return a region adjacency graph. + template <typename I, typename V> + util::graph + influence_zone_graph(const Image<I>& input_, + const Neighborhood<N>& nbh_, + const V& label_type, + unsigned iz_dmax); + + +# ifndef MLN_INCLUDE_ONLY + + template <typename I, typename V> + util::graph + influence_zone_graph(const Image<I>& input_, + const Neighborhood<N>& nbh_, + const V& label_type, + unsigned iz_dmax) + { + trace::entering("scribo::make::influence_zone_graph"); + + (void) label_type; + I& input = exact(input_); + const N& nbh = exact(nbh_); + + mlc_is_equal(mln_value(I), bool)::check(); + mln_assertion(input.is_valid()); + mln_assertion(nbh.is_valid()); + + V nlabels; + typedef mln_ch_value(I,V) lbl_t; + lbl_t lbl = labeling::blobs(input, nbh, nlabels); + + lbl_t iz = transform::influence_zone_geodesic(lbl, nbh, iz_dmax); + + util::graph g = make::graph(iz | (pw::value(iz) != pw::cst(literal::zero)), + nbh, nlabels); + + trace::exiting("scribo::make::influence_zone_graph"); + return g; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + + + +#endif // ! SCRIBO_MAKE_INFLUENCE_ZONE_GRAPH_HH + diff --git a/milena/sandbox/scribo/src/extract_text_double_link.cc b/milena/sandbox/scribo/src/extract_text_double_link.cc new file mode 100644 index 0000000..26fccff --- /dev/null +++ b/milena/sandbox/scribo/src/extract_text_double_link.cc @@ -0,0 +1,109 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#include <iostream> + +#include <mln/essential/2d.hh> + +#include <scribo/text/extract_bboxes.hh> +#include <scribo/text/grouping/group_with_single_left_link.hh> +#include <scribo/text/grouping/group_with_single_right_link.hh> +#include <scribo/debug/save_linked_textbboxes_image.hh> +#include <scribo/text/grouping/group_from_double_link.hh> + +#include <scribo/debug/save_textbboxes_image.hh> + +int usage(const char *name) +{ + std::cout << "Usage: " << name << " <input.pbm> " << std::endl; + return 1; +} + + +namespace scribo { namespace text { namespace grouping +{ + using namespace mln; + + template <typename I> + util::array< box<mln_site(I)> > + group_from_double_link(const util::array< box<mln_site(I)> >& textbboxes, + util::array<unsigned>& left_link, + util::array<unsigned>& right_link); +}}} + +int main(int argc, char* argv[]) +{ + using namespace scribo; + using namespace mln; + + if (argc < 1) + return usage(argv[0]); + + image2d<bool> input; + io::pbm::load(input, argv[1]); + + value::label_16 nbboxes; + util::array<box2d> textbboxes = text::extract_bboxes(input, c8(), nbboxes); + + util::array<unsigned> left_link + = text::grouping::group_with_single_left_link(input, + c8(), nbboxes, + textbboxes, 30); + util::array<unsigned> right_link + = text::grouping::group_with_single_right_link(input, + c8(), nbboxes, + textbboxes, 30); + + std::cout << "BEFORE - nbboxes = " << nbboxes << std::endl; + + scribo::debug::save_linked_textbboxes_image(input, + textbboxes, left_link, + literal::red, literal::cyan, + "test_double_link_left_linked.ppm"); + scribo::debug::save_linked_textbboxes_image(input, + textbboxes, right_link, + literal::red, literal::cyan, + "test_double_link_right_linked.ppm"); + + scribo::debug::save_linked_textbboxes_image(input, + textbboxes, left_link, right_link, + literal::red, literal::cyan, + literal::yellow, + "test_double_link_double_linked.ppm"); + + // With validation. + util::array< box<point2d> > grouped_textbboxes + = text::grouping::group_from_double_link(textbboxes, left_link, right_link); + + std::cout << "AFTER double grouping - nbboxes = " << grouped_textbboxes.nelements() << std::endl; + + scribo::debug::save_textbboxes_image(input, grouped_textbboxes, + literal::red, + "test_double_link_grouped_text.ppm"); + +} + diff --git a/milena/sandbox/scribo/src/extract_text_multiple_links.cc b/milena/sandbox/scribo/src/extract_text_multiple_links.cc new file mode 100644 index 0000000..9b181e6 --- /dev/null +++ b/milena/sandbox/scribo/src/extract_text_multiple_links.cc @@ -0,0 +1,78 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#include <iostream> + +#include <mln/essential/2d.hh> + +#include <scribo/text/extract_bboxes.hh> +#include <scribo/text/grouping/group_with_multiple_links.hh> +#include <scribo/text/grouping/group_from_multiple_links.hh> + +#include <scribo/debug/save_textbboxes_image.hh> +#include <scribo/debug/save_linked_textbboxes_image.hh> + +int usage(const char *name) +{ + std::cout << "Usage: " << name << " <input.pbm> " << std::endl; + return 1; +} + +int main(int argc, char* argv[]) +{ + using namespace scribo; + using namespace mln; + + if (argc < 1) + return usage(argv[0]); + + image2d<bool> input; + io::pbm::load(input, argv[1]); + + value::label_16 nbboxes; + util::array<box2d> textbboxes = text::extract_bboxes(input, c8(), nbboxes); + + util::graph g = text::grouping::group_with_multiple_links(input, + c8(), nbboxes, + textbboxes, 30); + + std::cout << "BEFORE - nbboxes = " << nbboxes << std::endl; + scribo::debug::save_linked_textbboxes_image(input, + textbboxes, g, + literal::red, literal::cyan, + "test_multiple_links_left_linked.ppm"); + + util::array< box<point2d> > grouped_textbboxes + = text::grouping::group_from_multiple_links(textbboxes, g); + + std::cout << "AFTER - nbboxes = " << grouped_textbboxes.nelements() << std::endl; + + scribo::debug::save_textbboxes_image(input, grouped_textbboxes, + literal::red, + "test_multiple_links_grouped_text.ppm"); +} + diff --git a/milena/sandbox/scribo/src/extract_text_single_link.cc b/milena/sandbox/scribo/src/extract_text_single_link.cc new file mode 100644 index 0000000..5b4e248 --- /dev/null +++ b/milena/sandbox/scribo/src/extract_text_single_link.cc @@ -0,0 +1,108 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#include <iostream> + +#include <mln/essential/2d.hh> + +#include <scribo/text/extract_bboxes.hh> +#include <scribo/text/grouping/group_with_single_left_link.hh> +#include <scribo/text/grouping/group_with_single_right_link.hh> +#include <scribo/text/grouping/group_from_single_link.hh> + +#include <scribo/debug/save_textbboxes_image.hh> +#include <scribo/debug/save_linked_textbboxes_image.hh> + +int usage(const char *name) +{ + std::cout << "Usage: " << name << " <input.pbm> " << std::endl; + return 1; +} + +int main(int argc, char* argv[]) +{ + using namespace scribo; + using namespace mln; + + if (argc < 1) + return usage(argv[0]); + + image2d<bool> input; + io::pbm::load(input, argv[1]); + + value::label_16 nbboxes; + util::array<box2d> textbboxes = text::extract_bboxes(input, c8(), nbboxes); + + { + std::cout << "* Left grouping" << std::endl; + util::array<unsigned> left_link + = text::grouping::group_with_single_left_link(input, + c8(), nbboxes, + textbboxes, 30); + + std::cout << "BEFORE - nbboxes = " << nbboxes << std::endl; + scribo::debug::save_linked_textbboxes_image(input, + textbboxes, left_link, + literal::red, literal::cyan, + "test_single_left_link_linked.ppm"); + + util::array< box<point2d> > grouped_textbboxes + = text::grouping::group_from_single_link(textbboxes, left_link); + + std::cout << "AFTER - nbboxes = " << grouped_textbboxes.nelements() << std::endl; + + scribo::debug::save_textbboxes_image(input, grouped_textbboxes, + literal::red, + "test_single_left_link_grouped_text.ppm"); + } + + { + std::cout << "* Left grouping" << std::endl; + util::array<unsigned> right_link + = text::grouping::group_with_single_right_link(input, + c8(), nbboxes, + textbboxes, 30); + + std::cout << "BEFORE - nbboxes = " << nbboxes << std::endl; + scribo::debug::save_linked_textbboxes_image(input, + textbboxes, right_link, + literal::red, literal::cyan, + "test_single_right_link_linked.ppm"); + + util::array< box<point2d> > grouped_textbboxes + = text::grouping::group_from_single_link(textbboxes, right_link); + + std::cout << "AFTER - nbboxes = " << grouped_textbboxes.nelements() << std::endl; + + scribo::debug::save_textbboxes_image(input, grouped_textbboxes, + literal::red, + "test_single_right_link_grouped_text.ppm"); + } + + +} + diff --git a/milena/sandbox/scribo/photo.cc b/milena/sandbox/scribo/src/photo.cc similarity index 100% copy from milena/sandbox/scribo/photo.cc copy to milena/sandbox/scribo/src/photo.cc diff --git a/milena/sandbox/scribo/table.cc b/milena/sandbox/scribo/src/table.cc similarity index 100% rename from milena/sandbox/scribo/table.cc rename to milena/sandbox/scribo/src/table.cc diff --git a/milena/sandbox/scribo/table/align_lines_horizontaly.hh b/milena/sandbox/scribo/table/align_lines_horizontaly.hh new file mode 100644 index 0000000..e75f2f2 --- /dev/null +++ b/milena/sandbox/scribo/table/align_lines_horizontaly.hh @@ -0,0 +1,97 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TABLE_ALIGN_LINES_HORIZONTALY_HH +# define SCRIBO_TABLE_ALIGN_LINES_HORIZONTALY_HH + +/// \file scribo/table/align_lines_horizontaly.hh +/// +/// Align line bounding boxes horizontaly. + + +# include <mln/core/concept/image.hh> +# include <mln/core/site_set/box.hh> +# include <mln/geom/nrows.hh> +# include <mln/geom/min_nrow.hh> +# include <mln/geom/max_nrow.hh> + +# include <mln/util/array.hh> + +# include <scribo/table/internal/align_lines.hh> + + +namespace scribo +{ + + namespace table + { + + /// Align line bounding boxes horizontaly. + /// + /// \param[in] input Image from which the line bboxes are + /// extracted from. + /// \param[in, out] lines_bboxes horizontal lines bounding boxes. + /// + /// \return A list of the resulting aligned rows. Each integer is actually + /// a row number. + template <typename I> + util::array<int> + align_lines_horizontaly(const Image<I>& input, + util::array<box<mln_site(I)> >& lines_bboxes); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + util::array<int> + align_lines_horizontaly(const Image<I>& input, + util::array<box<mln_site(I)> >& lines_bboxes) + { + trace::entering("scribo::table::align_lines_horizontaly"); + + mln_precondition(exact(input).is_valid()); + util::array<int> res = internal::align_lines(geom::nrows(input), + geom::min_row(input), + geom::max_row(input), + lines_bboxes, 0); + + trace::exiting("scribo::table::align_lines_horizontaly"); + return res; + } + + +# endif // ! MLN_INCLUDE_ONLY + + + } // end of namespace scribo::table + +} // end of namespace scribo + +#endif // ! SCRIBO_TABLE_ALIGN_LINES_HORIZONTALY_HH diff --git a/milena/sandbox/scribo/table/align_lines_verticaly.hh b/milena/sandbox/scribo/table/align_lines_verticaly.hh new file mode 100644 index 0000000..7e860d0 --- /dev/null +++ b/milena/sandbox/scribo/table/align_lines_verticaly.hh @@ -0,0 +1,98 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TABLE_ALIGN_LINES_VERTICALY_HH +# define SCRIBO_TABLE_ALIGN_LINES_VERTICALY_HH + +/// \file scribo/table/align_lines_verticaly.hh +/// +/// Align line bounding boxes verticaly. + + +# include <mln/core/concept/image.hh> +# include <mln/core/site_set/box.hh> +# include <mln/geom/ncols.hh> +# include <mln/geom/min_ncol.hh> +# include <mln/geom/max_ncol.hh> + +# include <mln/util/array.hh> + +# include <scribo/table/internal/align_lines.hh> + + + +namespace scribo +{ + + namespace table + { + + /// Align line bounding boxes verticaly. + /// + /// \param[in] input Image from which the line bboxes are + /// extracted from. + /// \param[in, out] lines_bboxes vertical lines bounding boxes. + /// + /// \return A list of the resulting aligned cols. Each integer is actually + /// a col number. + template <typename I> + util::array<int> + align_lines_verticaly(const Image<I>& input, + util::array<box<mln_site(I)> >& lines_bboxes); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + util::array<int> + align_lines_verticaly(const Image<I>& input, + util::array<box<mln_site(I)> >& lines_bboxes) + { + trace::entering("scribo::table::align_lines_horizontaly"); + + mln_precondition(exact(input).is_valid()); + util::array<int> res = internal::align_lines(geom::ncols(input), + geom::min_col(input), + geom::max_col(input), + lines_bboxes, 1); + + trace::exiting("scribo::table::align_lines_horizontaly"); + return res; + } + + +# endif // ! MLN_INCLUDE_ONLY + + + } // end of namespace scribo::table + +} // end of namespace scribo + +#endif // ! SCRIBO_TABLE_ALIGN_LINES_VERTICALY_HH diff --git a/milena/sandbox/scribo/table/connect_horizontal_lines.hh b/milena/sandbox/scribo/table/connect_horizontal_lines.hh new file mode 100644 index 0000000..7dcf638 --- /dev/null +++ b/milena/sandbox/scribo/table/connect_horizontal_lines.hh @@ -0,0 +1,94 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TABLE_CONNECT_HORIZONTAL_LINES_HH +# define SCRIBO_TABLE_CONNECT_HORIZONTAL_LINES_HH + +/// \file scribo/table/connect_horizontal_lines.hh +/// +/// Connect horizontal lines with the new aligned columns. + +# include <mln/core/concept/image.hh> +# include <mln/util/array.hh> +# include <mln/util/couple.hh> + +# include <scribo/table/internal/connect_lines.hh> + + +namespace scribo +{ + + namespace table + { + + + /// Connect horizontal lines with the new aligned columns. + /// + /// \param[in] aligned_cols a list of new aligned cols. + /// \param[in,out] tableboxes the vertical and horizontal lines + /// bounding boxes. + /// \param[in] input The image from where the lines are + /// extracted. + template <typename I> + void + connect_vertical_lines(const util::array<int>& aligned_rows, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > tableboxes, + const Image<I>& input) + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + void + connect_horizontal_lines(const util::array<int>& aligned_cols, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > tableboxes, + const Image<I>& input) + { + trace::entering("scribo::table::connect_horizontal_lines"); + mln_precondition(exact(input).is_valid()); + + internal::connect_lines(aligned_cols, tableboxes.second(), + 1, exact(input).ncols()); + + trace::exiting("scribo::table::connect_horizontal_lines"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + + } // end of namespace scribo::table + +} // end of namespace scribo + + +#endif // ! SCRIBO_TABLE_CONNECT_HORIZONTAL_LINES_HH diff --git a/milena/sandbox/scribo/table/connect_vertical_lines.hh b/milena/sandbox/scribo/table/connect_vertical_lines.hh new file mode 100644 index 0000000..98c39f7 --- /dev/null +++ b/milena/sandbox/scribo/table/connect_vertical_lines.hh @@ -0,0 +1,95 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TABLE_CONNECT_VERTICAL_LINES_HH +# define SCRIBO_TABLE_CONNECT_VERTICAL_LINES_HH + +/// \file scribo/table/connect_vertical_lines.hh +/// +/// Connect vertical lines with aligned rows. + +# include <mln/core/concept/image.hh> +# include <mln/util/array.hh> +# include <mln/util/couple.hh> + +# include <scribo/table/internal/connect_lines.hh> + + +namespace scribo +{ + + namespace table + { + + + /// Connect vertical lines with the new aligned rows. + /// + /// \param[in] aligned_rows a list of new aligned rows. + /// \param[in,out] tableboxes the vertical and horizontal lines + /// bounding boxes. + /// \param[in] input The image from where the lines are + /// extracted. + template <typename I> + void + connect_vertical_lines(const util::array<int>& aligned_rows, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > tableboxes, + const Image<I>& input) + + +# ifndef MLN_INCLUDE_ONLY + + + /// Connect vertical lines with the new aligned rows. + template <typename I> + void + connect_vertical_lines(const util::array<int>& aligned_rows, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > tableboxes, + const Image<I>& input) + { + trace::entering("scribo::table::connect_vertical_lines"); + mln_precondition(exact(input).is_valid()); + + internal::connect_lines(aligned_rows, tableboxes.first(), + 0, exact(input).nrows()); + + trace::exiting("scribo::table::connect_vertical_lines"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + + } // end of namespace scribo::table + +} // end of namespace scribo + + +#endif // ! SCRIBO_TABLE_CONNECT_VERTICAL_LINES_HH diff --git a/milena/sandbox/scribo/table/erase.hh b/milena/sandbox/scribo/table/erase.hh new file mode 100644 index 0000000..9d94edb --- /dev/null +++ b/milena/sandbox/scribo/table/erase.hh @@ -0,0 +1,97 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_TABLE_ERASE_HH +# define SCRIBO_TABLE_ERASE_HH + +/// \file scribo/table/erase.hh +/// +/// Erase the table lines in an image. + +# include <mln/core/concept/image.hh> +# include <mln/core/site_set/box.hh> +# include <mln/core/routine/duplicate.hh> + +# include <mln/util/array.hh> +# include <mln/util/couple.hh> + +# include <scribo/core/erase_bboxes.hh> + +namespace scribo +{ + + namespace table + { + + /// Erase table line bboxes from an image. + /// + /// \param[in] line_bboxes vertical and horizontal line bounding + /// boxes. + /// \param[in] in A binary image from which the table line + /// bboxes are extracted. + /// + /// + /// \return A copy of \p in where the table lines are removed. + template <typename I> + mln_concrete(I) + erase(const Image<I>& in, + const util::couple<util::array< box<mln_site(I)> >, + util::array< box<mln_site(I)> > >& line_bboxes); + + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + mln_concrete(I) + erase(const Image<I>& in, + const util::couple<util::array< box<mln_site(I)> >, + util::array< box<mln_site(I)> > >& line_bboxes) + { + trace::entering("scribo::internal::erase_table"); + mlc_equal(mln_value(I),bool)::check(); + mln_precondition(exact(in).is_valid()); + + I output = duplicate(in); + + erase_bboxes(output, line_bboxes.first()); + erase_bboxes(output, line_bboxes.second()); + + trace::exiting("scribo::internal::erase_table"); + return output; + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::table + +} // end of namespace scribo + +#endif // ! SCRIBO_TABLE_ERASE_HH diff --git a/milena/sandbox/scribo/table/extract_lines_with_rank.hh b/milena/sandbox/scribo/table/extract_lines_with_rank.hh new file mode 100644 index 0000000..242dc30 --- /dev/null +++ b/milena/sandbox/scribo/table/extract_lines_with_rank.hh @@ -0,0 +1,127 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TABLE_EXTRACT_LINES_WITH_RANK_HH +# define SCRIBO_TABLE_EXTRACT_LINES_WITH_RANK_HH + +/// \file scribo/table/extract_lines_with_rank.hh +/// +/// Extract table lines using a rank filter. + + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/window.hh> +# include <mln/core/concept/neighborhood.hh> +# include <mln/core/site_set/box.hh> + +# include <mln/morpho/rank_filter.hh> + +# include <mln/accu/bbox.hh> + +# include <mln/util/array.hh> +# include <mln/util/couple.hh> + + +# include <scribo/core/component_bboxes.hh> + +namespace scribo +{ + + namespace table + { + + /// Find table bboxes thanks to a rank filter. + /*! + * + * \param[in] input_ A binary image. + * \param[in] nbh_ The neighborhood used for labeling image components. + * \param[in] label_type The type used to store the labels. + * \param[in] vwin Window used to extract the vertical lines in the rank + * filter. + * \param[in] hwin Window used to extract the horizontal lines in the rank + * filter. + * \param[in] vrank_k Rank used for vertical lines filtering. + * \param[in] hrank_k Rank used for horizontal lines filtering. + * + * \return pair of array of bounding boxes. The first array holds the + * vertical lines bounding boxes and the second one the + * horizontal lines bounding boxes. + */ + template <typename I, typename N, typename V, typename HW, typename VW> + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > + extract_lines_with_rank(const Image<I>& input_, + const Neighborhood<N>& nbh_, const V& label_type, + const Window<HW>& vwin, const Window<VW>& hwin, + unsigned vrank_k, unsigned hrank_k); + + +# ifndef MLN_INCLUDE_ONLY + + template <typename I, typename N, typename V, typename HW, typename VW> + inline + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > > + extract_lines_with_rank(const Image<I>& input_, + const Neighborhood<N>& nbh_, const V& label_type, + const Window<HW>& vwin, const Window<VW>& hwin, + unsigned vrank_k, unsigned hrank_k) + { + trace::entering("scribo::table::extract_lines_with_rank"); + + mlc_equal(mln_value(I),bool)::check(); + mlc_is_a(V, mln::value::Symbolic)::check(); + const I& input = exact(input_); + const N& nbh = exact(nbh_); + mln_precondition(input.is_valid()); + mln_precondition(nbh.is_valid()); + mln_precondition(exact(vwin).is_valid()); + mln_precondition(exact(hwin).is_valid()); + + typedef accu::bbox<mln_psite(I)> A; + typedef util::array<mln_result(A)> boxes_t; + + // Vertical lines + mln_ch_value(I,bool) vfilter = morpho::rank_filter(input, vwin, vrank_k); + boxes_t vboxes = component_bboxes(vfilter, nbh, label_type); + + // Horizontal lines. + mln_ch_value(I,bool) hfilter = morpho::rank_filter(input, hwin, hrank_k); + boxes_t hboxes = component_bboxes(hfilter, nbh, label_type); + + return mln::make::couple(vboxes, hboxes); + } + +# endif // !MLN_INCLUDE_ONLY + + } // end of namespace scribo::table + +} // end of namespace scribo + +#endif // ! SCRIBO_TABLE_EXTRACT_LINES_WITH_RANK_HH diff --git a/milena/sandbox/scribo/table/internal/align_lines.hh b/milena/sandbox/scribo/table/internal/align_lines.hh new file mode 100644 index 0000000..aaad9fa --- /dev/null +++ b/milena/sandbox/scribo/table/internal/align_lines.hh @@ -0,0 +1,183 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TABLE_INTERNAL_ALIGN_LINES_HH +# define SCRIBO_TABLE_INTERNAL_ALIGN_LINES_HH + +/// \file scribo/table/internal/align_lines.hh +/// +/// Align table lines verticaly or horizontaly. + + +# include <mln/core/site_set/box.hh> +# include <mln/util/array.hh> +# include <mln/util/set.hh> + +# include <scribo/core/macros.hh> + + +namespace scribo +{ + + namespace table + { + + namespace internal + { + + /// Align table lines bboxes according to a given dimension. + /// + /// \return A list of the resulting aligned cols. Each integer is actually + /// a col number. + /* + ** + ** 0 1 3 4 5 6 + ** ------------ ------- + ** 0 |- - - - - | | {0,1} | + ** 1 | - - | | {0,1} | + ** 2 | | | {1} | + ** 3 | | | | + ** 4 | | | {2} | + ** 5 |- - | | {2} | + ** 6 | | | {2} | + ** + ** \p hboxes contains all the table lines bboxes. Each bbox is + ** associated with an id, its location in the array. + ** + ** For each bbox, its id is marked in a vector. The location is defined, + ** according to the given parameter \p dim, either by the row or the col + ** value of the pmin site. + ** + ** Ids are then propagated in the vector according a small delta value. + ** if bbox ids are at the same location in the vector, their related bboxes + ** are likely to be on the same line. + ** + ** Finally, iterate over the vector until all bboxes have been treated. + ** For each iteration, the set with a specific number of elements is found + ** and all bboxes referenced in this set are aligned on the same row or col. + ** + */ + template <typename P> + util::array<int> + align_lines(unsigned nsites, + int min_coord, + int max_coord, + util::array<box<P> >& line_boxes, + unsigned dim); + + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename P> + util::array<int> + align_lines(unsigned nsites, + int min_coord, + int max_coord, + util::array<box<P> >& line_boxes, + unsigned dim) + { + trace::entering("scribo::internal::align_lines"); + + mln_precondition(nsites > 0); + + util::array< util::set<unsigned> > lines; + lines.resize(nsites); + + // Map components with actual lines. + for_all_components(i, line_boxes) + { + int minline = line_boxes[i].pmin()[dim] - 5; + minline = (minline < min_coord ? min_coord : minline); + int maxline = line_boxes[i].pmax()[dim] + 5; + maxline = (maxline > max_coord ? max_coord : maxline); + + for (int line = minline; + line <= maxline; ++line) + lines[line].insert(i); + } + + // Init box2line + util::array<int> box2line; + box2line.resize(line_boxes.nelements()); + for_all_elements(i, box2line) + box2line[i] = -1; + + // Find the line with the highest element count. + unsigned max_nelts = 0; + for_all_elements(i, lines) + if (max_nelts < lines[i].nelements()) + max_nelts = lines[i].nelements(); + + // Aligning lines + // FIXME: not optimal... Make it faster! + // We may do too much iterations (while loop) and some of them may + // be done for nothing... + util::array<int> newlines; + while (max_nelts > 0) + { + for_all_elements(i, lines) + if (lines[i].nelements() == max_nelts) + { + accu::mean<unsigned> mean; + for_all_elements(j, lines[i]) + if (box2line[lines[i][j]] == -1) + mean.take(line_boxes[lines[i][j]].center()[dim]); + + if (mean.is_valid()) + { + for_all_elements(j, lines[i]) + if (box2line[lines[i][j]] == -1) + { + line_boxes[lines[i][j]].pmin()[dim] = mean.to_result(); + line_boxes[lines[i][j]].pmax()[dim] = mean.to_result(); + box2line[lines[i][j]] = mean.to_result(); + } + newlines.append(mean.to_result()); + } + } + --max_nelts; + } + + trace::exiting("scribo::internal::align_lines"); + return newlines; + } + + +# endif // ! MLN_INCLUDE_ONLY + + + } // end of namespace scribo::table::internal + + } // end of namespace scribo::table + +} // end of namespace scribo + +#endif // ! SCRIBO_TABLE_INTERNAL_ALIGN_LINES_HH diff --git a/milena/sandbox/scribo/table/internal/connect_lines.hh b/milena/sandbox/scribo/table/internal/connect_lines.hh new file mode 100644 index 0000000..d510fe2 --- /dev/null +++ b/milena/sandbox/scribo/table/internal/connect_lines.hh @@ -0,0 +1,118 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TABLE_INTERNAL_CONNECT_LINES_HH +# define SCRIBO_TABLE_INTERNAL_CONNECT_LINES_HH + +/// \file scribo/table/connect_vertical_lines.hh +/// +/// Connect vertical lines with aligned rows. + +# include <mln/core/image/image1d.hh> + +# include <mln/data/fill.hh> + +# include <mln/morpho/elementary/dilation.hh> +# include <mln/util/array.hh> +# include <mln/util/couple.hh> + +# include <mln/opt/at.hh> + +# include <scribo/core/macros.hh> + + +namespace scribo +{ + + namespace table + { + + namespace internal + { + + /// Connect vertical and horizontal lines if they are close to each other. + /// + /// ------ ------ + /// ---> | + /// | | + /// | | + /// + /// + /// FIXME: doc arguments. + template <typename P> + void + connect_lines(const util::array<int>& aligned_lines, + util::array< box<P> >& boxes, + unsigned dim, + unsigned dim_size); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename P> + void + connect_lines(const util::array<int>& aligned_lines, + util::array< box<P> >& boxes, + unsigned dim, + unsigned dim_size) + { + trace::entering("scribo::table::internal::connect_lines"); + + image1d<int> l(dim_size); + data::fill(l, -1); + + for_all_elements(i, aligned_lines) + opt::at(l, aligned_lines[i]) = i; + + for (unsigned i = 0; i < settings.max_dist_lines; ++i) + l = morpho::elementary::dilation(l, c2()); + + for_all_components(i, boxes) + { + util::couple<point2d, point2d> cp = central_sites(boxes[i], dim); + if (opt::at(l, cp.first()[dim]) != -1) + boxes[i].pmin()[dim] = aligned_lines[opt::at(l, cp.first()[dim])]; + if (opt::at(l, cp.second()[dim]) != -1) + boxes[i].pmax()[dim] = aligned_lines[opt::at(l, cp.second()[dim])]; + } + + trace::exiting("scribo::table::internal::connect_lines"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + + } // end of namespace scribo::table + +} // end of namespace scribo + + +#endif // ! SCRIBO_TABLE_INTERNAL_CONNECT_LINES_HH diff --git a/milena/sandbox/scribo/table/internal/repair_lines.hh b/milena/sandbox/scribo/table/internal/repair_lines.hh new file mode 100644 index 0000000..87f0933 --- /dev/null +++ b/milena/sandbox/scribo/table/internal/repair_lines.hh @@ -0,0 +1,132 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_TABLE_INTERNAL_REPAIR_LINES_HH +# define SCRIBO_TABLE_INTERNAL_REPAIR_LINES_HH + +/// \file scribo/table/internal/repair_lines.hh +/// +/// Repair lines which have small discontinuities. + +# include <vector> + +# include <mln/core/concept/image.hh> +# include <mln/core/image/extension_val.hh> +# include <mln/core/site_set/box.hh> +# include <mln/data/fill.hh> +# include <mln/util/couple.hh> +# include <mln/util/array.hh> +# include <mln/win/line.hh> +# include <mln/pw/all.hh> + +# include <scribo/core/central_sites.hh> +# include <scribo/table/internal/repair_lines.hh> + + +namespace scribo +{ + + namespace table + { + + namespace internal + { + + +# ifndef MLN_INCLUDE_ONLY + + /// Repair lines which have small discontinuities. + /// FIXME: buggy. Sometimes few lines move or shrink! + template <unsigned axis, typename I> + void + repair_lines(const Image<I>& input_, + util::array<box<mln_site(I)> >& tableboxes) + { + trace::entering("scribo::table::internal::repair_lines"); + + const I& input = exact(input_); + mln_precondition(input.is_valid()); + + typedef mln_site(I) P; + typedef win::line<mln_grid(P), axis, mln_coord(P)> line_t; + + // Initialization + mln_ch_value(I,unsigned) l(input.domain()); + data::fill(l, literal::zero); + for_all_components(i, tableboxes) + { + util::couple<P,P> cp = central_sites(tableboxes[i], axis); + l(cp.first()) = i; + l(cp.second()) = i; + } + + // Repair + extension_val<mln_ch_value(I,unsigned)> l_ext(l, literal::zero); + + util::array<box<P> > result; + std::vector<bool> to_keep(tableboxes.nelements(), true); + + mln_VAR(tbb_ima, extend(l | pw::value(l) != literal::zero, l)); + //FIXME: use a half window, just the bottom of the vertical line. + line_t vl(settings.repair_max_dist); + mln_piter(tbb_ima_t) p(tbb_ima.domain()); + mln_qiter(line_t) q(vl, p); + for_all(p) + for_all(q) + if (l_ext(q) != literal::zero && l_ext(q) != l_ext(p)) + { + to_keep[l_ext(q)] = false; + + tableboxes[l_ext(p)].pmax() = tableboxes[l_ext(q)].pmax(); + + util::couple<P,P> cp = central_sites(tableboxes[l_ext(q)], axis); + l_ext(cp.first()) = l_ext(p); + l_ext(cp.second()) = l_ext(p); + } + + + // Remove merged boxes. + for_all_elements(i, tableboxes) + if (to_keep[i]) + result.append(tableboxes[i]); + + tableboxes = result; + trace::exiting("scribo::table::internal::repair_lines"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::table::internal + + } // end of namespace scribo::table + +} // end of namespace scribo + + +#endif // ! SCRIBO_TABLE_INTERNAL_REPAIR_LINES_HH diff --git a/milena/sandbox/scribo/table/repair_horizontal_lines.hh b/milena/sandbox/scribo/table/repair_horizontal_lines.hh new file mode 100644 index 0000000..7c1edbd --- /dev/null +++ b/milena/sandbox/scribo/table/repair_horizontal_lines.hh @@ -0,0 +1,92 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_TABLE_REPAIR_HORIZONTAL_LINES_HH +# define SCRIBO_TABLE_REPAIR_HORIZONTAL_LINES_HH + +/// \file scribo/table/repair_horizontal_lines.hh +/// +/// Repair horizontal lines which have small discontinuities. + +# include <mln/core/concept/image.hh> +# include <mln/util/couple.hh> +# include <mln/util/array.hh> + +# include <scribo/table/internal/repair_lines.hh> + +namespace scribo +{ + + namespace table + { + + + /// Repair horizontal lines which have small discontinuities. + /*! + * ---- ---- ==> ---------- + * + * \param[in] input Image from which the table bounding + * boxes are extracted. + * \param[in,out] tablesboxes Table line bounding boxes. + * \param[in] max_discontinuity Repair discontinuity which are smaller + * than this value. + */ + template <typename I> + void + repair_horizontal_lines(const Image<I>& input, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > >& tableboxes, + unsigned max_discontinuity); + +# ifndef MLN_INCLUDE_ONLY + + template <typename I> + inline + void + repair_horizontal_lines(const Image<I>& input, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > >& tableboxes, + unsigned max_discontinuity) + { + trace::entering("scribo::table::repair_horizontal_lines"); + mln_precondition(exact(input).is_valid()); + + internal::repair_lines<1>(input, tableboxes.second(), max_discontinuity); + + trace::exiting("scribo::table::repair_horizontal_lines"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::table + +} // end of namespace scribo + + +#endif // ! SCRIBO_TABLE_REPAIR_HORIZONTAL_LINES_HH diff --git a/milena/sandbox/scribo/table/repair_vertical_lines.hh b/milena/sandbox/scribo/table/repair_vertical_lines.hh new file mode 100644 index 0000000..94ae3f1 --- /dev/null +++ b/milena/sandbox/scribo/table/repair_vertical_lines.hh @@ -0,0 +1,98 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_TABLE_REPAIR_VERTICAL_LINES_HH +# define SCRIBO_TABLE_REPAIR_VERTICAL_LINES_HH + +/// \file scribo/table/repair_vertical_lines.hh +/// +/// Repair vertical lines which have small discontinuities. + +# include <mln/core/concept/image.hh> +# include <mln/util/couple.hh> +# include <mln/util/array.hh> + +# include <scribo/table/internal/repair_lines.hh> + +namespace scribo +{ + + namespace table + { + + + /// Repair vertical lines which have small discontinuities. + /*! + * + * | | \n + * | | \n + * ==> | \n + * | | \n + * | | \n + * + * + * \param[in] input Image from which the table bounding + * boxes are extracted. + * \param[in,out] tablesboxes Table line bounding boxes. + * \param[in] max_discontinuity Repair discontinuity which are smaller + * than this value. + */ + template <typename I> + void + repair_vertical_lines(const Image<I>& input, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > >& tblboxes, + unsigned max_discontinuity); + +# ifndef MLN_INCLUDE_ONLY + + template <typename I> + inline + void + repair_vertical_lines(const Image<I>& input, + util::couple<util::array<box<mln_site(I)> >, + util::array<box<mln_site(I)> > >& tblboxes, + unsigned max_discontinuity) + { + trace::entering("scribo::table::repair_vertical_lines"); + mln_precondition(exact(input).is_valid()); + + internal::repair_lines<0>(input, tblboxes.first(), max_discontinuity); + + trace::exiting("scribo::table::repair_vertical_lines"); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::table + +} // end of namespace scribo + + +#endif // ! SCRIBO_TABLE_REPAIR_VERTICAL_LINES_HH diff --git a/milena/sandbox/scribo/text/extract_bboxes.hh b/milena/sandbox/scribo/text/extract_bboxes.hh new file mode 100644 index 0000000..0c88880 --- /dev/null +++ b/milena/sandbox/scribo/text/extract_bboxes.hh @@ -0,0 +1,95 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_EXTRACT_BBOXES_HH +# define SCRIBO_TEXT_GROUPING_EXTRACT_BBOXES_HH + +/// \file scribo/text/grouping/group_with_single_link.hh +/// +/// Extract text bounding boxes from a binary image. + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/neighborhood.hh> +# include <mln/core/site_set/box.hh> +# include <mln/labeling/blobs.hh> +# include <mln/labeling/compute.hh> +# include <mln/util/array.hh> + +# include <scribo/core/component_bboxes.hh> + +namespace scribo +{ + + namespace text + { + + using namespace mln; + + /// Extract text bounding boxes from a binary image. + /// + /// \param[in] input_ A binary image. + /// + /// \return an array of bounding boxes. The first bounding box is + /// the background's. + template <typename I, typename N, typename V> + inline + util::array< box<mln_site(I)> > + extract_bboxes(const Image<I>& input_, + const Neighborhood<N>& nbh, V& nbboxes); + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I, typename N, typename V> + inline + util::array< box<mln_site(I)> > + extract_bboxes(const Image<I>& input_, + const Neighborhood<N>& nbh, V& nbboxes) + { + trace::entering("scribo::text::extract_bboxes"); + + const I& input = exact(input_); + + mlc_equal(mln_value(I), bool)::check(); + mln_precondition(input.is_valid()); + + typedef util::array< box<mln_site(I)> > bboxes_t; + bboxes_t bboxes = component_bboxes(input, nbh, nbboxes); + + trace::exiting("scribo::text::extract_bboxes"); + return bboxes; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_GROUPING_EXTRACT_BBOXES_HH diff --git a/milena/sandbox/scribo/text/grouping/group_from_double_link.hh b/milena/sandbox/scribo/text/grouping/group_from_double_link.hh new file mode 100644 index 0000000..d178a80 --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/group_from_double_link.hh @@ -0,0 +1,108 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_GROUP_FROM_DOUBLE_LINK_HH +# define SCRIBO_TEXT_GROUPING_GROUP_FROM_DOUBLE_LINK_HH + +/// \file scribo/text/grouping/group_from_double_link.hh +/// +/// Link text bounding boxes with their left neighbor. + +# include <mln/core/concept/image.hh> +# include <mln/core/site_set/box.hh> + +# include <mln/accu/bbox.hh> + +# include <mln/util/array.hh> + +# include <scribo/text/grouping/internal/find_root.hh> + +# include <scribo/core/macros.hh> + + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + /// FIXME: Add much more doc! + template <typename P> + util::array< box<P> > + group_from_double_link(const util::array< box<P> >& textbboxes, + util::array<unsigned>& left_link, + util::array<unsigned>& right_link); + +# ifndef MLN_INCLUDE_ONLY + + template <typename P> + inline + util::array< box<P> > + group_from_double_link(const util::array< box<P> >& textbboxes, + util::array<unsigned>& left_link, + util::array<unsigned>& right_link) + { + trace::entering("scribo::text::grouping::group_from_double_link"); + + mln_precondition(left_link.nelements() == right_link.nelements()); + + util::array< accu::bbox<P> > tboxes; + tboxes.resize(textbboxes.nelements()); + for_all_components(i, textbboxes) + { + unsigned nbh = right_link[left_link[i]]; + if (nbh == i) + tboxes[left_link[i]].take(textbboxes[i]); + else + tboxes[i].take(textbboxes[i]); + } + + util::array< box<P> > result; + // component 0, the background, has an invalid box. + result.append(box<P>()); + for_all_components(i, tboxes) + if (tboxes[i].is_valid()) + result.append(tboxes[i]); + + trace::exiting("scribo::text::grouping::group_from_double_link"); + return result; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_GROUPING_GROUP_FROM_DOUBLE_LINK_HH diff --git a/milena/sandbox/scribo/text/grouping/group_from_multiple_links.hh b/milena/sandbox/scribo/text/grouping/group_from_multiple_links.hh new file mode 100644 index 0000000..612fb87 --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/group_from_multiple_links.hh @@ -0,0 +1,154 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_TEXT_GROUPING_GROUP_FROM_MULTIPLE_LINKS_HH +# define SCRIBO_TEXT_GROUPING_GROUP_FROM_MULTIPLE_LINKS_HH + +/// \file scribo/text/grouping/group_from_multiple_links.hh +/// +/// Link text bounding boxes with their neighbors. + +# include <mln/core/concept/graph.hh> +# include <mln/core/concept/image.hh> +# include <mln/core/site_set/box.hh> + +# include <mln/canvas/browsing/depth_first_search.hh> + +# include <mln/accu/bbox.hh> + +# include <mln/pw/all.hh> + +# include <mln/fun/i2v/array.hh> + +# include <mln/util/array.hh> + +# include <scribo/core/macros.hh> + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + /// FIXME: Add much more doc! + template <typename G, typename P> + util::array<box<P> > + group_from_multiple_links(util::array<box<P> >& textbboxes, + const Graph<G>& g_); + +# ifndef MLN_INCLUDE_ONLY + + + namespace internal + { + + /// Functor to be passed to depth_first_search. + /// Map each component vertex with its representative vertex id. + struct map_vertex_to_component_id_functor + { + template <typename G> + void init(const Graph<G>& g) + { + vertextocomp.resize(exact(g).v_nmax(), mln_max(unsigned)); + ncomp = 0; + } + + void final() + {} + + void next_component() + { ++ncomp; } + + void new_component_from_vertex(unsigned id) + { vertextocomp(id) = ncomp; } + + void added_to_queue(unsigned id) + { new_component_from_vertex(id); } + + void process_vertex(unsigned) + {} + + bool to_be_treated(unsigned id) + { return vertextocomp(id) == mln_max(unsigned); } + + bool to_be_queued(unsigned id) + { return to_be_treated(id); } + + unsigned ncomp; + fun::i2v::array<unsigned> vertextocomp; + }; + + } // end of namespace scribo::text::grouping::internal + + + + template <typename G, typename P> + inline + util::array<box<P> > + group_from_multiple_links(util::array<box<P> >& textbboxes, + const Graph<G>& g_) + { + trace::entering("scribo::text::grouping::group_from_multiple_links"); + + const G& g = exact(g_); + + mln_assertion(g.is_valid()); + + internal::map_vertex_to_component_id_functor f; + canvas::browsing::depth_first_search(g, f); + + util::array< accu::bbox<P> > tboxes; + tboxes.resize(textbboxes.nelements()); + for_all_components(i, textbboxes) + tboxes[f.vertextocomp(i)].take(textbboxes[i]); + + util::array< box<P> > result; + // Component 0 - the background has not valid bboxes. + result.append(box<P>()); + for_all_components(i, textbboxes) + if (tboxes[i].is_valid()) + result.append(tboxes[i].to_result()); + + mln_assertion(result.nelements() == f.ncomp); + + trace::exiting("scribo::text::grouping::group_from_multiple_links"); + return result; + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_GROUPING_GROUP_FROM_MULTIPLE_LINKS_HH diff --git a/milena/sandbox/scribo/text/grouping/group_from_single_link.hh b/milena/sandbox/scribo/text/grouping/group_from_single_link.hh new file mode 100644 index 0000000..eccde96 --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/group_from_single_link.hh @@ -0,0 +1,101 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_GROUP_FROM_SINGLE_LINK_HH +# define SCRIBO_TEXT_GROUPING_GROUP_FROM_SINGLE_LINK_HH + +/// \file scribo/text/grouping/group_from_single_link.hh +/// +/// Link text bounding boxes with their left neighbor. + +# include <mln/core/concept/image.hh> +# include <mln/core/site_set/box.hh> + +# include <mln/accu/bbox.hh> + +# include <mln/util/array.hh> + +# include <scribo/text/grouping/internal/find_root.hh> + +# include <scribo/core/macros.hh> + + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + /// FIXME: Add much more doc! + template <typename P> + util::array< box<P> > + group_from_single_link(const util::array< box<P> >& textbboxes, + util::array<unsigned>& link_array); + +# ifndef MLN_INCLUDE_ONLY + + template <typename P> + inline + util::array< box<P> > + group_from_single_link(const util::array< box<P> >& textbboxes, + util::array<unsigned>& link_array) + { + trace::entering("scribo::text::grouping::group_from_single_link"); + + for (unsigned i = 0; i < link_array.nelements(); ++i) + link_array[i] = internal::find_root(link_array, i); + + util::array< accu::bbox<P> > tboxes; + tboxes.resize(textbboxes.nelements()); + for_all_components(i, textbboxes) + tboxes[link_array[i]].take(textbboxes[i]); + + util::array< box<P> > result; + // component 0, the background, has an invalid box. + result.append(box<P>()); + for_all_components(i, tboxes) + if (tboxes[i].is_valid()) + result.append(tboxes[i]); + + trace::exiting("scribo::text::grouping::group_from_single_link"); + return result; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_GROUPING_GROUP_FROM_SINGLE_LINK_HH diff --git a/milena/sandbox/scribo/text/grouping/group_with_multiple_links.hh b/milena/sandbox/scribo/text/grouping/group_with_multiple_links.hh new file mode 100644 index 0000000..2f769fc --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/group_with_multiple_links.hh @@ -0,0 +1,115 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_GROUP_WITH_MULTIPLE_LINKS_HH +# define SCRIBO_TEXT_GROUPING_GROUP_WITH_MULTIPLE_LINKS_HH + +/// \file scribo/text/grouping/group_with_multiple_links.hh +/// +/// Group character bounding boxes with multiple links. + +# include <mln/core/concept/image.hh> + +# include <mln/math/abs.hh> + +# include <mln/util/array.hh> + +# include <scribo/core/macros.hh> +# include <scribo/text/grouping/internal/init_link_array.hh> +# include <scribo/text/grouping/internal/update_link_graph.hh> + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + /// Group character bounding boxes with multiple links. + /// Look up for neighbors on the left of each box. + template <typename I, typename N, typename V> + util::graph + group_with_multiple_links(const Image<I>& input_, + const Neighborhood<N>& nbh, V& nbboxes, + const util::array<box<mln_site(I)> >& textbboxes, + unsigned neighb_max_distance); + +# ifndef MLN_INCLUDE_ONLY + + template <typename I, typename N, typename V> + util::graph + group_with_multiple_links(const Image<I>& input_, + const Neighborhood<N>& nbh, V& nbboxes, + const util::array<box<mln_site(I)> >& textbboxes, + unsigned neighb_max_distance) + { + trace::entering("scribo::text::grouping::group_with_multiple_links"); + + const I& input = exact(input_); + + mlc_equal(mln_value(I), bool)::check(); + mln_assertion(input.is_valid()); + + typedef mln_ch_value(I,V) lbl_t; + lbl_t lbl = labeling::blobs(input, nbh, nbboxes); + + util::graph g(nbboxes.next()); + + for_all_ncomponents(i, nbboxes) + { + unsigned midcol = (textbboxes[i].pmax().col() + - textbboxes[i].pmin().col()) / 2; + int dmax = midcol + neighb_max_distance; + mln_site(I) c = textbboxes[i].center(); + + /// First site on the right of the central site + mln_site(I) p = c + right; + + while (lbl.domain().has(p) && (lbl(p) == literal::zero || lbl(p) == i) + && math::abs(p.col() - c.col()) < dmax) + ++p.col(); + + internal::update_link_graph(lbl, g, p, c, i, dmax); + } + + trace::exiting("scribo::text::grouping::group_with_multiple_links"); + return g; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_GROUPING_GROUP_WITH_MULTIPLE_LINKS_HH diff --git a/milena/sandbox/scribo/text/grouping/group_with_single_left_link.hh b/milena/sandbox/scribo/text/grouping/group_with_single_left_link.hh new file mode 100644 index 0000000..f1a3013 --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/group_with_single_left_link.hh @@ -0,0 +1,135 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_GROUP_WITH_SINGLE_LEFT_LINK_HH +# define SCRIBO_TEXT_GROUPING_GROUP_WITH_SINGLE_LEFT_LINK_HH + +/// \file scribo/text/grouping/group_with_single_left_link.hh +/// +/// Link text bounding boxes with their left neighbor. +/// +/// Merge code with text::grouping::group_with_single_right_link.hh + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/neighborhood.hh> + +# include <mln/math/abs.hh> + +# include <mln/util/array.hh> + +# include <scribo/core/macros.hh> +# include <scribo/text/grouping/internal/init_link_array.hh> +# include <scribo/text/grouping/internal/update_link_array.hh> + +//FIXME: not generic. +# include <mln/core/alias/dpoint2d.hh> + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + /// Map each character bounding box to its left bounding box neighbor + /// if possible. + /// Iterate to the right but link boxes to the left. + /// + /// \return an util::array. Map a bounding box to its left neighbor. + template <typename I, typename N, typename V> + inline + util::array<unsigned> + group_with_single_left_link(const Image<I>& input_, + const Neighborhood<N>& nbh, V& nbboxes, + const util::array< box<mln_site(I)> >& textbboxes, + unsigned neighb_max_distance); + +# ifndef MLN_INCLUDE_ONLY + + template <typename I, typename N, typename V> + inline + util::array<unsigned> + group_with_single_left_link(const Image<I>& input_, + const Neighborhood<N>& nbh_, V& nbboxes, + const util::array< box<mln_site(I)> >& textbboxes, + unsigned neighb_max_distance) + { + trace::entering("scribo::text::grouping::group_with_single_left_link"); + + const I& input = exact(input_); + const N& nbh = exact(nbh_); + + mlc_equal(mln_value(I), bool)::check(); + mln_assertion(input.is_valid()); + mln_precondition(nbh.is_valid()); + + typedef mln_ch_value(I,V) lbl_t; + lbl_t lbl = labeling::blobs(input, nbh, nbboxes); + + util::array<unsigned> left_link(nbboxes.next()); + internal::init_link_array(left_link); + + for_all_ncomponents(i, nbboxes) + { + unsigned midcol = (textbboxes[i].pmax().col() + - textbboxes[i].pmin().col()) / 2; + int dmax = midcol + neighb_max_distance; + mln_site(I) c = textbboxes[i].center(); + + /// + /// Find a neighbor on the right + /// + + ///FIXME: the following code is not generic... + /// First site on the right of the central site + mln_site(I) p = c + right; + + while (lbl.domain().has(p) && (lbl(p) == literal::zero || lbl(p) == i) + && math::abs(p.col() - c.col()) < dmax) + ++p.col(); + + internal::update_link_array(lbl, left_link, p, c, i, dmax); + + } + + trace::exiting("scribo::text::grouping::group_with_single_left_link"); + return left_link; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_GROUPING_GROUP_WITH_SINGLE_LEFT_LINK_HH diff --git a/milena/sandbox/scribo/text/grouping/group_with_single_right_link.hh b/milena/sandbox/scribo/text/grouping/group_with_single_right_link.hh new file mode 100644 index 0000000..7f1eeac --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/group_with_single_right_link.hh @@ -0,0 +1,135 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_GROUP_WITH_SINGLE_RIGHT_LINK_HH +# define SCRIBO_TEXT_GROUPING_GROUP_WITH_SINGLE_RIGHT_LINK_HH + +/// \file scribo/text/grouping/group_with_single_right_link.hh +/// +/// Link text bounding boxes with their right neighbor. +/// +/// Merge code with text::grouping::group_with_single_right_link.hh + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/neighborhood.hh> + +# include <mln/math/abs.hh> + +# include <mln/util/array.hh> + +# include <scribo/core/macros.hh> +# include <scribo/text/grouping/internal/init_link_array.hh> +# include <scribo/text/grouping/internal/update_link_array.hh> + +//FIXME: not generic. +# include <mln/core/alias/dpoint2d.hh> + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + /// Map each character bounding box to its right bounding box neighbor + /// if possible. + /// Iterate to the right but link boxes to the right. + /// + /// \return an util::array. Map a bounding box to its right neighbor. + template <typename I, typename N, typename V> + inline + util::array<unsigned> + group_with_single_right_link(const Image<I>& input_, + const Neighborhood<N>& nbh, V& nbboxes, + const util::array< box<mln_site(I)> >& textbboxes, + unsigned neighb_max_distance); + +# ifndef MLN_INCLUDE_ONLY + + template <typename I, typename N, typename V> + inline + util::array<unsigned> + group_with_single_right_link(const Image<I>& input_, + const Neighborhood<N>& nbh_, V& nbboxes, + const util::array< box<mln_site(I)> >& textbboxes, + unsigned neighb_max_distance) + { + trace::entering("scribo::text::grouping::group_with_single_right_link"); + + const I& input = exact(input_); + const N& nbh = exact(nbh_); + + mlc_equal(mln_value(I), bool)::check(); + mln_assertion(input.is_valid()); + mln_precondition(nbh.is_valid()); + + typedef mln_ch_value(I,V) lbl_t; + lbl_t lbl = labeling::blobs(input, nbh, nbboxes); + + util::array<unsigned> right_link(nbboxes.next()); + internal::init_link_array(right_link); + + for_all_ncomponents(i, nbboxes) + { + unsigned midcol = (textbboxes[i].pmax().col() + - textbboxes[i].pmin().col()) / 2; + int dmax = midcol + neighb_max_distance; + mln_site(I) c = textbboxes[i].center(); + + /// + /// Find a neighbor on the left + /// + + ///FIXME: the following code is not generic... + /// First site on the left of the central site + mln_site(I) p = c + left; + + while (lbl.domain().has(p) && (lbl(p) == literal::zero || lbl(p) == i) + && math::abs(p.col() - c.col()) < dmax) + --p.col(); + + internal::update_link_array(lbl, right_link, p, c, i, dmax); + + } + + trace::exiting("scribo::text::grouping::group_with_single_right_link"); + return right_link; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_GROUPING_GROUP_WITH_SINGLE_RIGHT_LINK_HH diff --git a/milena/sandbox/scribo/photo.cc b/milena/sandbox/scribo/text/grouping/internal/find_root.hh similarity index 52% copy from milena/sandbox/scribo/photo.cc copy to milena/sandbox/scribo/text/grouping/internal/find_root.hh index a173750..0b1e307 100644 --- a/milena/sandbox/scribo/photo.cc +++ b/milena/sandbox/scribo/text/grouping/internal/find_root.hh @@ -1,4 +1,5 @@ -// Copyright (C) 2008 EPITA Research and Development Laboratory +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) // // This file is part of the Olena Library. This library is free // software; you can redistribute it and/or modify it under the terms @@ -26,39 +27,54 @@ // Public License. -#include "demat.hh" +#ifndef SCRIBO_TEXT_GROUPING_INTERNAL_FIND_ROOT_LINK_ARRAY_HH +# define SCRIBO_TEXT_GROUPING_INTERNAL_FIND_ROOT_LINK_ARRAY_HH -int main(int argc, char*argv[]) +/// \file scribo/text/grouping/internal/find_root.hh +/// +/// Find root in a parent array arrays. + + +# include <mln/util/array.hh> + + +namespace scribo { - using namespace mln; - using value::int_u8; - if (argc < 2) + namespace text { - std::cout << argv[0] << " <in.pbm> <out.pgm> <l> <bbox_larger> <bbox_distance> <min_comp_nsites>" << std::endl - << std::endl << std::endl - << std::endl - << "==========" - << std::endl << std::endl - << "<in.pbm> B/W inverted input image." - << std::endl << std::endl -/* << "<out.ppm> RGB8 output image." - << std::endl << std::endl - << "<bbox_distance> Maximum distance between character bounding boxes. Used for bbox grouping." - << std::endl << std::endl - << "<min_comp_nsites> Minimum site count of a character/text component." - << std::endl - << " If a component have a site count lesser than this value, it is erased." - << std::endl << std::endl - << std::endl*/ - << "==========" - << std::endl << std::endl - << "HINT: compile with -DNOUT to avoid debug images." - << std::endl << std::endl; - return 1; - } - - scribo::demat_photo(argv); - - return 0; -} + + namespace grouping + { + + namespace internal + { + + /// Find root in a parent array arrays. + unsigned + find_root(util::array<unsigned>& parent, unsigned x); + +# ifndef MLN_INCLUDE_ONLY + + inline + unsigned + find_root(util::array<unsigned>& parent, unsigned x) + { + if (parent[x] == x) + return x; + else + return parent[x] = find_root(parent, parent[x]); + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping::internal + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + + +#endif // ! SCRIBO_TEXT_GROUPING_INTERNAL_FIND_ROOT_LINK_ARRAY_HH diff --git a/milena/sandbox/scribo/photo.cc b/milena/sandbox/scribo/text/grouping/internal/init_link_array.hh similarity index 52% rename from milena/sandbox/scribo/photo.cc rename to milena/sandbox/scribo/text/grouping/internal/init_link_array.hh index a173750..ba197f9 100644 --- a/milena/sandbox/scribo/photo.cc +++ b/milena/sandbox/scribo/text/grouping/internal/init_link_array.hh @@ -1,4 +1,5 @@ -// Copyright (C) 2008 EPITA Research and Development Laboratory +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) // // This file is part of the Olena Library. This library is free // software; you can redistribute it and/or modify it under the terms @@ -26,39 +27,51 @@ // Public License. -#include "demat.hh" +#ifndef SCRIBO_TEXT_GROUPING_INTERNAL_INIT_LINK_ARRAY_HH +# define SCRIBO_TEXT_GROUPING_INTERNAL_INIT_LINK_ARRAY_HH -int main(int argc, char*argv[]) +/// \file scribo/text/grouping/internal/init_link_array.hh +/// +/// Initialize a link array. + + +# include <mln/util/array.hh> + + +namespace scribo { - using namespace mln; - using value::int_u8; - if (argc < 2) + namespace text { - std::cout << argv[0] << " <in.pbm> <out.pgm> <l> <bbox_larger> <bbox_distance> <min_comp_nsites>" << std::endl - << std::endl << std::endl - << std::endl - << "==========" - << std::endl << std::endl - << "<in.pbm> B/W inverted input image." - << std::endl << std::endl -/* << "<out.ppm> RGB8 output image." - << std::endl << std::endl - << "<bbox_distance> Maximum distance between character bounding boxes. Used for bbox grouping." - << std::endl << std::endl - << "<min_comp_nsites> Minimum site count of a character/text component." - << std::endl - << " If a component have a site count lesser than this value, it is erased." - << std::endl << std::endl - << std::endl*/ - << "==========" - << std::endl << std::endl - << "HINT: compile with -DNOUT to avoid debug images." - << std::endl << std::endl; - return 1; - } - - scribo::demat_photo(argv); - - return 0; -} + + namespace grouping + { + + namespace internal + { + + /// Initialize a link array. + void init_link_array(util::array<unsigned>& link_array); + +# ifndef MLN_INCLUDE_ONLY + + inline + void + init_link_array(util::array<unsigned>& link_array) + { + for (unsigned i = 0; i < link_array.nelements(); ++i) + link_array[i] = i; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping::internal + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + + +#endif // ! SCRIBO_TEXT_GROUPING_INTERNAL_INIT_LINK_ARRAY_HH diff --git a/milena/sandbox/scribo/text/grouping/internal/update_link_array.hh b/milena/sandbox/scribo/text/grouping/internal/update_link_array.hh new file mode 100644 index 0000000..242d18f --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/internal/update_link_array.hh @@ -0,0 +1,93 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_INTERNAL_UPDATE_LINK_ARRAY_HH +# define SCRIBO_TEXT_GROUPING_INTERNAL_UPDATE_LINK_ARRAY_HH + +/// \file scribo/text/grouping/internal/update_link_array.hh +/// +/// Update a lookup table if a neighbor is found on the right of +/// the current bbox. + + +# include <mln/core/concept/image.hh> +# include <mln/util/array.hh> + + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + namespace internal + { + + /// Update the lookup table \p link_array if a neighbor is found + /// on the right of the current bbox. + template <typename I> + void + update_link_array(const Image<I>& lbl, util::array<unsigned>& link_array, + const mln_site(I)& p, const mln_site(I)& c, + unsigned i, int dmax); + +# ifndef MLN_INCLUDE_ONLY + + template <typename I> + inline + void + update_link_array(const Image<I>& lbl_, util::array<unsigned>& link_array, + const mln_site(I)& p, const mln_site(I)& c, + unsigned i, int dmax) + { + const I& lbl = exact(lbl_); + + mlc_is_a(mln_value(I), mln::value::Symbolic)::check(); + mln_assertion(lbl.is_valid()); + + if (lbl.domain().has(p) && lbl(p) != literal::zero && lbl(p) != i + && (math::abs(p.col() - c.col())) < dmax && link_array[lbl(p)] == lbl(p)) + link_array[lbl(p)] = i; + } + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping::internal + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + + +#endif // ! SCRIBO_TEXT_GROUPING_INTERNAL_UPDATE_LINK_ARRAY_HH diff --git a/milena/sandbox/scribo/text/grouping/internal/update_link_graph.hh b/milena/sandbox/scribo/text/grouping/internal/update_link_graph.hh new file mode 100644 index 0000000..e31c5e8 --- /dev/null +++ b/milena/sandbox/scribo/text/grouping/internal/update_link_graph.hh @@ -0,0 +1,92 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + + +#ifndef SCRIBO_TEXT_GROUPING_INTERNAL_UPDATE_LINK_GRAPH_HH +# define SCRIBO_TEXT_GROUPING_INTERNAL_UPDATE_LINK_GRAPH_HH + +/// \file scribo/text/grouping/internal/update_link_array.hh +/// +/// Update graph edges if a valid neighbor is found. + +# include <mln/core/concept/image.hh> +# include <mln/util/graph.hh> + +namespace scribo +{ + + namespace text + { + + namespace grouping + { + + namespace internal + { + + /// Update graph edges if a valid neighbor is found. + template <typename I> + void + update_link_graph(Image<I>& lbl_, util::graph& g, + const mln_site(I)& p, const mln_site(I)& c, + unsigned i, int dmax); + + +# ifndef MLN_INCLUDE_ONLY + + + template <typename I> + inline + void + update_link_graph(Image<I>& lbl_, util::graph& g, + const mln_site(I)& p, const mln_site(I)& c, + unsigned i, int dmax) + { + const I& lbl = exact(lbl_); + + mlc_is_a(mln_value(I), mln::value::Symbolic)::check(); + mln_precondition(exact(lbl).is_valid()); + + if (lbl.domain().has(p) && lbl(p) != 0u && lbl(p) != i + && (math::abs(p.col() - c.col())) < dmax) + g.add_edge(lbl(p), i); + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text::grouping::internal + + } // end of namespace scribo::text::grouping + + } // end of namespace scribo::text + +} // end of namespace scribo + + +#endif // ! SCRIBO_TEXT_GROUPING_INTERNAL_UPDATE_LINK_GRAPH_HH diff --git a/milena/sandbox/scribo/text/recognition.hh b/milena/sandbox/scribo/text/recognition.hh new file mode 100644 index 0000000..98c3b67 --- /dev/null +++ b/milena/sandbox/scribo/text/recognition.hh @@ -0,0 +1,142 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of the Olena Library. This library is free +// software; you can redistribute it and/or modify it under the terms +// of the GNU General Public License version 2 as published by the +// Free Software Foundation. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this library; see the file COPYING. If not, write to +// the Free Software Foundation, 51 Franklin Street, Fifth Floor, +// Boston, MA 02111-1307, USA. +// +// As a special exception, you may use this file as part of a free +// software library without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to +// produce an executable, this file does not by itself cause the +// resulting executable to be covered by the GNU General Public +// License. This exception does not however invalidate any other +// reasons why the executable file might be covered by the GNU General +// Public License. + +#ifndef SCRIBO_TEXT_RECOGNITION_HH +# define SCRIBO_TEXT_RECOGNITION_HH + +/// \file scribo/text/recognition.hh +/// +/// Passes the text bounding boxes to an OCR (Tesseract). + +# include <mln/core/concept/image.hh> +# include <mln/core/concept/neighborhood.hh> +# include <mln/core/site_set/box.hh> +# include <mln/util/array.hh> +# include <mln/labeling/blobs.hh> +# include <mln/data/fill.hh> +# include <mln/pw/all.hh> +# include <mln/debug/put_words.hh> + +# include <scribo/core/macros.hh> + +# include <tesseract/baseapi.h> + + + +namespace scribo +{ + + namespace text + { + + /// Passes the text bboxes to Tesseract (OCR) and store the result in + /// an image of characters. + /// + /// \param[in] input_ image from where the text bboxes are extracted. + /// \param[in] nbh_ The neighborhood used to label \p input_. + /// \param[in] nbboxes The value type used in the labeled image. + /// \param[in] textbboxes array of text bounding boxes. + /// \param[in] language the language which should be recognized by Tesseract. + /// (fra, en, ...) + /// + /// \return An image of characters. + /// + /// FIXME: For each text bbox, we create a new image. We may like to avoid that. + /// FIXME: Do not store the result in an image? + template <typename I, typename N, typename V> + mln_ch_value(I,char) + text_recognition(const Image<I>& input_, + const Neighborhood<N>& nbh_, V& nbboxes, + const util::array< box<mln_site(I)> >& textbboxes, + const char *language); + + +# ifndef MLN_INCLUDE_ONLY + + template <typename I, typename N, typename V> + mln_ch_value(I,char) + text_recognition(const Image<I>& input_, + const Neighborhood<N>& nbh_, V& nbboxes, + const util::array< box<mln_site(I)> >& textbboxes, + const char *language) + { + trace::entering("scribo::text::recognition"); + + mlc_equal(mln_value(I), bool)::check(); + const I& input = exact(input_); + const N& nbh = exact(nbh_); + mln_precondition(input.is_valid()); + mln_precondition(nbh.is_valid()); + + mln_ch_value(I,V) lbl = labeling::blobs(input, nbh, nbboxes); + + /// Use text bboxes with Tesseract + TessBaseAPI::InitWithLanguage(NULL, NULL, language, NULL, false, 0, NULL); + mln_ch_value(I,char) txt(input.domain()); + data::fill(txt, ' '); + + for_all_components(i, textbboxes) + { + if (textbboxes[i].is_valid()) + { + mln_ch_value(I,bool) b(textbboxes[i], 0); + data::fill(b, false); + data::fill((b | (pw::value(lbl) == pw::cst(i))).rw(), true); + + char* s = TessBaseAPI::TesseractRect( + (unsigned char*) b.buffer(), + sizeof (bool), // Pixel size. + b.ncols() * sizeof (bool), // Row_offset + 0, // Left + 0, // Top + b.ncols(), // n cols + b.nrows()); // n rows + + + + mln_site(I) p = textbboxes[i].center(); + p.col() -= (textbboxes[i].pmax().col() + - textbboxes[i].pmin().col()) / 2; + if (s != 0) + debug::put_word(txt, p, s); + free(s); + } + } + + trace::exiting("scribo::text::recognition"); + return text; + } + + +# endif // ! MLN_INCLUDE_ONLY + + } // end of namespace scribo::text + +} // end of namespace scribo + +#endif // ! SCRIBO_TEXT_RECOGNITION_HH -- 1.5.6.5