diff --git a/src/Screenshot/Screenshot_Gui.py b/src/Screenshot/Screenshot_Gui.py
index 0025695..326f626 100644
--- a/src/Screenshot/Screenshot_Gui.py
+++ b/src/Screenshot/Screenshot_Gui.py
@@ -20,7 +20,6 @@ from PyQt5.QtCore import QRect, Qt, QPoint, QMetaObject, QThread
from PyQt5.QtGui import QIcon, QFont, QColor, QImage, QPixmap, QPen, QPainter
from PyQt5.QtWidgets import (QMainWindow, QApplication, QDesktopWidget, QWidget, QFrame,
QLabel, QPushButton, QComboBox, QCheckBox, QSpinBox, QDialog, QFileDialog, QMessageBox)
-from py7zr import pack_7zarchive, unpack_7zarchive
from Screenshot.Audio_record import AudioRecord
from Screenshot.Screenshot_record import Screenshot, file_path
@@ -74,7 +73,7 @@ class Thread_screenshot(QThread):
self.screen.terminate()
-class Ui_MainWindow1(QMainWindow):
+class Ui_MainWindow2(QMainWindow):
"""主体窗口设置"""
def __init__(self):
diff --git a/src/Screenshot/Screenshot_record.py b/src/Screenshot/Screenshot_record.py
index a19e253..afe8094 100644
--- a/src/Screenshot/Screenshot_record.py
+++ b/src/Screenshot/Screenshot_record.py
@@ -4,14 +4,13 @@
import sys
from pathlib import Path
from time import sleep
-
import ffmpeg
def file_path(filename):
relative_path = Path('resource').joinpath(filename)
if getattr(sys, 'frozen', False):
- base_path = Path(sys._MEIPASS)
+ base_path = Path(sys.MEIPASS)
else:
base_path = Path('.').absolute()
return Path(base_path).joinpath(relative_path).as_posix()
diff --git a/src/Screenshot/__pycache__/Audio_record.cpython-39.pyc b/src/Screenshot/__pycache__/Audio_record.cpython-39.pyc
index b172c7d..037c86a 100644
Binary files a/src/Screenshot/__pycache__/Audio_record.cpython-39.pyc and b/src/Screenshot/__pycache__/Audio_record.cpython-39.pyc differ
diff --git a/src/Screenshot/__pycache__/Screenshot_Gui.cpython-39.pyc b/src/Screenshot/__pycache__/Screenshot_Gui.cpython-39.pyc
deleted file mode 100644
index a922dc6..0000000
Binary files a/src/Screenshot/__pycache__/Screenshot_Gui.cpython-39.pyc and /dev/null differ
diff --git a/src/Screenshot/__pycache__/Screenshot_record.cpython-39.pyc b/src/Screenshot/__pycache__/Screenshot_record.cpython-39.pyc
index 826f23e..64d31c2 100644
Binary files a/src/Screenshot/__pycache__/Screenshot_record.cpython-39.pyc and b/src/Screenshot/__pycache__/Screenshot_record.cpython-39.pyc differ
diff --git a/src/qt/.idea/.gitignore b/src/qt/.idea/.gitignore
new file mode 100644
index 0000000..35410ca
--- /dev/null
+++ b/src/qt/.idea/.gitignore
@@ -0,0 +1,8 @@
+# 默认忽略的文件
+/shelf/
+/workspace.xml
+# 基于编辑器的 HTTP 客户端请求
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml
diff --git a/src/qt/.idea/inspectionProfiles/profiles_settings.xml b/src/qt/.idea/inspectionProfiles/profiles_settings.xml
new file mode 100644
index 0000000..105ce2d
--- /dev/null
+++ b/src/qt/.idea/inspectionProfiles/profiles_settings.xml
@@ -0,0 +1,6 @@
+
+
+
+
+
+
\ No newline at end of file
diff --git a/src/qt/.idea/misc.xml b/src/qt/.idea/misc.xml
new file mode 100644
index 0000000..d6c20c7
--- /dev/null
+++ b/src/qt/.idea/misc.xml
@@ -0,0 +1,4 @@
+
+
+
+
\ No newline at end of file
diff --git a/src/qt/.idea/modules.xml b/src/qt/.idea/modules.xml
new file mode 100644
index 0000000..45b9f23
--- /dev/null
+++ b/src/qt/.idea/modules.xml
@@ -0,0 +1,8 @@
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/src/qt/.idea/qt.iml b/src/qt/.idea/qt.iml
new file mode 100644
index 0000000..72d96af
--- /dev/null
+++ b/src/qt/.idea/qt.iml
@@ -0,0 +1,14 @@
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/src/qt/.idea/vcs.xml b/src/qt/.idea/vcs.xml
new file mode 100644
index 0000000..b2bdec2
--- /dev/null
+++ b/src/qt/.idea/vcs.xml
@@ -0,0 +1,6 @@
+
+
+
+
+
+
\ No newline at end of file
diff --git a/src/qt/Mainwindow.py b/src/qt/Mainwindow.py
index ae72b35..f560ca1 100644
--- a/src/qt/Mainwindow.py
+++ b/src/qt/Mainwindow.py
@@ -11,6 +11,7 @@
import sys
from PyQt5.QtWidgets import *
from findpath_UI import *
+from Screenshot.Screenshot_Gui import *
import qdarkstyle
class Ui_MainWindow(object):
@@ -190,11 +191,18 @@ class Ui_MainWindow(object):
icon9.addPixmap(QtGui.QPixmap("window_png/路径分析-1.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.actionicon_2.setIcon(icon9)
self.actionicon_2.setObjectName("actionicon_2")
+ self.actionicon_3 = QtWidgets.QAction(MainWindow)
+ icon10 = QtGui.QIcon()
+ icon10.addPixmap(QtGui.QPixmap("window_png/录像.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)
+ self.actionicon_3.setIcon(icon10)
+ self.actionicon_3.setObjectName("actionicon_3")
self.menubar.addAction(self.menu.menuAction())
self.menubar.addAction(self.menu_2.menuAction())
self.toolBar.addAction(self.actionicon)
self.toolBar.addAction(self.actionicon_2)
+ self.toolBar.addAction(self.actionicon_3)
self.actionicon_2.triggered.connect(lambda: self.path())
+ self.actionicon_3.triggered.connect(lambda: self.video())
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
@@ -204,6 +212,10 @@ class Ui_MainWindow(object):
ui.setupUi(self.MainWindow1)
self.MainWindow1.show()
+ def video(self):
+ self.ui = Ui_MainWindow2()
+ self.ui.show()
+
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "无人机路径生成系统"))
diff --git a/src/qt/Screenshot/Audio_record.py b/src/qt/Screenshot/Audio_record.py
new file mode 100644
index 0000000..89f3561
--- /dev/null
+++ b/src/qt/Screenshot/Audio_record.py
@@ -0,0 +1,133 @@
+import time
+import wave
+from pathlib import Path
+from threading import Thread
+
+from pyaudio import PyAudio, paInt16, paContinue, paComplete
+
+
+class AudioRecord(PyAudio):
+
+ def __init__(self, channels=2):
+ super().__init__()
+ self.chunk = 1024 # 每个缓冲区的帧数
+ self.format_sample = paInt16 # 采样位数
+ self.channels = channels # 声道： 1，单声道；2，双声道
+ self.fps = 44100 # 采样频率
+ self.input_dict = None
+ self.output_dict = None
+ self.stream = None
+ self.filename = '~test.wav'
+ self.wf = None
+ self.stop_flag = False
+ self.kill = False
+
+ def callback_input(self, in_data, frame_count, time_info, status):
+ """录制回调函数"""
+ self.wf.writeframes(in_data)
+ if not self.stop_flag:
+ return (in_data, paContinue)
+ else:
+ return (in_data, paComplete)
+
+ def callback_output(self, in_data, frame_count, time_info, status):
+ """播放回调函数"""
+ data = self.wf.readframes(frame_count)
+ return (data, paContinue)
+
+ def open_stream(self, name):
+ """打开录制流"""
+ input_device_index = self.get_device_index(name, True) if name else None
+ return self.open(format=self.format_sample,
+ channels=self.channels,
+ rate=self.fps,
+ frames_per_buffer=self.chunk,
+ input=True,
+ input_device_index=input_device_index,
+ stream_callback=self.callback_input
+ )
+
+ def audio_record_run(self, name=None):
+ """音频录制"""
+ self.wf = self.save_audio_file(self.filename)
+ self.stream = self.open_stream(name)
+ self.stream.start_stream()
+ while self.stream.is_active():
+ time.sleep(0.1)
+ self.wf.close()
+ if self.kill:
+ Path(self.filename).unlink()
+ self.duration = self.get_duration()
+ self.terminate_run()
+
+ def run(self, filename=None, name=None, record=True):
+ """音频录制启动"""
+ if record:
+ if filename:
+ self.filename = filename
+ thread_1 = Thread(target=self.audio_record_run, args=(name,))
+ else:
+ if not filename:
+ raise Exception('未输入音频文件名，不能播放，请输入后再试！')
+ thread_1 = Thread(target=self.read_audio, args=(filename, name,))
+ thread_1.start()
+
+ def read_audio(self, filename, name=None):
+ """音频播放"""
+ output_device_index = self.get_device_index(name, False) if name else None
+ with wave.open(filename, 'rb') as self.wf:
+ # 获取音频长度
+ self.duration = self.get_duration()
+ self.stream = self.open(format=self.get_format_from_width(self.wf.getsampwidth()),
+ channels=self.wf.getnchannels(),
+ rate=self.wf.getframerate(),
+ output=True,
+ output_device_index=output_device_index,
+ stream_callback=self.callback_output
+ )
+ self.stream.start_stream()
+ while self.stream.is_active():
+ time.sleep(0.1)
+ print(self.duration)
+ self.terminate_run()
+
+ def get_duration(self):
+ """获取音频时长"""
+ return round(self.wf.getnframes() / self.wf.getframerate(), 2)
+
+ def get_in_out_devices(self):
+ """获取系统输入输出设备"""
+ self.input_dict = {}
+ self.output_dict = {}
+ for i in range(self.get_device_count()):
+ devinfo = self.get_device_info_by_index(i)
+ if not devinfo['hostApi'] and int(devinfo['defaultSampleRate']) == self.fps \
+ and '映射器' not in devinfo['name']:
+ if devinfo['maxInputChannels']:
+ self.input_dict[devinfo['name'].split(' ')[0]] = i
+ elif devinfo['maxOutputChannels']:
+ self.output_dict[devinfo['name'].split(' ')[0]] = i
+
+ def get_device_index(self, name, inp=True):
+ """获取选定设备索引"""
+ if inp and self.input_dict:
+ return self.input_dict.get(name, -1)
+ elif not inp and self.output_dict:
+ return self.output_dict.get(name, -1)
+
+ def save_audio_file(self, filename):
+ """音频文件保存"""
+ wf = wave.open(filename, 'wb')
+ wf.setnchannels(self.channels)
+ wf.setsampwidth(self.get_sample_size(self.format_sample))
+ wf.setframerate(self.fps)
+ return wf
+
+ def terminate_run(self):
+ """结束流录制或流播放"""
+ if self.stream is not None:
+ self.stream.stop_stream()
+ self.stream.close()
+ self.stream = None
+ self.wf = None
+ self.terminate()
diff --git a/src/qt/Screenshot/PyAudio-0.2.11-cp37-cp37m-win_amd64.whl b/src/qt/Screenshot/PyAudio-0.2.11-cp37-cp37m-win_amd64.whl
new file mode 100644
index 0000000..72fbfc8
Binary files /dev/null and b/src/qt/Screenshot/PyAudio-0.2.11-cp37-cp37m-win_amd64.whl differ
diff --git a/src/qt/Screenshot/README.md b/src/qt/Screenshot/README.md
new file mode 100644
index 0000000..4aa3f83
--- /dev/null
+++ b/src/qt/Screenshot/README.md
@@ -0,0 +1,39 @@
+# 使用python动手做一个屏幕录制工具
+
+*更新日期：2022-04-20*
+> windows 10
+> python 3.7
+
+- Screeshot_Gui.exe [58.5M] [下载地址](https://github.com/lk-itween/FunnyCodeRepository/releases/download/Screenshot_Gui/Screenshot_Gui.exe)
+先将ffmpeg.exe压缩成7z格式，再一并打包进入exe文件中，且使用pyinstaller打包，生成文件较大
+
+- Screenshot_Gui.py
+pyqt5制作的屏幕录制窗口，窗口程序入口
+
+- Screenshot_record.py
+使用ffmpeg工具，利用python调用ffmpeg命令行来录制屏幕
+
+- Audio_record.py
+pyaudio读取可录制音频设备，设定需要使用的设备进行录制音频
+
+- Screenshot_record_with_cv2.py
+使用opencv-python录制视频并保存
+
+- Screenshot_test.py
+无窗口化测试联动屏幕录制及音频录制，保持音视频同时录制和同时停止录制
+
+- resource
+保存程序运行过程中需要使用到的程序或图片
+
+- requirements.txt
+已将所需模块放在requirements.txt中，可直接pip install -r requirements.txt下载安装。
+
+- PyAudio-0.2.11-cp37-cp37m-win_amd64.whl
+python3.7环境的PyAudio的轮子
+
+
+**ps:**
+*1. ffmpeg.exe 请至此https://www.gyan.dev/ffmpeg/builds/#release-builds 下载符合的版本，并解压将bin下的ffmpeg.exe移动到resource文件夹下*
+*2. 如果PyAudio包不能使用pip在线安装，python3.7环境请下载PyAudio-0.2.11-cp37-cp37m-win_amd64.whl至本地，在该路径下打开cmd窗口，执行pip install 该whl文件*
+*3. 其他python环境可在https://www.lfd.uci.edu/~gohlke/pythonlibs/#pyaudio 下自行安装对应版本的whl文件*
+
diff --git a/src/qt/Screenshot/Screenshot_Gui.py b/src/qt/Screenshot/Screenshot_Gui.py
new file mode 100644
index 0000000..326f626
--- /dev/null
+++ b/src/qt/Screenshot/Screenshot_Gui.py
@@ -0,0 +1,359 @@
+# -*- coding: utf-8 -*-
+
+# Form implementation generated from reading ui file 'Screenshot_GUI.ui'
+#
+# Created by: PyQt5 UI code generator 5.15.4
+#
+# WARNING: Any manual changes made to this file will be lost when pyuic5 is
+# run again. Do not edit this file unless you know what you are doing.
+import sys
+import os
+
+sys.path.append(os.path.dirname(os.path.abspath(__file__)) + "/../../PaddleClas-release-2.3")
+import shutil
+
+from datetime import datetime
+from pathlib import Path
+
+from PIL import ImageGrab
+from PyQt5.QtCore import QRect, Qt, QPoint, QMetaObject, QThread
+from PyQt5.QtGui import QIcon, QFont, QColor, QImage, QPixmap, QPen, QPainter
+from PyQt5.QtWidgets import (QMainWindow, QApplication, QDesktopWidget, QWidget, QFrame,
+ QLabel, QPushButton, QComboBox, QCheckBox, QSpinBox, QDialog, QFileDialog, QMessageBox)
+
+from Screenshot.Audio_record import AudioRecord
+from Screenshot.Screenshot_record import Screenshot, file_path
+
+
+def unpack_7zip_to_exe(filename):
+ file_basename = Path(file_path(f'{filename}.exe'))
+ if file_basename.exists():
+ return True
+ elif file_basename.with_suffix('.7z').exists():
+ shutil.unpack_archive(file_basename.with_suffix('.7z'), file_basename.parent)
+ return unpack_7zip_to_exe(file_basename.stem)
+ else:
+ return False
+
+class Thread_screenshot(QThread):
+
+ def __init__(self):
+ super().__init__()
+ self.offset_x = 0
+ self.offset_y = 0
+ self.draw_mouse = 1
+ self.filename = ''
+ self.screen = Screenshot()
+
+ def config_modify(self, **kwargs):
+ self.screen.width = kwargs.get('width', self.screen.width)
+ self.screen.height = kwargs.get('height', self.screen.height)
+ self.screen.fps = kwargs.get('fps', self.screen.fps)
+ self.offset_x = kwargs.get('offset_x', self.offset_x)
+ self.offset_y = kwargs.get('offset_y', self.offset_y)
+ self.draw_mouse = kwargs.get('draw_mouse', self.draw_mouse)
+ self.filename = kwargs.get('filename', self.filename)
+
+ def compose(self):
+ video = self.filename + '.mp4'
+ audio = self.filename + '.mp3'
+ self.screen.compose_audio(video, audio, video.replace('缓存', ''))
+ QThread.sleep(1)
+ self.stop()
+ Path(audio).unlink()
+ Path(video).unlink()
+
+ def run(self):
+ if not self.filename:
+ raise Exception('请输入filename!')
+ filename = self.filename + '.mp4'
+ self.screen.record(filename, self.offset_x, self.offset_y, self.draw_mouse)
+
+ def stop(self):
+ self.screen.terminate()
+
+
+class Ui_MainWindow2(QMainWindow):
+ """主体窗口设置"""
+
+ def __init__(self):
+ super().__init__()
+ self.setObjectName("MainWindow")
+ self.setWindowTitle('屏幕录制')
+ windowico = file_path('gui_svg.svg')
+ self.setWindowIcon(QIcon(windowico))
+ self.resize(512, 352)
+ self.setFixedSize(512, 352)
+ self._dir = '.'
+ self.fname = ''
+ self.offset_x, self.offset_y = 0, 0
+ self.screen_x, self.screen_y = ImageGrab.grab().size
+ self.audio = AudioRecord()
+ self.audio.get_in_out_devices()
+ self.device_name = None
+ self.record_audio = True if self.audio.input_dict else False
+ self.hotkey_start = False
+ self.screenshot = Thread_screenshot()
+ self.setupUi()
+ ffmpeg_exists = unpack_7zip_to_exe('ffmpeg')
+ if not ffmpeg_exists:
+ QMessageBox.warning(self, '请下载ffmpeg！', '请将ffmpeg.exe放在resource目录下再录制！')
+ self.show()
+
+ def setupUi(self):
+ self.centralwidget = QWidget(self)
+ self.centralwidget.setObjectName("centralwidget")
+ self.label0 = QLabel(self.centralwidget)
+ self.label0.setGeometry(QRect(0, 0, 421, 351))
+ self.label0.setObjectName("scrollArea")
+ # 添加边框
+ self.label0.setFrameShape(QFrame.Box)
+ self.label0.setFrameShadow(QFrame.Sunken)
+ self.get_screen_grab((0, 0, self.screen_x, self.screen_y))
+ self.label0.setAlignment(Qt.AlignCenter)
+ self.label_1 = QLabel(self.centralwidget)
+ font = QFont()
+ font.setBold(True)
+ self.label_1.setGeometry(QRect(430, 0, 85, 31))
+ self.label_1.setTextFormat(Qt.AutoText)
+ self.label_1.setFont(font)
+ self.label_1.setScaledContents(True)
+ self.label_1.setObjectName("屏幕录制器")
+ self.label_6 = QLabel(self.centralwidget)
+ self.label_6.setGeometry(QRect(430, 20, 85, 31))
+ self.label_6.setTextFormat(Qt.AutoText)
+ self.label_6.setScaledContents(True)
+ self.label_6.setObjectName("注释1")
+ self.checkbox = QCheckBox(self.centralwidget)
+ self.checkbox.setGeometry(QRect(430, 100, 85, 20))
+ self.checkbox.setChecked(True)
+ self.checkbox.stateChanged.connect(self.mouse_draw)
+ self.checkbox.setObjectName("录制鼠标")
+ self.comboBox = QComboBox(self.centralwidget)
+ self.comboBox.setGeometry(QRect(430, 200, 85, 20))
+ self.comboBox.setEditable(False)
+ self.comboBox.addItems(self.audio.input_dict.keys())
+ self.comboBox.addItems(['无'])
+ self.comboBox.setMaxVisibleItems(9)
+ self.comboBox.currentTextChanged.connect(self.change_combox)
+ self.comboBox.setObjectName("音频来源")
+ self.comboBox_3 = QComboBox(self.centralwidget)
+ self.comboBox_3.setGeometry(QRect(430, 260, 80, 20))
+ self.comboBox_3.addItems(['双声道', '单声道'])
+ self.comboBox_3.currentIndexChanged.connect(self.change_combox)
+ self.comboBox_3.setObjectName("声道设置")
+ self.pushButton = QPushButton(self.centralwidget)
+ self.pushButton.setGeometry(QRect(470, 320, 41, 31))
+ self.pushButton.clicked.connect(self.stop)
+ self.pushButton.setObjectName("停止")
+ self.pushButton_2 = QPushButton(self.centralwidget)
+ self.pushButton_2.setGeometry(QRect(430, 320, 41, 31))
+ self.pushButton_2.setObjectName("开始")
+ self.pushButton_2.clicked.connect(self.start)
+ self.pushButton_3 = QPushButton(self.centralwidget)
+ self.pushButton_3.setGeometry(QRect(430, 290, 80, 23))
+ self.pushButton_3.setObjectName("保存目录")
+ self.pushButton_3.clicked.connect(self.open_dirpath)
+ self.pushButton_4 = QPushButton(self.centralwidget)
+ self.pushButton_4.setGeometry(QRect(430, 70, 80, 23))
+ self.pushButton_4.setCheckable(True)
+ self.pushButton_4.setObjectName("Size")
+ self.pushButton_4.clicked.connect(self.get_screen_area)
+ self.spinBox = QSpinBox(self.centralwidget)
+ self.spinBox.setValue(15)
+ self.spinBox.setRange(10, 60)
+ self.spinBox.setGeometry(QRect(430, 145, 80, 20))
+ self.spinBox.valueChanged.connect(self.screen_fps_mofify)
+ self.spinBox.setObjectName("spinBox")
+ self.label_2 = QLabel(self.centralwidget)
+ self.label_2.setGeometry(QRect(430, 50, 54, 16))
+ self.label_2.setTextFormat(Qt.AutoText)
+ self.label_2.setObjectName("选择区域")
+ self.label_3 = QLabel(self.centralwidget)
+ self.label_3.setGeometry(QRect(430, 125, 54, 16))
+ self.label_3.setObjectName("帧率")
+ self.label_4 = QLabel(self.centralwidget)
+ self.label_4.setGeometry(QRect(430, 240, 54, 16))
+ self.label_4.setObjectName("声道设置")
+ self.label_5 = QLabel(self.centralwidget)
+ self.label_5.setGeometry(QRect(430, 180, 54, 16))
+ self.label_5.setObjectName("音频来源")
+ self.setCentralWidget(self.centralwidget)
+
+ self.retranslateUi()
+ QMetaObject.connectSlotsByName(self)
+
+ def retranslateUi(self):
+ self.pushButton.setText("停止")
+ self.pushButton_2.setText("开始")
+ self.pushButton_3.setText("保存目录")
+ self.pushButton_4.setText("Size*Size")
+ self.checkbox.setText("录制鼠标")
+ self.label_1.setText("屏幕录制器")
+ self.label_2.setText("选择区域：")
+ self.label_3.setText("帧率：")
+ self.label_4.setText("声道设置：")
+ self.label_5.setText("音频来源：")
+ self.label_6.setText("F7：开始/停止")
+
+ def keyPressEvent(self, event):
+ """监测键盘是否按下F3，如果已开始录制则停止，否则开始录制"""
+ if event.key() == Qt.Key_F7:
+ if not self.hotkey_start:
+ self.start()
+ else:
+ self.stop()
+
+ def change_combox(self, event):
+ sendername = self.sender().objectName()
+ if sendername == '声道设置':
+ self.channels = 2 if event == 0 else 1
+ elif sendername == '音频来源':
+ if event == '无':
+ self.record_audio = False
+ else:
+ self.record_audio = True
+ self.device_name = event
+
+ def get_screen_grab(self, crop_size):
+ # 获取屏幕截图并更新至self.label0中
+ screen = ImageGrab.grab().convert('RGBA')
+ screen = screen.crop(box=crop_size) # box=(left, upper, right, lower)
+ data = screen.tobytes("raw", "RGBA")
+ qim = QImage(data, screen.size[0], screen.size[1], QImage.Format_RGBA8888)
+ pix = QPixmap.fromImage(qim).scaled(self.label0.size(), aspectRatioMode=Qt.KeepAspectRatio)
+ pix.detach() # QPixmap为另开线程操作的，避免程序崩溃，使用detach方法使用一个QImage对象直到QPixmap对象销毁
+ self.label0.setPixmap(pix)
+ self.label0.repaint()
+ return screen.size
+
+ def get_screen_area(self):
+ # 获取截图区间
+ SizeScreen = MousePaint()
+ SizeScreen.exec_()
+ self.offset_x, self.offset_y = SizeScreen.lastpoint
+ end_x, end_y = SizeScreen.endpoint
+ self.screen_x, self.screen_y = self.get_screen_grab((self.offset_x, self.offset_y, end_x, end_y))
+ self.screenshot.config_modify(offset_x=self.offset_x, offset_y=self.offset_y,
+ width=self.screen_x, height=self.screen_y)
+ self.pushButton_4.setText(f'{self.screen_x}*{self.screen_y}')
+
+ def screen_fps_mofify(self, fps):
+ self.screenshot.config_modify(fps=fps)
+
+ def mouse_draw(self, checked):
+ draw_mouse = 1 if checked else 0
+ self.screenshot.config_modify(draw_mouse=draw_mouse)
+
+ def open_dirpath(self):
+ fdir = QFileDialog.getExistingDirectory(self, '选择目录', self._dir)
+ if not fdir:
+ return None
+ self._dir = fdir
+ self.set_filename()
+
+ def set_filename(self):
+ self.fname = f'{self._dir}/屏幕录制缓存_{int(datetime.now().replace(microsecond=0).timestamp())}'
+ self.screenshot.config_modify(filename=self.fname)
+ self.audio = AudioRecord()
+ self.setWindowTitle(f'屏幕录制 {self.fname}')
+
+ def start(self):
+ if self.hotkey_start:
+ return
+ if not self.fname:
+ QMessageBox.warning(self, '请选择保存目录！', '请选择保存目录！')
+ return
+ self.screenshot.start()
+ if self.record_audio:
+ self.audio.stop_flag = False
+ self.audio.run(filename=self.fname + '.mp3', name=self.device_name)
+ self.hotkey_start = True
+
+ def stop(self):
+ if not self.hotkey_start:
+ return
+ self.audio.stop_flag = True
+ self.hotkey_start = False
+ self.screenshot.stop()
+ if self.record_audio:
+ self.screenshot.compose()
+ else:
+ filename = self.fname + '.mp4'
+ target = filename.replace('缓存', '')
+ Path(filename).replace(target)
+ self.set_filename()
+
+
+class MousePaint(QDialog):
+ """移动鼠标获取屏幕捕获范围"""
+
+ def __init__(self):
+ super().__init__()
+ self.setMouseTracking(True)
+ # 设置窗口布满整个屏幕
+ self.showFullScreen()
+ # 设置窗体无边框
+ self.setWindowFlags(Qt.FramelessWindowHint) # 窗口置顶，无边框
+ # 设置背景透明
+ self.setWindowOpacity(0.5)
+ self.initUI()
+ self.setFocus()
+
+ def initUI(self):
+ self.setGeometry(*(QDesktopWidget().screenGeometry()).getRect())
+ self.pix = QPixmap()
+ self.lastpoint = QPoint()
+ self.endpoint = QPoint()
+ self.pos = None
+ self.bline = 0
+
+ def mousePressEvent(self, event):
+ # 监听鼠标按压事件
+ if event.button() == Qt.LeftButton:
+ self.lastpoint = event.x(), event.y()
+ self.bline = 1
+ elif event.button() == Qt.RightButton:
+ self.close()
+ event.accept()
+
+ def mouseReleaseEvent(self, event):
+ # 监听鼠标释放事件
+ self.endpoint = event.x(), event.y()
+ self.bline = 0
+ event.accept()
+ self.close()
+
+ def mouseMoveEvent(self, event):
+ # 监听鼠标移动事件
+ if self.bline == 1:
+ self.pos = event.x(), event.y()
+ event.accept()
+ self.update()
+
+ def paintEvent(self, event):
+ # 绘画事件
+ if self.bline == 1:
+ pp = QPainter(self)
+ pen = QPen() # 定义笔格式对象
+ pen.setWidth(5) # 设置笔的宽度
+ pen.setColor(QColor(255, 0, 0))
+ pp.setPen(pen)
+ lpx, lpy = self.lastpoint
+ pp.drawRect(lpx, lpy, self.pos[0] - lpx, self.pos[1] - lpy)
+ event.accept()
+
+
+# def main():
+# """运行函数"""
+# # 将7zip压缩格式添加到shutil中
+# shutil.register_archive_format('7zip', pack_7zarchive, description='7zip archive')
+# shutil.register_unpack_format('7zip', ['.7z'], unpack_7zarchive, description='7zip archive')
+# app = QApplication(sys.argv)
+# app.setAttribute(Qt.AA_UseHighDpiPixmaps)
+# ui = Ui_MainWindow1()
+# sys.exit(app.exec_())
+
+
+#if __name__ == '__main__':
+# main()
diff --git a/src/qt/Screenshot/Screenshot_record.py b/src/qt/Screenshot/Screenshot_record.py
new file mode 100644
index 0000000..afe8094
--- /dev/null
+++ b/src/qt/Screenshot/Screenshot_record.py
@@ -0,0 +1,67 @@
+"""
+使用ffmpeg录制屏幕，并结合Pyaudio同步录制的音频合成带有声音的视频文件
+"""
+import sys
+from pathlib import Path
+from time import sleep
+import ffmpeg
+
+
+def file_path(filename):
+ relative_path = Path('resource').joinpath(filename)
+ if getattr(sys, 'frozen', False):
+ base_path = Path(sys.MEIPASS)
+ else:
+ base_path = Path('.').absolute()
+ return Path(base_path).joinpath(relative_path).as_posix()
+
+
+class Screenshot:
+
+ def __init__(self, width=1920, height=1080, fps=15):
+ self.width = width
+ self.height = height
+ self.fps = fps
+ self.process = None
+ self.ffmpeg_path = file_path('ffmpeg.exe')
+
+ def __call__(self, width, height, fps=None):
+ self.width = width
+ self.height = height
+ self.fps = fps if fps else self.fps
+
+ @staticmethod
+ def unlink(filename):
+ Path(filename).unlink()
+
+ def record(self, filename, offset_x=0, offset_y=0, draw_mouse=0):
+ self.process = (
+ ffmpeg.output(
+ ffmpeg.input(
+ filename='desktop', format='gdigrab', framerate=self.fps, offset_x=offset_x, offset_y=offset_y,
+ draw_mouse=draw_mouse, s=f'{self.width}x{self.height}'),
+ filename=filename, pix_fmt='yuv420p'
+ ).overwrite_output()
+ )
+ self.ffmpeg_async()
+
+ def compose_audio(self, video_path, audio_path, output_path):
+ self.process = (
+ ffmpeg.output(
+ ffmpeg.input(filename=video_path),
+ ffmpeg.input(filename=audio_path),
+ filename=output_path, vcodec='copy', acodec='aac', strict='experimental', pix_fmt='yuv420p'
+ ).overwrite_output()
+ )
+ sleep(1)
+ self.ffmpeg_async()
+
+ def ffmpeg_async(self):
+ self.process = self.process.run_async(cmd=self.ffmpeg_path, pipe_stdin=True, pipe_stdout=False,
+ pipe_stderr=False)
+
+ def terminate(self):
+ if self.process is not None:
+ self.process.communicate(str.encode("q"))
+ self.process.terminate()
+ self.process = None
diff --git a/src/qt/Screenshot/Screenshot_record_with_cv2.py b/src/qt/Screenshot/Screenshot_record_with_cv2.py
new file mode 100644
index 0000000..8f6a136
--- /dev/null
+++ b/src/qt/Screenshot/Screenshot_record_with_cv2.py
@@ -0,0 +1,151 @@
+import time
+from pathlib import Path
+from threading import Thread
+
+import cv2
+import numpy as np
+from PIL import ImageGrab
+from numba import jit
+from pynput import keyboard
+
+
+@jit(nopython=True)
+def average_n(x, y):
+ """Numpy计算趋近值"""
+ return ((x + y + y) // 3).astype(x.dtype)
+
+
+class ScreenshotVideo(Thread):
+
+ def __init__(self, width, high, path='', fps=15):
+ """初始化参数"""
+ super().__init__()
+ self.save_file = path
+ self.best_fps = fps
+ self.fps = fps
+ self.width = width
+ self.high = high
+ self.spend_time = 1
+ self.flag = False
+ self.kill = False
+ self.video = None
+
+ def __call__(self, path):
+ """重载视频路径，便于类的二次调用"""
+ self.save_file = Path(path)
+ self.video = self.init_videowriter(self.save_file)
+
+ @staticmethod
+ def screenshot():
+ """静态方法，屏幕截图，并转换为np.array数组"""
+ return np.array(ImageGrab.grab())
+
+ @staticmethod
+ def get_fourcc(name):
+ """视频编码字典"""
+ fourcc_maps = {'.avi': 'I420',
+ '.m4v': 'mp4v',
+ '.mp4': 'avc1',
+ '.ogv': 'THEO',
+ '.flv': 'FLV1',
+ }
+ return fourcc_maps.get(name)
+
+ def init_videowriter(self, path):
+ """获取视频编码并新建视频文件"""
+ if not path:
+ raise Exception('视频路径未设置，请设置\nvideo = ScreenshotVideo(fps,width,high)\nvideo = video(video_path)')
+ path = Path(path) if isinstance(path, str) else path
+ fourcc = cv2.VideoWriter_fourcc(*self.get_fourcc(path.suffix))
+ return cv2.VideoWriter(path.as_posix(), fourcc, self.fps, (self.width, self.high))
+
+ def video_record_doing(self, img):
+ """将BGR数组转换为RGB数组"""
+ im_cv = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+ self.video.write(im_cv)
+
+ def video_record_end(self):
+ """录制结束，根据条件判断文件是否保存"""
+ self.video.release()
+ cv2.destroyAllWindows()
+ if self.save_file and self.kill:
+ Path(self.save_file).unlink()
+
+ def video_best_fps(self, path):
+ """获取电脑录制视频的最优帧率"""
+ video = cv2.VideoCapture(path)
+ fps = video.get(cv2.CAP_PROP_FPS)
+ count = video.get(cv2.CAP_PROP_FRAME_COUNT)
+ self.best_fps = int(fps * ((int(count) / fps) / self.spend_time))
+ video.release()
+
+ def pre_video_record(self):
+ """预录制，以获取最佳fps值"""
+ self.video = self.init_videowriter('test.mp4')
+ start_time = time.time()
+ for _ in range(10):
+ im = self.screenshot()
+ self.video_record_doing(im)
+ self.spend_time = round(time.time() - start_time, 4)
+ self.video_record_end()
+ time.sleep(2)
+ self.video_best_fps('test.mp4')
+ Path('test.mp4').unlink()
+
+ def insert_frame_array(self, frame_list):
+ """Numpy增强截图信息"""
+ fps_n = round(self.fps / self.best_fps)
+ if fps_n <= 0:
+ return frame_list
+ times = int(np.log2(fps_n)) # 倍率
+ for _ in range(times):
+ frame_list2 = map(average_n, [frame_list[0]] + frame_list[:-1], frame_list)
+ frame_list = [[x, y] for x, y in zip(frame_list2, frame_list)]
+ frame_list = [j for i in frame_list for j in i]
+ return frame_list
+
+ def frame2video_run(self):
+ """使用opencv将连续型截图转换为视频"""
+ self.video = self.init_videowriter(self.save_file)
+ start_time = time.time()
+ frame_list = []
+ while True:
+ frame_list.append(self.screenshot())
+ if self.flag:
+ break
+ self.spend_time = round(time.time() - start_time, 4)
+ if not self.kill: # 视频录制不被终止将逐帧处理图像
+ frame_list = self.insert_frame_array(frame_list)
+ for im in frame_list:
+ self.video_record_doing(im)
+ self.video_record_end()
+
+ def hotkey(self):
+ """热键监听"""
+ with keyboard.Listener(on_press=self.on_press) as listener:
+ listener.join()
+
+ def on_press(self, key):
+ try:
+ if key.char == 't': # 录屏结束，保存视频
+ self.flag = True
+ elif key.char == 'k': # 录屏中止，删除文件
+ self.flag = True
+ self.kill = True
+ except Exception as e:
+ print(e)
+
+ def run(self):
+ # 运行函数
+ # 设置守护线程
+ Thread(target=self.hotkey, daemon=True).start()
+ # 运行截图函数
+ self.frame2video_run()
+
+
+screen = ImageGrab.grab()
+width, high = screen.size
+video = ScreenshotVideo(width, high, fps=60)
+video.pre_video_record() # 预录制获取最优fps
+video('test1.mp4')
+video.run()
diff --git a/src/qt/Screenshot/Screenshot_test.py b/src/qt/Screenshot/Screenshot_test.py
new file mode 100644
index 0000000..0f8276f
--- /dev/null
+++ b/src/qt/Screenshot/Screenshot_test.py
@@ -0,0 +1,32 @@
+from threading import Thread
+
+from pynput import keyboard # pip install pynput
+
+from Audio_record import AudioRecord
+from Screenshot_record import Screenshot
+
+
+def hotkey():
+ """热键监听"""
+ with keyboard.Listener(on_press=on_press) as listener:
+ listener.join()
+
+
+def on_press(key):
+ try:
+ video.terminate()
+ if key.char == 't': # t键，录制结束，保存音视频
+ audio.stop_flag = True
+ elif key.char == 'k': # k键，录制中止，删除文件
+ audio.stop_flag = True
+ audio.kill = True
+ video.unlink('test.mp4')
+ except Exception as e:
+ print(e)
+
+
+key_thread = Thread(target=hotkey, daemon=True)
+audio = AudioRecord()
+video = Screenshot()
+key_thread.start()
+video.record('test.mp4')
diff --git a/src/qt/Screenshot/__pycache__/Audio_record.cpython-39.pyc b/src/qt/Screenshot/__pycache__/Audio_record.cpython-39.pyc
new file mode 100644
index 0000000..037c86a
Binary files /dev/null and b/src/qt/Screenshot/__pycache__/Audio_record.cpython-39.pyc differ
diff --git a/src/qt/Screenshot/__pycache__/Screenshot_Gui.cpython-39.pyc b/src/qt/Screenshot/__pycache__/Screenshot_Gui.cpython-39.pyc
new file mode 100644
index 0000000..ee5472f
Binary files /dev/null and b/src/qt/Screenshot/__pycache__/Screenshot_Gui.cpython-39.pyc differ
diff --git a/src/qt/Screenshot/__pycache__/Screenshot_record.cpython-39.pyc b/src/qt/Screenshot/__pycache__/Screenshot_record.cpython-39.pyc
new file mode 100644
index 0000000..64d31c2
Binary files /dev/null and b/src/qt/Screenshot/__pycache__/Screenshot_record.cpython-39.pyc differ
diff --git a/src/qt/Screenshot/requirements.txt b/src/qt/Screenshot/requirements.txt
new file mode 100644
index 0000000..e36e104
--- /dev/null
+++ b/src/qt/Screenshot/requirements.txt
@@ -0,0 +1,9 @@
+ffmpeg-python==0.2.0
+numba==0.54.1
+numpy==1.20.3
+opencv-python==4.5.4.60
+Pillow==8.4.0
+PyAudio
+pynput==1.7.5
+PyQt5==5.15.4
+py7zr==0.18.4
diff --git a/src/qt/Screenshot/resource/ffmpeg.7z b/src/qt/Screenshot/resource/ffmpeg.7z
new file mode 100644
index 0000000..80bad7c
Binary files /dev/null and b/src/qt/Screenshot/resource/ffmpeg.7z differ
diff --git a/src/qt/Screenshot/resource/ffmpeg.exe b/src/qt/Screenshot/resource/ffmpeg.exe
new file mode 100644
index 0000000..71b5352
Binary files /dev/null and b/src/qt/Screenshot/resource/ffmpeg.exe differ
diff --git a/src/qt/Screenshot/resource/gui_ico.ico b/src/qt/Screenshot/resource/gui_ico.ico
new file mode 100644
index 0000000..4a5fb69
Binary files /dev/null and b/src/qt/Screenshot/resource/gui_ico.ico differ
diff --git a/src/qt/__pycache__/findpath_UI.cpython-39.pyc b/src/qt/__pycache__/findpath_UI.cpython-39.pyc
new file mode 100644
index 0000000..54f0718
Binary files /dev/null and b/src/qt/__pycache__/findpath_UI.cpython-39.pyc differ
diff --git a/src/qt/findpath_UI.py b/src/qt/findpath_UI.py
index 8cb4358..d3a2918 100644
--- a/src/qt/findpath_UI.py
+++ b/src/qt/findpath_UI.py
@@ -9,6 +9,7 @@
from PyQt5 import QtCore, QtGui, QtWidgets
+from PyQt5.QtCore import QTimer
from PyQt5.QtMultimediaWidgets import QVideoWidget
from PyQt5.QtMultimedia import QMediaContent, QMediaPlayer
from PyQt5.QtWidgets import QFileDialog
@@ -20,7 +21,7 @@ class Ui_MainWindow1(object):
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
self.frame = QtWidgets.QFrame(self.centralwidget)
- self.frame.setGeometry(QtCore.QRect(370, 770, 421, 71))
+ self.frame.setGeometry(QtCore.QRect(370, 770, 421, 70))
self.frame.setStyleSheet("QPushButton{\n"
" border:none\n"
"}")
@@ -71,6 +72,7 @@ class Ui_MainWindow1(object):
self.sld_duration.setGeometry(QtCore.QRect(20, 20, 1051, 22))
self.sld_duration.setOrientation(QtCore.Qt.Horizontal)
self.sld_duration.setObjectName("sld_duration")
+ self.sld_duration.setEnabled(True)
self.lab_duration = QtWidgets.QLabel(self.frame_2)
self.lab_duration.setGeometry(QtCore.QRect(1100, 20, 81, 21))
font = QtGui.QFont()
@@ -105,18 +107,63 @@ class Ui_MainWindow1(object):
self.actionl1_2.setObjectName("actionl1_2")
self.toolBar.addAction(self.actionl1)
self.toolBar.addAction(self.actionl1_2)
+ self.frame_3 = QtWidgets.QFrame(self.centralwidget)
+ self.frame_3.setGeometry(QtCore.QRect(800, 770, 150, 70))
+ self.frame.setStyleSheet("QPushButton{\n"
+" border:none\n"
+"}")
+ self.frame_3.setFrameShape(QtWidgets.QFrame.StyledPanel)
+ self.frame_3.setFrameShadow(QtWidgets.QFrame.Raised)
+ self.frame_3.setObjectName("frame_3")
+ self.sld = QtWidgets.QSlider(self.frame_3)
+ self.sld.setGeometry(QtCore.QRect(50, 20, 50, 30))
+ self.sld.setOrientation(QtCore.Qt.Horizontal)
+ self.sld.setObjectName("sld")
+ self.sld.setEnabled(True)
+ self.lab = QtWidgets.QLabel(self.frame_3)
+ self.lab.setGeometry(QtCore.QRect(100, 20, 40, 30))
+ font = QtGui.QFont()
+ font.setFamily("微软雅黑")
+ font.setPointSize(13)
+ font.setBold(False)
+ font.setWeight(50)
+ font.setStyleStrategy(QtGui.QFont.PreferAntialias)
+ self.lab_duration.setFont(font)
+ self.lab_duration.setStyleSheet("QLabel\n"
+ "{\n"
+ " color:white\n"
+ "}")
+ self.lab.setObjectName("lab")
+ self.pushButton_9 = QtWidgets.QPushButton(self.frame_3)
+ self.pushButton_9.setEnabled(False)
+ self.pushButton_9.move(0, 10)
+ self.pushButton_9.setMinimumSize(QtCore.QSize(50, 50))
+ self.pushButton_9.setMaximumSize(QtCore.QSize(50, 50))
+ self.pushButton_9.setText("")
+ icon9 = QtGui.QIcon()
+ icon9.addPixmap(QtGui.QPixmap("window_png/音量.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)
+ self.pushButton_9.setIcon(icon9)
+ self.pushButton_9.setIconSize(QtCore.QSize(50, 50))
+ self.pushButton_9.setObjectName("pushButton_9")
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
self.player = QMediaPlayer()
self.player.setVideoOutput(self.widget)
+ self.timer = QTimer()
+ self.timer.start(1000)
+ self.timer.timeout.connect(lambda: self.onTimerOut())
self.actionl1.triggered.connect(lambda: self.openfile())
- self.displayTime()
self.pushButton_2.clicked.connect(lambda: self.playpause())
+ self.pushButton_3.clicked.connect(lambda: self.uptime())
+ self.pushButton.clicked.connect(lambda: self.downtime())
+ self.sld_duration.sliderMoved.connect(lambda: self.changetime())
+ self.sld.sliderMoved.connect(lambda: self.changevoice())
def openfile(self):
- self.player.setMedia(QMediaContent(QFileDialog.getOpenFileUrl()[0]))
+ media_list = QMediaContent(QFileDialog.getOpenFileUrl()[0])
+ self.player.setMedia(media_list)
self.player.play()
self.player.pause()
@@ -133,17 +180,36 @@ class Ui_MainWindow1(object):
icon9.addPixmap(QtGui.QPixmap("window_png/暂停.png"), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.pushButton_2.setIcon(icon9)
self.pushButton_2.setIconSize(QtCore.QSize(50, 50))
- def getms(self):
- a = self.player.duration()
- self.sld_duration.setRange(0, a)
- self.sld_duration.setEnabled(True)
- return a
- def displayTime(self):
- ms = self.getms()
+
+ def onTimerOut(self):
+ alltime = self.player.duration()
+ ms = self.player.position()
minutes = int(ms / 60000)
seconds = int((ms - minutes * 60000) / 1000)
- self.lab_duration.setText('{}:{}'.format(minutes, seconds))
+ minute = int(alltime / 60000)
+ sec = int((alltime - minute * 60000) / 1000)
+ self.lab_duration.setText('{}:{}/{}:{}'.format(minutes, seconds, minute, sec))
+ self.sld_duration.setMaximum(alltime)
+ self.sld_duration.setValue(ms)
+ self.sld.setValue(self.player.volume())
+ self.lab.setText('{}%'.format(self.player.volume()))
+ def uptime(self):
+ num = self.player.position() + int(self.player.duration() / 20)
+ self.player.setPosition(num)
+
+ def downtime(self):
+ num = self.player.position() - int(self.player.duration() / 20)
+ self.player.setPosition(num)
+
+ def changetime(self):
+ self.timer.stop()
+ num = self.sld_duration.value()
+ self.player.setPosition(num)
+ self.timer.start()
+ def changevoice(self):
+ num = self.sld.value()
+ self.player.setVolume(num)
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
diff --git a/src/qt/resource/ffmpeg.7z b/src/qt/resource/ffmpeg.7z
new file mode 100644
index 0000000..80bad7c
Binary files /dev/null and b/src/qt/resource/ffmpeg.7z differ
diff --git a/src/qt/resource/ffmpeg.exe b/src/qt/resource/ffmpeg.exe
new file mode 100644
index 0000000..71b5352
Binary files /dev/null and b/src/qt/resource/ffmpeg.exe differ
diff --git a/src/qt/resource/gui_ico.ico b/src/qt/resource/gui_ico.ico
new file mode 100644
index 0000000..4a5fb69
Binary files /dev/null and b/src/qt/resource/gui_ico.ico differ
diff --git a/src/qt/resource/gui_svg.svg b/src/qt/resource/gui_svg.svg
new file mode 100644
index 0000000..37d2ebc
--- /dev/null
+++ b/src/qt/resource/gui_svg.svg
@@ -0,0 +1,4 @@
+
+
\ No newline at end of file
diff --git a/src/qt/window_png/录像.png b/src/qt/window_png/录像.png
new file mode 100644
index 0000000..04f717c
Binary files /dev/null and b/src/qt/window_png/录像.png differ
diff --git a/src/qt/window_png/音量.png b/src/qt/window_png/音量.png
new file mode 100644
index 0000000..f259469
Binary files /dev/null and b/src/qt/window_png/音量.png differ