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
|
#include "Reader.hpp"
#include <boost/algorithm/string/predicate.hpp>
#ifdef MAEPARSER_HAVE_BOOST_IOSTREAMS
#include <boost/iostreams/device/file.hpp>
#include <boost/iostreams/filter/gzip.hpp>
#include <boost/iostreams/filtering_stream.hpp>
#endif
#include <fstream>
#include <iostream>
#include <sstream>
#include <utility>
using boost::algorithm::ends_with;
#ifdef MAEPARSER_HAVE_BOOST_IOSTREAMS
using boost::iostreams::file_source;
using boost::iostreams::filtering_istream;
#endif
namespace schrodinger
{
namespace mae
{
Reader::Reader(FILE* file, size_t buffer_size)
{
m_mae_parser.reset(new MaeParser(file, buffer_size));
}
Reader::Reader(const std::shared_ptr<std::istream>& stream, size_t buffer_size)
{
m_mae_parser.reset(new MaeParser(stream, buffer_size));
}
Reader::Reader(const std::string& fname, size_t buffer_size)
{
const auto ios_mode = std::ios_base::in | std::ios_base::binary;
std::shared_ptr<std::istream> stream;
if (ends_with(fname, ".maegz") || ends_with(fname, ".mae.gz")) {
#ifdef MAEPARSER_HAVE_BOOST_IOSTREAMS
auto* gzip_stream = new filtering_istream();
gzip_stream->push(boost::iostreams::gzip_decompressor());
gzip_stream->push(file_source(fname, ios_mode));
stream.reset(static_cast<std::istream*>(gzip_stream));
#else
std::stringstream ss;
ss << "Unable to open " << fname << " for reading, "
<< "as maeparser was compiled without boost::iostreams support";
throw std::runtime_error(ss.str());
#endif
} else {
auto* file_stream = new std::ifstream(fname, ios_mode);
stream.reset(static_cast<std::istream*>(file_stream));
}
if (stream->fail()) {
std::stringstream ss;
ss << "Failed to open file \"" << fname << "\" for reading operation.";
throw std::runtime_error(ss.str());
}
m_mae_parser.reset(new MaeParser(stream, buffer_size));
}
Reader::Reader(std::shared_ptr<MaeParser> mae_parser)
: m_mae_parser(std::move(mae_parser))
{
}
std::shared_ptr<Block> Reader::next(const std::string& outer_block_name)
{
std::shared_ptr<Block> block;
do {
m_mae_parser->whitespace();
block = m_mae_parser->outerBlock();
} while (block != nullptr && block->getName() != outer_block_name);
return block;
}
} // namespace mae
} // namespace schrodinger
|
