mlaga97/PySimpleGUI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Better camera control for droid

Browse source
This commit is contained in:
PySimpleGUI 2019-12-08 12:50:16 -05:00
parent fa24ab4607
commit 68a0c79d1a
3 changed files with 77 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

62
DemoPrograms/PyDroid3/Demo_OpenCV_Webcam_ASCII.py
View file

@ -3,8 +3,12 @@ import cv2
from PIL import Image
import numpy as np
import PySimpleGUI as sg
font_size = 4
USING_QT = False
CAMERA_FRONT = 1
CAMERA_REAR = 0
USE_CAMERA = CAMERA_REAR
"""
Interesting program that shows your webcam's image as ASCII text. Runs in realtime, producing a stream of
@ -29,34 +33,48 @@ On Linux / Mac use pip3 instead of pip
chars = np.asarray(list(' .,:;irsXA253hMHGS#9B&@'))
SC, GCF, WCF = .1, 1, 7/4
sg.change_look_and_feel('Black') # make it look cool
sg.change_look_and_feel('Dark Black 1') # make it look cool
# define the window layout
# ---------------- A Quick Little Window - to get camera to use ----------------
win_get_cam = sg.Window('Which Camera?',[[sg.T('Which camera?')],[sg.Combo(['Front', 'Rear'], default_value='Rear', font='any 20')], [sg.Ok()]], location=(0,0))
event, values = win_get_cam.read()
win_get_cam.close()
if event != 'Ok': exit()
USE_CAMERA = [CAMERA_FRONT, CAMERA_REAR][values[0]=='Rear']
# ------------ Turn on camera and read a frame to find the number of lines of text needed ---------
cap = cv2.VideoCapture(USE_CAMERA) # Setup camera using your camera choice
ret, frame = cap.read()
img = Image.fromarray(frame) # create PIL image from frame
GCF = 1.0 # Variable user will be able to change
WCF = 1.75 # Variable user will be able to change
# ----------- magic that coverts the image to ascii -----------
S = (round(img.size[0] * SC * WCF), round(img.size[1] * SC))
img = np.sum(np.asarray(img.resize(S)), axis=2)
img -= img.min()
img = (1.0 - img / img.max()) ** GCF * (chars.size - 1)
# Find number of lines of text that will be drawn so can create the right size of window
NUM_LINES = len(chars[img.astype(int)])
LINE_LENGTH = len("".join(chars[img.astype(int)][0]))
print(f'line len = {LINE_LENGTH}')
# ------------- define the window layout -------------
# number of lines of text elements. Depends on cameras image size and the variable SC (scaller)
NUM_LINES = 65
if USING_QT:
layout = [[sg.Text(i, size_px=(800, 12),
font=('Courier', font_size),
key='-OUT-' + str(i))] for i in range(NUM_LINES)]
else:
layout = [[sg.Text(i, size=(120, 1), font=('Courier', font_size),
pad=(0, 0), key='-OUT-'+str(i))] for i in range(NUM_LINES)]
layout = [[sg.Text(i, size=(LINE_LENGTH, 1), font=('Courier', font_size), pad=(0, 0), key='-OUT-'+str(i))] for i in range(NUM_LINES)]
layout += [[sg.Button('Exit', size=(5, 1)),
sg.Text('GCF', size=(4, 1)),
sg.Spin([round(i, 2) for i in np.arange(0.1, 20.0, 0.1)],
initial_value=1, key='-SPIN-GCF-', size=(5, 1)),
sg.Text('WCF', size=(4, 1)),
sg.Slider((1, 4), resolution=.05, default_value=1.75,
orientation='h', key='-SLIDER-WCF-', size=(15, 15))]]
sg.Text('GCF', size=(4, 1)),
sg.Spin([round(i, 2) for i in np.arange(0.1, 20.0, 0.1)], initial_value=1, key='-SPIN-GCF-', size=(5, 1), font='any 20')],
[sg.Text('WCF', size=(4, 1)),
sg.Slider((1, 4), resolution=.05, default_value=1.75, orientation='h', key='-SLIDER-WCF-', size=(15, 30))]]
# create the window and show it without the plot
window = sg.Window('Demo Application - OpenCV Integration', layout,
location=(0,0), font='Any 10')
# ------------- create the window -------------
window = sg.Window('Demo Application - OpenCV Integration', layout, location=(0,0))
# ---===--- Event LOOP Read and display frames, operate the GUI --- #
# Setup the OpenCV capture device (webcam)
cap = cv2.VideoCapture(0)
while True:
event, values = window.read(timeout=0)