def addPoint(self, text, color, border=None, size=20.0): shape = QGraphicsEllipseItem(size / 2.0, size / 2.0, size, size) brush = makeBrush(color) shape.setBrush(brush) shape.setPen(makePen(border)) self.addEntry(MapLegendEntryItem(shape, text))
def addPoint(self, text, color, border=None, size=20.0): shape = QGraphicsEllipseItem(size / 2.0, size / 2.0, size, size) brush = makeBrush(color) shape.setBrush(brush) shape.setPen(makePen(border)) self.addEntry(MapLegendEntryItem(shape, text))
def updateTime(self, frame): if frame == self.previousframe: return self.previousframe = frame idx = self.currentPuff.currentIndex() currentPuff = self.group.puffs[idx] image1 = np.copy(self.I[frame]) mmax = np.max(self.I) image1 = image1 / mmax image2 = np.copy(self.I_fits[idx][frame]) image2 = image2 / mmax self.puff_3D.update_images(image1, image2) rel_frame = frame - (currentPuff.kinetics['t_start'] - currentPuff.kinetics['before']) for line in [self.vLine1, self.vLine2, self.vLine5]: # self.vLine3,self.vLine4,self.vLine5]: line.setValue(rel_frame) for bar in self.error_bars: bar.plot.removeItem(bar) self.error_bars = [] for item in self.addedItems: self.imageview.view.removeItem(item) self.addedItems = [] scatter = pg.ScatterPlotItem(size=8, pen=pg.mkPen(None)) self.imageview.view.addItem(scatter) self.addedItems.append(scatter) spots = [] for i in np.arange(len(self.group.puffs)): rel_frame2 = frame - ((currentPuff.kinetics['t_start'] - currentPuff.kinetics['before']) - (self.group.puffs[i].kinetics['t_start'] - self.group.puffs[i].kinetics['before'])) if self.puffCheckBoxes[i].isChecked( ) and rel_frame2 >= 0 and rel_frame2 < self.params[i].shape[0]: centers = self.params[i][:, :2] - self.bounds[1:, 0] center = centers[rel_frame2, :] amps = np.copy(self.params[i][:, -1]) amps[amps <= 0] = .00000001 amp = amps[rel_frame2] std = self.group.puffs[0].clusters.standard_deviations[i] color = np.array(self.colors[i]) color[3] = 50 ### FIRST ADD THE POINT TO THE SCATTER PLOT if 0.8 / amp < std: # if the amplitude>0 and the error (0.8/SNR) is within the fitting bounds ### FIRST PLOT THE """ WARNING!!! the amp should really be the signal to noise ratio, so it is only valid when the image has been ratiod by the standard deviation of the noise """ sigma = 0.8 / amp sigmas = 0.8 / amps # add error bars to x and y # first add X color[3] = 255 err = pg.ErrorBarItem( x=np.array([rel_frame]), y=np.array([self.params[i][rel_frame2, 0]]), top=np.array([sigma]), bottom=np.array([sigma]), beam=0.5, pen=pg.mkPen(pg.mkColor(tuple(color)))) self.p3.addItem(err) err.plot = self.p3 self.error_bars.append(err) # second add Y err = pg.ErrorBarItem( x=np.array([rel_frame]), y=np.array([self.params[i][rel_frame2, 1]]), top=np.array([sigma]), bottom=np.array([sigma]), beam=0.5, pen=pg.mkPen(pg.mkColor(tuple(color)))) self.p4.addItem(err) err.plot = self.p4 self.error_bars.append(err) reasonable_fit = sigmas < 2 * std bounds = QRectF( QPointF(center[0] - sigma, center[1] - sigma), QSizeF(2 * sigma, 2 * sigma)) ### plot outer circle pathitem = QGraphicsEllipseItem(bounds) color[-1] = 127 # alpha channel pathitem.setPen(pg.mkPen(pg.mkColor(tuple(color)))) pathitem.setBrush(pg.mkColor((0, 0, 0, 0))) self.imageview.view.addItem(pathitem) self.addedItems.append(pathitem) ### plot line frame_i = rel_frame2 - 6 if frame_i < 0: frame_i = 0 alpha = 255 for ii in np.arange(rel_frame2, frame_i, -1) - 1: if alpha <= 0 or not reasonable_fit[ii]: break color[-1] = alpha # alpha channel alpha = alpha - (255 * 1. / 6.) pathitem = QGraphicsPathItem() pathitem.setPen(pg.mkColor(tuple(color))) path = QPainterPath(QPointF(*centers[ii])) path.lineTo(QPointF(*centers[ii + 1])) pathitem.setPath(path) self.imageview.view.addItem(pathitem) self.addedItems.append(pathitem) color[3] = 255 # make the spot transparent spots.append({ 'pos': center, 'brush': pg.mkBrush(pg.mkColor(tuple(color))) }) scatter.addPoints(spots)