Add a save animation option and a try to dedup the turmit we keep

gte/__main__.py

@@ -87,6 +87,10 @@ parser_gen = subparsers.add_parser('generate', help='evolving help')
 
 parser_gen.add_argument('--prog', '-P', type=str, default=None)
 parser_gen.add_argument('--output', '-o', type=str, default='gte.png')
+parser_gen.add_argument('--output-dir', '-A', type=str, default=None,
+                        help="Save images for animation in this directory")
+parser_gen.add_argument('--anim-div', '-D', type=int, default=2,
+                        help='Save all X images (default 2) when -O given')
 parser_gen.add_argument('--world-height', '-y', type=int, metavar='HEIGHT',
                         default=512)
 parser_gen.add_argument('--world-width', '-x', type=int, metavar='WIDTH',
@@ -197,6 +201,10 @@ if 'pool_size' in args:
         keep_div = args.pool_div
         progs = []
         for prog in [ prog for _, prog, _ in scores[:len(scores)//keep_div]]:
+            for p2 in progs:
+                if prog == p2:
+                    prog = mutate(prog, force=True)
+                    break
             progs.append(prog)
             for cur_mut in range(keep_div - 1):
                 if args.exp_mutate:
@@ -229,9 +237,9 @@ else:
             logger.info(msg)
         for turmit in turmits:
             turmit()
-        #if step % 2 == 0:
-        #    #w.save('/tmp/anim/GTE_ANIM_%d.png' % (step/10))
-        #    w.save('/tmp/anim/GTE_ANIM_%d.png' % (step/2))
+        if args.output_dir is not None and step % args.anim_div == 0:
+            #w.save('/tmp/anim/GTE_ANIM_%d.png' % (step/10))
+            w.save('/tmp/anim/GTE_ANIM_%d.png' % (step/args.anim_div))
     stop = time.time()
 
     msg = 'Fractdim %.3f after %d steps in %.2fs (%dus per step)'