QDialog的封装实现自定义选择对话框

效果图：

注意事项：

1、该窗体是通过封装QDialog类实现的

2、提供了两个对外的接口

• 批量增加item
• 返回用户选择的item

3、该窗体提供查找、选择功能

4、该窗体类关闭时只是单纯的隐藏，没有释放内存（需要注意）

 

代码：

wpselectdialog.h：

#ifndef WPSELECTDIALOG_H
#define WPSELECTDIALOG_H

#include <QDialog>
#include <QLabel>
#include <QLineEdit>
#include <QPushButton>
#include <QTreeWidget>
#include <QTreeWidgetItem>
#include <QList>
#include <QVariant>


class WPSelectDialog : public QDialog
{
    Q_OBJECT
public:
    explicit WPSelectDialog(QWidget *parent = 0);

    void addTopItems(QList<QTreeWidgetItem *> &items);

    QVariant getUserSelected();

signals:

public slots:

private:
    QLabel *findLabel;
    QLineEdit *inputEdit;
    QPushButton *findButton;
    QTreeWidget *treeWidget;
    QPushButton *okButton;
    QPushButton *cancelButton;

    QVariant userSelected;

private slots:
    void slot_currentItemChanged(QTreeWidgetItem*,QTreeWidgetItem*);

    void slot_itemClicked(QTreeWidgetItem*,int);

    void slot_okButtonClicked();

    void slot_cancelButtonClicked();

    void slot_findButtonClicked();

};

#endif // WPSELECTDIALOG_H

wpselectdialog.cpp：

#include "wpselectdialog.h"

#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QTreeWidgetItemIterator>

//初始化组件
WPSelectDialog::WPSelectDialog(QWidget *parent) :
    QDialog(parent)
{
    this->findLabel=new QLabel("查找：");
    this->inputEdit=new QLineEdit;

    this->findButton=new QPushButton("查找");
    connect(this->findButton,SIGNAL(clicked()),this,SLOT(slot_findButtonClicked()));

    this->treeWidget=new QTreeWidget;
    this->treeWidget->setHeaderHidden(true);
    connect(this->treeWidget,SIGNAL(currentItemChanged(QTreeWidgetItem*,QTreeWidgetItem*)),
            this,SLOT(slot_currentItemChanged(QTreeWidgetItem*,QTreeWidgetItem*)));

    connect(this->treeWidget,SIGNAL(itemClicked(QTreeWidgetItem*,int)),
            this,SLOT(slot_itemClicked(QTreeWidgetItem*,int)));

    this->okButton=new QPushButton("确定");
    connect(this->okButton,SIGNAL(clicked()),this,SLOT(slot_okButtonClicked()));

    this->cancelButton=new QPushButton("取消");
    connect(this->cancelButton,SIGNAL(clicked()),this,SLOT(slot_cancelButtonClicked()));

    QHBoxLayout *topLayout=new QHBoxLayout;

    topLayout->addWidget(this->findLabel);
    topLayout->addWidget(this->inputEdit);
    topLayout->addWidget(this->findButton);

    QHBoxLayout *bottomLayout=new QHBoxLayout;

    bottomLayout->addWidget(this->okButton);
    bottomLayout->addWidget(this->cancelButton);

    QVBoxLayout *mainLayout=new QVBoxLayout;

    mainLayout->addLayout(topLayout);
    mainLayout->addWidget(this->treeWidget);
    mainLayout->addLayout(bottomLayout);

    this->setLayout(mainLayout);

    this->setWindowTitle(QString("场站选择列表框"));
    this->resize(300,400);

}

//对外接口，提供批量添加items
void WPSelectDialog::addTopItems(QList<QTreeWidgetItem *> &items)
{
    this->treeWidget->addTopLevelItems(items);
}

void WPSelectDialog::slot_currentItemChanged(QTreeWidgetItem *current, QTreeWidgetItem *previous)
{
    if(!previous)
    {
        return;
    }

    previous->setCheckState(0,Qt::Unchecked);

    current->setCheckState(0,Qt::Checked);
}

void WPSelectDialog::slot_itemClicked(QTreeWidgetItem *item, int column)
{
    if(column==0)
    {
        if(item->checkState(0)==Qt::Unchecked)
        {
            item->setCheckState(0,Qt::Checked);
        }
    }
}

void WPSelectDialog::slot_okButtonClicked()
{
    if(!(this->treeWidget->currentItem()))
    {
        return;
    }

    this->userSelected=this->treeWidget->currentItem()->data(0,Qt::UserRole);

    this->close();
}

//对外接口，返回用户选择的item的数据
QVariant WPSelectDialog::getUserSelected()
{
    return this->userSelected;
}

void WPSelectDialog::slot_cancelButtonClicked()
{
    this->close();
}

void WPSelectDialog::slot_findButtonClicked()
{
    if(this->inputEdit->text().isEmpty())
    {
        return;
    }

    QTreeWidgetItemIterator iter(this->treeWidget);

    while(*iter)
    {
        if((*iter)->text(0).contains(this->inputEdit->text().trimmed()))
        {
            this->treeWidget->setCurrentItem(*iter);
            break;
        }
        ++iter;
    }
}