bh_tomo.py

# -*- coding: utf-8 -*-
"""
Copyright 2017 Bernard Giroux, Elie Dumas-Lefebvre, Jerome Simon
email: Bernard.Giroux@ete.inrs.ca

This file is part of BhTomoPy.

BhTomoPy is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
"""

import os
import sys

from PyQt5 import QtGui, QtCore, QtWidgets

from database_ui import DatabaseUI
from manual_tt_ui import ManualttUI
from covar_ui import CovarUI
from inversion_ui import InversionUI
from interp_ui import InterpretationUI
from semi_auto_tt_ui import SemiAutottUI
from manual_amp_ui import ManualAmpUI


class BhTomoPy(QtWidgets.QWidget):

    def __init__(self, parent=None):
        super(BhTomoPy, self).__init__(parent)
        self.setWindowTitle("BhTomoPy")

        self.dbname = ''
        self.database = DatabaseUI()
        self.manual_tt = ManualttUI()
        self.semi_tt = SemiAutottUI()
        self.covar = CovarUI()
        self.inv = InversionUI()
        self.interp = InterpretationUI()
        self.manual_amp = ManualAmpUI()
        self.init_UI()

        self.setSizePolicy(QtWidgets.QSizePolicy.Minimum,
                           QtWidgets.QSizePolicy.Minimum)

    def choose_db(self):
        filename = QtWidgets.QFileDialog.getOpenFileName(self, 'Choose Database')[0]
        if filename:
            self.dbname = filename
            self.current_db.setText(os.path.basename(filename))

    def show(self):
        super(BhTomoPy, self).show()

        # Gets initial geometry of the widget
        qr = self.frameGeometry()

        # Shows it at the center of the screen
        cp = QtWidgets.QDesktopWidget().availableGeometry().center()

        # Moves the window's center at the center of the screen
        qr.moveCenter(cp)

        # Then moves it at the top left
        translation = qr.topLeft()

        self.move(translation)

        self.setFixedSize(self.size())
        self.__h1 = self.size().height()
        self.__h2 = self.tt_tool.size().height() + self.tl_tool.size().height()

    def init_UI(self):

        # ------- Widgets ------- #
        # --- Actions --- #
        ChooseDbAction = QtWidgets.QAction('Choose Database', self)
        ChooseDbAction.setShortcut('Ctrl+O')
        ChooseDbAction.triggered.connect(self.choose_db)

        ConvertDbAction = QtWidgets.QAction('Convert Database', self)
        ConvertDbAction.setShortcut('Ctrl+C')

        # --- Menubar --- #
        self.menu = QtWidgets.QMenuBar()
        filemenu = self.menu.addMenu('&File')
        editmenu = self.menu.addMenu('&Edit')
        filemenu.addAction(ChooseDbAction)
        editmenu.addAction(ConvertDbAction)

        # --- Buttons --- #
        btn_Database                            = QtWidgets.QPushButton("Database")
        btn_Automatic_Traveltime_Picking        = QtWidgets.QPushButton("Automatic (AIC-CWT)")
        btn_Semi_Automatic_Traveltime_Picking   = QtWidgets.QPushButton("Semi-Automatic (x-corr)")
        btn_Manual_Traveltime_Picking           = QtWidgets.QPushButton("Manual")
        btn_Manual_Amplitude_Picking            = QtWidgets.QPushButton("Manual Amplitude Picking")
        btn_Cov_Mod                             = QtWidgets.QPushButton("Covariance Model")
        btn_Inversion                           = QtWidgets.QPushButton("Inversion")
        btn_Interpretation                      = QtWidgets.QPushButton("Interpretation (GPR)")
        btn_Time_Lapse_Inversion                = QtWidgets.QPushButton("Inversion")
        btn_Time_Lapse_Visualisation            = QtWidgets.QPushButton("Visualisation")
        btn_Nano_Fluid                          = QtWidgets.QPushButton("Magnetic Nano Fluid Saturation")

        # - Buttons Disposition - #
        btn_Automatic_Traveltime_Picking.setDisabled(True)
        btn_Time_Lapse_Inversion.setDisabled(True)
        btn_Time_Lapse_Visualisation.setDisabled(True)
        # btn_Nano_Fluid.setDisabled(True)

        # - Buttons Actions - #
        btn_Database.clicked.connect(lambda: self.database.show(self.dbname))
        btn_Manual_Traveltime_Picking.clicked.connect(lambda: self.manual_tt.show(self.dbname))
        btn_Semi_Automatic_Traveltime_Picking.clicked.connect(lambda: self.semi_tt.show(self.dbname))
        btn_Cov_Mod.clicked.connect(lambda: self.covar.show(self.dbname))
        btn_Inversion.clicked.connect(lambda: self.inv.show(self.dbname))
        btn_Interpretation.clicked.connect(lambda: self.interp.show(self.dbname))
        btn_Manual_Amplitude_Picking.clicked.connect(lambda: self.manual_amp.show(self.dbname))

        # --- Image --- #
        pic = QtGui.QPixmap(os.getcwd() + "/BhTomoPy.png")
        image_label = QtWidgets.QLabel()
        image_label.setPixmap(pic)
        image_label.setAlignment(QtCore.Qt.AlignCenter)

#         # --- Title --- #
#         Title = QtWidgets.QLabel(
#             'BH TOMO \n Borehole Radar/Seismic Data Processing Center')
#         Title.setAlignment(QtCore.Qt.AlignHCenter)
#         Title.setContentsMargins(10, 10, 10, 30)
#         Title.setStyleSheet('color: Darkcyan')
#         serifFont = QtGui.QFont("Times", 10, QtGui.QFont.Bold)
#         Title.setFont(serifFont)

        # --- Edit --- #
        # Edit to hold the chosen database's name
        self.current_db = QtWidgets.QLineEdit()

        # - Edit Disposition - #
        self.current_db.setReadOnly(True)
        self.current_db.setAlignment(QtCore.Qt.AlignHCenter)

        # --- Image SubWidget --- #
        sub_image_widget = QtWidgets.QWidget()
        sub_image_grid = QtWidgets.QGridLayout()
        sub_image_grid.addWidget(image_label, 0, 0)
        sub_image_grid.setContentsMargins(20, 0, 20, 0)
        sub_image_widget.setLayout(sub_image_grid)

        # --- Traveltime ToolBox --- #
        travel_time_tool = QtWidgets.QWidget()
        travel_time_grid = QtWidgets.QGridLayout()
        travel_time_grid.addWidget(btn_Manual_Traveltime_Picking, 0, 0)
        travel_time_grid.addWidget(btn_Semi_Automatic_Traveltime_Picking, 1, 0)
        travel_time_grid.addWidget(btn_Automatic_Traveltime_Picking, 2, 0)
        travel_time_tool.setLayout(travel_time_grid)

        # --- Time Lapse Tool --- #
        time_lapse_tool = QtWidgets.QWidget()
        time_lapse_grid = QtWidgets.QGridLayout()
        time_lapse_grid.addWidget(btn_Time_Lapse_Inversion, 0, 0)
        time_lapse_grid.addWidget(btn_Time_Lapse_Visualisation, 1, 0)
        time_lapse_tool.setLayout(time_lapse_grid)

        # --- Traveltime ToolBox --- #
        tt_tool = MyQToolBox()
        tt_tool.setIcons(QtGui.QIcon('Icons/triangle_right.png'),
                         QtGui.QIcon('Icons/triangle_down.png'))
        tt_tool.addItem(travel_time_tool, 'Travel Time Picking')
        tt_tool.sizeChanged.connect(self.fit_height)
        self.tt_tool = tt_tool

        # --- Time Lapse ToolBox --- #
        tl_tool = MyQToolBox()
        tl_tool.setIcons(QtGui.QIcon('Icons/triangle_right.png'),
                         QtGui.QIcon('Icons/triangle_down.png'))
        tl_tool.addItem(time_lapse_tool, 'Time Lapse')
        tl_tool.sizeChanged.connect(self.fit_height)
        self.tl_tool = tl_tool

        # --- Connecting mutual closing --- #
        tl_tool.toolboxOpened.connect(tt_tool.closeAll)
        tt_tool.toolboxOpened.connect(tl_tool.closeAll)

        # --- Buttons SubWidget --- #
        Sub_button_widget = QtWidgets.QGroupBox()
        sub_button_grid = QtWidgets.QGridLayout()
        sub_button_grid.addWidget(btn_Database, 0, 0)
        sub_button_grid.addWidget(tt_tool, 1, 0)
        sub_button_grid.addWidget(btn_Manual_Amplitude_Picking, 2, 0)
        sub_button_grid.addWidget(btn_Cov_Mod, 3, 0)
        sub_button_grid.addWidget(btn_Inversion, 4, 0)
        sub_button_grid.addWidget(tl_tool, 5, 0)
        sub_button_grid.addWidget(btn_Interpretation, 6, 0)
        sub_button_grid.addWidget(btn_Nano_Fluid, 7, 0)
        sub_button_grid.setRowStretch(8, 100)
        Sub_button_widget.setLayout(sub_button_grid)

        # --- Main Widget--- #
        master_grid = QtWidgets.QGridLayout()
        master_grid.addWidget(self.menu, 0, 0, 1, 4)
        master_grid.addWidget(sub_image_widget, 1, 0, 1, 4)
        master_grid.addWidget(self.current_db, 2, 1, 1, 2)
        master_grid.addWidget(Sub_button_widget, 3, 0, 1, 4)
        sub_button_grid.setRowStretch(0, 0)
        sub_button_grid.setRowStretch(1, 0)
        master_grid.setContentsMargins(0, 0, 0, 0)
        master_grid.setVerticalSpacing(3)

        self.setLayout(master_grid)

    def fit_height(self, x):
        # shrink or expand height to fit the toolbox
        h = self.__h1 + x - self.__h2
        self.setFixedHeight(h)


class MyQToolBox(QtWidgets.QWidget):

    """
    A custom widget that mimicks the behavior of the "Tools" sidepanel in
    Adobe Acrobat. It is derived from a QToolBox with the following variants:

    1. Only one tool can be displayed at a time.
    2. Unlike the stock QToolBox widget, it is possible to hide all the tools.
    3. It is also possible to hide the current displayed tool by clicking on
       its header.
    4. Closed and Expanded arrows can be set from custom icons.
    5. The tools that are hidden are marked by a right-arrow icon, while the
       tool that is currently displayed is marked with a down-arrow icon.
    """
# =============================================================================

    sizeChanged = QtCore.pyqtSignal(float)
    toolboxOpened = QtCore.pyqtSignal()

    def __init__(self, parent=None):
        super(MyQToolBox, self).__init__(parent)

        self.__iclosed = QtWidgets.QWidget().style().standardIcon(
            QtWidgets.QStyle.SP_ToolBarHorizontalExtensionButton)
        self.__iexpand = QtWidgets.QWidget().style().standardIcon(
            QtWidgets.QStyle.SP_ToolBarVerticalExtensionButton)

        self.setLayout(QtWidgets.QGridLayout())
        self.layout().setContentsMargins(0, 0, 0, 0)

        self.__currentIndex = -1

    def setIcons(self, ar_right, ar_down):  # =================================
        self.__iclosed = ar_right
        self.__iexpand = ar_down

    def addItem(self, tool, text):  # =========================================

        N = self.layout().rowCount()

        # ---- Add Header ----

        head = QtWidgets.QPushButton(text)
        head.setIcon(self.__iclosed)
        head.clicked.connect(self.__isClicked__)
        head.setStyleSheet("QPushButton {text-align:left;}")
        head.sizeHint().height()

        self.layout().addWidget(head, N - 1, 0)

        # ---- Add Item in a ScrollArea ----

        scrollarea = QtWidgets.QScrollArea()
        scrollarea.setFrameStyle(0)
        scrollarea.hide()
        scrollarea.setStyleSheet("QScrollArea {background-color:transparent;}")
        scrollarea.setWidgetResizable(True)

        tool.setObjectName("myViewport")
        tool.setStyleSheet("#myViewport {background-color:transparent;}")
        scrollarea.setWidget(tool)

        self.layout().addWidget(scrollarea, N, 0)

    def __isClicked__(self):  # ===============================================

        for row in range(0, self.layout().rowCount() - 1, 2):

            head = self.layout().itemAtPosition(0, 0).widget()
            tool = self.layout().itemAtPosition(row + 1, 0).widget()

            if head == self.sender():
                if self.__currentIndex == row:
                    # if clicked tool is open, closes it
                    head.setIcon(self.__iclosed)
                    tool.hide()
                    self.__currentIndex = -1
                    self.sizeChanged.emit(self.sizeHint().height() + head.sizeHint().height())

                else:
                    # if clicked tool is closed, expands it
                    head.setIcon(self.__iexpand)
                    tool.show()
                    self.__currentIndex = row
                    self.toolboxOpened.emit()
                    self.sizeChanged.emit(self.sizeHint().height() + head.sizeHint().height())

    def closeAll(self):  # ====================================================

        # allows opening only one toolbox at a time using the 'toolboxOpened' signal

        for row in range(0, self.layout().rowCount() - 1, 2):

            head = self.layout().itemAtPosition(0, 0).widget()
            tool = self.layout().itemAtPosition(row + 1, 0).widget()

            if self.__currentIndex == row:

                    head.setIcon(self.__iclosed)
                    tool.hide()
                    self.__currentIndex = -1
                    self.sizeChanged.emit(self.sizeHint().height() + head.sizeHint().height())


if __name__ == '__main__':

    app = QtWidgets.QApplication(sys.argv)

    bh_tomo = BhTomoPy()
    bh_tomo.show()

    sys.exit(app.exec_())