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 Strainer_Y_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 QStrainer_Y(QDialog):

    def __init__(self):

        QDialog.__init__(self)

        self.ui = Strainer_Y_UI.Ui_Strainer_Y_Dialog()

        self.ui.setupUi(self)

        self._item = None

        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()

        return self.exec_()

    def initialize(self):

        self.ui.label_Img.setVisible(False)

    def set_controls(self):

        self.ui.label_Img.setVisible(True)

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

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

            elevation = matches[0].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:

                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)