$extrastylesheet
Olena  User documentation 2.1
An Image Processing Platform
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Groups Pages
show_links_single_right_dmax_ratio.cc
1 // Copyright (C) 2009, 2010, 2011 EPITA Research and Development
2 // Laboratory (LRDE)
3 //
4 // This file is part of Olena.
5 //
6 // Olena is free software: you can redistribute it and/or modify it under
7 // the terms of the GNU General Public License as published by the Free
8 // Software Foundation, version 2 of the License.
9 //
10 // Olena is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 // General Public License for more details.
14 //
15 // You should have received a copy of the GNU General Public License
16 // along with Olena. If not, see <http://www.gnu.org/licenses/>.
17 //
18 // As a special exception, you may use this file as part of a free
19 // software project without restriction. Specifically, if other files
20 // instantiate templates or use macros or inline functions from this
21 // file, or you compile this file and link it with other files to produce
22 // an executable, this file does not by itself cause the resulting
23 // executable to be covered by the GNU General Public License. This
24 // exception does not however invalidate any other reasons why the
25 // executable file might be covered by the GNU General Public License.
26 
27 #include <iostream>
28 
29 #include <mln/core/image/image2d.hh>
30 #include <mln/core/alias/neighb2d.hh>
31 
32 #include <mln/data/convert.hh>
33 
34 #include <mln/value/rgb8.hh>
35 #include <mln/value/label_16.hh>
36 #include <mln/literal/colors.hh>
37 
38 #include <mln/io/pbm/load.hh>
39 #include <mln/io/ppm/save.hh>
40 
41 #include <mln/draw/line.hh>
42 
43 #include <scribo/primitive/extract/components.hh>
44 #include <scribo/primitive/link/with_single_right_link_dmax_ratio.hh>
45 #include <scribo/primitive/link/internal/link_single_dmax_ratio_base.hh>
46 #include <scribo/primitive/link/compute.hh>
47 
48 #include <scribo/draw/bounding_boxes.hh>
49 
50 #include <scribo/debug/usage.hh>
51 
52 
53 namespace scribo
54 {
55 
56  template <typename I, typename L>
57  struct single_right_link_debug_functor
58  : primitive::link::internal::link_single_dmax_ratio_base<L,
59  primitive::link::internal::dmax_default,
60  single_right_link_debug_functor<I, L> >
61  {
62  typedef single_right_link_debug_functor<I, L> self_t;
63  typedef primitive::link::internal::dmax_default dmax_fun_t;
64  typedef
65  primitive::link::internal::link_single_dmax_ratio_base<L, dmax_fun_t, self_t> super_;
66 
67  public:
68  typedef mln_site(L) P;
69 
70  single_right_link_debug_functor(const I& input,
71  const component_set<L>& components,
72  float dmax_ratio)
73  : super_(components, anchor::Horizontal, dmax_fun_t(dmax_ratio))
74  {
75  output_ = data::convert(value::rgb8(), input);
76  scribo::draw::bounding_boxes(output_, components, literal::blue);
77  mln_postcondition(output_.is_valid());
78  }
79 
80  void validate_link_(unsigned current_object,
81  const P& start_point,
82  const P& p,
83  anchor::Type anchor)
84  {
85  mln::draw::line(output_, start_point, p, literal::green);
86 
87  super_::validate_link_(current_object, start_point, p, anchor);
88  }
89 
90 
91 
92  void invalidate_link_(unsigned current_object,
93  const P& start_point,
94  const P& p,
95  anchor::Type anchor)
96  {
97  if (output_.domain().has(p))
98  mln::draw::line(output_, start_point, p, literal::red);
99  else
100  {
101  P tmp = p;
102  --tmp.col();
103  mln::draw::line(output_, start_point, tmp, literal::red);
104  }
105 
106  super_::invalidate_link_(current_object, start_point, p, anchor);
107  }
108 
109 
110 
111  void compute_next_site_(P& p)
112  {
113  ++p.col();
114  }
115 
116 
117  image2d<value::rgb8> output_;
118  };
119 
120 
121 }
122 
123 
124 const char *args_desc[][2] =
125 {
126  { "input.pbm", "A binary image. True for objects and False for the background." },
127  { "max_nbh_dist_ratio", "Size ratio defining the maximum lookup distance. (common value : 3)" },
128  {0, 0}
129 };
130 
131 
132 int main(int argc, char* argv[])
133 {
134  using namespace scribo;
135  using namespace scribo::primitive::internal;
136  using namespace mln;
137 
138  if (argc != 4)
139  return scribo::debug::usage(argv,
140  "Show sucessful/unsuccessful right links "
141  "between components.",
142  "input.pbm max_nbh_dist_ratio output.ppm",
143  args_desc);
144 
145  typedef image2d<bool> I;
146  I input;
147  io::pbm::load(input, argv[1]);
148 
149  // Finding objects.
150  value::label_16 nbboxes;
151  typedef image2d<value::label_16> L;
152  component_set<L> components
153  = scribo::primitive::extract::components(input, c8(), nbboxes);
154 
155  // Write debug image.
156  single_right_link_debug_functor<I, L>
157  functor(input, components, atof(argv[2]));
158  primitive::link::compute(functor, anchor::MassCenter);
159 
160  io::ppm::save(functor.output_, argv[3]);
161 }