@@ -28,13 +28,17 @@ def gentle_compare(w0, w1):
2828 return True
2929
3030
31- def rms (a , b ):
31+ def rms (a , b , verbose = None , out_prefix = '' ):
3232 '''
3333 Returns RMS diff of raw bytes of two sequences.
3434 '''
35+ if verbose is True :
36+ verbose = 100000
3537 assert len (a ) == len (b )
3638 e = 0
37- for aa , bb in zip (a , b ):
39+ for i , (aa , bb ) in enumerate (zip (a , b )):
40+ if verbose and (i % verbose == 0 ):
41+ print (f'{ out_prefix } rms(): { i = } { e = } { aa = } { aa = } .' )
3842 e += (aa - bb ) ** 2
3943 rms = math .sqrt (e / len (a ))
4044 return rms
@@ -58,14 +62,9 @@ def pixmaps_rms(a, b, out_prefix=''):
5862 a_mv = a .samples_mv
5963 b_mv = b .samples_mv
6064 assert len (a_mv ) == len (b_mv )
61- e = 0
62- for i , (a_byte , b_byte ) in enumerate (zip (a_mv , b_mv )):
63- if i % 100000 == 0 :
64- print (f'{ out_prefix } compare_pixmaps(): { i = } { e = } { a_byte = } { b_byte = } .' )
65- e += (a_byte - b_byte ) ** 2
66- rms = math .sqrt (e / len (a_mv ))
67- print (f'{ out_prefix } compare_pixmaps(): { e = } { rms = } .' )
68- return rms
65+ ret = rms (a_mv , b_mv , verbose = True , out_prefix = out_prefix )
66+ print (f'{ out_prefix } pixmaps_rms(): { ret = } .' )
67+ return ret
6968
7069
7170def pixmaps_diff (a , b , out_prefix = '' ):
0 commit comments