Intelegentny_Pszczelarz/.venv/Lib/site-packages/pygame/examples/scaletest.py

155 lines
4.7 KiB
Python
Raw Normal View History

2023-03-18 12:55:22 +01:00
#!/usr/bin/env python
""" pygame.examples.scaletest
Shows an interactive image scaler.
"""
import sys
import time
import pygame as pg
def main(imagefile, convert_alpha=False, run_speed_test=False):
"""show an interactive image scaler
Args:
imagefile - name of source image (required)
convert_alpha - use convert_alpha() on the surf (default False)
run_speed_test - (default False)
"""
# initialize display
pg.display.init()
# load background image
background = pg.image.load(imagefile)
if run_speed_test:
if convert_alpha:
# convert_alpha() requires the display mode to be set
pg.display.set_mode((1, 1))
background = background.convert_alpha()
SpeedTest(background)
return
# start fullscreen mode
screen = pg.display.set_mode((1024, 768), pg.FULLSCREEN)
if convert_alpha:
background = background.convert_alpha()
# turn off the mouse pointer
pg.mouse.set_visible(0)
# main loop
bRunning = True
bUp = False
bDown = False
bLeft = False
bRight = False
cursize = [background.get_width(), background.get_height()]
while bRunning:
image = pg.transform.smoothscale(background, cursize)
imgpos = image.get_rect(centerx=512, centery=384)
screen.fill((255, 255, 255))
screen.blit(image, imgpos)
pg.display.flip()
for event in pg.event.get():
if event.type == pg.QUIT or (
event.type == pg.KEYDOWN and event.key == pg.K_ESCAPE
):
bRunning = False
if event.type == pg.KEYDOWN:
if event.key == pg.K_UP:
bUp = True
if event.key == pg.K_DOWN:
bDown = True
if event.key == pg.K_LEFT:
bLeft = True
if event.key == pg.K_RIGHT:
bRight = True
if event.type == pg.KEYUP:
if event.key == pg.K_UP:
bUp = False
if event.key == pg.K_DOWN:
bDown = False
if event.key == pg.K_LEFT:
bLeft = False
if event.key == pg.K_RIGHT:
bRight = False
if bUp:
cursize[1] -= 2
if cursize[1] < 1:
cursize[1] = 1
if bDown:
cursize[1] += 2
if bLeft:
cursize[0] -= 2
if cursize[0] < 1:
cursize[0] = 1
if bRight:
cursize[0] += 2
pg.quit()
def SpeedTest(image):
print(f"\nImage Scaling Speed Test - Image Size {str(image.get_size())}\n")
imgsize = [image.get_width(), image.get_height()]
duration = 0.0
for i in range(128):
shrinkx = (imgsize[0] * i) // 128
shrinky = (imgsize[1] * i) // 128
start = time.time()
tempimg = pg.transform.smoothscale(image, (shrinkx, shrinky))
duration += time.time() - start
del tempimg
print(f"Average transform.smoothscale shrink time: {duration / 128 * 1000:.4f} ms.")
duration = 0.0
for i in range(128):
expandx = (imgsize[0] * (i + 129)) // 128
expandy = (imgsize[1] * (i + 129)) // 128
start = time.time()
tempimg = pg.transform.smoothscale(image, (expandx, expandy))
duration += time.time() - start
del tempimg
print(f"Average transform.smoothscale expand time: {duration / 128 * 1000:.4f} ms.")
duration = 0.0
for i in range(128):
shrinkx = (imgsize[0] * i) // 128
shrinky = (imgsize[1] * i) // 128
start = time.time()
tempimg = pg.transform.scale(image, (shrinkx, shrinky))
duration += time.time() - start
del tempimg
print(f"Average transform.scale shrink time: {duration / 128 * 1000:.4f} ms.")
duration = 0.0
for i in range(128):
expandx = (imgsize[0] * (i + 129)) // 128
expandy = (imgsize[1] * (i + 129)) // 128
start = time.time()
tempimg = pg.transform.scale(image, (expandx, expandy))
duration += time.time() - start
del tempimg
print(f"Average transform.scale expand time: {duration / 128 * 1000:.4f} ms.")
if __name__ == "__main__":
# check input parameters
if len(sys.argv) < 2:
print(f"\nUsage: {sys.argv[0]} imagefile [-t] [-convert_alpha]")
print(" imagefile image filename (required)")
print(" -t run speed test")
print(" -convert_alpha use convert_alpha() on the image's " "surface\n")
else:
main(
sys.argv[1],
convert_alpha="-convert_alpha" in sys.argv,
run_speed_test="-t" in sys.argv,
)