Update to the latest coding guidelines. Use PPM encoding as it's been shown to be more efficient

This commit is contained in:
PySimpleGUI 2022-06-01 15:44:19 -04:00
parent 051fa57d15
commit 11319709e1
1 changed files with 24 additions and 22 deletions

View File

@ -1,16 +1,18 @@
#!/usr/bin/env python #!/usr/bin/env python
import PySimpleGUI as sg import PySimpleGUI as sg
from PIL import Image
import cv2 as cv import cv2 as cv
import io # from PIL import Image
# import io
""" """
Demo program to open and play a file using OpenCV Demo program to open and play a file using OpenCV
It's main purpose is to show you: It's main purpose is to show you:
1. How to get a frame at a time from a video file using OpenCV 1. How to get a frame at a time from a video file using OpenCV
2. How to display an image in a PySimpleGUI Window 2. How to display an image in a PySimpleGUI Window
For added fun, you can reposition the video using the slider. For added fun, you can reposition the video using the slider.
Copyright 2022 PySimpleGUI
""" """
@ -28,37 +30,39 @@ def main():
# ---===--- define the window layout --- # # ---===--- define the window layout --- #
layout = [[sg.Text('OpenCV Demo', size=(15, 1), font='Helvetica 20')], layout = [[sg.Text('OpenCV Demo', size=(15, 1), font='Helvetica 20')],
[sg.Image(filename='', key='-image-')], [sg.Image(key='-IMAGE-')],
[sg.Slider(range=(0, num_frames), [sg.Slider(range=(0, num_frames), size=(60, 10), orientation='h', key='-SLIDER-')],
size=(60, 10), orientation='h', key='-slider-')], [sg.Push(), sg.Button('Exit', font='Helvetica 14')]]
[sg.Button('Exit', size=(7, 1), pad=((600, 0), 3), font='Helvetica 14')]]
# create the window and show it without the plot # create the window and show it without the plot
window = sg.Window('Demo Application - OpenCV Integration', layout, no_titlebar=False, location=(0, 0)) window = sg.Window('Demo Application - OpenCV Integration', layout, no_titlebar=False, location=(0, 0))
# locate the elements we'll be updating. Does the search only 1 time # locate the elements we'll be updating. Does the search only 1 time
image_elem = window['-image-'] image_elem = window['-IMAGE-']
slider_elem = window['-slider-'] slider_elem = window['-SLIDER-']
timeout = 1000//fps # time in ms to use for window reads
# ---===--- LOOP through video file by frame --- # # ---===--- LOOP through video file by frame --- #
cur_frame = 0 cur_frame = 0
while vidFile.isOpened(): while vidFile.isOpened():
event, values = window.read(timeout=0) event, values = window.read(timeout=timeout)
if event in ('Exit', None): if event in ('Exit', None):
break break
ret, frame = vidFile.read() ret, frame = vidFile.read()
if not ret: # if out of data stop looping if not ret: # if out of data stop looping
break break
# if someone moved the slider manually, the jump to that frame # if someone moved the slider manually, the jump to that frame
if int(values['-slider-']) != cur_frame-1: if int(values['-SLIDER-']) != cur_frame-1:
cur_frame = int(values['-slider-']) cur_frame = int(values['-SLIDER-'])
vidFile.set(cv.CAP_PROP_POS_FRAMES, cur_frame) vidFile.set(cv.CAP_PROP_POS_FRAMES, cur_frame)
slider_elem.update(cur_frame) slider_elem.update(cur_frame)
cur_frame += 1 cur_frame += 1
imgbytes = cv.imencode('.png', frame)[1].tobytes() # ditto imgbytes = cv.imencode('.ppm', frame)[1].tobytes() # can also use png. ppm found to be more efficient
image_elem.update(data=imgbytes) image_elem.update(data=imgbytes)
main()
############# #############
# | | # # | | #
# | | # # | | #
@ -75,5 +79,3 @@ def main():
imgbytes = bio.getvalue() # this can be used by OpenCV hopefully imgbytes = bio.getvalue() # this can be used by OpenCV hopefully
image_elem.update(data=imgbytes) image_elem.update(data=imgbytes)
""" """
main()