389 lines
14 KiB
Python
389 lines
14 KiB
Python
import os
|
||
import tkinter as tk
|
||
from tkinter import messagebox
|
||
import subprocess
|
||
import sys
|
||
import time
|
||
|
||
# =============== 全局命令配置 ===============
|
||
COMMANDS_TEMPLATE = [
|
||
"conda activate aiden-p3d",
|
||
"d:",
|
||
"cd aiden-p3d",
|
||
]
|
||
|
||
# 扫描路径配置（默认F盘根目录）
|
||
SCAN_PATH = "E:\\" # 修改为您需要的默认扫描路径
|
||
|
||
# 新增文件夹检测间隔（秒）
|
||
DETECTION_INTERVAL = 5 # 每5秒检测一次新文件夹
|
||
|
||
# 游戏配置映射
|
||
GAME_COMMANDS = {
|
||
"赛博朋克2077": {
|
||
"command": "python load_point_cloud_saibo_new.py \"{}\"",
|
||
"description": "赛博朋克2077 - python load_point_cloud_saibo_new.py"
|
||
},
|
||
"巫师3": {
|
||
"command": "python load_point_cloud_wushi_new.py \"{}\"",
|
||
"description": "巫师3 - python load_point_cloud_wushi_new.py"
|
||
},
|
||
"孤岛危机": {
|
||
"command": "python load_point_cloud_crysis_new.py \"{}\"",
|
||
"description": "孤岛危机 - python load_point_cloud_crysis_new.py"
|
||
},
|
||
"地平线零之曙光": {
|
||
"command": "python load_point_horizon_new.py \"{}\"",
|
||
"description": "地平线零之曙光 - python load_point_horizon_new.py"
|
||
},
|
||
"原子之心": {
|
||
"command": "python debug_atomicHeart.py -d \"{}\"",
|
||
"description": "原子之心 - python debug_atomicHeart.py -d"
|
||
},
|
||
"巴士模拟18": {
|
||
"command": "python debug_bus18.py -d \"{}\"",
|
||
"description": "巴士模拟18 - python debug_bus18.py -d"
|
||
},
|
||
"超自然车旅": {
|
||
"command": "python debug_pacificdrive.py -d \"{}\"",
|
||
"description": "超自然车旅 - python debug_pacificdrive.py -d"
|
||
},
|
||
"印蒂卡": {
|
||
"command": "python debug_indika.py -d \"{}\"",
|
||
"description": "印蒂卡 - python debug_indika.py -d"
|
||
},
|
||
"黑神话：悟空": {
|
||
"command": "python restruct_b1.py --data-dir \"{}\"",
|
||
"description": "黑神话：悟空 - python restruct_b1.py --data-dir"
|
||
},
|
||
"新游戏选项": {
|
||
"command": "conda activate worldgen && E: && cd datacheck && run.bat \"{}\"",
|
||
"description": "新游戏选项 - E: && cd datacheck && run.bat"
|
||
}
|
||
}
|
||
|
||
# MeshLab路径
|
||
MESHLAB_PATH = r"C:\Program Files\VCG\MeshLab\meshlab.exe"
|
||
|
||
# 输出目录
|
||
OUTPUT_DIR = r"E:\datacheck\outputs"
|
||
|
||
# =============== 结束配置 ===============
|
||
|
||
class FolderPlayerApp:
|
||
def __init__(self, root):
|
||
self.root = root
|
||
self.root.title("CV Saved Folders Player")
|
||
self.root.geometry("700x250") # 增加窗口宽度以适应新按钮
|
||
self.root.resizable(False, False)
|
||
|
||
self.selected_game = tk.StringVar(value="赛博朋克2077")
|
||
self.scan_folders()
|
||
|
||
if not self.folders:
|
||
messagebox.showerror("错误", f"在路径 {SCAN_PATH} 中没有找到以'cv_saved_'或'25'为前缀的文件夹")
|
||
sys.exit(1)
|
||
|
||
self.current_index = 0
|
||
self.process = None
|
||
self.last_scan_time = time.time()
|
||
|
||
self.create_ui()
|
||
self.update_display()
|
||
self.start_detection_timer()
|
||
self._force_focus_timer()
|
||
|
||
def _force_focus_timer(self):
|
||
self.root.focus_force()
|
||
self.root.after(500, self._force_focus_timer)
|
||
|
||
def start_detection_timer(self):
|
||
self.check_new_folders()
|
||
self.root.after(DETECTION_INTERVAL * 1000, self.start_detection_timer)
|
||
|
||
def check_new_folders(self):
|
||
current_dir = SCAN_PATH
|
||
try:
|
||
all_items = os.listdir(current_dir)
|
||
except Exception as e:
|
||
print(f"检测新文件夹时出错: {e}")
|
||
return
|
||
|
||
processed_folders = set()
|
||
for filename in ["passed_files.txt", "rejected_files.txt"]:
|
||
if os.path.exists(filename):
|
||
with open(filename, 'r', encoding='utf-8') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if line and not line.startswith('#'):
|
||
processed_folders.add(line)
|
||
|
||
new_folders = [
|
||
os.path.join(current_dir, item)
|
||
for item in all_items
|
||
if ((item.startswith("cv_saved_") or item.startswith("25")) and
|
||
os.path.isdir(os.path.join(current_dir, item)) and
|
||
item not in processed_folders and
|
||
os.path.join(current_dir, item) not in self.folders)
|
||
]
|
||
|
||
if new_folders:
|
||
new_folders.sort()
|
||
was_empty = len(self.folders) == 0
|
||
self.folders.extend(new_folders)
|
||
|
||
if was_empty and self.folders:
|
||
self.current_index = 0
|
||
self.update_display()
|
||
|
||
messagebox.showinfo("提示", f"检测到 {len(new_folders)} 个新文件夹，已添加到列表")
|
||
|
||
def scan_folders(self):
|
||
current_dir = SCAN_PATH
|
||
try:
|
||
all_items = os.listdir(current_dir)
|
||
except FileNotFoundError:
|
||
messagebox.showerror("错误", f"指定的扫描路径不存在: {SCAN_PATH}")
|
||
sys.exit(1)
|
||
except PermissionError:
|
||
messagebox.showerror("错误", f"没有权限访问指定路径: {SCAN_PATH}")
|
||
sys.exit(1)
|
||
|
||
processed_folders = set()
|
||
for filename in ["passed_files.txt", "rejected_files.txt"]:
|
||
if os.path.exists(filename):
|
||
with open(filename, 'r', encoding='utf-8') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if line and not line.startswith('#'):
|
||
processed_folders.add(line)
|
||
|
||
self.folders = [
|
||
os.path.join(current_dir, item)
|
||
for item in all_items
|
||
if ((item.startswith("cv_saved_") or item.startswith("25")) and
|
||
os.path.isdir(os.path.join(current_dir, item)) and
|
||
item not in processed_folders)
|
||
]
|
||
self.folders.sort()
|
||
|
||
def create_ui(self):
|
||
game_frame = tk.Frame(self.root)
|
||
game_frame.pack(fill=tk.X, padx=10, pady=5)
|
||
|
||
tk.Label(game_frame, text="选择游戏:", font=("Arial", 10)).pack(side=tk.LEFT)
|
||
game_dropdown = tk.OptionMenu(game_frame, self.selected_game, *GAME_COMMANDS.keys())
|
||
game_dropdown.pack(side=tk.LEFT, padx=5)
|
||
|
||
self.status_label = tk.Label(self.root, text="", pady=5)
|
||
self.status_label.pack()
|
||
|
||
self.folder_label = tk.Label(self.root, text="", font=("Arial", 12), fg="blue")
|
||
self.folder_label.pack()
|
||
|
||
button_frame = tk.Frame(self.root)
|
||
button_frame.pack(side=tk.BOTTOM, pady=20)
|
||
|
||
self.prev_button = tk.Button(
|
||
button_frame,
|
||
text="上一条(<-)",
|
||
width=10,
|
||
command=self.prev_folder,
|
||
state=tk.DISABLED if self.current_index == 0 else tk.NORMAL
|
||
)
|
||
self.prev_button.pack(side=tk.LEFT, padx=5)
|
||
|
||
self.play_button = tk.Button(
|
||
button_frame,
|
||
text="播放(空格)",
|
||
width=10,
|
||
command=self.play_current
|
||
)
|
||
self.play_button.pack(side=tk.LEFT, padx=5)
|
||
|
||
self.pass_button = tk.Button(
|
||
button_frame,
|
||
text="通过(p)",
|
||
width=10,
|
||
command=self.mark_as_passed,
|
||
bg="green",
|
||
fg="white"
|
||
)
|
||
self.pass_button.pack(side=tk.LEFT, padx=5)
|
||
|
||
self.reject_button = tk.Button(
|
||
button_frame,
|
||
text="不通过(r)",
|
||
width=10,
|
||
command=self.mark_as_rejected,
|
||
bg="red",
|
||
fg="white"
|
||
)
|
||
self.reject_button.pack(side=tk.LEFT, padx=5)
|
||
|
||
self.next_button = tk.Button(
|
||
button_frame,
|
||
text="下一条(->)",
|
||
width=10,
|
||
command=self.next_folder,
|
||
state=tk.DISABLED if self.current_index == len(self.folders) - 1 else tk.NORMAL
|
||
)
|
||
self.next_button.pack(side=tk.LEFT, padx=5)
|
||
|
||
# 新增打开pts.ply文件按钮
|
||
self.open_ply_button = tk.Button(
|
||
button_frame,
|
||
text="打开PLY(o)",
|
||
width=10,
|
||
command=self.open_ply_file,
|
||
bg="orange",
|
||
fg="white"
|
||
)
|
||
self.open_ply_button.pack(side=tk.LEFT, padx=5)
|
||
|
||
self.root.bind('<Left>', lambda event: self.prev_folder())
|
||
self.root.bind('<Right>', lambda event: self.next_folder())
|
||
self.root.bind('<space>', lambda event: self.play_current())
|
||
self.root.bind('<p>', lambda event: self.mark_as_passed())
|
||
self.root.bind('<r>', lambda event: self.mark_as_rejected())
|
||
self.root.bind('<o>', lambda event: self.open_ply_file()) # 绑定o键
|
||
|
||
def update_display(self):
|
||
if not self.folders:
|
||
self.status_label.config(text="所有文件夹已处理完毕")
|
||
self.folder_label.config(text="")
|
||
return
|
||
|
||
current_path = self.folders[self.current_index]
|
||
folder_name = os.path.basename(current_path)
|
||
|
||
self.status_label.config(text=f"正在处理: 第 {self.current_index + 1}/{len(self.folders)} 个")
|
||
self.folder_label.config(text=folder_name)
|
||
|
||
self.prev_button.config(state=tk.NORMAL if self.current_index > 0 else tk.DISABLED)
|
||
self.next_button.config(state=tk.NORMAL if self.current_index < len(self.folders) - 1 else tk.DISABLED)
|
||
|
||
def prev_folder(self):
|
||
if self.current_index > 0:
|
||
self.current_index -= 1
|
||
self.update_display()
|
||
|
||
def next_folder(self):
|
||
if self.current_index < len(self.folders) - 1:
|
||
self.current_index += 1
|
||
self.update_display()
|
||
|
||
def play_current(self):
|
||
if not self.folders:
|
||
return
|
||
|
||
current_path = self.folders[self.current_index]
|
||
selected_game_name = self.selected_game.get()
|
||
game_config = GAME_COMMANDS.get(selected_game_name)
|
||
|
||
if not game_config:
|
||
messagebox.showerror("错误", f"未找到游戏 {selected_game_name} 的配置")
|
||
return
|
||
|
||
try:
|
||
self.close_command_prompt()
|
||
time.sleep(0.5)
|
||
|
||
if selected_game_name == "新游戏选项":
|
||
# 执行新游戏选项命令（不使用COMMANDS_TEMPLATE）
|
||
game_command = game_config["command"]
|
||
formatted_command = game_command.format(current_path)
|
||
|
||
# 启动命令（不等待）
|
||
self.process = subprocess.Popen(f'start cmd /k "{formatted_command}"', shell=True)
|
||
else:
|
||
# 其他游戏使用原有的COMMANDS_TEMPLATE
|
||
game_command = game_config["command"]
|
||
formatted_command = game_command.format(current_path)
|
||
commands = COMMANDS_TEMPLATE.copy()
|
||
commands.append(formatted_command)
|
||
full_command = " && ".join(commands)
|
||
self.process = subprocess.Popen(f'start cmd /k "{full_command}"', shell=True)
|
||
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"执行命令时出错:\n{str(e)}")
|
||
|
||
def open_ply_file(self):
|
||
"""打开output目录中的pts.ply文件"""
|
||
ply_file = os.path.join(OUTPUT_DIR, "pts.ply")
|
||
|
||
if os.path.exists(ply_file):
|
||
try:
|
||
subprocess.Popen([MESHLAB_PATH, ply_file])
|
||
messagebox.showinfo("成功", f"已用MeshLab打开文件: {ply_file}")
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"打开文件时出错:\n{str(e)}")
|
||
else:
|
||
messagebox.showwarning("警告", f"未找到文件: {ply_file}")
|
||
|
||
def close_command_prompt(self):
|
||
try:
|
||
if self.process:
|
||
self.process.terminate()
|
||
|
||
subprocess.Popen('taskkill /f /im cmd.exe', shell=True,
|
||
stdout=subprocess.PIPE, stderr=subprocess.PIPE,
|
||
creationflags=subprocess.CREATE_NO_WINDOW)
|
||
except Exception as e:
|
||
print(f"关闭命令行时出错: {e}")
|
||
|
||
def mark_as_passed(self):
|
||
self._mark_file("passed_files.txt")
|
||
|
||
def mark_as_rejected(self):
|
||
self._mark_file("rejected_files.txt")
|
||
|
||
def _mark_file(self, filename):
|
||
if not self.folders:
|
||
return
|
||
|
||
current_path = self.folders[self.current_index]
|
||
folder_name = os.path.basename(current_path)
|
||
|
||
try:
|
||
if not os.path.exists(filename):
|
||
with open(filename, 'w', encoding='utf-8') as f:
|
||
pass
|
||
|
||
with open(filename, 'a', encoding='utf-8') as f:
|
||
if os.path.getsize(filename) == 0:
|
||
f.write(f"{folder_name}\n")
|
||
else:
|
||
needs_newline = True
|
||
with open(filename, 'rb+') as f_check:
|
||
f_check.seek(-1, os.SEEK_END)
|
||
last_char = f_check.read(1)
|
||
needs_newline = (last_char != b'\n')
|
||
|
||
if needs_newline:
|
||
f.write('\n')
|
||
f.write(f"{folder_name}\n")
|
||
|
||
status = "通过" if filename == "passed_files.txt" else "不通过"
|
||
messagebox.showinfo("成功", f"已标记为{status}: {folder_name}")
|
||
|
||
self.folders.pop(self.current_index)
|
||
|
||
if self.current_index >= len(self.folders):
|
||
if len(self.folders) > 0:
|
||
self.current_index = len(self.folders) - 1
|
||
else:
|
||
messagebox.showinfo("完成", "所有文件夹已处理完毕！")
|
||
self.update_display()
|
||
return
|
||
|
||
self.update_display()
|
||
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"写入文件时出错:\n{str(e)}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
root = tk.Tk()
|
||
app = FolderPlayerApp(root)
|
||
root.mainloop() |