$extrastylesheet
Olena  User documentation 2.1
An Image Processing Platform
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Groups Pages
show_links_single_left_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_left_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_left_link_debug_functor
58  : primitive::link::internal::link_single_dmax_ratio_base<L,
59  primitive::link::internal::dmax_default,
60  single_left_link_debug_functor<I, L> >
61  {
62  typedef single_left_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,
66  dmax_fun_t,
67  self_t> super_;
68 
69  public:
70  typedef mln_site(L) P;
71 
72  single_left_link_debug_functor(const I& input,
73  const component_set<L>& components,
74  float dmax_ratio)
75  : super_(components, anchor::Horizontal, dmax_fun_t(dmax_ratio))
76  {
77  output_ = data::convert(value::rgb8(), input);
78  scribo::draw::bounding_boxes(output_, components, literal::blue);
79  mln_postcondition(output_.is_valid());
80  }
81 
82  void validate_link_(unsigned current_object,
83  const P& start_point,
84  const P& p,
85  anchor::Type anchor)
86  {
87  mln::draw::line(output_, start_point, p, literal::green);
88 
89  super_::validate_link_(current_object, start_point, p, anchor);
90  }
91 
92 
93 
94  void invalidate_link_(unsigned current_object,
95  const P& start_point,
96  const P& p,
97  anchor::Type anchor)
98  {
99  if (output_.domain().has(p))
100  mln::draw::line(output_, start_point, p, literal::red);
101  else
102  {
103  P tmp = p;
104  ++tmp.col();
105  mln::draw::line(output_, start_point, tmp, literal::red);
106  }
107 
108  super_::invalidate_link_(current_object, start_point, p, anchor);
109  }
110 
111 
112 
113  void compute_next_site_(P& p)
114  {
115  --p.col();
116  }
117 
118 
119  image2d<value::rgb8> output_;
120  };
121 
122 
123 }
124 
125 
126 const char *args_desc[][2] =
127 {
128  { "input.pbm", "A binary image. True for objects and False for the "
129  "background." },
130  { "max_nbh_dist_ratio", "Size ratio defining the maximum lookup distance."
131  "(common value : 3)" },
132  {0, 0}
133 };
134 
135 
136 int main(int argc, char* argv[])
137 {
138  using namespace scribo;
139  using namespace scribo::primitive::internal;
140  using namespace mln;
141 
142  if (argc != 4)
143  return scribo::debug::usage(argv,
144  "Show sucessful/unsuccessful left links "
145  "between components.",
146  "input.pbm max_nbh_dist_ratio output.ppm",
147  args_desc);
148 
149  typedef image2d<bool> I;
150  I input;
151  io::pbm::load(input, argv[1]);
152 
153  // Finding objects.
154  value::label_16 nbboxes;
155  typedef image2d<value::label_16> L;
156  component_set<L> components
157  = scribo::primitive::extract::components(input, c8(), nbboxes);
158 
159  // Write debug image.
160  single_left_link_debug_functor<I, L>
161  functor(input, components, atof(argv[2]));
162  primitive::link::compute(functor, anchor::MassCenter);
163 
164  io::ppm::save(functor.output_, argv[3]);
165 }