Mã:
#include <QtGui>
#include "tree.h"
#include "adddialog.h"
Tree::Tree(QWidget *parent)
: QTreeWidget(parent)
{
QStringList labels;
labels << trUtf8("Tên") << trUtf8("Số Điện Thoại") << trUtf8("Địa Chỉ");
header()->setResizeMode(QHeaderView::Stretch);
setHeaderLabels(labels);
Pre_Icon.addPixmap(style()->standardPixmap(QStyle::SP_ArrowDown),
QIcon::Normal, QIcon::Off);
Pre_Icon.addPixmap(style()->standardPixmap(QStyle::SP_ArrowForward),
QIcon::Normal, QIcon::On);
Person_Icon.addPixmap(style()->standardPixmap(QStyle::SP_TitleBarMinButton));
}
bool Tree::read(QIODevice *device)
{
QString errorStr;
int errorLine;
int errorColumn;
if (!domDocument.setContent(device, true, &errorStr, &errorLine,
&errorColumn)) {
QMessageBox::information(window(), trUtf8("Error"),
trUtf8("Parser XML lỗi tại dòng %1, cột %2:\n%3")
.arg(errorLine)
.arg(errorColumn)
.arg(errorStr));
return false;
}
QDomElement root = domDocument.documentElement();
if (root.tagName() != "qldb") {
QMessageBox::information(window(), trUtf8("Error"),
trUtf8("Chỉ đọc file XML quản lý danh bạ."));
return false;
}
clear();
disconnect(this, SIGNAL(itemChanged(QTreeWidgetItem*,int)),
this, SLOT(updateDomElement(QTreeWidgetItem*,int)));
disconnect(this,SIGNAL(itemClicked(QTreeWidgetItem*,int)),this,
SLOT(show_output(QTreeWidgetItem*,int,QString&rtrn)));
QDomElement child = root.firstChildElement("pre");
while (!child.isNull()) {
QDomElement nd = child.firstChildElement("person");
if(nd.isNull()){
child.setAttribute("folded","no");
child = child.nextSiblingElement("pre");
}
else{
Parse(child);
child = child.nextSiblingElement("pre");
}
}
connect(this,SIGNAL(itemClicked(QTreeWidgetItem*,int)),this,
SLOT(show_output(QTreeWidgetItem*,int,QString&rtrn)));
connect(this, SIGNAL(itemChanged(QTreeWidgetItem*,int)),
this, SLOT(updateDomElement(QTreeWidgetItem*,int)));
return true;
}
bool Tree::write(QIODevice *device)
{
const int IndentSize = 4;
QTextStream out(device);
domDocument.save(out, IndentSize);
return true;
}
void Tree::updateDomElement(QTreeWidgetItem *item, int column)
{
QDomElement element = domElementForItem.value(item);
if (!element.isNull()) {
if (column == 0) {
QDomElement oldTitleElement = element.firstChildElement("name");
QDomElement newTitleElement = domDocument.createElement("name");
QDomText newTitleText = domDocument.createTextNode(item->text(0));
newTitleElement.appendChild(newTitleText);
element.replaceChild(newTitleElement, oldTitleElement);
}
if(column == 1){
QDomElement oldAddElement = element.firstChildElement("add");
QDomElement newAddElement = domDocument.createElement("add");
QDomText newAddText = domDocument.createTextNode(item->text(1));
newAddElement.appendChild(newAddText);
element.replaceChild(newAddElement,oldAddElement);
}
if(column == 2)
{
QDomElement oldPhoneElement = element.firstChildElement("phonenum");
QDomElement newPhoneElement = domDocument.createElement("phonenum");
QDomText newPhoneText = domDocument.createTextNode(item->text(2));
newPhoneElement.appendChild(newPhoneText);
element.replaceChild(newPhoneElement,oldPhoneElement);
}
}
}
void Tree::Parse(const QDomElement &element,QTreeWidgetItem *parentItem)
{
QTreeWidgetItem *item = createItem(element, parentItem);
QString title = element.attribute("name");
if (title.isEmpty())
title = QObject::tr("Undefine");
item->setFlags(item->flags() | Qt::ItemIsEditable);
item->setIcon(0, Pre_Icon);
item->setText(0, title);
bool folded = (element.attribute("folded") != "yes");
setItemExpanded(item, !folded);
QDomElement child = element.firstChildElement();
while (!child.isNull()) {
if (child.tagName() == "pre") {
Parse(child, item);
} else if (child.tagName() == "person") {
QTreeWidgetItem *childItem = createItem(child, item);
QString title = child.firstChildElement("name").text();
if (title.isEmpty())
title = QObject::tr("Undefine");
childItem->setFlags(item->flags() | Qt::ItemIsEditable);
childItem->setIcon(0, Person_Icon);
childItem->setText(0, title);
childItem->setFlags(item->flags() | Qt::ItemIsEditable);
childItem->setIcon(0, Person_Icon);
childItem->setText(0, title);
childItem->setText(1, child.firstChildElement("phonenum").text());
childItem->setFlags(item->flags() | Qt::ItemIsEditable);
childItem->setIcon(0, Person_Icon);
childItem->setText(0, title);
childItem->setText(2, child.firstChildElement("add").text());
}
child = child.nextSiblingElement();
}
}
QTreeWidgetItem *Tree::createItem(const QDomElement &element,QTreeWidgetItem *parentItem)
{
QTreeWidgetItem *item;
if (parentItem) {
item = new QTreeWidgetItem(parentItem);
} else {
item = new QTreeWidgetItem(this);
}
domElementForItem.insert(item, element);
return item;
}
bool Tree::insert(QString &pr, QString &n, QString &a, QString &p){
QDomElement person = domDocument.createElement("person");
QDomElement name = domDocument.createElement("name");
QDomElement address = domDocument.createElement("add");
QDomElement phonenum = domDocument.createElement("phonenum");
QDomText name_value = domDocument.createTextNode(n);
QDomText address_value = domDocument.createTextNode(a);
QDomText phonenum_value = domDocument.createTextNode(p);
person.appendChild(name);
person.appendChild(phonenum);
person.appendChild(address);
name.appendChild(name_value);
address.appendChild(address_value);
phonenum.appendChild(phonenum_value);
QDomElement root = domDocument.documentElement();
QDomElement child = root.firstChildElement("pre");
while (!child.isNull()) {
if(child.attribute("name") == pr)
{
child.setAttribute("folded","yes");
child.appendChild(person);
break;
}
child=child.nextSiblingElement();
}
return true;
}
void Tree::show_output(QTreeWidgetItem *item,int column,QString &rtrn){
QDomElement element = domElementForItem.value(item);
if (!element.isNull()) {
if (column == 0) {
QDomElement out = element.firstChildElement("name");
rtrn = out.text();
}
if (column == 1) {
QDomElement out = element.firstChildElement("phonenum");
rtrn = out.text();
}
if (column == 2) {
QDomElement out = element.firstChildElement("add");
rtrn = out.text();
}
}
else{
rtrn = (trUtf8("Nhập SĐT"));
}
}
bool Tree::remove(QString &p){
QDomElement root = domDocument.documentElement();
QDomElement pre = root.firstChildElement("pre");
while (!pre.isNull())
{
QDomElement person = pre.firstChildElement("person");
while(!person.isNull())
{
QDomElement phonenum = person.firstChildElement("phonenum");
if(phonenum.text() == p){
pre.removeChild(person);
return true;
}
person = person.nextSiblingElement();
}
pre = pre.nextSiblingElement();
}
return false;
}
bool Tree::find_value(QString &p){
QDomElement root = domDocument.documentElement();
QDomElement pre = root.firstChildElement("pre");
while (!pre.isNull())
{
QDomElement person = pre.firstChildElement("person");
while(!person.isNull())
{
QDomElement phonenum = person.firstChildElement("phonenum");
if(phonenum.text() == p){
QMessageBox::information(this,trUtf8("Tìm kiếm"),
trUtf8("<font color=green size=4>Tìm thấy trong danh mục: </font>")
+"<font color=green size=5><b>"+(pre.attribute("name"))+"</b></font>");
return true;
}
person = person.nextSiblingElement();
}
pre = pre.nextSiblingElement();
}
return false;
}
