nanovna-saver/src/NanoVNASaver/Charts/SParam.py

#  NanoVNASaver
#
#  A python program to view and export Touchstone data from a NanoVNA
#  Copyright (C) 2019, 2020  Rune B. Broberg
#  Copyright (C) 2020,2021 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
from typing import List

from PyQt5 import QtGui

from NanoVNASaver.RFTools import Datapoint
from NanoVNASaver.Charts.Chart import Chart
from NanoVNASaver.Charts.Frequency import FrequencyChart
from NanoVNASaver.Charts.LogMag import LogMagChart

logger = logging.getLogger(__name__)


class SParameterChart(FrequencyChart):
    def __init__(self, name=""):
        super().__init__(name)

        self.minDisplayValue = -1
        self.maxDisplayValue = 1
        self.fixedValues = True

        self.y_action_automatic.setChecked(False)
        self.y_action_fixed_span.setChecked(True)

        self.minValue = 0
        self.maxValue = 1
        self.span = 1

        self.isInverted = False

    def drawChart(self, qp: QtGui.QPainter):
        qp.setPen(QtGui.QPen(Chart.color.text))
        qp.drawText(self.dim.width // 2 - 20, 15, f"{self.name}")
        qp.drawText(10, 15, "Real")
        qp.drawText(self.leftMargin + self.dim.width - 15, 15, "Imag")
        qp.setPen(QtGui.QPen(Chart.color.foreground))
        qp.drawLine(self.leftMargin,
                    self.topMargin - 5,
                    self.leftMargin,
                    self.topMargin + self.dim.height + 5)
        qp.drawLine(self.leftMargin - 5,
                    self.topMargin + self.dim.height,
                    self.leftMargin + self.dim.width,
                    self.topMargin + self.dim.height)

    def drawValues(self, qp: QtGui.QPainter):
        if len(self.data) == 0 and len(self.reference) == 0:
            return
        self._set_start_stop()
        if self.bands.enabled:
            self.drawBands(qp, self.fstart, self.fstop)
        if self.fixedValues:
            maxValue = self.maxDisplayValue
            minValue = self.minDisplayValue
        else:
            minValue = -1
            maxValue = 1
        self.minValue = minValue
        self.maxValue = maxValue
        span = maxValue - minValue
        if span == 0:
            span = 0.01
        self.span = span
        tick_count = self.dim.height // 60
        tick_step = self.span / tick_count
        for i in range(tick_count):
            val = int(minValue + i * tick_step)
            y = self.topMargin + (maxValue - val) // span * self.dim.height
            qp.setPen(QtGui.QPen(Chart.color.foreground))
            qp.drawLine(self.leftMargin - 5, y,
                        self.leftMargin + self.dim.width, y)
            if val > minValue and val != maxValue:
                qp.setPen(QtGui.QPen(Chart.color.text))
                qp.drawText(3, y + 4, str(round(val, 2)))
        qp.setPen(QtGui.QPen(Chart.color.foreground))
        qp.drawLine(self.leftMargin - 5, self.topMargin,
                    self.leftMargin + self.dim.width, self.topMargin)

        qp.setPen(Chart.color.text)
        qp.drawText(3, self.topMargin + 4, f"{maxValue}")
        qp.drawText(3, self.dim.height + self.topMargin, f"{minValue}")
        self.drawFrequencyTicks(qp)
        self.drawData(qp, self.data, Chart.color.sweep, self.getReYPosition)
        self.drawData(qp, self.reference, Chart.color.reference,
                      self.getReYPosition)
        self.drawData(qp, self.data, Chart.color.sweep_secondary,
                      self.getImYPosition)
        self.drawData(qp, self.reference,
                      Chart.color.reference_secondary, self.getImYPosition)

        self.drawMarkers(qp, y_function=self.getReYPosition)
        self.drawMarkers(qp, y_function=self.getImYPosition)

    def getYPosition(self, d: Datapoint) -> int:
        return int(
            self.topMargin + (self.maxValue - d.re) / self.span *
            self.dim.height)

    def getReYPosition(self, d: Datapoint) -> int:
        return int(
            self.topMargin + (self.maxValue - d.re) / self.span *
            self.dim.height)

    def getImYPosition(self, d: Datapoint) -> int:
        return int(
            self.topMargin + (self.maxValue - d.im) / self.span *
            self.dim.height)

    def valueAtPosition(self, y) -> List[float]:
        absy = y - self.topMargin
        val = -1 * ((absy / self.dim.height * self.span) - self.maxValue)
        return [val]

    def logMag(self, p: Datapoint) -> float:
        return -p.gain if self.isInverted else p.gain

    def copy(self):
        new_chart: LogMagChart = super().copy()
        new_chart.isInverted = self.isInverted
        new_chart.span = self.span
        return new_chart