SimpleImageComp/imcomp_app.py at main · Addi90/SimpleImageComp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
from re import S
from sys import orig_argv
from kivy.app import App
from kivy.uix.image import Image
from kivy.uix.label import Label
from kivy.uix.button import Button
from kivy.core.window import Window
from kivy.uix.gridlayout import GridLayout
from kivy.uix.boxlayout import BoxLayout

import threading
import os.path
import imcomp_funs

fpaths = []

class FileDropArea(Label):

    # Add every FileDropArea to a list in main app, so that main app can call
    # all on_filedrop functions of all instances of FileDropArea.

    # In on_filedrop, check Mouse position on a file drop to find out which
    # instance a file was dropped on
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        app = App.get_running_app()
        app.file_drops.append(self.on_filedrop)

        self.orig_text = self.text

    # This function is called when a file is dropped on the FileDropArea
    def on_filedrop(self,widget,file_path):
        if self.parent.collide_point(*Window.mouse_pos):
            app = App.get_running_app()
            self.parent.clear_widgets([i for i in self.parent.children if type(i) is Image])

            fpath = os.path.abspath(file_path)
            self.text = fpath.decode('utf-8')
            fpaths.append(self.text)
            img = Image(source= self.text)
            self.parent.add_widget(img)


class AppLayout(GridLayout):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)


    def on_enterbtn_pressed(self,instance):
        self.background_color=(1,0,0,1)
        app = App.get_running_app()

        mse_thread = threading.Thread(
            target=imcomp_funs.calc_mse,
            args=(self.ids['orig_fda'].text,self.ids['comp_fda'].text,self.set_mse_lb)
            )
        psnr_thread = threading.Thread(
            target=imcomp_funs.calc_psnr,
            args=(self.ids['orig_fda'].text,self.ids['comp_fda'].text,self.set_psnr_lb)
            )
        ssim_thread = threading.Thread(
            target=imcomp_funs.calc_ssim,
            args=(self.ids['orig_fda'].text,self.ids['comp_fda'].text,self.set_ssim_lb)
            )

        mse_thread.start()
        psnr_thread.start()
        ssim_thread.start()

    def set_mse_lb(self, mse_txt):
        print(mse_txt)
        if(mse_txt != -1):
            self.ids['mse_lb'].text = 'Mean Squared Error (MSE): ' + str(mse_txt)
        else:
            self.ids['mse_lb'].text = 'Mean Squared Error (MSE): ERROR'

    def set_psnr_lb(self, psnr_txt):
        print(psnr_txt)
        if(psnr_txt != -1):
            self.ids['psnr_lb'].text = 'Peak Signal-Noise Ratio (PSNR): ' + str(psnr_txt)
        else:
            self.ids['psnr_lb'].text = 'Peak Signal-Noise Ratio (PSNR): ERROR'

    def set_ssim_lb(self, ssim_txt):
        print(ssim_txt)
        if(ssim_txt != -1):
            self.ids['ssim_lb'].text = 'Structural Similarity Index (SSIM): ' + str(ssim_txt)
        else:
            self.ids['ssim_lb'].text = 'Structural Similarity Index (SSIM): ERROR'


class ImcompApp(App):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.file_drops =[]

    def build(self):

        self.window = AppLayout()
        Window.bind(on_dropfile=self.handle_filedrop)
        self.title = "Simple Image Compare"
        return self.window

    def handle_filedrop(self,*args):
        for f in self.file_drops:
            f(*args)

    def get_img_paths(self):
        return self.orig_path,self.comp_path


if __name__ == "__main__":
    ImcompApp().run()