Russian Qt Forum

class QComboCheckBoxModel : public QSqlQueryModel {
        Q_OBJECT
    public:
        QComboCheckBoxModel(QObject *parent = 0);
        QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const;
    private:

        Qt::ItemFlags flags(const QModelIndex &index) const;
        bool setData(const QModelIndex &index, const QVariant &value, int role);

};


class QComboCheckBox: public QComboBox
{
      Q_OBJECT;

public:

      QComboCheckBox(QWidget *widget = 0);
      virtual ~QComboCheckBox();


      bool eventFilter(QObject * watched, QEvent * event);
      void showPopup();

private:
    QListView * m_listView;
    bool m_opening;

};

////////////////////////////////////////////////////////////////////////////////////
// конструктор
////////////////////////////////////////////////////////////////////////////////////
QComboCheckBox::QComboCheckBox(QWidget *parent) :
    QComboBox(parent)
{
    // заменяем стандартный вьювер
    m_listView = new QListView(this);
    setView(m_listView);
    // устанавливаем перехват событий
    m_listView->viewport()->installEventFilter(this);
    // флаг открытия комбобокса
    m_opening = false;
}

QComboCheckBox::~QComboCheckBox()
{
}

////////////////////////////////////////////////////////////////////////////////
//  Перехватчик событий для работы чекбоксов
////////////////////////////////////////////////////////////////////////////////
bool QComboCheckBox::eventFilter(QObject * watched, QEvent * event)
{
    // проверка тика отловленного события
    if (event->type() == QEvent::MouseButtonRelease)
    {
        // блокируем смену галочки при открытии
        if (m_opening)
        {
            m_opening = false;
            return QObject::eventFilter(watched, event);
        }
        // проверяем тип
        if (watched->parent()->inherits("QListView"))
        {
            // приводим к нужным типам
            QListView *tmp = (QListView *)(watched->parent());
            QMouseEvent *mEvent = (QMouseEvent *)event;
            QModelIndex ind = tmp->indexAt(mEvent->pos());
            // меняем состояние cheched
            bool checked = tmp->model()->data(ind,Qt::CheckStateRole).toBool();
            tmp->model()->setData(ind,!checked,Qt::CheckStateRole);
            // блокируем закрытие комбобокса
            return true;
        }
    }
    return QObject::eventFilter(watched, event);
}

////////////////////////////////////////////////////////////////////////////////
// функция раскрытия комбобокса
////////////////////////////////////////////////////////////////////////////////
void QComboCheckBox::showPopup()
{
    // флаг открытия комбобокса
    m_opening = true;
    // вызываем функцию класса - предка
    QComboBox::showPopup();
};

////////////////////////////////////////////////////////////////////////////////
// определяем модель для QComboCheckBox
////////////////////////////////////////////////////////////////////////////////
QComboCheckBoxModel::QComboCheckBoxModel(QObject *parent)
                 : QSqlQueryModel(parent) {
};

QVariant QComboCheckBoxModel::data(
             const QModelIndex &index,
             int role) const {
    QVariant value = QSqlQueryModel::data(index, role);
    if (index.column() == 0) value = value.toBool();
    return value;
}

Qt::ItemFlags QComboCheckBoxModel::flags(
        const QModelIndex &index) const
{
    Qt::ItemFlags flags = QSqlQueryModel::flags(index);
    if (index.column() == 0)
          flags = flags | Qt::ItemIsUserCheckable | Qt::ItemIsEditable;
    return flags;
}

bool QComboCheckBoxModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if (index.column() == 0) return QSqlQueryModel::setData(index, value, role);
    return true;
}