DockWidget folder fixed

1 files changed, 220 insertions, 0 deletions

diff --git a/DockWidgets/DockWidget.py b/DockWidgets/DockWidget.py
new file mode 100644
index 0000000..6fc7731
--- /dev/null
+++ b/DockWidgets/DockWidget.py
@@ -0,0 +1,220 @@
+from PyQt5.QtCore import *
+from PyQt5.QtWidgets import *
+from PyQt5.QtGui import *
+from PyQt5.uic import loadUiType
+import pandas as pd
+from functools import partial
+from ComponentSelector import *
+from collections import defaultdict
+from Graphics import *
+
+ui_dialog,_ = loadUiType('DockWidgets/DockWidget.ui')
+
+class DockWidget(QDockWidget,ui_dialog):
+
+    def __init__(self,name,comptype,obj,container, parent=None):
+        QDockWidget.__init__(self,parent)
+        self.setupUi(self)
+        self.setWindowTitle(obj.name)
+        self.name=name
+        self.obj=obj
+        self.type = comptype
+        self.input_dict = {}
+        self.x_pclist = []
+        self.modes()
+        self.comboBox.currentIndexChanged.connect(self.mode_selection)
+       
+        print("constructor ", self.input_dict)
+        self.pushButton_2.clicked.connect(self.param)
+        self.dict = {}
+
+        self.name_type = None
+        self.container = container
+        
+    # input data tab
+    def modes(self):
+        modes_list = self.obj.modes_list
+        if(modes_list):
+            for j in modes_list:
+                self.comboBox.addItem(str(self.obj.variables[j]['name']))
+            self.mode_selection()
+        else:
+            self.comboBox.setDisabled(True)
+            self.input_dict= {}
+            self.input_dict = self.obj.param_getter()
+            self.input_params_list()
+
+    def mode_selection(self):
+        self.input_dict= {}
+        for i in reversed(range(self.formLayout.count())):
+            self.formLayout.removeRow(i) 
+        print(self.comboBox.currentText())
+        for i in self.obj.variables:
+            if self.obj.variables[i]['name'] == self.comboBox.currentText():
+                currentText = i
+                break
+        self.input_dict = self.obj.param_getter(currentText)
+        print('mode selection ', self.input_dict)
+        self.input_params_list()
+            
+    def input_params_list(self):
+        try:
+            print("input_params_list ", self.input_dict)
+            for c,i in enumerate(self.input_dict):
+                if i == None:
+                    continue
+                if(i=="thermo_package"):
+                    print("thermo1")
+                    combo = QComboBox()
+                    self.lines = [line.rstrip('\n') for line in open('thermopackage.txt')]
+                    print("thermo2")
+                    for j in self.lines:
+                        combo.addItem(str(j))
+                    combo.setMinimumContentsLength(15)
+                    lay = QGridLayout()
+                    lay.addWidget(QLabel(i+":"), 0,0, alignment=Qt.AlignLeft)
+                    lay.addWidget(combo, 0, 1, alignment=Qt.AlignCenter)
+                    lay.addWidget(QLabel(''), 0, 2, alignment = Qt.AlignCenter)
+                    self.formLayout.addRow(lay)
+                    self.input_dict[i] = combo   
+                    print("thermo")
+                elif(i=="Ctype"):
+                    combo = QComboBox()
+                    self.lines = ["Total","Partial"]
+                    for j in self.lines:
+                        combo.addItem(str(j))
+                    combo.setMinimumContentsLength(15)
+                    lay = QGridLayout()
+                    lay.addWidget(QLabel("Condensor Type :"), 0, 0, alignment=Qt.AlignLeft)
+                    lay.addWidget(combo, 0, 1, alignment=Qt.AlignCenter)
+                    lay.addWidget(QLabel(''), 0, 2, alignment = Qt.AlignCenter)
+                    self.formLayout.addRow(lay)
+                    self.input_dict[i] = combo
+                elif(i=="x_pc"):
+                    noc = len(compound_selected)
+                    print(noc)
+                    self.x_pclist.clear()
+                    gp = QGroupBox("Compounds")
+                    lay = QGridLayout()
+                    for j in range(noc):
+                        l = QLineEdit()    
+                        self.input_dict[i] = "x_pc"
+                        lay.addWidget(QLabel(str(compound_selected[j])+":"),j,0, alignment=Qt.AlignLeft)
+                        lay.addWidget(l,j,1, alignment=Qt.AlignCenter)
+                        lay.addWidget(QLabel(self.obj.variables[i]['unit']),j,2, alignment=Qt.AlignCenter)                                      
+                        self.x_pclist.append(l)
+                    gp.setLayout(lay)
+                    self.formLayout.addRow(gp)       
+                else:
+                    print("elseloop")
+                    print(i)
+                    if i == None:
+                        continue
+                    l = QLineEdit()
+                    if self.input_dict[i] != None:
+                        l.setText(str(self.input_dict[i]))
+                    print('before lay')
+                    lay = QGridLayout()
+                    lay.addWidget(QLabel(self.obj.variables[i]['name']+":"),0,0, alignment=Qt.AlignLeft) #self.obj.variables[i]['name']
+                    lay.addWidget(l,0,1, alignment=Qt.AlignCenter)
+                    print('after lay')
+                    if(i != 'MolFlow'):
+                        lay.addWidget(QLabel(self.obj.variables[i]['unit']),0,2, alignment=Qt.AlignCenter)
+                    else:
+                        lay.addWidget(QLabel('mol/s'),0,2, alignment=Qt.AlignCenter)
+                    print('after all')
+                    self.formLayout.addRow(lay)
+                    self.input_dict[i] = l
+                    
+            
+        except Exception as e:
+            print(e)
+
+    def show_error(self):
+        QMessageBox.about(self, 'Important', "Please fill all fields with data")
+
+    def param(self):
+        try:
+            self.dict={}
+            print("param.input_dict ", self.input_dict)
+            for i in self.input_dict:
+                if (self.input_dict[i] == None):
+                    continue                  
+                if(i=="thermo_package"):
+                    if (self.input_dict[i].currentText()):
+                        self.dict[i] = self.input_dict[i].currentText()
+                    else:
+                        self.show_error()
+                        break
+                elif(i=="Ctype"):
+                    if (self.input_dict[i].currentText()):
+                        self.dict[i] = self.input_dict[i].currentText()
+                    else:
+                        self.show_error()
+                        break
+                elif(i =="x_pc"):
+                    l=[]
+                    mf = []
+                    total_moles = 0
+                    for mol_frac in self.x_pclist:
+                        if (mol_frac.text()):
+                            l.append(mol_frac.text())
+                            total_moles += float(l[-1])
+                        else:
+                            self.show_error()
+                            break
+                    for c in range(len(compound_selected)):
+                        mf.append(str(float(l[c])/total_moles))
+                        self.x_pclist[c].setText(mf[-1])
+                    self.dict[i] = ",".join(mf)
+                elif(i == 'Pout' and self.obj.type == 'Mixer' or i == '' or i == 'HKey' or i == 'LKey'):
+                    self.dict[i] = self.input_dict[i].currentText()
+                else:
+                    print(self.input_dict[i], i, self.obj.type)
+                    if (self.input_dict[i].text()):
+                        self.dict[i] = self.input_dict[i].text()
+                    else:
+                        print(self.input_dict[i].text())
+                        self.show_error()
+                        break
+            
+            print("param ", self.dict)
+            self.obj.param_setter(self.dict)
+            self.hide()
+            
+        except Exception as e:
+            print(e)
+
+    @staticmethod
+    def show_result(lst):
+        #DockWidget1.flag = True
+        for i in lst:
+            i.results_category(i.name)
+            #i.show()
+        
+    # result data tab
+    def results_category(self,name):
+        flag = True
+        try:
+            print("Under result category name ", name)
+            result=self.container.result
+            obj = self.container.fetch_object(name)
+            self.tableWidget.setRowCount(0)
+            variKeys = list(obj.variables.keys())
+            print(variKeys)
+            for i, val in enumerate(variKeys):
+                propertyname = name + '.' + val
+                print(i,val, propertyname)
+                if propertyname in result[0]:
+                    ind = result[0].index(propertyname)
+                    resultval = str(result[-1][ind])
+                    print("######Resultsfetch####",val,resultval)
+                    rowPosition = self.tableWidget.rowCount()
+                    self.tableWidget.insertRow(rowPosition)
+                    self.tableWidget.setItem(rowPosition , 0, QTableWidgetItem(obj.variables[val]['name']))
+                    self.tableWidget.setItem(rowPosition , 1, QTableWidgetItem(resultval))
+                    self.tableWidget.setItem(rowPosition , 2, QTableWidgetItem(obj.variables[val]['unit']))
+                    self.tableWidget.resizeColumnsToContents()
+
+        except Exception as e:
+            print(e)
\ No newline at end of file