$extrastylesheet
Olena  User documentation 2.1
An Image Processing Platform
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Groups Pages
recorder-wst.cc
1 // Copyright (C) 2011 EPITA Research and Development Laboratory (LRDE)
2 //
3 // This file is part of Olena.
4 //
5 // Olena is free software: you can redistribute it and/or modify it under
6 // the terms of the GNU General Public License as published by the Free
7 // Software Foundation, version 2 of the License.
8 //
9 // Olena is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 // General Public License for more details.
13 //
14 // You should have received a copy of the GNU General Public License
15 // along with Olena. If not, see <http://www.gnu.org/licenses/>.
16 //
17 // As a special exception, you may use this file as part of a free
18 // software project without restriction. Specifically, if other files
19 // instantiate templates or use macros or inline functions from this
20 // file, or you compile this file and link it with other files to produce
21 // an executable, this file does not by itself cause the resulting
22 // executable to be covered by the GNU General Public License. This
23 // exception does not however invalidate any other reasons why the
24 // executable file might be covered by the GNU General Public License.
25 
36 
37 #include <string>
38 
39 #include <mln/core/image/image2d.hh>
40 #include <mln/core/alias/neighb2d.hh>
41 
42 #include <mln/value/int_u8.hh>
43 #include <mln/value/label_8.hh>
44 #include <mln/value/rgb8.hh>
45 #include <mln/literal/colors.hh>
46 
47 #include <mln/morpho/gradient.hh>
48 #include <mln/morpho/closing/area.hh>
49 #include <mln/morpho/watershed/flooding.hh>
50 
51 #include <mln/test/predicate.hh>
52 #include <mln/pw/value.hh>
53 
54 #include <mln/labeling/colorize.hh>
55 
56 #include <mln/io/pgm/load.hh>
57 #include <mln/io/ppm/save.hh>
58 
59 #include "apps/morphers/recorder.hh"
60 
61 #include "apps/data.hh"
62 
63 
64 int main()
65 {
66  using namespace mln;
67  using mln::value::int_u8;
68  using mln::value::label_8;
69 
70  typedef image2d<int_u8> I;
71  I lena = io::pgm::load<int_u8>(MLN_IMG_DIR "/tiny.pgm");
72 
73  // Gradient.
74  I grad = morpho::gradient(lena, win_c4p());
75  // Simplification of the gradient.
76  unsigned lambda = 5;
77  I closed_grad = morpho::closing::area(grad, c4(), lambda);
78  // Attach recorder to closed gradient.
79  decorated_image< I, recorder<I> > closed_grad_rec = record(closed_grad);
80 
81  // Watershed transform.
82  typedef label_8 L;
83  L nbasins;
84  typedef mln_ch_value_(I, L) J;
85  decorated_image< J, recorder<J> > wst_rec =
86  morpho::watershed::flooding(closed_grad_rec, c4(), nbasins);
87  std::cout << nbasins << std::endl;
88 
89  // Dump recorded frames.
90  /* FIXME: Hand-made (it partly duplicates ppm::save() for
91  decorated_image's). We should be able to process sequences of
92  images. */
93  for (size_t i = 0; i < wst_rec.decoration().sequence.size(); ++i)
94  {
95  J frame = wst_rec.decoration().sequence[i];
96 
97  // Skip ``empty'' frames.
98  if (test::predicate(frame.domain(), pw::value(frame) == 0))
99  continue;
100 
101  std::stringstream s;
102  s << std::setfill ('0') << std::setw (6) << i;
103  io::ppm::save(labeling::colorize(value::rgb8(), frame, nbasins),
104  std::string("lena-wst") + s.str() + ".ppm");
105  }
106 }