Merge pull request #18 from CJ-Wright/waterfall_callback

Waterfall callback
xpdAcq · Aug 28, 2017 · 338c1bd · 338c1bd
2 parents 869a78e + 56915ca
commit 338c1bd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/xpdview/callbacks.py b/xpdview/callbacks.py
@@ -1,5 +1,6 @@
 from bluesky.callbacks.broker import BrokerCallbackBase
 from .waterfall import Waterfall
+import matplotlib.pyplot as plt
 
 
 class LiveWaterfall(BrokerCallbackBase):
@@ -25,11 +26,11 @@ def __init__(self, x_name, y_name, db=None, units=None):
         self.y_name = y_name
         self.units = units
 
-        from matplotlib.figure import Figure
-        self.fig = Figure()
+        self.fig = plt.figure()
 
         self.wf = Waterfall(fig=self.fig, unit=self.units)
         self.i = 0
+        self.fig.show()
 
     def start(self, doc):
         self.i = 0