class HistPlot(QtWidgets.QWidget): def __init__(self, parent): super(HistPlot, self).__init__() self.verticalScaleDivision = ScaleDivision(-140, 0, 100) self.verticalScaleTransform = CoordinateTransform(-140, 0, 100, 0, 0) self.verticalScale = VerticalScaleWidget(self, self.verticalScaleDivision, self.verticalScaleTransform) self.verticalScale.setTitle("PSD (dB A)") self.horizontalScaleDivision = ScaleDivision(44, 22000, 100) self.horizontalScaleTransform = CoordinateTransform(44, 22000, 100, 0, 0) self.horizontalScale = HorizontalScaleWidget(self, self.horizontalScaleDivision, self.horizontalScaleTransform) self.horizontalScale.setTitle("Frequency (Hz)") self.canvasWidget = CanvasWidget(self, self.verticalScaleTransform, self.horizontalScaleTransform) self.canvasWidget.setTrackerFormatter(lambda x, y: "%d Hz, %.1f dB" % (x, y)) plot_layout = QtWidgets.QGridLayout() plot_layout.setSpacing(0) plot_layout.setContentsMargins(0, 0, 0, 0) plot_layout.addWidget(self.verticalScale, 0, 0) plot_layout.addWidget(self.canvasWidget, 0, 1) plot_layout.addWidget(self.horizontalScale, 1, 1) self.setLayout(plot_layout) self.needfullreplot = False self.horizontalScaleTransform.setScale(fscales.Logarithmic) self.horizontalScaleDivision.setScale(fscales.Logarithmic) # insert an additional plot item for the peak bar self.bar_peak = HistogramPeakBarItem() self.canvasWidget.attach(self.bar_peak) self.peak = zeros((1,)) self.peak_int = 0 self.peak_decay = PEAK_DECAY_RATE self.histogram = HistogramItem() self.histogram.set_color(Qt.Qt.darkGreen) self.canvasWidget.attach(self.histogram) # need to replot here for the size Hints to be computed correctly (depending on axis scales...) self.update() def setdata(self, fl, fh, fc, y): self.histogram.setData(fl, fh, fc, y) self.compute_peaks(y) self.bar_peak.setData(fl, fh, self.peak, self.peak_int, y) # only draw on demand # self.draw() def draw(self): if self.needfullreplot: self.needfullreplot = False self.verticalScaleDivision.setLength(self.canvasWidget.height()) self.verticalScaleTransform.setLength(self.canvasWidget.height()) start_border, end_border = self.verticalScale.spacingBorders() self.verticalScaleTransform.setBorders(start_border, end_border) self.verticalScale.update() self.horizontalScaleDivision.setLength(self.canvasWidget.width()) self.horizontalScaleTransform.setLength(self.canvasWidget.width()) start_border, end_border = self.horizontalScale.spacingBorders() self.horizontalScaleTransform.setBorders(start_border, end_border) self.horizontalScale.update() x_major_tick = self.horizontalScaleDivision.majorTicks() x_minor_tick = self.horizontalScaleDivision.minorTicks() y_major_tick = self.verticalScaleDivision.majorTicks() y_minor_tick = self.verticalScaleDivision.minorTicks() self.canvasWidget.setGrid(array(x_major_tick), array(x_minor_tick), array(y_major_tick), array(y_minor_tick)) self.canvasWidget.update() # redraw when the widget is resized to update coordinates transformations def resizeEvent(self, event): self.needfullreplot = True self.draw() def compute_peaks(self, y): if len(self.peak) != len(y): y_ones = ones(y.shape) self.peak = y_ones * (-500.) self.peak_int = zeros(y.shape) self.peak_decay = y_ones * 20. * log10(PEAK_DECAY_RATE) * 5000 mask1 = (self.peak < y) mask2 = (~mask1) mask2_a = mask2 * (self.peak_int < 0.2) mask2_b = mask2 * (self.peak_int >= 0.2) self.peak[mask1] = y[mask1] self.peak[mask2_a] = self.peak[mask2_a] + self.peak_decay[mask2_a] self.peak_decay[mask1] = 20. * log10(PEAK_DECAY_RATE) * 5000 self.peak_decay[mask2_a] += 20. * log10(PEAK_DECAY_RATE) * 5000 self.peak_int[mask1] = 1. self.peak_int[mask2_b] *= 0.975 def setspecrange(self, spec_min, spec_max): self.verticalScaleTransform.setRange(spec_min, spec_max) self.verticalScaleDivision.setRange(spec_min, spec_max) # notify that sizeHint has changed (this should be done with a signal emitted from the scale division to the scale bar) self.verticalScale.scaleBar.updateGeometry() self.needfullreplot = True self.update() def setweighting(self, weighting): if weighting is 0: title = "PSD (dB)" elif weighting is 1: title = "PSD (dB A)" elif weighting is 2: title = "PSD (dB B)" else: title = "PSD (dB C)" self.verticalScale.setTitle(title)
class ImagePlot(QtWidgets.QWidget): def __init__(self, parent): super(ImagePlot, self).__init__(parent) self.verticalScaleDivision = ScaleDivision(20, 20000, 100) self.verticalScaleTransform = CoordinateTransform(20, 20000, 100, 0, 0) self.verticalScale = VerticalScaleWidget(self, self.verticalScaleDivision, self.verticalScaleTransform) self.verticalScale.setTitle("Frequency (Hz)") self.verticalScale.scaleBar.setTickFormatter(tickFormatter) self.horizontalScaleDivision = ScaleDivision(0, 10, 100) self.horizontalScaleTransform = CoordinateTransform(0, 10, 100, 0, 0) self.horizontalScale = HorizontalScaleWidget( self, self.horizontalScaleDivision, self.horizontalScaleTransform) self.horizontalScale.setTitle("Time (s)") self.colorScaleDivision = ScaleDivision(-140, 0, 100) self.colorScaleTransform = CoordinateTransform(-140, 0, 100, 0, 0) self.colorScale = ColorScaleWidget(self, self.colorScaleDivision, self.colorScaleTransform) self.colorScale.setTitle("PSD (dB A)") self.canvasWidget = CanvasWidget(self, self.verticalScaleTransform, self.horizontalScaleTransform) self.canvasWidget.setTrackerFormatter(lambda x, y: "%.2f s, %d Hz" % (x, y)) plotLayout = QtWidgets.QGridLayout() plotLayout.setSpacing(0) plotLayout.setContentsMargins(0, 0, 0, 0) plotLayout.addWidget(self.verticalScale, 0, 0) plotLayout.addWidget(self.canvasWidget, 0, 1) plotLayout.addWidget(self.colorScale, 0, 2) plotLayout.addWidget(self.horizontalScale, 1, 1) self.setLayout(plotLayout) self.needfullreplot = False # attach a plot image self.plotImage = PlotImage() self.canvasWidget.attach(self.plotImage) self.setfreqscale(fscales.Linear) self.setspecrange(-140., 0.) # need to replot here for the size Hints to be computed correctly (depending on axis scales...) self.update() def addData(self, freq, xyzs, last_data_time): self.plotImage.addData(freq, xyzs, self.freqscale, last_data_time) def draw(self): if self.needfullreplot: self.needfullreplot = False self.verticalScaleDivision.setLength(self.canvasWidget.height()) self.verticalScaleTransform.setLength(self.canvasWidget.height()) startBorder, endBorder = self.verticalScale.spacingBorders() self.verticalScaleTransform.setBorders(startBorder, endBorder) self.verticalScale.update() self.horizontalScaleDivision.setLength(self.canvasWidget.width()) self.horizontalScaleTransform.setLength(self.canvasWidget.width()) startBorder, endBorder = self.horizontalScale.spacingBorders() self.horizontalScaleTransform.setBorders(startBorder, endBorder) self.horizontalScale.update() self.colorScaleDivision.setLength(self.canvasWidget.height()) self.colorScaleTransform.setLength(self.canvasWidget.height()) startBorder, endBorder = self.colorScale.spacingBorders() self.colorScaleTransform.setBorders(startBorder, endBorder) self.colorScale.update() self.canvasWidget.update() # redraw when the widget is resized to update coordinates transformations def resizeEvent(self, event): self.needfullreplot = True self.draw() def pause(self): self.plotImage.pause() def restart(self): self.plotImage.restart() def setfreqscale(self, scale): self.freqscale = scale self.plotImage.erase() self.plotImage.setfreqscale(scale) self.verticalScaleTransform.setScale(scale) self.verticalScaleDivision.setScale(scale) # notify that sizeHint has changed (this should be done with a signal emitted from the scale division to the scale bar) self.verticalScale.scaleBar.updateGeometry() self.needfullreplot = True self.update() def settimerange(self, timerange_seconds, dT_seconds): self.plotImage.settimerange(timerange_seconds, dT_seconds) self.horizontalScaleTransform.setRange(0, timerange_seconds) self.horizontalScaleDivision.setRange(0, timerange_seconds) # notify that sizeHint has changed (this should be done with a signal emitted from the scale division to the scale bar) self.horizontalScale.scaleBar.updateGeometry() self.needfullreplot = True self.update() def set_sfft_rate(self, rate_frac): self.plotImage.set_sfft_rate(rate_frac) def setfreqrange(self, minfreq, maxfreq): self.plotImage.setfreqrange(minfreq, maxfreq) self.verticalScaleTransform.setRange(minfreq, maxfreq) self.verticalScaleDivision.setRange(minfreq, maxfreq) # notify that sizeHint has changed (this should be done with a signal emitted from the scale division to the scale bar) self.verticalScale.scaleBar.updateGeometry() self.needfullreplot = True self.update() def setspecrange(self, spec_min, spec_max): self.colorScaleTransform.setRange(spec_min, spec_max) self.colorScaleDivision.setRange(spec_min, spec_max) # notify that sizeHint has changed (this should be done with a signal emitted from the scale division to the scale bar) self.colorScale.scaleBar.updateGeometry() self.needfullreplot = True self.update() def setweighting(self, weighting): if weighting is 0: title = "PSD (dB)" elif weighting is 1: title = "PSD (dB A)" elif weighting is 2: title = "PSD (dB B)" else: title = "PSD (dB C)" self.colorScale.setTitle(title)
class SpectrumPlotWidget(QtWidgets.QWidget): def __init__(self, parent): super(SpectrumPlotWidget, self).__init__(parent) self.x1 = array([0.1, 0.5, 1.]) self.x2 = array([0.5, 1., 2.]) self.needtransform = False self.paused = False self.peaks_enabled = True self.peak = zeros((3, )) self.peak_int = zeros((3, )) self.peak_decay = ones((3, )) * PEAK_DECAY_RATE self.verticalScaleDivision = ScaleDivision(0, 1, 100) self.verticalScaleTransform = CoordinateTransform(0, 1, 100, 0, 0) self.verticalScale = VerticalScaleWidget(self, self.verticalScaleDivision, self.verticalScaleTransform) self.verticalScale.setTitle("PSD (dB)") self.horizontalScaleDivision = ScaleDivision(0, 22000, 100) self.horizontalScaleTransform = CoordinateTransform( 0, 22000, 100, 0, 0) self.horizontalScale = HorizontalScaleWidget( self, self.horizontalScaleDivision, self.horizontalScaleTransform) self.horizontalScale.setTitle("Frequency (Hz)") self.canvasWidget = GlCanvasWidget(self, self.verticalScaleTransform, self.horizontalScaleTransform) self.canvasWidget.setTrackerFormatter(lambda x, y: "%d Hz, %.1f dB" % (x, y)) self.canvasWidget.resized.connect(self.canvasResized) def r_peak(p): return 1. + 0. * p def g_peak(p): return 1. - p def b_peak(p): return 1. - p self.peakQuadsItem = QuadsItem(r_peak, g_peak, b_peak) self.canvasWidget.attach(self.peakQuadsItem) def r_signal(p): return 0. * p def g_signal(p): return 0.3 + 0.5 * p def b_signal(p): return 0. * p self.quadsItem = QuadsItem(r_signal, g_signal, b_signal) self.canvasWidget.attach(self.quadsItem) plotLayout = QtWidgets.QGridLayout() plotLayout.setSpacing(0) plotLayout.setContentsMargins(0, 0, 0, 0) plotLayout.addWidget(self.verticalScale, 0, 0) plotLayout.addWidget(self.canvasWidget, 0, 1) plotLayout.addWidget(self.horizontalScale, 1, 1) self.setLayout(plotLayout) def setfreqscale(self, scale): self.horizontalScaleTransform.setScale(scale) self.horizontalScaleDivision.setScale(scale) self.needtransform = True self.draw() def setfreqrange(self, minfreq, maxfreq): self.xmin = minfreq self.xmax = maxfreq self.horizontalScaleTransform.setRange(minfreq, maxfreq) self.horizontalScaleDivision.setRange(minfreq, maxfreq) # notify that sizeHint has changed (this should be done with a signal emitted from the scale division to the scale bar) self.horizontalScale.scaleBar.updateGeometry() self.needtransform = True self.draw() def setspecrange(self, spec_min, spec_max): if spec_min > spec_max: spec_min, spec_max = spec_max, spec_min self.verticalScaleTransform.setRange(spec_min, spec_max) self.verticalScaleDivision.setRange(spec_min, spec_max) # notify that sizeHint has changed (this should be done with a signal emitted from the scale division to the scale bar) self.verticalScale.scaleBar.updateGeometry() self.needtransform = True self.draw() def setweighting(self, weighting): if weighting is 0: title = "PSD (dB)" elif weighting is 1: title = "PSD (dB A)" elif weighting is 2: title = "PSD (dB B)" else: title = "PSD (dB C)" self.verticalScale.setTitle(title) self.needtransform = True self.draw() def setShowFreqLabel(self, showFreqLabel): self.canvasWidget.setShowFreqLabel(showFreqLabel) def set_peaks_enabled(self, enabled): self.peaks_enabled = enabled self.canvasWidget.detachAll() if enabled: self.canvasWidget.attach(self.peakQuadsItem) self.canvasWidget.attach(self.quadsItem) def set_baseline_displayUnits(self, baseline): self.quadsItem.set_baseline_displayUnits(baseline) def set_baseline_dataUnits(self, baseline): self.quadsItem.set_baseline_dataUnits(baseline) def setdata(self, x, y, fmax): x1 = zeros(x.shape) x2 = zeros(x.shape) x1[0] = 1e-10 x1[1:] = (x[1:] + x[:-1]) / 2. x2[:-1] = x1[1:] x2[-1] = float(SAMPLING_RATE / 2) if len(x1) != len(self.x1): self.needtransform = True self.x1 = x1 self.x2 = x2 if not self.paused: self.canvasWidget.setfmax(fmax) M = np.max(y) m = self.verticalScaleTransform.coord_min y_int = (y - m) / (np.abs(M - m) + 1e-3) self.quadsItem.setData(x1, x2, y, y_int) if self.peaks_enabled: self.compute_peaks(y) self.peakQuadsItem.setData(x1, x2, self.peak, self.peak_int) def draw(self): if self.needtransform: self.verticalScaleDivision.setLength(self.canvasWidget.height()) self.verticalScaleTransform.setLength(self.canvasWidget.height()) startBorder, endBorder = self.verticalScale.spacingBorders() self.verticalScaleTransform.setBorders(startBorder, endBorder) self.verticalScale.update() self.horizontalScaleDivision.setLength(self.canvasWidget.width()) self.horizontalScaleTransform.setLength(self.canvasWidget.width()) startBorder, endBorder = self.horizontalScale.spacingBorders() self.horizontalScaleTransform.setBorders(startBorder, endBorder) self.horizontalScale.update() self.canvasWidget.setGrid( array(self.horizontalScaleDivision.majorTicks()), array(self.horizontalScaleDivision.minorTicks()), array(self.verticalScaleDivision.majorTicks()), array(self.verticalScaleDivision.minorTicks())) self.quadsItem.transformUpdate() self.peakQuadsItem.transformUpdate() self.needtransform = False def pause(self): self.paused = True self.canvasWidget.pause() def restart(self): self.paused = False self.canvasWidget.restart() # redraw when the widget is resized to update coordinates transformations # QOpenGlWidget does not like that we override resizeEvent def canvasResized(self, canvasWidth, canvasHeight): self.needtransform = True self.draw() def canvasUpdate(self): self.canvasWidget.update() def compute_peaks(self, y): if len(self.peak) != len(y): y_ones = ones(y.shape) self.peak = y_ones * (-500.) self.peak_int = zeros(y.shape) self.peak_decay = y_ones * 20. * log10(PEAK_DECAY_RATE) * 5000 mask1 = (self.peak < y) mask2 = (~mask1) mask2_a = mask2 * (self.peak_int < 0.2) mask2_b = mask2 * (self.peak_int >= 0.2) self.peak[mask1] = y[mask1] self.peak[mask2_a] = self.peak[mask2_a] + self.peak_decay[mask2_a] self.peak_decay[mask1] = 20. * log10(PEAK_DECAY_RATE) * 5000 self.peak_decay[mask2_a] += 20. * log10(PEAK_DECAY_RATE) * 5000 self.peak_int[mask1] = 1. self.peak_int[mask2_b] *= 0.975