Add option to update legend in main figure using items from ratio plot

demarley · Sep 14, 2018 · 1d78e7e · 1d78e7e
1 parent 1f2078a
commit 1d78e7e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/histogram1D.py b/python/histogram1D.py
@@ -286,6 +286,11 @@ def plotRatio(self):
         formatter = FormatStrFormatter('%g')
         self.ax2.set_yticklabels(np.array([formatter(i) for i in axis_ticks]))
 
+        if self.ratio.update_legend:
+            extra_handles,extra_labels = self.ax2.get_legend_handles_labels()
+            self.legend['extra_handles'] = extra_handles
+            self.legend['extra_labels']  = extra_labels
+
         return
 
 
@@ -365,6 +370,7 @@ def __init__(self):
         self.ylim   = None
         self.yticks = None
         self.ylabel = ''
+        self.update_legend = False   # add ratio values to legend on main plot
 
     def initialize(self):
         """Set some default options if not set by the user"""