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, Qt

import PressureVariation_UI

import math

def is_float(s):

    try:

        if s:

            float(s)

            return True

        else:

            return False

    except ValueError:

        return False

def set_item_properties(name, alignment, color=None):

    if name is None:

        name = ''

    item = QTableWidgetItem(str(name))

    item.setTextAlignment(alignment)

    if color:

        item.setBackground(color)

    return item

def is_not_blank(s):

    return bool(s and s.strip())

class QPressureVariation(QDialog):

    def __init__(self):

        QDialog.__init__(self)

        self.ui = PressureVariation_UI.Ui_Dialog()

        self.ui.setupUi(self)

        self.units = None

        self.item = None

        self.init_units()

        self.initialize()

    def show_dialog(self, item):

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

        self.item = item

        self.load_data()

        return self.exec_()

    def init_units(self):

        app_doc_data = AppDocData.instance()

        active_drawing = app_doc_data.activeDrawing

        for attr in active_drawing.attrs:

            if attr[0] == 'Units':

                self.units = attr[1]

    def initialize(self):

        self.ui.tableWidget.setColumnCount(21)

        col_names = ['No.\n-', 'Element\n-', 'Diameter\n{}'.format(self.units['Pipe_Diameter']), 'Length\n{}'.format(self.units['Length']), 'Angle\n(deg)', 'K\n-',

                     'Pressure\n{}(g)'.format(self.units['Pressure']), 'Void\n-', 'Quality\n-', 'Density\n{}'.format(self.units['Density']), 'V.Den\n{}'.format(self.units['Density']),

                     'Mean Vel\n{}'.format(self.units['Velocity']), 'Max Vel\n{}'.format(self.units['Velocity']), 'Ero Vel\n{}'.format(self.units['Velocity']),

                     'Pattern\nX', 'Pattern\nY', 'Regime\n-', 'Friction\n{}/{}'.format(self.units['Pressure'], self.units['Length']), 'Gravity\n{}/{}'.format(self.units['Pressure'], self.units['Length']),

                     'Momentum\n{}/{}'.format(self.units['Pressure'], self.units['Length']), 'Total\n{}'.format(self.units['Pressure'])]

        self.ui.tableWidget.setHorizontalHeaderLabels(col_names)

        self.ui.tableWidget.verticalHeader().setVisible(False)

        self.ui.tableWidget.setSelectionMode(QAbstractItemView.SingleSelection)

        self.ui.tableWidget.setSelectionBehavior(QAbstractItemView.SelectRows)

        self.ui.tableWidget.setEditTriggers(QAbstractItemView.NoEditTriggers)

        self.ui.tableWidget.horizontalHeader().setStretchLastSection(True)

    def add_data(self, no, element, diameter, length, angle, k, pressure, void, quality, density, v_den, mean_vel, max_vel, ero_vel, x, y, regime, friction, gravity, momentum, total):

        row = self.ui.tableWidget.rowCount()

        self.ui.tableWidget.setRowCount(row + 1)

        self.ui.tableWidget.setItem(row, 0, set_item_properties(no, Qt.AlignHCenter | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 1, set_item_properties(element, Qt.AlignLeft | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 2, set_item_properties(diameter, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 3, set_item_properties(length, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 4, set_item_properties(angle, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 5, set_item_properties(k, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 6, set_item_properties(pressure, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 7, set_item_properties(void, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 8, set_item_properties(quality, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 9, set_item_properties(density, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 10, set_item_properties(v_den, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 11, set_item_properties(mean_vel, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 12, set_item_properties(max_vel, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 13, set_item_properties(ero_vel, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 14, set_item_properties(x, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 15, set_item_properties(y, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 16, set_item_properties(regime, Qt.AlignHCenter | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 17, set_item_properties(friction, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 18, set_item_properties(gravity, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 19, set_item_properties(momentum, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.setItem(row, 20, set_item_properties(total, Qt.AlignRight | Qt.AlignVCenter))

        self.ui.tableWidget.resizeRowsToContents()

        self.ui.tableWidget.resizeColumnsToContents()

    def load_data(self):

        data = self.item.mixed_calculation_result

        for no in range(data.no):

            element = data.element[no] if no in data.element else None

            inside_diameter = round(data.inside_diameter[no], 3) if no in data.inside_diameter else None

            length = round(data.length[no], 3) if no in data.length else None

            angle = data.angle[no] if no in data.angle else None

            k = round(data.k[no], 3) if no in data.k else None

            pressure = round(data.pressure[no], 3) if no in data.pressure else None

            void = round(data.void[no], 3) if no in data.void else None

            quality = round(data.quality[no], 3) if no in data.quality else None

            mean_den = round(data.mean_den[no], 3) if no in data.mean_den else None

            v_density = round(data.v_density[no], 3) if no in data.v_density else None

            homo_vel = round(data.homo_vel[no], 3) if no in data.homo_vel else None

            max_vel = round(data.max_vel[no], 3) if no in data.max_vel else None

            ero_vel = round(data.ero_vel[no], 3) if no in data.ero_vel else None

            x = round(data.x[no], 3) if no in data.x else None

            y = round(data.y[no], 3) if no in data.y else None

            regime = data.regime[no] if no in data.regime else None

            dp_fric = round(data.dp_fric[no], 5) if no in data.dp_fric else None

            dp_stat = round(data.dp_stat[no], 5) if no in data.dp_stat else None

            dp_momen = round(data.dp_momen[no], 5) if no in data.dp_momen else None

            total_length = round(data.total_length[no], 5) if no in data.total_length else None

            self.add_data(no, element, inside_diameter, length, angle, k, pressure, void, quality, mean_den, v_density, homo_vel, max_vel, ero_vel, x, y, regime, dp_fric, dp_stat, dp_momen, total_length)

    def accept(self):

        QDialog.accept(self)

    def reject(self):

        QDialog.reject(self)