57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""Test liveview avec Pygame. Lance: python3 scripts/test_pygame.py"""
|
|
import time
|
|
import io
|
|
import gphoto2 as gp
|
|
import pygame
|
|
|
|
cam = gp.Camera()
|
|
cam.init()
|
|
print("Camera connectee")
|
|
|
|
pygame.init()
|
|
screen = pygame.display.set_mode((960, 640))
|
|
pygame.display.set_caption("Test liveview Pygame")
|
|
clock = pygame.time.Clock()
|
|
font = pygame.font.SysFont(None, 36)
|
|
|
|
t_start = time.time()
|
|
frame_count = 0
|
|
premier_frame = None
|
|
|
|
running = True
|
|
while running:
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT or (event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE):
|
|
running = False
|
|
|
|
try:
|
|
t0 = time.time()
|
|
f = cam.capture_preview()
|
|
data = bytes(f.get_data_and_size())
|
|
dt = time.time() - t0
|
|
|
|
if premier_frame is None:
|
|
premier_frame = time.time() - t_start
|
|
print(f"1ere image en : {premier_frame:.2f}s")
|
|
|
|
img_surface = pygame.image.load(io.BytesIO(data))
|
|
img_surface = pygame.transform.scale(img_surface, (960, 640))
|
|
screen.blit(img_surface, (0, 0))
|
|
|
|
frame_count += 1
|
|
fps = frame_count / (time.time() - t_start)
|
|
txt = font.render(f"FPS: {fps:.1f} frame: {dt*1000:.0f}ms", True, (255, 255, 0))
|
|
screen.blit(txt, (10, 10))
|
|
pygame.display.flip()
|
|
|
|
except Exception as e:
|
|
print(f"Erreur: {e}")
|
|
break
|
|
|
|
clock.tick(30)
|
|
|
|
cam.exit()
|
|
pygame.quit()
|
|
print(f"Total frames: {frame_count}, FPS moyen: {frame_count/(time.time()-t_start):.1f}")
|