Add info after scale bar has been drawn (ppinard#67)

Author: ppinard

README.md

@@ -108,6 +108,35 @@ ax.add_artist(scalebar)
 fig.savefig("original_resolution.png", dpi=dpi)
 ```
 
+### Information about the drawn scale bar
+
+After the scale bar has been drawn, the `info` property returns the following dataclass.
+
+```python
+@dataclasses.dataclass
+class ScaleBarInfo:
+    length_px: int
+    value: float
+    units: str
+    scale_text: str
+    window_extent: matplotlib.transforms.Bbox
+```
+
+Note that the `info` property returns a `ValueError` exception if the scale bar has not been drawn.
+
+```python
+fig, ax = plt.subplots()
+
+scalebar = ScaleBar(0.08, "cm", length_fraction=0.25)
+ax.add_artist(scalebar)
+
+print(scalebar.info) # raises a ValueError exception
+
+fig.canvas.draw()
+
+print(scalebar.info) # works
+```
+
 ## ScaleBar arguments
 
 Here are arguments of the **ScaleBar** class constructor and examples how to use them.

matplotlib_scalebar/scalebar.py

@@ -43,6 +43,7 @@
 # Standard library modules.
 import bisect
 import warnings
+import dataclasses
 
 # Third party modules.
 import matplotlib
@@ -151,6 +152,15 @@ def _validate_legend_loc(loc):
 }
 
 
+@dataclasses.dataclass
+class ScaleBarInfo:
+    length_px: int
+    value: float
+    units: str
+    scale_text: str
+    window_extent: matplotlib.transforms.Bbox
+
+
 class ScaleBar(Artist):
     zorder = 6
 
@@ -379,6 +389,7 @@ def __init__(
         self.rotation = rotation
         self.bbox_to_anchor = bbox_to_anchor
         self.bbox_transform = bbox_transform
+        self._info = None
 
     def _calculate_best_length(self, length_px):
         dx = self.dx
@@ -404,6 +415,8 @@ def _calculate_exact_length(self, value, units):
         return newvalue / self.dx
 
     def draw(self, renderer, *args, **kwargs):
+        self._info = None
+
         if not self.get_visible():
             return
         if self.dx == 0:
@@ -566,6 +579,10 @@ def _get_value(attr, default):
         box.patch.set_alpha(box_alpha)
         box.draw(renderer)
 
+        self._info = ScaleBarInfo(
+            length_px, value, units, scale_text, box.get_window_extent(renderer)
+        )
+
     def get_dx(self):
         return self._dx
 
@@ -864,3 +881,9 @@ def set_bbox_transform(self, bbox_transform):
         self._bbox_transform = bbox_transform
 
     bbox_transform = property(get_bbox_transform, set_bbox_transform)
+
+    @property
+    def info(self):
+        if self._info is None:
+            raise ValueError("Scale bar has not been drawn. Call figure.canvas.draw()")
+        return self._info