diff options
Diffstat (limited to 'util/minetestmapper.py')
| -rw-r--r--[-rwxr-xr-x] | util/minetestmapper.py | 62 |
1 files changed, 42 insertions, 20 deletions
diff --git a/util/minetestmapper.py b/util/minetestmapper.py index c7ec0774e..a1f1e3473 100755..100644 --- a/util/minetestmapper.py +++ b/util/minetestmapper.py @@ -56,6 +56,25 @@ TRANSLATION_TABLE = { 28: 0x813, # CONTENT_PAPYRUS 29: 0x814} # CONTENT_BOOKSHELF +class Bytestream: + def __init__(self, stream): + self.stream = stream + self.pos = 0 + + # So you can use files also + if hasattr(self.stream, 'read'): + self.read = self.stream.read + + def __len__(self): + return len(self.stream) + + def read(self, length = None): + if length is None: + length = len(self) + self.pos += length + return self.stream[self.pos - length:self.pos] + + def close(self): pass def hex_to_int(h): i = int(h, 16) @@ -88,9 +107,19 @@ def int_to_hex4(i): def getBlockAsInteger(p): return p[2]*16777216 + p[1]*4096 + p[0] -def getIntegerAsBlock(i): - return i%4096, int(i/4096)%4096, int(i/16777216)%4096 +def unsignedToSigned(i, max_positive): + if i < max_positive: + return i + else: + return i - 2*max_positive +def getIntegerAsBlock(i): + x = unsignedToSigned(i % 4096, 2048) + i = int((i - x) / 4096) + y = unsignedToSigned(i % 4096, 2048) + i = int((i - y) / 4096) + z = unsignedToSigned(i % 4096, 2048) + return x,y,z def limit(i, l, h): if(i > h): @@ -160,9 +189,9 @@ if path[-1:] != "/" and path[-1:] != "\\": # Load color information for the blocks. colors = {} try: - f = file("colors.txt") + f = file("colors.txt") except IOError: - f = file(os.path.join(os.path.dirname(__file__), "colors.txt")) + f = file(os.path.join(os.path.dirname(__file__), "colors.txt")) for line in f: values = string.split(line) colors[int(values[0], 16)] = ( @@ -190,7 +219,7 @@ if os.path.exists(path + "map.sqlite"): if not r: break - x, y, z = getIntegerAsBlock (r[0]) + x, y, z = getIntegerAsBlock(r[0]) if x < sector_xmin or x > sector_xmax: continue @@ -223,6 +252,9 @@ if os.path.exists(path + "sectors"): xlist.append(x) zlist.append(z) +# Get rid of doubles +xlist, zlist = zip(*sorted(set(zip(xlist, zlist)))) + minx = min(xlist) minz = min(zlist) maxx = max(xlist) @@ -372,7 +404,8 @@ for n in range(len(xlist)): if sectortype == "": continue - ylist.sort() + #ylist.sort() + ylist = sorted(set(ylist)) # Make a list of pixels of the sector that are to be looked for. pixellist = [] @@ -395,17 +428,14 @@ for n in range(len(xlist)): r = cur.fetchone() if not r: continue - filename = "mtm_tmp" - f = file(filename, 'wb') - f.write(r[0]) - f.close() + f = Bytestream(r[0]) else: if sectortype == "old": filename = path + "sectors/" + sector1 + "/" + yhex.lower() else: filename = path + "sectors2/" + sector2 + "/" + yhex.lower() - f = file(filename, "rb") + f = file(filename, "rb") # Let's just memorize these even though it's not really necessary. version = ord(f.read(1)) @@ -431,12 +461,7 @@ for n in range(len(xlist)): r = cur.fetchone() if not r: continue - filename = "mtm_tmp" - f = file(filename, 'wb') - f.write(r[0]) - f.close() - - f = file(filename, "rb") + f = Bytestream(r[0]) version = ord(f.read(1)) flags = f.read(1) @@ -557,8 +582,5 @@ if drawplayers: except OSError: pass -if os.path.isfile("mtm_tmp"): - os.remove("mtm_tmp") - print "Saving" im.save(output) |