save.hh 3.94 KB
Newer Older
1
2
// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE)
//
3
// This file is part of Olena.
4
//
5
6
7
8
9
// 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,
10
11
12
13
14
// 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
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
21
22
23
24
// 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.
25

26
27
#ifndef MLN_IO_DUMP_SAVE_HH
# define MLN_IO_DUMP_SAVE_HH
28

29
/// \file
30
31
///
/// Save a Milena image by dumping its data to a file.
Guillaume Lazzara's avatar
Guillaume Lazzara committed
32
///
Guillaume Lazzara's avatar
Guillaume Lazzara committed
33
/// \todo handle endianness.
34
35
36
37
38
39
40
41

# include <iostream>
# include <fstream>

# include <mln/core/concept/image.hh>
# include <mln/core/box_runstart_piter.hh>
# include <mln/core/pixel.hh>
# include <mln/data/memcpy_.hh>
42
# include <mln/trait/value_.hh>
43

Guillaume Lazzara's avatar
Guillaume Lazzara committed
44

45
46
47
48
49
50
namespace mln
{

  namespace io
  {

51
    namespace dump
52
53
54
55
56
57
    {

      /// Save a Milena image by dumping its data to a file.
      ///
      /// \param[in] ima_ The image to save.
      /// \param[in] filename the destination.
Guillaume Lazzara's avatar
Guillaume Lazzara committed
58
      //
59
60
61
62
      template <typename I>
      void save(const Image<I>& ima_, const std::string& filename);


Guillaume Lazzara's avatar
Guillaume Lazzara committed
63

64
65
# ifndef MLN_INCLUDE_ONLY

Guillaume Lazzara's avatar
Guillaume Lazzara committed
66

67
68
69
70
71
72
73
74
      namespace internal
      {

        template <typename I>
        inline
        void save_header(const I& ima,
			 std::ofstream& file)
	{
75
76
77
78
	  // Milena's file type
	  file << "milena/dump" << std::endl;

	  // Dimension
79
	  typedef mln_site(I) P;
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
	  file << P::dim << std::endl;

	  // Image size.
	  typedef algebra::vec<P::dim, unsigned> vec_t;
	  vec_t size = ima.domain().pmax() - ima.domain().pmin();
	  for (unsigned i = 0; i < P::dim - 1; ++i)
	    file << size[i] + 1 << " ";
	  file << size[P::dim - 1] + 1 << std::endl;

	  // Value type name
	  // WARNING: value type name limited to 255 characters...
	  file << mln_trait_value_name(mln_value(I)) << std::endl;

	  // Empty line - may be used for a new information.
	  file << std::endl;

	  // Empty line - may be used for a new information.
	  file << std::endl;
98

99
	  // Pmin
100
101
102
	  mln_site(I) p = ima.domain().pmin();
	  file.write((char*) (&p), sizeof (P));

103
	  // Pmax
104
105
106
107
108
109
110
111
112
	  p = ima.domain().pmax();
	  file.write((char*) (&p), sizeof (P));
	}


	template <typename I>
        inline
        void save_data(I& ima, std::ofstream& file)
	{
Guillaume Lazzara's avatar
Guillaume Lazzara committed
113
	  // Handle padding.
114
115
116
117
118
119
120
121
122
123
	  unsigned data_size = sizeof (mln_value(I)) + sizeof (mln_value(I)) % 2;

	  mln_box_runstart_piter(I) p(ima.domain());
	  for_all(p)
	  {
	    pixel<I> src(ima, p);
	    file.write((char*) (&src.val()), p.run_length() * data_size);
	  }
	}

124
      } // end of namespace mln::io::dump::internal
125
126


Guillaume Lazzara's avatar
Guillaume Lazzara committed
127
128
129

      // Facade

130
131
132
      template <typename I>
      void save(const Image<I>& ima_, const std::string& filename)
      {
133
	trace::entering("mln::io::dump::save");
134
135
136
137
138
139
140
141
142
143
144
145
146

	const I& ima = exact(ima_);

	std::ofstream file(filename.c_str());
	if (! file)
	{
	  std::cerr << "error: cannot open file '" << filename << "'!";
	  abort();
	}

	internal::save_header(ima, file);
	internal::save_data(ima, file);

Guillaume Lazzara's avatar
Guillaume Lazzara committed
147
148
149

	file.close();

150
	trace::exiting("mln::io::dump::save");
151
152
153
154
155
      }


# endif // ! MLN_INCLUDE_ONLY

156
    } // end of namespace mln::io::dump
157
158
159
160
161

  } // end of namespace mln::io

} // end of namespace mln

162
#endif // ! MLN_IO_DUMP_SAVE_HH