1 files changed, 99 insertions, 0 deletions
diff --git a/DockWidgets/DockWidgetCompoundSeparator.py b/DockWidgets/DockWidgetCompoundSeparator.py
new file mode 100644
index 0000000..1818290
--- /dev/null
+++ b/DockWidgets/DockWidgetCompoundSeparator.py
@@ -0,0 +1,99 @@
+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/DockWidgetCompoundSeparator.ui')
+
+class DockWidgetCompoundSeparator(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 = []
+        
+        print("constructor ", self.input_dict)
+        self.input_params_list()
+        self.dict = []
+            
+    def input_params_list(self):
+        try:
+            print("input_params_list ", self.input_dict)
+            if self.type == 'CompoundSeparator':
+
+                # self.gridLayout = QGridLayout()
+                calculationGroupBox = QGroupBox('Calculation Parameters')
+                
+                calculationLayout = QGridLayout()
+
+                r1 = QRadioButton('Stream 1')
+                r1.setChecked(True)
+                r1.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
+                r2 = QRadioButton('Stream 2')
+                r2.setChecked(False)
+                r2.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
+
+                lst = [r1, r2]
+                calculationLayout.addWidget(r1, 0, 1)
+                calculationLayout.addWidget(r2, 0, 2)
+
+                for k,val in enumerate(self.obj.compounds):
+                    combo = QComboBox()
+                    print("CompoundSeparator combo")
+                    for j in self.obj.SepFact_modes:
+                        combo.addItem(str(j))   
+                    combo.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
+                    l = QLineEdit()
+                    l.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
+                    calculationLayout.addWidget(QLabel(val+" :"), k+1,0, alignment=Qt.AlignLeft)
+                    calculationLayout.addWidget(combo, k+1, 1, alignment=Qt.AlignCenter)
+                    calculationLayout.addWidget(l,k+1,2, alignment=Qt.AlignCenter)
+                    lst.append(combo)
+                    lst.append(l)
+
+                calculationLayout.setColumnStretch(3, len(self.obj.compounds)+1)
+                calculationGroupBox.setLayout(calculationLayout)
+                
+                btn = QPushButton('Submit')
+                btn.clicked.connect(self.param)
+                
+                self.gridLayout.setVerticalSpacing(5)
+                self.gridLayout.addWidget(calculationGroupBox,0,0)
+                self.gridLayout.addWidget(btn,1,0)
+
+                self.input_dict = lst            
+                        
+        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)
+           
+            self.dict = [self.input_dict[0].isChecked(), self.input_dict[1].isChecked()]
+            j = 2
+            for i in range(len(self.obj.compounds)):
+                print(j+i)
+                self.dict.append(self.input_dict[j+i].currentText())
+                self.dict.append(self.input_dict[j+i+1].text())
+                j += 1
+            
+            print("param ", self.dict)
+            self.obj.param_setter(self.dict)
+            self.hide()
+            
+        except Exception as e:
+            print(e)