diff --git a/Automation/Main.py b/Automation/Main.py new file mode 100644 index 0000000..d6f3309 --- /dev/null +++ b/Automation/Main.py @@ -0,0 +1,460 @@ +import os +import sys +import csv +import numpy as np +import matplotlib.pyplot as plt +from scipy.optimize import curve_fit +import time +from PyQt6 import QtCore +from PyQt6.QtGui import QAction, QIcon, QPixmap +from PyQt6.QtWidgets import (QHBoxLayout, + QApplication, + QCheckBox, + QComboBox, + QDateEdit, + QDateTimeEdit, + QDial, + QDoubleSpinBox, + QFontComboBox, + QLabel, + QLCDNumber, + QLineEdit, + QMainWindow, + QProgressBar, + QPushButton, + QRadioButton, + QSlider, + QSpinBox, + QTimeEdit, + QVBoxLayout, + QWidget, + QTableWidget, + QGridLayout, + QMenu, + QTableWidgetItem, + QHeaderView, + QTextBrowser, + ) + + +class AbstractWindow(QMainWindow): + def __init__(self): + super().__init__() + self.create_actions() + self.create_menuBar() + + def create_menuBar(self): + menuBar = self.menuBar() + self.fileMenu = QMenu("&Файл", self) + menuBar.addMenu(self.fileMenu) + self.helpMenu = QMenu("&Помощь") + menuBar.addMenu(self.helpMenu) + self.fileMenu.addAction(self.exitAction) + self.helpMenu.addAction(self.helpContentAction) + self.helpMenu.addAction(self.aboutAction) + + def create_actions(self): + self.exitAction = QAction("&Выход", self) + self.exitAction.triggered.connect(self.exit_click) + self.helpContentAction = QAction("&Инструкция", self) + self.helpContentAction.triggered.connect(self.help_click) + self.aboutAction = QAction("&О программе", self) + self.aboutAction.triggered.connect(self.about_click) + + def exit_click(self): + self.close() + + def help_click(self): + print('Help') + # TODO + + def about_click(self): + print('About') + # TODO + + +class Start: + def __init__(self): + self.number = 0 + self.foldername = '' + if not QApplication.instance(): + self.app = QApplication(sys.argv) + else: + self.app = QApplication.instance() + self.window = StartWindow(self) + self.draw() + self.app.exec() + + def __del__(self): + print('destruct') + + def draw(self): + self.window.show() + + def change_number(self): + if self.number == 20: + self.window.close() + self.window = MainExperimentDataWindow(self) + if self.number == 21: + self.window.close() + self.window = MainExperimentChartWindow(self) + + if self.number == 10: + pass + self.draw() + + +class StartWindow(AbstractWindow): + def __init__(self, parent): + super().__init__() + + self.setWindowTitle('Эффект Холла в полупроводниках') + self.parent = parent + self.centralwidget = QWidget() + self.resize(1400, 800) + self.setCentralWidget(self.centralwidget) + self.lineEdit = QLineEdit(placeholderText='Введите фамилию') + self.lineEdit.returnPressed.connect(self.enter_name) + + self.flow = QPushButton('измерение потока') + self.flow.clicked.connect(self.flow_click) + self.flow.setEnabled(False) + self.main = QPushButton('основной эксперимент') + self.main.clicked.connect(self.main_click) + self.main.setEnabled(False) + + self.hbox_layout = QGridLayout(self.centralwidget) + self.hbox_layout.setRowStretch(1, 1) + self.hbox_layout.addWidget(self.lineEdit, 1, 0, 1, 2) + self.hbox_layout.addWidget(self.flow, 2, 0) + self.hbox_layout.addWidget(self.main, 2, 1) + + def flow_click(self): + self.parent.number = 10 + self.parent.change_number() + + def main_click(self): + self.parent.number = 20 + self.parent.change_number() + + def enter_name(self): + # make folder + self.parent.foldername = self.lineEdit.text() + self.parent.folder = os.path.join(os.getcwd(), self.parent.foldername) + if not os.path.exists(self.parent.folder): + os.mkdir(self.parent.folder) + + self.flow.setEnabled(True) + self.main.setEnabled(True) + self.lineEdit.setReadOnly(True) + + +class ThreadData(QtCore.QThread): + signal = QtCore.pyqtSignal(str) + + def __init__(self, parent): + QtCore.QThread.__init__(self) + self.running = False + self.parent = parent + + def run(self): + self.running = True + while self.running: + self.parent.no_data() + self.sleep(1) + + +class MainExperimentDataWindow(AbstractWindow): + def __init__(self, parent): + # TODO : clean code + super().__init__() + + self.setWindowTitle('Основной эксперимент. Получение данных') + self.start_time = round(time.time()*1000) + self.parent = parent + self.data_thread = ThreadData(self) + self.resize(1400, 800) + + # make masthead + self.dataname = 'data.csv' + head_1 = 'I_0,mA' + head_2 = 'U_34,mV' + head_3 = 't,s' + with open(os.path.join(self.parent.folder, self.dataname), 'w') as file: + wr = csv.writer(file) + wr.writerow([head_1, head_2, head_3]) + + self.centralwidget = QWidget() + self.setCentralWidget(self.centralwidget) + + self.start = QPushButton('Старт') + self.start.clicked.connect(self.start_clicked) + self.start.setEnabled(False) + + self.stop = QPushButton('Стоп') + self.stop.clicked.connect(self.stop_clicked) + self.stop.setEnabled(False) + + self.next = QPushButton(self) + self.next.setIcon(QIcon('arrow.png')) + self.next.setEnabled(False) + self.next.clicked.connect(self.next_clicked) + + grid_layout = QGridLayout(self.centralwidget) + + self.table = QTableWidget(self) # Create a self.table + self.table.setColumnCount(3) # Set three columns + self.table.setRowCount(0) + + self.table.setHorizontalHeaderLabels([head_1, head_2, head_3]) + + header = self.table.horizontalHeader() + header.setSectionResizeMode(0, QHeaderView.ResizeMode.Stretch) + header.setSectionResizeMode(1, QHeaderView.ResizeMode.Stretch) + header.setSectionResizeMode(2, QHeaderView.ResizeMode.Stretch) + self.table.setItem(0, 0, QTableWidgetItem("Text in column 1")) + self.table.setItem(0, 1, QTableWidgetItem("Text in column 2")) + self.table.setItem(0, 2, QTableWidgetItem("Text in column 3")) + self.table.resizeColumnsToContents() + + self.lineEdit = QLineEdit(placeholderText='Введите что-то') + self.lineEdit.returnPressed.connect(self.enter_smth) + + # Adding the table to the grid + grid_layout.addWidget(self.table, 0, 0, -1, 1) + grid_layout.addWidget(self.lineEdit, 0, 2, -1, -1) + grid_layout.addWidget(self.next, 2, 2, -1, -1) + grid_layout.addWidget(self.start, 1, 2) + grid_layout.addWidget(self.stop, 1, 3) + + self.parent.draw() + + def enter_smth(self): + # TODO + self.parent.smth = self.lineEdit.text() + self.start.setEnabled(True) + self.lineEdit.setReadOnly(True) + + def start_clicked(self): + self.stop.setEnabled(True) + self.start.setEnabled(False) + if not self.data_thread.isRunning(): + self.data_thread.start() + + def no_data(self): + current_time = round(time.time()*1000) + v = str((current_time-self.start_time)/60) + a = v + t = str(current_time-self.start_time) + with open(os.path.join(self.parent.folder, self.dataname), 'a') as file: + wr = csv.writer(file) + wr.writerow([v, a, str(current_time-self.start_time)]) + self.table.insertRow(self.table.rowCount()) + self.table.setItem(self.table.rowCount()-1, 0, QTableWidgetItem(v)) + self.table.setItem(self.table.rowCount()-1, 1, QTableWidgetItem(a)) + self.table.setItem(self.table.rowCount()-1, 2, QTableWidgetItem(t)) + + def stop_clicked(self): + self.data_thread.running = False + self.stop.setEnabled(False) + self.next.setEnabled(True) + + def next_clicked(self): + self.parent.number = 21 + self.parent.change_number() + + def take_data(self): + # measure voltage and current + volt_name = os.path.join('/dev', 'usbtmc1') + f_volt = open(volt_name, 'w') + f_volt.write('Measure:Voltage:DC?\n') + f_volt.close() + amp_name = os.path.join('/dev', 'usbtmc2') + f_amp = open(amp_name, 'w') + f_amp.write('Measure:Current:DC?\n') + f_amp.close() + + f_volt = open(volt_name, 'r') + v = f_volt.read(15) + f_amp = open(amp_name, 'r') + a = f_amp.read(15) + f_volt.close() + f_amp.close() + current_time = round(time.time()*1000) + with open(os.path.join(self.parent.folder, self.dataname), 'a') as file: + wr = csv.writer(file) + wr.writerow([v, a, str(current_time-self.start_time)]) + + +class MainExperimentChartWindow(AbstractWindow): + def __init__(self, parent): + super().__init__() + + self.setWindowTitle('Основной эксперимент. Обработка данных') + self.parent = parent + self.resize(1400, 800) + + self.centralwidget = QWidget() + self.setCentralWidget(self.centralwidget) + + pixmap = QPixmap('arrow.png') + self.label = QLabel(self) + self.label.setPixmap(pixmap) + self.label.resize(pixmap.width(), pixmap.height()) + + self.text = QTextBrowser() + self.text.setText('text') + + self.hbox_layout = QGridLayout(self.centralwidget) + self.hbox_layout.addWidget(self.label, 0, 0) + self.hbox_layout.addWidget(self.text, 0, 1) + + +class Data: + def __init__(self, x, y, xlabel, ylabel, caption, xerr, yerr, through_0, + data_filename, color=None, centering=None, size=15, + coefficient=[0.9, 1.1]): + self.x = x + self.y = y + self.xlabel = xlabel + self.ylabel = ylabel + self.caption = caption + if type(yerr) == float or type(yerr) == int: + self.yerr = [yerr for _ in self.y] + else: + self.yerr = yerr + if type(xerr) == float or type(xerr) == int: + self.xerr = [xerr for _ in self.x] + else: + self.xerr = xerr + self.through_0 = through_0 + if not self.color: + self.color = ['limegreen', 'indigo'] + else: + self.color = color + self.centering = centering + self.size = size + self.coefficient = coefficient + self.data_filename=data_filename + + def read_csv(self): + with open(self.data_filename) as file: + reader = list(csv.reader(file, delimiter=';', + quotechar=',', quoting=csv.QUOTE_MINIMAL)) + data = np.array(reader) + data = np.transpose(data) + dic = {} + for i in range(len(data)): + dic[data[i][0]] = np.array(data[i][1:]).astype(np.float) + data = dic + self.data=data + + def make_point_grafic(self): + + if self.xerr[1] != 0 or self.yerr[1] != 0: + plt.errorbar(self.x, self.y, yerr=self.yerr, xerr=self.xerr, linewidth=4, + linestyle='', label=self.caption, color=self.color, + ecolor=self.color, elinewidth=1, capsize=3.4, + capthick=1.4) + else: + plt.scatter(self.x, self.y, linewidth=0.005, label=self.caption, + color=self.color, edgecolor='black', s=self.size) + + if not self.centering: + plt.xlabel(self.xlabel) + plt.ylabel(self.ylabel) + else: + self.ax = plt.gca() + self.ax.spines['top'].set_visible(False) + self.ax.spines['right'].set_visible(False) + self.ax.spines['bottom'].set_position('zero') + self.ax.spines['left'].set_position('zero') + self.ax.set_xlabel(self.ylabel, labelpad=-180, fontsize=14) + self.ax.set_ylabel(self.xlabel, labelpad=-260, + rotation=0, fontsize=14) + + def make_line_grafic(self, k, b,): + if min(self.x) > 0: + xmin = min(self.x)*self.coefficient[0] + else: + xmin = min(self.x)*self.coefficient[1] + + if max(self.x) > 0: + xmax = max(self.x)*self.coefficient[1] + else: + xmax = max(self.x)*self.coefficient[0] + x = np.arange(xmin, xmax, (xmax-xmin)/10000) + plt.plot(x, k*x+b, label=self.caption, linewidth=2.4, + linestyle='-') + + def make_graffic(self, named_by_points=True): + + if named_by_points: + cap_point = self.caption + line_point = None + else: + line_point = self.caption + cap_point = None + self.make_point_grafic() + k, b, sigma = self.approx() + sigma[0] = abs(k*((sigma[0]/k)**2+(np.mean(self.yerr)/np.mean(self.y))**2 + + (np.mean(self.xerr)/np.mean(self.x))**2)**0.5) + if (b != 0): + sigma[1] = abs(b*((sigma[1]/b)**2+(np.mean(self.yerr)/np.mean(self.y))**2 + + (np.mean(self.xerr)/np.mean(self.x))**2)**0.5) + else: + sigma[1] = 0 + + self.make_line_grafic() + plt.legend() + return k, b, sigma + + def approx(self): + if self.yerr[0] != 0: + sigma_y = [1/i**2 for i in self.yerr] + else: + sigma_y = np.array([1 for _ in self.yerr]) + if self.through_0 == 0: + def f(x, k): + return k*x + k, sigma = curve_fit(f, xdata=self.x, ydata=self.y, sigma=sigma_y) + sigma = np.sqrt(np.diag(sigma)) + return k[0], 0, [sigma[0], 0] + else: + def f(x, k, b): + return x*k + b + k, sigma = curve_fit(f, xdata=self.x, ydata=self.y, sigma=sigma_y) + sigma_b = np.sqrt(sigma[1][1]) + b = k[1] + k = k[0] + sigma = np.sqrt(sigma[0][0]) + return k, b, [sigma, sigma_b] + + +start = Start() + +# ============================================================================= +# import time +# import serial +# +# ser=serial.Serial( +# port='/dev/ttyUSB0', +# baudrate=9600, +# timeout=1 +# ) +# ser.isOpen() +# +# msg='SYSTem:REMote\n' +# ser.write(msg.encode('ascii')) +# +# while 1: +# +# msg='Read?\n' +# ser.write(msg.encode('ascii')) +# time.sleep(1) +# +# bytesToRead=ser.inWaiting() +# data=ser.read(bytesToRead) +# print(data) +# +# ============================================================================= diff --git a/Automation/arrow.png b/Automation/arrow.png new file mode 100644 index 0000000..b72358d Binary files /dev/null and b/Automation/arrow.png differ diff --git a/Automation/j/data.csv b/Automation/j/data.csv new file mode 100644 index 0000000..f19e2b9 --- /dev/null +++ b/Automation/j/data.csv @@ -0,0 +1,3 @@ +"I_0,mA","U_34,mV","t,s" +68.26666666666667,68.26666666666667,4096 +84.95,84.95,5097