diff options
Diffstat (limited to 'src/NBT')
-rw-r--r-- | src/NBT/ByteArrayTag.hpp | 20 | ||||
-rw-r--r-- | src/NBT/ByteTag.hpp | 2 | ||||
-rw-r--r-- | src/NBT/DoubleTag.hpp | 13 | ||||
-rw-r--r-- | src/NBT/FloatTag.hpp | 9 | ||||
-rw-r--r-- | src/NBT/IntArrayTag.hpp | 20 | ||||
-rw-r--r-- | src/NBT/IntTag.hpp | 9 | ||||
-rw-r--r-- | src/NBT/ListTag.hpp | 5 | ||||
-rw-r--r-- | src/NBT/LongTag.hpp | 13 | ||||
-rw-r--r-- | src/NBT/ShortTag.hpp | 7 | ||||
-rw-r--r-- | src/NBT/StringTag.hpp | 13 | ||||
-rw-r--r-- | src/NBT/Tag.cpp | 9 |
11 files changed, 41 insertions, 79 deletions
diff --git a/src/NBT/ByteArrayTag.hpp b/src/NBT/ByteArrayTag.hpp index ee0d7d1..2081759 100644 --- a/src/NBT/ByteArrayTag.hpp +++ b/src/NBT/ByteArrayTag.hpp @@ -38,22 +38,26 @@ class ByteArrayTag : public Tag { private: friend class Tag; - std::vector<uint8_t> value; + uint32_t len; + const uint8_t *value; ByteArrayTag(Buffer *buffer) { - uint32_t len = uint32_t(buffer->get()) << 24; - len |= uint32_t(buffer->get()) << 16; - len |= uint32_t(buffer->get()) << 8; - len |= uint32_t(buffer->get()); - - value.resize(len); - buffer->getData(value.data(), len); + len = buffer->get32(); + value = buffer->get(len); } public: virtual Type getType() const { return Type::ByteArray; } + + uint32_t getLength() const { + return len; + } + + const uint8_t & operator[](size_t i) const { + return value[i]; + } }; } diff --git a/src/NBT/ByteTag.hpp b/src/NBT/ByteTag.hpp index cc4aba9..64bdaa1 100644 --- a/src/NBT/ByteTag.hpp +++ b/src/NBT/ByteTag.hpp @@ -39,7 +39,7 @@ private: uint8_t value; ByteTag(Buffer *buffer) { - value = buffer->get(); + value = buffer->get8(); } public: diff --git a/src/NBT/DoubleTag.hpp b/src/NBT/DoubleTag.hpp index 327dcf5..dace193 100644 --- a/src/NBT/DoubleTag.hpp +++ b/src/NBT/DoubleTag.hpp @@ -36,17 +36,10 @@ class DoubleTag : public Tag { private: friend class Tag; + const uint8_t *ptr; + DoubleTag(Buffer *buffer) { - uint64_t value; - - value = uint64_t(buffer->get()) << 56; - value |= uint64_t(buffer->get()) << 48; - value |= uint64_t(buffer->get()) << 40; - value |= uint64_t(buffer->get()) << 32; - value |= uint64_t(buffer->get()) << 24; - value |= uint64_t(buffer->get()) << 16; - value |= uint64_t(buffer->get()) << 8; - value |= uint64_t(buffer->get()); + ptr = buffer->get(8); } public: diff --git a/src/NBT/FloatTag.hpp b/src/NBT/FloatTag.hpp index 8859f94..ad9cc64 100644 --- a/src/NBT/FloatTag.hpp +++ b/src/NBT/FloatTag.hpp @@ -36,13 +36,10 @@ class FloatTag : public Tag { private: friend class Tag; - FloatTag(Buffer *buffer) { - uint32_t value; + const uint8_t *ptr; - value = uint32_t(buffer->get()) << 24; - value |= uint32_t(buffer->get()) << 16; - value |= uint32_t(buffer->get()) << 8; - value |= uint32_t(buffer->get()); + FloatTag(Buffer *buffer) { + ptr = buffer->get(4); } public: diff --git a/src/NBT/IntArrayTag.hpp b/src/NBT/IntArrayTag.hpp index e9b0d3d..22f84b2 100644 --- a/src/NBT/IntArrayTag.hpp +++ b/src/NBT/IntArrayTag.hpp @@ -38,24 +38,12 @@ class IntArrayTag : public Tag { private: friend class Tag; - std::vector<uint32_t> value; + uint32_t len; + const uint8_t *ptr; IntArrayTag(Buffer *buffer) { - uint32_t len = uint32_t(buffer->get()) << 24; - len |= uint32_t(buffer->get()) << 16; - len |= uint32_t(buffer->get()) << 8; - len |= uint32_t(buffer->get()); - - value.resize(len); - - for (uint32_t i = 0; i < len; i++) { - uint32_t v = uint32_t(buffer->get()) << 24; - v |= uint32_t(buffer->get()) << 16; - v |= uint32_t(buffer->get()) << 8; - v |= uint32_t(buffer->get()); - - value[i] = v; - } + len = buffer->get32(); + ptr = buffer->get(4*len); } public: diff --git a/src/NBT/IntTag.hpp b/src/NBT/IntTag.hpp index 9e7b3db..b0337c7 100644 --- a/src/NBT/IntTag.hpp +++ b/src/NBT/IntTag.hpp @@ -36,13 +36,10 @@ class IntTag : public Tag { private: friend class Tag; - uint32_t value; + const uint8_t *ptr; IntTag(Buffer *buffer) { - value = uint32_t(buffer->get()) << 24; - value |= uint32_t(buffer->get()) << 16; - value |= uint32_t(buffer->get()) << 8; - value |= uint32_t(buffer->get()); + ptr = buffer->get(4); } public: @@ -51,7 +48,7 @@ public: } uint32_t getValue() const { - return value; + return Buffer::parse32(ptr); } }; diff --git a/src/NBT/ListTag.hpp b/src/NBT/ListTag.hpp index 1a48062..757a0aa 100644 --- a/src/NBT/ListTag.hpp +++ b/src/NBT/ListTag.hpp @@ -53,10 +53,7 @@ private: Type type; ListTag(Type type0, Buffer *buffer) : type(type0) { - uint32_t len = uint32_t(buffer->get()) << 24; - len |= uint32_t(buffer->get()) << 16; - len |= uint32_t(buffer->get()) << 8; - len |= uint32_t(buffer->get()); + uint32_t len = buffer->get32(); this->resize(len); diff --git a/src/NBT/LongTag.hpp b/src/NBT/LongTag.hpp index 6cd5f8f..b07bc1a 100644 --- a/src/NBT/LongTag.hpp +++ b/src/NBT/LongTag.hpp @@ -36,17 +36,10 @@ class LongTag : public Tag { private: friend class Tag; - uint64_t value; + const uint8_t *ptr; LongTag(Buffer *buffer) { - value = uint64_t(buffer->get()) << 56; - value |= uint64_t(buffer->get()) << 48; - value |= uint64_t(buffer->get()) << 40; - value |= uint64_t(buffer->get()) << 32; - value |= uint64_t(buffer->get()) << 24; - value |= uint64_t(buffer->get()) << 16; - value |= uint64_t(buffer->get()) << 8; - value |= uint64_t(buffer->get()); + ptr = buffer->get(8); } public: @@ -55,7 +48,7 @@ public: } uint64_t getValue() const { - return value; + return Buffer::parse64(ptr); } }; diff --git a/src/NBT/ShortTag.hpp b/src/NBT/ShortTag.hpp index a44ca9b..7415317 100644 --- a/src/NBT/ShortTag.hpp +++ b/src/NBT/ShortTag.hpp @@ -36,11 +36,10 @@ class ShortTag : public Tag { private: friend class Tag; - uint16_t value; + const uint8_t *ptr; ShortTag(Buffer *buffer) { - value = buffer->get() << 8; - value |= buffer->get(); + ptr = buffer->get(2); } public: @@ -49,7 +48,7 @@ public: } uint16_t getValue() const { - return value; + return Buffer::parse16(ptr); } }; diff --git a/src/NBT/StringTag.hpp b/src/NBT/StringTag.hpp index 0144081..42432f2 100644 --- a/src/NBT/StringTag.hpp +++ b/src/NBT/StringTag.hpp @@ -36,23 +36,18 @@ class StringTag : public Tag { private: friend class Tag; - std::string value; + uint16_t len; + const uint8_t *ptr; StringTag(Buffer *buffer) { - uint16_t len = buffer->get() << 8; - len |= buffer->get(); - - value = buffer->getString(len); + len = buffer->get16(); + ptr = buffer->get(len); } public: virtual Type getType() const { return Type::String; } - - const std::string & getValue() const { - return value; - } }; } diff --git a/src/NBT/Tag.cpp b/src/NBT/Tag.cpp index 1829dc5..c75cb9f 100644 --- a/src/NBT/Tag.cpp +++ b/src/NBT/Tag.cpp @@ -90,7 +90,7 @@ std::shared_ptr<const Tag> Tag::readTag(Type type, Buffer *buffer) { } std::shared_ptr<const Tag> Tag::readList(Buffer *buffer) { - Type type = static_cast<Type>(buffer->get()); + Type type = static_cast<Type>(buffer->get8()); switch (type) { case Type::End: @@ -135,13 +135,12 @@ std::shared_ptr<const Tag> Tag::readList(Buffer *buffer) { } std::pair<std::string, std::shared_ptr<const Tag>> Tag::readNamedTag(Buffer *buffer) { - Type type = static_cast<Type>(buffer->get()); + Type type = static_cast<Type>(buffer->get8()); if (type == Type::End) return std::make_pair("", std::shared_ptr<EndTag>(new EndTag())); - uint16_t len = buffer->get() << 8; - len |= buffer->get(); - std::string name = buffer->getString(len); + uint16_t len = buffer->get16(); + std::string name(reinterpret_cast<const char*>(buffer->get(len)), len); return std::make_pair(name, readTag(type, buffer)); } |