HYTOS

# -*- coding: utf-8 -*-

"""This is AirFinCooler module"""

from PyQt5 import QtCore, QtGui, QtWidgets

from PyQt5.QtWidgets import *

import os

from AppDocData import AppDocData

import AirFinCooler_UI

import math

def is_float(s):

    try:

        if s:

            float(s)

            return True

        else:

            return False

    except ValueError:

        return False

def is_blank(s):

    return not (s and s.strip())

def is_not_blank(s):

    return bool(s and s.strip())

def convert_to_fixed_point(value):

    if is_float(str(value)):

        tokens = f"{float(value):.10f}".split('.')

        if len(tokens) == 2:

            # 소수점 아래가 있을 경우 소수점 아래의 trailing zero를 제거한다.

            if is_blank(tokens[1].rstrip('0')):

                return tokens[0]

            else:

                tokens[1] = tokens[1].rstrip('0')

                return '.'.join(tokens)

        else:

            return tokens[0]

    else:

        return value

class QAirFinCooler(QDialog):

    def __init__(self):

        QDialog.__init__(self)

        self.ui = AirFinCooler_UI.Ui_AirFinCoolerDialog()

        self.ui.setupUi(self)

        self.ui.buttonBox.button(QDialogButtonBox.Ok).setIcon(QtGui.QIcon(':/images/OK.svg'))

        self.ui.buttonBox.button(QDialogButtonBox.Cancel).setIcon(QtGui.QIcon(':/images/Cancel.svg'))

        self._item = None

        self._modified = False

        self.ui.lineEdit_Pressure_Drop.setValidator(QtGui.QDoubleValidator(self.ui.lineEdit_Pressure_Drop))

        self.ui.lineEdit_Elevation.setValidator(QtGui.QDoubleValidator(self.ui.lineEdit_Elevation))

        self.initialize()

    def show_dialog(self, item):

        self.setWindowFlags(self.windowFlags() & ~QtCore.Qt.WindowContextHelpButtonHint)

        self._item = item

        self.ui.lineEdit_TagNo.setFocus()

        self.set_controls()

        self.load_data()

        self.exec_()

        return self._modified

    def initialize(self):

        self.ui.label_Img.setVisible(False)

    def set_controls(self):

        self.ui.label_Img.setVisible(True)

    def is_modified(self):

        """ Tag No 도 Output Explorer에 사용되기에 Modified 적용"""

        old_tag_no = self.ui.lineEdit_TagNo.placeholderText()

        new_tag_no = self.ui.lineEdit_TagNo.text()

        if old_tag_no != new_tag_no:

            self._modified = True

            return

        old_pressure_drop = self.ui.lineEdit_Pressure_Drop.placeholderText()

        new_pressure_drop = self.ui.lineEdit_Pressure_Drop.text()

        if old_pressure_drop != new_pressure_drop:

            self._modified = True

            return

        old_elevation = self.ui.lineEdit_Elevation.placeholderText()

        new_elevation = self.ui.lineEdit_Elevation.text()

        if old_elevation != new_elevation:

            self._modified = True

            return

    def load_data(self):

        """ load tag no and nozzle data """

        tag_no = self._item.tag_no

        if tag_no != 'None' and is_not_blank(tag_no):

            self.ui.lineEdit_TagNo.setText(tag_no)

            self.ui.lineEdit_TagNo.setPlaceholderText(tag_no)

        app_doc_data = AppDocData.instance()

        drawing = app_doc_data.activeDrawing

        if drawing:

            for attr in drawing.attrs:

                if attr[0] == 'Units':

                    self.ui.label_PressureUnit.setText(attr[1]['Pressure'])

                    self.ui.label_ElevationUnit.setText(attr[1]['Length'])

        matches = [connector.data for connector in self._item.connectors if connector.data]

        if matches:

            pressure_drop = matches[0].pressure_drop

            if pressure_drop is not None:

                self.ui.lineEdit_Pressure_Drop.setText(str(convert_to_fixed_point(pressure_drop)))

                self.ui.lineEdit_Pressure_Drop.setPlaceholderText(str(convert_to_fixed_point(pressure_drop)))

            elevation = matches[0].elevation

            if elevation is not None:

                self.ui.lineEdit_Elevation.setText(str(convert_to_fixed_point(elevation)))

                self.ui.lineEdit_Elevation.setPlaceholderText(str(convert_to_fixed_point(elevation)))

    def accept(self):

        """ check changed value """

        self.is_modified()

        """ set tag no and nozzle data """

        from EngineeringConnectorItem import NozzleData

        tag_no = self.ui.lineEdit_TagNo.text()

        if is_not_blank(tag_no):

            self._item.tag_no = tag_no

            for connector in self._item.connectors:

                if not connector.data:

                    connector.data = NozzleData()

                pressure_drop = self.ui.lineEdit_Pressure_Drop.text()

                if pressure_drop:

                    connector.data.pressure_drop = float(pressure_drop)

                else:

                    connector.data.pressure_drop = 0

                elevation = self.ui.lineEdit_Elevation.text()

                if elevation:

                    connector.data.elevation = float(elevation)

                else:

                    connector.data.elevation = 0

            QDialog.accept(self)

        else:

            QMessageBox.information(self, self.tr('Information'), self.tr('Please Input [Tag No.]'))

    def reject(self):

        QDialog.reject(self)