compute SNR for accs

tests/test_base.py

@@ -162,24 +162,26 @@ def acceleration_test(c, m = 3, species = 9):
     num_acc2 = sum(fc[2, n-i]*pos[n-i:pos.shape[0]-i-n] for i in range(-n, n+1)) / dt**2
 
     pid = np.unravel_index(np.argmax(np.abs(num_acc1[:] - acc[n:-n])), dims = num_acc1.shape)
-    print np.abs(num_acc1[:] - acc[n:-n])[pid[0], pid[1]]
-    for cc in range(3):
-        a.plot(num_acc1[:, pid[1], cc], color = col[cc])
-        #a.plot(num_acc2[:, pid[1], cc], color = col[cc], dashes = (2, 2))
-        a.plot(acc[m:, pid[1], cc], color = col[cc], dashes = (1, 1))
-
-    for n in range(1, m):
-        fc = get_fornberg_coeffs(0, range(-n, n+1))
-        dt = d['parameters/dt'].value*d['parameters/niter_part'].value
-
-        num_acc1 = sum(fc[1, n-i]*vel[n-i:vel.shape[0]-i-n] for i in range(-n, n+1)) / dt
-        num_acc2 = sum(fc[2, n-i]*pos[n-i:pos.shape[0]-i-n] for i in range(-n, n+1)) / dt**2
-
-        for cc in range(3):
-            a.plot(num_acc1[m-n:, pid[1], cc], color = col[cc])
-            #a.plot(num_acc2[m-n:, pid[1], cc], color = col[cc], dashes = (2, 2))
-    fig.tight_layout()
-    fig.savefig('acc_test_{0}_{1}.pdf'.format(c.simname, species))
+    snrv = -20*np.log10(np.average((num_acc1[:] - acc[n:-n])**2) / np.average((acc[n:-n])**2))
+    snrp = -20*np.log10(np.average((num_acc2[:] - acc[n:-n])**2) / np.average((acc[n:-n])**2))
+    print('SNR diffvel vs acc {0}, SNR diffpos vs acc {1}'.format(snrv, snrp))
+    #for cc in range(3):
+    #    a.plot(num_acc1[:, pid[1], cc], color = col[cc])
+    #    #a.plot(num_acc2[:, pid[1], cc], color = col[cc], dashes = (2, 2))
+    #    a.plot(acc[m:, pid[1], cc], color = col[cc], dashes = (1, 1))
+
+    #for n in range(1, m):
+    #    fc = get_fornberg_coeffs(0, range(-n, n+1))
+    #    dt = d['parameters/dt'].value*d['parameters/niter_part'].value
+
+    #    num_acc1 = sum(fc[1, n-i]*vel[n-i:vel.shape[0]-i-n] for i in range(-n, n+1)) / dt
+    #    num_acc2 = sum(fc[2, n-i]*pos[n-i:pos.shape[0]-i-n] for i in range(-n, n+1)) / dt**2
+
+    #    for cc in range(3):
+    #        a.plot(num_acc1[m-n:, pid[1], cc], color = col[cc])
+    #        #a.plot(num_acc2[m-n:, pid[1], cc], color = col[cc], dashes = (2, 2))
+    #fig.tight_layout()
+    #fig.savefig('acc_test_{0}_{1}.pdf'.format(c.simname, species))
     return None
 
 if __name__ == '__main__':