Last active
June 17, 2023 19:21
-
-
Save metaphox/a915cb47f70c0886df565ef7a87aef8a to your computer and use it in GitHub Desktop.
probing frame number
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python | |
| ############################################################################# | |
| ## | |
| ## Copyright (C) 2013 Riverbank Computing Limited. | |
| ## Copyright (C) 2013 Digia Plc and/or its subsidiary(-ies). | |
| ## All rights reserved. | |
| ## | |
| ## This file is part of the examples of PyQt. | |
| ## | |
| ## $QT_BEGIN_LICENSE:BSD$ | |
| ## You may use this file under the terms of the BSD license as follows: | |
| ## | |
| ## "Redistribution and use in source and binary forms, with or without | |
| ## modification, are permitted provided that the following conditions are | |
| ## met: | |
| ## * Redistributions of source code must retain the above copyright | |
| ## notice, this list of conditions and the following disclaimer. | |
| ## * Redistributions in binary form must reproduce the above copyright | |
| ## notice, this list of conditions and the following disclaimer in | |
| ## the documentation and/or other materials provided with the | |
| ## distribution. | |
| ## * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor | |
| ## the names of its contributors may be used to endorse or promote | |
| ## products derived from this software without specific prior written | |
| ## permission. | |
| ## | |
| ## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
| ## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
| ## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
| ## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | |
| ## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
| ## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | |
| ## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
| ## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
| ## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
| ## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
| ## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." | |
| ## $QT_END_LICENSE$ | |
| ## | |
| ############################################################################# | |
| from PyQt5.QtCore import (pyqtSignal, pyqtSlot, Q_ARG, QAbstractItemModel, | |
| QFileInfo, qFuzzyCompare, QModelIndex, Qt, QTime, QUrl) | |
| from PyQt5.QtGui import QPalette | |
| from PyQt5.QtMultimedia import (QAbstractVideoBuffer, QMediaContent, | |
| QMediaMetaData, QMediaPlayer, QMediaPlaylist, QVideoFrame, QVideoProbe) | |
| from PyQt5.QtMultimediaWidgets import QVideoWidget | |
| from PyQt5.QtWidgets import (QApplication, QComboBox, QDialog, QFileDialog, | |
| QFormLayout, QHBoxLayout, QLabel, QListView, QMessageBox, QPushButton, | |
| QSizePolicy, QSlider, QStyle, QToolButton, QVBoxLayout, QWidget, QMainWindow) | |
| class VideoWidget(QVideoWidget): | |
| def __init__(self, parent=None): | |
| super(VideoWidget, self).__init__(parent) | |
| self.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored) | |
| p = self.palette() | |
| p.setColor(QPalette.Window, Qt.black) | |
| self.setPalette(p) | |
| self.setAttribute(Qt.WA_OpaquePaintEvent) | |
| def keyPressEvent(self, event): | |
| if event.key() == Qt.Key_Escape and self.isFullScreen(): | |
| self.setFullScreen(False) | |
| event.accept() | |
| elif event.key() == Qt.Key_Enter and event.modifiers() & Qt.Key_Alt: | |
| self.setFullScreen(not self.isFullScreen()) | |
| event.accept() | |
| else: | |
| super(VideoWidget, self).keyPressEvent(event) | |
| def mouseDoubleClickEvent(self, event): | |
| self.setFullScreen(not self.isFullScreen()) | |
| event.accept() | |
| class PlaylistModel(QAbstractItemModel): | |
| Title, ColumnCount = range(2) | |
| def __init__(self, parent=None): | |
| super(PlaylistModel, self).__init__(parent) | |
| self.m_playlist = None | |
| def rowCount(self, parent=QModelIndex()): | |
| return self.m_playlist.mediaCount() if self.m_playlist is not None and not parent.isValid() else 0 | |
| def columnCount(self, parent=QModelIndex()): | |
| return self.ColumnCount if not parent.isValid() else 0 | |
| def index(self, row, column, parent=QModelIndex()): | |
| return self.createIndex(row, column) if self.m_playlist is not None and not parent.isValid() and row >= 0 and row < self.m_playlist.mediaCount() and column >= 0 and column < self.ColumnCount else QModelIndex() | |
| def parent(self, child): | |
| return QModelIndex() | |
| def data(self, index, role=Qt.DisplayRole): | |
| if index.isValid() and role == Qt.DisplayRole: | |
| if index.column() == self.Title: | |
| location = self.m_playlist.media(index.row()).canonicalUrl() | |
| return QFileInfo(location.path()).fileName() | |
| return self.m_data[index] | |
| return None | |
| def playlist(self): | |
| return self.m_playlist | |
| def setPlaylist(self, playlist): | |
| if self.m_playlist is not None: | |
| self.m_playlist.mediaAboutToBeInserted.disconnect( | |
| self.beginInsertItems) | |
| self.m_playlist.mediaInserted.disconnect(self.endInsertItems) | |
| self.m_playlist.mediaAboutToBeRemoved.disconnect( | |
| self.beginRemoveItems) | |
| self.m_playlist.mediaRemoved.disconnect(self.endRemoveItems) | |
| self.m_playlist.mediaChanged.disconnect(self.changeItems) | |
| self.beginResetModel() | |
| self.m_playlist = playlist | |
| if self.m_playlist is not None: | |
| self.m_playlist.mediaAboutToBeInserted.connect( | |
| self.beginInsertItems) | |
| self.m_playlist.mediaInserted.connect(self.endInsertItems) | |
| self.m_playlist.mediaAboutToBeRemoved.connect( | |
| self.beginRemoveItems) | |
| self.m_playlist.mediaRemoved.connect(self.endRemoveItems) | |
| self.m_playlist.mediaChanged.connect(self.changeItems) | |
| self.endResetModel() | |
| def beginInsertItems(self, start, end): | |
| self.beginInsertRows(QModelIndex(), start, end) | |
| def endInsertItems(self): | |
| self.endInsertRows() | |
| def beginRemoveItems(self, start, end): | |
| self.beginRemoveRows(QModelIndex(), start, end) | |
| def endRemoveItems(self): | |
| self.endRemoveRows() | |
| def changeItems(self, start, end): | |
| self.dataChanged.emit(self.index(start, 0), | |
| self.index(end, self.ColumnCount)) | |
| class PlayerControls(QWidget): | |
| play = pyqtSignal() | |
| pause = pyqtSignal() | |
| stop = pyqtSignal() | |
| next = pyqtSignal() | |
| previous = pyqtSignal() | |
| changeVolume = pyqtSignal(int) | |
| changeMuting = pyqtSignal(bool) | |
| changeRate = pyqtSignal(float) | |
| def __init__(self, parent=None): | |
| super(PlayerControls, self).__init__(parent) | |
| self.playerState = QMediaPlayer.StoppedState | |
| self.playerMuted = False | |
| self.playButton = QToolButton(clicked=self.playClicked) | |
| self.playButton.setIcon(self.style().standardIcon(QStyle.SP_MediaPlay)) | |
| self.stopButton = QToolButton(clicked=self.stop) | |
| self.stopButton.setIcon(self.style().standardIcon(QStyle.SP_MediaStop)) | |
| self.stopButton.setEnabled(False) | |
| self.nextButton = QToolButton(clicked=self.next) | |
| self.nextButton.setIcon( | |
| self.style().standardIcon(QStyle.SP_MediaSkipForward)) | |
| self.previousButton = QToolButton(clicked=self.previous) | |
| self.previousButton.setIcon( | |
| self.style().standardIcon(QStyle.SP_MediaSkipBackward)) | |
| self.muteButton = QToolButton(clicked=self.muteClicked) | |
| self.muteButton.setIcon( | |
| self.style().standardIcon(QStyle.SP_MediaVolume)) | |
| self.volumeSlider = QSlider(Qt.Horizontal, | |
| sliderMoved=self.changeVolume) | |
| self.volumeSlider.setRange(0, 100) | |
| self.rateBox = QComboBox(activated=self.updateRate) | |
| self.rateBox.addItem("0.5x", 0.5) | |
| self.rateBox.addItem("1.0x", 1.0) | |
| self.rateBox.addItem("2.0x", 2.0) | |
| self.rateBox.setCurrentIndex(1) | |
| layout = QHBoxLayout() | |
| layout.setContentsMargins(0, 0, 0, 0) | |
| layout.addWidget(self.stopButton) | |
| layout.addWidget(self.previousButton) | |
| layout.addWidget(self.playButton) | |
| layout.addWidget(self.nextButton) | |
| layout.addWidget(self.muteButton) | |
| layout.addWidget(self.volumeSlider) | |
| layout.addWidget(self.rateBox) | |
| self.setLayout(layout) | |
| def state(self): | |
| return self.playerState | |
| def setState(self,state): | |
| if state != self.playerState: | |
| self.playerState = state | |
| if state == QMediaPlayer.StoppedState: | |
| self.stopButton.setEnabled(False) | |
| self.playButton.setIcon( | |
| self.style().standardIcon(QStyle.SP_MediaPlay)) | |
| elif state == QMediaPlayer.PlayingState: | |
| self.stopButton.setEnabled(True) | |
| self.playButton.setIcon( | |
| self.style().standardIcon(QStyle.SP_MediaPause)) | |
| elif state == QMediaPlayer.PausedState: | |
| self.stopButton.setEnabled(True) | |
| self.playButton.setIcon( | |
| self.style().standardIcon(QStyle.SP_MediaPlay)) | |
| def volume(self): | |
| return self.volumeSlider.value() | |
| def setVolume(self, volume): | |
| self.volumeSlider.setValue(volume) | |
| def isMuted(self): | |
| return self.playerMuted | |
| def setMuted(self, muted): | |
| if muted != self.playerMuted: | |
| self.playerMuted = muted | |
| self.muteButton.setIcon( | |
| self.style().standardIcon( | |
| QStyle.SP_MediaVolumeMuted if muted else QStyle.SP_MediaVolume)) | |
| def playClicked(self): | |
| if self.playerState in (QMediaPlayer.StoppedState, QMediaPlayer.PausedState): | |
| self.play.emit() | |
| elif self.playerState == QMediaPlayer.PlayingState: | |
| self.pause.emit() | |
| def muteClicked(self): | |
| self.changeMuting.emit(not self.playerMuted) | |
| def playbackRate(self): | |
| return self.rateBox.itemData(self.rateBox.currentIndex()) | |
| def setPlaybackRate(self, rate): | |
| for i in range(self.rateBox.count()): | |
| if qFuzzyCompare(rate, self.rateBox.itemData(i)): | |
| self.rateBox.setCurrentIndex(i) | |
| return | |
| self.rateBox.addItem("%dx" % rate, rate) | |
| self.rateBox.setCurrentIndex(self.rateBox.count() - 1) | |
| def updateRate(self): | |
| self.changeRate.emit(self.playbackRate()) | |
| class FrameCounterWidget(QLabel): | |
| def __init__(self, parent=None): | |
| super(FrameCounterWidget, self).__init__(parent) | |
| self.frame_cnt = 0 | |
| self.setText('0') | |
| @pyqtSlot(QVideoFrame) | |
| def processFrame(self, frame): | |
| self.frame_cnt = self.frame_cnt + 1 | |
| self.setText(str(self.frame_cnt)) | |
| class Player(QWidget): | |
| fullScreenChanged = pyqtSignal(bool) | |
| def __init__(self, playlist, parent=None): | |
| super(Player, self).__init__(parent) | |
| self.colorDialog = None | |
| self.trackInfo = "" | |
| self.statusInfo = "" | |
| self.duration = 0 | |
| self.player = QMediaPlayer() | |
| self.playlist = QMediaPlaylist() | |
| self.player.setPlaylist(self.playlist) | |
| self.player.durationChanged.connect(self.durationChanged) | |
| self.player.positionChanged.connect(self.positionChanged) | |
| self.player.metaDataChanged.connect(self.metaDataChanged) | |
| self.playlist.currentIndexChanged.connect(self.playlistPositionChanged) | |
| self.player.mediaStatusChanged.connect(self.statusChanged) | |
| self.player.bufferStatusChanged.connect(self.bufferingProgress) | |
| self.player.videoAvailableChanged.connect(self.videoAvailableChanged) | |
| self.player.error.connect(self.displayErrorMessage) | |
| self.videoWidget = VideoWidget() | |
| self.player.setVideoOutput(self.videoWidget) | |
| self.playlistModel = PlaylistModel() | |
| self.playlistModel.setPlaylist(self.playlist) | |
| self.playlistView = QListView() | |
| self.playlistView.setModel(self.playlistModel) | |
| self.playlistView.setCurrentIndex( | |
| self.playlistModel.index(self.playlist.currentIndex(), 0)) | |
| self.playlistView.activated.connect(self.jump) | |
| self.slider = QSlider(Qt.Horizontal) | |
| self.slider.setRange(0, self.player.duration() / 1000) | |
| self.labelDuration = QLabel() | |
| self.slider.sliderMoved.connect(self.seek) | |
| self.frameCounter = FrameCounterWidget() | |
| self.probe = QVideoProbe() | |
| self.probe.videoFrameProbed.connect(self.frameCounter.processFrame) | |
| self.probe.setSource(self.player) | |
| openButton = QPushButton("Open", clicked=self.open) | |
| controls = PlayerControls() | |
| controls.setState(self.player.state()) | |
| controls.setVolume(self.player.volume()) | |
| controls.setMuted(controls.isMuted()) | |
| controls.play.connect(self.player.play) | |
| controls.pause.connect(self.player.pause) | |
| controls.stop.connect(self.player.stop) | |
| controls.next.connect(self.playlist.next) | |
| controls.previous.connect(self.previousClicked) | |
| controls.changeVolume.connect(self.player.setVolume) | |
| controls.changeMuting.connect(self.player.setMuted) | |
| controls.changeRate.connect(self.player.setPlaybackRate) | |
| controls.stop.connect(self.videoWidget.update) | |
| self.player.stateChanged.connect(controls.setState) | |
| self.player.volumeChanged.connect(controls.setVolume) | |
| self.player.mutedChanged.connect(controls.setMuted) | |
| self.fullScreenButton = QPushButton("FullScreen") | |
| self.fullScreenButton.setCheckable(True) | |
| displayLayout = QHBoxLayout() | |
| displayLayout.addWidget(self.videoWidget, 2) | |
| displayLayout.addWidget(self.playlistView) | |
| controlLayout = QHBoxLayout() | |
| controlLayout.setContentsMargins(0, 0, 0, 0) | |
| controlLayout.addWidget(openButton) | |
| controlLayout.addStretch(1) | |
| controlLayout.addWidget(controls) | |
| controlLayout.addStretch(1) | |
| controlLayout.addWidget(self.fullScreenButton) | |
| layout = QVBoxLayout() | |
| layout.addLayout(displayLayout) | |
| hLayout = QHBoxLayout() | |
| hLayout.addWidget(self.slider) | |
| hLayout.addWidget(self.labelDuration) | |
| hLayout.addWidget(self.frameCounter) | |
| layout.addLayout(hLayout) | |
| layout.addLayout(controlLayout) | |
| self.setLayout(layout) | |
| if not self.player.isAvailable(): | |
| QMessageBox.warning(self, "Service not available", | |
| "The QMediaPlayer object does not have a valid service.\n" | |
| "Please check the media service plugins are installed.") | |
| controls.setEnabled(False) | |
| self.playlistView.setEnabled(False) | |
| openButton.setEnabled(False) | |
| self.fullScreenButton.setEnabled(False) | |
| self.metaDataChanged() | |
| self.addToPlaylist(playlist) | |
| def open(self): | |
| fileNames, _ = QFileDialog.getOpenFileNames(self, "Open Files") | |
| self.addToPlaylist(fileNames) | |
| def addToPlaylist(self, fileNames): | |
| for name in fileNames: | |
| fileInfo = QFileInfo(name) | |
| if fileInfo.exists(): | |
| url = QUrl.fromLocalFile(fileInfo.absoluteFilePath()) | |
| if fileInfo.suffix().lower() == 'm3u': | |
| self.playlist.load(url) | |
| else: | |
| self.playlist.addMedia(QMediaContent(url)) | |
| else: | |
| url = QUrl(name) | |
| if url.isValid(): | |
| self.playlist.addMedia(QMediaContent(url)) | |
| def durationChanged(self, duration): | |
| duration /= 1000 | |
| self.duration = duration | |
| self.slider.setMaximum(duration) | |
| def positionChanged(self, progress): | |
| progress /= 1000 | |
| if not self.slider.isSliderDown(): | |
| self.slider.setValue(progress) | |
| self.updateDurationInfo(progress) | |
| def metaDataChanged(self): | |
| if self.player.isMetaDataAvailable(): | |
| self.setTrackInfo("%s - %s" % ( | |
| self.player.metaData(QMediaMetaData.AlbumArtist), | |
| self.player.metaData(QMediaMetaData.Title))) | |
| def previousClicked(self): | |
| # Go to the previous track if we are within the first 5 seconds of | |
| # playback. Otherwise, seek to the beginning. | |
| if self.player.position() <= 5000: | |
| self.playlist.previous() | |
| else: | |
| self.player.setPosition(0) | |
| def jump(self, index): | |
| if index.isValid(): | |
| self.playlist.setCurrentIndex(index.row()) | |
| self.player.play() | |
| def playlistPositionChanged(self, position): | |
| self.playlistView.setCurrentIndex( | |
| self.playlistModel.index(position, 0)) | |
| def seek(self, seconds): | |
| self.player.setPosition(seconds * 1000) | |
| def statusChanged(self, status): | |
| self.handleCursor(status) | |
| if status == QMediaPlayer.LoadingMedia: | |
| self.setStatusInfo("Loading...") | |
| elif status == QMediaPlayer.StalledMedia: | |
| self.setStatusInfo("Media Stalled") | |
| elif status == QMediaPlayer.EndOfMedia: | |
| QApplication.alert(self) | |
| elif status == QMediaPlayer.InvalidMedia: | |
| self.displayErrorMessage() | |
| else: | |
| self.setStatusInfo("") | |
| def handleCursor(self, status): | |
| if status in (QMediaPlayer.LoadingMedia, QMediaPlayer.BufferingMedia, QMediaPlayer.StalledMedia): | |
| self.setCursor(Qt.BusyCursor) | |
| else: | |
| self.unsetCursor() | |
| def bufferingProgress(self, progress): | |
| self.setStatusInfo("Buffering %d%" % progress) | |
| def videoAvailableChanged(self, available): | |
| if available: | |
| self.fullScreenButton.clicked.connect( | |
| self.videoWidget.setFullScreen) | |
| self.videoWidget.fullScreenChanged.connect( | |
| self.fullScreenButton.setChecked) | |
| if self.fullScreenButton.isChecked(): | |
| self.videoWidget.setFullScreen(True) | |
| else: | |
| self.fullScreenButton.clicked.disconnect( | |
| self.videoWidget.setFullScreen) | |
| self.videoWidget.fullScreenChanged.disconnect( | |
| self.fullScreenButton.setChecked) | |
| self.videoWidget.setFullScreen(False) | |
| def setTrackInfo(self, info): | |
| self.trackInfo = info | |
| if self.statusInfo != "": | |
| self.setWindowTitle("%s | %s" % (self.trackInfo, self.statusInfo)) | |
| else: | |
| self.setWindowTitle(self.trackInfo) | |
| def setStatusInfo(self, info): | |
| self.statusInfo = info | |
| if self.statusInfo != "": | |
| self.setWindowTitle("%s | %s" % (self.trackInfo, self.statusInfo)) | |
| else: | |
| self.setWindowTitle(self.trackInfo) | |
| def displayErrorMessage(self): | |
| self.setStatusInfo(self.player.errorString()) | |
| def updateDurationInfo(self, currentInfo): | |
| duration = self.duration | |
| if currentInfo or duration: | |
| currentTime = QTime((currentInfo/3600)%60, (currentInfo/60)%60, | |
| currentInfo%60, (currentInfo*1000)%1000) | |
| totalTime = QTime((duration/3600)%60, (duration/60)%60, | |
| duration%60, (duration*1000)%1000); | |
| format = 'hh:mm:ss' if duration > 3600 else 'mm:ss' | |
| tStr = currentTime.toString(format) + " / " + totalTime.toString(format) | |
| else: | |
| tStr = "" | |
| self.labelDuration.setText(tStr) | |
| if __name__ == '__main__': | |
| import sys | |
| app = QApplication(sys.argv) | |
| player = Player(sys.argv[1:]) | |
| w = QMainWindow() | |
| w.setCentralWidget(player) | |
| w.statusBar() | |
| w.show() | |
| sys.exit(app.exec_()) |
you made it possible for me to complete the work by the deadline
Thank you so much for sharing this, very appreciated. There are a few hiccups here and there.
- Warnings about conversion to int that might get deprecated, so I changed it explicitly using
int()in a few places QWidget::paintEngine: Should no longer be calledwhen resizing the window. Not the end of the world but 🤷
Most important thing is:
- Your counter only goes up. If the slider is moved as rewinding, the frame count keeps increasing.
I understand this is a problematic issue with the library itself (somehow the library doesn't even have a frame counter implemented). I happen to have a separate file with timestamps of my camera so I can find the closest frame index to the current seconds. Alternatively one could assume a frame rate as shown in the code below but it might have certain issues with cameras that don't function at the proper fps
def seek(self, seconds):
self.player.setPosition(seconds * 1000)
# expressed in milliseconds since the beginning of the media
position_ms = self.player.position()
# update the framecounter to the proper position assuming constant framerate (duration / fps ?)
current_frame = ...
self.frameCounter.frame_cnt = int(current_frame)
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hello, thank u sooo much bro ))))