''' Created on Jul 9, 2014 @author: roj-idl71 ''' import os import datetime import numpy from time import sleep from Queue import Queue from threading import Lock # from threading import Thread from schainpy.model.proc.jroproc_base import Operation from schainpy.model.serializer.data import obj2Dict, dict2Obj from jroplot_correlation import * from jroplot_heispectra import * from jroplot_parameters import * from jroplot_spectra import * from jroplot_voltage import * class Plotter(Operation): isConfig = None name = None __queue = None def __init__(self, plotter_name, plotter_queue=None): Operation.__init__(self) self.isConfig = False self.name = plotter_name self.__queue = plotter_queue def getSubplots(self): nrow = self.nplots ncol = 1 return nrow, ncol def setup(self, **kwargs): print "Initializing ..." def run(self, dataOut, id=None, **kwargs): """ Input: dataOut : id : """ packDict = {} packDict['id'] = id packDict['name'] = self.name packDict['kwargs'] = kwargs packDict['data'] = obj2Dict(dataOut) self.__queue.put(packDict) # class PlotManager(Thread): class PlotManager(): __stop = False controllerThreadObj = None def __init__(self, plotter_queue): # Thread.__init__(self) # self.setDaemon(True) self.__queue = plotter_queue self.__lock = Lock() self.plotInstanceDict = {} self.__stop = False def run(self): 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() n = int(self.__queue.qsize()/3 + 1) for i in range(n): if self.__queue.empty(): break serial_data = self.__queue.get() self.__queue.task_done() 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) self.__lock.release() def isEmpty(self): return self.__queue.empty() def stop(self): 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() 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()