You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
6.3 KiB
126 lines
6.3 KiB
import subprocess
|
|
import os
|
|
from minetestinfo import *
|
|
#python_exe_path is from:
|
|
from pythoninfo import *
|
|
#from PIL import Image, ImageDraw, ImageFont, ImageColor
|
|
from PIL import Image
|
|
|
|
class ChunkymapOfflineRenderer:
|
|
|
|
minetestmapper_fast_sqlite_path = None
|
|
minetestmapper_custom_path = None
|
|
minetestmapper_py_path = None
|
|
backend_string = None
|
|
world_path = None
|
|
world_name = None
|
|
|
|
def __init__(self):
|
|
|
|
self.backend_string = get_world_var("backend")
|
|
|
|
#region the following is also in singleimage.py
|
|
self.minetestmapper_fast_sqlite_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "minetestmapper-numpy.py")
|
|
self.minetestmapper_custom_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "minetestmapper-expertmm.py")
|
|
self.minetestmapper_py_path = self.minetestmapper_fast_sqlite_path
|
|
if (self.backend_string!="sqlite3"):
|
|
# Since minetestmapper-numpy has trouble with leveldb:
|
|
self.minetestmapper_py_path = self.minetestmapper_custom_path
|
|
print("Chose image generator script: "+self.minetestmapper_py_path)
|
|
if not os.path.isfile(self.minetestmapper_py_path):
|
|
print("ERROR: script does not exist, so exiting "+__file__+".")
|
|
sys.exit(2)
|
|
self.colors_path = os.path.join(os.path.dirname(os.path.abspath(self.minetestmapper_py_path)), "colors.txt")
|
|
if not os.path.isfile(self.colors_path):
|
|
print("ERROR: missing '"+self.colors_path+"', so exiting "+__file__+".")
|
|
sys.exit(2)
|
|
self.world_path = minetestinfo.get_var("primary_world_path")
|
|
if not os.path.isdir(self.world_path):
|
|
print("ERROR: missing world '"+self.world_path+"', so exiting "+__file__+".")
|
|
sys.exit(2)
|
|
else:
|
|
self.world_name = os.path.basename(self.world_path)
|
|
#endregion the following is also in singleimage.py
|
|
|
|
def RenderSingleImage(self):
|
|
genresults_folder_path = os.path.join( os.path.join(os.path.dirname(os.path.abspath(__file__)), "chunkymap-genresults"), self.world_name)
|
|
if not os.path.isdir(genresults_folder_path):
|
|
os.makedirs(genresults_folder_path)
|
|
genresult_path = os.path.join(genresults_folder_path, "singleimage"+genresult_name_closer_string)
|
|
cmd_suffix = " > \""+genresult_path+"\""
|
|
|
|
#blank since singleimage mode:
|
|
#geometry_string = str(min_x)+":"+str(min_z)+"+"+str(int(max_x)-int(min_x)+1)+"+"+str(int(max_z)-int(min_z)+1) # +1 since max-min is exclusive and width must be inclusive for minetestmapper.py
|
|
#geometry_param = " --geometry "+geometry_string
|
|
geometry_param = ""
|
|
#expertmm_region_string = str(min_x) + ":" + str(max_x) + "," + str(min_z) + ":" + str(max_z)
|
|
|
|
#cmd_no_out_string = python_exe_path+" "+self.minetestmapper_py_path+" --bgcolor '"+self.FLAG_EMPTY_HEXCOLOR+"' --input \""+minetestinfo.get_var("primary_world_path")+"\" --geometry "+geometry_string+" --output \""+tmp_png_path+"\""
|
|
png_name = "singleimage.png"
|
|
|
|
tmp_png_path = os.path.join(genresults_folder_path, png_name)
|
|
squote = ""
|
|
if os_name!="windows":
|
|
squote = "'"
|
|
io_string = " --input \""+self.world_path+"\" --output \""+tmp_png_path+"\""
|
|
if "numpy" in self.minetestmapper_py_path:
|
|
io_string = " \""+self.world_path+"\" \""+tmp_png_path+"\""
|
|
#geometry_param = " --region " + str(min_x) + " " + str(max_x) + " " + str(min_z) + " " + str(max_z)
|
|
cmd_no_out_string = python_exe_path+" "+self.minetestmapper_py_path+" --bgcolor "+squote+FLAG_EMPTY_HEXCOLOR+squote+io_string
|
|
cmd_string = cmd_no_out_string + cmd_suffix
|
|
print("")
|
|
print("")
|
|
print("Running")
|
|
print(" "+cmd_string)
|
|
print(" # (this may take a while...)")
|
|
if os.path.isfile(tmp_png_path):
|
|
os.remove(tmp_png_path)
|
|
subprocess.call(cmd_string, shell=True)
|
|
final_png_path = tmp_png_path
|
|
www_chunkymapdata_path = os.path.join(minetestinfo.get_var("www_minetest_path"), "chunkymapdata")
|
|
www_chunkymapdata_worlds_path = os.path.join(www_chunkymapdata_path, "worlds")
|
|
www_chunkymapdata_world_path = os.path.join(www_chunkymapdata_worlds_path, self.world_name)
|
|
if os.path.isfile(tmp_png_path):
|
|
if not os.path.isdir(www_chunkymapdata_world_path):
|
|
os.makedirs(www_chunkymapdata_world_path)
|
|
if minetestinfo.contains("www_minetest_path"):
|
|
dest_png_path = os.path.join(www_chunkymapdata_world_path, png_name)
|
|
if os.path.isfile(dest_png_path):
|
|
os.remove(dest_png_path)
|
|
os.rename(tmp_png_path, dest_png_path)
|
|
final_png_path = dest_png_path
|
|
print("Png image saved to:")
|
|
print(" "+final_png_path)
|
|
print("Converting to jpg...")
|
|
pngim = Image.open(final_png_path)
|
|
#jpgim = Image.new('RGB', pngim.size, (0, 0, 0))
|
|
#jpgim.paste(pngim.convert("RGB"), (0,0,pngim.size[0],pngim.size[0]))
|
|
jpg_name = "singleimage.jpg"
|
|
dest_jpg_path = os.path.join(www_chunkymapdata_world_path, jpg_name)
|
|
if os.path.isfile(dest_jpg_path):
|
|
os.remove(dest_jpg_path)
|
|
if not os.path.isfile(dest_jpg_path):
|
|
print(" removed old '"+dest_jpg_path+"'")
|
|
else:
|
|
print(" failed to remove'"+dest_jpg_path+"'")
|
|
#jpgim.save(dest_jpg_path)
|
|
pngim.save(dest_jpg_path, 'JPEG')
|
|
if os.path.isfile(dest_jpg_path):
|
|
print("jpg image saved to:")
|
|
print(" "+dest_jpg_path)
|
|
else:
|
|
print("Could not write '"+dest_jpg_path+"'")
|
|
if os.path.isfile(genresult_path):
|
|
print("Results:")
|
|
print(" "+genresult_path)
|
|
mtchunk = MTChunk()
|
|
mtchunk.set_from_genresult(genresult_path)
|
|
mtchunk.metadata["is_traversed"] = True
|
|
dest_yaml_name = "singleimage.yml"
|
|
dest_yaml_path = os.path.join(www_chunkymapdata_world_path, dest_yaml_name)
|
|
mtchunk.save_yaml(dest_yaml_path)
|
|
else:
|
|
print("No image could be generated from '"+self.world_path+"'")
|
|
|
|
cmor = ChunkymapOfflineRenderer()
|
|
cmor.RenderSingleImage()
|
|
|