439 lines
16 KiB
Python
439 lines
16 KiB
Python
|
import subprocess
|
|||
|
|
import psutil
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import json
|
|||
|
|
import socket
|
|||
|
|
import time
|
|||
|
|
import re
|
|||
|
|
from loguru import logger
|
|||
|
|
from PySide6.QtWidgets import (QApplication, QMainWindow, QPushButton, QMessageBox,
|
|||
|
|
QVBoxLayout, QWidget, QLabel, QGroupBox, QGridLayout,
|
|||
|
|
QStatusBar, QFrame, QDialog, QLineEdit, QFormLayout,
|
|||
|
|
QDialogButtonBox, QFileDialog, QHBoxLayout, QComboBox,
|
|||
|
|
QSpinBox, QCheckBox)
|
|||
|
|
from PySide6.QtCore import Qt, QThread, Signal
|
|||
|
|
from PySide6.QtGui import QFont
|
|||
|
|
|
|||
|
|
logger.add(sys.stderr, format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {message}", level="INFO")
|
|||
|
|
|
|||
|
|
CONFIG_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "dual_screen_config.json")
|
|||
|
|
|
|||
|
|
def load_config():
|
|||
|
|
if os.path.exists(CONFIG_FILE):
|
|||
|
|
with open(CONFIG_FILE, 'r', encoding='utf-8') as f:
|
|||
|
|
return json.load(f)
|
|||
|
|
return {
|
|||
|
|
"server_ip": "192.168.1.100",
|
|||
|
|
"mediamtx_path": r"D:\ScreenCast\mediamtx\mediamtx.exe",
|
|||
|
|
"ffmpeg_path": r"D:\ScreenCast\ffmpeg\bin\ffmpeg.exe",
|
|||
|
|
"front_stream_path": "front",
|
|||
|
|
"back_stream_path": "back"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def save_config(config):
|
|||
|
|
with open(CONFIG_FILE, 'w', encoding='utf-8') as f:
|
|||
|
|
json.dump(config, f, ensure_ascii=False, indent=4)
|
|||
|
|
logger.info(f"配置已保存到 {CONFIG_FILE}")
|
|||
|
|
|
|||
|
|
def get_displays():
|
|||
|
|
displays = []
|
|||
|
|
try:
|
|||
|
|
result = subprocess.run(
|
|||
|
|
['powershell', '-Command',
|
|||
|
|
'Add-Type -AssemblyName System.Windows.Forms; '
|
|||
|
|
'[System.Windows.Forms.Screen]::AllScreens | ForEach-Object { '
|
|||
|
|
'$dev = $_.DeviceName; '
|
|||
|
|
'$w = $_.Bounds.Width; '
|
|||
|
|
'$h = $_.Bounds.Height; '
|
|||
|
|
'$x = $_.Bounds.X; '
|
|||
|
|
'$y = $_.Bounds.Y; '
|
|||
|
|
'$isPrimary = $_.Primary; '
|
|||
|
|
'Write-Output "$dev|$w|$h|$x|$y|$isPrimary" }'],
|
|||
|
|
capture_output=True, text=True, timeout=10
|
|||
|
|
)
|
|||
|
|
for line in result.stdout.strip().split('\n'):
|
|||
|
|
if line:
|
|||
|
|
parts = line.split('|')
|
|||
|
|
if len(parts) >= 6:
|
|||
|
|
displays.append({
|
|||
|
|
'name': parts[0],
|
|||
|
|
'width': int(parts[1]),
|
|||
|
|
'height': int(parts[2]),
|
|||
|
|
'x': int(parts[3]),
|
|||
|
|
'y': int(parts[4]),
|
|||
|
|
'primary': parts[5] == 'True'
|
|||
|
|
})
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"获取显示器信息失败: {e}")
|
|||
|
|
return displays
|
|||
|
|
|
|||
|
|
class ConfigDialog(QDialog):
|
|||
|
|
def __init__(self, config, parent=None):
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self.config = config
|
|||
|
|
self.setWindowTitle("设置")
|
|||
|
|
self.setFixedSize(450, 280)
|
|||
|
|
|
|||
|
|
layout = QFormLayout(self)
|
|||
|
|
|
|||
|
|
self.server_ip_edit = QLineEdit(config["server_ip"])
|
|||
|
|
self.mediamtx_path_edit = QLineEdit(config["mediamtx_path"])
|
|||
|
|
self.ffmpeg_path_edit = QLineEdit(config["ffmpeg_path"])
|
|||
|
|
self.front_stream_edit = QLineEdit(config["front_stream_path"])
|
|||
|
|
self.back_stream_edit = QLineEdit(config["back_stream_path"])
|
|||
|
|
|
|||
|
|
self.mediamtx_btn = QPushButton("浏览...")
|
|||
|
|
self.mediamtx_btn.clicked.connect(lambda: self.browse("mediamtx"))
|
|||
|
|
self.ffmpeg_btn = QPushButton("浏览...")
|
|||
|
|
self.ffmpeg_btn.clicked.connect(lambda: self.browse("ffmpeg"))
|
|||
|
|
|
|||
|
|
path_layout = QGridLayout()
|
|||
|
|
path_layout.addWidget(self.mediamtx_path_edit, 0, 0)
|
|||
|
|
path_layout.addWidget(self.mediamtx_btn, 0, 1)
|
|||
|
|
path_layout.addWidget(self.ffmpeg_path_edit, 1, 0)
|
|||
|
|
path_layout.addWidget(self.ffmpeg_btn, 1, 1)
|
|||
|
|
|
|||
|
|
layout.addRow("服务器IP:", self.server_ip_edit)
|
|||
|
|
layout.addRow("MediaMTX路径:", self.mediamtx_path_edit)
|
|||
|
|
layout.addRow("FFmpeg路径:", self.ffmpeg_path_edit)
|
|||
|
|
layout.addRow("前置屏幕流名:", self.front_stream_edit)
|
|||
|
|
layout.addRow("后置屏幕流名:", self.back_stream_edit)
|
|||
|
|
|
|||
|
|
buttons = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
|
|||
|
|
buttons.accepted.connect(self.accept)
|
|||
|
|
buttons.rejected.connect(self.reject)
|
|||
|
|
layout.addRow(buttons)
|
|||
|
|
|
|||
|
|
def browse(self, field):
|
|||
|
|
if field == "mediamtx":
|
|||
|
|
path = QFileDialog.getOpenFileName(self, "选择MediaMTX", "", "Executable (*.exe)")[0]
|
|||
|
|
if path:
|
|||
|
|
self.mediamtx_path_edit.setText(path)
|
|||
|
|
elif field == "ffmpeg":
|
|||
|
|
path = QFileDialog.getOpenFileName(self, "选择FFmpeg", "", "Executable (*.exe)")[0]
|
|||
|
|
if path:
|
|||
|
|
self.ffmpeg_path_edit.setText(path)
|
|||
|
|
|
|||
|
|
def get_config(self):
|
|||
|
|
return {
|
|||
|
|
"server_ip": self.server_ip_edit.text().strip(),
|
|||
|
|
"mediamtx_path": self.mediamtx_path_edit.text().strip(),
|
|||
|
|
"ffmpeg_path": self.ffmpeg_path_edit.text().strip(),
|
|||
|
|
"front_stream_path": self.front_stream_edit.text().strip() or "front",
|
|||
|
|
"back_stream_path": self.back_stream_edit.text().strip() or "back"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
class ConnectionChecker(QThread):
|
|||
|
|
status_update = Signal(dict)
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
while True:
|
|||
|
|
cfg = self.config
|
|||
|
|
status = {
|
|||
|
|
"mediamtx": self.check_mediamtx(cfg["mediamtx_path"]),
|
|||
|
|
"ffmpeg": self.check_ffmpeg(cfg["ffmpeg_path"]),
|
|||
|
|
"server": self.check_server(cfg["server_ip"]),
|
|||
|
|
"server_port": self.check_port(cfg["server_ip"], 8554),
|
|||
|
|
}
|
|||
|
|
self.status_update.emit(status)
|
|||
|
|
time.sleep(3)
|
|||
|
|
|
|||
|
|
def set_config(self, config):
|
|||
|
|
self.config = config
|
|||
|
|
|
|||
|
|
def check_mediamtx(self, path):
|
|||
|
|
for proc in psutil.process_iter(['name']):
|
|||
|
|
if proc.info['name'] == 'mediamtx.exe':
|
|||
|
|
return True
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def check_ffmpeg(self, path):
|
|||
|
|
return os.path.exists(path)
|
|||
|
|
|
|||
|
|
def check_server(self, ip):
|
|||
|
|
try:
|
|||
|
|
socket.create_connection((ip, 8554), timeout=2)
|
|||
|
|
return True
|
|||
|
|
except:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def check_port(self, ip, port):
|
|||
|
|
try:
|
|||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|||
|
|
sock.settimeout(2)
|
|||
|
|
result = sock.connect_ex((ip, port))
|
|||
|
|
sock.close()
|
|||
|
|
return result == 0
|
|||
|
|
except:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
class MainWindow(QMainWindow):
|
|||
|
|
def __init__(self):
|
|||
|
|
super().__init__()
|
|||
|
|
self.config = load_config()
|
|||
|
|
self.processes = []
|
|||
|
|
self.setWindowTitle("双屏投屏控制")
|
|||
|
|
self.setFixedSize(600, 500)
|
|||
|
|
|
|||
|
|
self.status_bar = QStatusBar()
|
|||
|
|
self.setStatusBar(self.status_bar)
|
|||
|
|
|
|||
|
|
central_widget = QWidget()
|
|||
|
|
self.setCentralWidget(central_widget)
|
|||
|
|
main_layout = QVBoxLayout(central_widget)
|
|||
|
|
|
|||
|
|
title_layout = QHBoxLayout()
|
|||
|
|
title_layout.addStretch()
|
|||
|
|
self.btn_settings = QPushButton("⚙ 设置")
|
|||
|
|
self.btn_settings.setFixedSize(70, 30)
|
|||
|
|
self.btn_settings.clicked.connect(self.open_settings)
|
|||
|
|
title_layout.addWidget(self.btn_settings)
|
|||
|
|
|
|||
|
|
title_label = QLabel("双屏投屏系统")
|
|||
|
|
title_font = QFont("Microsoft YaHei", 14, QFont.Bold)
|
|||
|
|
title_label.setFont(title_font)
|
|||
|
|
title_label.setAlignment(Qt.AlignCenter)
|
|||
|
|
|
|||
|
|
title_main_layout = QVBoxLayout()
|
|||
|
|
title_main_layout.addLayout(title_layout)
|
|||
|
|
title_main_layout.addWidget(title_label)
|
|||
|
|
main_layout.addLayout(title_main_layout)
|
|||
|
|
|
|||
|
|
self.displays = get_displays()
|
|||
|
|
display_info = QLabel(f"检测到 {len(self.displays)} 个显示器")
|
|||
|
|
display_info.setAlignment(Qt.AlignCenter)
|
|||
|
|
main_layout.addWidget(display_info)
|
|||
|
|
|
|||
|
|
self.display_group = QGroupBox("显示器信息")
|
|||
|
|
display_layout = QGridLayout()
|
|||
|
|
|
|||
|
|
self.display_labels = []
|
|||
|
|
for i, display in enumerate(self.displays):
|
|||
|
|
label = QLabel(f"显示器{i+1}: {display['width']}x{display['height']} "
|
|||
|
|
f"(@ {display['x']},{display['y']}) "
|
|||
|
|
f"{'主屏' if display['primary'] else ''}")
|
|||
|
|
label.setStyleSheet("padding: 5px;")
|
|||
|
|
display_layout.addWidget(label, i, 0)
|
|||
|
|
self.display_labels.append(label)
|
|||
|
|
|
|||
|
|
self.display_group.setLayout(display_layout)
|
|||
|
|
main_layout.addWidget(self.display_group)
|
|||
|
|
|
|||
|
|
config_group = QGroupBox("本地配置")
|
|||
|
|
config_layout = QGridLayout()
|
|||
|
|
|
|||
|
|
self.mediamtx_label = QLabel("MediaMTX: 未检测")
|
|||
|
|
self.ffmpeg_label = QLabel("FFmpeg: 未检测")
|
|||
|
|
self.server_label = QLabel("服务器连接: 未检测")
|
|||
|
|
|
|||
|
|
self.mediamtx_indicator = QLabel("●")
|
|||
|
|
self.mediamtx_indicator.setFixedWidth(20)
|
|||
|
|
self.ffmpeg_indicator = QLabel("●")
|
|||
|
|
self.ffmpeg_indicator.setFixedWidth(20)
|
|||
|
|
self.server_indicator = QLabel("●")
|
|||
|
|
self.server_indicator.setFixedWidth(20)
|
|||
|
|
|
|||
|
|
config_layout.addWidget(self.mediamtx_indicator, 0, 0)
|
|||
|
|
config_layout.addWidget(self.mediamtx_label, 0, 1)
|
|||
|
|
config_layout.addWidget(self.ffmpeg_indicator, 1, 0)
|
|||
|
|
config_layout.addWidget(self.ffmpeg_label, 1, 1)
|
|||
|
|
config_layout.addWidget(self.server_indicator, 2, 0)
|
|||
|
|
config_layout.addWidget(self.server_label, 2, 1)
|
|||
|
|
|
|||
|
|
config_group.setLayout(config_layout)
|
|||
|
|
main_layout.addWidget(config_group)
|
|||
|
|
|
|||
|
|
self.info_group = QGroupBox("流信息")
|
|||
|
|
self.update_info_group()
|
|||
|
|
main_layout.addWidget(self.info_group)
|
|||
|
|
|
|||
|
|
btn_layout = QHBoxLayout()
|
|||
|
|
|
|||
|
|
self.btn_front = QPushButton("🎬 投送前置屏幕")
|
|||
|
|
self.btn_front.setFixedHeight(45)
|
|||
|
|
self.btn_front.setFont(QFont("Microsoft YaHei", 10))
|
|||
|
|
self.btn_front.setStyleSheet("""
|
|||
|
|
QPushButton {
|
|||
|
|
background-color: #1976D2;
|
|||
|
|
color: white;
|
|||
|
|
border: none;
|
|||
|
|
border-radius: 8px;
|
|||
|
|
font-weight: bold;
|
|||
|
|
}
|
|||
|
|
QPushButton:hover {
|
|||
|
|
background-color: #1565C0;
|
|||
|
|
}
|
|||
|
|
""")
|
|||
|
|
self.btn_front.clicked.connect(lambda: self.push_screen("front"))
|
|||
|
|
|
|||
|
|
self.btn_back = QPushButton("🎬 投送后置屏幕")
|
|||
|
|
self.btn_back.setFixedHeight(45)
|
|||
|
|
self.btn_back.setFont(QFont("Microsoft YaHei", 10))
|
|||
|
|
self.btn_back.setStyleSheet("""
|
|||
|
|
QPushButton {
|
|||
|
|
background-color: #388E3C;
|
|||
|
|
color: white;
|
|||
|
|
border: none;
|
|||
|
|
border-radius: 8px;
|
|||
|
|
font-weight: bold;
|
|||
|
|
}
|
|||
|
|
QPushButton:hover {
|
|||
|
|
background-color: #2E7D32;
|
|||
|
|
}
|
|||
|
|
""")
|
|||
|
|
self.btn_back.clicked.connect(lambda: self.push_screen("back"))
|
|||
|
|
|
|||
|
|
btn_layout.addWidget(self.btn_front)
|
|||
|
|
btn_layout.addWidget(self.btn_back)
|
|||
|
|
main_layout.addLayout(btn_layout)
|
|||
|
|
|
|||
|
|
self.btn_stop = QPushButton("⏹ 停止所有推流")
|
|||
|
|
self.btn_stop.setFixedHeight(40)
|
|||
|
|
self.btn_stop.setFont(QFont("Microsoft YaHei", 10))
|
|||
|
|
self.btn_stop.setStyleSheet("""
|
|||
|
|
QPushButton {
|
|||
|
|
background-color: #D32F2F;
|
|||
|
|
color: white;
|
|||
|
|
border: none;
|
|||
|
|
border-radius: 8px;
|
|||
|
|
font-weight: bold;
|
|||
|
|
}
|
|||
|
|
QPushButton:hover {
|
|||
|
|
background-color: #C62828;
|
|||
|
|
}
|
|||
|
|
""")
|
|||
|
|
self.btn_stop.clicked.connect(self.stop_all)
|
|||
|
|
main_layout.addWidget(self.btn_stop)
|
|||
|
|
|
|||
|
|
self.log_label = QLabel("日志: 等待操作...")
|
|||
|
|
self.log_label.setWordWrap(True)
|
|||
|
|
self.log_label.setFrameShape(QFrame.Shape.StyledPanel)
|
|||
|
|
self.log_label.setFixedHeight(50)
|
|||
|
|
main_layout.addWidget(self.log_label)
|
|||
|
|
|
|||
|
|
self.checker = ConnectionChecker()
|
|||
|
|
self.checker.set_config(self.config)
|
|||
|
|
self.checker.status_update.connect(self.update_status)
|
|||
|
|
self.checker.start()
|
|||
|
|
|
|||
|
|
logger.info("双屏投屏控制界面已初始化")
|
|||
|
|
|
|||
|
|
def update_info_group(self):
|
|||
|
|
cfg = self.config
|
|||
|
|
if self.info_group.layout():
|
|||
|
|
QWidget().setLayout(self.info_group.layout())
|
|||
|
|
info_layout = QGridLayout()
|
|||
|
|
info_layout.addWidget(QLabel("服务器IP:"), 0, 0)
|
|||
|
|
info_layout.addWidget(QLabel(cfg["server_ip"]), 0, 1)
|
|||
|
|
info_layout.addWidget(QLabel("前置屏幕流:"), 1, 0)
|
|||
|
|
info_layout.addWidget(QLabel(f"rtsp://{cfg['server_ip']}:8554/{cfg['front_stream_path']}"), 1, 1)
|
|||
|
|
info_layout.addWidget(QLabel("后置屏幕流:"), 2, 0)
|
|||
|
|
info_layout.addWidget(QLabel(f"rtsp://{cfg['server_ip']}:8554/{cfg['back_stream_path']}"), 2, 1)
|
|||
|
|
self.info_group.setLayout(info_layout)
|
|||
|
|
|
|||
|
|
def open_settings(self):
|
|||
|
|
dialog = ConfigDialog(self.config, self)
|
|||
|
|
if dialog.exec():
|
|||
|
|
self.config = dialog.get_config()
|
|||
|
|
save_config(self.config)
|
|||
|
|
self.checker.set_config(self.config)
|
|||
|
|
self.update_info_group()
|
|||
|
|
self.log_label.setText("日志: 配置已更新")
|
|||
|
|
|
|||
|
|
def update_status(self, status):
|
|||
|
|
if status["mediamtx"]:
|
|||
|
|
self.mediamtx_indicator.setStyleSheet("color: green; font-size: 16px;")
|
|||
|
|
self.mediamtx_label.setText("MediaMTX: ✓ 运行中")
|
|||
|
|
else:
|
|||
|
|
self.mediamtx_indicator.setStyleSheet("color: red; font-size: 16px;")
|
|||
|
|
self.mediamtx_label.setText("MediaMTX: ✗ 未运行")
|
|||
|
|
|
|||
|
|
if status["ffmpeg"]:
|
|||
|
|
self.ffmpeg_indicator.setStyleSheet("color: green; font-size: 16px;")
|
|||
|
|
self.ffmpeg_label.setText("FFmpeg: ✓ 已安装")
|
|||
|
|
else:
|
|||
|
|
self.ffmpeg_indicator.setStyleSheet("color: red; font-size: 16px;")
|
|||
|
|
self.ffmpeg_label.setText("FFmpeg: ✗ 未找到")
|
|||
|
|
|
|||
|
|
if status["server"]:
|
|||
|
|
self.server_indicator.setStyleSheet("color: green; font-size: 16px;")
|
|||
|
|
self.server_label.setText("服务器连接: ✓ 可连接")
|
|||
|
|
else:
|
|||
|
|
self.server_indicator.setStyleSheet("color: orange; font-size: 16px;")
|
|||
|
|
self.server_label.setText("服务器连接: ✗ 连接失败")
|
|||
|
|
|
|||
|
|
def push_screen(self, screen_type):
|
|||
|
|
cfg = self.config
|
|||
|
|
displays = self.displays
|
|||
|
|
|
|||
|
|
if len(displays) < 2:
|
|||
|
|
QMessageBox.warning(self, "错误", "未检测到足够的显示器！")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if screen_type == "front":
|
|||
|
|
display = displays[0]
|
|||
|
|
stream_path = cfg["front_stream_path"]
|
|||
|
|
screen_name = "前置屏幕"
|
|||
|
|
else:
|
|||
|
|
display = displays[1] if len(displays) > 1 else displays[0]
|
|||
|
|
stream_path = cfg["back_stream_path"]
|
|||
|
|
screen_name = "后置屏幕"
|
|||
|
|
|
|||
|
|
logger.info(f"开始投送{screen_name}: {display['width']}x{display['height']}")
|
|||
|
|
|
|||
|
|
cmd = [
|
|||
|
|
cfg["ffmpeg_path"],
|
|||
|
|
"-f", "gdigrab",
|
|||
|
|
"-offset_x", str(display['x']),
|
|||
|
|
"-offset_y", str(display['y']),
|
|||
|
|
"-video_size", f"{display['width']}x{display['height']}",
|
|||
|
|
"-framerate", "30",
|
|||
|
|
"-i", "desktop",
|
|||
|
|
"-c:v", "libx264",
|
|||
|
|
"-preset", "ultrafast",
|
|||
|
|
"-tune", "zerolatency",
|
|||
|
|
"-f", "rtsp",
|
|||
|
|
f"rtsp://{cfg['server_ip']}:8554/{stream_path}"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
logger.info(f"执行FFmpeg命令: {' '.join(cmd)}")
|
|||
|
|
process = subprocess.Popen(cmd, creationflags=subprocess.CREATE_NEW_CONSOLE)
|
|||
|
|
self.processes.append(process)
|
|||
|
|
|
|||
|
|
self.log_label.setText(f"日志: {screen_name}正在投送... "
|
|||
|
|
f"访问 http://{cfg['server_ip']}:8889/{stream_path}")
|
|||
|
|
QMessageBox.information(self, "提示",
|
|||
|
|
f"{screen_name}投送已启动！\n\n"
|
|||
|
|
f"观众访问：http://{cfg['server_ip']}:8889/{stream_path}")
|
|||
|
|
|
|||
|
|
def stop_all(self):
|
|||
|
|
for proc in self.processes:
|
|||
|
|
try:
|
|||
|
|
proc.terminate()
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
self.processes.clear()
|
|||
|
|
|
|||
|
|
for proc in psutil.process_iter(['name']):
|
|||
|
|
try:
|
|||
|
|
if proc.info['name'] == 'ffmpeg.exe':
|
|||
|
|
proc.terminate()
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
self.log_label.setText("日志: 所有推流已停止")
|
|||
|
|
logger.info("所有推流已停止")
|
|||
|
|
|
|||
|
|
def closeEvent(self, event):
|
|||
|
|
self.stop_all()
|
|||
|
|
self.checker.terminate()
|
|||
|
|
event.accept()
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
logger.info("应用程序启动")
|
|||
|
|
app = QApplication(sys.argv)
|
|||
|
|
app.setStyle("Fusion")
|
|||
|
|
window = MainWindow()
|
|||
|
|
window.show()
|
|||
|
|
sys.exit(app.exec())
|