melage 0.0.65__py3-none-any.whl

melage/widgets/HistImage.py

@@ -0,0 +1,153 @@
+__AUTHOR__ = 'Bahram Jafrasteh'
+
+import sys
+from PyQt5.QtWidgets import QDialog, QApplication, QPushButton, QVBoxLayout
+from PyQt5.QtCore import Qt
+
+from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg, FigureCanvasQT
+from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
+import matplotlib.pyplot as plt
+from PyQt5 import QtCore, QtGui, QtWidgets
+from PyQt5.QtCore import Qt
+from PyQt5.QtCore import pyqtSignal
+import random
+import matplotlib
+matplotlib.use('Agg')
+
+from matplotlib.figure import Figure
+
+class FigureCanvas(FigureCanvasQTAgg):
+    """
+    Class implemented to show figures
+    """
+    def __init__(self, parent=None, width=5, height=4, dpi=1000):
+        super(FigureCanvas, self).__init__(parent)
+        fig = Figure(figsize=(width, height), dpi=dpi)
+        self.axes = fig.add_subplot(111)
+        self._canvas = FigureCanvasQT(parent)
+
+    def mousePressEvent(self, a0) -> None:
+        x, y = a0.x(), a0.y()
+        inaxes = self._canvas.inaxes((x, y))
+        if inaxes is not None:
+            try:
+                trans = inaxes.transData.inverted()
+                xdata, ydata = trans.transform((x, y))
+            except ValueError:
+                pass
+            else:
+                print(xdata, ydata)
+        super(FigureCanvas, self).mousePressEvent(a0)
+
+
+class HistImage(QDialog):
+    """
+    Histogram images
+    """
+    closeSig = pyqtSignal()
+    def __init__(self, parent=None):
+        super(HistImage, self).__init__(parent)
+
+        Form = self.window()
+        Form.setObjectName("N4")
+        Form.resize(600, 600)
+        self.grid_main = QtWidgets.QGridLayout(self)
+        self.grid_main.setContentsMargins(0,0,0,0)
+        self.grid_main.setObjectName("gridLayout")
+        self.tabWidget = QtWidgets.QTabWidget()
+        self.tabWidget.setGeometry(QtCore.QRect(50, 50, 500, 500))
+        self.tabWidget.setObjectName("tabWidget")
+        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding, QtWidgets.QSizePolicy.MinimumExpanding)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        sizePolicy.setHeightForWidth(self.tabWidget.sizePolicy().hasHeightForWidth())
+        self.tabWidget.setSizePolicy(sizePolicy)
+        self.tabWidget.setMinimumSize(QtCore.QSize(100, 100))
+        #self.tabWidget.setObjectName("tabWidget")
+        self.widget = QtWidgets.QWidget(self)
+        self.widget.setGeometry(QtCore.QRect(20, 20, 400, 400))
+        self.widget.setObjectName("widget")
+        self.gridLayout = QtWidgets.QGridLayout(self.widget)
+        self.gridLayout.setContentsMargins(0, 0, 0, 0)
+        self.gridLayout.setObjectName("gridLayout")
+
+        # a figure instance to plot on
+        self.figure = plt.figure()
+
+
+        self.canvas = FigureCanvas(self.figure)
+
+        self.canvas.setParent(self)
+
+        self.gridLayout.addWidget(self.canvas, 0, 0, 2, 2)
+
+        self.figure2 = plt.figure()
+
+        self.widget2 = QtWidgets.QWidget()
+        self.widget2.setGeometry(QtCore.QRect(20, 20, 400, 400))
+        self.widget2.setObjectName("widget")
+        
+        self.gridLayout2 = QtWidgets.QGridLayout(self.widget2)
+        self.canvas2 = FigureCanvas(self.figure2)
+
+        self.gridLayout2.addWidget(self.canvas2, 0, 0, 2, 2)
+        self.gridLayout2.setContentsMargins(0, 0, 0, 0)
+        self.gridLayout2.setObjectName("gridLayout")
+        self.tabWidget.addTab(self.widget2,"")
+        self.tabWidget.addTab(self.widget, "")
+        _translate = QtCore.QCoreApplication.translate
+        self.tabWidget.setTabText(self.tabWidget.indexOf(self.widget), _translate("Dialog", "View 2"))
+        self.tabWidget.setTabText(self.tabWidget.indexOf(self.widget2), _translate("Dialog", "View 1"))
+
+        self.grid_main.addWidget(self.tabWidget)
+
+        Form.setWindowTitle(_translate("Form", "Image Histogram"))
+
+    def closeEvent(self, a0: QtGui.QCloseEvent) -> None:
+        self.closeSig.emit()
+        super(HistImage, self).closeEvent(a0)
+
+    def UpdateName(self, a, b):
+        _translate = QtCore.QCoreApplication.translate
+        if a is not None:
+            self.tabWidget.setTabText(self.tabWidget.indexOf(self.widget), _translate("Dialog", a))
+        if b is not None:
+            self.tabWidget.setTabText(self.tabWidget.indexOf(self.widget2), _translate("Dialog", b))
+    def plot(self, data, id):
+        if data is None:
+            return #data = [random.random() for i in range(10)]
+        if id == 1:
+            figure, canvas = self.figure, self.canvas
+        elif id == 2:
+            figure, canvas = self.figure2, self.canvas2
+        else:
+            return
+
+
+        figure.clear()
+
+        ax = figure.add_subplot(111)
+
+        a = data.mean(axis=2).flatten()
+        a = a[a>0]
+
+        ax.hist(a, 25)
+        ax.grid('on')
+        ax.tick_params(axis='both', which='major', labelsize=10)
+        ax.tick_params(axis='both', which='minor', labelsize=10)
+        ax.set_title('Image histogram', fontsize=6 * 1.7, weight='bold')
+        ax.set_xlabel('Range', fontsize=6 * 1.5, weight='bold')
+
+        ax.set_ylabel('Frequency', fontsize=6 * 1.5, weight='bold')
+        ax.set_xlim([0,a.max()+1])
+        figure.tight_layout(pad=2)
+
+        canvas.draw()
+
+if __name__ == '__main__':
+    app = QApplication(sys.argv)
+
+    main = HistImage()
+    main.show()
+
+    sys.exit(app.exec_())

melage/widgets/ImageThresholding.py

@@ -0,0 +1,222 @@
+__AUTHOR__ = 'Bahram Jafrasteh'
+
+from PyQt5.QtWidgets import QDialog, QApplication, QPushButton, QVBoxLayout
+import matplotlib.pyplot as plt
+from PyQt5 import QtCore, QtGui, QtWidgets
+from PyQt5.QtCore import Qt
+from PyQt5.QtCore import pyqtSignal
+import random
+import matplotlib
+from melage.widgets.HistImage import FigureCanvas
+matplotlib.use('Agg')
+
+
+class ThresholdingImage(QDialog):
+    """
+    Image thresholding
+    """
+    closeSig = pyqtSignal()
+    applySig = pyqtSignal()
+    repltSig = pyqtSignal()
+    histeqSig = pyqtSignal(int)
+
+    def __init__(self, parent=None):
+        super(ThresholdingImage, self).__init__(parent)
+
+        Form = self.window()
+        Form.setObjectName("N4")
+        Form.resize(600, 600)
+        #self.tabWidget = QtWidgets.QTabWidget(Form)
+        #self.tabWidget.setGeometry(QtCore.QRect(50, 50, 500, 500))
+        #self.tabWidget.setObjectName("tabWidget")
+        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.MinimumExpanding,
+                                           QtWidgets.QSizePolicy.MinimumExpanding)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        #sizePolicy.setHeightForWidth(self.tabWidget.sizePolicy().hasHeightForWidth())
+
+        # self.tabWidget.setObjectName("tabWidget")
+        self.widget = QtWidgets.QWidget(self)
+        #self.widget.setGeometry(QtCore.QRect(20, 20, 400, 400))
+        self.widget.setObjectName("widget")
+        self.gridLayout_main = QtWidgets.QVBoxLayout(self)
+        self.gridLayout = QtWidgets.QGridLayout()
+        self.gridLayout.setContentsMargins(0, 0, 0, 0)
+        self.gridLayout.setObjectName("gridLayout")
+
+        # a figure instance to plot on
+        self.figure = plt.figure()
+
+        # this is the Canvas Widget that displays the `figure`
+        # it takes the `figure` instance as a parameter to __init__
+        self.canvas = FigureCanvas(self.figure)
+        # self.toolbar = NavigationToolbar(self.canvas, self)
+        self.canvas.setParent(self)
+
+        self.gridLayout.addWidget(self.canvas, 5, 5, 2, 2)
+        # this is the Navigation widget
+        # it takes the Canvas widget and a parent
+        self.gridLayout2 = QtWidgets.QGridLayout()
+
+        # Just some button connected to `plot` method
+        self.button = QPushButton('Plot')
+        self.button.clicked.connect(self.applySig)
+        # self.gridLayout.addWidget(self.button, 7, 0, 1, 1)
+        self._number_class = QtWidgets.QComboBox(self.widget)
+
+        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Minimum)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        self._number_class.setMaximumSize(50,50)
+        sizePolicy.setHeightForWidth(self._number_class.sizePolicy().hasWidthForHeight())
+        self._number_class.setSizePolicy(sizePolicy)
+        self._number_class.setObjectName("+")
+        self.gridLayout2.addWidget(self._number_class, 0, 2, 1, 2)
+        self._label_numberClasses = QtWidgets.QLabel(self.widget)
+        self._label_numberClasses.setObjectName("_label_numberClasses")
+
+        self._histeq = QtWidgets.QCheckBox(self.widget)
+        self._histeq.setObjectName("check box")
+        self._histeq.setChecked(False)
+        self._histeq.clicked.connect(self.histEQ)
+
+        self.gridLayout2.addWidget(self.button, 0, 4, 1, 1)
+        self.gridLayout2.addWidget(self._histeq, 0,3, 1, 1)
+        self.gridLayout2.addWidget(self._label_numberClasses, 0,1, 1, 1)
+
+        # self.toolbar2 = NavigationToolbar(self.canvas2, self)
+        # self.canvas2.setParent(self)
+        for i in range(9):
+            self._number_class.setItemData(i, Qt.AlignCenter, Qt.TextAlignmentRole)
+            self._number_class.addItem(" {} ".format(i+1))
+        _translate = QtCore.QCoreApplication.translate
+        self._label_numberClasses.setText(_translate("Form", "No. classes"))
+        self._histeq.setText(_translate("Form", "Hist EQ."))
+        self.button.setText(_translate("Form", "Apply"))
+        self.comboBox_image = QtWidgets.QComboBox(self.widget)
+        self.comboBox_image.setObjectName("comboBox_image")
+        self.comboBox_image.addItem("")
+        self.comboBox_image.addItem("")
+        self.gridLayout2.addWidget(self.comboBox_image,0, 0, 1, 1)
+        Form.setWindowTitle(_translate("Form", "Image Thresholding"))
+        self.comboBox_image.setItemText(0, _translate("Form", "        View 1        "))
+        self.comboBox_image.setItemText(1, _translate("Form", "      View 2      "))
+        #self._number_class.currentIndexChanged.connect(self.plot)
+
+        self.gridLayout_main.addLayout(self.gridLayout, 0)
+        self.gridLayout_main.addLayout(self.gridLayout2, 1)
+
+        self.comboBox_image.currentIndexChanged.connect(self.repltSig)
+        self.button.clicked.connect(self.applySig)
+        self._number_class.currentIndexChanged.connect(self.repltSig)
+
+        #self.plot([])
+
+    def histEQ(self):
+        if not hasattr(self, '_a'):
+            return
+        if self._histeq.isChecked():
+            a = self._a
+            from melage.utils.utils import histogram_equalization
+            self.im_rec = histogram_equalization(a)
+            self.plot(self.im_rec)
+            self.histeqSig.emit(True)
+        else:
+            self.histeqSig.emit(0)
+
+
+
+    def new_thresholds(self, a):
+
+        self._a = a
+        numc = self._number_class.currentIndex()+2
+        from utils.utils import Threshold_MultiOtsu
+        thresholds = Threshold_MultiOtsu(a, numc)
+        return thresholds
+
+
+    def closeEvent(self, a0: QtGui.QCloseEvent) -> None:
+        self.closeSig.emit()
+        if hasattr(self, '_a'):
+            delattr(self,'_a')
+        self._number_class.setCurrentIndex(0)
+        self._histeq.setChecked(False)
+        super(ThresholdingImage, self).closeEvent(a0)
+
+
+
+    def emptyPlot(self):
+
+
+        figure, canvas = self.figure, self.canvas
+
+        ''' plot some random stuff '''
+        # random data
+
+        # instead of ax.hold(False)
+        figure.clear()
+        #import numpy as np
+        #data = np.random.rand(40,40,40)*255
+        # create an axis
+        ax = figure.add_subplot(111)
+
+        # ax.axis('off')
+        #ax.grid('on')
+        ax.tick_params(axis='both', which='major', labelsize=10)
+        ax.tick_params(axis='both', which='minor', labelsize=10)
+        ax.set_title('Image histogram', fontsize=6 * 1.7, weight='bold')
+        ax.set_xlabel('Range', fontsize=6 * 1.5, weight='bold')
+
+        ax.set_ylabel('Frequency', fontsize=6 * 1.5, weight='bold')
+        #ax.set_xlim([0, a.max() + 1])
+        figure.tight_layout(pad=2)
+
+        # ax.legend(loc="upper right", prop={'size': 8 * 2})
+        # refresh canvas
+        canvas.draw()
+
+
+
+    def plot(self, data):
+        if data is None:
+            return  # data = [random.random() for i in range(10)]
+
+        figure, canvas = self.figure, self.canvas
+
+
+        figure.clear()
+
+        # create an axis
+        ax = figure.add_subplot(111)
+
+        a = data.flatten()
+        a = a[a > 0]
+
+        ax.hist(a, 25)
+        self._currentThresholds = self.new_thresholds(data)
+        yl = ax.get_ylim()[1]
+        for el in self._currentThresholds:
+            ax.plot([el,el], [0, yl], )
+        # ax.axis('off')
+        #ax.grid('on')
+        ax.tick_params(axis='both', which='major', labelsize=10)
+        ax.tick_params(axis='both', which='minor', labelsize=10)
+        ax.set_title('Image histogram', fontsize=6 * 1.7, weight='bold')
+        ax.set_xlabel('Range', fontsize=6 * 1.5, weight='bold')
+
+        ax.set_ylabel('Frequency', fontsize=6 * 1.5, weight='bold')
+        #ax.set_xlim([0, a.max() + 1])
+        figure.tight_layout(pad=2)
+
+
+        canvas.draw()
+
+
+def run():
+    import sys
+    app = QtWidgets.QApplication(sys.argv)
+    window = ThresholdingImage()
+    window.show()
+    sys.exit(app.exec_())
+if __name__ == '__main__':
+    run()

melage/widgets/MaskOperations.py

@@ -0,0 +1,147 @@
+__AUTHOR__ = 'Bahram Jafrasteh'
+
+
+from PyQt5 import QtCore, QtGui, QtWidgets
+from PyQt5.QtCore import Qt
+from PyQt5.QtCore import pyqtSignal
+
+
+
+class MaskOperations(QtWidgets.QDialog):
+    """
+    Image MASKING OPERATION
+    """
+    buttonpressed = pyqtSignal(object)
+    closeSig = pyqtSignal()
+
+    def __init__(self, parent=None):
+        QtWidgets.QDialog.__init__(self, parent)
+        self.setWindowTitle("Child Window!")
+        self.inputimage = None
+
+        self.MessageBox = QtWidgets.QMessageBox(self)
+        self.setupUi()
+
+    def setupUi(self):
+        Form = self.window()
+        Form.setObjectName("N4")
+        Form.resize(759, 140)
+        self.widget = QtWidgets.QWidget(Form)
+        self.widget.setGeometry(QtCore.QRect(43, 20, 644, 89))
+        self.widget.setObjectName("widget")
+        self.gridLayout = QtWidgets.QGridLayout(self.widget)
+        self.gridLayout.setContentsMargins(0, 0, 0, 0)
+        self.gridLayout.setObjectName("gridLayout")
+
+        self.splitter = QtWidgets.QSplitter(self.widget)
+        self.splitter.setOrientation(QtCore.Qt.Horizontal)
+        self.splitter.setObjectName("splitter")
+
+        #self.gridLayout.addWidget(self.splitter, 0, 1, 1, 1)
+
+        self.mask_operation = QtWidgets.QComboBox(self.widget)
+
+        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Minimum)
+        sizePolicy.setHorizontalStretch(0)
+        sizePolicy.setVerticalStretch(0)
+        self.mask_operation.setMaximumSize(50,50)
+        sizePolicy.setHeightForWidth(self.mask_operation.sizePolicy().hasWidthForHeight())
+        #self.mask_operation.setSizePolicy(sizePolicy)
+        self.mask_operation.setObjectName("+")
+        self.gridLayout.addWidget(self.mask_operation, 0, 2, 1, 1)
+        self._combobox_colors = QtWidgets.QComboBox(self.widget)
+        self._combobox_colors.setObjectName("_combobox_colors")
+
+        self._combobox_colors2 = QtWidgets.QComboBox(self.widget)
+        self._combobox_colors2.setObjectName("_combobox_colors")
+        self.gridLayout.addWidget(self._combobox_colors2, 0, 0, 1, 2)
+        self.gridLayout.addWidget(self._combobox_colors, 0,5, 1, 2)
+
+
+
+        #self.pushButton_2 = QtWidgets.QPushButton(self.widget)
+        #self.pushButton_2.setObjectName("pushButton")
+        #self.gridLayout.addWidget(self.pushButton_2, 0, 6, 1, 1)
+        self.pushButton = QtWidgets.QPushButton(self.widget)
+        self.pushButton.setObjectName("pushButton_2")
+        self.gridLayout.addWidget(self.pushButton, 2, 7, 1, 1)
+
+        #self.gridLayout.addWidget(self.progressBar, 1, 0, 1, 5)
+        #spacerItem = QtWidgets.QSpacerItem(40, 20, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
+        #self.gridLayout.addItem(spacerItem, 0, 1, 1, 4)
+        self.comboBox_image = QtWidgets.QComboBox(self.widget)
+        self.comboBox_image.setObjectName("comboBox_image")
+        self.comboBox_image.addItem("")
+        self.comboBox_image.addItem("")
+        self.gridLayout.addWidget(self.comboBox_image, 1, 0, 1, 1)
+
+
+        self.retranslateUi(Form)
+        QtCore.QMetaObject.connectSlotsByName(Form)
+
+        for i, operation in enumerate(['+', '-']):
+            self.mask_operation.setItemData(i, Qt.AlignCenter, Qt.TextAlignmentRole)
+            self.mask_operation.addItem(" {} ".format(operation))
+
+        self.pushButton.clicked.connect(self.accepted_emit)
+
+    def closeEvent(self, a0: QtGui.QCloseEvent) -> None:
+        self.closeSig.emit()
+        super(MaskOperations, self).closeEvent(a0)
+
+    def accept_emit(self):
+        self.mask_operation.text()
+        num = self.mask_operation.text()
+        if num not in ['+', '-', '']:
+            self.message()
+            return False
+        return True
+
+    def setComboBoxColors(self, color_name):
+        self._combobox_colors.clear()
+        self._combobox_colors2.clear()
+        for colr in color_name:
+            #self._combobox_colors.setItemData(i, Qt.AlignCenter, Qt.TextAlignmentRole)
+            self._combobox_colors.addItem("   {}   ".format(colr))
+            self._combobox_colors2.addItem("   {}   ".format(colr))
+
+    def retranslateUi(self, Form):
+        _translate = QtCore.QCoreApplication.translate
+        Form.setWindowTitle(_translate("Form", "Mask Operations"))
+
+        self.comboBox_image.setItemText(0, _translate("Form", "        View 1        "))
+        self.comboBox_image.setItemText(1, _translate("Form", "      View 2      "))
+        #self.label_timer.setText(_translate("Form", "00:00"))
+        self.pushButton.setText(_translate("Form", "Apply"))
+        self.pushButton.setVisible(True)
+
+    def message(self):
+
+        self.MessageBox.setText('Please fill with +,-')
+        self.mask_operation.setText('+')
+        self.MessageBox.setWindowTitle('Warning')
+        self.MessageBox.show()
+
+    def accepted_emit(self):
+
+        #self._selectedColor = int(float(self._combobox_colors.currentText()))
+
+        index_image = self.comboBox_image.currentIndex()
+
+        index_color = self._combobox_colors.currentIndex()
+        index_color2 = self._combobox_colors2.currentIndex()
+        if index_color2==index_color:
+            return
+        operation = self.mask_operation.currentText().strip(' ')
+        self.buttonpressed.emit([index_image, index_color2, index_color, operation])
+
+
+
+def run():
+    import sys
+    app = QtWidgets.QApplication(sys.argv)
+    window = MaskOperations()
+    window.show()
+    sys.exit(app.exec_())
+if __name__ == '__main__':
+    run()