class ControlImage(ControlBase):
_imageWidget = None
def init_form(self):
control_path = tools.getFileInSameDirectory(__file__, "image.ui")
self._form = uic.loadUi(control_path)
self._imageWidget = VideoGLWidget()
self._form.imageLayout.addWidget(self._imageWidget)
super(ControlImage, self).init_form()
def save_form(self, data, path=None):
if type(self.value) is np.ndarray: data['value'] = self._value
@property
def value(self):
"""
This property returns or set what the control should manage or store.
"""
return self._value
@value.setter
def value(self, value):
oldvalue = self._value
self._value = value
self._imageWidget.paint([value])
if (type(oldvalue) is np.ndarray and type(value) is np.ndarray and oldvalue.any()!=value.any()) or \
(type(oldvalue) is np.ndarray and type(value) is not np.ndarray) or \
(type(oldvalue) is not np.ndarray and type(value) is np.ndarray):
self.changed_event()
def init_form(self): # Get the current path of the file rootPath = os.path.dirname(__file__) # Load the UI for the self instance uic.loadUi(os.path.join(rootPath, "video.ui"), self) # Define the icon for the Play button self.videoPlay.setIcon(conf.PYFORMS_ICON_VIDEOPLAYER_PAUSE_PLAY) self.detach_btn.setIcon(conf.PYFORMS_ICON_VIDEOPLAYER_DETACH) self.detach_btn.clicked.connect(self.__detach_player_evt) self._video_widget = VideoGLWidget() self._video_widget._control = self self.videoLayout.addWidget(self._video_widget) self.videoPlay.clicked.connect(self.videoPlay_clicked) self.videoFrames.valueChanged.connect(self.video_frames_value_changed) self.videoProgress.valueChanged.connect(self.videoProgress_valueChanged) self.videoProgress.sliderReleased.connect(self.videoProgress_sliderReleased) self._timer = QtCore.QTimer(self) self._timer.timeout.connect(self.call_next_frame) self.form.horizontalSlider.valueChanged.connect(self.__rotateZ) self.form.verticalSlider.valueChanged.connect(self.__rotateX) self._current_frame = None self.view_in_3D = False
class ControlPlayer(ControlBase, QFrame):
def __init__(self, *args, **kwargs):
self._video_widget = None # GL widget
QFrame.__init__(self)
ControlBase.__init__(self, *args, **kwargs)
self._multiple_files = kwargs.get('multiple_files', False)
self._current_frame = None # current frame image
self._current_frame_index = None # current frame index
self.process_frame_event = kwargs.get('process_frame_event', self.process_frame_event)
self._speed = 1
self.logger = logging.getLogger('pyforms')
self._update_video_frame = True # if true update the spinbox with the current frame
self._update_video_slider = True # if true update the slider with the current frame
def init_form(self):
# Get the current path of the file
rootPath = os.path.dirname(__file__)
# Load the UI for the self instance
uic.loadUi(os.path.join(rootPath, "video.ui"), self)
# Define the icon for the Play button
self.videoPlay.setIcon(conf.PYFORMS_ICON_VIDEOPLAYER_PAUSE_PLAY)
self.detach_btn.setIcon(conf.PYFORMS_ICON_VIDEOPLAYER_DETACH)
self.detach_btn.clicked.connect(self.__detach_player_evt)
self._video_widget = VideoGLWidget()
self._video_widget._control = self
self.videoLayout.addWidget(self._video_widget)
self.videoPlay.clicked.connect(self.videoPlay_clicked)
self.videoFrames.valueChanged.connect(self.video_frames_value_changed)
self.videoProgress.valueChanged.connect(self.videoProgress_valueChanged)
self.videoProgress.sliderReleased.connect(self.videoProgress_sliderReleased)
self._timer = QtCore.QTimer(self)
self._timer.timeout.connect(self.call_next_frame)
self.form.horizontalSlider.valueChanged.connect(self.__rotateZ)
self.form.verticalSlider.valueChanged.connect(self.__rotateX)
self._current_frame = None
self.view_in_3D = False
##########################################################################
############ FUNCTIONS ###################################################
##########################################################################
def play(self):
try:
self.videoPlay.setChecked(True)
self._timer.start( 1000.0/float(self.fps+1) )
except Exception as e:
self.videoPlay.setChecked(False)
logger.error(e, exc_info=True)
def stop(self):
self.videoPlay.setChecked(False)
self._timer.stop()
def hide(self): QFrame.hide(self)
def show(self): QFrame.show(self)
def refresh(self):
if self._current_frame is not None:
frame = self.process_frame_event(self._current_frame.copy())
if isinstance(frame, list) or isinstance(frame, tuple):
self._video_widget.paint(frame)
else:
self._video_widget.paint([frame])
else:
self._video_widget.paint(None)
def save_form(self, data, path=None): return data
def load_form(self, data, path=None): pass
##########################################################################
############ EVENTS ######################################################
##########################################################################
def process_frame_event(self, frame): return frame
@property
def double_click_event(self): return self._video_widget.onDoubleClick
@double_click_event.setter
def double_click_event(self, value): self._video_widget.onDoubleClick = value
@property
def click_event(self): return self._video_widget.onClick
@click_event.setter
def click_event(self, value): self._video_widget.onClick = value
@property
def drag_event(self): return self._video_widget.onDrag
@drag_event.setter
def drag_event(self, value): self._video_widget.onDrag = value
@property
def end_drag_event(self): return self._video_widget.onEndDrag
@end_drag_event.setter
def end_drag_event(self, value): self._video_widget.onEndDrag = value
@property
def key_press_event(self):
return self._video_widget.on_key_press
@key_press_event.setter
def key_press_event(self, value):
self._video_widget.on_key_press = value
@property
def key_release_event(self): return self._video_widget.on_key_release
@key_release_event.setter
def key_release_event(self, value): self._video_widget.on_key_release = value
##########################################################################
############ PROPERTIES ##################################################
##########################################################################
@property
def next_frame_step(self): return self._speed
@next_frame_step.setter
def next_frame_step(self, value): self._speed = value
@property
def view_in_3D(self): return self._video_widget.onEndDrag
@view_in_3D.setter
def view_in_3D(self, value):
self.form.horizontalSlider.setVisible(value)
self.form.verticalSlider.setVisible(value)
@property
def video_index(self): return int(self._value.get(1)) if self._value else None
@video_index.setter
def video_index(self, value): self._value.set(1, value)
@property
def max(self):
if self._value is None or self._value=='':
return 0
return int(self._value.get(7))
@property
def frame(self): return self._current_frame
@frame.setter
def frame(self, value):
if isinstance(value, list) or isinstance(value, tuple):
self._video_widget.paint(value)
elif value is not None:
self._video_widget.paint([value])
else:
self._video_widget.paint(None)
QApplication.processEvents()
@property
def fps(self):
"""
Return the video frames per second
"""
return self._value.get(5)
@property
def help_text(self): return self._video_widget._helpText
@help_text.setter
def help_text(self, value): self._video_widget._helpText = value
@property
def form(self): return self
@property
def frame_width(self): return self._value.get(3)
@property
def frame_height(self): return self._value.get(4)
@property
def is_playing(self): return self._timer.isActive()
@property
def value(self): return ControlBase.value.fget(self)
@value.setter
def value(self, value):
self.form.setUpdatesEnabled(False)
if value is None:
self.stop()
self.videoControl.setEnabled(False)
self.refresh()
self._video_widget.reset()
if value == 0:
self._value = cv2.VideoCapture(0)
elif isinstance(value, str) and value:
open_multiplefiles = self._multiple_files
if open_multiplefiles:
open_multiplefiles = len(MultipleVideoCapture.search_files(value))>0
if open_multiplefiles:
msg = "Multiple files were found with the same name, do you wish to combine then in a single video?\n\n"
for filepath in MultipleVideoCapture.search_files(value):
msg += "- {filename}\n".format(filename=os.path.basename(filepath))
reply = QMessageBox(
QMessageBox.Question,
'Open multiple files',
msg,
QMessageBox.No | QMessageBox.Yes
).exec_()
if reply == QMessageBox.Yes:
open_multiplefiles = True
else:
open_multiplefiles = False
if open_multiplefiles:
self._value = MultipleVideoCapture(value)
else:
self._value = cv2.VideoCapture(value)
else:
self._value = value
if self._value and value != 0:
self.videoProgress.setMinimum(0)
self.videoProgress.setValue(0)
self.videoProgress.setMaximum(
self._value.get(7))
self.videoFrames.setMinimum(0)
self.videoFrames.setValue(0)
self.videoFrames.setMaximum(
self._value.get(7))
if self._value:
self.videoControl.setEnabled(True)
self.refresh()
self.form.setUpdatesEnabled(True)
##########################################################################
############ PRIVATE FUNCTIONS ###########################################
##########################################################################
def __rotateX(self):
self._video_widget.rotateX = self.form.verticalSlider.value()
self.refresh()
def __rotateZ(self):
self._video_widget.rotateZ = self.form.horizontalSlider.value()
self.refresh()
def call_next_frame(self, update_slider=True, update_number=True, increment_frame=True):
# move the player to the next frame
self.form.setUpdatesEnabled(False)
self._current_frame_index = self.video_index
# if the player is not visible, stop
if not self.visible:
self.stop()
self.form.setUpdatesEnabled(True)
return
# if no video is selected
if self.value is None:
self._current_frame = None
self._current_frame_index = None
return
# read next frame
(success, self._current_frame) = self.value.read()
# increment frame index if the step is bigger than 1
if increment_frame and self.next_frame_step > 1:
self.video_index += self.next_frame_step
# no frame available. leave the function
if not success:
self.stop()
self.form.setUpdatesEnabled(True)
return
frame = self.process_frame_event(
self._current_frame.copy()
)
# draw the frame
if isinstance(frame, list) or isinstance(frame, tuple):
self._video_widget.paint(frame)
else:
self._video_widget.paint([frame])
if not self.videoProgress.isSliderDown():
if update_slider and self._update_video_slider:
self._update_video_slider = False
self.videoProgress.setValue(self._current_frame_index)
self._update_video_slider = True
if update_number:
self._update_video_frame = False
self.videoFrames.setValue(self._current_frame_index)
self._update_video_frame = True
self.form.setUpdatesEnabled(True)
def __detach_player_evt(self):
"""
Called by the detach button
"""
self._old_layout = self.parentWidget().layout()
self._old_layout_index = self._old_layout.indexOf(self)
self._detach_win = QMainWindow(parent=self.parent)
self._detach_win.setWindowTitle('Player')
self._detach_win.setCentralWidget(self)
self.detach_btn.hide()
self._detach_win.closeEvent = self.__detach_win_closed_evt
self._detach_win.show()
def __detach_win_closed_evt(self, event):
"""
Called when the detached window is closed
"""
self._old_layout.insertWidget(self._old_layout_index, self)
self.detach_btn.show()
self._detach_win.close()
del self._detach_win
def videoPlay_clicked(self):
"""Slot for Play/Pause functionality."""
if self.is_playing:
self.stop()
else:
self.play()
def convertFrameToTime(self, totalMilliseconds):
# totalMilliseconds = totalMilliseconds*(1000.0/self._value.get(5))
if math.isnan(totalMilliseconds): return 0, 0, 0
totalseconds = int(totalMilliseconds / 1000)
minutes = int(totalseconds / 60)
seconds = totalseconds - (minutes * 60)
milliseconds = totalMilliseconds - (totalseconds * 1000)
return (minutes, seconds, milliseconds)
def videoProgress_valueChanged(self):
milli = self._value.get(0)
(minutes, seconds, milliseconds) = self.convertFrameToTime(milli)
self.videoTime.setText(
"%02d:%02d:%03d" % (minutes, seconds, milliseconds))
def videoProgress_sliderReleased(self):
if not self.is_playing and self._update_video_slider:
new_index = self.videoProgress.value()
self._value.set(1, new_index)
self.call_next_frame(update_slider=False, increment_frame=False)
def video_frames_value_changed(self, pos):
if not self.is_playing and self._update_video_frame:
self._value.set(1, pos) # set the video position
self.call_next_frame(update_number=False, increment_frame=False)
def jump_forward(self):
self.video_index += 20 * self.fps
def jump_backward(self):
self.video_index -= 20 * self.fps
def init_form(self):
control_path = tools.getFileInSameDirectory(__file__, "image.ui")
self._form = uic.loadUi(control_path)
self._imageWidget = VideoGLWidget()
self._form.imageLayout.addWidget(self._imageWidget)
super(ControlImage, self).init_form()
class ControlPlayer(ControlBase, QFrame):
_videoWidget = None
_currentFrame = None
def __init__(self, *args, **kwargs):
QFrame.__init__(self)
ControlBase.__init__(self, *args, **kwargs)
self.process_frame_event = kwargs.get('process_frame_event', self.process_frame_event)
self._speed = 1
self.logger = logging.getLogger('pyforms')
self._updateVideoFrame = True
def init_form(self):
# Get the current path of the file
rootPath = os.path.dirname(__file__)
# Load the UI for the self instance
uic.loadUi(os.path.join(rootPath, "video.ui"), self)
# Define the icon for the Play button
self.videoPlay.setIcon(conf.PYFORMS_ICON_VIDEOPLAYER_PAUSE_PLAY)
self._videoWidget = VideoGLWidget()
self._videoWidget._control = self
self.videoLayout.addWidget(self._videoWidget)
self.videoPlay.clicked.connect(self.videoPlay_clicked)
self.videoFrames.valueChanged.connect(self.videoFrames_valueChanged)
self.videoProgress.valueChanged.connect(self.videoProgress_valueChanged)
self.videoProgress.sliderReleased.connect(self.videoProgress_sliderReleased)
self._timer = QtCore.QTimer(self)
self._timer.timeout.connect(self.update_frame)
self.form.horizontalSlider.valueChanged.connect(self.__rotateZ)
self.form.verticalSlider.valueChanged.connect(self.__rotateX)
self._currentFrame = None
self.view_in_3D = False
##########################################################################
############ FUNCTIONS ###################################################
##########################################################################
def play(self):
self.videoPlay.setChecked(True)
self._timer.start( 1000.0/float(self.fps+1) )
def stop(self):
self.videoPlay.setChecked(False)
self._timer.stop()
def hide(self): QFrame.hide(self)
def show(self): QFrame.show(self)
def refresh(self):
if self._currentFrame is not None:
frame = self.process_frame_event(self._currentFrame.copy())
if isinstance(frame, list) or isinstance(frame, tuple):
self._videoWidget.paint(frame)
else:
self._videoWidget.paint([frame])
else:
self._videoWidget.paint(None)
def save_form(self, data, path=None): return data
def load_form(self, data, path=None): pass
##########################################################################
############ EVENTS ######################################################
##########################################################################
def process_frame_event(self, frame): return frame
@property
def double_click_event(self): return self._videoWidget.onDoubleClick
@double_click_event.setter
def double_click_event(self, value): self._videoWidget.onDoubleClick = value
@property
def click_event(self): return self._videoWidget.onClick
@click_event.setter
def click_event(self, value): self._videoWidget.onClick = value
@property
def drag_event(self): return self._videoWidget.onDrag
@drag_event.setter
def drag_event(self, value): self._videoWidget.onDrag = value
@property
def end_drag_event(self): return self._videoWidget.onEndDrag
@end_drag_event.setter
def end_drag_event(self, value): self._videoWidget.onEndDrag = value
@property
def key_release_event(self): return self._videoWidget.on_key_release
@key_release_event.setter
def key_release_event(self, value): self._videoWidget.on_key_release = value
##########################################################################
############ PROPERTIES ##################################################
##########################################################################
@property
def next_frame_step(self): return self._speed
@next_frame_step.setter
def next_frame_step(self, value): self._speed = value
@property
def view_in_3D(self): return self._videoWidget.onEndDrag
@view_in_3D.setter
def view_in_3D(self, value):
self.form.horizontalSlider.setVisible(value)
self.form.verticalSlider.setVisible(value)
@property
def video_index(self): return int(self._value.get(1)) if self._value else None
@video_index.setter
def video_index(self, value): self._value.set(1, value)
@property
def max(self): return int(self._value.get(7))
@property
def frame(self): return self._currentFrame
@frame.setter
def frame(self, value):
if isinstance(value, list) or isinstance(value, tuple):
self._videoWidget.paint(value)
elif value is not None:
self._videoWidget.paint([value])
else:
self._videoWidget.paint(None)
QApplication.processEvents()
@property
def fps(self):
"""
Return the video frames per second
"""
return self._value.get(5)
@property
def help_text(self): return self._videoWidget._helpText
@help_text.setter
def help_text(self, value): self._videoWidget._helpText = value
@property
def form(self): return self
@property
def frame_width(self): return self._value.get(3)
@property
def frame_height(self): return self._value.get(4)
@property
def is_playing(self): return self._timer.isActive()
@property
def value(self): return ControlBase.value.fget(self)
@value.setter
def value(self, value):
self.form.setUpdatesEnabled(False)
if value is None:
self.stop()
self.videoControl.setEnabled(False)
self.refresh()
self._videoWidget.reset()
if value == 0:
self._value = cv2.VideoCapture(0)
elif isinstance(value, str) and value:
self._value = cv2.VideoCapture(value)
else:
self._value = value
if self._value and value != 0:
self.videoProgress.setMinimum(0)
self.videoProgress.setValue(0)
self.videoProgress.setMaximum(
self._value.get(7))
self.videoFrames.setMinimum(0)
self.videoFrames.setValue(0)
self.videoFrames.setMaximum(
self._value.get(7))
if self._value:
self.videoControl.setEnabled(True)
self.refresh()
self.form.setUpdatesEnabled(True)
##########################################################################
############ PRIVATE FUNCTIONS ###########################################
##########################################################################
def __rotateX(self):
self._videoWidget.rotateX = self.form.verticalSlider.value()
self.refresh()
def __rotateZ(self):
self._videoWidget.rotateZ = self.form.horizontalSlider.value()
self.refresh()
def update_frame(self):
self.form.setUpdatesEnabled(False)
if not self.visible: self.stop()
if self.value is None:
self._currentFrame = None
return
if self.next_frame_step>1: self.video_index += self.next_frame_step
(success, frame) = self.value.read()
if not success: self.stop()
if frame is not None: self._currentFrame = frame
if self._currentFrame is not None:
frame = self.process_frame_event(self._currentFrame.copy())
if isinstance(frame, list) or isinstance(frame, tuple):
self._videoWidget.paint(frame)
else:
self._videoWidget.paint([frame])
if not self.videoProgress.isSliderDown():
currentFrame = self.video_index
self.videoProgress.setValue(currentFrame)
if self._updateVideoFrame:
self.videoFrames.setValue(currentFrame)
self.form.setUpdatesEnabled(True)
def videoPlay_clicked(self):
"""Slot for Play/Pause functionality."""
if self.is_playing:
self.stop()
else:
self.play()
def convertFrameToTime(self, totalMilliseconds):
# totalMilliseconds = totalMilliseconds*(1000.0/self._value.get(5))
if math.isnan(totalMilliseconds): return 0, 0, 0
totalseconds = int(totalMilliseconds / 1000)
minutes = int(totalseconds / 60)
seconds = totalseconds - (minutes * 60)
milliseconds = totalMilliseconds - (totalseconds * 1000)
return (minutes, seconds, milliseconds)
def videoProgress_valueChanged(self):
milli = self._value.get(0)
(minutes, seconds, milliseconds) = self.convertFrameToTime(milli)
self.videoTime.setText(
"%02d:%02d:%03d" % (minutes, seconds, milliseconds))
def videoProgress_sliderReleased(self):
jump2Frame = self.videoProgress.value()
self._value.set(1, jump2Frame)
self._updateVideoFrame = False
self.videoFrames.setValue(jump2Frame)
self._value.set(1, jump2Frame)
self._updateVideoFrame = True
def videoFrames_valueChanged(self, i):
if not self.is_playing:
jump2Frame = self.videoProgress.value()
diff = jump2Frame - i
self._value.set(1, jump2Frame - diff)
self._updateVideoFrame = False
self.update_frame()
self._updateVideoFrame = True