AstralTransRocketries/pymclevel
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactored compressTag and decompressTag so we can use the compressors individually

Browse Source
This commit is contained in:
David Vierra 2011-04-05 19:59:37 -10:00
parent 80f1fd4944
commit 6578ac2aeb
1 changed files with 25 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
mclevel.py
View File

@ -2450,26 +2450,37 @@ class MCInfdevOldLevel(MCLevel):
return MCRegionFile.VERSION_DEFLATE
else:
return MCRegionFile.VERSION_GZIP
def compressTagGzip(self, root_tag):
buf = StringIO.StringIO()
with closing(gzip.GzipFile(fileobj=buf, mode='wb', compresslevel=2)) as gzipper:
root_tag.save(buf=gzipper)
return buf.getvalue()
def compressTagDeflate(self, root_tag):
buf = StringIO.StringIO()
root_tag.save(buf=buf)
return deflate(buf.getvalue())
def compressTag(self, root_tag):
if self.compressMode == MCRegionFile.VERSION_GZIP:
buf = StringIO.StringIO()
with closing(gzip.GzipFile(fileobj=buf, mode='wb', compresslevel=2)) as gzipper:
root_tag.save(buf=gzipper)
return buf.getvalue()
return self.compressTagGzip(root_tag)
if self.compressMode == MCRegionFile.VERSION_DEFLATE:
buf = StringIO.StringIO()
root_tag.save(buf=buf)
return deflate(buf.getvalue())
return self.compressTagDeflate(root_tag)
def decompressTagGzip(self, data):
with closing(gzip.GzipFile(fileobj=StringIO.StringIO(data))) as gz:
return nbt.load(buf=gz.read())
def decompressTagDeflate(self, data):
return nbt.load(buf=inflate(data))
def decompressTag(self, data):
if self.compressMode == MCRegionFile.VERSION_GZIP:
with closing(gzip.GzipFile(fileobj=StringIO.StringIO(data))) as gz:
return nbt.load(buf=gz.read())
return self.compressTagGzip(data)
if self.compressMode == MCRegionFile.VERSION_DEFLATE:
return nbt.load(buf=inflate(data))
return self.decompressTagDeflate(data)
@property