diff transformations/testtransformations.py @ 80:4407ab3f7805

small display change in testtransformation
author Xavier Glorot <glorotxa@iro.umontreal.ca>
date Wed, 10 Feb 2010 15:03:40 -0500
parents 6846136b2201
children 104e0b8a1c70
line wrap: on
line diff
--- a/transformations/testtransformations.py	Wed Feb 10 14:52:13 2010 -0500
+++ b/transformations/testtransformations.py	Wed Feb 10 15:03:40 2010 -0500
@@ -15,20 +15,22 @@
 from BruitGauss import BruitGauss
 from DistorsionGauss import DistorsionGauss
 from PermutPixel import PermutPixel
-#from gimp_script import GIMPTransformation
+#from gimp_script import GIMP1
 from Rature import Rature
 from contrast import Contrast
 from local_elastic_distortions import LocalElasticDistorter
 from slant import Slant
-#from Occlusion import Occlusion
+from Occlusion import Occlusion
 from add_background_image import AddBackground
 from affine_transform import AffineTransformation
 
 ###---------------------order of transformation module
-MODULE_INSTANCES = [Thick(),Slant(),AffineTransformation(), LocalElasticDistorter(), PermutPixel(), Rature(), BruitGauss(),PoivreSel(), Contrast()]
+MODULE_INSTANCES = [Slant(),Thick(),AffineTransformation(),LocalElasticDistorter(), PermutPixel(), Rature(), Occlusion(), BruitGauss(),PoivreSel(),AddBackground(), Contrast()]
 
 ###---------------------complexity associated to each of them
-complexity = [0.6,0.7,0.5,0.4,0.1,0.5,0.03,0.03,0.5]
+complexity = [0.5,0.6,0.5,0.4,0.1,0.5,0.5,0.5,0.5,0.5,0.5]
+complexity = [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.,0.]
+
 
 
 nbmodule = len(MODULE_INSTANCES)
@@ -42,7 +44,7 @@
 pygame.surfarray.use_arraytype('numpy')
 
 pygame.display.init()
-screen = pygame.display.set_mode((4*(nbmodule+1)*32,4*32+20),0,8)
+screen = pygame.display.set_mode((4*(nbmodule+1)/2*32,2*(4*32+20)),0,8)
 anglcolorpalette=[(x,x,x) for x in xrange(0,256)]
 screen.set_palette(anglcolorpalette)
 
@@ -59,6 +61,7 @@
     screen.blit(new,(0,0))
     
     offset = 4*32
+    offset2 = 0
     ct = 0
     for j in MODULE_INSTANCES:
         #max dilation
@@ -73,11 +76,15 @@
         new=pygame.transform.scale2x(new)
         new=pygame.transform.scale2x(new)
         new.set_palette(anglcolorpalette)
-        screen.blit(new,(offset,0))
+        screen.blit(new,(offset,offset2))
         font = pygame.font.SysFont('liberationserif',18)
         text = font.render(j.__module__,0,(255,255,255),(0,0,0))
-        screen.blit(text,(offset,4*32))
-        offset += 4*32
+        screen.blit(text,(offset,offset2+4*32))
+        if ct == len(MODULE_INSTANCES)/2-1:
+            offset = 0
+            offset2 = 4*32+20
+        else:
+            offset += 4*32
         ct+=1
     pygame.display.update()
     raw_input('Press Enter')