Optionally allow the legend label to be specified directly.

rootpy · Mar 8, 2013 · bbcb64a · pwaller · Jun 6, 2013 · ekfriis
1 parent 4687e37
commit bbcb64a
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/rootpy/plotting/legend.py b/rootpy/plotting/legend.py
@@ -49,7 +49,15 @@ def Draw(self, *args, **kwargs):
         self.pad.Modified()
         self.pad.Update()
 
-    def AddEntry(self, thing, legendstyle=None):
+    def AddEntry(self, thing, legendstyle=None, label=None):
+        """
+        Add an entry to the legend.
+
+        If legendstyle is None, it will be taken from thing's
+        'legendstyle' attribute.
+
+        If label is None, the thing's title will be used as the label.
+        """
 
         if isinstance(thing, HistStack):
             things = thing
@@ -60,8 +68,10 @@ def AddEntry(self, thing, legendstyle=None):
                             type(thing))
         for hist in things:
             if hist.inlegend:
+                if label is None:
+                    label = hist.GetTitle()
                 if legendstyle is None:
                     legendstyle = hist.legendstyle
-                ROOT.TLegend.AddEntry(self, hist, hist.GetTitle(), legendstyle)
+                ROOT.TLegend.AddEntry(self, hist, label, legendstyle)
         self.pad.Modified()
         self.pad.Update()