--- milena/sandbox/ChangeLog | 4 + milena/sandbox/modules/icdar/input_to_words.cc | 225 ++++++++++++++++++++++++ 2 files changed, 229 insertions(+), 0 deletions(-) create mode 100644 milena/sandbox/modules/icdar/input_to_words.cc diff --git a/milena/sandbox/ChangeLog b/milena/sandbox/ChangeLog index cf48639..6e960f6 100644 --- a/milena/sandbox/ChangeLog +++ b/milena/sandbox/ChangeLog @@ -1,3 +1,7 @@ +2009-07-07 Guillaume Lazzara <lazzara@lrde.epita.fr> + + * modules/icdar/input_to_words.cc: Cleanup code for ICDAR. + 2009-07-07 Thierry Geraud <thierry.geraud@lrde.epita.fr> New directories for icdar code release. diff --git a/milena/sandbox/modules/icdar/input_to_words.cc b/milena/sandbox/modules/icdar/input_to_words.cc new file mode 100644 index 0000000..55ef385 --- /dev/null +++ b/milena/sandbox/modules/icdar/input_to_words.cc @@ -0,0 +1,225 @@ +// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) +// +// This file is part of Olena. +// +// Olena is free software: you can redistribute it and/or modify it under +// the terms of the GNU General Public License as published by the Free +// Software Foundation, version 2 of the License. +// +// Olena 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 Olena. If not, see <http://www.gnu.org/licenses/>. +// +// As a special exception, you may use this file as part of a free +// software project 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 <mln/core/image/image2d.hh> +#include <mln/core/image/imorph/tr_image.hh> +#include <mln/core/image/dmorph/image_if.hh> +#include <mln/core/concept/function.hh> +#include <mln/core/alias/neighb2d.hh> +#include <mln/core/var.hh> + +#include <mln/fun/v2b/threshold.hh> +#include <mln/data/transform.hh> +#include <mln/data/fill.hh> +#include <mln/data/paste.hh> +#include <mln/labeling/wrap.hh> +#include <mln/labeling/colorize.hh> +#include <mln/morpho/watershed/superpose.hh> +#include <mln/logical/not.hh> +#include <mln/world/binary_2d/subsample.hh> +#include <mln/morpho/watershed/flooding.hh> +#include <mln/morpho/elementary/dilation.hh> +#include <mln/morpho/closing/area.hh> +#include <mln/morpho/closing/structural.hh> + +#include <mln/transform/distance_front.hh> + +#include <mln/make/w_window2d_int.hh> + +#include <mln/io/pbm/all.hh> +#include <mln/io/pgm/all.hh> +#include <mln/io/ppm/all.hh> + +#include <mln/value/int_u8.hh> + + + +namespace mln +{ + + + // Forward declaration. + template <typename P> struct to_larger; + + + template <typename P> + struct to_smaller + : Function_v2v < to_smaller<P> > + { + typedef P result; + typedef to_larger<P> invert; + + P operator()(const P& p) const + { + P tmp(p); + + for (unsigned i = 0; i < P::dim; ++i) + tmp[i] /= 2; + + return tmp; + } + + + invert inv() const + { + return to_larger<P>(); + } + }; + + + template <typename P> + struct to_larger + : Function_v2v < to_larger<P> > + { + typedef P result; + typedef to_smaller<P> invert; + + P operator()(const P& p) const + { + P tmp(p); + + for (unsigned i = 0; i < P::dim; ++i) + tmp[i] *= 2; + + return tmp; + } + + + invert inv() const + { + return to_smaller<P>(); + } + + }; + + +} + + +int main(int argc, char *argv[]) +{ + using namespace mln; + + typedef value::int_u8 L; + typedef image2d<bool> I; + + + // Parameters. + + // Related to the first structural closing. + const unsigned + height = 5, + width = 11; + + // Related to the distance map computation. + const unsigned + l_area = 500; + + // end of Parameters. + + + + I input; + io::pbm::load(input, argv[1]); + + + /// Reduction and invertion. + image2d<L> + small_gl = world::binary_2d::subsample(input, 2); + I small = data::transform(small_gl, fun::v2b::threshold<L>(191)); + logical::not_inplace(small); + +#ifdef LOG + io::pbm::save(small, "small.pbm"); +#endif // ! LOG + + + /// Structural closing. + I clo = morpho::closing::structural(small, + win::rectangle2d(height, + width)); + +#ifdef LOG + io::pbm::save(clo, "tmp_clo.pbm"); +#endif // ! LOG + + + /// Distance map computation. + int ww[] = { 00, 11, 0, 11, 0, + 11, 7, 5, 7, 11, + 00, 5, 0, 5, 0, + 11, 7, 5, 7, 11, + 00, 11, 0, 11, 0 }; + + typedef value::int_u<12> int_u12; + image2d<int_u12> + dmap = transform::distance_front(clo, + c4(), make::w_window2d_int(ww), + mln_max(int_u12)), + dmap_ = morpho::closing::area(dmap, c4(), l_area); + + +#ifdef LOG + io::pgm::save(labeling::wrap(dmap), "tmp_dmap.pgm"); + io::pgm::save(labeling::wrap(dmap_), "tmp_dmap_.pgm"); +#endif // ! LOG + + /// Watershed. + L n_words; + image2d<L> ws_ = morpho::watershed::flooding(dmap_, + c4(), + n_words); + + /// Remove non-significant regional minima. + image2d<L> ws = morpho::elementary::dilation(ws_, + c4()); + +#ifdef LOG + io::pgm::save(labeling::wrap(ws), "tmp_ws.pgm"); + io::pgm::save(labeling::wrap(ws_), "tmp_ws_.pgm"); + + /// Adjust the domain of the watershed image. + mln_VAR(ws_large_, transposed_image(input.domain(), ws_, + to_larger<point2d>())); + io::ppm::save(morpho::watershed::superpose(input, ws_large_), + "tmp_ws_on_input.ppm"); +#endif // ! LOG + + + /// Adjust the domain of the watershed image. + mln_VAR(ws_large, transposed_image(input.domain(), ws, + to_larger<point2d>())); + + image2d<L> output(input.domain()); + data::fill(output, literal::zero); + + data::paste(ws_large | (pw::value(input) == pw::cst(false)), output); + + io::pgm::save(output, "ws_large.pgm"); + io::ppm::save(labeling::colorize(output, n_words), "ws_large.ppm"); + +} + -- 1.5.6.5