mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 11:40:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add --resume

Browse Source
This commit is contained in:
Steven Robertson 2011-10-15 22:21:49 -04:00
parent 6e6321bb76
commit 8e99c9c463
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.py
View File

@ -40,13 +40,16 @@ def fmt_time(time):
return '%s_%03d' % (dcml, frac) return '%s_%03d' % (dcml, frac)
return dcml 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): def save(args, time, raw):
noalpha = raw[:,:,:3] noalpha = raw[:,:,:3]
if args.raw: if args.raw:
real_stdout.write(buffer(np.uint8(noalpha * 255.0))) real_stdout.write(buffer(np.uint8(noalpha * 255.0)))
return 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 = scipy.misc.toimage(noalpha, cmin=0, cmax=1)
img.save(name+'.png') img.save(name+'.png')
@ -124,9 +127,10 @@ def main(args):
if (args.start is None or t >= args.start) if (args.start is None or t >= args.start)
and (args.end is None or t < args.end)] and (args.end is None or t < args.end)]
render._AnimRenderer.sync = args.sync or args.sleep if args.resume:
if args.sleep: times = [t for t in times
render._AnimRenderer.sleep = args.sleep / 1000. if not os.path.isfile(fmt_filename(args, t)+'.png')]
anim = render.Animation(genomes) anim = render.Animation(genomes)
if args.debug: if args.debug:
anim.cmp_options.append('-G') anim.cmp_options.append('-G')
@ -192,6 +196,8 @@ if __name__ == "__main__":
help="Prefix to use when saving files (default is basename of input)") help="Prefix to use when saving files (default is basename of input)")
parser.add_argument('-o', metavar='DIR', type=str, dest='dir', parser.add_argument('-o', metavar='DIR', type=str, dest='dir',
help="Output directory", default='.') 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', parser.add_argument('--raw', action='store_true', dest='raw',
help="Do not write files; instead, send raw RGBA data to stdout.") 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', genome.add_argument('--height', type=int, metavar='PIXELS',
help="Use this height (does *not* auto-set scale)") help="Use this height (does *not* auto-set scale)")
debug = parser.add_argument_group('Debug options') debug = parser.add_argument_group('Debug options')
debug.add_argument('--test', action='store_true', dest='test', debug.add_argument('--test', action='store_true', dest='test',
help='Run some internal tests') help='Run some internal tests')