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
|
/*************************************************************************
* odil - Copyright (C) Universite de Strasbourg
* Distributed under the terms of the CeCILL-B license, as published by
* the CEA-CNRS-INRIA. Refer to the LICENSE file or to
* http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
* for details.
************************************************************************/
#include <boost/property_tree/xml_parser.hpp>
#include <pybind11/pybind11.h>
#include <boost/version.hpp>
#include "odil/DataSet.h"
#include "odil/StringStream.h"
#include "odil/xml_converter.h"
#include "opaque_types.h"
#include "type_casters.h"
namespace
{
std::string as_xml(std::shared_ptr<odil::DataSet> data_set, bool pretty_print)
{
auto const xml = odil::as_xml(data_set);
std::ostringstream stream;
#if BOOST_VERSION >= 105600
typedef boost::property_tree::xml_writer_settings<std::string> SettingsType;
#else
typedef boost::property_tree::xml_writer_settings<char> SettingsType;
#endif
SettingsType settings;
if(pretty_print)
{
settings = SettingsType('\t', 1);
}
boost::property_tree::write_xml(stream, xml, settings);
return stream.str();
}
}
void wrap_xml_converter(pybind11::module & m)
{
using namespace pybind11;
using namespace odil;
m.def(
"as_xml",
[](std::shared_ptr<odil::DataSet> data_set, bool pretty_print)
{
auto const xml = odil::as_xml(data_set);
std::string data;
OStringStream stream(data);
#if BOOST_VERSION >= 105600
typedef boost::property_tree::xml_writer_settings<std::string> SettingsType;
#else
typedef boost::property_tree::xml_writer_settings<char> SettingsType;
#endif
SettingsType settings;
if(pretty_print)
{
settings = SettingsType('\t', 1);
}
boost::property_tree::write_xml(stream, xml, settings);
return data;
});
m.def(
"from_xml",
[](std::string const & xml_string)
{
IStringStream stream(&xml_string[0], xml_string.size());
boost::property_tree::ptree xml;
boost::property_tree::read_xml(stream, xml);
return as_dataset(xml);
}
);
}
|
