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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
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
|
#ifndef IMG_IO_H_
#define IMG_IO_H_
// minimal input/output
#include <iostream>
#include <fstream>
#include <limits>
namespace img {
template<int Channels, typename ScalarType, bool Safe>
inline bool saveNativeFormat(const Image<Channels,ScalarType,Safe> &image, const char *filename)
{
assert(image.isValid());
if(Safe){
if(!image.isValid()) throw ImageException("Invalid image");
}
using namespace std;
ofstream output (filename, ios::out|ios::binary);
if (output.is_open()) {
int channels=Channels;
output.write(reinterpret_cast<const char*>(& channels), sizeof(int));
int scalartype=0;
if(typeid(ScalarType) == typeid(float)) {
scalartype=1;
} else if(typeid(ScalarType) == typeid(double)) {
scalartype=2;
} else {
assert(0);
}
output.write(reinterpret_cast<const char*>(& scalartype), sizeof(int));
int width=image.width();
output.write(reinterpret_cast<const char*>(& width), sizeof(int));
int height=image.height();
output.write(reinterpret_cast<const char*>(& height), sizeof(int));
output.write(reinterpret_cast<const char*>(& image.attributes), sizeof(ImgAttributes<ScalarType>));
output.write(reinterpret_cast<const char*>(image.dataValues()), sizeof(ScalarType) * image.dataValuesSize());
output.flush();
output.close();
return true;
}
if(Safe)
throw ImageException("Unable to save file");
return false;
}
template<int Channels, typename ScalarType, bool Safe>
inline bool openNativeFormat(const char *filename, Image<Channels,ScalarType,Safe> &image)
{
using namespace std;
ifstream input (filename, ios::in|ios::binary);
if (input.is_open()) {
int channels;
input.read(reinterpret_cast<char*>(&channels), sizeof(int));
assert(channels==Channels);
int scalartype;
input.read(reinterpret_cast<char*>(&scalartype), sizeof(int));
if(typeid(ScalarType) == typeid(float)) {
assert(scalartype==1);
} else if(typeid(ScalarType) == typeid(double)) {
assert(scalartype==2);
} else {
assert(0);
}
int width;
input.read(reinterpret_cast<char*>(&width), sizeof(int));
int height;
input.read(reinterpret_cast<char*>(&height), sizeof(int));
image.setZero(width,height);
input.read(reinterpret_cast<char*>(& image.attributes), sizeof(ImgAttributes<ScalarType>));
input.read(reinterpret_cast<char*>(image.dataValues()), sizeof(ScalarType) * image.dataValuesSize());
input.close();
return true;
}
if(Safe)
throw ImageException("Unable to open file");
return false;
}
} //end namespace img
#endif /*IMG_IO_H_*/
|
