Source code for tuiview.plotscalingdialog

Module that contains the PlotScalingDialog class
# This file is part of 'TuiView' - a simple Raster viewer
# Copyright (C) 2012  Sam Gillingham
# 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 2
# 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
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

from PyQt5.QtWidgets import QDialog, QFormLayout, QHBoxLayout, QCheckBox
from PyQt5.QtWidgets import QSpinBox, QDoubleSpinBox, QPushButton, QVBoxLayout
from PyQt5.QtCore import Qt
import numpy

[docs]class PlotScalingDialog(QDialog): """ Dialog that allows the user to set the scale for the X axis """ def __init__(self, parent, currentScale, data): QDialog.__init__(self, parent) minScale, maxScale = currentScale self.setWindowTitle("Plot Scaling") self.formLayout = QFormLayout() self.minAutoCheck = QCheckBox() self.minAutoCheck.setText("Auto") floatingPoint = numpy.issubdtype(data.dtype, numpy.floating) if floatingPoint: info = numpy.finfo(data.dtype) else: info = numpy.iinfo(data.dtype) if floatingPoint: self.minValueSpin = QDoubleSpinBox() self.minValueSpin.setRange(info.min, info.max) else: self.minValueSpin = QSpinBox() self.minValueSpin.setRange(info.min, info.max) self.minAutoCheck.stateChanged.connect(self.onMinAuto) if minScale is None: self.minAutoCheck.setCheckState(Qt.Checked) self.minValueSpin.setEnabled(False) self.minValueSpin.setValue(data.min()) else: self.minValueSpin.setValue(minScale) self.minLayout = QVBoxLayout() self.minLayout.addWidget(self.minAutoCheck) self.minLayout.addWidget(self.minValueSpin) self.formLayout.addRow("Minimum Value:", self.minLayout) self.maxAutoCheck = QCheckBox() self.maxAutoCheck.setText("Auto") if floatingPoint: self.maxValueSpin = QDoubleSpinBox() self.maxValueSpin.setRange(info.min, info.max) else: self.maxValueSpin = QSpinBox() self.maxValueSpin.setRange(info.min, info.max) self.maxAutoCheck.stateChanged.connect(self.onMaxAuto) if maxScale is None: self.maxAutoCheck.setCheckState(Qt.Checked) self.maxValueSpin.setEnabled(False) self.maxValueSpin.setValue(data.max()) else: self.maxValueSpin.setValue(maxScale) self.maxLayout = QVBoxLayout() self.maxLayout.addWidget(self.maxAutoCheck) self.maxLayout.addWidget(self.maxValueSpin) self.formLayout.addRow("Maximum Value:", self.maxLayout) self.okButton = QPushButton() self.okButton.setText("OK") self.okButton.clicked.connect(self.accept) self.cancelButton = QPushButton() self.cancelButton.setText("Cancel") self.cancelButton.clicked.connect(self.reject) self.buttonLayout = QHBoxLayout() self.buttonLayout.addWidget(self.okButton) self.buttonLayout.addWidget(self.cancelButton) self.mainLayout = QVBoxLayout(self) self.mainLayout.addLayout(self.formLayout) self.mainLayout.addLayout(self.buttonLayout) self.setLayout(self.mainLayout)
[docs] def onMaxAuto(self, state): "Called when auto state changed" self.maxValueSpin.setEnabled(state != Qt.Checked)
[docs] def onMinAuto(self, state): "Called when auto state changed" self.minValueSpin.setEnabled(state != Qt.Checked)
[docs] def getScale(self): """ Return the tupe of min, max scaling. None for Auto """ if self.minAutoCheck.isChecked(): minValue = None else: minValue = self.minValueSpin.value() if self.maxAutoCheck.isChecked(): maxValue = None else: maxValue = self.maxValueSpin.value() return (minValue, maxValue)