diff options
Diffstat (limited to 'plot_sens.py')
| -rwxr-xr-x | plot_sens.py | 40 |
1 files changed, 39 insertions, 1 deletions
diff --git a/plot_sens.py b/plot_sens.py index 79aa6a9..35fed5c 100755 --- a/plot_sens.py +++ b/plot_sens.py @@ -35,6 +35,34 @@ def process_args(args): if args.data is not DataType.REAL: args.injected_ratio = fr_utils.normalise_fr(args.injected_ratio) + # Anon points + anon = [] + if args.dimensions[0] == 3: + anon.append([0.825, 0.845]) + anon.append([0.865, 0.875]) + anon.append([0.875, 0.885]) + anon.append([0.905, 0.915]) + anon.append([0.925, 0.935]) + if args.dimensions[0] == 4: + anon.append([0.165, 0.175]) + anon.append([0.805, 0.825]) + anon.append([0.835, 0.845]) + anon.append([0.855, 0.885]) + anon.append([0.965, 0.975]) + if args.dimensions[0] == 5: + anon.append([0.895, 0.905]) + anon.append([0.955, 0.965]) + if args.dimensions[0] == 6: + anon.append([0.115, 0.125]) + anon.append([0.855, 0.865]) + if args.dimensions[0] == 7: + # anon.append([0.815, 0.835]) + anon.append([0.875, 0.885]) + if args.dimensions[0] == 8: + anon.append([0.915, 0.935]) + anon.append([0.875, 0.895]) + anon.append([0.845, 0.855]) + if args.source_ratios is not None: if args.x_segments is not None: raise ValueError('Cannot do both --source-ratios and --x-segments') @@ -50,7 +78,14 @@ def process_args(args): x_array = np.linspace(0, 1, args.x_segments) sources = [] for x in x_array: - if x > MASK_X[0] and x < MASK_X[1]: continue + if x >= MASK_X[0] and x <= MASK_X[1]: continue + skip = False + for a in anon: + if (a[1] > x) & (x > a[0]): + print 'Skipping src', x + skip = True + break + if skip: continue sources.append([x, 1-x, 0]) args.source_ratios = sources else: @@ -124,6 +159,8 @@ def main(): srcs = len(args.source_ratios) if args.texture is Texture.NONE: textures = [Texture.OEU, Texture.OET, Texture.OUT] + if args.plot_table: + textures = [Texture.OET, Texture.OUT] else: textures = [args.texture] texs = len(textures) @@ -149,6 +186,7 @@ def main(): for isrc, src in enumerate(args.source_ratios): argsc.source_ratio = src + for itex, texture in enumerate(textures): argsc.texture = texture |