1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
# Copyright (C) 2023 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
from enum import Enum, auto
from PySide6.QtMultimediaWidgets import QVideoWidget
from PySide6.QtMultimedia import (QCapturableWindow, QMediaCaptureSession,
QScreenCapture, QWindowCapture)
from PySide6.QtWidgets import (QGridLayout, QLabel, QListView,
QMessageBox, QPushButton, QWidget)
from PySide6.QtGui import QAction, QGuiApplication
from PySide6.QtCore import QItemSelection, Qt, Slot
from screenlistmodel import ScreenListModel
from windowlistmodel import WindowListModel
class SourceType(Enum):
Screen = auto()
Window = auto()
class ScreenCapturePreview(QWidget):
def __init__(self, parent=None):
super().__init__(parent)
self._source = SourceType.Screen
self._screen_capture = QScreenCapture(self)
self._media_capture_session = QMediaCaptureSession(self)
self._video_widget = QVideoWidget(self)
self._screen_list_view = QListView(self)
self._screen_label = QLabel("Select screen to capture:", self)
self._video_widget_label = QLabel("Capture output:", self)
self._start_stop_button = QPushButton(self)
self._status_label = QLabel(self)
self._screen_list_model = ScreenListModel(self)
# Setup QScreenCapture with initial source:
self.setScreen(QGuiApplication.primaryScreen())
self._screen_capture.start()
self._media_capture_session.setScreenCapture(self._screen_capture)
self._media_capture_session.setVideoOutput(self._video_widget)
self._screen_list_view.setModel(self._screen_list_model)
self._window_list_view = QListView(self)
self._window_capture = QWindowCapture(self)
self._media_capture_session.setWindowCapture(self._window_capture)
self._window_label = QLabel("Select window to capture:", self)
self._window_list_model = WindowListModel(self)
self._window_list_view.setModel(self._window_list_model)
update_action = QAction("Update windows List", self)
update_action.triggered.connect(self._window_list_model.populate)
self._window_list_view.addAction(update_action)
self._window_list_view.setContextMenuPolicy(Qt.ActionsContextMenu)
grid_layout = QGridLayout(self)
grid_layout.addWidget(self._screen_label, 0, 0)
grid_layout.addWidget(self._screen_list_view, 1, 0)
grid_layout.addWidget(self._start_stop_button, 4, 0)
grid_layout.addWidget(self._video_widget_label, 0, 1)
grid_layout.addWidget(self._video_widget, 1, 1, 4, 1)
grid_layout.addWidget(self._window_label, 2, 0)
grid_layout.addWidget(self._window_list_view, 3, 0)
grid_layout.addWidget(self._status_label, 5, 0, 1, 2)
grid_layout.setColumnStretch(1, 1)
grid_layout.setRowStretch(1, 1)
grid_layout.setColumnMinimumWidth(0, 400)
grid_layout.setColumnMinimumWidth(1, 400)
grid_layout.setRowMinimumHeight(3, 1)
selection_model = self._screen_list_view.selectionModel()
selection_model.selectionChanged.connect(self.on_current_screen_selection_changed)
selection_model = self._window_list_view.selectionModel()
selection_model.selectionChanged.connect(self.on_current_window_selection_changed)
self._start_stop_button.clicked.connect(self.on_start_stop_button_clicked)
self._screen_capture.errorOccurred.connect(self.on_screen_capture_error_occured,
Qt.QueuedConnection)
self._window_capture.errorOccurred.connect(self.on_window_capture_error_occured,
Qt.QueuedConnection)
self.update_active(SourceType.Screen, True)
@Slot(QItemSelection)
def on_current_screen_selection_changed(self, selection):
self.clear_error_string()
indexes = selection.indexes()
if indexes:
self._screen_capture.setScreen(self._screen_list_model.screen(indexes[0]))
self.update_active(SourceType.Screen, self.is_active())
self._window_list_view.clearSelection()
else:
self._screen_capture.setScreen(None)
@Slot(QItemSelection)
def on_current_window_selection_changed(self, selection):
self.clear_error_string()
indexes = selection.indexes()
if indexes:
window = self._window_list_model.window(indexes[0])
if not window.isValid():
m = "The window is no longer valid. Update the list of windows?"
answer = QMessageBox.question(self, "Invalid window", m)
if answer == QMessageBox.Yes:
self.update_active(SourceType.Window, False)
self._window_list_view.clearSelection()
self._window_list_model.populate()
return
self._window_capture.setWindow(window)
self.update_active(SourceType.Window, self.is_active())
self._screen_list_view.clearSelection()
else:
self._window_capture.setWindow(QCapturableWindow())
@Slot(QWindowCapture.Error, str)
def on_window_capture_error_occured(self, error, error_string):
self.set_error_string("QWindowCapture: Error occurred " + error_string)
@Slot(QScreenCapture.Error, str)
def on_screen_capture_error_occured(self, error, error_string):
self.set_error_string("QScreenCapture: Error occurred " + error_string)
def set_error_string(self, t):
self._status_label.setStyleSheet("background-color: rgb(255, 0, 0);")
self._status_label.setText(t)
def clear_error_string(self):
self._status_label.clear()
self._status_label.setStyleSheet("")
@Slot()
def on_start_stop_button_clicked(self):
self.clear_error_string()
self.update_active(self._source_type, not self.is_active())
def update_start_stop_button_text(self):
active = self.is_active()
if self._source_type == SourceType.Window:
m = "Stop window capture" if active else "Start window capture"
self._start_stop_button.setText(m)
elif self._source_type == SourceType.Screen:
m = "Stop screen capture" if active else "Start screen capture"
self._start_stop_button.setText(m)
def update_active(self, source_type, active):
self._source_type = source_type
self._screen_capture.setActive(active and source_type == SourceType.Screen)
self._window_capture.setActive(active and source_type == SourceType.Window)
self.update_start_stop_button_text()
def is_active(self):
if self._source_type == SourceType.Window:
return self._window_capture.isActive()
if self._source_type == SourceType.Screen:
return self._screen_capture.isActive()
return False
|
