@@ -28,13 +28,17 @@ def gentle_compare(w0, w1):
28
28
return True
29
29
30
30
31
- def rms (a , b ):
31
+ def rms (a , b , verbose = None , out_prefix = '' ):
32
32
'''
33
33
Returns RMS diff of raw bytes of two sequences.
34
34
'''
35
+ if verbose is True :
36
+ verbose = 100000
35
37
assert len (a ) == len (b )
36
38
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 = } .' )
38
42
e += (aa - bb ) ** 2
39
43
rms = math .sqrt (e / len (a ))
40
44
return rms
@@ -58,14 +62,9 @@ def pixmaps_rms(a, b, out_prefix=''):
58
62
a_mv = a .samples_mv
59
63
b_mv = b .samples_mv
60
64
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
69
68
70
69
71
70
def pixmaps_diff (a , b , out_prefix = '' ):
0 commit comments