50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
|
import subprocess
|
|||
|
|
import psutil
|
|||
|
|
import os
|
|||
|
|
import time
|
|||
|
|
import tkinter as tk
|
|||
|
|
from tkinter import messagebox
|
|||
|
|
|
|||
|
|
# 配置项（仅需改这里的服务器IP）
|
|||
|
|
SERVER_IP = "192.168.1.100" # MediaMTX服务器内网IP
|
|||
|
|
MEDIAMTX_PATH = r"D:\ScreenCast\mediamtx\mediamtx.exe"
|
|||
|
|
FFMPEG_PATH = r"D:\ScreenCast\ffmpeg\bin\ffmpeg.exe"
|
|||
|
|
STREAM_PATH = "screen" # 投屏流路径（模式1/2）
|
|||
|
|
|
|||
|
|
# 检查MediaMTX是否运行
|
|||
|
|
def check_mediamtx_running():
|
|||
|
|
for proc in psutil.process_iter(['name']):
|
|||
|
|
if proc.info['name'] == 'mediamtx.exe':
|
|||
|
|
return True
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 启动MediaMTX（如果未运行）
|
|||
|
|
def start_mediamtx():
|
|||
|
|
if not check_mediamtx_running():
|
|||
|
|
subprocess.Popen([MEDIAMTX_PATH], cwd=r"D:\ScreenCast\mediamtx", creationflags=subprocess.CREATE_NEW_CONSOLE)
|
|||
|
|
time.sleep(3) # 等待启动
|
|||
|
|
messagebox.showinfo("提示", "MediaMTX已启动！")
|
|||
|
|
|
|||
|
|
# 屏幕推流（全屏）
|
|||
|
|
def push_full_screen():
|
|||
|
|
start_mediamtx()
|
|||
|
|
# 封装FFmpeg命令，低延迟推流
|
|||
|
|
cmd = [
|
|||
|
|
FFMPEG_PATH,
|
|||
|
|
"-f", "gdigrab", "-framerate", "30", "-i", "desktop",
|
|||
|
|
"-c:v", "libx264", "-preset", "ultrafast", "-tune", "zerolatency",
|
|||
|
|
"-f", "rtsp", f"rtsp://{SERVER_IP}:8554/{STREAM_PATH}"
|
|||
|
|
]
|
|||
|
|
subprocess.Popen(cmd, creationflags=subprocess.CREATE_NEW_CONSOLE)
|
|||
|
|
messagebox.showinfo("提示", f"全屏投屏已启动！接收端可打开浏览器访问：\n http://{SERVER_IP}:8889/webrtc.html?src=screen ")
|
|||
|
|
|
|||
|
|
# 主界面（点击按钮即可）
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
root = tk.Tk()
|
|||
|
|
root.title("投屏源控制")
|
|||
|
|
root.geometry("300x150")
|
|||
|
|
|
|||
|
|
btn1 = tk.Button(root, text="一键全屏投屏", command=push_full_screen, width=20, height=2)
|
|||
|
|
btn1.pack(pady=20)
|
|||
|
|
|
|||
|
|
root.mainloop()
|