moulder/configure_dialog.py

from __future__ import print_function
from future.builtins import super

import os
import sys
import numpy
import matplotlib
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg
from PyQt5.QtGui import QIcon, QFont
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QSizePolicy, QMainWindow, QApplication, QAction
from PyQt5.QtWidgets import QMenu, QWidget, QVBoxLayout, QMessageBox
from PyQt5.QtWidgets import QSlider, QHBoxLayout, QLabel, QDialog, QPushButton
from PyQt5.QtWidgets import QDialogButtonBox, QGridLayout, QRadioButton, QLineEdit


class ConfigureMeassurementDialog(QDialog):

    def __init__(self, parent):
        super().__init__(parent)
        self.setModal(False)
        self.setWindowTitle("Configure Meassurement Points")
        self._completed = False
        self._init_ui()

        self.regular_grid_btn.toggled.connect(self._radio_button_callback)
        self.custom_grid_btn.toggled.connect(self._radio_button_callback)
        self.cancel_btn.clicked.connect(self._button_pushed_callback)
        self.apply_btn.clicked.connect(self._button_pushed_callback)

    @property
    def x(self):
        if self.regular_grid_btn.isChecked():
            entries = self._read_regular_grid_entries()
            if entries:
                x1, x2, step, z = entries[:]
                return numpy.arange(x1, x2, step, dtype=numpy.float64)
            else:
                return None
        elif self.custom_grid_btn.isChecked():
            # Need to be completed
            pass

    @property
    def z(self):
        if self.regular_grid_btn.isChecked():
            entries = self._read_regular_grid_entries()
            if entries:
                x1, x2, step, z = entries[:]
                return z*numpy.ones_like(self.x)
            else:
                return None
        elif self.custom_grid_btn.isChecked():
            # Need to be completed
            pass

    def is_completed(self):
        return self._completed

    def _init_ui(self):
        self.regular_grid_btn = QRadioButton("Regular grid (in meters)")
        self.regular_grid_btn.setChecked(True)
        self.custom_grid_btn = QRadioButton("Custom grid")
        self.from_input = QLineEdit()
        self.to_input = QLineEdit()
        self.step_input = QLineEdit()
        self.height_input = QLineEdit()
        self.apply_btn = QPushButton("Apply Changes")
        self.cancel_btn = QPushButton("Cancel")
        self.apply_btn.setDefault(True)

        bold_font = QFont()
        bold_font.setBold(True)

        layout = QVBoxLayout()
        layout.addWidget(QLabel("Create Meassurement Points", font=bold_font))
        layout.addWidget(self.regular_grid_btn)

        grid = QGridLayout()
        grid.setContentsMargins(25, 0, 0, 0)
        grid.addWidget(QLabel("From:"), 0, 0)
        grid.addWidget(self.from_input, 0, 1)
        grid.addWidget(QLabel("To:"), 0, 2)
        grid.addWidget(self.to_input, 0, 3)
        grid.addWidget(QLabel("Step:"), 0, 4)
        grid.addWidget(self.step_input, 0, 5)
        grid.addWidget(QLabel("Height:"), 1, 0)
        grid.addWidget(self.height_input, 1, 1, 1, 5)
        layout.addLayout(grid)

        layout.addWidget(self.custom_grid_btn)

        hbox = QHBoxLayout()
        hbox.setAlignment(Qt.AlignRight)
        hbox.addWidget(self.cancel_btn)
        hbox.addWidget(self.apply_btn)
        layout.addLayout(hbox)

        self.setLayout(layout)

    def _button_pushed_callback(self):
        sender_text = self.sender().text()
        if sender_text == "Cancel":
            self.close()
        elif sender_text == "Apply Changes":
            filled_entries = self._check_filled_entries()
            if filled_entries:
                self._completed = True
                self.close()
            else:
                QMessageBox.warning(self, "Warning",
                                    "Some entries are not properly " +
                                    "completed or are incomplete.")

    def _radio_button_callback(self):
        regular_grid_lines = [self.from_input, self.to_input,
                              self.step_input, self.height_input]
        if self.sender().text() == "Custom grid":
            for line_edit in regular_grid_lines:
                line_edit.setDisabled(True)
        else:
            for line_edit in regular_grid_lines:
                line_edit.setEnabled(True)

    def _check_filled_entries(self):
        if self.regular_grid_btn.isChecked():
            entries = self._read_regular_grid_entries()
            if entries:
                return True
            else:
                # Show messagebox with warning for not completed entries
                pass
        elif self.custom_grid_btn.isChecked():
            # Needed to be completed
            return False

    def _read_regular_grid_entries(self):
        x1, x2 = self.from_input.text(), self.to_input.text()
        step = self.step_input.text()
        z = self.height_input.text()
        try:
            x1, x2, step = float(x1), float(x2), float(step)
            z = float(z)
        except ValueError:
            return False
        return x1, x2, step, z