import logging
from typing import Any, Tuple
try:
from PIL import Image
except ImportError as ex:
raise ImportError("Texture loader 'PillowLoader' requires Pillow: {}".format(ex))
from moderngl_window.loaders.base import BaseLoader
from moderngl_window.exceptions import ImproperlyConfigured
logger = logging.getLogger(__name__)
class PillowLoader(BaseLoader):
"""Base loader using PIL/Pillow"""
kind = '__unknown__'
def __init__(self, meta):
super().__init__(meta)
self.image = None
def load(self) -> Any:
raise NotImplementedError()
def _open_image(self):
if self.meta.image:
self.image = self.meta.image
else:
self.meta.resolved_path = self.find_texture(self.meta.path)
logger.info("loading %s", self.meta.resolved_path)
if not self.meta.resolved_path:
raise ImproperlyConfigured("Cannot find texture: {}".format(self.meta.path))
self.image = Image.open(self.meta.resolved_path)
if self.meta.flip:
self.image = self.image.transpose(Image.FLIP_TOP_BOTTOM)
def _close_image(self):
self.image.close()
def image_data(image: Image) -> Tuple[int, bytes]:
"""Get components and bytes for an image.
The number of components is assumed by image
size and the byte length of the raw data.
Returns:
Tuple[int, bytes]: Number of components, byte data
"""
# NOTE: We might want to check the actual image.mode
# and convert to an acceptable format.
# At the moment we load the data as is.
data = image.tobytes()
components = len(data) // (image.size[0] * image.size[1])
logger.debug("image_data size=[%s, %s] components=%s bytes=%s", image.size[0], image.size[1], components, len(data))
return components, data