class Actor(object):
def __init__(self, name, num_data, color, ix, max_actors, actor_height):
self.name = name
self.num_data = num_data
self.color = color
self.ix = ix
self.max_actors = max_actors
self.actor_height = actor_height
self.offset = ix * actor_height
self.data = np.array([self.offset + 30] * num_data)
self.head = 0
self.pre_head = 0
self.plotItem = PlotCurveItem(pen=pg.mkPen(color, width=3), width=4, name=name)
self.plotPoint = ScatterPlotItem(pen=pg.mkPen("w", width=5), brush=pg.mkBrush(color), size=5)
self.render()
def __str__(self):
return "<Actor name:%r, position=%r>" % (self.name, self.head)
def __repr__(self):
return "Actor(%r, %r, %r, %r, %r, %r)" % (self.name, self.num_data,
self.color, self.ix, self.max_actors, self.actor_height)
def add_value(self, value):
self.pre_head = self.head
self.data[self.pre_head] = value / self.max_actors + self.offset
self.head = (self.pre_head + 1) % self.num_data
def fill_missing(self, count):
dp = self.head
for i in range(count):
self.data[dp] = self.offset
dp = (dp + 1) % self.num_data
self.pre_head = (dp - 1) % self.num_data
self.head = dp
def render(self):
self.plotItem.setData(y=self.data, clear=True)
self.plotPoint.setData(x=[self.pre_head], y=[self.data[self.pre_head]])
class ClusterPlotItem(object):
def __init__(self, cluster, plot, pen=None, brush=None):
self.__cluster=cluster;
if(pen is None):
self.__pen=pg.mkPen(QColor("#DFDFDF"));
else:
self.__pen=pen;
if(brush is None):
self.__brush=QBrush(QColor("#DFDFDF"));
else:
self.__brush=brush;
# self.__plotData=pg.ScatterPlotItem(x=[0, 1], y=[0, 1],
# symbol="s", __pen=pg.mkPen("w"),
# size=1);
self.__plotData=FastScatterPlotItem(x=[0], y=[0],
symbol="s", pen=self.__pen,
size=1, pointMode=True);
# self.__plotData=ScatterPlotItem(x=[0], y=[0], symbol="s",
# pen=self.__pen, brush=self.__brush, size=1);
self.__plotDataBigP=ScatterPlotItem(x=[0], y=[0], symbol="s",
pen=self.__pen, brush=self.__brush, size=2);
self.__plotBoundaryData=pg.PlotDataItem(x=[0], y=[0], pen=self.__pen);
self.__plot=plot;
self.__waveStartInd=0;
self.__selPtsSequence=None;
self.__calcSelPtsSequence();
self.__selDispWaves=np.zeros(self.__cluster.getSelectArray().shape, dtype="bool");
def setPlotData(self, xChN, yChN, xChParamT, yChParamT, drawType=0):
if(drawType==0):
self.__plotData.setData(x=self.__cluster.getParam(xChN, xChParamT),
y=self.__cluster.getParam(yChN, yChParamT),
symbol="s", pen=self.__pen,
size=1, pointMode=True);
# self.__plotData.setData(x=self.__cluster.getParam(xChN, xChParamT),
# y=self.__cluster.getParam(yChN, yChParamT),
# symbol="s", pen=self.__pen, brush=self.__brush,
# size=1);
else:
self.__plotDataBigP.setData(x=self.__cluster.getParam(xChN, xChParamT),
y=self.__cluster.getParam(yChN, yChParamT),
symbol="s", pen=self.__pen, brush=self.__brush,
size=2);
boundaries=self.__cluster.getBoundaries([xChN, yChN], [xChParamT, yChParamT]);
if(len(boundaries)>0):
points=np.zeros((2, 0));
connArr=np.zeros(0, dtype="bool");
for i in boundaries:
curPoints=i.getPoints();
points=np.hstack((points, curPoints));
curCon=np.zeros(curPoints.shape[1], dtype="bool");
curCon[:-1]=True;
connArr=np.hstack((connArr, curCon));
self.__plotBoundaryData.setData(x=points[0, :], y=points[1, :],
connect=connArr);
else:
self.__plotBoundaryData.setData(x=[0], y=[0], connect="all");
def addToPlot(self, drawType=0):
if(drawType==0):
self.__plot.addItem(self.__plotData);
else:
self.__plot.addItem(self.__plotDataBigP);
self.__plot.addItem(self.__plotBoundaryData);
def removeFromPlot(self, plotType=0):
if(plotType==0):
self.__plot.removeItem(self.__plotData);
else:
self.__plot.RemoveItem(self.__plotDataBigP);
self.__plot.removeItem(self.__plotBoundaryData);
def getCurPen(self):
return self.__pen;
def __calcSelPtsSequence(self):
sBA=self.__cluster.getSelectArray();
self.__selPtsSequence=np.cumsum(sBA);
def getPrevWaves(self, indIncSize, trigChOnly):
self.__calcSelPtsSequence();
if(self.__waveStartInd<=0):
return (None, None, None, None, None);
self.__waveStartInd=self.__waveStartInd-indIncSize;
if(self.__waveStartInd+indIncSize>=self.__getNumSelPts()):
self.__waveStartInd=self.__getNumSelPts()-indIncSize;
# self.__waveStartInd=self.__waveStartInd%self.__getNumSelPts();
if(self.__waveStartInd<=0):
self.__waveStartInd=0;
sBA=self.__getSelPointsByRange(self.__waveStartInd, self.__waveStartInd+indIncSize);
self.__selDispWaves=self.__selDispWaves | sBA;
(nptsPerCh, waves, xvals, connArrs)=self.__cluster.getWaveforms(sBA, None, trigChOnly);
return (self.__cluster.getNumChannels(), nptsPerCh, waves, xvals, connArrs);
def getNextWaves(self, indIncSize, trigChOnly):
self.__calcSelPtsSequence();
if(self.__waveStartInd>=self.__getNumSelPts()):
return (None, None, None, None, None);
sBA=self.__getSelPointsByRange(self.__waveStartInd, self.__waveStartInd+indIncSize);
self.__selDispWaves=self.__selDispWaves | sBA;
self.__waveStartInd=self.__waveStartInd+indIncSize;
(nptsPerCh, waves, xvals, connArrs)=self.__cluster.getWaveforms(sBA, None, trigChOnly);
return (self.__cluster.getNumChannels(), nptsPerCh, waves, xvals, connArrs);
def resetWaveInd(self):
self.__waveStartInd=0;
def clearSelDispWaves(self):
self.__selDispWaves[:]=False;
def resetViewWaves(self):
self.__calcSelPtsSequence();
def getSelDispWaves(self):
(wAvg, wSEM, w25P, wMed, w75P)=self.__cluster.calcWaveStats();
(nptsPerCh, waves, xvals, conArrs)=self.__cluster.getWaveforms(self.__selDispWaves,
None, False)
return (waves, wAvg, wSEM, w25P, wMed, w75P);
def __getSelPointsByRange(self, startInd, endInd):
sBA=self.__cluster.getSelectArray();
return sBA & ((self.__selPtsSequence>=startInd) & (self.__selPtsSequence<endInd));
def __getNumSelPts(self):
return self.__selPtsSequence[-1];