add option to not show lvatmo limits in table plot

3 files changed, 8 insertions, 5 deletions

diff --git a/misc/command b/misc/command
index db34921..3ec2d6c 100644
--- a/misc/command
+++ b/misc/command
@@ -5,3 +5,4 @@ python plot_sens.py --data real --datadir /data/user/smandalia/flavour_ratio/dat
 python mc_unitary.py --burnin 100 --datadir=/data/user/smandalia/flavour_ratio/data/mc_unitary --nsteps 1000 --nwalkers 60 --run-mcmc True --seed 26 --source-ratio 1 2 0 --threads 1
 python mc_x.py --burnin 200 --datadir=/data/user/smandalia/flavour_ratio/data/mc_x --nsteps 1000 --nwalkers 60 --seed 26 --threads max
 python fr.py --data real --datadir ./test/ --dimension 6 --nsteps 20 --nwalkers 60 --burnin 20 --seed 26 --source-ratio 0 1 0 --texture oet --threads max --debug True
+python plot_sens.py --data real --datadir /data/user/smandalia/flavour_ratio/data/sensitivity/ --dimensions 3 4 5 6 7 8 --plot-table True --segments 10 --source-ratios 0 1 0 1 0 0 --split-jobs True --stat-method bayesian --plot-statistic False
diff --git a/plot_sens.py b/plot_sens.py
index 35fed5c..bd8c72b 100755
--- a/plot_sens.py
+++ b/plot_sens.py
@@ -284,6 +284,7 @@ def main():
             outfile   = baseoutfile + '/hese_table',
             outformat = ['png', 'pdf'],
             args      = args,
+            show_lvatmo = True
         )
 
 
diff --git a/utils/plot.py b/utils/plot.py
index 8513317..d19a52e 100644
--- a/utils/plot.py
+++ b/utils/plot.py
@@ -619,7 +619,7 @@ def plot_statistic(data, outfile, outformat, args, scale_param, label=None):
         fig.savefig(outfile+'.'+of, bbox_inches='tight', dpi=150)
 
 
-def plot_table_sens(data, outfile, outformat, args):
+def plot_table_sens(data, outfile, outformat, args, show_lvatmo=True):
     print 'Making TABLE sensitivity plot'
     argsc = deepcopy(args)
 
@@ -726,7 +726,7 @@ def plot_table_sens(data, outfile, outformat, args):
                               edgecolor=rgb_co[isrc]+[1], label=label)
                     )
 
-            if itex == len(textures)-1:
+            if itex == len(textures)-1 and show_lvatmo:
                 LV_lim = np.log10(LV_ATMO_90PC_LIMITS[dim])
                 ax.add_patch(patches.Rectangle(
                     (LV_lim[1], ylims[0]), LV_lim[0]-LV_lim[1], np.diff(ylims),
@@ -739,9 +739,10 @@ def plot_table_sens(data, outfile, outformat, args):
     ax.tick_params(axis='x', labelsize=16)
 
     purple = [0.5019607843137255, 0.0, 0.5019607843137255]
-    legend_elements.append(
-        Patch(facecolor='none', hatch='\\\\', edgecolor='k', label='IceCube, Nature.Phy.14,961(2018)')
-    )
+    if show_lvatmo:
+        legend_elements.append(
+            Patch(facecolor='none', hatch='\\\\', edgecolor='k', label='IceCube, Nature.Phy.14,961(2018)')
+        )
     legend_elements.append(
         Patch(facecolor=purple+[0.7], edgecolor=purple+[1], label='Planck Scale Expectation')
     )