Added EC plotter

[nepi.git] / src / nepi / execution / ec.py

diff --git a/src/nepi/execution/ec.py b/src/nepi/execution/ec.py
index cf41983..31e11ef 100644 (file)
--- a/src/nepi/execution/ec.py
+++ b/src/nepi/execution/ec.py
@@ -25,6 +25,7 @@ from nepi.execution.resource import ResourceFactory, ResourceAction, \
 from nepi.execution.scheduler import HeapScheduler, Task, TaskStatus
 from nepi.execution.trace import TraceAttr
 from nepi.util.serializer import ECSerializer, SFormats
+from nepi.util.plotter import ECPlotter, PFormats
 
 # TODO: use multiprocessing instead of threading
 # TODO: Allow to reconnect to a running experiment instance! (reconnect mode vs deploy mode)
@@ -381,6 +382,12 @@ class ExperimentController(object):
 
                 time.sleep(0.5)
 
+    def plot(self, fpath = None, format= PFormats.FIGURE, persist = False):
+        plotter = ECPlotter()
+        fpath = plotter.plot(self, fpath = fpath, format= format, 
+                persist = persist)
+        return fpath
+
     def serialize(self, format = SFormats.XML):
         serializer = ECSerializer()
         sec = serializer.load(self, format = format)