jroplotter.py.svn-base
239 lines
| 6.2 KiB
| text/plain
|
TextLexer
ebocanegra
|
r965 | ''' | |
| Created on Jul 9, 2014 | |||
| @author: roj-idl71 | |||
| ''' | |||
| import os, sys | |||
| import datetime | |||
| import numpy | |||
| import traceback | |||
| from time import sleep | |||
| from threading import Lock | |||
| # from threading import Thread | |||
| import schainpy | |||
| import schainpy.admin | |||
| 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) | |||
| packDict['data'] = dataOut | |||
| self.__queue.put(packDict) | |||
| # class PlotManager(Thread): | |||
| class PlotManager(): | |||
| __err = False | |||
| __stop = False | |||
| __realtime = False | |||
| controllerThreadObj = None | |||
| plotterList = ['Scope', | |||
| 'SpectraPlot', 'RTIPlot', | |||
| 'SpectraCutPlot', | |||
| 'CrossSpectraPlot', 'CoherenceMap', | |||
| 'PowerProfilePlot', 'Noise', 'BeaconPhase', | |||
| 'CorrelationPlot', | |||
| 'SpectraHeisScope','RTIfromSpectraHeis'] | |||
| def __init__(self, plotter_queue): | |||
| # Thread.__init__(self) | |||
| # self.setDaemon(True) | |||
| self.__queue = plotter_queue | |||
| self.__lock = Lock() | |||
| self.plotInstanceDict = {} | |||
| self.__err = False | |||
| self.__stop = False | |||
| self.__realtime = False | |||
| def __handleError(self, name="", send_email=False): | |||
| err = traceback.format_exception(sys.exc_info()[0], | |||
| sys.exc_info()[1], | |||
| sys.exc_info()[2]) | |||
| print "***** Error occurred in PlotManager *****" | |||
| print "***** [%s]: %s" %(name, err[-1]) | |||
| message = "\nError ocurred in %s:\n" %name | |||
| message += "".join(err) | |||
| sys.stderr.write(message) | |||
| if not send_email: | |||
| return | |||
| import socket | |||
| subject = "SChain v%s: Error running %s\n" %(schainpy.__version__, name) | |||
| subtitle = "%s:\n" %(name) | |||
| subtitle += "Hostname: %s\n" %socket.gethostbyname(socket.gethostname()) | |||
| subtitle += "Working directory: %s\n" %os.path.abspath("./") | |||
| # subtitle += "Configuration file: %s\n" %self.filename | |||
| subtitle += "Time: %s\n" %str(datetime.datetime.now()) | |||
| adminObj = schainpy.admin.SchainNotify() | |||
| adminObj.sendAlert(message=message, | |||
| subject=subject, | |||
| subtitle=subtitle) | |||
| def run(self): | |||
| if self.__queue.empty(): | |||
| return | |||
| if self.__err: | |||
| serial_data = self.__queue.get() | |||
| self.__queue.task_done() | |||
| 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) | |||
| dataPlot = serial_data['data'] | |||
| if plot_id not in self.plotInstanceDict.keys(): | |||
| className = eval(plot_name) | |||
| self.plotInstanceDict[plot_id] = className() | |||
| plotter = self.plotInstanceDict[plot_id] | |||
| try: | |||
| plotter.run(dataPlot, plot_id, **kwargs) | |||
| except: | |||
| self.__err = True | |||
| self.__handleError(plot_name, send_email=True) | |||
| break | |||
| 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() | |||
| def isErrorDetected(self): | |||
| self.__lock.acquire() | |||
| err = self.__err | |||
| self.__lock.release() | |||
| return err |