import sys
from PyQt5 import QtWidgets, QtGui
from PyQt5.QtWidgets import *
from DIP import Ui_DigitalImageProcessing
from GeometricalChange import GeoChange
from LogicalAlgorithm import logicalAlgorithm
from Histogram import Histogram
from Hough import Hough
from EdgeDetection import edgeDetect
from morphology import morphology
from NoiseCanceller import noiseCanceller
from SpatialSmoothing import spatial
from FrequencySmoothing import frequency_filter


class run(QtWidgets.QMainWindow, Ui_DigitalImageProcessing):

    def __init__(self):
        super(run, self).__init__()
        self.setupUi(self)
        # 原图像路径，后面将作为图像处理的参数，表示对哪张图片进行图像处理。
        self.img_path = ""

    # 打开图像函数
    def openImg(self):
        imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "", "All Files(*);")
        self.img_path = imgName
        # 展示选中的图片
        jpg = QtGui.QPixmap(imgName).scaled(self.original_img.width(), self.original_img.height())
        self.original_img.setPixmap(jpg)

    # 展示修改后图片
    def showImg(self, pic_path):
        jpg = QtGui.QPixmap(pic_path).scaled(self.fixed_img.width(), self.fixed_img.height())
        self.fixed_img.setPixmap(jpg)

    # 此处开始处理图像逻辑运算的槽函数, alAnd为图像与操作
    def alAnd(self):
        imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "", "All Files(*);")
        al = logicalAlgorithm(self.img_path, imgName)
        al.alAnd()
        self.showImg("saved Img/And.bmp")

    # 图像或操作槽函数
    def alOr(self):
        imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "", "All Files(*);")
        al = logicalAlgorithm(self.img_path, imgName)
        al.alOr()
        self.showImg("saved Img/Or.bmp")

    # 图像非操作槽函数
    def alNegation(self):
        al = logicalAlgorithm(self.img_path, self.img_path)
        al.alNegation()
        self.showImg("saved Img/Negation.bmp")

    # 图像加法操作槽函数
    def alAdd(self):
        imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "", "All Files(*);")
        al = logicalAlgorithm(self.img_path, imgName)
        al.alAdd()
        self.showImg("saved Img/Add.bmp")

    # 图像减法操作槽函数
    def alSubtract(self):
        imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "", "All Files(*);")
        al = logicalAlgorithm(self.img_path, imgName)
        al.alSubtract()
        self.showImg("saved Img/Subtract.bmp")

    # 图像乘法操作槽函数
    def alMulty(self):
        imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "", "All Files(*);")
        al = logicalAlgorithm(self.img_path, imgName)
        al.alMulty()
        self.showImg("saved Img/Multiply.bmp")

    # 图像除法操作槽函数
    def alDivide(self):
        imgName, imgType = QFileDialog.getOpenFileName(self, "打开图片", "", "All Files(*);")
        al = logicalAlgorithm(self.img_path, imgName)
        al.alDivide()
        self.showImg("saved Img/Divide.bmp")

    # 此处开始处理图像几何变化的槽函数，extend是图像放缩槽函数
    def extend(self):
        size, _ = QInputDialog.getText(self, "放缩倍数", "请输入放缩倍数, 倍数应为非负数。\n由于修改后图片预览是固定大小的，因此无法体现放缩效果\n请前往图片保存地址进行查看。", QLineEdit.Normal, "2")
        try:
            size = float(size)
            geo = GeoChange(self.img_path)
            geo.extend(size)
            self.showImg("saved Img/size.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    # 图像平移槽函数
    def moveImg(self):
        xny, _ = QInputDialog.getText(self, "平移参数", "请输入平移参数\n输入格式为a,b其中a与b以英文逗号隔开。\n", QLineEdit.Normal, "20,20")
        try:
            ipt = xny.split(",")
            a, b = [float(x) for x in ipt]
            geo = GeoChange(self.img_path)
            geo.move(a, b)
            self.showImg("saved Img/move.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误!请检查输入格式与内容。")

    # 图像旋转槽函数
    def rotation(self):
        degree, _ = QInputDialog.getText(self, "旋转角度", "请输入旋转角度，角度应为实数", QLineEdit.Normal, "90")
        try:
            degree = float(degree)
            geo = GeoChange(self.img_path)
            geo.rotation(degree)
            self.showImg("saved Img/rotation.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误!请检查输入格式与内容。")

    # 图像水平翻折槽函数
    def horizonal(self):
        geo = GeoChange(self.img_path)
        geo.horizon_flip()
        self.showImg("saved Img/horizon.bmp")

    # 图像垂直翻折槽函数
    def vertical(self):
        geo = GeoChange(self.img_path)
        geo.vertical_flip()
        self.showImg("saved Img/vertical.bmp")

    # 仿射变换槽函数
    def affine(self):
        geo = GeoChange(self.img_path)
        geo.affine()
        self.showImg("saved Img/affine.bmp")

    # 下面开始对数字图像的直方图进行绘制，histGrey为绘制灰度直方图的槽函数
    def histGrey(self):
        hist = Histogram(self.img_path)
        hist.histGrey()
        self.showImg(self.img_path)

    # 彩色直方图绘制
    def histRGB(self):
        hist = Histogram(self.img_path)
        hist.histRGB()
        self.showImg(self.img_path)

    # 绘制直方图
    def histD(self):
        hist = Histogram(self.img_path)
        hist.hist()
        self.showImg(self.img_path)

    # 修改灰度直方图
    def editHist(self):
        hist = Histogram(self.img_path)
        hist.editHist()
        self.showImg(self.img_path)

    # 以下是使用opencv实现线条变化检测的函数
    def houghLines(self):
        hough = Hough(self.img_path)
        hough.line_change_detection()
        self.showImg("saved Img/result.bmp")

    # 使用Hough P实现线条变化检测
    def houghLinesP(self):
        hough = Hough(self.img_path)
        hough.line_change_detection_P()
        self.showImg('saved Img/result_p.bmp')

    # 图像边缘检测部分
    def basic_of_edge_detection(self):
        edge = edgeDetect(self.img_path)
        edge.graphicsEnhance()
        self.showImg('saved Img/sharp.bmp')

    # Roberts算子
    def roberts(self):
        edge = edgeDetect(self.img_path)
        edge.roberts()
        self.showImg("saved Img/roberts.bmp")

    # Prewitt & sobel
    def sobel(self):
        edge = edgeDetect(self.img_path)
        edge.sobel()
        self.showImg("saved Img/sobel.bmp")

    # laplacian
    def laplacian(self):
        edge = edgeDetect(self.img_path)
        edge.laplacian()
        self.showImg("saved Img/laplacian.bmp")

    # LoG
    def LoG(self):
        edge = edgeDetect(self.img_path)
        edge.LoG()
        self.showImg("saved Img/log.bmp")

    # canny
    def canny(self):
        edge = edgeDetect(self.img_path)
        edge.canny()
        self.showImg("saved Img/canny.bmp")

    # 以下为图像形态学
    # 腐蚀
    def corrode(self):
        struct_size, _ = QInputDialog.getText(self, "结构元大小", "请输入结构元大小，大小应为实数", QLineEdit.Normal, "5")
        try:
            morph = morphology(self.img_path)
            morph.corrode(struct_size)
            self.showImg("saved Img/erosion.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    # 膨胀
    def dilation(self):
        struct_size, _ = QInputDialog.getText(self, "结构元大小", "请输入结构元大小，大小应为实数", QLineEdit.Normal, "5")
        try:
            morph = morphology(self.img_path)
            morph.dilation(struct_size)
            self.showImg("saved Img/dilate.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    # 开运算
    def open(self):
        struct_size, _ = QInputDialog.getText(self, "结构元大小", "请输入结构元大小，大小应为实数", QLineEdit.Normal, "5")
        try:
            morph = morphology(self.img_path)
            morph.open(struct_size)
            self.showImg("saved Img/open.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    # 闭运算
    def close(self):
        struct_size, _ = QInputDialog.getText(self, "结构元大小", "请输入结构元大小，大小应为实数", QLineEdit.Normal, "5")
        try:
            morph = morphology(self.img_path)
            morph.close(struct_size)
            self.showImg("saved Img/close.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    # 接下来的部分是图像噪声处理
    # describe是噪声描述器的槽函数
    def describe(self):
        noise = noiseCanceller(self.img_path)
        noise.noiseDescriber()
        self.showImg("saved Img/out.bmp")

    # mean filter是均值滤波器的槽函数
    def meanFilter(self):
        noise = noiseCanceller(self.img_path)
        noise.meanFilter()
        self.showImg("saved Img/mean filter.bmp")

    # 排序类滤波器的槽函数
    def sortFilter(self):
        noise = noiseCanceller(self.img_path)
        noise.sortFilter()
        self.showImg("saved Img/sort filter.bmp")

    # 选择性滤波器槽函数
    def selectiveFilter(self):
        noise = noiseCanceller(self.img_path)
        noise.selectiveFilter()
        self.showImg("saved Img/selective filter.bmp")

    # 以下为图像空域的平滑
    # 均值
    def avg_blur(self):
        ksize, _ = QInputDialog.getText(self, "滤波核大小", "请输入滤波核的大小", QLineEdit.Normal, "3")
        try:
            blur = spatial(self.img_path)
            blur.average_blur(ksize)
            self.showImg("saved Img/avg_blur.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    def mid_blur(self):
        ksize, _ = QInputDialog.getText(self, "滤波核大小", "请输入滤波核的大小", QLineEdit.Normal, "3")
        try:
            blur = spatial(self.img_path)
            blur.mid_blur(ksize)
            self.showImg("saved Img/mid_blur.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    def low_pass(self):
        fre = frequency_filter(self.img_path)
        fre.Ideal_low_pass_filtering_smooth()
        self.showImg("saved Img/ideal_low_pass.bmp")

    def butterworth(self):
        D0, _ = QInputDialog.getText(self, "截止频率", "请输入截止频率", QLineEdit.Normal, "20")
        n, _ = QInputDialog.getText(self, "阶数", "请输入阶数", QLineEdit.Normal, "2")
        try:
            fre = frequency_filter(self.img_path)
            fre.butterworth(D0, n)
            self.showImg("saved Img/butterworth.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")

    def gauss(self):
        D0, _ = QInputDialog.getText(self, "截止频率", "请输入截止频率", QLineEdit.Normal, "20")
        try:
            fre = frequency_filter(self.img_path)
            fre.gauss(D0)
            self.showImg("saved Img/gauss.bmp")
        except Exception as error:
            QMessageBox.about(self.window(), "提示", f"{error}\n输入有误！请检查输入格式与内容。")


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    now_Status = run()
    now_Status.show()
    sys.exit(app.exec_())