import chunk
import world
import tempfile
import glob
import time
import cProfile
import os
import sys
import shutil

# Simple Benchmarking script.  Usage and example:

#  $ python contrib/benchmark.py World4/
#  Rendering 50 chunks...
#  Took 20.290062 seconds or 0.405801 seconds per chunk, or 2.464261 chunks per second


# create a new, empty, cache dir
cachedir = tempfile.mkdtemp(prefix="benchmark_cache", dir=".")
if os.path.exists("benchmark.prof"): os.unlink("benchmark.prof")

w = world.WorldRenderer("World4", cachedir)

numchunks = 50
chunklist = w._find_chunkfiles()[:numchunks]

print "Rendering %d chunks..." % (numchunks)
def go():
    for f in chunklist:
        chunk.render_and_save(f[2], w.cachedir, w, (None,None), None)
start = time.time()
if "-profile" in sys.argv:
    cProfile.run("go()", 'benchmark.prof')
else:
    go()
stop = time.time()

delta = stop - start

print "Took %f seconds or %f seconds per chunk, or %f chunks per second" % (delta, delta/numchunks, numchunks/delta)

if "-profile" in sys.argv:
    print "Profile is below:\n----\n"

    import pstats
    p = pstats.Stats('benchmark.prof')

    p.strip_dirs().sort_stats("cumulative").print_stats(20)


shutil.rmtree(cachedir)