mirror of
https://github.com/dptech-corp/Uni-Lab-OS.git
synced 2025-12-14 13:14:39 +00:00
334 lines
15 KiB
Python
334 lines
15 KiB
Python
import tkinter as tk
|
||
from tkinter import ttk # 使用 ttk 替换 tk 控件
|
||
from tkinter import messagebox
|
||
from tkinter.font import Font
|
||
from threading import Thread
|
||
from ttkthemes import ThemedTk
|
||
import time
|
||
import matplotlib.pyplot as plt
|
||
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
|
||
import clr # pythonnet library
|
||
import sys
|
||
import threading
|
||
import datetime
|
||
|
||
jifenshijian = 10 #积分时间
|
||
shuaxinshijian = 0.01 #刷新时间
|
||
zihaodaxiao = 16 #字号大小
|
||
ymax = 70000
|
||
ymin = -2000
|
||
|
||
# 加载DLL
|
||
dll_path = "C:\\auto\\UV_spec\\idea-sdk 3.0.9\\idea-sdk.UPI\\IdeaOptics.dll"
|
||
clr.AddReference(dll_path)
|
||
from IdeaOptics import Wrapper
|
||
|
||
# 初始化Wrapper对象和光谱仪
|
||
wrapper = Wrapper()
|
||
number_of_spectrometers = wrapper.OpenAllSpectrometers()
|
||
if number_of_spectrometers > 0:
|
||
spectrometer_index = 0 # 假设使用第一个光谱仪
|
||
integration_time = jifenshijian # 设置积分时间
|
||
wrapper.setIntegrationTime(spectrometer_index, integration_time)
|
||
|
||
class App:
|
||
def __init__(self, root):
|
||
self.root = root
|
||
self.root.title("光谱测试")
|
||
self.is_continuous = False
|
||
self.root.protocol("WM_DELETE_WINDOW", self.on_closing)
|
||
self.stop_event = threading.Event() # 使用Event代替布尔标志
|
||
self.continuous_thread = None # 在这里初始化
|
||
self.background_spectrum = None
|
||
self.correct_background = False
|
||
self.test_count = 0
|
||
self.background_count = 0
|
||
|
||
self.source_spectrum = None # 初始化光源强度变量
|
||
self.transmission_mode = False # 初始化透射模式标志
|
||
|
||
self.data_ready = False
|
||
|
||
# 使用 grid 布局
|
||
self.root.columnconfigure(0, weight=1)
|
||
self.root.rowconfigure(0, weight=1)
|
||
self.root.rowconfigure(1, weight=1)
|
||
self.root.rowconfigure(2, weight=1)
|
||
self.root.rowconfigure(3, weight=1)
|
||
|
||
self.current_ylim = [-100, 1000] # 初始化y轴范围
|
||
|
||
# 创建一个 Style 对象
|
||
style = ttk.Style()
|
||
|
||
# 定义一个新的样式
|
||
style.configure('Custom.TButton', font=('Helvetica', zihaodaxiao, 'bold'), foreground='white')
|
||
|
||
# 创建滑动条和按钮的容器 Frame
|
||
control_frame = ttk.Frame(self.root)
|
||
control_frame.grid(row=0, column=0, sticky="ew")
|
||
|
||
# 创建一个滑动条来选择平滑次数
|
||
self.boxcar_width_slider = tk.Scale(control_frame, from_=0, to=10, orient=tk.HORIZONTAL, length=300, label="平滑次数", font=("Helvetica", zihaodaxiao, 'bold'))
|
||
self.boxcar_width_slider.grid(row=0, column=0, padx=10, pady=10)
|
||
|
||
# 创建一个滑动条来选择平均次数
|
||
self.scans_to_average_slider = tk.Scale(control_frame, from_=1, to=10, orient=tk.HORIZONTAL, length=300, label="平均次数", font=("Helvetica", zihaodaxiao, 'bold'))
|
||
self.scans_to_average_slider.grid(row=0, column=1, padx=10, pady=10)
|
||
|
||
# 调整 Scale 控件的外观
|
||
self.boxcar_width_slider.config(bg='grey', fg='white')
|
||
self.scans_to_average_slider.config(bg='grey', fg='white')
|
||
|
||
# 字体设置
|
||
entry_font = ('Helvetica', zihaodaxiao, 'bold')
|
||
|
||
# 添加输入框的容器 Frame
|
||
entry_frame = ttk.Frame(self.root)
|
||
entry_frame.grid(row=1, column=0, sticky="ew")
|
||
|
||
# 创建并放置"积分时间(ms)"输入框
|
||
ttk.Label(entry_frame, text="积分时间(ms):", font=entry_font).grid(row=0, column=0, padx=10, pady=10)
|
||
self.integration_time_entry = ttk.Entry(entry_frame, font=entry_font)
|
||
self.integration_time_entry.grid(row=0, column=1, padx=10, pady=10)
|
||
self.integration_time_entry.insert(0, "10") # 设置默认值
|
||
|
||
# 创建并放置"刷新间隔(s)"输入框
|
||
ttk.Label(entry_frame, text="刷新间隔(s):", font=entry_font).grid(row=0, column=2, padx=10, pady=10)
|
||
self.refresh_interval_entry = ttk.Entry(entry_frame, font=entry_font)
|
||
self.refresh_interval_entry.grid(row=0, column=3, padx=10, pady=10)
|
||
self.refresh_interval_entry.insert(0, "0.01") # 设置默认值
|
||
|
||
# 创建按钮的容器 Frame
|
||
button_frame = ttk.Frame(self.root)
|
||
button_frame.grid(row=2, column=0, sticky="ew")
|
||
|
||
# 创建并放置按钮
|
||
ttk.Button(button_frame, text="测试一下", style='Custom.TButton', command=self.single_test).grid(row=0, column=0, padx=10, pady=10)
|
||
ttk.Button(button_frame, text="连续测试", style='Custom.TButton', command=self.start_continuous_test).grid(row=0, column=1, padx=10, pady=10)
|
||
ttk.Button(button_frame, text="停止测试", style='Custom.TButton', command=self.stop_continuous_test).grid(row=0, column=2, padx=10, pady=10)
|
||
|
||
# 创建背景相关按钮的容器 Frame
|
||
background_frame = ttk.Frame(self.root)
|
||
background_frame.grid(row=3, column=0, sticky="ew")
|
||
|
||
# 创建并放置“采集背景”按钮
|
||
self.collect_background_button = ttk.Button(background_frame, text="采集背景", style='Custom.TButton', command=self.collect_background)
|
||
self.collect_background_button.grid(row=0, column=0, padx=10, pady=10)
|
||
|
||
# 创建并放置“背景校正”按钮
|
||
self.correct_background_button = ttk.Button(background_frame, text="背景校正", style='Custom.TButton', command=self.toggle_background_correction)
|
||
self.correct_background_button.grid(row=0, column=1, padx=10, pady=10)
|
||
|
||
# 创建“光源采集”按钮
|
||
ttk.Button(background_frame, text="光源采集", style='Custom.TButton', command=self.collect_source).grid(row=0, column=2, padx=10, pady=10)
|
||
|
||
# 创建“透射模式”按钮
|
||
self.transmission_button = ttk.Button(background_frame, text="透射模式", style='Custom.TButton', command=self.toggle_transmission_mode)
|
||
self.transmission_button.grid(row=0, column=3, padx=10, pady=10)
|
||
|
||
# 创建 matplotlib 画布
|
||
plt.style.use('ggplot') # 使用预定义的样式，如 'ggplot'
|
||
self.figure, self.ax = plt.subplots(figsize=(10, 8))
|
||
self.canvas = FigureCanvasTkAgg(self.figure, self.root)
|
||
self.canvas_widget = self.canvas.get_tk_widget()
|
||
self.canvas_widget.grid(row=3, column=0, sticky="ew")
|
||
|
||
# 使用 grid 布局来放置 matplotlib 画布
|
||
self.canvas_widget = self.canvas.get_tk_widget()
|
||
self.canvas_widget.grid(row=4, column=0, sticky="ew")
|
||
|
||
# 创建文件名并打开文件
|
||
start_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
|
||
self.data_file = open(f"C:\\auto\\UV_spec\\data\\{start_time}.txt", "w")
|
||
|
||
def get_spectrum_data(self):
|
||
# 获取波长和光谱值
|
||
pixels = wrapper.getNumberOfPixels(spectrometer_index)
|
||
spectrum = wrapper.getSpectrum(spectrometer_index)
|
||
wavelengths = wrapper.getWavelengths(spectrometer_index)
|
||
|
||
# 转换.NET数组到Python列表
|
||
spectrum_list = [spectrum[i] for i in range(pixels)]
|
||
wavelengths_list = [wavelengths[i] for i in range(pixels)]
|
||
|
||
self.data_ready = True
|
||
return wavelengths_list, spectrum_list
|
||
|
||
def collect_source(self):
|
||
# 采集光源强度数据
|
||
wavelengths, self.source_spectrum = self.get_spectrum_data()
|
||
conditions = f"jifenshijian = {jifenshijian} shuaxinshijian = {shuaxinshijian} zihaodaxiao = {zihaodaxiao}"
|
||
self.write_data_to_file("source", 1, conditions, self.source_spectrum)
|
||
self.update_plot(wavelengths, self.source_spectrum)
|
||
|
||
def toggle_transmission_mode(self):
|
||
# 切换透射模式
|
||
self.transmission_mode = not self.transmission_mode
|
||
self.transmission_button.config(text="正在透射" if self.transmission_mode else "透射模式")
|
||
|
||
def calculate_transmission(self, spectrum):
|
||
# 计算透射率
|
||
transmission = []
|
||
for s, b, src in zip(spectrum, self.background_spectrum, self.source_spectrum):
|
||
denominator = max(src - b, 0.1)
|
||
trans_value = (s - b) / denominator * 100
|
||
trans_value = max(0, min(trans_value, 100))
|
||
transmission.append(trans_value)
|
||
return transmission
|
||
|
||
def update_plot(self, wavelengths, spectrum, plot_type='spectrum'):
|
||
|
||
if not self.data_ready:
|
||
return
|
||
|
||
self.ax.clear()
|
||
|
||
if plot_type == 'transmission':
|
||
# 透射率模式的绘图设置
|
||
self.ax.plot(wavelengths, spectrum, label='Transmission (%)')
|
||
self.ax.set_ylim(-10, 110) # 设置y轴范围为0%到100%
|
||
self.ax.set_ylabel('Transmission (%)', fontname='Arial', fontsize=zihaodaxiao)
|
||
else:
|
||
# 普通光谱模式的绘图设置
|
||
self.ax.plot(wavelengths, spectrum)
|
||
self.ax.set_ylim(self.current_ylim) # 使用当前y轴范围
|
||
|
||
# 计算新的最大值和最小值
|
||
new_min, new_max = min(spectrum), max(spectrum)
|
||
|
||
# 检查新的最大值或最小值是否超过当前y轴范围
|
||
while new_min < self.current_ylim[0] or new_max > self.current_ylim[1]:
|
||
# 扩大y轴范围
|
||
self.current_ylim = [self.current_ylim[0] * 2, self.current_ylim[1] * 2]
|
||
|
||
# 确保新的y轴范围不超过最大限制
|
||
if self.current_ylim[0] < ymin:
|
||
self.current_ylim[0] = ymin
|
||
if self.current_ylim[1] > ymax:
|
||
self.current_ylim[1] = ymax
|
||
break
|
||
|
||
self.ax.set_ylabel('Intensity', fontname='Arial', fontsize=zihaodaxiao)
|
||
|
||
self.ax.set_xlabel('Wavelength (nm)', fontname='Arial', fontsize=zihaodaxiao)
|
||
self.ax.set_title('Spectrum', fontname='Arial', fontsize=zihaodaxiao)
|
||
|
||
self.canvas.draw()
|
||
|
||
self.data_ready = False
|
||
|
||
def draw_plot(self):
|
||
self.canvas.draw()
|
||
|
||
def write_data_to_file(self, test_type, test_number, conditions, spectrum):
|
||
data_str = " ".join(map(str, spectrum))
|
||
self.data_file.write(f"{test_type}{test_number}\n{conditions}\n{data_str}\n\n")
|
||
self.data_file.flush()
|
||
|
||
def collect_background(self):
|
||
# 设置平滑次数
|
||
boxcar_width = self.boxcar_width_slider.get()
|
||
wrapper.setBoxcarWidth(spectrometer_index, boxcar_width)
|
||
|
||
# 设置平均次数
|
||
scans_to_average = self.scans_to_average_slider.get()
|
||
wrapper.setScansToAverage(spectrometer_index, scans_to_average)
|
||
|
||
# 采集背景数据
|
||
wavelengths, self.background_spectrum = self.get_spectrum_data()
|
||
conditions = f"jifenshijian = {jifenshijian} shuaxinshijian = {shuaxinshijian} zihaodaxiao = {zihaodaxiao} pinghuacishu = {self.boxcar_width_slider.get()} pingjuncishu = {self.scans_to_average_slider.get()}"
|
||
self.background_count += 1
|
||
self.write_data_to_file("background", self.background_count, conditions, self.background_spectrum)
|
||
self.update_plot(wavelengths, self.background_spectrum)
|
||
|
||
def toggle_background_correction(self):
|
||
self.correct_background = not self.correct_background
|
||
self.correct_background_button.config(text="正在校正" if self.correct_background else "背景校正")
|
||
|
||
def apply_background_correction(self, spectrum):
|
||
if self.background_spectrum is not None and self.correct_background:
|
||
return [s - b for s, b in zip(spectrum, self.background_spectrum)]
|
||
return spectrum
|
||
|
||
def single_test(self):
|
||
# 获取输入框的值
|
||
jifenshijian = float(self.integration_time_entry.get())
|
||
shuaxinshijian = float(self.refresh_interval_entry.get())
|
||
|
||
# 设置平滑次数
|
||
boxcar_width = self.boxcar_width_slider.get()
|
||
wrapper.setBoxcarWidth(spectrometer_index, boxcar_width)
|
||
|
||
# 设置平均次数
|
||
scans_to_average = self.scans_to_average_slider.get()
|
||
wrapper.setScansToAverage(spectrometer_index, scans_to_average)
|
||
|
||
conditions = f"jifenshijian = {jifenshijian} shuaxinshijian = {shuaxinshijian} zihaodaxiao = {zihaodaxiao} pinghuacishu = {self.boxcar_width_slider.get()} pingjuncishu = {self.scans_to_average_slider.get()}"
|
||
self.test_count += 1
|
||
|
||
wavelengths, spectrum = self.get_spectrum_data()
|
||
|
||
# 在透射模式下计算透射率，否则应用背景校正
|
||
if self.transmission_mode and self.background_spectrum is not None and self.source_spectrum is not None:
|
||
transmission = self.calculate_transmission(spectrum)
|
||
self.update_plot(wavelengths, transmission, plot_type='transmission')
|
||
else:
|
||
corrected_spectrum = self.apply_background_correction(spectrum)
|
||
self.update_plot(wavelengths, corrected_spectrum, plot_type='spectrum')
|
||
|
||
def continuous_test(self):
|
||
while not self.stop_event.is_set():
|
||
# 获取输入框的值
|
||
jifenshijian = float(self.integration_time_entry.get())
|
||
shuaxinshijian = float(self.refresh_interval_entry.get())
|
||
|
||
# 设置平滑次数和平均次数
|
||
boxcar_width = self.boxcar_width_slider.get()
|
||
wrapper.setBoxcarWidth(spectrometer_index, boxcar_width)
|
||
scans_to_average = self.scans_to_average_slider.get()
|
||
wrapper.setScansToAverage(spectrometer_index, scans_to_average)
|
||
|
||
conditions = f"jifenshijian = {jifenshijian} shuaxinshijian = {shuaxinshijian} zihaodaxiao = {zihaodaxiao} pinghuacishu = {self.boxcar_width_slider.get()} pingjuncishu = {self.scans_to_average_slider.get()}"
|
||
self.test_count += 1
|
||
wavelengths, spectrum = self.get_spectrum_data()
|
||
self.write_data_to_file("test", self.test_count, conditions, spectrum)
|
||
|
||
# 根据当前模式计算并更新图表
|
||
if self.transmission_mode and self.background_spectrum is not None and self.source_spectrum is not None:
|
||
transmission = self.calculate_transmission(spectrum)
|
||
self.update_plot(wavelengths, transmission, plot_type='transmission')
|
||
else:
|
||
corrected_spectrum = self.apply_background_correction(spectrum)
|
||
self.update_plot(wavelengths, corrected_spectrum)
|
||
|
||
time.sleep(shuaxinshijian)
|
||
|
||
def start_continuous_test(self):
|
||
self.stop_event.clear() # 重置事件
|
||
self.continuous_thread = Thread(target=self.continuous_test)
|
||
self.continuous_thread.start()
|
||
|
||
def stop_continuous_test(self):
|
||
self.stop_event.set() # 设置事件通知线程停止
|
||
self.continuous_thread = None
|
||
|
||
def on_closing(self):
|
||
if self.data_file:
|
||
self.data_file.close()
|
||
if messagebox.askyesno("退出", "实验g了？"):
|
||
self.stop_continuous_test()
|
||
self.root.destroy()
|
||
sys.exit()
|
||
|
||
if __name__ == "__main__":
|
||
# 使用 ThemedTk 而不是普通的 Tk
|
||
root = ThemedTk(theme="equilux") # 使用 'arc' 主题
|
||
|
||
# 由于我们已经使用了 ttkthemes 来设置主题，下面这些行可以省略
|
||
# style = ttk.Style()
|
||
# style.theme_use('arc')
|
||
|
||
app = App(root)
|
||
root.mainloop()
|