jroplotter.py
166 lines
| 4.0 KiB
| text/x-python
|
PythonLexer
Miguel Valdez
|
r691 | ''' | |
| Created on Jul 9, 2014 | |||
| @author: roj-idl71 | |||
| ''' | |||
| import os | |||
| import datetime | |||
| import numpy | |||
|
Miguel Valdez
|
r698 | from time import sleep | |
| from Queue import Queue | |||
|
Miguel Valdez
|
r707 | from threading import Lock | |
| # from threading import Thread | |||
|
Miguel Valdez
|
r691 | ||
|
Miguel Valdez
|
r698 | from schainpy.model.proc.jroproc_base import Operation | |
| from schainpy.model.serializer.data import obj2Dict, dict2Obj | |||
|
Miguel Valdez
|
r701 | from jroplot_correlation import * | |
| from jroplot_heispectra import * | |||
| from jroplot_parameters import * | |||
| from jroplot_spectra import * | |||
| from jroplot_voltage import * | |||
|
Miguel Valdez
|
r698 | ||
| class Plotter(Operation): | |||
|
Miguel Valdez
|
r691 | ||
| isConfig = None | |||
| name = None | |||
|
Miguel Valdez
|
r698 | __queue = None | |
|
Miguel Valdez
|
r691 | ||
| def __init__(self, plotter_name, plotter_queue=None): | |||
|
Miguel Valdez
|
r698 | Operation.__init__(self) | |
|
Miguel Valdez
|
r691 | self.isConfig = False | |
| self.name = plotter_name | |||
|
Miguel Valdez
|
r698 | self.__queue = plotter_queue | |
|
Miguel Valdez
|
r691 | ||
| def getSubplots(self): | |||
| nrow = self.nplots | |||
| ncol = 1 | |||
| return nrow, ncol | |||
| def setup(self, **kwargs): | |||
| print "Initializing ..." | |||
|
Miguel Valdez
|
r698 | def run(self, dataOut, id=None, **kwargs): | |
|
Miguel Valdez
|
r691 | ||
| """ | |||
| Input: | |||
| dataOut : | |||
| id : | |||
| """ | |||
|
Miguel Valdez
|
r698 | packDict = {} | |
| packDict['id'] = id | |||
| packDict['name'] = self.name | |||
| packDict['kwargs'] = kwargs | |||
| packDict['data'] = obj2Dict(dataOut) | |||
| self.__queue.put(packDict) | |||
|
Miguel Valdez
|
r707 | # class PlotManager(Thread): | |
| class PlotManager(): | |||
|
Miguel Valdez
|
r698 | __stop = False | |
|
Miguel Valdez
|
r716 | controllerThreadObj = None | |
|
Miguel Valdez
|
r698 | ||
| def __init__(self, plotter_queue): | |||
|
Miguel Valdez
|
r707 | # Thread.__init__(self) | |
| # self.setDaemon(True) | |||
|
Miguel Valdez
|
r703 | ||
|
Miguel Valdez
|
r698 | self.__queue = plotter_queue | |
|
Miguel Valdez
|
r707 | self.__lock = Lock() | |
|
Miguel Valdez
|
r698 | ||
| self.plotInstanceDict = {} | |||
| self.__stop = False | |||
| def run(self): | |||
|
Miguel Valdez
|
r707 | if self.__queue.empty(): | |
| return | |||
| self.__lock.acquire() | |||
| # if self.__queue.full(): | |||
| # for i in range(int(self.__queue.qsize()/2)): | |||
| # serial_data = self.__queue.get() | |||
| # self.__queue.task_done() | |||
|
Miguel Valdez
|
r698 | ||
|
Miguel Valdez
|
r707 | n = int(self.__queue.qsize()/3 + 1) | |
| for i in range(n): | |||
|
Miguel Valdez
|
r698 | ||
|
Miguel Valdez
|
r716 | if self.__queue.empty(): | |
| break | |||
|
Miguel Valdez
|
r707 | serial_data = self.__queue.get() | |
| self.__queue.task_done() | |||
|
Miguel Valdez
|
r698 | ||
| plot_id = serial_data['id'] | |||
| plot_name = serial_data['name'] | |||
| kwargs = serial_data['kwargs'] | |||
| dataDict = serial_data['data'] | |||
| dataPlot = dict2Obj(dataDict) | |||
| if plot_id not in self.plotInstanceDict.keys(): | |||
| className = eval(plot_name) | |||
| self.plotInstanceDict[plot_id] = className() | |||
| plotter = self.plotInstanceDict[plot_id] | |||
| plotter.run(dataPlot, plot_id, **kwargs) | |||
|
Miguel Valdez
|
r691 | ||
|
Miguel Valdez
|
r707 | self.__lock.release() | |
|
Miguel Valdez
|
r710 | ||
| def isEmpty(self): | |||
| return self.__queue.empty() | |||
|
Miguel Valdez
|
r698 | def stop(self): | |
|
Miguel Valdez
|
r691 | ||
|
Miguel Valdez
|
r707 | self.__lock.acquire() | |
| self.__stop = True | |||
| self.__lock.release() | |||
| def close(self): | |||
| self.__lock.acquire() | |||
| for plot_id in self.plotInstanceDict.keys(): | |||
| plotter = self.plotInstanceDict[plot_id] | |||
| plotter.close() | |||
| self.__lock.release() | |||
|
Miguel Valdez
|
r716 | ||
| def setController(self, controllerThreadObj): | |||
| self.controllerThreadObj = controllerThreadObj | |||
| def start(self): | |||
| if not self.controllerThreadObj.isRunning(): | |||
| raise RuntimeError, "controllerThreadObj has not been initialized. Use controllerThreadObj.start() before call this method" | |||
| self.join() | |||
| def join(self): | |||
| #Execute plotter while controller is running | |||
| while self.controllerThreadObj.isRunning(): | |||
| self.run() | |||
| self.controllerThreadObj.stop() | |||
| #Wait until plotter queue is empty | |||
| while not self.isEmpty(): | |||
| self.run() | |||
| self.close() |