bboxes_enlarged_image.hh 4.21 KB
Newer Older
1
// Copyright (C) 2010, 2013 EPITA Research and Development Laboratory (LRDE)
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
//
// This file is part of Olena.
//
// Olena is free software: you can redistribute it and/or modify it under
// the terms of the GNU General Public License as published by the Free
// Software Foundation, version 2 of the License.
//
// Olena is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Olena.  If not, see <http://www.gnu.org/licenses/>.
//
// As a special exception, you may use this file as part of a free
// software project without restriction.  Specifically, if other files
// instantiate templates or use macros or inline functions from this
// file, or you compile this file and link it with other files to produce
// an executable, this file does not by itself cause the resulting
// executable to be covered by the GNU General Public License.  This
// exception does not however invalidate any other reasons why the
// executable file might be covered by the GNU General Public License.

#ifndef SCRIBO_DEBUG_BBOXES_ENLARGED_IMAGE_HH
# define SCRIBO_DEBUG_BBOXES_ENLARGED_IMAGE_HH

/// \file
///
/// \brief Compute an image of enlarged component bounding boxes.


# include <mln/core/concept/image.hh>
# include <mln/data/convert.hh>
# include <mln/value/rgb8.hh>
# include <mln/literal/colors.hh>
# include <mln/draw/box.hh>

# include <scribo/core/line_set.hh>

Guillaume Lazzara's avatar
Guillaume Lazzara committed
42
# include <scribo/text/look_like_text_lines.hh>
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102


namespace scribo
{

  namespace debug
  {

    using namespace mln;


    /// \brief Compute an image of enlarged component bounding boxes.
    ///
    /// This check whether each line looks like a text line. If it is
    /// a text line, its extended bounding box is drawn, otherwise, it
    /// is normal bounding box.
    ///
    /// This rountine uses scribo::internal::looks_like_a_text_line to
    /// check if a component looks like a text line.
    ///
    ///
    /// \param[in] input An image convertible towards a color image.
    /// \param[in] lines A line set.
    /// \param[in] text_value The color used to draw bounding boxes
    ///                       of components looking like a text line.
    /// \param[in] non_text_value The color used to draw bounding
    ///                           boxes of components NOT looking like
    ///                           a text line.
    ///
    /// \return A color image.
    //
    template <typename I, typename L>
    mln_ch_value(I,value::rgb8)
    bboxes_enlarged_image(const Image<I>& input,
			  const line_set<L>& lines,
			  const value::rgb8& text_value,
			  const value::rgb8& non_text_value);

    /// \overload
    /// text_value is set to literal::green.
    /// non_text_value is set to literal::red.
    template <typename I, typename L>
    mln_ch_value(I,value::rgb8)
    bboxes_enlarged_image(const Image<I>& input,
			  const line_set<L>& lines);


# ifndef MLN_INCLUDE_ONLY


    template <typename I, typename L>
    mln_ch_value(I,value::rgb8)
    bboxes_enlarged_image(const Image<I>& input,
			  const line_set<L>& lines,
			  const value::rgb8& text_value,
			  const value::rgb8& non_text_value)
    {
      trace::entering("scribo::debug::bboxes_enlarged_image");
      mln_precondition(exact(input).is_valid());

103
      mln_ch_value(I,value::rgb8) output = data::convert(value::rgb8(), input);
104
105

      for_all_lines(l, lines)
106
	if (! lines(l).is_hidden())
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
	{
	  if (text::internal::looks_like_a_text_line(lines(l)))
	    mln::draw::box(output, lines(l).ebbox(), text_value);
	  else
	    mln::draw::box(output, lines(l).bbox(), non_text_value);
	}

      trace::exiting("scribo::debug::bboxes_enlarged_image");
      return output;
    }


    template <typename I, typename L>
    mln_ch_value(I,value::rgb8)
    bboxes_enlarged_image(const Image<I>& input,
			  const line_set<L>& lines)
    {
      return bboxes_enlarged_image(input, lines,
				   literal::green, literal::red);
    }

# endif // ! MLN_INCLUDE_ONLY

  } // end of namespace scribo::debug

} // end of namespace scribo


#endif // ! SCRIBO_DEBUG_BBOXES_ENLARGED_IMAGE_HH