Ajout d'un main de test

L'essentiel du main sera déplacé dans un module

gte/__main__.py

@@ -0,0 +1,99 @@
+import argparse
+import logging
+import numpy as np
+from random import randint
+
+logging.basicConfig(level=logging.DEBUG,
+                    format="%(created)d %(asctime)s:%(module)s(%(lineno)d):%(levelname)s:%(message)s %(relativeCreated)dms")
+logger = logging.getLogger(__name__)
+logger.debug('Logger started')
+
+from . import turmit, rpnlib
+from .turmit import Turmit
+
+parser = argparse.ArgumentParser(description='Genetic Turmit Evolver')
+
+parser.add_argument('--list-op', action='store_const', default=False,
+                    const=True)
+parser.add_argument('--list-var', action='store_const', default=False,
+                    const=True)
+parser.add_argument('--rand-sym', '-r', action='store_const', default=False,
+                    const=True)
+
+
+args = parser.parse_args()
+
+if args.list_op:
+    for i,(op, desc) in enumerate(rpnlib._op_list.items()):
+        print('#%02X %7s%s' % (i, op.upper(), desc[0].__doc__))
+    exit(0)
+elif args.list_var:
+    print([op.upper() for op in rpnlib._op_list.keys()])
+    exit(0)
+elif args.rand_sym:
+    for _ in range(10):
+        print(rpnlib.RpnSymbol.random())
+    exit(0)
+
+x = 512
+y = 512
+nb_turmit = 10
+steps = 1024
+
+img = [[(0,0,0) for _ in range(x)] for _ in range(y)]
+
+prog = rpnlib.RpnExpr.random(10)
+prog.append(rpnlib.RpnSymbol.rand_op())
+
+turmits = [Turmit(prog=prog) for _ in range(nb_turmit)]
+turmits_color = [(randint(0,255), randint(0,255), randint(0,255))
+                 for _ in range(nb_turmit)]
+turmits_pos = [(randint(0,y-1), randint(0,x-1)) for _ in range(nb_turmit)]
+
+import time
+
+print(str(prog))
+start = time.time()
+for step in range(steps):
+    for i in range(nb_turmit):
+        ty, tx = turmits_pos[i]
+        r,g,b = img[ty][tx]
+        turmit = turmits[i]
+        tdir = turmit(x=tx,y=ty,r=r,g=g,b=b) % 8
+
+        img[ty][tx] = turmits_color[i]
+
+        if tdir == 0 or tdir == 4 or tdir == 5:
+            tx += 1
+        if tdir == 1 or tdir == 4 or tdir == 6:
+            ty += 1
+        if tdir == 2 or tdir == 6 or tdir == 7:
+            if tx == 0:
+                tx = x -1
+            else:
+                tx -= 1
+        if tdir == 3 or tdir == 5 or tdir == 7:
+            if ty == 0:
+                ty = y - 1
+            else:
+                ty -= 1
+        tx %= x
+        ty %= y
+        #print('Turmit #%d dir:%d x:%d y:%d prog="%s"' % (i, tdir, tx, ty,
+        #                                                 str(turmit._prog)))
+        turmits_pos[i] = (ty,tx)
+    if step % 100 == 0:
+        print('%d/%d' % (step, steps))
+stop = time.time()
+print('Exec %d steps in %.2fs : %.4fs/step' % (steps, stop - start,
+                                                (stop - start) / steps))
+from .fractdim import *
+I = rgb2gray(np.array(img))
+print(fractal_dimension(I))
+
+import scipy.misc
+scipy.misc.imsave('/tmp/GTE.jpg', img)
+print('saved to : /tmp/GTE.jpg')
+print()
+            
+    

gte/fractdim.py

@@ -51,5 +51,7 @@ def fractal_dimension(Z, threshold=None):
     coeffs = np.polyfit(np.log(sizes), np.log(counts), 1)
     return -coeffs[0]
 
-I = rgb2gray(scipy.misc.imread(sys.argv[1]))
-print("%f:%s" % (fractal_dimension(I), sys.argv[1]))
+if __name__ == '__main__':
+    I = rgb2gray(scipy.misc.imread(sys.argv[1]))
+    print(I)
+    print("%f:%s" % (fractal_dimension(I), sys.argv[1]))