add flow window
This commit is contained in:
nazarchuk.as
parent
d879a6a6da
commit
059abdc5e6
47
Automation/Abstract_window.py
Normal file
47
Automation/Abstract_window.py
Normal file
|
|
@ -0,0 +1,47 @@
|
|||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Created on Fri Nov 11 19:52:27 2022
|
||||
|
||||
@author: anna
|
||||
"""
|
||||
|
||||
from PyQt6.QtWidgets import (QMainWindow,
|
||||
QMenu)
|
||||
|
||||
from PyQt6.QtGui import QAction
|
||||
|
||||
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
|
||||
149
Automation/Analisis_data.py
Normal file
149
Automation/Analisis_data.py
Normal file
|
|
@ -0,0 +1,149 @@
|
|||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Created on Fri Nov 11 19:49:57 2022
|
||||
|
||||
@author: anna
|
||||
"""
|
||||
import csv
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
from scipy.optimize import curve_fit
|
||||
|
||||
class Data:
|
||||
def __init__(self, x=[], y=[], xlabel='', ylabel='', caption='', xerr=None,
|
||||
yerr=None, through_0=0, data_filename='', color=None,
|
||||
centering=None, size=15, coefficient=[0.9, 1.1], saving=None):
|
||||
self.x = x
|
||||
self.y = y
|
||||
self.xlabel = xlabel
|
||||
self.ylabel = ylabel
|
||||
self.caption = caption
|
||||
self.xerr=xerr
|
||||
self.yerr=yerr
|
||||
self.make_errors(xerr, yerr)
|
||||
self.through_0 = through_0
|
||||
if not color:
|
||||
self.color = ['limegreen', 'indigo']
|
||||
else:
|
||||
self.color = color
|
||||
self.centering = centering
|
||||
self.size = size
|
||||
self.coefficient = coefficient
|
||||
self.data_filename=data_filename
|
||||
self.saving=saving
|
||||
|
||||
def make_errors(self, xerr=None, yerr=None):
|
||||
if not xerr: xerr=self.xerr
|
||||
if not yerr: yerr=self.yerr
|
||||
if not xerr: xerr=0
|
||||
if not yerr: yerr=0
|
||||
if not yerr or type(yerr) == float or type(yerr) == int:
|
||||
self.yerr = [yerr for _ in self.y]
|
||||
else:
|
||||
self.yerr = yerr
|
||||
if not xerr or type(xerr) == float or type(xerr) == int:
|
||||
self.xerr = [xerr for _ in self.x]
|
||||
else:
|
||||
self.xerr = xerr
|
||||
|
||||
def read_csv(self):
|
||||
with open(self.data_filename) as file:
|
||||
reader = list(csv.reader(file))
|
||||
data = np.array(reader)
|
||||
dic=[]
|
||||
for i in range(len(data[0])):
|
||||
micro_data = []
|
||||
for j in range(len(data)):
|
||||
micro_data.append(data[j][i])
|
||||
dic.append(micro_data)
|
||||
data=dic
|
||||
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):
|
||||
self.make_errors()
|
||||
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[0],
|
||||
ecolor=self.color[0], elinewidth=1, capsize=3.4,
|
||||
capthick=1.4)
|
||||
else:
|
||||
plt.scatter(self.x, self.y, linewidth=0.005, label=self.caption,
|
||||
color=self.color[0], 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='-', color=self.color[1])
|
||||
|
||||
def make_grafic(self, named_by_points=True):
|
||||
self.make_errors()
|
||||
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(k, b)
|
||||
plt.legend()
|
||||
if self.saving:
|
||||
plt.savefig(self.saving)
|
||||
return k, b, sigma
|
||||
|
||||
def approx(self):
|
||||
if self.yerr[0] != 0:
|
||||
sigma_y = [1/i**2 for i in self.y]
|
||||
else:
|
||||
sigma_y = np.array([1 for _ in self.y])
|
||||
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]
|
||||
|
|
@ -1,11 +1,7 @@
|
|||
import os
|
||||
import sys
|
||||
import csv
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
from scipy.optimize import curve_fit
|
||||
import time
|
||||
import pandas as pd
|
||||
from PyQt6 import QtCore
|
||||
from PyQt6.QtGui import QAction, QIcon, QPixmap
|
||||
from PyQt6.QtWidgets import (QHBoxLayout,
|
||||
|
|
@ -36,42 +32,8 @@ from PyQt6.QtWidgets import (QHBoxLayout,
|
|||
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
|
||||
from Abstract_window import AbstractWindow
|
||||
from Main_experiment_window import *
|
||||
|
||||
|
||||
class Start:
|
||||
|
|
@ -99,9 +61,12 @@ class Start:
|
|||
if self.number == 21:
|
||||
self.window.close()
|
||||
self.window = MainExperimentChartWindow(self)
|
||||
|
||||
if self.number == 10:
|
||||
pass
|
||||
self.window.close()
|
||||
self.window = FlowWindow(self)
|
||||
if self.number == 0:
|
||||
self.window.close()
|
||||
self.window = StartWindow(self)
|
||||
self.draw()
|
||||
|
||||
|
||||
|
|
@ -150,58 +115,31 @@ class StartWindow(AbstractWindow):
|
|||
self.lineEdit.setReadOnly(True)
|
||||
|
||||
|
||||
class ThreadData(QtCore.QThread):
|
||||
signal = QtCore.pyqtSignal(str)
|
||||
|
||||
class FlowWindow(AbstractWindow):
|
||||
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.parent.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.parent.dataname), 'w') as file:
|
||||
wr = csv.writer(file)
|
||||
wr.writerow([head_1, head_2, head_3])
|
||||
|
||||
self.parent = parent
|
||||
self.centralwidget = QWidget()
|
||||
self.resize(1400, 800)
|
||||
self.setCentralWidget(self.centralwidget)
|
||||
grid_layout = QGridLayout(self.centralwidget)
|
||||
|
||||
self.start = QPushButton('Старт')
|
||||
self.start.clicked.connect(self.start_clicked)
|
||||
self.start.setEnabled(False)
|
||||
self.start.setEnabled(True)
|
||||
|
||||
self.stop = QPushButton('Стоп')
|
||||
self.stop.clicked.connect(self.stop_clicked)
|
||||
self.stop.setEnabled(False)
|
||||
self.lineEdit = QLineEdit(placeholderText='Индукция B, мТл')
|
||||
self.lineEdit.returnPressed.connect(self.enter_value)
|
||||
self.lineEdit.setReadOnly(True)
|
||||
|
||||
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.parent.flow_dataname = 'Induction_data.csv'
|
||||
head_1 = 'B,mTl'
|
||||
head_2 = 'U,mV'
|
||||
head_3 = 't,s'
|
||||
with open(os.path.join(self.parent.folder, self.parent.flow_dataname), 'w') as file:
|
||||
wr = csv.writer(file)
|
||||
wr.writerow([head_1, head_2, head_3])
|
||||
|
||||
self.table = QTableWidget(self) # Create a self.table
|
||||
self.table.setColumnCount(3) # Set three columns
|
||||
|
|
@ -217,249 +155,49 @@ class MainExperimentDataWindow(AbstractWindow):
|
|||
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.parent.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)
|
||||
|
||||
self.next = QPushButton(self)
|
||||
self.next.setIcon(QIcon('arrow.png'))
|
||||
self.next.setEnabled(False)
|
||||
self.next.clicked.connect(self.next_clicked)
|
||||
|
||||
grid_layout.addWidget(self.start, 0, 0)
|
||||
grid_layout.addWidget(self.lineEdit, 1, 0)
|
||||
grid_layout.addWidget(self.table, 1, 3)
|
||||
grid_layout.addWidget(self.next, 2, 0)
|
||||
|
||||
|
||||
def next_clicked(self):
|
||||
self.parent.number = 21
|
||||
self.parent.number = 20
|
||||
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()
|
||||
def start_clicked(self):
|
||||
self.start.setEnabled(False)
|
||||
self.start_time = round(time.time()*1000)
|
||||
self.lineEdit.setReadOnly(False)
|
||||
self.next.setEnabled(True)
|
||||
|
||||
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()
|
||||
def enter_value(self):
|
||||
self.lineEdit.setReadOnly(True)
|
||||
self.get_data()
|
||||
self.lineEdit.clear()
|
||||
self.lineEdit.setReadOnly(False)
|
||||
|
||||
def get_data(self):
|
||||
current_time = round(time.time()*1000)
|
||||
with open(os.path.join(self.parent.folder, self.parent.dataname), 'a') as file:
|
||||
b = self.lineEdit.text()
|
||||
u = str((current_time-self.start_time)/60)
|
||||
t = str(current_time-self.start_time)
|
||||
with open(os.path.join(self.parent.folder, self.parent.flow_dataname), 'a') as file:
|
||||
wr = csv.writer(file)
|
||||
wr.writerow([v, a, str(current_time-self.start_time)])
|
||||
wr.writerow([b, u, str(t)])
|
||||
self.table.insertRow(self.table.rowCount())
|
||||
self.table.setItem(self.table.rowCount()-1, 0, QTableWidgetItem(b))
|
||||
self.table.setItem(self.table.rowCount()-1, 1, QTableWidgetItem(u))
|
||||
self.table.setItem(self.table.rowCount()-1, 2, QTableWidgetItem(t))
|
||||
|
||||
|
||||
class MainExperimentChartWindow(AbstractWindow):
|
||||
def __init__(self, parent):
|
||||
super().__init__()
|
||||
|
||||
self.parent = parent
|
||||
self.parent.chartname = 'Chart'
|
||||
self.parent.data = Data(data_filename=os.path.join(self.parent.folder, self.parent.dataname),
|
||||
saving=os.path.join(self.parent.folder, self.parent.chartname))
|
||||
self.parent.data.read_csv()
|
||||
self.parent.data.x=self.parent.data.data['I_0,mA']
|
||||
self.parent.data.y=self.parent.data.data['U_34,mV']
|
||||
self.parent.data.make_grafic()
|
||||
|
||||
self.setWindowTitle('Основной эксперимент. Обработка данных')
|
||||
self.resize(1400, 800)
|
||||
|
||||
self.centralwidget = QWidget()
|
||||
self.setCentralWidget(self.centralwidget)
|
||||
|
||||
pixmap = QPixmap(os.path.join(self.parent.folder, self.parent.chartname))
|
||||
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=None,
|
||||
yerr=None, through_0=0, data_filename='', color=None,
|
||||
centering=None, size=15, coefficient=[0.9, 1.1], saving=None):
|
||||
self.x = x
|
||||
self.y = y
|
||||
self.xlabel = xlabel
|
||||
self.ylabel = ylabel
|
||||
self.caption = caption
|
||||
self.xerr=xerr
|
||||
self.yerr=yerr
|
||||
self.make_errors(xerr, yerr)
|
||||
self.through_0 = through_0
|
||||
if not color:
|
||||
self.color = ['limegreen', 'indigo']
|
||||
else:
|
||||
self.color = color
|
||||
self.centering = centering
|
||||
self.size = size
|
||||
self.coefficient = coefficient
|
||||
self.data_filename=data_filename
|
||||
self.saving=saving
|
||||
|
||||
def make_errors(self, xerr=None, yerr=None):
|
||||
if not xerr: xerr=self.xerr
|
||||
if not yerr: yerr=self.yerr
|
||||
if not xerr: xerr=0
|
||||
if not yerr: yerr=0
|
||||
if not yerr or type(yerr) == float or type(yerr) == int:
|
||||
self.yerr = [yerr for _ in self.y]
|
||||
else:
|
||||
self.yerr = yerr
|
||||
if not xerr or type(xerr) == float or type(xerr) == int:
|
||||
self.xerr = [xerr for _ in self.x]
|
||||
else:
|
||||
self.xerr = xerr
|
||||
|
||||
def read_csv(self):
|
||||
with open(self.data_filename) as file:
|
||||
reader = list(csv.reader(file))
|
||||
data = np.array(reader)
|
||||
dic=[]
|
||||
for i in range(len(data[0])):
|
||||
micro_data = []
|
||||
for j in range(len(data)):
|
||||
micro_data.append(data[j][i])
|
||||
dic.append(micro_data)
|
||||
data=dic
|
||||
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):
|
||||
self.make_errors()
|
||||
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[0],
|
||||
ecolor=self.color[0], elinewidth=1, capsize=3.4,
|
||||
capthick=1.4)
|
||||
else:
|
||||
plt.scatter(self.x, self.y, linewidth=0.005, label=self.caption,
|
||||
color=self.color[0], 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='-', color=self.color[1])
|
||||
|
||||
def make_grafic(self, named_by_points=True):
|
||||
self.make_errors()
|
||||
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(k, b)
|
||||
plt.legend()
|
||||
if self.saving:
|
||||
plt.savefig(self.saving)
|
||||
return k, b, sigma
|
||||
|
||||
def approx(self):
|
||||
if self.yerr[0] != 0:
|
||||
sigma_y = [1/i**2 for i in self.y]
|
||||
else:
|
||||
sigma_y = np.array([1 for _ in self.y])
|
||||
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()
|
||||
|
||||
|
|
|
|||
211
Automation/Main_experiment_window.py
Normal file
211
Automation/Main_experiment_window.py
Normal file
|
|
@ -0,0 +1,211 @@
|
|||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Created on Fri Nov 11 20:00:45 2022
|
||||
|
||||
@author: anna
|
||||
"""
|
||||
import os
|
||||
import csv
|
||||
import time
|
||||
from PyQt6 import QtCore
|
||||
from PyQt6.QtGui import 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,
|
||||
)
|
||||
from Analisis_data import Data
|
||||
from Abstract_window import AbstractWindow
|
||||
|
||||
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.parent.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.parent.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_time = round(time.time()*1000)
|
||||
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.parent.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.parent.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.parent = parent
|
||||
self.parent.chartname = 'Chart'
|
||||
self.parent.data = Data(data_filename=os.path.join(self.parent.folder, self.parent.dataname),
|
||||
saving=os.path.join(self.parent.folder, self.parent.chartname))
|
||||
self.parent.data.read_csv()
|
||||
self.parent.data.x=self.parent.data.data['I_0,mA']
|
||||
self.parent.data.y=self.parent.data.data['U_34,mV']
|
||||
self.parent.data.make_grafic()
|
||||
|
||||
self.setWindowTitle('Основной эксперимент. Обработка данных')
|
||||
self.resize(1400, 800)
|
||||
|
||||
self.centralwidget = QWidget()
|
||||
self.setCentralWidget(self.centralwidget)
|
||||
|
||||
pixmap = QPixmap(os.path.join(self.parent.folder, self.parent.chartname))
|
||||
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)
|
||||
|
||||
BIN
Automation/__pycache__/Abstract_window.cpython-39.pyc
Normal file
BIN
Automation/__pycache__/Abstract_window.cpython-39.pyc
Normal file
Binary file not shown.
BIN
Automation/__pycache__/Analisis_data.cpython-39.pyc
Normal file
BIN
Automation/__pycache__/Analisis_data.cpython-39.pyc
Normal file
Binary file not shown.
BIN
Automation/__pycache__/Main_experiment_window.cpython-39.pyc
Normal file
BIN
Automation/__pycache__/Main_experiment_window.cpython-39.pyc
Normal file
Binary file not shown.
Binary file not shown.
|
Before Width: | Height: | Size: 11 KiB After Width: | Height: | Size: 10 KiB |
1
Automation/j/Induction_data.csv
Normal file
1
Automation/j/Induction_data.csv
Normal file
|
|
@ -0,0 +1 @@
|
|||
"B,mTl","U,mV","t,s"
|
||||
|
|
|
@ -1,5 +1,2 @@
|
|||
"I_0,mA","U_34,mV","t,s"
|
||||
52.78333333333333,52.78333333333333,3167
|
||||
69.46666666666667,69.46666666666667,4168
|
||||
86.15,86.15,5169
|
||||
102.83333333333333,102.83333333333333,6170
|
||||
0.016666666666666666,0.016666666666666666,1
|
||||
|
|
|
|||
|
Loading…
Reference in New Issue
Block a user