298 lines
14 KiB
Python
Executable File
298 lines
14 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# This file is part of the Minecraft Overviewer.
|
|
#
|
|
# Minecraft Overviewer is free software: you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or (at
|
|
# your option) any later version.
|
|
#
|
|
# Minecraft Overviewer is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
|
|
# Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with the Overviewer. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import sys
|
|
if not (sys.version_info[0] == 2 and sys.version_info[1] >= 6):
|
|
print "Sorry, the Overviewer requires at least Python 2.6 to run" # Python3.0 is not supported either
|
|
sys.exit(1)
|
|
|
|
import os
|
|
import os.path
|
|
from configParser import ConfigOptionParser
|
|
import re
|
|
import subprocess
|
|
import multiprocessing
|
|
import time
|
|
import logging
|
|
import util
|
|
import platform
|
|
|
|
logging.basicConfig(level=logging.INFO,format="%(asctime)s [%(levelname)s] %(message)s")
|
|
|
|
this_dir = util.get_program_path()
|
|
|
|
# make sure the c_overviewer extension is available
|
|
try:
|
|
import c_overviewer
|
|
except ImportError:
|
|
## try to find the build extension
|
|
ext = os.path.join(this_dir, "c_overviewer.%s" % ("pyd" if platform.system() == "Windows" else "so"))
|
|
if os.path.exists(ext):
|
|
print "Something has gone wrong importing the c_overviewer extension. Please"
|
|
print "make sure it is up-to-date (clean and rebuild)"
|
|
sys.exit(1)
|
|
|
|
print "You need to compile the c_overviewer module to run Minecraft Overviewer."
|
|
print "Run `python setup.py build`, or see the README for details."
|
|
sys.exit(1)
|
|
|
|
if hasattr(sys, "frozen"):
|
|
pass # we don't bother with a compat test since it should always be in sync
|
|
elif "extension_version" in dir(c_overviewer):
|
|
# check to make sure the binary matches the headers
|
|
if os.path.exists(os.path.join(this_dir, "src", "overviewer.h")):
|
|
with open(os.path.join(this_dir, "src", "overviewer.h")) as f:
|
|
lines = f.readlines()
|
|
lines = filter(lambda x: x.startswith("#define OVERVIEWER_EXTENSION_VERSION"), lines)
|
|
if lines:
|
|
l = lines[0]
|
|
if int(l.split()[2].strip()) != c_overviewer.extension_version():
|
|
print "Please rebuild your c_overviewer module. It is out of date!"
|
|
sys.exit(1)
|
|
else:
|
|
print "Please rebuild your c_overviewer module. It is out of date!"
|
|
sys.exit(1)
|
|
|
|
|
|
import optimizeimages
|
|
import world
|
|
import quadtree
|
|
import googlemap
|
|
import rendernode
|
|
|
|
helptext = """
|
|
%prog [OPTIONS] <World # / Name / Path to World> <tiles dest dir>
|
|
%prog -d <World # / Name / Path to World / Path to cache dir> [tiles dest dir]"""
|
|
|
|
|
|
|
|
def main():
|
|
try:
|
|
cpus = multiprocessing.cpu_count()
|
|
except NotImplementedError:
|
|
cpus = 1
|
|
|
|
avail_rendermodes = c_overviewer.get_render_modes()
|
|
|
|
parser = ConfigOptionParser(usage=helptext, config="settings.py")
|
|
parser.add_option("-V", "--version", dest="version", help="Displays version information and then exits", action="store_true")
|
|
parser.add_option("-p", "--processes", dest="procs", help="How many worker processes to start. Default %s" % cpus, default=cpus, action="store", type="int")
|
|
parser.add_option("-z", "--zoom", dest="zoom", help="Sets the zoom level manually instead of calculating it. This can be useful if you have outlier chunks that make your world too big. This value will make the highest zoom level contain (2**ZOOM)^2 tiles", action="store", type="int", configFileOnly=True)
|
|
parser.add_option("-d", "--delete", dest="delete", help="Clear all caches. Next time you render your world, it will have to start completely over again. This is probably not a good idea for large worlds. Use this if you change texture packs and want to re-render everything.", action="store_true", commandLineOnly=True)
|
|
parser.add_option("--regionlist", dest="regionlist", help="A file containing, on each line, a path to a regionlist to update. Instead of scanning the world directory for regions, it will just use this list. Normal caching rules still apply.")
|
|
parser.add_option("--forcerender", dest="forcerender", help="Force re-rendering the entire map (or the given regionlist). Useful for re-rendering without deleting the entire map with --delete.", action="store_true")
|
|
parser.add_option("--rendermodes", dest="rendermode", help="Specifies the render types, separated by commas. Use --list-rendermodes to list them all.", type="choice", choices=avail_rendermodes, required=True, default=avail_rendermodes[0], listify=True)
|
|
parser.add_option("--list-rendermodes", dest="list_rendermodes", action="store_true", help="List available render modes and exit.", commandLineOnly=True)
|
|
parser.add_option("--imgformat", dest="imgformat", help="The image output format to use. Currently supported: png(default), jpg.", configFileOnly=True )
|
|
parser.add_option("--imgquality", dest="imgquality", default=95, help="Specify the quality of image output when using imgformat=\"jpg\".", type="int", configFileOnly=True)
|
|
parser.add_option("--bg_color", dest="bg_color", help="Configures the background color for the GoogleMap output. Specify in #RRGGBB format", configFileOnly=True, type="string", default="#1A1A1A")
|
|
parser.add_option("--optimize-img", dest="optimizeimg", help="If using png, perform image file size optimizations on the output. Specify 1 for pngcrush, 2 for pngcrush+advdef and 3 for pngcrush-advdef with more agressive settings. This may double (or more) render times, but will produce up to 30% smaller images. NOTE: requires corresponding programs in $PATH or %PATH%", configFileOnly=True)
|
|
parser.add_option("--web-assets-hook", dest="web_assets_hook", help="If provided, run this function after the web assets have been copied, but before actual tile rendering begins. It should accept a QuadtreeGen object as its only argument.", action="store", metavar="SCRIPT", type="function", configFileOnly=True)
|
|
parser.add_option("--web-assets-path", dest="web_assets_path", help="Specifies a non-standard web_assets directory to use. Files here will overwrite the default web assets.", metavar="PATH", type="string", configFileOnly=True)
|
|
parser.add_option("--textures-path", dest="textures_path", help="Specifies a non-standard textures path, from which terrain.png and other textures are loaded.", metavar="PATH", type="string", configFileOnly=True)
|
|
parser.add_option("-q", "--quiet", dest="quiet", action="count", default=0, help="Print less output. You can specify this option multiple times.")
|
|
parser.add_option("-v", "--verbose", dest="verbose", action="count", default=0, help="Print more output. You can specify this option multiple times.")
|
|
parser.add_option("--skip-js", dest="skipjs", action="store_true", help="Don't output marker.js or regions.js")
|
|
parser.add_option("--display-config", dest="display_config", action="store_true", help="Display the configuration parameters, but don't render the map. Requires all required options to be specified", commandLineOnly=True)
|
|
#parser.add_option("--write-config", dest="write_config", action="store_true", help="Writes out a sample config file", commandLineOnly=True)
|
|
|
|
options, args = parser.parse_args()
|
|
|
|
|
|
if options.version:
|
|
print "Minecraft-Overviewer"
|
|
print "Git version: %s" % util.findGitVersion()
|
|
try:
|
|
import overviewer_version
|
|
if hasattr(sys, "frozen"):
|
|
print "py2exe version build on %s" % overviewer_version.BUILD_DATE
|
|
print "Build machine: %s %s" % (overviewer_version.BUILD_PLATFORM, overviewer_version.BUILD_OS)
|
|
except:
|
|
pass
|
|
sys.exit(0)
|
|
|
|
if options.list_rendermodes:
|
|
rendermode_info = map(c_overviewer.get_render_mode_info, avail_rendermodes)
|
|
name_width = max(map(lambda i: len(i['name']), rendermode_info))
|
|
for info in rendermode_info:
|
|
print "{name:{0}} {description}".format(name_width, **info)
|
|
sys.exit(0)
|
|
|
|
if len(args) < 1:
|
|
logging.error("You need to give me your world number or directory")
|
|
parser.print_help()
|
|
list_worlds()
|
|
sys.exit(1)
|
|
worlddir = args[0]
|
|
|
|
if not os.path.exists(worlddir):
|
|
# world given is either world number, or name
|
|
worlds = world.get_worlds()
|
|
|
|
# if there are no worlds found at all, exit now
|
|
if not worlds:
|
|
parser.print_help()
|
|
logging.error("Invalid world path")
|
|
sys.exit(1)
|
|
|
|
try:
|
|
worldnum = int(worlddir)
|
|
worlddir = worlds[worldnum]['path']
|
|
except ValueError:
|
|
# it wasn't a number or path, try using it as a name
|
|
try:
|
|
worlddir = worlds[worlddir]['path']
|
|
except KeyError:
|
|
# it's not a number, name, or path
|
|
parser.print_help()
|
|
logging.error("Invalid world name or path")
|
|
sys.exit(1)
|
|
except KeyError:
|
|
# it was an invalid number
|
|
parser.print_help()
|
|
logging.error("Invalid world number")
|
|
sys.exit(1)
|
|
|
|
if len(args) < 2:
|
|
if options.delete:
|
|
return delete_all(worlddir, None)
|
|
logging.error("Where do you want to save the tiles?")
|
|
sys.exit(1)
|
|
elif len(args) > 2:
|
|
if options.delete:
|
|
return delete_all(worlddir, None)
|
|
parser.print_help()
|
|
logging.error("Sorry, you specified too many arguments")
|
|
sys.exit(1)
|
|
|
|
|
|
destdir = args[1]
|
|
if options.display_config:
|
|
# just display the config file and exit
|
|
parser.display_config()
|
|
sys.exit(0)
|
|
|
|
|
|
if options.delete:
|
|
return delete_all(worlddir, destdir)
|
|
|
|
if options.regionlist:
|
|
regionlist = map(str.strip, open(options.regionlist, 'r'))
|
|
else:
|
|
regionlist = None
|
|
|
|
if options.imgformat:
|
|
if options.imgformat not in ('jpg','png'):
|
|
parser.error("Unknown imgformat!")
|
|
else:
|
|
imgformat = options.imgformat
|
|
else:
|
|
imgformat = 'png'
|
|
|
|
if options.optimizeimg:
|
|
optimizeimg = int(options.optimizeimg)
|
|
optimizeimages.check_programs(optimizeimg)
|
|
else:
|
|
optimizeimg = None
|
|
|
|
logging.getLogger().setLevel(
|
|
logging.getLogger().level + 10*options.quiet)
|
|
logging.getLogger().setLevel(
|
|
logging.getLogger().level - 10*options.verbose)
|
|
|
|
logging.info("Welcome to Minecraft Overviewer!")
|
|
logging.debug("Current log level: {0}".format(logging.getLogger().level))
|
|
|
|
useBiomeData = os.path.exists(os.path.join(worlddir, 'biomes'))
|
|
if not useBiomeData:
|
|
logging.info("Notice: Not using biome data for tinting")
|
|
|
|
# First do world-level preprocessing
|
|
w = world.World(worlddir, useBiomeData=useBiomeData, regionlist=regionlist)
|
|
w.go(options.procs)
|
|
|
|
logging.info("Rending the following tilesets: %s", ",".join(options.rendermode))
|
|
|
|
bgcolor = (int(options.bg_color[1:3],16), int(options.bg_color[3:5],16), int(options.bg_color[5:7],16), 0)
|
|
|
|
# create the quadtrees
|
|
# TODO chunklist
|
|
q = []
|
|
qtree_args = {'depth' : options.zoom, 'imgformat' : imgformat, 'imgquality' : options.imgquality, 'optimizeimg' : optimizeimg, 'bgcolor' : bgcolor, 'forcerender' : options.forcerender}
|
|
for rendermode in options.rendermode:
|
|
if rendermode == 'normal':
|
|
qtree = quadtree.QuadtreeGen(w, destdir, rendermode=rendermode, tiledir='tiles', **qtree_args)
|
|
else:
|
|
qtree = quadtree.QuadtreeGen(w, destdir, rendermode=rendermode, **qtree_args)
|
|
q.append(qtree)
|
|
|
|
# do quadtree-level preprocessing
|
|
for qtree in q:
|
|
qtree.go(options.procs)
|
|
|
|
# create the distributed render
|
|
r = rendernode.RenderNode(q, options)
|
|
|
|
# write out the map and web assets
|
|
m = googlemap.MapGen(q, configInfo=options)
|
|
m.go(options.procs)
|
|
|
|
# render the tiles!
|
|
r.go(options.procs)
|
|
|
|
# finish up the map
|
|
m.finalize()
|
|
|
|
|
|
def delete_all(worlddir, tiledir):
|
|
# TODO should we delete tiledir here too?
|
|
|
|
# delete the overviewer.dat persistant data file
|
|
datfile = os.path.join(worlddir,"overviewer.dat")
|
|
if os.path.exists(datfile):
|
|
os.unlink(datfile)
|
|
logging.info("Deleting {0}".format(datfile))
|
|
|
|
def list_worlds():
|
|
"Prints out a brief summary of saves found in the default directory"
|
|
print
|
|
worlds = world.get_worlds()
|
|
if not worlds:
|
|
print 'No world saves found in the usual place'
|
|
return
|
|
print "Detected saves:"
|
|
for name, info in sorted(worlds.iteritems()):
|
|
if isinstance(name, basestring) and name.startswith("World") and len(name) == 6:
|
|
try:
|
|
world_n = int(name[-1])
|
|
# we'll catch this one later, when it shows up as an
|
|
# integer key
|
|
continue
|
|
except ValueError:
|
|
pass
|
|
timestamp = time.strftime("%Y-%m-%d %H:%M",
|
|
time.localtime(info['LastPlayed'] / 1000))
|
|
playtime = info['Time'] / 20
|
|
playstamp = '%d:%02d' % (playtime / 3600, playtime / 60 % 60)
|
|
size = "%.2fMB" % (info['SizeOnDisk'] / 1024. / 1024.)
|
|
print "World %s: %s Playtime: %s Modified: %s" % (name, size, playstamp, timestamp)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
multiprocessing.freeze_support()
|
|
main()
|