\r
#include "ShoppingTreeItem.h"\r
#include <QFile>\r
+#include <QtGui>\r
\r
ShoppingTreeModel::ShoppingTreeModel(const QString &xmlFileName,\r
QObject *parent) :\r
if(!m_document.setContent(&file, true, &error, &errLine, &errColumn))\r
{\r
emit xmlParseError(error, errLine, errColumn);\r
- file.close(();\r
+ file.close();\r
return;\r
}\r
file.close();\r
return;\r
}\r
else if(root.attribute("version") == "1.0")\r
+ {\r
// set column titles\r
QVector<QVariant> rootData;\r
rootData << "Category/Item name"\r
{\r
rootItem->child(rootItem->childCount() - 1)->setData(column, columnData[column]);\r
}\r
+ m_domElementForItem.insert(rootItem->child(rootItem->childCount() - 1),\r
+ child);\r
+ }\r
+}\r
+\r
+ShoppingTreeModel::~ShoppingTreeModel()\r
+{\r
+ delete rootItem;\r
+}\r
+\r
+QVariant ShoppingTreeModel::data(const QModelIndex &index, int role) const\r
+{\r
+ if(!index.isValid())\r
+ return QVariant();\r
+\r
+ if(role != Qt::DisplayRole && role != Qt::EditRole)\r
+ return QVariant();\r
+\r
+ ShoppingTreeItem *item = getItem(index);\r
+ return item->data(index.column());\r
+}\r
+\r
+Qt::ItemFlags ShoppingTreeModel::flags(const QModelIndex &index) const\r
+{\r
+ if(!index.isValid())\r
+ return 0;\r
+\r
+ return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable;\r
+}\r
+\r
+ShoppingTreeItem *ShoppingTreeModel::getItem(const QModelIndex &index) const\r
+{\r
+ if(index.isValid()){\r
+ ShoppingTreeItem *item = static_cast<ShoppingTreeItem*>(index.internalPointer());\r
+ if(item) return item;\r
}\r
+\r
+ return rootItem;\r
+}\r
+\r
+QVariant ShoppingTreeModel::headerData(int section, Qt::Orientation orientation,\r
+ int role) const\r
+{\r
+ if(orientation == Qt::Horizontal && role == Qt::DisplayRole)\r
+ return rootItem->data(section);\r
+\r
+ return QVariant();\r
+}\r
+\r
+QModelIndex ShoppingTreeModel::index(int row, int column, const QModelIndex &parent) const\r
+{\r
+ if(parent.isValid() && parent.column() != 0)\r
+ return QModelIndex();\r
+\r
+ ShoppingTreeItem *parentItem = getItem(parent);\r
+\r
+ ShoppingTreeItem *childItem = parentItem->child(row);\r
+ if(childItem)\r
+ return createIndex(row, column, childItem);\r
+ else\r
+ return QModelIndex();\r
+}\r
+\r
+bool ShoppingTreeModel::insertColumns(int position, int columns, const QModelIndex &parent)\r
+{\r
+ bool success;\r
+\r
+ beginInsertColumns(parent, position, position + columns - 1);\r
+ success = rootItem->insertColumns(position, columns);\r
+ endInsertColumns();\r
+\r
+ return success;\r
+}\r
+\r
+bool ShoppingTreeModel::insertRows(int position, int rows, const QModelIndex &parent)\r
+{\r
+ ShoppingTreeItem *parentItem = getItem(parent);\r
+ bool success;\r
+\r
+ beginInsertRows(parent, position, position + rows - 1);\r
+ success = parentItem->insertChildren(position, rows, rootItem->columnCount());\r
+ endInsertRows();\r
+\r
+ return success;\r
+}\r
+\r
+QModelIndex ShoppingTreeModel::parent(const QModelIndex &index) const\r
+{\r
+ if(!index.isValid())\r
+ return QModelIndex();\r
+\r
+ ShoppingTreeItem *childItem = getItem(index);\r
+ ShoppingTreeItem *parentItem = childItem->parent();\r
+\r
+ if(parentItem == rootItem)\r
+ return QModelIndex();\r
+\r
+ return createIndex(parentItem->childNumber(), 0, parentItem);\r
+}\r
+\r
+bool ShoppingTreeModel::removeColumns(int position, int columns, const QModelIndex &parent)\r
+{\r
+ bool success;\r
+\r
+ beginRemoveColumns(parent, position, position + columns - 1);\r
+ success = rootItem->removeColumns(position, columns);\r
+ endRemoveColumns();\r
+\r
+ if(rootItem->columnCount() != 0)\r
+ removeRows(0, rowCount());\r
+\r
+ return success;\r
+}\r
+\r
+bool ShoppingTreeModel::removeRows(int position, int rows, const QModelIndex &parent)\r
+{\r
+ ShoppingTreeItem *parentItem = getItem(parent);\r
+ bool success;\r
+\r
+ beginRemoveRows(parent, position, position + rows - 1);\r
+ success = parentItem->removeChildren(position, rows);\r
+ endRemoveRows();\r
+\r
+ return success;\r
+}\r
+\r
+int ShoppingTreeModel::rowCount(const QModelIndex &parent) const\r
+{\r
+ ShoppingTreeItem *parentItem = getItem(parent);\r
+\r
+ return parentItem->childCount();\r
+}\r
+\r
+int ShoppingTreeModel::columnCount(const QModelIndex &parent) const\r
+{\r
+ return rootItem->columnCount();\r
+}\r
+bool ShoppingTreeModel::setData(const QModelIndex &index, const QVariant &value, int role)\r
+{\r
+ if(role != Qt::EditRole)\r
+ return false;\r
+\r
+ ShoppingTreeItem *item = getItem(index);\r
+ bool result = item->setData(index.column(),value);\r
+\r
+ if(result)\r
+ emit dataChanged(index, index);\r
+\r
+ return result;\r
+}\r
+\r
+bool ShoppingTreeModel::setHeaderData(int section, Qt::Orientation orientation,\r
+ const QVariant &value, int role)\r
+{\r
+ if(role != Qt::EditRole || orientation != Qt::Horizontal)\r
+ return false;\r
+\r
+ bool result = rootItem->setData(section, value);\r
+\r
+ if(result)\r
+ emit headerDataChanged(orientation, section, section);\r
+\r
+ return result;\r
}\r
\r
void ShoppingTreeModel::parseCategoryElement(const QDomElement &element,\r
ShoppingTreeItem *parentItem)\r
{\r
+ // if parent is null then add category to root\r
if(!parentItem)\r
parentItem = rootItem;\r
- // TODO\r
\r
+ ShoppingTreeItem *item;\r
+ QString title = element.firstChildElement("title").text();\r
+ if(!title.isEmpty())\r
+ {\r
+ parentItem->insertChildren(parentItem->childCount(), 1,\r
+ rootItem->columnCount());\r
+\r
+ parentItem->child(parentItem->childCount() - 1)->setData(0, title);\r
+ m_domElementForItem.insert(parentItem->child(parentItem->childCount() - 1),\r
+ element);\r
+ item = parentItem->child(parentItem->childCount() - 1);\r
+ }\r
+ else\r
+ {\r
+ emit invalidDocument();\r
+ return;\r
+ }\r
+\r
+ // add each sub category and item to the tree\r
+ QDomElement child = element.firstChildElement();\r
+ while(!child.isNull())\r
+ {\r
+ if(child.tagName() == "category")\r
+ {\r
+ parseCategoryElement(child, parentItem);\r
+ }\r
+ else if(child.tagName() == "item")\r
+ {\r
+ item->insertChildren(\r
+ item->childCount(), 1,\r
+ rootItem->columnCount());\r
+ QVector<QVariant> columnData =\r
+ getColumnsFromItemElement(child);\r
+ for(int column = 0; column < columnData.size(); column++)\r
+ {\r
+ item->child(item->childCount() - 1)->setData(column, columnData[column]);\r
+ }\r
+ m_domElementForItem.insert(item->child(item->childCount() - 1),\r
+ child);\r
+ }\r
+ else\r
+ {\r
+ emit invalidDocument();\r
+ return;\r
+ }\r
+\r
+ child = child.nextSiblingElement();\r
+ }\r
}\r
\r
QVector<QVariant> ShoppingTreeModel::getColumnsFromItemElement(const QDomElement &element)\r
{\r
- QString title = child.firstChildElement("title").text();\r
- int quantity = child.firstChildElement("quantity").text().toInt();\r
- QString store = child.firstChildElement("store").text();\r
+ QVector<QVariant> data;\r
+ QString title = element.firstChildElement("title").text();\r
+ int quantity = element.firstChildElement("quantity").text().toInt();\r
+ QString store = element.firstChildElement("store").text();\r
if(title.isEmpty() || quantity < 0)\r
{\r
- emit invalidDocument;\r
- return;\r
+ emit invalidDocument();\r
+ return data;\r
}\r
\r
- QVector<QVariant> data;\r
data << title << quantity << store;\r
return data;\r
}\r