ProjektAI/venv/Lib/site-packages/pygame/examples/scaletest.py
2020-04-20 22:51:16 +02:00

141 lines
4.6 KiB
Python

#!/usr/bin/env python
import sys, time
import pygame
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
pygame.display.init()
# load background image
background = pygame.image.load(imagefile)
if run_speed_test:
if convert_alpha:
# convert_alpha() requires the display mode to be set
pygame.display.set_mode((1, 1))
background = background.convert_alpha()
SpeedTest(background)
return
# start fullscreen mode
screen = pygame.display.set_mode((1024, 768), pygame.FULLSCREEN)
if convert_alpha:
background = background.convert_alpha()
# turn off the mouse pointer
pygame.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 = pygame.transform.smoothscale(background, cursize)
imgpos = image.get_rect(centerx=512, centery=384)
screen.fill((255,255,255))
screen.blit(image, imgpos)
pygame.display.flip()
for event in pygame.event.get():
if event.type == pygame.QUIT or (event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE):
bRunning = False
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_UP: bUp = True
if event.key == pygame.K_DOWN: bDown = True
if event.key == pygame.K_LEFT: bLeft = True
if event.key == pygame.K_RIGHT: bRight = True
if event.type == pygame.KEYUP:
if event.key == pygame.K_UP: bUp = False
if event.key == pygame.K_DOWN: bDown = False
if event.key == pygame.K_LEFT: bLeft = False
if event.key == pygame.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
def SpeedTest(image):
print("\nImage Scaling Speed Test - Image Size %s\n" % str(
image.get_size()))
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 = pygame.transform.smoothscale(image, (shrinkx, shrinky))
duration += (time.time() - start)
del tempimg
print("Average transform.smoothscale shrink time: %.4f ms." % (
duration / 128 * 1000))
duration = 0
for i in range(128):
expandx = (imgsize[0] * (i + 129)) // 128
expandy = (imgsize[1] * (i + 129)) // 128
start = time.time()
tempimg = pygame.transform.smoothscale(image, (expandx, expandy))
duration += (time.time() - start)
del tempimg
print("Average transform.smoothscale expand time: %.4f ms." % (
duration / 128 * 1000))
duration = 0.0
for i in range(128):
shrinkx = (imgsize[0] * i) // 128
shrinky = (imgsize[1] * i) // 128
start = time.time()
tempimg = pygame.transform.scale(image, (shrinkx, shrinky))
duration += (time.time() - start)
del tempimg
print("Average transform.scale shrink time: %.4f ms." % (
duration / 128 * 1000))
duration = 0
for i in range(128):
expandx = (imgsize[0] * (i + 129)) // 128
expandy = (imgsize[1] * (i + 129)) // 128
start = time.time()
tempimg = pygame.transform.scale(image, (expandx, expandy))
duration += (time.time() - start)
del tempimg
print("Average transform.scale expand time: %.4f ms." % (
duration / 128 * 1000))
if __name__ == '__main__':
# check input parameters
if len(sys.argv) < 2:
print("\nUsage: %s imagefile [-t] [-convert_alpha]" % sys.argv[0])
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)