|
|
# -*- coding: utf-8 -*-
|
|
|
# Time : 2023/8/9 10:20
|
|
|
# Author : lirunsheng
|
|
|
# User : l'r's
|
|
|
# Software: PyCharm
|
|
|
# File : X4.py
|
|
|
|
|
|
# -*- encoding: utf-8 -*-
|
|
|
"""
|
|
|
@Author: packy945
|
|
|
@FileName: main.py
|
|
|
@DateTime: 2023/5/30 15:29
|
|
|
@SoftWare: PyCharm
|
|
|
"""
|
|
|
import sys
|
|
|
from fitting import *
|
|
|
import numpy as np
|
|
|
import data as gl_data
|
|
|
import input
|
|
|
import tkinter as tk
|
|
|
from tkinter import *
|
|
|
import tkinter.filedialog # 注意次数要将文件对话框导入
|
|
|
# from demo1 import Addfunc
|
|
|
|
|
|
global Q_root,F_root
|
|
|
global root_window
|
|
|
global label1,label2,label3
|
|
|
|
|
|
class Addfunc():
|
|
|
def __init__(self):
|
|
|
self.r = 7.5 # 设置按钮大小
|
|
|
self.rr = 2.5# 设置按钮大小
|
|
|
self.window = tk.Tk() # 新建函数增加窗口
|
|
|
self.window.title("新增函数曲线")#设置窗口标题
|
|
|
self.window.geometry('400x300')#设置窗口大小
|
|
|
self.show()#显示按钮
|
|
|
self.window.mainloop()#运行UI界面
|
|
|
|
|
|
def show(self):
|
|
|
for widget in self.window.winfo_children():
|
|
|
widget.destroy()#清空已有组件
|
|
|
self.add_cv = tk.Canvas(self.window, width=400, height=300) # 新建画布
|
|
|
self.add_cv.place(x=0, y=0) # 放置画布
|
|
|
cur = 0#记录已有按钮个数
|
|
|
for f in function.Fun:#遍历已有函数
|
|
|
self.add_cv.create_oval(20 - self.r, 20 + 15 + cur * 30 - self.r, 20 + self.r, 20 + 15 + cur * 30 + self.r
|
|
|
, fill="white", width=1, outline="black")# 绘制圆圈
|
|
|
if f.show == 1:#若为当前选中函数，则标记
|
|
|
self.add_cv.create_oval(20 - self.rr, 20 + 15 + cur * 30 - self.rr, 20 + self.rr, 20 + 15 + cur * 30 + self.rr
|
|
|
, fill="black", width=1, outline="black")
|
|
|
button = tk.Button(self.window, text=f.name + f.demo, bd=0, font=("微软雅黑", 12)
|
|
|
, command=lambda x=f.no: self.change(x), anchor=W)#设置选择按钮
|
|
|
button.place(x=40, y=20 + cur * 30, width=300, height=30)#放置选择按钮
|
|
|
cur += 1#计数加一
|
|
|
|
|
|
def change(self, no):#切换函数显示情况
|
|
|
function.Show[no] = not function.Show[no]
|
|
|
function.Fun[no].show = not function.Fun[no].show
|
|
|
self.show()#更新函数选择面板
|
|
|
f_button()#更新函数显示界面
|
|
|
# class AddFunc(Addfunc):
|
|
|
# def change(self, no):
|
|
|
# # super().change(no)
|
|
|
# function.Show[no] = not function.Show[no]
|
|
|
# function.Fun[no].show = not function.Fun[no].show
|
|
|
# self.show() # 更新函数选择面板
|
|
|
# f_button()
|
|
|
|
|
|
def addfunc():
|
|
|
Addfunc()
|
|
|
# AddFunc()
|
|
|
def window():
|
|
|
global root_window
|
|
|
global label1, label2, label3
|
|
|
root_window = Tk()
|
|
|
root_window.title('函数拟合')
|
|
|
root_window.geometry('900x600') # 设置窗口大小:宽x高,注,此处不能为 "*",必须使用 "x"
|
|
|
# 设置主窗口的背景颜色,颜色值可以是英文单词，或者颜色值的16进制数,除此之外还可以使用Tk内置的颜色常量
|
|
|
root_window["background"] = "white"
|
|
|
root_window.resizable(0, 0) # 防止用户调整尺寸
|
|
|
label1 = tk.Label(root_window, text="样本数据\n集文件", font=('Times', 8), bg="white",
|
|
|
width=13, height=3, # 设置标签内容区大小
|
|
|
padx=0, pady=0, borderwidth=0, )
|
|
|
label1.place(x=10, y=4) # 设置填充区距离、边框宽度和其样式（凹陷式）
|
|
|
label3 = tk.Label(root_window, text="", font=('Times', 8), bg="white", fg="black",
|
|
|
width=39, height=2, padx=0, pady=0, borderwidth=0, relief="ridge", highlightcolor="blue")
|
|
|
label3.place(x=122, y=10)
|
|
|
# 使用按钮控件调用函数
|
|
|
tk.Button(root_window, text="装载", relief=RAISED, command=lambda: askfile()).place(x=370, y=12)
|
|
|
label2 = tk.Label(root_window, text="拟合曲线类型", font=('Times', 12), bg="white",
|
|
|
width=20, height=3, # 设置标签内容区大小
|
|
|
padx=0, pady=0, borderwidth=0, )
|
|
|
# 设置填充区距离、边框宽度和其样式（凹陷式）
|
|
|
label2.place(x=450, y=4)
|
|
|
gl_data.Canvas2 = tk.Canvas(root_window, bg='white', width=450, height=330)
|
|
|
gl_data.Canvas2.place(x=4, y=60)
|
|
|
# 定义一个处理文件的相关函数
|
|
|
def askfile():
|
|
|
# 从本地选择一个文件，并返回文件的路径
|
|
|
filename = tkinter.filedialog.askopenfilename()
|
|
|
if filename != '':#若选中了一个文件，则对文件进行读取
|
|
|
label3.config(text=filename)#显示文件路径
|
|
|
read_sample_data(filename)#将文件读取并存到sampleData中
|
|
|
# print_sample_data(filename)#将文件读取并逐行输出
|
|
|
selfdata_show(gl_data.X, gl_data.Y, gl_data.LOW, gl_data.HIGH)
|
|
|
else:#若未选择文件，则显示为空
|
|
|
label3.config(text='')
|
|
|
|
|
|
def print_sample_data(file_path):#打开对应文件
|
|
|
with open(file_path, 'r') as file:
|
|
|
for line in file:#逐行读取文件
|
|
|
line = line.strip('\n')#移除换行符
|
|
|
sx, sy = line.split(' ')#以空格分割x，y
|
|
|
print(f'sx: {float(sx)}, sy: {float(sy)}')#将sx、sy转换为浮点数并打印
|
|
|
|
|
|
def read_sample_data(file_path):
|
|
|
x, y = [], []#初始化x，y
|
|
|
with open(file_path, 'r') as file:
|
|
|
for line in file:#逐行读文件
|
|
|
line = line.strip('\n')#移除换行符
|
|
|
sx, sy = line.split(' ')#以空格分割x，y
|
|
|
x.append(float(sx))#将sx转换为浮点数并加入数组
|
|
|
y.append(float(sy))#将sy转换为浮点数并加入数组
|
|
|
gl_data.X, gl_data.Y = np.array(x), np.array(y)#x，y数组转为array并赋值给全局变量
|
|
|
|
|
|
def buttons():
|
|
|
# 随机生成样本点并显示
|
|
|
b_dot = tk.Button(root_window, text="生成\n数据集", relief=RAISED, bd=4, bg="white", font=("微软雅黑", 10),
|
|
|
command=lambda: generate_and_plot_sample_data(gl_data.LOW, gl_data.HIGH))
|
|
|
b_dot.place(x=455, y=410)
|
|
|
# 显示当前样本点
|
|
|
b_show = tk.Button(root_window, text="显示\n数据集", relief=RAISED, bd=4, bg="white", font=("微软雅黑", 10),
|
|
|
command=lambda: selfdata_show(gl_data.X, gl_data.Y, gl_data.LOW, gl_data.HIGH))
|
|
|
b_show.place(x=510, y=410)
|
|
|
# 显示数据集与曲线
|
|
|
b_line = tk.Button(root_window, text="显示数据\n集与曲线", relief=RAISED, bd=4, bg="white", font=("微软雅黑", 10),
|
|
|
command=lambda: draw_dots_and_line(gl_data.Xian_index, gl_data.LOW, gl_data.HIGH, gl_data.X, gl_data.Y))
|
|
|
b_line.place(x=565, y=410)
|
|
|
# 手动输入数据集
|
|
|
b_input = tk.Button(root_window, text="手动输入数据集", relief=RAISED, bd=4, bg="white", pady=7, font=("微软雅黑", 13),
|
|
|
command=lambda: input.input_num(root_window))
|
|
|
b_input.place(x=633, y=410)
|
|
|
# 拟合并输出拟合结果
|
|
|
b_fit = tk.Button(root_window, text="拟合", relief=RAISED, bd=4, bg="white", pady=7, font=("微软雅黑", 13),
|
|
|
command=lambda: fit_data(gl_data.Xian_index, gl_data.X, gl_data.Y))
|
|
|
b_fit.place(x=771, y=410)
|
|
|
def show_fit():
|
|
|
L = tk.Label(root_window, text='结果输出：', bg='white', font=("微软雅黑", 16)
|
|
|
, anchor=W)
|
|
|
L.place(x=20, y=480, width=100, height=30)
|
|
|
sout = str(gl_data.Out)
|
|
|
ans = tk.Label(root_window, text=sout, font=("微软雅黑", 14)
|
|
|
, anchor=W, justify='left')
|
|
|
ans.place(x=120, y=480, width=760, height=100)
|
|
|
print(sout)
|
|
|
def fit_data(xian_index, sx, sy):
|
|
|
# gl_data.yvals_pow = []
|
|
|
cur = function.Fun[xian_index] # 装载正在选择的函数
|
|
|
func = cur.get_fun()#获取当前函数func
|
|
|
popt, pcov = curve_fit(func, sx, sy)# 用curve_fit来对点进行拟合
|
|
|
yvals_pow = func(gl_data.X, *popt)
|
|
|
rr = gl_data.goodness_of_fit(yvals_pow, gl_data.Y)# 计算本次拟合的R*R值，用于表示拟合优度
|
|
|
# 输出拟合后的各个参数值
|
|
|
ans = '\n函数系数：'
|
|
|
for i in range(cur.variable):
|
|
|
if i == 4:
|
|
|
ans += '\n'
|
|
|
if i != 0:
|
|
|
ans += ', '
|
|
|
ans += chr(ord('a') + i) + '=' + '{:.2e}'.format(popt[i]) # str(round(gl_data.popt[i], 3))
|
|
|
gl_data.Out = '函数形式：' + cur.name + ' '
|
|
|
gl_data.Out += cur.demo
|
|
|
gl_data.Out += ans
|
|
|
gl_data.Out += '\n拟合优度(R\u00b2)：' + str(round(rr, 5))
|
|
|
show_fit()# 显示函数信息到输出框
|
|
|
|
|
|
def change_Q(no):
|
|
|
gl_data.Quadrant = no #更改全局变量的象限显示
|
|
|
if no:#若为一象限，则修改显示下限为0
|
|
|
gl_data.LOW = 0
|
|
|
else:#若为四象限，则修改显示下限为-gl_data.maxV
|
|
|
gl_data.LOW = -gl_data.MAXV
|
|
|
q_button()#更新象限显示面板
|
|
|
|
|
|
def q_button():
|
|
|
r = 7.5
|
|
|
rr = 2.5
|
|
|
for widget in Q_root.winfo_children():
|
|
|
widget.destroy()
|
|
|
q_cv = tk.Canvas(Q_root, width=400, height=50)
|
|
|
q_cv.place(x=0, y=0)
|
|
|
l = tk.Label(Q_root, text='坐标轴', bd=0, font=("微软雅黑", 16)
|
|
|
, anchor=W)
|
|
|
l.place(x=20, y=0, width=80, height=50,)
|
|
|
# 四象限按钮
|
|
|
b1 = tk.Button(Q_root, text='四象限', bd=0, font=("微软雅黑", 16)
|
|
|
, command=lambda: change_Q(0), anchor=W)
|
|
|
b1.place(x=170, y=0, width=80, height=50,)
|
|
|
# 一象限按钮
|
|
|
b2 = tk.Button(Q_root, text='一象限', bd=0, font=("微软雅黑", 16)
|
|
|
, command=lambda: change_Q(1), anchor=W)
|
|
|
b2.place(x=320, y=0, width=80, height=50,)
|
|
|
# 绘制标记框
|
|
|
q_cv.create_oval(140 - r, 25 - r, 140 + r, 25 + r
|
|
|
, fill="white", width=1, outline="black")
|
|
|
q_cv.create_oval(290 - r, 25 - r, 290 + r, 25 + r
|
|
|
, fill="white", width=1, outline="black")
|
|
|
# 根据当前的象限选择值来填充标记框
|
|
|
if gl_data.Quadrant == 0:
|
|
|
q_cv.create_oval(140 - rr, 25 - rr, 140 + rr, 25 + rr
|
|
|
, fill="black", width=1, outline="black")
|
|
|
else:
|
|
|
q_cv.create_oval(290 - rr, 25 - rr, 290 + rr, 25 + rr
|
|
|
, fill="black", width=1, outline="black")
|
|
|
def change_f(no):
|
|
|
gl_data.Xian_index = no#设置全局函数编号为该函数
|
|
|
f_button()#重新绘制函数显示界面
|
|
|
|
|
|
def f_button():
|
|
|
r = 7.5#设置按钮大小
|
|
|
rr = 2.5#设置按钮大小
|
|
|
for widget in F_root.winfo_children():
|
|
|
widget.destroy()#清空原有按钮
|
|
|
f_cv = tk.Canvas(F_root, width=400, height=330)#新建画布
|
|
|
f_cv.place(x=0, y=0)#放置画布
|
|
|
f_cv.create_rectangle(2, 2, 398, 328, fill='white', outline="#0099ff")#设置画布边框及底色
|
|
|
cur = 0#函数计数
|
|
|
for fun in function.Fit_type_library:#遍历函数库
|
|
|
f = function.Fit_type_library.get(fun)#获取函数具体信息
|
|
|
if function.Show[f[0]] == 1:# 如果show为1则显示
|
|
|
f_cv.create_oval(20 - r, 20 + 15 + cur * 30 - r, 20 + r, 20 + 15 + cur * 30 + r
|
|
|
, fill="white", width=1, outline="black")# 绘制标记框
|
|
|
if f[0] == gl_data.Xian_index:# 若选择为当前函数则标记
|
|
|
f_cv.create_oval(20 - rr, 20 + 15 + cur * 30 - rr, 20 + rr, 20 + 15 + cur * 30 + rr
|
|
|
, fill="black", width=1, outline="black")
|
|
|
# 绘制切换按钮,单击则将当前使用函数切换为该函数
|
|
|
button = tk.Button(F_root, text=f[2] + ' ' + f[1], bd=0, bg="white", font=("微软雅黑", 12)
|
|
|
, command=lambda x=f[0]: change_f(x), anchor=W)
|
|
|
button.place(x=40, y=20 + cur * 30, width=300, height=30)
|
|
|
cur += 1#计数加一
|
|
|
|
|
|
def close_window():
|
|
|
sys.exit()
|
|
|
def main():
|
|
|
global Q_root, F_root,root_window
|
|
|
gl_data.X = []
|
|
|
gl_data.X = []
|
|
|
window()
|
|
|
# 函数选择相关界面
|
|
|
F_root = tk.Frame(root_window, width=400, height=330, bg='white', )
|
|
|
F_root.place(x=490, y=60)
|
|
|
# 象限选择相关界面
|
|
|
Q_root = tk.Frame(root_window, width=400, height=50, bg='white', )
|
|
|
Q_root.place(x=20, y=410)
|
|
|
function.f_read()
|
|
|
buttons()
|
|
|
f_button()
|
|
|
q_button()
|
|
|
show_fit()
|
|
|
tk.Button(root_window, text="新增拟合曲线类型", bd=4, command=addfunc).place(x=740, y=12)
|
|
|
root_window.protocol("WM_DELETE_WINDOW", close_window)
|
|
|
root_window.mainloop()
|
|
|
if __name__ == '__main__':
|
|
|
main()
|
|
|
pass
|
