From 8e99c9c463f7b9cf0b43fb782d76777e377acbe2 Mon Sep 17 00:00:00 2001 From: Steven Robertson Date: Sat, 15 Oct 2011 22:21:49 -0400 Subject: [PATCH] Add --resume --- main.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 99e6940..a7f7fea 100644 --- a/main.py +++ b/main.py @@ -40,13 +40,16 @@ def fmt_time(time): return '%s_%03d' % (dcml, frac) return dcml +def fmt_filename(args, time): + return os.path.join(args.dir, '%s_%s' % (args.name, fmt_time(time))) + def save(args, time, raw): noalpha = raw[:,:,:3] if args.raw: real_stdout.write(buffer(np.uint8(noalpha * 255.0))) return - name = os.path.join(args.dir, '%s_%s' % (args.name, fmt_time(time))) + name = fmt_filename(args, time) img = scipy.misc.toimage(noalpha, cmin=0, cmax=1) img.save(name+'.png') @@ -124,9 +127,10 @@ def main(args): if (args.start is None or t >= args.start) and (args.end is None or t < args.end)] - render._AnimRenderer.sync = args.sync or args.sleep - if args.sleep: - render._AnimRenderer.sleep = args.sleep / 1000. + if args.resume: + times = [t for t in times + if not os.path.isfile(fmt_filename(args, t)+'.png')] + anim = render.Animation(genomes) if args.debug: anim.cmp_options.append('-G') @@ -192,6 +196,8 @@ if __name__ == "__main__": help="Prefix to use when saving files (default is basename of input)") parser.add_argument('-o', metavar='DIR', type=str, dest='dir', help="Output directory", default='.') + parser.add_argument('--resume', action='store_true', dest='resume', + help="Do not render any frame for which a .png already exists.") parser.add_argument('--raw', action='store_true', dest='raw', help="Do not write files; instead, send raw RGBA data to stdout.") @@ -227,7 +233,6 @@ if __name__ == "__main__": genome.add_argument('--height', type=int, metavar='PIXELS', help="Use this height (does *not* auto-set scale)") - debug = parser.add_argument_group('Debug options') debug.add_argument('--test', action='store_true', dest='test', help='Run some internal tests')