view cagou/core/image.py @ 439:12d188cb1206

core: use of new profileConnected method: bookmarks cache is now retrieved in profileConnected, as getting it in ProfilePlugged could result in an exception is client was not fully connected (if plugins were not all initialised).
author Goffi <goffi@goffi.org>
date Sat, 07 Mar 2020 00:05:49 +0100
parents aa204c813f07
children b5e8e470f7f7
line wrap: on
line source

#!/usr/bin/env python3


# Cagou: desktop/mobile frontend for Salut à Toi XMPP client
# Copyright (C) 2016-2020 Jérôme Poisson (goffi@goffi.org)

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.


from sat.core import log as logging
log = logging.getLogger(__name__)
from kivy.uix import image as kivy_img
from kivy.core.image import Image as CoreImage
from kivy.resources import resource_find
import io
import PIL


class Image(kivy_img.Image):
    """Image widget which accept source without extension"""

    def texture_update(self, *largs):
        if not self.source:
            self.texture = None
        else:
            filename = resource_find(self.source)
            self._loops = 0
            if filename is None:
                return log.error('Image: Error reading file {filename}'.
                                    format(filename=self.source))
            mipmap = self.mipmap
            if self._coreimage is not None:
                self._coreimage.unbind(on_texture=self._on_tex_change)
            try:
                self._coreimage = ci = CoreImage(filename, mipmap=mipmap,
                                                 anim_delay=self.anim_delay,
                                                 keep_data=self.keep_data,
                                                 nocache=self.nocache)
            except Exception:
                # loading failed probably because of unmanaged extention,
                # we try our luck with with PIL
                try:
                    im = PIL.Image.open(filename)
                    ext = im.format.lower()
                    del im
                    # we can't use im.tobytes as it would use the
                    # internal decompressed representation from pillow
                    # and im.save would need processing to handle format
                    data = io.BytesIO(open(filename, "rb").read())
                    cache_filename = "{}.{}".format(filename,ext) # needed for kivy's Image to use cache
                    self._coreimage = ci = CoreImage(data, ext=ext,
                                                     filename=cache_filename, mipmap=mipmap,
                                                     anim_delay=self.anim_delay,
                                                     keep_data=self.keep_data,
                                                     nocache=self.nocache)
                except Exception as e:
                    log.warning("Can't load image: {}".format(e))
                    self._coreimage = ci = None

            if ci:
                ci.bind(on_texture=self._on_tex_change)
                self.texture = ci.texture


class AsyncImage(kivy_img.AsyncImage):
    """AsyncImage which accept file:// schema"""

    def _load_source(self, *args):
        if self.source.startswith('file://'):
            self.source = self.source[7:]
        else:
            super(AsyncImage, self)._load_source(*args)