class VideoPlayer(QWidget):
def __init__(self, parent=None):
super(VideoPlayer, self).__init__(parent)
self.mediaPlayer = QMediaPlayer(None, QMediaPlayer.VideoSurface)
self.videoItem = QGraphicsVideoItem()
self.videoItem.setSize(QSizeF(640, 480))
scene = QGraphicsScene(self)
graphicsView = QGraphicsView(scene)
scene.addItem(self.videoItem)
rotateSlider = QSlider(Qt.Horizontal)
rotateSlider.setRange(-180, 180)
rotateSlider.setValue(0)
rotateSlider.valueChanged.connect(self.rotateVideo)
openButton = QPushButton("Open...")
openButton.clicked.connect(self.openFile)
self.playButton = QPushButton()
self.playButton.setEnabled(False)
self.playButton.setIcon(self.style().standardIcon(QStyle.SP_MediaPlay))
self.playButton.clicked.connect(self.play)
self.positionSlider = QSlider(Qt.Horizontal)
self.positionSlider.setRange(0, 0)
self.positionSlider.sliderMoved.connect(self.setPosition)
controlLayout = QHBoxLayout()
controlLayout.setContentsMargins(0, 0, 0, 0)
controlLayout.addWidget(openButton)
controlLayout.addWidget(self.playButton)
controlLayout.addWidget(self.positionSlider)
layout = QVBoxLayout()
layout.addWidget(graphicsView)
layout.addWidget(rotateSlider)
layout.addLayout(controlLayout)
self.setLayout(layout)
self.mediaPlayer.setVideoOutput(self.videoItem)
self.mediaPlayer.stateChanged.connect(self.mediaStateChanged)
self.mediaPlayer.positionChanged.connect(self.positionChanged)
self.mediaPlayer.durationChanged.connect(self.durationChanged)
def sizeHint(self):
return QSize(800, 600)
def openFile(self):
fileName, _ = QFileDialog.getOpenFileName(self, "Open Movie",
QDir.homePath())
if fileName != '':
self.mediaPlayer.setMedia(
QMediaContent(QUrl.fromLocalFile(fileName)))
self.playButton.setEnabled(True)
def play(self):
if self.mediaPlayer.state() == QMediaPlayer.PlayingState:
self.mediaPlayer.pause()
else:
self.mediaPlayer.play()
def mediaStateChanged(self, state):
if self.mediaPlayer.state() == QMediaPlayer.PlayingState:
self.playButton.setIcon(self.style().standardIcon(
QStyle.SP_MediaPause))
else:
self.playButton.setIcon(self.style().standardIcon(
QStyle.SP_MediaPlay))
def positionChanged(self, position):
self.positionSlider.setValue(position)
def durationChanged(self, duration):
self.positionSlider.setRange(0, duration)
def setPosition(self, position):
self.mediaPlayer.setPosition(position)
def rotateVideo(self, angle):
x = self.videoItem.boundingRect().width() / 2.0
y = self.videoItem.boundingRect().height() / 2.0
self.videoItem.setTransform(QTransform().translate(
x, y).rotate(angle).translate(-x, -y))
class VideoPlayer(QWidget):
def __init__(self, parent=None):
super(VideoPlayer, self).__init__(parent)
self.mediaPlayer = QMediaPlayer(None, QMediaPlayer.VideoSurface)
self.videoItem = QGraphicsVideoItem()
self.videoItem.setSize(QSizeF(640, 480))
scene = QGraphicsScene(self)
graphicsView = QGraphicsView(scene)
scene.addItem(self.videoItem)
rotateSlider = QSlider(Qt.Horizontal)
rotateSlider.setRange(-180, 180)
rotateSlider.setValue(0)
rotateSlider.valueChanged.connect(self.rotateVideo)
openButton = QPushButton("Open...")
openButton.clicked.connect(self.openFile)
self.playButton = QPushButton()
self.playButton.setEnabled(False)
self.playButton.setIcon(self.style().standardIcon(QStyle.SP_MediaPlay))
self.playButton.clicked.connect(self.play)
self.positionSlider = QSlider(Qt.Horizontal)
self.positionSlider.setRange(0, 0)
self.positionSlider.sliderMoved.connect(self.setPosition)
controlLayout = QHBoxLayout()
controlLayout.setContentsMargins(0, 0, 0, 0)
controlLayout.addWidget(openButton)
controlLayout.addWidget(self.playButton)
controlLayout.addWidget(self.positionSlider)
layout = QVBoxLayout()
layout.addWidget(graphicsView)
layout.addWidget(rotateSlider)
layout.addLayout(controlLayout)
self.setLayout(layout)
self.mediaPlayer.setVideoOutput(self.videoItem)
self.mediaPlayer.stateChanged.connect(self.mediaStateChanged)
self.mediaPlayer.positionChanged.connect(self.positionChanged)
self.mediaPlayer.durationChanged.connect(self.durationChanged)
def sizeHint(self):
return QSize(800, 600)
def openFile(self):
fileName, _ = QFileDialog.getOpenFileName(self, "Open Movie",
QDir.homePath())
if fileName != '':
self.mediaPlayer.setMedia(
QMediaContent(QUrl.fromLocalFile(fileName)))
self.playButton.setEnabled(True)
def play(self):
if self.mediaPlayer.state() == QMediaPlayer.PlayingState:
self.mediaPlayer.pause()
else:
self.mediaPlayer.play()
def mediaStateChanged(self, state):
if self.mediaPlayer.state() == QMediaPlayer.PlayingState:
self.playButton.setIcon(
self.style().standardIcon(QStyle.SP_MediaPause))
else:
self.playButton.setIcon(
self.style().standardIcon(QStyle.SP_MediaPlay))
def positionChanged(self, position):
self.positionSlider.setValue(position)
def durationChanged(self, duration):
self.positionSlider.setRange(0, duration)
def setPosition(self, position):
self.mediaPlayer.setPosition(position)
def rotateVideo(self, angle):
x = self.videoItem.boundingRect().width() / 2.0
y = self.videoItem.boundingRect().height() / 2.0
self.videoItem.setTransform(
QTransform().translate(x, y).rotate(angle).translate(-x, -y))
class VideoScene(QGraphicsScene):
regionSelected = Signal(QRect)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.videoItem = QGraphicsVideoItem()
self.videoItem.setPos(0, 0)
self.videoItem.nativeSizeChanged.connect(self.videoSizeChanged)
self.addItem(self.videoItem)
self.rectangle = QGraphicsRectItem()
self.rectangle.setPen(QColor(0, 122, 217))
self.addItem(self.rectangle)
def videoSizeChanged(self, size: QSizeF):
logging.debug("%s", size)
if not size.isEmpty():
self.videoItem.setSize(size)
self.setSceneRect(self.videoItem.boundingRect())
self.clearSelection()
def clearSelection(self) -> None:
self.rectangle.setRect(QRectF())
self.regionSelected.emit(self.selection)
@property
def selection(self) -> QRect:
return self.rectangle.rect().toRect()
@staticmethod
def coerceInside(point: QPointF, rect: QRectF):
return QPointF(
min(max(point.x(), rect.left()), rect.right()),
min(max(point.y(), rect.top()), rect.bottom()),
)
def coerceInsideVideo(self, point: QPointF):
return self.coerceInside(point, self.videoItem.boundingRect())
def mousePressEvent(self, event: QGraphicsSceneMouseEvent) -> None:
self.startPos = self.coerceInsideVideo(event.scenePos())
return super().mousePressEvent(event)
def mouseMoveEvent(self, event: QGraphicsSceneMouseEvent) -> None:
endPos = self.coerceInsideVideo(event.scenePos())
self.rectangle.setRect(QRectF(self.startPos, endPos).normalized())
return super().mouseMoveEvent(event)
def mouseReleaseEvent(self, event: QGraphicsSceneMouseEvent) -> None:
logging.info("Selection: %s", self.selection)
super().mouseReleaseEvent(event)
self.regionSelected.emit(self.selection)
def paintGrid(self, painter: QPainter, rect: QRectF, gridSize: int):
left, top, right, bottom = rect.getCoords()
x_min = (int(left - 1) // gridSize + 1) * gridSize
y_min = (int(top - 1) // gridSize + 1) * gridSize
x_max = (int(right) // gridSize) * gridSize
y_max = (int(bottom) // gridSize) * gridSize
# logging.debug(
# "x=(%s<=%s..%s<=%s) y=(%s<=%s..%s<=%s)",
# *(left, x_min, x_max, right),
# *(top, y_min, y_max, bottom)
# )
painter.setPen(QPen(Qt.gray, 2))
painter.drawLines(QLineF(0, top, 0, bottom), QLineF(left, 0, right, 0))
painter.setPen(QPen(Qt.gray, 1))
painter.drawLines(
QLineF(x, top, x, bottom)
for x in range(x_min, x_max + 1, gridSize) if x)
painter.drawLines(
QLineF(left, y, right, y)
for y in range(y_min, y_max + 1, gridSize) if y)
def drawBackground(self, painter: QPainter, rect: QRectF) -> None:
super().drawBackground(painter, rect)
self.paintGrid(painter, rect, 20)
