MTG/dunya-desktop

import os
import platform

from PyQt5.QtWidgets import (QFrame, QGridLayout, QSizePolicy, QLayout,
                             QHBoxLayout, QToolButton, QSpacerItem)
from PyQt5.QtGui import QCursor, QIcon
from PyQt5.QtCore import Qt, QSize

from .combobox import ComboBox

# platform dependent margins/spacings
if platform.system() == 'Linux':
    MARGIN = 2
    SPACING = 3
    SPACE = 3
else:  # now, only for mac
    MARGIN = 5
    SPACING = 5
    SPACE = 7

# css paths
QUERY_ICON = os.path.join(os.path.dirname(__file__), '..', 'ui_files',
                          'icons', 'magnifying-glass.png')


class AudioAttFrame(QFrame):
    """Frame contains the comboboxes of attributes (such as makams, forms,
    etc) and query button"""

    def __init__(self, qwidget_parent=None):
        QFrame.__init__(self, qwidget_parent)
        self._set_size_attributes()

        layout = QGridLayout(self)
        self._set_layout(layout)
        self._retranslate_status_tips()

        self.toolButton_query.setDisabled(True)

        # signals
        self.comboBox_melodic.currentIndexChanged.connect(self.set_toolbutton)
        self.comboBox_form.currentIndexChanged.connect(self.set_toolbutton)
        self.comboBox_rhythm.currentIndexChanged.connect(self.set_toolbutton)
        self.comboBox_composer.currentIndexChanged.connect(self.set_toolbutton)
        self.comboBox_performer.currentIndexChanged.connect(
            self.set_toolbutton)
        self.comboBox_instrument.currentIndexChanged.connect(
            self.set_toolbutton)

    def _set_size_attributes(self):
        """Sets the size policies of frame"""
        size_policy = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Fixed)
        size_policy.setHorizontalStretch(0)
        size_policy.setVerticalStretch(0)
        size_policy.setHeightForWidth(self.sizePolicy().hasHeightForWidth())
        self.setSizePolicy(size_policy)
        self.setCursor(QCursor(Qt.ArrowCursor))
        self.setFrameShape(QFrame.StyledPanel)
        self.setFrameShadow(QFrame.Raised)
        self.setLineWidth(1)

    def _set_layout(self, layout):
        """Sets the size policies of layout and initializes the comoboxes and
        query button."""
        layout.setSizeConstraint(QLayout.SetNoConstraint)
        layout.setMargin(MARGIN)
        layout.setSpacing(SPACING)

        # combo boxes
        # melodic structure
        self.comboBox_melodic = ComboBox(self)
        layout.addWidget(self.comboBox_melodic, 0, 0, 1, 1)

        # form structure
        self.comboBox_form = ComboBox(self)
        layout.addWidget(self.comboBox_form, 0, 2, 1, 1)

        # rhythmic structure
        self.comboBox_rhythm = ComboBox(self)
        layout.addWidget(self.comboBox_rhythm, 0, 4, 1, 1)

        # composer
        self.comboBox_composer = ComboBox(self)
        layout.addWidget(self.comboBox_composer, 1, 0, 1, 1)
        self.comboBox_composer.set_placeholder_text('Composer')

        # performer
        self.comboBox_performer = ComboBox(self)
        layout.addWidget(self.comboBox_performer, 1, 2, 1, 1)
        self.comboBox_performer.set_placeholder_text('Performer')

        # instrument
        self.comboBox_instrument = ComboBox(self)
        layout.addWidget(self.comboBox_instrument, 1, 4, 1, 1)
        self.comboBox_instrument.set_placeholder_text('Instrument')

        # spacers between the comboboxes
        spacer_item1 = QSpacerItem(SPACE, 20, QSizePolicy.Minimum,
                                   QSizePolicy.Fixed)

        spacer_item2 = QSpacerItem(SPACE, 20, QSizePolicy.Minimum,
                                   QSizePolicy.Fixed)

        if platform.system() != 'Linux':
            spacer_item3 = QSpacerItem(SPACE, 20, QSizePolicy.Minimum,
                                       QSizePolicy.Fixed)
            layout.addItem(spacer_item3, 1, 5, 1, 1)

        layout.addItem(spacer_item1, 1, 1, 1, 1)
        layout.addItem(spacer_item2, 1, 3, 1, 1)

        # query button and layout
        self.horizontalLayout_query = QHBoxLayout()
        self.horizontalLayout_query.setSpacing(0)

        self.toolButton_query = QToolButton(self)
        self.toolButton_query.setFixedSize(QSize(50, 50))
        self.toolButton_query.setIcon(QIcon(QUERY_ICON))
        self.toolButton_query.setIconSize(QSize(25, 25))
        self.horizontalLayout_query.addWidget(self.toolButton_query)
        layout.addLayout(self.horizontalLayout_query, 0, 6, 2, 1)

    def _retranslate_status_tips(self):
        """Sets the status tips of comboboxes and query button"""
        self.comboBox_melodic.setStatusTip("Select melodic attribute")
        self.comboBox_form.setStatusTip("Select form attribute")
        self.comboBox_rhythm.setStatusTip("Select rhythm attribute")
        self.comboBox_composer.setStatusTip("Select composer")
        self.comboBox_performer.setStatusTip("Select performer")
        self.comboBox_instrument.setStatusTip("Select instrument")
        self.toolButton_query.setStatusTip("Query your selection")

    def set_toolbutton(self):
        """Checks the comboboxes and enables/disables the query button"""
        index_melodic = self.comboBox_melodic.currentIndex()
        index_form = self.comboBox_form.currentIndex()
        index_rhythm = self.comboBox_rhythm.currentIndex()
        index_composer = self.comboBox_composer.currentIndex()
        index_performer = self.comboBox_performer.currentIndex()
        index_instrument = self.comboBox_instrument.currentIndex()

        if index_melodic is -1 and index_form is -1 and index_rhythm is -1 \
                and index_composer is -1 and index_performer is -1 and \
                        index_instrument is -1:
            self.toolButton_query.setDisabled(True)
        else:
            self.toolButton_query.setEnabled(True)