camera.py

#full

import pygame
import pygame.camera
import subprocess
import random
from weirdimage import transforms
import threading
import ui
from aspectscale import aspect_scale

pygame.camera.init()

camsize = (640, 480)

camlist = pygame.camera.list_cameras()
if not camlist: exit("No cameras detected")
cam = pygame.camera.Camera(camlist[0], camsize)
cam.start()
camimg = pygame.Surface((0, 0))
filterimg = pygame.Surface((0, 0))

screensize = [camsize[0] * 2, camsize[1]]
screen = pygame.display.set_mode(screensize, pygame.RESIZABLE)
pygame.display.set_caption("Weird Image (Camera)")

pygame.font.init()
ui.init(screen, pygame.font.SysFont(pygame.font.get_default_font(), 30))
filter = ui.menu("Select Transform", [*[x.__name__ for x in transforms], "", "All"])
if filter > len(transforms):
	filter = False
else:
	filter = transforms[filter]

def filterasync():
	global filterimg
	if filter == False:
		t = transforms.copy()
		random.shuffle(t)
		i = camimg.copy()
		for f in t:
			i = f(i)
			filterimg = i.copy()
	else:
		i = camimg.copy()
		filterimg = filter(i)

running = True
c = pygame.time.Clock()
while running:
	for event in pygame.event.get():
		if event.type == pygame.QUIT:
			running = False
		elif event.type == pygame.MOUSEBUTTONDOWN:
			_camimg = camimg.copy()
			_filterimg = filterimg.copy()
			camimgsize = aspect_scale(_camimg, screensize[0] / 2, screensize[1] / 2)
			filterimgsize = aspect_scale(_filterimg, screensize[0] / 2, screensize[1] / 2)
			u = ui.UI()
			u.add(ui.Header("Select Image"))
			u.add(ui.Button("Camera Image"))
			u.add(ui.Image(pygame.transform.scale(_camimg, camimgsize)))
			u.add(ui.Button("Filtered Image"))
			u.add(ui.Image(pygame.transform.scale(_filterimg, filterimgsize)))
			u.add(ui.Button("Cancel"))
			choice = ui.uimenu(u)
			if choice == 1:
				pygame.image.save(_camimg, "camera_image.png")
				subprocess.Popen(["python3", "imgopen.py", "camera_image.png"])
			elif choice == 3:
				pygame.image.save(_filterimg, "camera_image.png")
				subprocess.Popen(["python3", "imgopen.py", "camera_image.png"])
		elif event.type == pygame.VIDEORESIZE:
			screensize = [*event.size]
			#screen = pygame.display.set_mode(screensize, pygame.RESIZABLE)
			# I sure hope you have Pygame 2.0
	# Get images
	c = pygame.time.Clock()
	while cam.query_image():
		camimg = cam.get_image()
	if threading.active_count() < 2: threading.Thread(target=filterasync, name="filter").start()
	err = True
	while err:
		try:
			screen.fill((255, 255, 255))
			imgsize = aspect_scale(camimg, screensize[0] / 2, screensize[1])
			screen.blit(pygame.transform.scale(camimg, imgsize), (0, 0))
			imgsize = aspect_scale(filterimg, screensize[0] / 2, screensize[1])
			imgsize = pygame.transform.scale(filterimg, imgsize)
			screen.blit(imgsize, (screensize[0] - imgsize.get_width(), 0))
			err = False
		except: err = True
	pygame.display.flip()
	c.tick(60)