From bdf1358b0a8936e0ee5d4743640bbab020878497 Mon Sep 17 00:00:00 2001
From: xiaji <rembme@163.com>
Date: Fri, 20 Mar 2026 10:38:39 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=8F=8C=E5=B1=8F=E6=8A=95?=
 =?UTF-8?q?=E5=B1=8F=E8=84=9A=E6=9C=ACdual=5Fscreen=5Fpush.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dual_screen_push.py | 439 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 439 insertions(+)
 create mode 100644 dual_screen_push.py

diff --git a/dual_screen_push.py b/dual_screen_push.py
new file mode 100644
index 0000000..51df6c0
--- /dev/null
+++ b/dual_screen_push.py
@@ -0,0 +1,439 @@
+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())
\ No newline at end of file