-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.py
182 lines (157 loc) · 7.68 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
# -*- coding: utf-8 -*-
import sys
import pandas as pd
from func import Quantile_Granger, logging
import func
from PyQt5 import QtCore, QtGui
from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog
from PyQt5.QtCore import pyqtSlot, pyqtSignal, QThread
from beauty_UI import beauty
from pyqt5界面.ui_MainWindow import Ui_MainWindow
from pyqt5界面.child_GUI.read_windows import read_Window
from pyqt5界面.child_GUI.QVAR_windows import QVAR
class Runthread(QThread):
# 通过类成员对象定义信号对象
_signal = pyqtSignal(str)
def __init__(self, form):
super(Runthread, self).__init__()
self.form = form
def __del__(self):
self.wait()
def run(self):
form = self.form
# 加载参数
start = form.ui.doubleSpinBox.value() # 区间起点
end = form.ui.doubleSpinBox_2.value() # 区间终点
num = form.ui.spinBox.value() # 区间个数
sign = form.ui.comboBox_2.currentText() # 模式选择
max_lag = int(form.ui.comboBox.currentText()) # 最大滞后阶数选择，默认为5
info_type = form.ui.comboBox_3.currentText() # 信息准则选择
WaldNum = form.ui.spinBox_3.value() # 估计wald个数，默认1000个
sign_num = int(form.ui.comboBox_4.currentText()) # 有效数字，默认为3
AicNum = form.ui.spinBox_2.value() # 估计各区间最优滞后阶数，默认50个
# 开始计算
qrange, qr_name = form.func.set_range(start, end, num)
DataList = form.func.pattern(form.df, sign)
INFO = [self._signal.emit]
if form.ui.checkBox_2.isChecked():
INFO.append(logging.info)
form.func.calculate(DataList, qrange, qr_name,
max_lag, info_type, WaldNum, sign_num, AicNum, objects=INFO)
class QmyMainWindow(QMainWindow):
def __init__(self, parent=None):
super().__init__(parent) # 调用父类构造函数，创建窗体
self.ui = Ui_MainWindow() # 创建UI对象
self.ui.setupUi(self) # 构造UI界面
self.resize(700, 700)
beauty(self) # 美化界面
self.func = Quantile_Granger() # 导入主函数
self.init_parameter() # 初始化各参数
# 默认导入测试数据
self.df = pd.read_excel(r'.\data\测试数据.xlsx')
self.df = self.df.drop(self.df.columns[0], axis=1)
# ==============自定义功能函数========================
def init_parameter(self):
'''
初始化各参数
'''
self.ui.doubleSpinBox.setValue(0.1) # 区间起点为0.1
self.ui.doubleSpinBox.setSingleStep(0.01)
self.ui.doubleSpinBox_2.setValue(0.9) # 区间终点为0.9
self.ui.doubleSpinBox_2.setSingleStep(0.01)
self.ui.spinBox.setValue(17) # 个数默认为17个
self.ui.spinBox_2.setValue(30) # 滞后估计数默认30个
self.ui.spinBox_3.setMinimum(10)
self.ui.spinBox_3.setMaximum(1500) # 设置上界
self.ui.spinBox_3.setValue(1000) # 设置估计wald的个数为1000
self.ui.textEdit.setText(
' '*20+'使用说明:\n1.点击导入数据按钮加载xlsx数据；\n2.进行参数设定，具体参数信息请参见README.md文件；\n3.点击开始运行按钮，程序开始计算')
self.ui.lineEdit.setText(
'[0.10-0.15]、[0.15-0.20]...[0.50-0.55]...[0.80-0.85]、[0.85-0.90]')
self.ui.checkBox.setChecked(True) # 勾选日期
self.ui.checkBox_2.setChecked(True) # 勾选日志
self.ui.comboBox_4.setCurrentIndex(3) # 设置默认保留三位小数
self.ui.comboBox.setCurrentIndex(4) # 最大阶数默认为5
self.ui.comboBox_3.setCurrentIndex(1) # 信息准则默认为BIC
self.ui.comboBox_2.setCurrentIndex(0) # 模式设定默认为1
def call_backlog(self, msg):
self.ui.textEdit.append(msg) # 将线程的参数传入
# =============================================================================
# # 当把边框给隐藏掉，边框将无法拖动移动，这个可以通过重写函数来实现。
# def mousePressEvent(self, event):
# if event.button()==QtCore.Qt.LeftButton:
# self.m_flag=True
# self.m_Position=event.globalPos()-self.pos() #获取鼠标相对窗口的位置
# event.accept()
# self.setCursor(QtGui.QCursor(QtCore.Qt.OpenHandCursor)) #更改鼠标图标
#
# def mouseMoveEvent(self, QMouseEvent):
# if QtCore.Qt.LeftButton and self.m_flag:
# self.move(QMouseEvent.globalPos()-self.m_Position)#更改窗口位置
# QMouseEvent.accept()
#
# def mouseReleaseEvent(self, QMouseEvent):
# self.m_flag=False
# self.setCursor(QtGui.QCursor(QtCore.Qt.ArrowCursor))
# =============================================================================
# ==========由connectSlotsByName()自动连接的槽函数============
@pyqtSlot()
def on_action_triggered(self): # 导入数据按钮
download_path = QFileDialog.getOpenFileName(None, "浏览", None,
"Text Files (*.xlsx);;Text Files (*.xls)")
try:
self.df = pd.read_excel(download_path[0])
self.ui.textEdit.setText("\n 恭喜,数据读取成功!")
if self.ui.checkBox.isChecked():
# 判断date是否勾选，如果勾选，删除第一列日期数据
self.df = self.df.drop(self.df.columns[0], axis=1)
except:
self.ui.textEdit.setText("\n 数据读取失败,请重新导入数据!")
@pyqtSlot()
def on_action_2_triggered(self): # 开始运行按钮
func.EXIT = False
self.ui.textEdit.setText('')
# 创建线程
self.thread = Runthread(self)
# 连接信号
self.thread._signal.connect(self.call_backlog) # 进程连接回传到GUI的事件
# 开始线程
self.thread.start()
@pyqtSlot()
def on_action_4_triggered(self): # 初始化按钮
self.init_parameter()
self.ui.textEdit.setText(
' '*20+'使用说明:\n1.点击导入数据按钮加载xlsx数据；\n2.进行参数设定，具体参数信息请参见README.md文件；\n3.点击开始运行按钮，程序开始计算\n\n 初始化成功！')
@pyqtSlot()
def on_action_3_triggered(self): # 最小化按钮
self.showMinimized()
@pyqtSlot()
def on_action_5_triggered(self): # 终止运行按钮
func.EXIT = True
@pyqtSlot()
def on_pushButton_clicked(self): # 生成区间按钮
start = self.ui.doubleSpinBox.value() # 区间起点
end = self.ui.doubleSpinBox_2.value() # 区间终点
num = self.ui.spinBox.value() # 区间个数
_, qr_name = self.func.set_range(start, end, num)
if len(qr_name) > 5:
display = f"{qr_name[0]}、{qr_name[1]}...{qr_name[int(len(qr_name)/2)]}...{qr_name[-2]}、{qr_name[-1]}"
else:
display = '、'.join(qr_name)
self.ui.lineEdit.setText(display)
@pyqtSlot()
def on_action_6_triggered(self): # 查看数据
read = read_Window(self)
read.display(self.df)
read.show()
@pyqtSlot()
def on_actionQVAR_triggered(self): # QVAR估计
QVARdata = self.func.pattern(self.df, self.ui.comboBox_2.currentText())
QVAR_win = QVAR(QVARdata, self)
QVAR_win.show()
# ============窗体测试程序 ================================
if __name__ == "__main__":
app = QApplication(sys.argv) # 创建GUI应用程序
form = QmyMainWindow() # 创建窗体
form.show()
sys.exit(app.exec_())