Milena (Olena)  User documentation 2.0a Id
 All Classes Namespaces Functions Variables Typedefs Enumerator Groups Pages
pbm/load.hh
1 // Copyright (C) 2009 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_IO_PBM_LOAD_HH
27 # define MLN_IO_PBM_LOAD_HH
28 
33 
34 
35 # include <iostream>
36 # include <fstream>
37 # include <string>
38 
39 # include <mln/core/image/image2d.hh>
40 # include <mln/core/image/image3d.hh>
41 # include <mln/io/pnm/load_header.hh>
42 
43 
44 namespace mln
45 {
46 
47  namespace io
48  {
49 
50  namespace pbm
51  {
52 
53 
60  void load(image2d<bool>& ima,
61  const std::string& filename);
62 
69  image2d<bool> load(const std::string& filename);
70 
71 
72 # ifndef MLN_INCLUDE_ONLY
73 
74  namespace internal
75  {
76 
78  template <typename I>
79  inline
80  void load_ascii(std::ifstream& file, I& ima)
81  {
82  mln_fwd_piter(I) p(ima.domain());
83  for_all(p)
84  {
85  unsigned char value;
86  file >> value;
87 
88  mln_assertion(value == '0' || value == '1');
89  ima(p) = (value == '0'); // In pbm, '0' means 'white' so 'object', thus 'true'!
90  }
91  }
92 
93 
95  template <typename I>
96  inline
97  void load_raw_2d(std::ifstream& file, I& ima)
98  {
99  point2d p = point2d(0, ima.domain().pmin().col());
100  typedef mln_value(I) V;
101  const mln_deduce(I, site, coord)
102  min_row = geom::min_row(ima),
103  max_row = geom::max_row(ima),
104  min_col = geom::min_col(ima),
105  max_col = geom::max_col(ima);
106 
107  char c;
108  int i;
109  for (p.row() = min_row; p.row() <= max_row; ++p.row())
110  {
111  i = 0;
112  for (p.col() = min_col; p.col() <= max_col; ++p.col())
113  {
114  if (i % 8 == 0)
115  file.read((char*)(&c), 1);
116  ima(p) = !(c & 128);
117  c = static_cast<char>(c * 2);
118  ++i;
119  }
120  }
121  }
122 
123 
124  } // end of namespace mln::io::internal
125 
126 
127  inline
128  image2d<bool> load(const std::string& filename)
129  {
130  trace::entering("mln::io::pbm::load");
131  std::ifstream file(filename.c_str());
132  if (! file)
133  {
134  std::cerr << "error: file '" << filename
135  << "' not found!";
136  abort();
137  }
138  char type;
139  int nrows, ncols;
140  io::pnm::read_header('1', '4', file, type, nrows, ncols);
141 
142  image2d<bool> ima(nrows, ncols);
143  if (type == '4')
144  internal::load_raw_2d(file, ima);
145  else
146  if (type == '1')
147  internal::load_ascii(file, ima);
148 
149  trace::exiting("mln::io::pbm::load");
150 
151  return ima;
152  }
153 
154 
155  inline
156  void load(image2d<bool>& ima,
157  const std::string& filename)
158  {
159  ima = load(filename);
160  }
161 
162 
163 # endif // ! MLN_INCLUDE_ONLY
164 
165  } // end of namespace mln::io::pbm
166 
167  } // end of namespace mln::io
168 
169 } // end of namespace mln
170 
171 
172 #endif // ! MLN_IO_PBM_LOAD_HH