Save plots.

libraries/base/common/python/try_round_weight.py

@@ -45,10 +45,12 @@
 #       . Choose sigmas_ratio_qq_sq_T < 10%   ==> -r 3
 #       . Choose sigmas_ratio_qq_sq_T <  1%   ==> -r 6
 #       . Choose sigmas_ratio_qq_sq_T <  0.1% ==> -r 9
+#       . Hence the disturbance (jumps) on the sigma is about proportional
+#         to 1/2**r
 # Note:
 # . For values exactly halfway between rounded decimal values, NumPy of
 #   Python3 rounds to the nearest even value. Thus 1.5 and 2.5 round to 2.0,
-#   -0.5 and 0.5 round to 0.0, etc. (Python2 rounds half away from zero).
+#   -0.5 and 0.5 round to 0.0, etc. Python2 rounds half away from zero.
 # Usage:
 #   > python3 try_round_weight.py -N 195312
 
@@ -87,6 +89,11 @@ _parser = argparse.ArgumentParser(
         # resolution before rounding
         > python try_round_weight.py --w_lo 0.7 --w_hi 0.8 --w_step 0.0001 --s_lo 1 --s_hi 10 --s_step 1 -N 195312 -S 0 -r 6
         > python try_round_weight.py --w_lo 0.3 --w_hi 1.1 --w_step 0.001 --s_lo 1 --s_hi 10 --s_step 1 -N 195312 -S 0 -r 6
+
+        # Reproduce plots/
+        > python try_round_weight.py --w_lo 0.3 --w_hi 1.1 --w_step 0.001 --s_lo 1 --s_hi 10 --s_step 1 -N 195312 -S 0 -r 0 --noplot --save
+        > python try_round_weight.py --w_lo 0.3 --w_hi 1.1 --w_step 0.001 --s_lo 1 --s_hi 10 --s_step 1 -N 195312 -S 0 -r 4 --noplot --save
+        > python try_round_weight.py --w_lo 0.3 --w_hi 1.1 --w_step 0.001 --s_lo 1 --s_hi 10 --s_step 1 -N 195312 -S 0 -r 6 --noplot --save
         \n""")),
     formatter_class=argparse.RawTextHelpFormatter)
 _parser.add_argument('-S', default=0, type=int, help='Random number seed')
@@ -99,6 +106,8 @@ _parser.add_argument('--s_step', default=0.1, type=float, help='Step sigma')
 _parser.add_argument('--w_lo', default=0.3, type=float, help='Lowest weight')
 _parser.add_argument('--w_hi', default=2.0, type=float, help='Highest weight')
 _parser.add_argument('--w_step', default=0.1, type=float, help='Step weight')
+_parser.add_argument('--noplot', action="store_true", help='Do not show plots')
+_parser.add_argument('--save', action="store_true", help='Do save plots')
 args = _parser.parse_args()
 
 np.random.seed(args.S)
@@ -167,6 +176,7 @@ sigmas_sq_T = sigmas_sq.transpose()
 
 # Plot results
 figNr = 0
+dpi = 254  # 10 dots per mm
 s_colors = plt.cm.jet(np.linspace(0, 1, N_sigmas))
 w_colors = plt.cm.jet(np.linspace(0, 1, N_weights))
 
@@ -181,6 +191,8 @@ plt.xlabel("Weight")
 plt.ylabel("Sigma_qq")
 plt.legend(loc='upper right')
 plt.grid()
+if args.save:
+    plt.savefig('plots/try_round_weight_r%d_w_sigma_qq.jpg' % resolution, dpi=dpi)
 
 figNr += 1
 plt.figure(figNr)
@@ -196,6 +208,8 @@ plt.xlabel("Weight")
 plt.ylabel("Sigma_qq")
 plt.legend(loc='upper right')
 plt.grid()
+if args.save:
+    plt.savefig('plots/try_round_weight_r%d_w_sigma_qq_normalized.jpg' % resolution, dpi=dpi)
 
 figNr += 1
 plt.figure(figNr)
@@ -210,6 +224,8 @@ plt.xlabel("Weight")
 plt.ylabel("Relative sigma difference")
 plt.legend(loc='upper right')
 plt.grid()
+if args.save:
+    plt.savefig('plots/try_round_weight_r%d_w_sigmas_ratio_sq_input.jpg' % resolution, dpi=dpi)
 
 figNr += 1
 plt.figure(figNr)
@@ -224,6 +240,8 @@ plt.xlabel("Weight")
 plt.ylabel("Relative sigma difference")
 plt.legend(loc='upper right')
 plt.grid()
+if args.save:
+    plt.savefig('plots/try_round_weight_r%d_w_sigmas_ratio_qq_sq.jpg' % resolution, dpi=dpi)
 
 figNr += 1
 plt.figure(figNr)
@@ -236,6 +254,8 @@ plt.xlabel("Sigma")
 plt.ylabel("Relative sigma difference (s_qq / s_sq)")
 plt.legend(loc='upper right')
 plt.grid()
+if args.save:
+    plt.savefig('plots/try_round_weight_r%d_s_sigmas_ratio_qq_sq.jpg' % resolution, dpi=dpi)
 
 figNr += 1
 plt.figure(figNr)
@@ -245,5 +265,8 @@ plt.title("Relative sigma difference of weighting after / before quantisation")
 plt.xlabel("Sigma")
 plt.ylabel("Weight")
 plt.grid()
+if args.save:
+    plt.savefig('plots/try_round_weight_r%d_sw_sigmas_ratio_qq_sq.jpg' % resolution, dpi=dpi)
 
-plt.show()
+if not args.noplot:
+    plt.show()