MyImageView.py

# coding: utf-8

import ui, os, sys


class MyImageView(ui.View):
    def __init__(self):
        self.root = os.path.expanduser("~")
        self.rootlen = len(self.root)
        self.path = os.getcwd()
        self.root = self.path[self.rootlen :]
        self.color = "white"
        self.x_off = 0
        self.y_off = 0
        self.scr_height = None
        self.scr_width = None
        self.scr_cor = 2.0
        self.ratio = 1.0
        self.files = []
        self.index = 0
        for entry in sorted(os.listdir(self.path)):
            if os.path.isfile(self.path + "/" + entry):
                if entry.find(".jpg") >= 0:
                    self.files.append(entry)
        self.nr_files = len(self.files)
        print("files = " + str(self.nr_files))
        if self.nr_files > 0:
            self.img = ui.Image.named(self.files[self.index])
            self.img_width, self.img_height = self.img.size
            self.name = self.root + "/" + self.files[self.index]
        else:
            print("Sorry, no images in this directory.")
            sys.exit()

    def draw(self):
        self.scr_height = self.height
        self.scr_width = self.width
        path = ui.Path.rect(0, 0, self.scr_width, self.scr_height)
        ui.set_color(self.color)
        path.fill()
        self.x_off = (self.scr_width - (self.img_width * self.ratio / self.scr_cor)) / 2
        self.y_off = (
            self.scr_height - (self.img_height * self.ratio / self.scr_cor)
        ) / 2
        self.img.draw(
            self.x_off,
            self.y_off,
            self.img_width * self.ratio / self.scr_cor,
            self.img_height * self.ratio / self.scr_cor,
        )

    def touch_began(self, touch):
        if (self.index + 1) < self.nr_files:
            self.index += 1
        else:
            self.index = 0
        self.img = ui.Image.named(self.files[self.index])
        self.img_width, self.img_height = self.img.size
        self.name = self.root + "/" + self.files[self.index]
        self.layout()
        self.set_needs_display()

    def layout(self):
        scr_height_real = self.height * self.scr_cor
        scr_width_real = self.width * self.scr_cor
        y_ratio = scr_height_real / self.img_height
        x_ratio = scr_width_real / self.img_width
        # 1.0 = okay, <1.0 = Image to small, >1.0 = Image to big
        if x_ratio == 1.0 and y_ratio == 1.0:
            self.ratio = 1.0  # perfect size
        elif x_ratio == 1.0 and y_ratio > 1.0:
            self.ratio = 1.0  # perfect width
        elif x_ratio > 1.0 and y_ratio == 1.0:
            self.ratio = 1.0  # perfect height
        elif x_ratio > 1.0 and y_ratio > 1.0:
            self.ratio = 1.0  # show image in original size
        elif x_ratio >= 1.0 and y_ratio < 1.0:
            self.ratio = y_ratio  # shrink height
        elif x_ratio < 1.0 and y_ratio >= 1.0:
            self.ratio = x_ratio  # shrink width
        elif x_ratio < 1.0 and y_ratio < 1.0:
            if x_ratio < y_ratio:  # which side?
                self.ratio = x_ratio
            else:
                self.ratio = y_ratio
        else:
            print("This should never happen. :(")


view = MyImageView()
view.present("fullscreen")