Milena (Olena)  User documentation 2.0a Id
 All Classes Namespaces Functions Variables Typedefs Enumerator Groups Pages
histo/equalize.hh
1 // Copyright (C) 2010 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 
26 #ifndef MLN_HISTO_EQUALIZATION_HH
27 # define MLN_HISTO_EQUALIZATION_HH
28 
29 # include <mln/core/concept/image.hh>
30 # include <mln/histo/all.hh>
31 
35 
36 namespace mln
37 {
38 
39  namespace histo
40  {
41 
46  template <typename I>
47  mln_concrete(I)
48  equalize(const Image<I>& input);
49 
50 
51 # ifndef MLN_INCLUDE_ONLY
52 
53  template <typename I>
54  mln_concrete(I)
55  equalize(const Image<I>& input_)
56  {
57  trace::entering("histo::equalize");
58 
59  const I& input = exact(input_);
60  mln_concrete(I) out_image;
61 
62  array<mln_value(I)> histogram = compute(input);
63  array<mln_value(I)> histogram_correction;
64 
65  unsigned cumulation = 0;
66  int number_of_pixels = input.nsites();
67  //int number_of_colors=histogram.nvalues();
68  //int number_of_colors=mln_card(mln_value(I));
69  int max_color = mln_max(mln_value(I));
70  mln_piter(I) p(input.domain());
71 
72  mln_viter(mln::value::set<mln_value(I)>) v(histogram.vset());
73  for_all(v)
74  if (histogram(v) != 0)
75  {
76  cumulation += histogram(v);
77  histogram_correction(v) = (/*number_of_colors-1*/max_color)
78  * cumulation / number_of_pixels;
79  }
80 
81  initialize(out_image, input);
82  for_all(p)
83  out_image(p) = histogram_correction[input(p)];
84 
85  trace::exiting("histo::equalize");
86  return out_image;
87  }
88 
89 # endif // ! MLN_INCLUDE_ONLY
90 
91  } // end of namespace mln::histo
92 
93 } // end of namespace mln
94 
95 
96 
97 #endif // ! MLN_HISTO_EQUALIZATION_HH