def draw_frequency_marker(self, x_pos, frequency):
if frequency is None:
self.clear_frequency_marker()
return
y1 = self.sceneRect().y()
y2 = self.sceneRect().y() + self.sceneRect().height()
if self.frequency_marker is None:
pen = QPen(constants.LINECOLOR, 0)
self.frequency_marker = [None, None]
self.frequency_marker[0] = self.addLine(x_pos, y1, x_pos, y2, pen)
self.frequency_marker[1] = self.addSimpleText("")
self.frequency_marker[1].setBrush(QBrush(constants.LINECOLOR))
font = QFont()
font.setBold(True)
font.setPointSize(int(font.pointSize() * 1.25)+1)
self.frequency_marker[1].setFont(font)
self.frequency_marker[0].setLine(x_pos, y1, x_pos, y2)
scale_x, scale_y = util.calc_x_y_scale(self.sceneRect(), self.parent())
self.frequency_marker[1].setTransform(QTransform.fromScale(scale_x, scale_y), False)
self.frequency_marker[1].setText("Tune to " + Formatter.big_value_with_suffix(frequency, decimals=3))
font_metric = QFontMetrics(self.frequency_marker[1].font())
text_width = font_metric.width("Tune to") * scale_x
text_width += (font_metric.width(" ") * scale_x) / 2
self.frequency_marker[1].setPos(x_pos-text_width, 0.95*y1)
def redraw_legend(self, force_show=False):
if not (force_show or self.always_show_symbols_legend):
self.hide_legend()
return
num_captions = len(self.centers) + 1
if num_captions != len(self.captions):
self.clear_legend()
fmt = "{0:0" + str(self.bits_per_symbol) + "b}"
for i in range(num_captions):
font = QFont()
font.setPointSize(16)
font.setBold(True)
self.captions.append(self.addSimpleText(fmt.format(i), font))
view_rect = self.parent().view_rect() # type: QRectF
padding = 0
fm = QFontMetrics(self.captions[0].font())
for i, caption in enumerate(self.captions):
caption.show()
scale_x, scale_y = util.calc_x_y_scale(self.separation_areas[i].rect(), self.parent())
try:
caption.setPos(view_rect.x() + view_rect.width() - fm.width(caption.text()) * scale_x,
self.centers[i] + padding)
except IndexError:
caption.setPos(view_rect.x() + view_rect.width() - fm.width(caption.text()) * scale_x,
self.centers[i - 1] - padding - fm.height() * scale_y)
caption.setTransform(QTransform.fromScale(scale_x, scale_y), False)
def drawBackground(self, painter: QPainter, rect: QRectF):
# freqs = np.fft.fftfreq(len(w), 1 / self.sample_rate)
if self.draw_grid and len(self.frequencies) > 0:
painter.setPen(QPen(painter.pen().color(), 0))
parent_width = self.parent().width() if hasattr(
self.parent(), "width") else 750
view_rect = self.parent().view_rect() if hasattr(
self.parent(), "view_rect") else rect
font_width = self.font_metrics.width(
Formatter.big_value_with_suffix(self.center_freq) + " ")
x_grid_size = int(view_rect.width() / parent_width * font_width)
# x_grid_size = int(0.1 * view_rect.width()) if 0.1 * view_rect.width() > 1 else 1
y_grid_size = 1
x_mid = np.where(self.frequencies == 0)[0]
x_mid = int(x_mid[0]) if len(x_mid) > 0 else 0
left = int(rect.left()) - (int(rect.left()) % x_grid_size)
left = left if left > 0 else 0
top = rect.top() - (rect.top() % y_grid_size)
bottom = rect.bottom() - (rect.bottom() % y_grid_size)
right_border = int(
rect.right()) if rect.right() < len(self.frequencies) else len(
self.frequencies)
x_range = list(range(x_mid, left, -x_grid_size)) + list(
range(x_mid, right_border, x_grid_size))
lines = [QLineF(x, rect.top(), x, bottom) for x in x_range] \
+ [QLineF(rect.left(), y, rect.right(), y) for y in np.arange(top, bottom, y_grid_size)]
painter.drawLines(lines)
scale_x, scale_y = util.calc_x_y_scale(rect, self.parent())
painter.scale(scale_x, scale_y)
counter = -1 # Counter for Label for every second line
for x in x_range:
freq = self.frequencies[x]
counter += 1
if freq != 0 and (counter % 2 !=
0): # Label for every second line
continue
if freq != 0:
prefix = "+" if freq > 0 else ""
value = prefix + Formatter.big_value_with_suffix(freq, 2)
else:
counter = 0
value = Formatter.big_value_with_suffix(6800e6 -
self.center_freq +
self.status_k)
font_width = self.font_metrics.width(value)
painter.drawText(x / scale_x - font_width / 2,
bottom / scale_y, value)
def redraw_legend(self, force_show=False):
if not (force_show or self.always_show_symbols_legend):
if self.zeros_caption is not None:
self.zeros_caption.hide()
if self.ones_caption is not None:
self.ones_caption.hide()
return
if self.ones_caption is None:
font = QFont()
font.setPointSize(32)
font.setBold(True)
self.ones_caption = self.addSimpleText("1", font)
if self.zeros_caption is None:
font = QFont()
font.setPointSize(32)
font.setBold(True)
self.zeros_caption = self.addSimpleText("0", font)
view_rect = self.parent().view_rect() # type: QRectF
self.ones_caption.show()
self.zeros_caption.show()
padding = view_rect.height() / 20
scale_x, scale_y = util.calc_x_y_scale(self.ones_area.rect(),
self.parent())
y_mid = self.y_mid
fm = QFontMetrics(self.ones_caption.font())
self.ones_caption.setPos(
view_rect.x() + view_rect.width() - fm.width("1") * scale_x,
y_mid - fm.height() * scale_y - padding)
self.ones_caption.setTransform(QTransform.fromScale(scale_x, scale_y),
False)
scale_x, scale_y = util.calc_x_y_scale(self.zeros_area.rect(),
self.parent())
self.zeros_caption.setPos(
view_rect.x() + view_rect.width() - fm.width("0") * scale_x,
y_mid + padding)
self.zeros_caption.setTransform(QTransform.fromScale(scale_x, scale_y),
False)