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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
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
136
137
138
139
140
141
142
143
144
145
146
147
148
|
// ************************************************************************************************
//
// BornAgain: simulate and fit reflection and scattering
//
//! @file Device/IO/IOFactory.cpp
//! @brief Implements data import/export functions.
//!
//! @homepage http://www.bornagainproject.org
//! @license GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
// ************************************************************************************************
#include "Device/IO/IOFactory.h"
#include "Base/Util/Assert.h"
#include "Device/Data/Datafield.h"
#include "Device/IO/DiffUtil.h"
#include "Device/IO/ImportSettings.h"
#include "Device/IO/ReadReflectometry.h"
#include "Device/IO/ReadWrite2DTable.h"
#include "Device/IO/ReadWriteINT.h"
#include "Device/IO/ReadWriteNicos.h"
#include "Device/IO/ReadWriteTiff.h"
#include "Device/IO/ZipUtil.h"
#include <sstream>
#include <stdexcept>
IO::Filetype1D IO::filename2type1D(const std::string& fname)
{
const std::string ext = ZipUtil::uncompressedExtension(fname);
if (ext == ".int")
return bornagain1D;
if (ext == ".mft")
return mft;
return csv1D;
}
IO::Filetype2D IO::filename2type2D(const std::string& fname)
{
const std::string ext = ZipUtil::uncompressedExtension(fname);
if (ext == ".int")
return bornagain2D;
if (ext == ".001")
return nicos2D;
if (ext == ".tif" || ext == ".tiff") // as in function isNativeBinary
return tiff;
return csv2D;
}
Datafield IO::readData1D(const std::string& fname, Filetype1D ftype,
const ImportSettings1D* importSettings)
{
ASSERT(ftype != unknown1D);
std::stringstream s = ZipUtil::file2stream(fname);
if (ftype == csv1D) {
if (!importSettings)
throw std::runtime_error("No import settings given for 'other legacy' data");
return Util::RW::readReflectometryTable(s, *importSettings);
}
if (importSettings)
throw std::runtime_error("Import settings given in spite of fully specified data format");
if (ftype == mft)
return Util::RW::readMotofit(s);
if (ftype == bornagain1D)
return Util::RW::readBAInt(s);
ASSERT_NEVER;
}
Datafield IO::readData1D(const std::string& fname)
{
return readData1D(fname, filename2type1D(fname));
}
Datafield IO::readData2D(const std::string& fname, Filetype2D ftype,
const ImportSettings2D* importSettings)
{
ASSERT(ftype != unknown2D);
std::stringstream s = ZipUtil::file2stream(fname);
if (ftype == csv2D)
return Util::RW::read2DTable(s, importSettings);
if (ftype == bornagain2D)
return Util::RW::readBAInt(s);
if (ftype == nicos2D)
return Util::RW::readNicos(s);
#ifdef BA_TIFF_SUPPORT
if (ftype == tiff)
return Util::RW::readTiff(s);
#endif
ASSERT_NEVER;
}
Datafield IO::readData2D(const std::string& fname)
{
return readData2D(fname, filename2type2D(fname));
}
void IO::writeDatafield(const Datafield& data, const std::string& fname)
{
const std::string ext = ZipUtil::uncompressedExtension(fname);
try {
std::stringstream s;
if (ext == ".int")
Util::RW::writeBAInt(data, s);
#ifdef BA_TIFF_SUPPORT
else if (ext == ".tiff" || ext == ".tif")
Util::RW::writeTiff(data, s);
#endif
else
Util::RW::write2DTable(data, s);
ZipUtil::stream2file(fname, s);
} catch (const std::exception& ex) {
throw std::runtime_error("Failed writing to " + fname + ": " + ex.what());
}
}
bool IO::Test::dataMatchesFile(const Datafield& data, const std::string& refFileName, double tol)
{
std::unique_ptr<Datafield> refDat;
try {
refDat =
std::make_unique<Datafield>(IO::readData2D(refFileName, IO::Filetype2D::bornagain2D));
} catch (const std::exception& ex) {
std::cerr << "File comparison: Could not read reference data from file " << refFileName
<< std::endl;
std::cerr << "Error message: " << ex.what() << std::endl;
return false;
}
return DiffUtil::checkRelativeDifference(data, *refDat, tol);
}
|
