You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
162 lines
4.6 KiB
Python
162 lines
4.6 KiB
Python
# importing libraries
|
|
import cv2
|
|
import numpy as np
|
|
import os
|
|
import random
|
|
import simpleaudio as sa
|
|
import threading
|
|
import time
|
|
|
|
MEDIA_FOLDER = "media"
|
|
WINDOW_NAME = "Image Viewer"
|
|
|
|
WINDOW_WIDTH = 800
|
|
WINDOW_HEIGHT = 480
|
|
|
|
|
|
mediaFolder = os.listdir(MEDIA_FOLDER)
|
|
current_file = False
|
|
|
|
def clear_frame():
|
|
# Create a black frame
|
|
black_frame = np.zeros((WINDOW_HEIGHT, WINDOW_WIDTH, 3), dtype=np.uint8)
|
|
cv2.imshow(WINDOW_NAME, black_frame)
|
|
cv2.waitKey(1) # Add a small delay to ensure the frame is displayed
|
|
|
|
|
|
def resize_and_center_frame(frame):
|
|
# Get original frame dimensions
|
|
height, width = frame.shape[:2]
|
|
|
|
# Calculate aspect ratio
|
|
aspect_ratio = width / height
|
|
|
|
# Determine resizing dimensions while maintaining aspect ratio
|
|
if width > height:
|
|
new_width = WINDOW_WIDTH
|
|
new_height = int(new_width / aspect_ratio)
|
|
else:
|
|
new_height = WINDOW_HEIGHT
|
|
new_width = int(new_height * aspect_ratio)
|
|
|
|
# Resize the frame
|
|
resized_frame = cv2.resize(frame, (new_width, new_height))
|
|
|
|
# Calculate offset to center the resized frame in the window
|
|
x_offset = max(0, (WINDOW_WIDTH - new_width) // 2)
|
|
y_offset = max(0, (WINDOW_HEIGHT - new_height) // 2)
|
|
|
|
# Create a black background of the window size
|
|
window_background = np.zeros((WINDOW_HEIGHT, WINDOW_WIDTH, 3), dtype=np.uint8)
|
|
|
|
# Overlay the resized frame onto the background at the calculated offset
|
|
window_background[
|
|
y_offset : y_offset + new_height, x_offset : x_offset + new_width
|
|
] = resized_frame
|
|
|
|
return window_background
|
|
|
|
|
|
# Show an image on the canvas
|
|
def showImage(file):
|
|
# Read the image
|
|
current_image = cv2.imread(MEDIA_FOLDER + "/" + file)
|
|
|
|
if current_image is None:
|
|
print(f"Error loading image: {image_path}")
|
|
return
|
|
a = cv2.cvtColor(current_image, cv2.COLOR_BGR2GRAY)
|
|
resized_frame = resize_and_center_frame(current_image)
|
|
cv2.imshow(WINDOW_NAME, resized_frame)
|
|
|
|
k = cv2.waitKey(0)
|
|
if k < 58 and k > 48:
|
|
print("number is pressed")
|
|
print(k - 48)
|
|
load_media(k - 48)
|
|
elif k == ord("q"):
|
|
print("destroy all windows")
|
|
cv2.destroyAllWindows()
|
|
else:
|
|
print("key pressed: " + str(k))
|
|
|
|
|
|
def play_audio(file):
|
|
clear_frame()
|
|
sound = sa.WaveObject.from_wave_file("media/" + file)
|
|
play_obj = sound.play()
|
|
|
|
# TODO: check for sensor data change
|
|
while play_obj.is_playing():
|
|
# Add a small delay to prevent high CPU usage
|
|
time.sleep(0.1)
|
|
load_media()
|
|
|
|
def get_random_file(previous=None):
|
|
random.shuffle(mediaFolder)
|
|
file = mediaFolder[0]
|
|
if file == previous:
|
|
return get_random_file(previous)
|
|
return file
|
|
|
|
|
|
# create a function with an optional parameter
|
|
def load_media(region=None, override=None):
|
|
global current_file
|
|
|
|
if region:
|
|
print("i should now load a media from region ")
|
|
print(region)
|
|
else:
|
|
print("region is not defined, just do something!")
|
|
|
|
if(override):
|
|
file = override
|
|
else:
|
|
file = get_random_file(current_file)
|
|
|
|
current_file = file
|
|
|
|
# Check if the file is an image or a video
|
|
if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"):
|
|
# Read and resize the image
|
|
print("ive loaded an image: " + file)
|
|
showImage(file)
|
|
elif file.endswith(".mov") or file.endswith(".mp4") or file.endswith(".avi"):
|
|
# Create a VideoCapture object and read from input file
|
|
cap = cv2.VideoCapture("media/" + file)
|
|
print("ive loaded a video" + file)
|
|
|
|
# show the input video file
|
|
while cap.isOpened():
|
|
ret, frame = cap.read()
|
|
# resized = cv2.resize(frame, (800, 480))
|
|
resized_frame = resize_and_center_frame(frame)
|
|
# if q is pressed, the app is closed
|
|
|
|
if ret == True:
|
|
cv2.imshow(WINDOW_NAME, resized_frame)
|
|
if cv2.waitKey(3) & 0xFF == ord("1"):
|
|
print("key 2 pressed")
|
|
load_media()
|
|
break
|
|
elif cv2.waitKey(3) & 0xFF == ord("q"):
|
|
print("destroy all windows")
|
|
cv2.destroyAllWindows()
|
|
break
|
|
else:
|
|
print("ois the video finished?")
|
|
load_media()
|
|
break
|
|
elif file.endswith(".wav") or file.endswith(".mp3"):
|
|
print("ive loaded an audio file: " + file)
|
|
play_audio(file)
|
|
|
|
else:
|
|
print("Unsupported file format: " + file)
|
|
load_media(region)
|
|
|
|
|
|
print("start script ")
|
|
load_media()
|