##// END OF EJS Templates
schain
RSS
Using threading.Lock in jroplotter.py
Miguel Valdez -
r707:17810a6cdcd3
parent child
Show More
Show file before | Show file after | Hide comments
@@ -1,111 +1,137
1 '''
1 '''
2 Created on Jul 9, 2014
2 Created on Jul 9, 2014
3
3
4 @author: roj-idl71
4 @author: roj-idl71
5 '''
5 '''
6 import os
6 import os
7 import datetime
7 import datetime
8 import numpy
8 import numpy
9
9
10 from time import sleep
10 from time import sleep
11 from Queue import Queue
11 from Queue import Queue
12 from threading import Thread
12 from threading import Lock
13 # from threading import Thread
13
14
14 from schainpy.model.proc.jroproc_base import Operation
15 from schainpy.model.proc.jroproc_base import Operation
15 from schainpy.model.serializer.data import obj2Dict, dict2Obj
16 from schainpy.model.serializer.data import obj2Dict, dict2Obj
16 from jroplot_correlation import *
17 from jroplot_correlation import *
17 from jroplot_heispectra import *
18 from jroplot_heispectra import *
18 from jroplot_parameters import *
19 from jroplot_parameters import *
19 from jroplot_spectra import *
20 from jroplot_spectra import *
20 from jroplot_voltage import *
21 from jroplot_voltage import *
21
22
22
23
23 class Plotter(Operation):
24 class Plotter(Operation):
24
25
25 isConfig = None
26 isConfig = None
26 name = None
27 name = None
27 __queue = None
28 __queue = None
28
29
29 def __init__(self, plotter_name, plotter_queue=None):
30 def __init__(self, plotter_name, plotter_queue=None):
30
31
31 Operation.__init__(self)
32 Operation.__init__(self)
32
33
33 self.isConfig = False
34 self.isConfig = False
34 self.name = plotter_name
35 self.name = plotter_name
35 self.__queue = plotter_queue
36 self.__queue = plotter_queue
36
37
37 def getSubplots(self):
38 def getSubplots(self):
38
39
39 nrow = self.nplots
40 nrow = self.nplots
40 ncol = 1
41 ncol = 1
41 return nrow, ncol
42 return nrow, ncol
42
43
43 def setup(self, **kwargs):
44 def setup(self, **kwargs):
44
45
45 print "Initializing ..."
46 print "Initializing ..."
46
47
47
48
48 def run(self, dataOut, id=None, **kwargs):
49 def run(self, dataOut, id=None, **kwargs):
49
50
50 """
51 """
51
52
52 Input:
53 Input:
53 dataOut :
54 dataOut :
54 id :
55 id :
55 """
56 """
56
57
57 packDict = {}
58 packDict = {}
58
59
59 packDict['id'] = id
60 packDict['id'] = id
60 packDict['name'] = self.name
61 packDict['name'] = self.name
61 packDict['kwargs'] = kwargs
62 packDict['kwargs'] = kwargs
62
63
63 packDict['data'] = obj2Dict(dataOut)
64 packDict['data'] = obj2Dict(dataOut)
64
65
65 self.__queue.put(packDict)
66 self.__queue.put(packDict)
66
67
67 class PlotterManager(Thread):
68 # class PlotManager(Thread):
68
69 class PlotManager():
69 __stop = False
70 __stop = False
70
71
71 def __init__(self, plotter_queue):
72 def __init__(self, plotter_queue):
72
73
73 Thread.__init__(self)
74 # Thread.__init__(self)
74
75 # self.setDaemon(True)
75 self.setDaemon(True)
76
76
77 self.__queue = plotter_queue
77 self.__queue = plotter_queue
78 self.__lock = Lock()
78
79
79 self.plotInstanceDict = {}
80 self.plotInstanceDict = {}
80 self.__stop = False
81 self.__stop = False
81
82
82 def run(self):
83 def run(self):
83
84
84 while True:
85 if self.__queue.empty():
85
86 return
86 if self.__stop:
87
87 break
88 self.__lock.acquire()
89
90 # if self.__queue.full():
91 # for i in range(int(self.__queue.qsize()/2)):
92 # serial_data = self.__queue.get()
93 # self.__queue.task_done()
88
94
89 if self.__queue.empty():
95 n = int(self.__queue.qsize()/3 + 1)
90 sleep(0.1)
96
91 continue
97 for i in range(n):
92
98
93 serial_data = self.__queue.get(True)
99 serial_data = self.__queue.get()
100 self.__queue.task_done()
94
101
95 plot_id = serial_data['id']
102 plot_id = serial_data['id']
96 plot_name = serial_data['name']
103 plot_name = serial_data['name']
97 kwargs = serial_data['kwargs']
104 kwargs = serial_data['kwargs']
98 dataDict = serial_data['data']
105 dataDict = serial_data['data']
99
106
100 dataPlot = dict2Obj(dataDict)
107 dataPlot = dict2Obj(dataDict)
101
108
102 if plot_id not in self.plotInstanceDict.keys():
109 if plot_id not in self.plotInstanceDict.keys():
103 className = eval(plot_name)
110 className = eval(plot_name)
104 self.plotInstanceDict[plot_id] = className()
111 self.plotInstanceDict[plot_id] = className()
105
112
106 plotter = self.plotInstanceDict[plot_id]
113 plotter = self.plotInstanceDict[plot_id]
107 plotter.run(dataPlot, plot_id, **kwargs)
114 plotter.run(dataPlot, plot_id, **kwargs)
108
115
116
117
118 self.__lock.release()
119
109 def stop(self):
120 def stop(self):
110
121
111 self.__stop = True No newline at end of file
122 self.__lock.acquire()
123
124 self.__stop = True
125
126 self.__lock.release()
127
128 def close(self):
129
130 self.__lock.acquire()
131
132 for plot_id in self.plotInstanceDict.keys():
133 plotter = self.plotInstanceDict[plot_id]
134 plotter.close()
135
136 self.__lock.release()
137 No newline at end of file
General Comments 0
You need to be logged in to leave comments. Login now