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
|
// 2010 © Václav Šmilauer <eudoxos@arcig.cz>
#pragma once
#include <boost/algorithm/string.hpp>
#include <boost/archive/basic_archive.hpp>
#include <boost/archive/binary_iarchive.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/archive/codecvt_null.hpp>
#include <boost/archive/xml_iarchive.hpp>
#include <boost/archive/xml_oarchive.hpp>
#include <boost/iostreams/device/file.hpp>
#include <boost/iostreams/filter/bzip2.hpp>
#include <boost/iostreams/filter/gzip.hpp>
#include <boost/iostreams/filtering_stream.hpp>
#include <boost/math/special_functions/nonfinite_num_facets.hpp>
#include <locale>
#include <string>
namespace yade {
/* Utility template functions for (de)serializing objects using boost::serialization from/to streams or files.
Includes boost::math::nonfinite_num_{put,get} for gracefully handling nan's and inf's.
*/
struct ObjectIO {
// tell whether given filename looks like XML
static bool isXmlFilename(const std::string f)
{
return boost::algorithm::ends_with(f, ".xml") || boost::algorithm::ends_with(f, ".xml.bz2") || boost::algorithm::ends_with(f, ".xml.gz");
}
// save to given stream and archive format
template <class T, class oarchive> static void save(std::ostream& ofs, const std::string& objectTag, T& object)
{
std::locale default_locale(std::locale::classic(), new boost::archive::codecvt_null<char>);
std::locale locale2(default_locale, new boost::math::nonfinite_num_put<char>);
ofs.imbue(locale2);
oarchive oa(ofs, boost::archive::no_codecvt);
oa << boost::serialization::make_nvp(objectTag.c_str(), object);
ofs.flush();
}
// load from given stream and archive format
template <class T, class iarchive> static void load(std::istream& ifs, const std::string& objectTag, T& object)
{
std::locale default_locale(std::locale::classic(), new boost::archive::codecvt_null<char>);
std::locale locale2(default_locale, new boost::math::nonfinite_num_get<char>);
ifs.imbue(locale2);
iarchive ia(ifs, boost::archive::no_codecvt);
ia >> boost::serialization::make_nvp(objectTag.c_str(), object);
}
// save to given file, guessing compression and XML/binary from extension
template <class T> static void save(const std::string fileName, const std::string& objectTag, T& object)
{
boost::iostreams::filtering_ostream out;
if (boost::algorithm::ends_with(fileName, ".bz2")) out.push(boost::iostreams::bzip2_compressor());
if (boost::algorithm::ends_with(fileName, ".gz")) out.push(boost::iostreams::gzip_compressor());
out.push(boost::iostreams::file_sink(fileName));
if (!out.good()) throw std::runtime_error("Error opening file " + fileName + " for writing.");
if (isXmlFilename(fileName)) save<T, boost::archive::xml_oarchive>(out, objectTag, object);
else
save<T, boost::archive::binary_oarchive>(out, objectTag, object);
}
// load from given file, guessing compression and XML/binary from extension
template <class T> static void load(const std::string& fileName, const std::string& objectTag, T& object)
{
boost::iostreams::filtering_istream in;
if (boost::algorithm::ends_with(fileName, ".bz2")) in.push(boost::iostreams::bzip2_decompressor());
if (boost::algorithm::ends_with(fileName, ".gz")) in.push(boost::iostreams::gzip_decompressor());
in.push(boost::iostreams::file_source(fileName));
if (!in.good()) throw std::runtime_error("Error opening file " + fileName + " for reading.");
if (isXmlFilename(fileName)) load<T, boost::archive::xml_iarchive>(in, objectTag, object);
else
load<T, boost::archive::binary_iarchive>(in, objectTag, object);
}
};
} // namespace yade
|
