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