HYTOS

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

# Form implementation generated from reading ui file 'ProjectDialog.ui'

#

# Created by: PyQt5 UI code generator 5.6

#

# WARNING! All changes made in this file will be lost!

from PyQt5 import QtCore, QtGui, QtWidgets

from PyQt5.QtWidgets import *

import os

from AppDocData import AppDocData

import Kompressor_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 QKompressor(QDialog):

    def __init__(self):

        QDialog.__init__(self)

        self.ui = Kompressor_UI.Ui_KompressorDialog()

        self.ui.setupUi(self)

        self._item = None

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

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

    def show_dialog(self, item):

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

        self._item = item

        self.ui.lineEdit_TagNo.setFocus()

        self.load_data()

        return self.exec_()

    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)

        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'])

        for connector in self._item.connectors:

            index = connector._conn_index

            if connector.data:

                pressure_drop = connector.data.pressure_drop

                if pressure_drop is not None:

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

                elevation = connector.data.elevation

                if elevation is not None:

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

    def accept(self):

        """ 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:

                index = connector._conn_index

                if not connector.data:

                    connector.data = NozzleData()

                pressure_drop = self.ui.lineEdit_Diff_Pressure.text()

                if pressure_drop:

                    connector.data.pressure_drop = float(pressure_drop)

                else:

                    connector.data.pressure_drop = None

                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)