Upstream/ViaNBT
1
0
Fork 0
mirror of https://github.com/ViaVersion/ViaNBT.git synced 2024-11-22 11:35:16 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added get, size, values, keyset, iterator, etc methods to ListTag and CompoundTag.

Browse Source
This commit is contained in:
Steveice10 2012-03-25 10:01:34 -07:00
parent e426b3d8e1
commit e9a259edd9
3 changed files with 36 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/main/java/ch/spacebase/opennbt/stream/NBTOutputStream.java
View File

@ -4,7 +4,6 @@ import java.io.Closeable;
import java.io.DataOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.List;
import java.util.zip.GZIPOutputStream;
import ch.spacebase.opennbt.NBTConstants;
@ -182,7 +181,7 @@ public final class NBTOutputStream implements Closeable {
* @throws IOException if an I/O error occurs.
*/
private void writeCompoundTagPayload(CompoundTag tag) throws IOException {
for(Tag childTag : tag.getValue().values()) {
for(Tag childTag : tag.values()) {
writeTag(childTag);
}
os.writeByte((byte) 0); // end tag - better way?
@ -193,16 +192,14 @@ public final class NBTOutputStream implements Closeable {
* @param tag The tag.
* @throws IOException if an I/O error occurs.
*/
@SuppressWarnings("unchecked")
private void writeListTagPayload(ListTag<?> tag) throws IOException {
Class<? extends Tag> clazz = tag.getType();
List<Tag> tags = (List<Tag>) tag.getValue();
int size = tags.size();
int size = tag.size();
os.writeByte(NBTUtils.getTypeCode(clazz));
os.writeInt(size);
for(int i = 0; i < size; i++) {
writeTagPayload(tags.get(i));
for(Tag t : tag.getValue()) {
this.writeTagPayload(t);
}
}

21
src/main/java/ch/spacebase/opennbt/tag/CompoundTag.java
View File

@ -34,14 +34,13 @@ package ch.spacebase.opennbt.tag;
* POSSIBILITY OF SUCH DAMAGE.
*/
import java.util.Collection;
import java.util.Collections;
import java.util.Map;
import java.util.Set;
import ch.spacebase.opennbt.NBTUtils;
/**
* The <code>TAG_Compound</code> tag.
*/
@ -67,6 +66,22 @@ public final class CompoundTag extends Tag {
return value;
}
public Tag get(String tagName) {
return this.value.get(tagName);
}
public Set<String> keySet() {
return this.value.keySet();
}
public Collection<Tag> values() {
return this.value.values();
}
public int size() {
return this.value.size();
}
@Override
public String toString() {
String name = getName();

15
src/main/java/ch/spacebase/opennbt/tag/ListTag.java
View File

@ -36,6 +36,7 @@ package ch.spacebase.opennbt.tag;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import ch.spacebase.opennbt.NBTUtils;
@ -46,7 +47,7 @@ import ch.spacebase.opennbt.NBTUtils;
/**
* The <code>TAG_List</code> tag.
*/
public final class ListTag<T extends Tag> extends Tag {
public final class ListTag<T extends Tag> extends Tag implements Iterable<T> {
/**
* The type.
@ -83,6 +84,18 @@ public final class ListTag<T extends Tag> extends Tag {
return value;
}
public T get(int index) {
return this.value.get(index);
}
public Iterator<T> iterator() {
return this.value.iterator();
}
public int size() {
return this.value.size();
}
@Override
public String toString() {
String name = getName();