added minetestmapper in utils/

1 files changed, 444 insertions, 0 deletions

diff --git a/util/minetestmapper.py b/util/minetestmapper.py
new file mode 100755
index 000000000..5f3aab456
--- /dev/null
+++ b/util/minetestmapper.py
@@ -0,0 +1,444 @@
+#!/usr/bin/python2
+# -*- coding: windows-1252 -*-
+
+# This program is free software. It comes without any warranty, to
+# the extent permitted by applicable law. You can redistribute it
+# and/or modify it under the terms of the Do What The Fuck You Want
+# To Public License, Version 2, as published by Sam Hocevar. See
+# COPYING for more details.
+
+# Made by Jogge, modified by celeron55
+# 2011-05-29: j0gge: initial release
+# 2011-05-30: celeron55: simultaneous support for sectors/sectors2, removed 
+# 2011-06-02: j0gge: command line parameters, coordinates, players, ...
+# 2011-06-04: celeron55: added #!/usr/bin/python2 and converted \r\n to \n
+#                        to make it easily executable on Linux
+
+# Requires Python Imaging Library: http://www.pythonware.com/products/pil/
+
+# Some speed-up: ...lol, actually it slows it down.
+#import psyco ; psyco.full() 
+#from psyco.classes import *
+
+import zlib
+import Image, ImageDraw, ImageFont, ImageColor
+import os
+import string
+import time
+import getopt
+import sys
+
+def hex_to_int(h):
+	i = int(h, 16)
+	if(i > 2047):
+		i -= 4096
+	return i
+
+def hex4_to_int(h):
+	i = int(h, 16)
+	if(i > 32767):
+		i -= 65536
+	return i
+
+def int_to_hex3(i):
+	if(i < 0):
+		return "%03X" % (i + 4096)
+	else:
+		return "%03X" % i
+
+def int_to_hex4(i):
+	if(i < 0):
+		return "%04X" % (i + 65536)
+	else:
+		return "%04X" % i
+
+def limit(i, l, h):
+	if(i > h):
+		i = h
+	if(i < l):
+		i = l
+	return i
+
+def usage():
+	print "TODO: Help"
+try:
+	opts, args = getopt.getopt(sys.argv[1:], "hi:o:", ["help", "input=", "output=", "bgcolor=", "scalecolor=", "origincolor=", "playercolor=", "draworigin", "drawplayers", "drawscale"])
+except getopt.GetoptError, err:
+	# print help information and exit:
+	print str(err) # will print something like "option -a not recognized"
+	usage()
+	sys.exit(2)
+
+path = "../world/"
+output = "uloste.png"
+border = 0
+scalecolor = "black"
+bgcolor = "white"
+origincolor = "red"
+playercolor = "red"
+drawscale = False
+drawplayers = False
+draworigin = False
+
+sector_xmin = -1500 / 16
+sector_xmax = 1500 / 16
+sector_zmin = -1500 / 16
+sector_zmax = 1500 / 16
+
+for o, a in opts:
+	if o in ("-h", "--help"):
+		usage()
+		sys.exit()
+	elif o in ("-i", "--input"):
+		path = a
+	elif o in ("-o", "--output"):
+		output = a
+	elif o == "--bgcolor":
+		bgcolor = ImageColor.getrgb(a)
+	elif o == "--scalecolor":
+		scalecolor = ImageColor.getrgb(a)
+	elif o == "--playercolor":
+		playercolor = ImageColor.getrgb(a)
+	elif o == "--origincolor":
+		origincolor = ImageColor.getrgb(a)
+	elif o == "--drawscale":
+		drawscale = True
+		border = 40
+	elif o == "--drawplayers":
+		drawplayers = True
+	elif o == "--draworigin":
+		draworigin = True
+	else:
+		assert False, "unhandled option"
+	
+if path[-1:]!="/" and path[-1:]!="\\":
+	path = path + "/"
+
+# Load color information for the blocks.
+colors = {}
+f = file("colors.txt")
+for line in f:
+	values = string.split(line)
+	colors[int(values[0])] = (int(values[1]), int(values[2]), int(values[3]))
+f.close()
+
+xlist = []
+zlist = []
+
+# List all sectors to memory and calculate the width and heigth of the resulting picture.
+try:
+	for filename in os.listdir(path + "sectors2"):
+		for filename2 in os.listdir(path + "sectors2/" + filename):
+			x = hex_to_int(filename)
+			z = hex_to_int(filename2)
+			if x < sector_xmin or x > sector_xmax:
+				continue
+			if z < sector_zmin or z > sector_zmax:
+				continue
+			xlist.append(x)
+			zlist.append(z)
+except OSError:
+	pass
+try:
+	for filename in os.listdir(path + "sectors"):
+		x = hex4_to_int(filename[:4])
+		z = hex4_to_int(filename[-4:])
+		if x < sector_xmin or x > sector_xmax:
+			continue
+		if z < sector_zmin or z > sector_zmax:
+			continue
+		xlist.append(x)
+		zlist.append(z)
+except OSError:
+	pass
+
+minx = min(xlist)
+minz = min(zlist)
+maxx = max(xlist)
+maxz = max(zlist)
+
+w = (maxx - minx) * 16 + 16
+h = (maxz - minz) * 16 + 16
+
+print "w="+str(w)+" h="+str(h)
+
+im = Image.new("RGB", (w + border, h + border), bgcolor)
+draw = ImageDraw.Draw(im)
+impix = im.load()
+
+stuff = {}
+
+starttime = time.time()
+
+# Go through all sectors.
+for n in range(len(xlist)):
+	#if n > 500:
+	#	break
+	if n % 200 == 0:
+		nowtime = time.time()
+		dtime = nowtime - starttime
+		try:
+			n_per_second = 1.0 * n / dtime
+		except ZeroDivisionError:
+			n_per_second = 0
+		if n_per_second != 0:
+			seconds_per_n = 1.0 / n_per_second
+			time_guess = seconds_per_n * len(xlist)
+			remaining_s = time_guess - dtime
+			remaining_minutes = int(remaining_s / 60)
+			remaining_s -= remaining_minutes * 60;
+			print("Processing sector "+str(n)+" of "+str(len(xlist))
+					+" ("+str(round(100.0*n/len(xlist), 1))+"%)"
+					+" (ETA: "+str(remaining_minutes)+"m "
+					+str(int(remaining_s))+"s)")
+
+	xpos = xlist[n]
+	zpos = zlist[n]
+	
+	xhex = int_to_hex3(xpos)
+	zhex = int_to_hex3(zpos)
+	xhex4 = int_to_hex4(xpos)
+	zhex4 = int_to_hex4(zpos)
+	
+	sector1 = xhex4.lower() + zhex4.lower()
+	sector2 = xhex.lower() + "/" + zhex.lower()
+	
+	ylist = []
+	
+	sectortype = ""
+	
+	try:
+		for filename in os.listdir(path + "sectors/" + sector1):
+			if(filename != "meta"):
+				pos = int(filename, 16)
+				if(pos > 32767):
+					pos -= 65536
+				ylist.append(pos)
+				sectortype = "old"
+	except OSError:
+		pass
+	
+	if sectortype != "old":
+		try:
+			for filename in os.listdir(path + "sectors2/" + sector2):
+				if(filename != "meta"):
+					pos = int(filename, 16)
+					if(pos > 32767):
+						pos -= 65536
+					ylist.append(pos)
+					sectortype = "new"
+		except OSError:
+			pass
+	
+	if sectortype == "":
+		continue
+
+	ylist.sort()
+	
+	# Make a list of pixels of the sector that are to be looked for.
+	pixellist = []
+	water = {}
+	for x in range(16):
+		for z in range(16):
+			pixellist.append((x, z))
+			water[(x, z)] = 0
+	
+	# Go through the Y axis from top to bottom.
+	ylist2=[]
+	for ypos in reversed(ylist):
+		
+		yhex = int_to_hex4(ypos)
+
+		filename = ""
+		if sectortype == "old":
+			filename = path + "sectors/" + sector1 + "/" + yhex.lower()
+		else:
+			filename = path + "sectors2/" + sector2 + "/" + yhex.lower()
+
+		f = file(filename, "rb")
+
+		version = f.read(1)
+		flags = f.read(1)
+		
+		# Checking day and night differs -flag
+		if not ord(flags) & 2:
+			ylist2.append((ypos,filename))
+			f.close()
+			continue
+
+		dec_o = zlib.decompressobj()
+		try:
+			mapdata = dec_o.decompress(f.read())
+		except:
+			mapdata = []
+			
+		f.close()
+		
+		if(len(mapdata) < 4096):
+			print "bad: " + xhex + "/" + zhex + "/" + yhex + " " + str(len(mapdata))
+		else:
+			chunkxpos = xpos * 16
+			chunkypos = ypos * 16
+			chunkzpos = zpos * 16
+			for (x, z) in reversed(pixellist):
+				for y in reversed(range(16)):
+					datapos = x + y * 16 + z * 256
+					if(ord(mapdata[datapos]) != 254 and ord(mapdata[datapos]) in colors):
+						if(ord(mapdata[datapos]) == 2 or ord(mapdata[datapos]) == 9):
+							water[(x, z)] += 1
+							# Add dummy stuff for drawing sea without seabed
+							stuff[(chunkxpos + x, chunkzpos + z)] = (chunkypos + y, ord(mapdata[datapos]), water[(x, z)])
+						else:
+							pixellist.remove((x, z))
+							# Memorize information on the type and height of the block and for drawing the picture.
+							stuff[(chunkxpos + x, chunkzpos + z)] = (chunkypos + y, ord(mapdata[datapos]), water[(x, z)])
+							break
+					elif(ord(mapdata[datapos]) != 254 and ord(mapdata[datapos]) not in colors):
+						print "strange block: " + xhex + "/" + zhex + "/" + yhex + " x: " + str(x) + " y: " + str(y) + " z: " + str(z) + " palikka: " + str(ord(mapdata[datapos]))
+		
+		# After finding all the pixels in the sector, we can move on to the next sector without having to continue the Y axis.
+		if(len(pixellist) == 0):
+			break
+	
+	if len(pixellist) > 0:
+		for (ypos, filename) in ylist2:
+			f = file(filename, "rb")
+
+			version = f.read(1)
+			flags = f.read(1)
+
+			dec_o = zlib.decompressobj()
+			try:
+				mapdata = dec_o.decompress(f.read())
+			except:
+				mapdata = []
+				
+			f.close()
+			
+			if(len(mapdata) < 4096):
+				print "bad: " + xhex + "/" + zhex + "/" + yhex + " " + str(len(mapdata))
+			else:
+				chunkxpos = xpos * 16
+				chunkypos = ypos * 16
+				chunkzpos = zpos * 16
+				for (x, z) in reversed(pixellist):
+					for y in reversed(range(16)):
+						datapos = x + y * 16 + z * 256
+						if(ord(mapdata[datapos]) != 254 and ord(mapdata[datapos]) in colors):
+							if(ord(mapdata[datapos]) == 2 or ord(mapdata[datapos]) == 9):
+								water[(x, z)] += 1
+								# Add dummy stuff for drawing sea without seabed
+								stuff[(chunkxpos + x, chunkzpos + z)] = (chunkypos + y, ord(mapdata[datapos]), water[(x, z)])
+							else:
+								pixellist.remove((x, z))
+								# Memorize information on the type and height of the block and for drawing the picture.
+								stuff[(chunkxpos + x, chunkzpos + z)] = (chunkypos + y, ord(mapdata[datapos]), water[(x, z)])
+								break
+						elif(ord(mapdata[datapos]) != 254 and ord(mapdata[datapos]) not in colors):
+							print "outo palikka: " + xhex + "/" + zhex + "/" + yhex + " x: " + str(x) + " y: " + str(y) + " z: " + str(z) + " palikka: " + str(ord(mapdata[datapos]))
+			
+			# After finding all the pixels in the sector, we can move on to the next sector without having to continue the Y axis.
+			if(len(pixellist) == 0):
+				break
+
+print "Drawing image"
+# Drawing the picture
+starttime = time.time()
+n = 0
+for (x, z) in stuff.iterkeys():
+	if n % 500000 == 0:
+		nowtime = time.time()
+		dtime = nowtime - starttime
+		try:
+			n_per_second = 1.0 * n / dtime
+		except ZeroDivisionError:
+			n_per_second = 0
+		if n_per_second != 0:
+			listlen = len(stuff)
+			seconds_per_n = 1.0 / n_per_second
+			time_guess = seconds_per_n * listlen
+			remaining_s = time_guess - dtime
+			remaining_minutes = int(remaining_s / 60)
+			remaining_s -= remaining_minutes * 60;
+			print("Drawing pixel "+str(n)+" of "+str(listlen)
+					+" ("+str(round(100.0*n/listlen, 1))+"%)"
+					+" (ETA: "+str(remaining_minutes)+"m "
+					+str(int(remaining_s))+"s)")
+	n += 1
+
+	(r, g, b) = colors[stuff[(x,z)][1]]
+	# Comparing heights of a couple of adjacent blocks and changing brightness accordingly.
+	try:
+		c1 = stuff[(x - 1, z)][1]
+		c2 = stuff[(x, z + 1)][1]
+		c = stuff[(x, z)][1]
+		if c1 != 2 and c1 != 9 and c2 != 2 and c2 != 9 and c != 2 and c != 9:
+			y1 = stuff[(x - 1, z)][0]
+			y2 = stuff[(x, z + 1)][0]
+			y = stuff[(x, z)][0]
+			
+			d = ((y - y1) + (y - y2)) * 12
+		else:
+			d = 0
+		
+		if(d > 36):
+			d = 36
+			
+		r = limit(r + d, 0, 255)
+		g = limit(g + d, 0, 255)
+		b = limit(b + d, 0, 255)
+	except:
+		pass
+	
+	# Water
+	if(stuff[(x,z)][2] > 0):
+		r=int(r * .15 + colors[2][0] * .85)
+		g=int(g * .15 + colors[2][1] * .85)
+		b=int(b * .15 + colors[2][2] * .85)
+		
+	impix[x - minx * 16 + border, h - 1 - (z - minz * 16) + border] = (r, g, b)
+
+
+if draworigin:
+	draw.ellipse((minx * -16 - 5 + border, h - minz * -16 - 6 + border, minx * -16 + 5 + border, h - minz * -16 + 4 + border), outline = origincolor)
+
+font = ImageFont.load_default()
+
+if drawscale:
+	draw.text((24, 0), "X", font = font, fill = scalecolor)
+	draw.text((2, 24), "Z", font = font, fill = scalecolor)
+
+	for n in range(int(minx / -4) * -4, maxx, 4):
+		draw.text((minx * -16 + n * 16 + 2 + border, 0), str(n * 16), font = font, fill = scalecolor)
+		draw.line((minx * -16 + n * 16 + border, 0, minx * -16 + n * 16 + border, border - 1), fill = scalecolor)
+
+	for n in range(int(maxz / 4) * 4, minz, -4):
+		draw.text((2, h - 1 - (n * 16 - minz * 16) + border), str(n * 16), font = font, fill = scalecolor)
+		draw.line((0, h - 1 - (n * 16 - minz * 16) + border, border - 1, h - 1 - (n * 16 - minz * 16) + border), fill = scalecolor)
+
+if drawplayers:
+	try:
+		for filename in os.listdir(path + "players"):
+			f = file(path + "players/" + filename)
+			lines = f.readlines()
+			name=""
+			position=[]
+			for line in lines:
+				p = string.split(line)
+				if p[0] == "name":
+					name = p[2]
+					print filename + ": name = " + name
+				if p[0] == "position":
+					position = string.split(p[2][1:-1], ",")
+					print filename + ": position = " + p[2]
+			if len(name) > 0 and len(position) == 3:
+				x=(int(float(position[0]) / 10 - minx * 16))
+				z=int(h - (float(position[2]) / 10 - minz * 16))
+				draw.ellipse((x - 2 + border, z - 2 + border, x + 2 + border, z + 2 + border), outline = playercolor)
+				draw.text((x + 2 + border, z + 2 + border), name, font = font, fill = playercolor)
+			f.close()
+	except OSError:
+		pass
+
+print "Saving"
+im.save(output)