save.hh 4.02 KB
Newer Older
1
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
// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE)
//
// This file is part of the Olena Library.  This library is free
// software; you can redistribute it and/or modify it under the terms
// of the GNU General Public License version 2 as published by the
// Free Software Foundation.
//
// This library 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 this library; see the file COPYING.  If not, write to
// the Free Software Foundation, 51 Franklin Street, Fifth Floor,
// Boston, MA 02111-1307, USA.
//
// As a special exception, you may use this file as part of a free
// software library 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.

28
29
#ifndef MLN_IO_DUMP_SAVE_HH
# define MLN_IO_DUMP_SAVE_HH
30

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

# 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>
43
# include <mln/trait/value_.hh>
44
45
46
47
48
49
50

namespace mln
{

  namespace io
  {

51
    namespace dump
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
    {

      /// Save a Milena image by dumping its data to a file.
      ///
      /// \param[in] ima_ The image to save.
      /// \param[in] filename the destination.
      template <typename I>
      void save(const Image<I>& ima_, const std::string& filename);


# ifndef MLN_INCLUDE_ONLY

      namespace internal
      {

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

	  // Dimension
76
	  typedef mln_site(I) P;
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
	  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;
95

96
	  // Pmin
97
98
99
	  mln_site(I) p = ima.domain().pmin();
	  file.write((char*) (&p), sizeof (P));

100
	  // Pmax
101
102
103
104
105
106
107
108
109
	  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
110
	  // Handle padding.
111
112
113
114
115
116
117
118
119
120
	  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);
	  }
	}

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


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

	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);

141
	trace::exiting("mln::io::dump::save");
142
143
144
145
146
      }


# endif // ! MLN_INCLUDE_ONLY

147
    } // end of namespace mln::io::dump
148
149
150
151
152

  } // end of namespace mln::io

} // end of namespace mln

153
#endif // ! MLN_IO_DUMP_SAVE_HH