610 lines
20 KiB
Python
610 lines
20 KiB
Python
import sys
|
||
import pandas as pd
|
||
|
||
import check_connection
|
||
import rw_device_params
|
||
import analize
|
||
import ammeter
|
||
import voltmeter
|
||
import power_unit
|
||
|
||
from device import Ui_Device
|
||
from main_window import Ui_MainWindow
|
||
|
||
from PyQt6 import QtCore
|
||
from PyQt6.QtWidgets import QApplication, QMainWindow, QWidget, QFileDialog
|
||
import pyqtgraph as pg
|
||
|
||
style_sheet_button_enabled = 'QPushButton{font-size:16px;border-width: 1px;border-style: solid;border-radius: 10px;background-color:green;}QPushButton::hover{background-color:red;}'
|
||
style_sheet_button_unabled = 'QPushButton{font-size:16px;border-width: 1px;border-style: solid;border-radius: 10px;background-color:rgb(44,47,52);}QPushButton::hover{background-color:red;}'
|
||
style_sheet_polarity_plus = 'QPushButton{font-size:16px;border-width: 1px;border-style: solid;border-radius: 10px;background-color:rgb(139,0,0);}QPushButton::hover{background-color:red;}'
|
||
style_sheet_polarity_minus = 'QPushButton{font-size:16px;border-width: 1px;border-style: solid;border-radius: 10px;background-color:rgb(2,14,93);}QPushButton::hover{background-color:red;}'
|
||
|
||
|
||
class ThreadCurrent(QtCore.QThread):
|
||
completed = QtCore.pyqtSignal()
|
||
|
||
def __init__(self, parent):
|
||
QtCore.QThread.__init__(self)
|
||
self.running = False
|
||
self.parent = parent
|
||
self.ammeter = ammeter
|
||
|
||
def run(self):
|
||
self.running = True
|
||
self.parent.measuredI = self.parent.ammeter.getCurrentDC()
|
||
self.completed.emit()
|
||
|
||
|
||
class ThreadVoltage(QtCore.QThread):
|
||
completed = QtCore.pyqtSignal()
|
||
|
||
def __init__(self, parent):
|
||
QtCore.QThread.__init__(self)
|
||
self.running = False
|
||
self.parent = parent
|
||
|
||
def run(self):
|
||
self.running = True
|
||
self.parent.measuredV = self.parent.voltmeter.getVoltageDC()
|
||
self.completed.emit()
|
||
|
||
|
||
class ThreadMeasureVac(QtCore.QThread):
|
||
|
||
def __init__(self, parent, start_voltage, end_voltage, step_voltage, polarity, A2, V2, PU):
|
||
QtCore.QThread.__init__(self)
|
||
self.running = False
|
||
self.parent = parent
|
||
|
||
self.step_voltage = step_voltage
|
||
self.current_voltage = start_voltage
|
||
self.end_voltage = end_voltage
|
||
self.ammeter = A2
|
||
self.voltmeter = V2
|
||
self.power_unit = PU
|
||
self.polarity = polarity
|
||
|
||
# Threads for voltmeter and ammeter
|
||
self.currThread = ThreadCurrent(self)
|
||
self.voltThread = ThreadVoltage(self)
|
||
|
||
# Variables to save measured data
|
||
self.measuredV = 0
|
||
self.measuredI = 0
|
||
|
||
def run(self):
|
||
|
||
while self.current_voltage < self.end_voltage:
|
||
self.power_unit.setVoltageDC(self.current_voltage)
|
||
self.sleep(1)
|
||
self.current_voltage += self.step_voltage
|
||
self.get_measure()
|
||
|
||
if self.polarity == True:
|
||
self.parent.I.append(self.measuredI*1e6)
|
||
self.parent.V.append(self.measuredV)
|
||
else:
|
||
self.parent.I.append(-1*self.measuredI*1e6)
|
||
self.parent.V.append(-1*self.measuredV)
|
||
self.power_unit.setVoltageDC(0)
|
||
|
||
def get_measure(self):
|
||
self.currThread.start()
|
||
self.voltThread.start()
|
||
self.sleep(2)
|
||
|
||
|
||
class ThreadMeasure(QtCore.QThread):
|
||
|
||
def __init__(self, parent, A1, V1):
|
||
QtCore.QThread.__init__(self)
|
||
self.running = False
|
||
self.parent = parent
|
||
|
||
self.ammeter = A1
|
||
self.voltmeter = V1
|
||
|
||
# Threads for voltmeter and ammeter
|
||
self.currThread = ThreadCurrent(self)
|
||
self.voltThread = ThreadVoltage(self)
|
||
|
||
# Variables to save measured data
|
||
self.measuredV = 0
|
||
self.measuredI = 0
|
||
|
||
def run(self):
|
||
while self.running != False:
|
||
self.get_measure()
|
||
self.parent.I.append(self.measuredI*1000)
|
||
self.parent.V.append(self.measuredV)
|
||
|
||
def get_measure(self):
|
||
self.currThread.start()
|
||
self.voltThread.start()
|
||
self.sleep(2)
|
||
|
||
|
||
class Window_Device(QWidget, Ui_Device):
|
||
|
||
def __init__(self, device_name):
|
||
|
||
super().__init__()
|
||
|
||
self.setupUi(self)
|
||
|
||
self.name = device_name
|
||
|
||
self.params = rw_device_params.read_device_params(self.name)
|
||
|
||
self.id_edit.setText(str(self.params[3]))
|
||
|
||
self.device_name.setText(str(self.params[2]))
|
||
|
||
self.device_status.setText(str(self.params[1]))
|
||
|
||
self.baud_rate_edit.setText(self.params[5])
|
||
|
||
self.connectSignalSlots()
|
||
|
||
self.setWindowTitle('Параметры {}'.format(self.name))
|
||
|
||
self.setFixedSize(506, 374)
|
||
|
||
def connectSignalSlots(self):
|
||
|
||
self.baud_rate_edit.textChanged.connect(self.baud_rate_changed)
|
||
self.id_edit.textChanged.connect(self.id_changed)
|
||
|
||
def baud_rate_changed(self, s):
|
||
rw_device_params.write_device_params(self.name, 'baud_rate', s)
|
||
|
||
def id_changed(self, s):
|
||
rw_device_params.write_device_params(self.name, 'id', s)
|
||
|
||
|
||
class Window(QMainWindow, Ui_MainWindow):
|
||
def __init__(self):
|
||
|
||
# Init window
|
||
super().__init__()
|
||
self.setupUi(self)
|
||
|
||
# Refresh list of connected deivces
|
||
self.update_devices()
|
||
|
||
self.connectSignalsSlots()
|
||
|
||
# Timer for updating window
|
||
self.timer_update_1 = QtCore.QTimer()
|
||
self.timer_update_1.setInterval(500)
|
||
self.timer_update_1.timeout.connect(self.update_measure_vac)
|
||
|
||
self.timer_update = QtCore.QTimer()
|
||
self.timer_update.setInterval(500)
|
||
self.timer_update.timeout.connect(self.update_measure)
|
||
|
||
self.graph_vac_1.setLabel(
|
||
'left', "<span style=\"color:white;font-size:30px\">I, мкА</span>")
|
||
self.graph_vac_1.setLabel(
|
||
'bottom', "<span style=\"color:white;font-size:30px\">U, В</span>")
|
||
|
||
self.graph_vac.setLabel(
|
||
'left', "<span style=\"color:white;font-size:30px\">I, мА</span>")
|
||
self.graph_vac.setLabel(
|
||
'bottom', "<span style=\"color:white;font-size:30px\">U, В</span>")
|
||
|
||
self.graph_vac_4.setLabel(
|
||
'left', "<span style=\"color:white;font-size:30px\">I, мкА</span>")
|
||
self.graph_vac_4.setLabel(
|
||
'bottom', "<span style=\"color:white;font-size:30px\">U, В</span>")
|
||
self.graph_vac_4.addLegend(
|
||
labelTextColor='white', labelTextSize='16pt')
|
||
|
||
# Variables
|
||
self.start_voltage = 0
|
||
self.end_voltage = 0
|
||
self.step_voltage = 0
|
||
self.exp_A = 0
|
||
self.I = []
|
||
self.V = []
|
||
self.polarity = True
|
||
self.time_estimated = 0
|
||
self.save_directory = ''
|
||
self.file_name = ''
|
||
self.thread = QtCore.QThread()
|
||
|
||
# Connect all signal slots
|
||
def connectSignalsSlots(self):
|
||
|
||
self.properties_V1.triggered.connect(
|
||
lambda: self.device('V1'))
|
||
self.properties_V2.triggered.connect(
|
||
lambda: self.device('V2'))
|
||
self.properties_A1.triggered.connect(
|
||
lambda: self.device('A1'))
|
||
self.properties_A2.triggered.connect(
|
||
lambda: self.device('A2'))
|
||
|
||
self.directory.clicked.connect(
|
||
lambda: self.choose_directory())
|
||
|
||
self.choose_folder.triggered.connect(
|
||
lambda: self.choose_directory())
|
||
|
||
self.begin_measure_1.clicked.connect(
|
||
lambda: self.measure_vac())
|
||
|
||
self.end_measure_1.clicked.connect(
|
||
lambda: self.end_of_measure_vac())
|
||
|
||
self.change_polarity_1.clicked.connect(
|
||
lambda: self.change_polarity())
|
||
|
||
self.confirm_measure_1.clicked.connect(
|
||
lambda: self.confirm_measure())
|
||
|
||
self.navigation.currentChanged.connect(
|
||
lambda: self.plot_to_run())
|
||
|
||
self.begin_measure.clicked.connect(
|
||
lambda: self.measure())
|
||
|
||
self.end_measure.clicked.connect(
|
||
lambda: self.end_of_measure())
|
||
|
||
# Updates list of connected devices
|
||
def update_devices(self):
|
||
|
||
self.devices = check_connection.check_devices_connection()
|
||
|
||
self.device_status()
|
||
|
||
# Updates device status on main_window
|
||
def device_status(self):
|
||
self.A1_status.setText(self.devices[2])
|
||
if self.A1_status.text() == 'Online':
|
||
self.A1_status.setStyleSheet('color : rgb(144,238,144);')
|
||
else:
|
||
self.A1_status.setStyleSheet('color : red;')
|
||
self.A2_status.setText(self.devices[3])
|
||
if self.A2_status.text() == 'Online':
|
||
self.A2_status.setStyleSheet('color : rgb(144,238,144);')
|
||
else:
|
||
self.A2_status.setStyleSheet('color : red;')
|
||
self.V1_Status.setText(self.devices[0])
|
||
if self.V1_Status.text() == 'Online':
|
||
self.V1_Status.setStyleSheet('color : rgb(144,238,144);')
|
||
else:
|
||
self.V1_Status.setStyleSheet('color : red;')
|
||
self.V2_status.setText(self.devices[1])
|
||
if self.V2_status.text() == 'Online':
|
||
self.V2_status.setStyleSheet('color : rgb(144,238,144);')
|
||
else:
|
||
self.V2_status.setStyleSheet('color : red;')
|
||
|
||
# Open window with device params
|
||
def device(self, device_name):
|
||
|
||
self.device_window = Window_Device(device_name)
|
||
|
||
self.device_window.show()
|
||
|
||
# Open a folder_window to choose folder
|
||
def choose_directory(self):
|
||
dlg = QFileDialog
|
||
folderpath = dlg.getExistingDirectory(self, 'Выбор директории')
|
||
self.save_directory = folderpath
|
||
self.chosen_directory.setText(self.save_directory)
|
||
|
||
# Start measurement
|
||
def measure(self):
|
||
|
||
if self.save_directory != '':
|
||
|
||
# Create voltmeter and ammeter
|
||
try:
|
||
self.A1 = ammeter.B7_78('/dev/ttyUSB0')
|
||
self.V1 = voltmeter.AKIP('/dev/usbtmc0')
|
||
except:
|
||
self.error.setText(QtCore.QTime.currentTime().toString(
|
||
)+' - [Errno2]')
|
||
# Start thread for measurments
|
||
self.thread = ThreadMeasure(
|
||
self, self.A1, self.V1)
|
||
self.thread.start()
|
||
self.thread.running = True
|
||
|
||
# Start timer
|
||
self.timer_update.start()
|
||
|
||
# Set graph params
|
||
self.graph_vac.setXRange(29, 36, padding=0)
|
||
self.graph_vac.setYRange(0, 6, padding=0)
|
||
self.graph_vac.showGrid(x=True, y=True, alpha=1)
|
||
|
||
# Set buttons properties
|
||
self.begin_measure.setEnabled(False)
|
||
self.end_measure.setEnabled(True)
|
||
self.end_measure.setStyleSheet(style_sheet_button_enabled)
|
||
self.begin_measure.setStyleSheet(style_sheet_button_unabled)
|
||
|
||
# Set experiment status
|
||
self.exp_status.setText('В процессе')
|
||
|
||
else:
|
||
self.error.setText(QtCore.QTime.currentTime().toString(
|
||
)+' - [Errno0]')
|
||
|
||
# Start measurment for vac
|
||
def measure_vac(self):
|
||
# Create voltmeter, ammeter and power_unit
|
||
try:
|
||
self.A2 = ammeter.B7_78('/dev/ttyUSB0')
|
||
self.PU = power_unit.Agilent('/dev/ttyUSB1')
|
||
self.V2 = voltmeter.AKIP('/dev/usbtmc0')
|
||
|
||
# Start thread for measurments
|
||
self.thread = ThreadMeasureVac(
|
||
self, self.start_voltage, self.end_voltage, self.step_voltage, self.polarity, self.A2, self.V2, self.PU)
|
||
self.thread.start()
|
||
|
||
# Start timer
|
||
self.timer_update_1.start()
|
||
|
||
# Set graph params
|
||
self.graph_vac_1.setXRange(-26, 26, padding=0)
|
||
self.graph_vac_1.setYRange(-130, 130, padding=0)
|
||
self.graph_vac_1.showGrid(x=True, y=True, alpha=1)
|
||
# Set buttons properties
|
||
self.begin_measure_1.setEnabled(False)
|
||
self.end_measure_1.setEnabled(True)
|
||
self.change_polarity_1.setEnabled(True)
|
||
self.change_polarity_1.setStyleSheet(style_sheet_button_enabled)
|
||
self.end_measure_1.setStyleSheet(style_sheet_button_enabled)
|
||
self.begin_measure_1.setStyleSheet(style_sheet_button_unabled)
|
||
|
||
# Set experiment status
|
||
self.exp_status_1.setText('В процессе')
|
||
except:
|
||
self.error.setText(QtCore.QTime.currentTime().toString(
|
||
)+' - [Errno2]')
|
||
|
||
# End measurement for vac
|
||
def end_of_measure_vac(self):
|
||
|
||
# Stop measurement thread
|
||
self.thread.quit()
|
||
|
||
self.timer_update_1.stop()
|
||
|
||
# Set experiment status
|
||
self.exp_status_1.setText('Окончен')
|
||
|
||
# Update button properties
|
||
self.change_polarity_1.setStyleSheet(style_sheet_button_unabled)
|
||
self.change_polarity_1.setEnabled(False)
|
||
self.confirm_measure_1.setEnabled(True)
|
||
self.begin_measure_1.setEnabled(False)
|
||
self.confirm_measure_1.setStyleSheet(style_sheet_button_enabled)
|
||
self.begin_measure_1.setStyleSheet(style_sheet_button_unabled)
|
||
|
||
self.exp_A_1.setEnabled(True)
|
||
self.start_voltage_1.setEnabled(True)
|
||
self.end_voltage_1.setEnabled(True)
|
||
self.step_voltage_1.setEnabled(True)
|
||
|
||
# Save collected data
|
||
df = pd.DataFrame(list(zip(self.V, self.I)), columns=['U', 'I'])
|
||
df.to_csv(self.save_directory+'/'+self.file_name+'mA.csv')
|
||
|
||
# Build fitted curve
|
||
self.I = df['I']
|
||
self.U = df['U']
|
||
|
||
df, params, pcov = analize.getApproxValues(pd.DataFrame(
|
||
list(zip(self.U, self.I)), columns=['U', 'I']),)
|
||
self.graph_vac_1.plot(
|
||
df['U'], df['I'], pen=pg.mkPen('yellow', width=1))
|
||
|
||
# Plot fitted curve
|
||
self.graph_vac_1.plot(self.U, self.I, symbol='o',
|
||
pen=None, symbolSize=10, symbolPen='yellow')
|
||
|
||
tmp = analize.discrepancy([self.I, self.U], params)
|
||
|
||
error = pg.ErrorBarItem(
|
||
x=self.U, y=self.I, top=tmp, bottom=tmp, beam=0.5)
|
||
self.graph_vac_1.addItem(error)
|
||
|
||
# Update value labels
|
||
self.I_n_1.setText(str(int(params[0])))
|
||
self.T_e_1.setText(str(int(params[1])))
|
||
self.I_n_6.setText(str(int(params[0])))
|
||
self.T_e_6.setText(str(int(params[1])))
|
||
# Zero variables
|
||
|
||
self.U = []
|
||
self.I = []
|
||
self.time_estimated = 0
|
||
|
||
# End measurement
|
||
def end_of_measure(self):
|
||
|
||
# Stop measurement thread
|
||
self.thread.running = False
|
||
|
||
self.timer_update.stop()
|
||
|
||
# Set experiment status
|
||
self.exp_status.setText('Окончен')
|
||
|
||
# Update button properties
|
||
self.begin_measure.setEnabled(False)
|
||
self.begin_measure.setStyleSheet(style_sheet_button_unabled)
|
||
# Save collected data
|
||
df = pd.DataFrame(list(zip(self.V, self.I)), columns=['U', 'I'])
|
||
df.to_csv(self.save_directory+'/'+'discharge_VAC.csv')
|
||
|
||
# Build fitted curve
|
||
|
||
# Update value labels
|
||
|
||
# Zero variables
|
||
self.U = []
|
||
self.I = []
|
||
self.time_estimated = 0
|
||
|
||
# Update labels and graph during vac experiment
|
||
def update_measure_vac(self):
|
||
# Update time
|
||
self.time_estimated += 0.5
|
||
|
||
# Update value labels
|
||
self.amount_of_dots_1.setText(str(len(self.I)))
|
||
self.time_from_begin_1.setText(str(self.time_estimated))
|
||
|
||
try:
|
||
self.last_A_1.setText(str(self.I[-1]))
|
||
self.last_V_1.setText(str(self.V[-1]))
|
||
except (IndexError, ValueError, KeyError):
|
||
pass
|
||
|
||
# Plot current V and I
|
||
self.graph_vac_1.plot(self.V, self.I, symbol='o',
|
||
pen=None, symbolSize=10, symbolPen='yellow')
|
||
|
||
# Update labels and graph during discharge experiment
|
||
def update_measure(self):
|
||
|
||
# Update time
|
||
self.time_estimated += 0.5
|
||
|
||
# Update value labels
|
||
self.amount_of_dots.setText(str(len(self.I)))
|
||
self.time_from_begin.setText(str(self.time_estimated))
|
||
|
||
try:
|
||
self.last_A.setText(str(self.I[-1]))
|
||
self.last_V.setText(str(self.V[-1]))
|
||
except (IndexError, ValueError):
|
||
pass
|
||
|
||
# Plot current V and I
|
||
self.graph_vac.plot(self.V, self.I, symbol='o',
|
||
pen=None, symbolSize=10, symbolPen='yellow')
|
||
|
||
# Start a thread with changed polarity
|
||
def change_polarity(self):
|
||
|
||
# Stop existing thread
|
||
self.thread.quit()
|
||
|
||
# Update polarity value
|
||
self.polarity = not self.polarity
|
||
|
||
# Creat new thread for measurement with changed polarity
|
||
self.thread = ThreadMeasureVac(
|
||
self, self.start_voltage, self.end_voltage, self.step_voltage, self.polarity, self.A2, self.V2, self.PU)
|
||
self.thread.start()
|
||
|
||
# Update button properties
|
||
if self.polarity == True:
|
||
self.change_polarity_1.setStyleSheet(style_sheet_polarity_plus)
|
||
else:
|
||
self.change_polarity_1.setStyleSheet(style_sheet_polarity_minus)
|
||
|
||
# Set experment params for vac
|
||
def confirm_measure(self):
|
||
if self.save_directory != '':
|
||
try:
|
||
|
||
# Save measure params from line_edits
|
||
self.exp_A = str(self.exp_A_1.text())
|
||
self.start_voltage = float(self.start_voltage_1.text())
|
||
self.end_voltage = float(self.end_voltage_1.text())
|
||
self.step_voltage = float(self.step_voltage_1.text())
|
||
self.file_name = str(self.exp_A_1.text())
|
||
# Set button properties
|
||
self.confirm_measure_1.setEnabled(False)
|
||
self.begin_measure_1.setEnabled(True)
|
||
|
||
self.confirm_measure_1.setStyleSheet(
|
||
style_sheet_button_unabled)
|
||
self.begin_measure_1.setStyleSheet(style_sheet_button_enabled)
|
||
|
||
self.exp_A_1.setEnabled(False)
|
||
self.start_voltage_1.setEnabled(False)
|
||
self.end_voltage_1.setEnabled(False)
|
||
self.step_voltage_1.setEnabled(False)
|
||
self.graph_vac_1.clear()
|
||
self.I_n_1.setText('Не определено')
|
||
self.T_e_1.setText('Не определено')
|
||
self.time_from_begin_1.setText('None')
|
||
self.amount_of_dots_1.setText('None')
|
||
self.last_V_1.setText('None')
|
||
self.last_A_1.setText('None')
|
||
self.exp_status_1.setText('Не начат')
|
||
self.graph_vac_1.plotItem.clear()
|
||
except ValueError:
|
||
self.error.setText(QtCore.QTime.currentTime().toString(
|
||
)+' - [Errno1]')
|
||
else:
|
||
self.error.setText(QtCore.QTime.currentTime().toString(
|
||
)+' - [Errno0]')
|
||
|
||
# Check which tab has been opened
|
||
def plot_to_run(self):
|
||
if self.navigation.currentIndex() == 3:
|
||
self.plot_all()
|
||
else:
|
||
self.graph_vac_4.clear()
|
||
|
||
# Plot all data on 3rd tab
|
||
def plot_all(self):
|
||
if self.save_directory != '':
|
||
T = [self.T_e_4, self.T_e_5, self.T_e_6]
|
||
I = [self.I_n_4, self.I_n_5, self.I_n_6]
|
||
B = [self.B_koef_4, self.B_koef_5, self.B_koef_6]
|
||
for file, color, T_e, I_n, B_koef in zip(['1.5mA.csv', '3.0mA.csv', '5.0mA.csv'], ['green', 'yellow', 'orange'], T, I, B):
|
||
try:
|
||
|
||
df = pd.read_csv(self.save_directory+'/'+file)
|
||
self.I = df['I']
|
||
self.U = df['U']
|
||
|
||
df, params, error = analize.getApproxValues(pd.DataFrame(
|
||
list(zip(self.U, self.I)), columns=['U', 'I']))
|
||
T_e.setText(str(int(params[1])) +
|
||
u"\u00B1" + str(int(error[1])))
|
||
I_n.setText(str(int(params[0])) +
|
||
u"\u00B1" + str(int(error[0])))
|
||
B_koef.setText(str('%.2f' % params[3]) +
|
||
u"\u00B1" + str('%.2f' % error[3]))
|
||
self.graph_vac_4.plot(
|
||
df['U'], df['I'], pen=pg.mkPen(color, width=1), name=file.rstrip('.csv'))
|
||
|
||
# Plot fitted curve
|
||
self.graph_vac_4.plot(self.U, self.I, symbol='o',
|
||
pen=None, symbolSize=7, symbolPen=color)
|
||
self.graph_vac_4.showGrid(x=True, y=True, alpha=1)
|
||
|
||
except FileNotFoundError:
|
||
self.error.setText(QtCore.QTime.currentTime().toString(
|
||
)+' - [Errno3]')
|
||
pass
|
||
else:
|
||
self.error.setText(QtCore.QTime.currentTime().toString(
|
||
)+' - [Errno0]')
|
||
|
||
|
||
if __name__ == "__main__":
|
||
|
||
app = QApplication(sys.argv)
|
||
|
||
win = Window()
|
||
|
||
win.show()
|
||
|
||
sys.exit(app.exec())
|