from PySide6.QtWidgets import QApplication, QMainWindow, QTableWidget, QTableWidgetItem, QHeaderView from PySide6.QtGui import QPixmap from PySide6.QtCore import Qt from qt import Ui_MainWindow from qt_material import apply_stylesheet import pandas as pd import sys import os import re import balloon lat = 52.21 g = 9.76 tag_balloon_once = [ [ "地形重力波", "垂直传播方向", "水平传播方向", "本征（固有）频率", "本征周期", "本征频率与固有频率的比值（即长短轴之比）", ], [ "垂直波长", "水平波长", "纬向本征相速度", "经向本征相速度", "垂直本征相速度", "波动能", ], [ "势能", "纬向动量通量", "经向动量通量", "纬向风扰动振幅", "经向风扰动振幅", "温度扰动振幅", ], ] def set_table_balloon_once(table: QTableWidget, data: list): for i in range(3): for j in range(6): table.setItem(i, j * 2, QTableWidgetItem(tag_balloon_once[i][j])) if data[0] == True: data[0] = "是" else: data[0] = "否" if data[1] == 1: data[1] = "上" else: data[1] = "下" for i in range(2, len(data)): data[i] = f"{data[i]} " data[2] += "°" data[3] += "rad/s" data[4] += "h" data[6] += "km" data[7] += "km" for i in [8, 9, 10, 15, 16]: data[i] += "m/s" data[11] += "J/kg" data[12] += "J/kg" data[13] += "mPa" data[14] += "mPa" data[17] += "K" for i in range(18): table.setItem(i // 6, (i % 6) * 2 + 1, QTableWidgetItem(data[i])) class MainWindow(QMainWindow, Ui_MainWindow): comboType = [ "探空气球", "流星雷达", "Saber", "TIDI", "COSMIC", ] comboMode = [ ["重力波单次", "重力波统计"], ["重力波月统计", "潮汐波单次", "潮汐波月统计"], ["行星波月统计", "重力波单次", "重力波月统计"], ["行星波月统计"], ["行星波月统计"], ] comboDate = [ [["年", "时间"], ["起始年", "终止年"]], [["年", "月"], ["年", "日期"], ["年", "月"]], [["起始月", "-"], ["月", "日"], ["月", "-"]], [["起始月", "-"]], [["起始月", "-"]], ] comboMap = [ [1, 2], [3, 4, 5], [6, 7, 8], [9], [10], ] def get_date0_items(self): match self.dataType: case 0: try: self.currentDate0 = os.listdir( f"./data/{self.comboType[self.dataType]}/{self.dataStation}") except FileNotFoundError: return [] return [i[-4:] for i in self.currentDate0] case 1: self.currentDate0 = ["2024"] return self.currentDate0 case 2: try: self.currentDate0 = os.listdir( f"./data/{self.comboType[self.dataType]}/{self.dataStation}") except FileNotFoundError: return [] return [re.search(r"_([A-z]+)\d{4}_", i).group(1) for i in self.currentDate0] case _: self.currentDate0 = ["Feb"] return self.currentDate0 def get_date1_items(self): if self.dataType == 0 and self.dataMode == 0: try: self.currentDate1 = os.listdir( f"./data/{self.comboType[self.dataType]}/{self.dataStation}/{self.currentDate0[self.dataDate0]}") except FileNotFoundError: return [] return [re.search(r"_(\d{8}T\d{6}_\d+)", i).group(1) for i in self.currentDate1] elif self.dataType == 0 and self.dataMode == 1: return [i[-4:] for i in [end for j, end in enumerate(self.currentDate0) if j >= self.dataDate0]] elif self.dataType == 1 and self.dataMode == 0: self.currentDate1 = ["01"] return self.currentDate1 elif self.dataType == 1 and self.dataMode == 1: self.currentDate1 = ["20240317T000000"] return self.currentDate1 elif self.dataType == 1 and self.dataMode == 2: if self.dataStation == "黑龙江漠河": self.currentDate1 = ["01"] else: self.currentDate1 = ["03"] return self.currentDate1 elif self.dataType == 2 and self.dataMode == 1: return [f"{i:02d}" for i in range(1, 31)] else: self.currentDate1 = [f"{i:02d}" for i in range(1, 13)] return self.currentDate1 def __init__(self): super(MainWindow, self).__init__() self.setupUi(self) self.stackedWidget.setCurrentIndex(0) self.pixmap_logo = QPixmap("./data/logo.png") self.dataType = -1 self.dataMode = -1 self.currentDate0 = [] self.currentDate1 = [] self.funcMap = [ [self.on_balloon_once_change, self.on_balloon_year_change], [self.on_meteor_g_change, self.on_meteor_once_change, self.on_meteor_month_change], [self.on_Saber_twice_change, self.on_Saber_g_once_change, self.on_Saber_g_month_change], [self.on_TIDI_twice_change], [self.on_COSMIC_twice_change], ] self.clear_combo(0) self.combo_type.addItems(self.comboType) self.combo_type.currentTextChanged.connect(self.on_change_type) self.combo_mode.currentTextChanged.connect(self.on_change_mode) self.combo_station.currentTextChanged.connect(self.on_change_station) self.combo_date0.currentTextChanged.connect(self.on_change_date0) self.combo_date1.currentTextChanged.connect(self.on_change_date1) s = self.size() self.resize(s.width() + 1, s.height() + 1) self.resize(s) def resizeEvent(self, event): self.on_resize() super().resizeEvent(event) def on_resize(self): tmp = [ None, None, None, self.label_meteor_g, self.label_meteor_once, self.label_meteor_month, self.label_Saber_twice, self.label_Saber_g_once, self.label_Saber_g_month, self.label_TIDI_twice, self.label_COSMIC_twice, ] match self.stackedWidget.currentIndex(): case 0: l = self.label_idle_logo p = self.pixmap_logo l.setPixmap(p.scaled(l.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)) case 1: l = self.label_balloon_once p = self.pixmap_balloon_once if p.isNull(): self.clear_pixmap() l.setText("无重力波") else: l.setPixmap( p.scaled(l.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)) case 2: l = self.label_balloon_year p = self.pixmap_balloon_year if p.isNull(): self.clear_pixmap() l.setText("无数据") else: l.setPixmap( p.scaled(l.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)) case 5: i = self.stackedWidget.currentIndex() l = tmp[i] if self.dataStation == "黑龙江漠河": p = QPixmap("./output/tmp_5_01.png") else: p = QPixmap("./output/tmp_5_03.png") l.setPixmap(p.scaled(l.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)) l.setCursor(Qt.CursorShape.PointingHandCursor) case _: i = self.stackedWidget.currentIndex() l = tmp[i] p = QPixmap(f"./output/tmp_{i}.png") l.setPixmap(p.scaled(l.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)) l.setCursor(Qt.CursorShape.PointingHandCursor) def clear_pixmap(self): self.pixmap_balloon_once = QPixmap() self.pixmap_balloon_year = QPixmap() self.pixmap_meteor_g = QPixmap() self.pixmap_meteor_once = QPixmap() self.pixmap_meteor_month = QPixmap() self.pixmap_Saber_twice = QPixmap() self.pixmap_Saber_g_once = QPixmap() self.pixmap_Saber_g_month = QPixmap() self.pixmap_TIDI_twice = QPixmap() self.pixmap_COSMIC_twice = QPixmap() self.table_balloon_once.setVisible(False) for l in [ self.label_balloon_once, self.label_balloon_year, self.label_meteor_g, self.label_meteor_once, self.label_meteor_month, self.label_Saber_twice, self.label_Saber_g_once, self.label_Saber_g_month, self.label_TIDI_twice, self.label_COSMIC_twice, ]: l.setCursor(Qt.CursorShape.ArrowCursor) def clear_combo(self, level: int): self.clear_pixmap() self.on_resize() self.stackedWidget.setCurrentIndex(0) for i, combo in enumerate( [ self.combo_type, self.combo_mode, self.combo_station, self.combo_date0, self.combo_date1, ] ): if i >= level: combo.clear() combo.setEnabled(i <= level) def on_change_type(self): if self.combo_type.currentIndex() < 0: return self.clear_combo(1) self.dataType = self.combo_type.currentIndex() self.combo_mode.addItems(self.comboMode[self.dataType]) self.combo_date0.setPlaceholderText("-") self.combo_date1.setPlaceholderText("-") def on_change_mode(self): if self.combo_mode.currentIndex() < 0: return self.clear_combo(2) self.dataMode = self.combo_mode.currentIndex() if self.dataType < 2: self.combo_station.setPlaceholderText("台站") else: self.combo_station.setPlaceholderText("年") list_station = [] try: # list_station = [i for i in os.listdir(f"./data/{self.comboType[self.dataType]}") if i.encode("utf-8").isupper()] list_station = [i for i in os.listdir( f"./data/{self.comboType[self.dataType]}")] except FileNotFoundError: return self.combo_station.addItems(list_station) date0, date1 = self.comboDate[self.dataType][self.dataMode] self.combo_date0.setPlaceholderText(date0) self.combo_date1.setPlaceholderText(date1) def on_change_station(self): if self.combo_station.currentIndex() < 0: return self.clear_combo(3) self.dataStation = self.combo_station.currentText() self.combo_date0.addItems(self.get_date0_items()) def on_change_date0(self): if self.combo_date0.currentIndex() < 0: return self.clear_combo(4) self.dataDate0 = self.combo_date0.currentIndex() if self.dataType < 2 or (self.dataType == 2 and self.dataMode == 1): self.combo_date1.addItems(self.get_date1_items()) return self.combo_date1.setEnabled(False) self.stackedWidget.setCurrentIndex( self.comboMap[self.dataType][self.dataMode]) if f := self.funcMap[self.dataType][self.dataMode]: f() def on_change_date1(self): if self.combo_date1.currentIndex() < 0: return self.dataDate1 = self.combo_date1.currentIndex() self.stackedWidget.setCurrentIndex( self.comboMap[self.dataType][self.dataMode]) if f := self.funcMap[self.dataType][self.dataMode]: f() def on_balloon_once_change(self): path = os.path.join("./data/探空气球", self.dataStation, self.currentDate0[self.dataDate0], self.currentDate1[self.dataDate1]) try: data = balloon.read_data(path) except Exception as e: print(e) return path = "./output" + path[6:-2] + "png" os.makedirs(os.path.dirname(path), exist_ok=True) result = balloon.plot_once(data, path, lat, g) if len(result) == 0: self.pixmap_balloon_once = QPixmap() self.on_resize() return self.pixmap_balloon_once = QPixmap(path) self.label_balloon_once.setCursor(Qt.CursorShape.PointingHandCursor) t = self.table_balloon_once t.clear() t.setRowCount(3) t.setColumnCount(12) t.setVisible(True) t.verticalHeader().setHidden(True) t.horizontalHeader().setHidden(True) t.verticalScrollBar().setHidden(True) t.horizontalHeader().setSectionResizeMode(QHeaderView.ResizeMode.Stretch) set_table_balloon_once(t, result) t.setMaximumHeight(t.rowHeight(0) * 3) s = self.size() self.resize(s.width() + 1, s.height() + 1) self.resize(s) def on_balloon_year_change(self): columns = [ "file_name", "c", "a", "b", "omega_upper", "w_f", "ver_wave_len", "hori_wave_len", "c_x", "c_y", "c_z", "Ek", "E_p", "MFu", "MFv", "u1", "v1", "T1", "zhou_qi", ] combined_df = pd.DataFrame() for i in range(self.dataDate1 + 1): path = f"./output/探空气球/{self.dataStation}/{self.currentDate0[self.dataDate0 + i]}.csv" os.makedirs(os.path.dirname(path), exist_ok=True) if os.path.exists(path): year_df = pd.read_csv(path) combined_df = pd.concat([combined_df, year_df]) continue folder = "./data" + path[8:-4] year_df = pd.DataFrame() for file in os.listdir(folder): print(file) data = balloon.read_data(os.path.join(folder, file)) try: wave = balloon.extract_wave(data, lat, g) except Exception: wave = [] if len(wave) == 0: continue c = balloon.is_terrain_wave(data, lat, g) wave.insert(0, c) wave.insert(0, file) line = pd.DataFrame([wave], columns=columns) year_df = pd.concat([year_df, line]) combined_df = pd.concat([combined_df, year_df]) year_df.to_csv(path, index=False) path = f"./output/探空气球/{self.dataStation}/{self.currentDate0[self.dataDate0]}_{self.dataDate1}.png" if not balloon.plot_year(combined_df, path, lat, g): self.pixmap_balloon_year = QPixmap() self.on_resize() return self.pixmap_balloon_year = QPixmap(path) self.label_balloon_year.setCursor(Qt.CursorShape.PointingHandCursor) s = self.size() self.resize(s.width() + 1, s.height() + 1) self.resize(s) def on_meteor_g_change(self): self.on_resize() def on_meteor_once_change(self): self.on_resize() def on_meteor_month_change(self): self.on_resize() def on_Saber_twice_change(self): self.spin_Saber_twice_lat.setValue(0) self.spin_Saber_twice_alt.setValue(90) self.on_resize() def on_Saber_g_once_change(self): self.spin_Saber_g_once_lat.setValue(0) self.spin_Saber_g_once_lon.setValue(0) self.spin_Saber_g_once_alt.setValue(90) self.on_resize() def on_Saber_g_month_change(self): self.spin_Saber_g_month_lat.setValue(0) self.spin_Saber_g_month_alt.setValue(90) self.on_resize() def on_TIDI_twice_change(self): self.spin_TIDI_twice_lat.setValue(0) self.spin_TIDI_twice_alt.setValue(90) self.on_resize() def on_COSMIC_twice_change(self): self.spin_COSMIC_twice_lat.setValue(0) self.spin_COSMIC_twice_alt.setValue(40) self.on_resize() if __name__ == "__main__": app = QApplication(sys.argv) window = MainWindow() apply_stylesheet(app, theme="light_blue.xml", invert_secondary=True) window.show() sys.exit(app.exec())