Fix minetestmapper.py

[dragonfireclient.git] / util / minetestmapper.py

diff --git a/util/minetestmapper.py b/util/minetestmapper.py
index cf2a5776d7625ae83f9057fe10f0446e7f913bc4..1914576e8859a965f718f374a7ec9262e7007b28 100755 (executable)
--- a/util/minetestmapper.py
+++ b/util/minetestmapper.py
@@ -85,6 +85,23 @@ def int_to_hex4(i):
         return "%04X" % i
 
 
+def getBlockAsInteger(p):
+    return p[2]*16777216 + p[1]*4096 + p[0]
+
+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):
         i = h
@@ -152,7 +169,10 @@ if path[-1:] != "/" and path[-1:] != "\\":
 
 # Load color information for the blocks.
 colors = {}
-f = file("colors.txt")
+try:
+       f = file("colors.txt")
+except IOError:
+       f = file(os.path.join(os.path.dirname(__file__), "colors.txt"))
 for line in f:
     values = string.split(line)
     colors[int(values[0], 16)] = (
@@ -166,6 +186,30 @@ zlist = []
 
 # List all sectors to memory and calculate the width and heigth of the
 # resulting picture.
+
+conn = None
+cur = None
+if os.path.exists(path + "map.sqlite"):
+    import sqlite3
+    conn = sqlite3.connect(path + "map.sqlite")
+    cur = conn.cursor()
+    
+    cur.execute("SELECT `pos` FROM `blocks`")
+    while True:
+        r = cur.fetchone()
+        if not r:
+            break
+        
+        x, y, z = getIntegerAsBlock    (r[0])
+        
+        if x < sector_xmin or x > sector_xmax:
+            continue
+        if z < sector_zmin or z > sector_zmax:
+            continue
+        
+        xlist.append(x)
+        zlist.append(z)
+
 if os.path.exists(path + "sectors2"):
     for filename in os.listdir(path + "sectors2"):
         for filename2 in os.listdir(path + "sectors2/" + filename):
@@ -302,6 +346,16 @@ for n in range(len(xlist)):
 
     sectortype = ""
 
+    if cur:
+        ps = getBlockAsInteger((xpos, 0, zpos))
+        cur.execute("SELECT `pos` FROM `blocks` WHERE `pos`>=? AND `pos`<?", (ps, ps + 4096))
+        while True:
+            r = cur.fetchone()
+            if not r:
+                break
+            pos = getIntegerAsBlock(r[0])[1]
+            ylist.append(pos)
+            sectortype = "sqlite"
     try:
         for filename in os.listdir(path + "sectors/" + sector1):
             if(filename != "meta"):
@@ -313,7 +367,7 @@ for n in range(len(xlist)):
     except OSError:
         pass
 
-    if sectortype != "old":
+    if sectortype == "":
         try:
             for filename in os.listdir(path + "sectors2/" + sector2):
                 if(filename != "meta"):
@@ -345,10 +399,21 @@ for n in range(len(xlist)):
         yhex = int_to_hex4(ypos)
 
         filename = ""
-        if sectortype == "old":
-            filename = path + "sectors/" + sector1 + "/" + yhex.lower()
+        if sectortype == "sqlite":
+            ps = getBlockAsInteger((xpos, ypos, zpos))
+            cur.execute("SELECT `data` FROM `blocks` WHERE `pos`==? LIMIT 1", (ps,))
+            r = cur.fetchone()
+            if not r:
+                continue
+            filename = "mtm_tmp"
+            f = file(filename, 'wb')
+            f.write(r[0])
+            f.close()
         else:
-            filename = path + "sectors2/" + sector2 + "/" + yhex.lower()
+            if sectortype == "old":
+                filename = path + "sectors/" + sector1 + "/" + yhex.lower()
+            else:
+                filename = path + "sectors2/" + sector2 + "/" + yhex.lower()
 
         f = file(filename, "rb")
 
@@ -371,6 +436,16 @@ for n in range(len(xlist)):
 
     if len(pixellist) > 0:
         for (ypos, filename) in ylist2:
+            ps = getBlockAsInteger((xpos, ypos, zpos))
+            cur.execute("SELECT `data` FROM `blocks` WHERE `pos`==? LIMIT 1", (ps,))
+            r = cur.fetchone()
+            if not r:
+                continue
+            filename = "mtm_tmp"
+            f = file(filename, 'wb')
+            f.write(r[0])
+            f.close()
+            
             f = file(filename, "rb")
 
             version = ord(f.read(1))
@@ -492,5 +567,8 @@ if drawplayers:
     except OSError:
         pass
 
+if os.path.isfile("mtm_tmp"):
+    os.remove("mtm_tmp")
+
 print "Saving"
 im.save(output)