kopia lustrzana https://github.com/NanoVNA-Saver/nanovna-saver
83 wiersze
2.8 KiB
Python
83 wiersze
2.8 KiB
Python
# NanoVNASaver - a python program to view and export Touchstone data from a NanoVNA
|
|
# Copyright (C) 2019. Rune B. Broberg
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
import collections
|
|
from typing import List
|
|
|
|
from PyQt5 import QtGui, QtWidgets, QtCore
|
|
from PyQt5.QtCore import pyqtSignal
|
|
|
|
Datapoint = collections.namedtuple('Datapoint', 'freq re im')
|
|
|
|
|
|
class Marker(QtCore.QObject):
|
|
name = "Marker"
|
|
frequency = 0
|
|
color = QtGui.QColor()
|
|
location = -1
|
|
|
|
updated = pyqtSignal()
|
|
|
|
def __init__(self, name, initialColor, frequency=""):
|
|
super().__init__()
|
|
self.name = name
|
|
|
|
if frequency.isnumeric():
|
|
self.frequency = int(frequency)
|
|
self.frequencyInput = QtWidgets.QLineEdit(frequency)
|
|
self.frequencyInput.setAlignment(QtCore.Qt.AlignRight)
|
|
self.frequencyInput.returnPressed.connect(lambda: self.setFrequency(self.frequencyInput.text()))
|
|
|
|
self.btnColorPicker = QtWidgets.QPushButton("█")
|
|
self.btnColorPicker.setFixedWidth(20)
|
|
self.setColor(initialColor)
|
|
self.btnColorPicker.clicked.connect(lambda: self.setColor(QtWidgets.QColorDialog.getColor()))
|
|
|
|
self.layout = QtWidgets.QHBoxLayout()
|
|
self.layout.addWidget(self.frequencyInput)
|
|
self.layout.addWidget(self.btnColorPicker)
|
|
|
|
def setFrequency(self, frequency):
|
|
if frequency.isnumeric():
|
|
self.frequency = int(frequency)
|
|
self.updated.emit()
|
|
else:
|
|
self.frequency = 0
|
|
return
|
|
|
|
def setColor(self, color):
|
|
self.color = color
|
|
p = self.btnColorPicker.palette()
|
|
p.setColor(QtGui.QPalette.ButtonText, self.color)
|
|
self.btnColorPicker.setPalette(p)
|
|
|
|
def getRow(self):
|
|
return (QtWidgets.QLabel(self.name), self.layout)
|
|
|
|
def findLocation(self, data: List[Datapoint]):
|
|
self.location = -1
|
|
if self.frequency == 0:
|
|
# No frequency set for this marker
|
|
return
|
|
if len(data) == 0:
|
|
# Set the frequency before loading any data
|
|
return
|
|
|
|
stepsize = data[1].freq-data[0].freq
|
|
for i in range(len(data)):
|
|
if abs(data[i].freq-self.frequency) <= (stepsize/2):
|
|
self.location = i
|
|
return
|