summaryrefslogtreecommitdiffstats
path: root/depends/libnbtplusplus/include/io/stream_writer.h
blob: 8938b73bedc5f08dc3b2e8230671fc1d8be5ea8d (plain)
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
/*
 * libnbt++ - A library for the Minecraft Named Binary Tag format.
 * Copyright (C) 2013, 2015  ljfa-ag
 *
 * This file is part of libnbt++.
 *
 * libnbt++ is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * libnbt++ is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with libnbt++.  If not, see <http://www.gnu.org/licenses/>.
 */
#ifndef STREAM_WRITER_H_INCLUDED
#define STREAM_WRITER_H_INCLUDED

#include "tag.h"
#include "endian_str.h"
#include <iosfwd>

#include "nbt++_export.h"

namespace nbt
{
namespace io
{

/* Not sure if that is even needed
///Exception that gets thrown when writing is not successful
class output_error : public std::runtime_error
{
    using std::runtime_error::runtime_error;
};*/

/**
* @brief Writes a named tag into the stream, including the tag type
* @param key the name of the tag
* @param t the tag
* @param os the stream to write to
* @param e the byte order of the written data. The Java edition
* of Minecraft uses Big Endian, the Pocket edition uses Little Endian
*/
NBT___EXPORT void write_tag(const std::string& key, const tag& t, std::ostream& os, endian::endian e = endian::big);

/**
 * @brief Helper class for writing NBT tags to output streams
 *
 * Can be reused to write multiple tags
 */
class NBT___EXPORT stream_writer
{
public:
    ///Maximum length of an NBT string (16 bit unsigned)
    static constexpr size_t max_string_len = UINT16_MAX;
    ///Maximum length of an NBT list or array (32 bit signed)
    static constexpr uint32_t max_array_len = INT32_MAX;

    /**
     * @param os the stream to write to
     * @param e the byte order of the written data. The Java edition
     * of Minecraft uses Big Endian, the Pocket edition uses Little Endian
     */
    explicit stream_writer(std::ostream& os, endian::endian e = endian::big) noexcept:
        os(os), endian(e)
    {}

    ///Returns the stream
    std::ostream& get_ostr() const { return os; }
    ///Returns the byte order
    endian::endian get_endian() const { return endian; }

    /**
     * @brief Writes a named tag into the stream, including the tag type
     */
    void write_tag(const std::string& key, const tag& t);

    /**
     * @brief Writes the given tag's payload into the stream
     */
    void write_payload(const tag& t) { t.write_payload(*this); }

    /**
     * @brief Writes a tag type to the stream
     */
    void write_type(tag_type tt) { write_num(static_cast<int8_t>(tt)); }

    /**
     * @brief Writes a binary number to the stream
     */
    template<class T>
    void write_num(T x);

    /**
     * @brief Writes an NBT string to the stream
     *
     * An NBT string consists of two bytes indicating the length, followed by
     * the characters encoded in modified UTF-8.
     * @throw std::length_error if the string is too long for NBT
     */
    void write_string(const std::string& str);

private:
    std::ostream& os;
    const endian::endian endian;
};

template<class T>
void stream_writer::write_num(T x)
{
    endian::write(os, x, endian);
}

}
}

#endif // STREAM_WRITER_H_INCLUDED