# -*- coding: utf-8 -*-
'''
一、統計所有 pixel 值量
二、計數 pixel 大於平均數的數量
'''
import cv2
import sys
import numpy as np
from tkinter import filedialog
import tkinter as tk
from PIL import ImageTk, Image
from matplotlib import pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
root = tk.Tk()
radioValue = tk.IntVar()
mediaFrame = tk.Frame(root).pack()
media = tk.Label(mediaFrame)
media.pack()
fig = plt.figure()
plot =fig.add_subplot(111)
canva = FigureCanvasTkAgg(fig,root)
canva.get_tk_widget().pack(side='right')
def cv_imread(filePath):
    cv_img=cv2.imdecode(np.fromfile(filePath,dtype=np.uint8),-1)
    return cv_img
def oas():
    plt.cla()
    global mode
    try:
        sfname = filedialog.askopenfilename(title='選擇',filetypes=[('All Files','*'),("jpeg files","*.jpg"),("png files","*.png"),("gif files","*.gif")])
        gray = cv2.cvtColor(cv2.resize(cv_imread(sfname),(500,250)),cv2.COLOR_BGR2GRAY)
        y = np.zeros((256))
        sun = 0
        count = 0
        arr = np.array(gray)
        (mean,stddv) = cv2.meanStdDev(gray)
        print(mean)
        print(stddv)
        for i in range(0,gray.shape[0]):
            for j in range(0,gray.shape[1]):
                y[gray[i,j]] += 1
                sun += arr[(i,j)]
                if(arr[i,j]>mean):
                    count += 1
        print(sun,count)
        print((count/(sun-count))**0.5)
        aa = ((count/(sun-count))**0.5)
        print(((sun-count)/count)**0.5)
        bb = (((sun-count)/count)**0.5)
        print(stddv*aa)
        print(stddv*bb)
        print(mean-stddv*aa)
        print(mean+stddv*bb)
        a = mean-stddv*aa
        b = mean+stddv*bb
        ri = gray.shape[0]
        rj = gray.shape[1]
        '''
        s = 0
        c = 0
        for i in range(0,ri,4):
            for j in range(0,rj,4):
                a = arr[i,j]
                s += a 
                if(a>mean):
                    c += 1
        print(s,c)
        print((c/(s-c))**0.5)
        print(((s-c)/c)**0.5)
        '''
        for i in range(0,gray.shape[0]):
            for j in range(0,gray.shape[1]):
                if(arr[(i,j)]>mean):
                    arr[(i,j)]=b
                if(arr[(i,j)]<=mean):
                    arr[(i,j)]=a
        btc=Image.fromarray(arr)
        btc.show()
        plt.bar(np.arange(0,256),y,color="gray",align="center")
        cvphoto = Image.fromarray(gray)
        imgtk = ImageTk.PhotoImage(image=cvphoto)
        media.imgtk = imgtk
        media.configure(image=imgtk)
        canva.draw()
    except:
        print('no open file or no choose mode'+sys.exc_info())
b1 = tk.Button(root, text="打開",command = oas).pack()
root.mainloop()