@@ -56,9 +56,13 @@ def __init__(self):
5656 self ._datatypes = (np .uint8 , np .uint16 , np .float )
5757 self ._datatype_index = 0
5858 self ._theta = _theta_generator ()
59+ self .numbering = True
5960 # Font for rendering counter in images.
6061 self ._font = ImageFont .load_default ()
6162
63+ def enable_numbering (self , enab ):
64+ self .numbering = enab
65+
6266 def get_data_types (self ):
6367 return (t .__name__ for t in self ._datatypes )
6468
@@ -86,7 +90,7 @@ def get_image(self, width, height, dark=0, light=255, index=None):
8690 d = self ._datatypes [self ._datatype_index ]
8791 #return Image.fromarray(m(width, height, dark, light).astype(d), 'L')
8892 data = m (width , height , dark , light ).astype (d )
89- if index is not None :
93+ if self . numbering and index is not None :
9094 text = "%d" % index
9195 size = tuple (d + 2 for d in self ._font .getsize (text ))
9296 img = Image .new ('L' , size )
@@ -150,6 +154,10 @@ def __init__(self, **kwargs):
150154 self ._image_generator .data_type ,
151155 self ._image_generator .set_data_type ,
152156 self ._image_generator .get_data_types )
157+ self .add_setting ('display image number' , 'bool' ,
158+ lambda : self ._image_generator .numbering ,
159+ self ._image_generator .enable_numbering ,
160+ None )
153161 # Software buffers and parameters for data conversion.
154162 self ._a_setting = 0
155163 self .add_setting ('a_setting' , 'int' ,
0 commit comments