67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
#!/usr/bin/python
|
|
|
|
import os
|
|
import sys
|
|
import os.path
|
|
from optparse import OptionParser
|
|
import re
|
|
|
|
import world
|
|
|
|
helptext = """
|
|
%prog [-c] <Path to World> <image out.png>
|
|
%prog -d [-c] <Path to World>"""
|
|
|
|
def remove_images(worlddir, cavemode):
|
|
if cavemode:
|
|
cavestr = "cave"
|
|
else:
|
|
cavestr = "nocave"
|
|
imgre = r"img\.[^.]+\.[^.]+\.{0}\.\w+\.png$".format(cavestr)
|
|
matcher = re.compile(imgre)
|
|
|
|
for dirpath, dirnames, filenames in os.walk(worlddir):
|
|
for f in filenames:
|
|
if matcher.match(f):
|
|
filepath = os.path.join(dirpath, f)
|
|
print "Deleting {0}".format(filepath)
|
|
os.unlink(filepath)
|
|
|
|
def confirm(imgfile):
|
|
answer = raw_input("Overwrite existing image at %r? [Y/n]" % imgfile).strip()
|
|
if not answer or answer.lower().startswith("y"):
|
|
return True
|
|
return False
|
|
|
|
def main():
|
|
parser = OptionParser(usage=helptext)
|
|
parser.add_option("-c", "--caves", dest="caves", help="Render only caves", action="store_true")
|
|
parser.add_option("-d", "--delete-cache", dest="delete", help="Deletes the image files cached in your world directory", action="store_true")
|
|
parser.add_option("-p", "--processes", dest="procs", help="How many chunks to render in parallel. A good number for this is 1 more than the number of cores in your computer. Default 2", default=2, action="store", type="int")
|
|
|
|
options, args = parser.parse_args()
|
|
|
|
if len(args) < 1:
|
|
print "You need to give me your world directory"
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
worlddir = args[0]
|
|
|
|
if options.delete:
|
|
remove_images(worlddir, options.caves)
|
|
else:
|
|
if len(args) != 2:
|
|
parser.error("What do you want to save the image as?")
|
|
imageout = args[1]
|
|
if not imageout.endswith(".png"):
|
|
imageout = imageout + ".png"
|
|
if os.path.exists(imageout) and not confirm(imageout):
|
|
return
|
|
imageobj = world.render_world(worlddir, options.caves, options.procs)
|
|
print "Saving image..."
|
|
imageobj.save(imageout)
|
|
print "Saved as", imageout
|
|
|
|
if __name__ == "__main__":
|
|
main()
|