Source code for pystreaming.video.testimages

TEST_S = 0
TEST_M = 1
TEST_L = 2
IMAG_S = 3
IMAG_M = 4
IMAG_L = 5

_lookup = [
    "640x480_c.png",
    "1280x720_c.png",
    "1920x1080_c.png",
    "640x480_i.png",
    "1280x720_i.png",
    "1920x1080_i.png",
]


[docs]def loadimage(enum): """Load a test image or a test card. Args: enum (int): One of pystreaming.TEST_S pystreaming.TEST_M pystreaming.TEST_L pystreaming.IMAG_S pystreaming.IMAG_M pystreaming.IMAG_L Raises: IndexError: Raised when received enum is not defined. Returns: PIL.Image: Image requested. """ from PIL import Image import os try: truepath = os.path.join( os.path.dirname(os.path.realpath(__file__)), _lookup[enum] ) return Image.open(truepath) except IndexError: raise IndexError(f"Unrecognized image option: {enum}")