Merge pull request #412 from NanoVNA-Saver/Development

Development
pull/446/head
Holger Müller 2021-06-29 09:08:05 +02:00 zatwierdzone przez GitHub
commit 5b1666b7e9
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
6 zmienionych plików z 140 dodań i 5 usunięć

Wyświetl plik

@ -17,7 +17,7 @@
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
VERSION = "0.3.10-pre02"
VERSION = "0.3.10-pre03"
VERSION_URL = (
"https://raw.githubusercontent.com/"
"NanoVNA-Saver/nanovna-saver/master/NanoVNASaver/About.py")

Wyświetl plik

@ -32,6 +32,7 @@ from NanoVNASaver.Hardware.NanoVNA_F_V2 import NanoVNA_F_V2
from NanoVNASaver.Hardware.NanoVNA_H import NanoVNA_H
from NanoVNASaver.Hardware.NanoVNA_H4 import NanoVNA_H4
from NanoVNASaver.Hardware.NanoVNA_V2 import NanoVNA_V2
from NanoVNASaver.Hardware.TinySA import TinySA
from NanoVNASaver.Hardware.Serial import drain_serial, Interface
@ -75,6 +76,7 @@ def get_interfaces() -> List[Interface]:
iface = Interface('serial', t.name)
iface.port = d.device
interfaces.append(iface)
logger.debug("Interfaces: %s", interfaces)
return interfaces
@ -111,6 +113,9 @@ def get_VNA(iface: Interface) -> 'VNA':
if info.find("NanoVNA") >= 0:
logger.info("Type: Generic NanoVNA")
return NanoVNA(iface)
if info.find("tinySA") >= 0:
logger.info("Type: tinySA Spectrum Analyser")
return TinySA(iface)
logger.warning("Did not recognize NanoVNA type from firmware.")
return NanoVNA(iface)
@ -159,4 +164,5 @@ def get_info(serial_port: serial.Serial) -> str:
logger.debug("Needed retries: %s", retries)
break
lines.append(line)
logger.debug("Info output: %s", lines)
return "\n".join(lines)

Wyświetl plik

@ -214,8 +214,8 @@ class NanoVNA_V2(VNA):
sleep(WRITE_SLEEP)
resp = self.serial.read(2)
if len(resp) != 2:
logger.error("Timeout reading version registers")
return None
logger.error("Timeout reading version registers. Got: %s", resp)
raise IOError("Timeout reading version registers")
result = Version(f"{resp[0]}.0.{resp[1]}")
logger.debug("readVersion: %s", result)
return result

Wyświetl plik

@ -0,0 +1,125 @@
# NanoVNASaver
#
# A python program to view and export Touchstone data from a NanoVNA
# Copyright (C) 2019, 2020 Rune B. Broberg
# Copyright (C) 2020 NanoVNA-Saver Authors
#
# 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 logging
import struct
from typing import List
import serial
import numpy as np
from PyQt5 import QtGui
from NanoVNASaver.Hardware.Serial import drain_serial, Interface
from NanoVNASaver.Hardware.VNA import VNA
from NanoVNASaver.Version import Version
logger = logging.getLogger(__name__)
class TinySA(VNA):
name = "tinySA"
screenwidth = 320
screenheight = 240
valid_datapoints = (290, )
def __init__(self, iface: Interface):
super().__init__(iface)
self.features = set('Screenshots')
logger.debug("Setting initial start,stop")
self.start, self.stop = self._get_running_frequencies()
self.sweep_max_freq_Hz = 950e6
self._sweepdata = []
def _get_running_frequencies(self):
logger.debug("Reading values: frequencies")
try:
frequencies = super().readValues("frequencies")
return frequencies[0], frequencies[-1]
except Exception as e:
logger.warning("%s reading frequencies", e)
logger.info("falling back to generic")
return VNA._get_running_frequencies(self)
def _capture_data(self) -> bytes:
timeout = self.serial.timeout
with self.serial.lock:
drain_serial(self.serial)
timeout = self.serial.timeout
self.serial.write("capture\r".encode('ascii'))
self.serial.readline()
self.serial.timeout = 4
image_data = self.serial.read(
self.screenwidth * self.screenheight * 2)
self.serial.timeout = timeout
self.serial.timeout = timeout
return image_data
def _convert_data(self, image_data: bytes) -> bytes:
rgb_data = struct.unpack(
f">{self.screenwidth * self.screenheight}H",
image_data)
rgb_array = np.array(rgb_data, dtype=np.uint32)
return (0xFF000000 +
((rgb_array & 0xF800) << 8) +
((rgb_array & 0x07E0) << 5) +
((rgb_array & 0x001F) << 3))
def getScreenshot(self) -> QtGui.QPixmap:
logger.debug("Capturing screenshot...")
if not self.connected():
return QtGui.QPixmap()
try:
rgba_array = self._convert_data(self._capture_data())
image = QtGui.QImage(
rgba_array,
self.screenwidth,
self.screenheight,
QtGui.QImage.Format_ARGB32)
logger.debug("Captured screenshot")
return QtGui.QPixmap(image)
except serial.SerialException as exc:
logger.exception(
"Exception while capturing screenshot: %s", exc)
return QtGui.QPixmap()
def resetSweep(self, start: int, stop: int):
return
def setSweep(self, start, stop):
self.start = start
self.stop = stop
list(self.exec_command(f"sweep {start} {stop} {self.datapoints}"))
list(self.exec_command("trigger auto"))
def readFrequencies(self) -> List[int]:
logger.debug("readFrequencies")
return [int(line) for line in self.exec_command("frequencies")]
def readValues(self, value) -> List[str]:
logger.debug("Read: %s", value)
if value == "data 0":
self._sweepdata = []
for line in self.exec_command("data"):
self._sweepdata.append(f"0 {line.strip()}")
return self._sweepdata
if value == "data 0":
return [x[0] for x in self._sweepdata]
if value == "data 1":
return [x[0] for x in self._sweepdata]

Wyświetl plik

@ -23,6 +23,7 @@ from collections import OrderedDict
from time import sleep, strftime, localtime
from PyQt5 import QtWidgets, QtCore, QtGui
from numpy import exp
from .Windows import (
AboutWindow, AnalysisWindow, CalibrationWindow,
@ -94,7 +95,10 @@ class NanoVNASaver(QtWidgets.QWidget):
self.bands = BandsModel()
self.interface = Interface("serial", "None")
self.vna = VNA(self.interface)
try:
self.vna = VNA(self.interface)
except IOError as exc:
self.showError(f"{exc}\n\nPlease try reconnect")
self.dataLock = threading.Lock()
self.data = Touchstone()

Wyświetl plik

@ -13,7 +13,7 @@ url = https://github.com/NanoVNA-Saver/nanovna-saver
version = attr: NanoVNASaver.About.VERSION
[options]
packages = NanoVNASaver
packages = find_namespace:
install_requires=
pyserial
PyQt5