diff options
Diffstat (limited to 'src/main/python/utils/Bin_Phase_env.py')
| -rw-r--r-- | src/main/python/utils/Bin_Phase_env.py | 240 |
1 files changed, 240 insertions, 0 deletions
diff --git a/src/main/python/utils/Bin_Phase_env.py b/src/main/python/utils/Bin_Phase_env.py new file mode 100644 index 0000000..ee973f9 --- /dev/null +++ b/src/main/python/utils/Bin_Phase_env.py @@ -0,0 +1,240 @@ +import sys +import pandas as pd +import numpy as np +import os +import csv +from subprocess import Popen, PIPE + +from PyQt5.QtCore import * +from PyQt5.QtWidgets import * +from PyQt5.QtGui import * +import PyQt5.QtGui as QtGui +import PyQt5.QtCore as QtCore +import PyQt5.QtWidgets as QtWidgets +from PyQt5.uic import loadUiType + + +import pyqtgraph as pg +import pyqtgraph.exporters + +current = os.path.dirname(os.path.realpath(__file__)) +parent = os.path.dirname(current) +parentPath = os.path.dirname(parent) +sys.path.append(parentPath) + +ui_dialog,_ = loadUiType(parentPath+'/ui/utils/Binary_Phase_Env.ui') + +class BinPhaseEnv(QWidget,ui_dialog): + def __init__(self,comp): + QWidget.__init__(self) + self.setupUi(self) + + self.comp = comp + self.compunds = self.comp.get_compounds() + + for i in self.compunds: + self.comboBox.addItem(str(i)) + + for i in range(len(self.compunds)): + if i!=0: + self.comboBox_2.addItem(str(self.compunds[i])) + + self.comboBox_2.addItem(str(self.compunds[0])) + + self.lines = [line.rstrip('\n') for line in open(parentPath+'/python/utils/thermopackage.txt')] + for j in self.lines: + self.comboBox_3.addItem(str(j)) + + self.radioButton.setChecked(False) + self.radioButton_2.setChecked(False) + + self.button_handler() + self.counter = 1 + + def button_handler(self): + self.radioButton.clicked.connect(lambda:self.T_xy()) + self.radioButton_2.clicked.connect(lambda:self.P_xy()) + self.pushButton_2.clicked.connect(lambda:self.plot()) + + def T_xy(self): + self.radioButton.setChecked(True) + self.radioButton_2.setChecked(False) + for i in reversed(range(self.formLayout.count())): + self.formLayout.itemAt(i).widget().setParent(None) + + self.first = QLineEdit() + self.type = "P" + self.other = "T" + self.otherunit = "(K)" + self.formLayout.addRow(QLabel("P(Pa)"),self.first) + self.points = QLineEdit() + self.points.setText("40") + self.formLayout.addRow(QLabel("Number of data points"),self.points) + + def P_xy(self): + self.radioButton_2.setChecked(True) + self.radioButton.setChecked(False) + for i in reversed(range(self.formLayout.count())): + self.formLayout.itemAt(i).widget().setParent(None) + + self.first = QLineEdit() + self.type = "T" + self.other = "P" + self.otherunit = "(Pa)" + self.points = QLineEdit() + self.points.setText("40") + self.formLayout.addRow(QLabel("T(K)"),self.first) + self.formLayout.addRow(QLabel("Number of data points"),self.points) + + def get_omc_path(self): + try: + self.omhome = os.environ.get('OPENMODELICAHOME') + if self.omhome is None: + self.omhome = os.path.split(os.path.split(os.path.realpath(spawn.find_executable("omc")))[0])[0] + elif os.path.exists('/opt/local/bin/omc'): + self.omhome = '/opt/local' + elif os.path.exists('/usr/bin/omc'): + self.omhome = '/usr' + return os.path.join(self.omhome, 'bin', 'omc') + except BaseException: + print("The OpenModelica compiler is missing in the System path please install it" ) + raise + + def plot(self): + try: + val = int(self.first.text(),10) + except: + val = 0 + try: + data_points = int(self.points.text(),10) + except: + data_points = 0 + + self.curr_path = os.getcwd() + self.sim_dir_path = os.path.join(self.curr_path,'./../Simulator') + self.Graphmo_path = os.path.join(self.sim_dir_path,'Graph.mo') + self.plot_mos_path = os.path.join(self.sim_dir_path,'PlotGraph.mos') + + self.data = [] + + self.comp1 = self.comboBox.currentText() + self.comp2 = self.comboBox_2.currentText() + self.comp_1 = self.comboBox.currentText().split('(')[0] + self.comp_2 = self.comboBox_2.currentText().split('(')[0] + + self.thermoPack = self.comboBox_3.currentText() + + self.data.append("model Graph\n") + self.data.append("import data = Simulator.Files.ChemsepDatabase;\n") + self.data.append("parameter data."+self.comp_1+" comp1;\n") + self.data.append("parameter data."+self.comp_2+" comp2;\n") + self.data.append("extends BinaryEnvelopes."+self.thermoPack+"(Nc = 2, data_points = "+str(data_points)+ ", comp = { comp1, comp2 }, "+self.type+" = fill( "+str(val)+", "+str(data_points)+"));\n") + self.data.append("end Graph;") + + with open(self.Graphmo_path, 'w') as txtfile: + for d in self.data: + txtfile.write(str(d)) + + with open(self.plot_mos_path, 'w') as mosFile: + mosFile.write("loadModel(Modelica);\n") + mosFile.write("loadFile(\"Simulator/package.mo\");\n") + mosFile.write("loadFile(\"BinaryEnvelopes.mo\");\n") + mosFile.write("loadFile(\"Graph.mo\");\n") + mosFile.write("simulate(Graph, outputFormat=\"csv\", stopTime=1.0, numberOfIntervals=1);\n") + + self.resdata = [] + self.omc_path = self.get_omc_path() + simpath = self.plot_mos_path + os.chdir(self.sim_dir_path) + + process = Popen([self.omc_path, '-s',simpath], stdout=PIPE, stderr=PIPE) + self.stdout, self.stderr = process.communicate() + + os.chdir(self.curr_path) + + csvpath = os.path.join(self.sim_dir_path,'Graph_res.csv') + + self.datay = [] + self.datax1 = [] + self.datax2 = [] + self.rows = [] + + with open (csvpath,'r') as resultFile: + self.resdata = [] + csvreader = csv.reader(resultFile,delimiter=',') + for row in csvreader: + self.resdata.append(row) + self.rows.append(row) + + if self.type=='T': + for k in range(len(self.rows[0])): + if self.rows[0][k][0]=='P': + self.datay.append(float(self.rows[1][k])) + length = len(self.rows[0][k]) + if self.rows[0][k][0]=='x' and self.rows[0][k][length-2]=='1': + self.datax1.append(float(self.rows[1][k])) + if self.rows[0][k][0]=='y' and self.rows[0][k][length-2]=='1': + self.datax2.append(float(self.rows[1][k])) + else: + for k in range(len(self.rows[0])): + if self.rows[0][k][0]=='T': + self.datay.append(float(self.rows[1][k])) + length = len(self.rows[0][k]) + if self.rows[0][k][0]=='x' and self.rows[0][k][length-2]=='1': + self.datax1.append(float(self.rows[1][k])) + + if self.rows[0][k][0]=='y' and self.rows[0][k][length-2]=='1': + self.datax2.append(float(self.rows[1][k])) + + plt = pg.PlotWidget() + plt.showGrid(x=True,y=True) + plt.addLegend() + plt.setXRange(0,1) + + c1 = plt.plot(self.datax1, self.datay,pen=pg.mkPen('b',width = 1), name='dew points') + c2 = plt.plot(self.datax2, self.datay,pen=pg.mkPen('r',width = 1), name='bubble points') + view_box = plt.plotItem.vb + self.tool_tip = "" + + def press_event(evt): + a = 10 + pos = evt + mousepoint = view_box.mapSceneToView(pos) + roi = pg.ROI(pos) + find_color = plt.mapToGlobal(pos.toPoint()) + + screen = QGuiApplication.primaryScreen() + image = screen.grabWindow(QApplication.desktop().winId()).toImage() + colour = QtGui.QColor(image.pixel(find_color.x(),find_color.y())) + + if colour.red()==255 or colour.blue()==255: + self.lineEdit_x.setText(str(round(mousepoint.x(),3))) + self.lineEdit_y.setText(str(round(mousepoint.y(),3))) + self.tool_tip = str(round(mousepoint.x(),3)) + ", " + str(round(mousepoint.y(),3)) + QApplication.setOverrideCursor(QCursor(QtCore.Qt.CrossCursor)) + else: + self.lineEdit_x.setText("") + self.lineEdit_y.setText("") + self.tool_tip = "" + QApplication.setOverrideCursor(QCursor(QtCore.Qt.ArrowCursor)) + + def entered(items): + for i in items: + if i.__class__.__name__ =="LegendItem": + self.lineEdit_x.setText("") + self.lineEdit_y.setText("") + QApplication.setOverrideCursor(QCursor(QtCore.Qt.ArrowCursor)) + else: + i.setToolTip(self.tool_tip) + + plt.scene().sigMouseMoved.connect(press_event) + plt.scene().sigMouseHover.connect(entered) + + plt.setLabel('left',self.other+self.otherunit,units = '') + plt.setLabel('bottom',self.comp1+'(mol. frac.)',units = '') + + self.new_tab = plt + self.new_tab.setObjectName("Plot "+str(self.counter)) + + self.tabWidget.addTab(self.new_tab,"Plot "+str(self.counter)) + self.counter+=1 |