def save_player(self, player):
tag = NBTFile()
tag.name = ""
tag["Pos"] = TAG_List(type=TAG_Double)
tag["Pos"].tags = [
TAG_Double(i)
for i in (player.location.x, player.location.y, player.location.z)
]
tag["Rotation"] = TAG_List(type=TAG_Double)
tag["Rotation"].tags = [
TAG_Double(i) for i in (player.location.yaw, player.location.pitch)
]
tag["Inventory"] = self._save_inventory_to_tag(player.inventory)
fp = self.folder.child("players")
if not fp.exists():
fp.makedirs()
fp = fp.child("%s.dat" % player.username)
self._write_tag(fp, tag)
def _save_chunk_to_tag(self, chunk):
tag = NBTFile()
tag.name = ""
level = TAG_Compound()
tag["Level"] = level
level["Blocks"] = TAG_Byte_Array()
level["HeightMap"] = TAG_Byte_Array()
level["BlockLight"] = TAG_Byte_Array()
level["Data"] = TAG_Byte_Array()
level["SkyLight"] = TAG_Byte_Array()
level["Blocks"].value = chunk.blocks.tostring()
level["HeightMap"].value = chunk.heightmap.tostring()
level["BlockLight"].value = pack_nibbles(chunk.blocklight)
level["Data"].value = pack_nibbles(chunk.metadata)
level["SkyLight"].value = pack_nibbles(chunk.skylight)
level["TerrainPopulated"] = TAG_Byte(chunk.populated)
level["Entities"] = TAG_List(type=TAG_Compound)
for entity in chunk.entities:
try:
entitytag = self._save_entity_to_tag(entity)
level["Entities"].tags.append(entitytag)
except KeyError:
print "Unknown entity %s" % entity.name
level["TileEntities"] = TAG_List(type=TAG_Compound)
for tile in chunk.tiles.itervalues():
try:
tiletag = self._save_tile_to_tag(tile)
level["TileEntities"].tags.append(tiletag)
except KeyError:
print "Unknown tile entity %s" % tile.name
return tag
def load_chunk(self, chunk):
region = name_for_region(chunk.x, chunk.z)
fp = self.folder.child("region").child(region)
if not fp.exists():
return
x, z = chunk.x % 32, chunk.z % 32
if region not in self.regions:
self.cache_region_pages(region)
positions = self.regions[region][0]
if (x, z) not in positions:
return
position, pages = positions[x, z]
if not position or not pages:
return
with fp.open("r") as handle:
handle.seek(position * 4096)
data = handle.read(pages * 4096)
length = unpack(">L", data[:4])[0] - 1
version = ord(data[4])
data = data[5:length+5]
if version == 1:
data = data.decode("gzip")
fileobj = GzipFile(fileobj=StringIO(data))
elif version == 2:
fileobj = StringIO(data.decode("zlib"))
tag = NBTFile(buffer=fileobj)
return self._load_chunk_from_tag(chunk, tag)
def setUp(self):
self.golden_value = "\x0A\0\x04Test\x08\0\x0Cempty string\0\0\0"
self.nbtfile = NBTFile(buffer=StringIO(self.golden_value))
def setUp(self):
self.nbtfile = NBTFile()
def testWriteback(self):
mynbt = NBTFile(self.f.name)
mynbt.write_file()
def testWriteBig(self):
mynbt = NBTFile(self.f.name)
output = StringIO()
mynbt.write_file(buffer=output)
self.assertTrue(GzipFile(self.f.name).read() == output.getvalue())
def testReadBig(self):
mynbt = NBTFile(self.f.name)
self.assertTrue(mynbt.filename != None)
self.assertEqual(len(mynbt.tags), 11)
def _read_tag(self, fp):
if fp.exists() and fp.getsize():
return NBTFile(fileobj=fp.open("r"))
return None
#!/usr/bin/env python
import sys
from bravo.nbt import NBTFile
if len(sys.argv) < 2:
print "Usage: %s <file>" % __name__
f = NBTFile(sys.argv[1])
print f.pretty_tree()
