-/*\r
- * This file is part of family-shop-mgr.\r
- *\r
- * family-shop-mgr is free software: you can redistribute it and/or modify\r
- * it under the terms of the GNU General Public License as published by\r
- * the Free Software Foundation, either version 3 of the License, or\r
- * (at your option) any later version.\r
- *\r
- * family-shop-mgr is distributed in the hope that it will be useful,\r
- * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\r
- * GNU General Public License for more details.\r
- *\r
- * You should have received a copy of the GNU General Public License\r
- * along with family-shop-mgr. If not, see <http://www.gnu.org/licenses/>.\r
- *\r
- * Author: Unai IRIGOYEN\r
- * Date: 12/07/2009\r
- *\r
- */\r
-\r
-#include <QtXml>\r
-\r
-#include "ShoppingTreeItem.h"\r
-#include "ShoppingTreeModel.h"\r
-\r
-ShoppingTreeModel::ShoppingTreeModel(QString fileName, QObject *parent)\r
- : QAbstractItemModel(parent),m_fileName(fileName)\r
-{\r
- connect(this, SIGNAL(documentChanged()), this, SLOT(save()));\r
- QFile file(m_fileName);\r
- if (!file.open(QIODevice::ReadOnly))\r
- return;\r
- if (!domDocument.setContent(&file)) {\r
- file.close();\r
- return;\r
- }\r
- file.close();\r
-\r
- rootItemNode = domDocument.firstChildElement("shoppingList");\r
- rootItem = new ShoppingTreeItem(rootItemNode, 0);\r
-}\r
-\r
-/*******************************************************************/\r
-ShoppingTreeModel::~ShoppingTreeModel()\r
-{\r
- delete rootItem;\r
-}\r
-\r
-/*******************************************************************/\r
-int ShoppingTreeModel::columnCount(const QModelIndex &/*parent*/) const\r
-{\r
- return 4;\r
-}\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)\r
- return QVariant();\r
-\r
- ShoppingTreeItem *item = static_cast<ShoppingTreeItem*>(index.internalPointer());\r
-\r
- QDomNode node = item->node();\r
-\r
- switch (index.column()) {\r
- case 0:\r
- return node.firstChildElement("title").text();\r
- case 1:\r
- if(node.toElement().nodeName() == "item")\r
- {\r
- return node.firstChildElement("quantity").text();\r
- }\r
- return QVariant();\r
- case 2:\r
- if(node.toElement().nodeName() == "item")\r
- {\r
- return node.firstChildElement("store").text();\r
- }\r
- return QVariant();\r
- case 3:\r
- if(node.toElement().nodeName() == "item")\r
- {\r
- return (bool) node.firstChildElement("checked").text().toInt();\r
- }\r
- return QVariant();\r
- default:\r
- return QVariant();\r
- }\r
-}\r
-\r
-/*******************************************************************/\r
-Qt::ItemFlags ShoppingTreeModel::flags(const QModelIndex &index) const\r
-{\r
- if (!index.isValid())\r
- return 0;\r
-\r
- return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;\r
-}\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
- switch (section) {\r
- case 0:\r
- return tr("Category/Item name");\r
- case 1:\r
- return tr("Quantity");\r
- case 2:\r
- return tr("Store");\r
- case 3:\r
- return tr("Checked");\r
- default:\r
- return QVariant();\r
- }\r
- }\r
-\r
- return QVariant();\r
-}\r
-\r
-/*******************************************************************/\r
-QModelIndex ShoppingTreeModel::index(int row, int column, const QModelIndex &parent)\r
- const\r
-{\r
- if (!hasIndex(row, column, parent))\r
- return QModelIndex();\r
-\r
- ShoppingTreeItem *parentItem;\r
-\r
- if (!parent.isValid())\r
- parentItem = rootItem;\r
- else\r
- parentItem = static_cast<ShoppingTreeItem*>(parent.internalPointer());\r
-\r
- ShoppingTreeItem *childItem = parentItem->child(row);\r
- if (childItem)\r
- return createIndex(row, column, childItem);\r
- else\r
- return QModelIndex();\r
-}\r
-\r
-/*******************************************************************/\r
-QModelIndex ShoppingTreeModel::parent(const QModelIndex &child) const\r
-{\r
- if (!child.isValid())\r
- return QModelIndex();\r
-\r
- ShoppingTreeItem *childItem = static_cast<ShoppingTreeItem*>(child.internalPointer());\r
- ShoppingTreeItem *parentItem = childItem->parent();\r
-\r
- if (!parentItem || parentItem == rootItem)\r
- return QModelIndex();\r
-\r
- return createIndex(parentItem->row(), 0, parentItem);\r
-}\r
-\r
-/*******************************************************************/\r
-int ShoppingTreeModel::rowCount(const QModelIndex &parent) const\r
-{\r
- if (parent.column() > 0)\r
- return 0;\r
-\r
- ShoppingTreeItem *parentItem;\r
-\r
- if (!parent.isValid())\r
- parentItem = rootItem;\r
- else\r
- parentItem = static_cast<ShoppingTreeItem*>(parent.internalPointer());\r
-\r
- int childCount = 0;\r
- for(QDomElement childElement = parentItem->node().firstChildElement();\r
- !childElement.isNull(); childElement = childElement.nextSiblingElement())\r
- {\r
- if(childElement.nodeName() == "category" || childElement.nodeName() == "item")\r
- childCount++;\r
- }\r
- return childCount;\r
-}\r
-\r
-/*******************************************************************/\r
-bool ShoppingTreeModel::insertRows(int row, int count, const QModelIndex &parent, ItemType type)\r
-{\r
- ShoppingTreeItem *parentItem;\r
-\r
- if (!parent.isValid())\r
- parentItem = rootItem;\r
- else\r
- parentItem = static_cast<ShoppingTreeItem*>(parent.internalPointer());\r
-\r
- for(int i = 0; i < count; i++)\r
- {\r
- QDomElement newElement;\r
- QDomElement titleElement;\r
- titleElement.setTagName("title");\r
- if(type == category)\r
- {\r
- newElement.setTagName("category");\r
- titleElement.setNodeValue(tr("new category"));\r
- newElement.appendChild(titleElement);\r
- parentItem->node().insertAfter(newElement, parentItem->child(row + i)->node());\r
- }\r
- else if(type == item)\r
- {\r
- newElement.setTagName("item");\r
- titleElement.setNodeValue(tr("new item"));\r
- newElement.appendChild(titleElement);\r
- QDomElement quantityElement;\r
- quantityElement.setTagName("quantity");\r
- newElement.appendChild(quantityElement);\r
- QDomElement storeElement;\r
- storeElement.setTagName("store");\r
- newElement.appendChild(storeElement);\r
- QDomElement checkedElement;\r
- checkedElement.setTagName("checked");\r
- checkedElement.setNodeValue("0");\r
- newElement.appendChild(checkedElement);\r
- QDomElement lastModifiedElement;\r
- lastModifiedElement.setTagName("lastModified");\r
- lastModifiedElement.setNodeValue(QDateTime::currentDateTime().toString("dd/MM/yyyy-hh:mm:ss"));\r
- newElement.appendChild(lastModifiedElement);\r
- parentItem->node().insertAfter(newElement, parentItem->child(row + i)->node());\r
- }\r
- }\r
- emit documentChanged();\r
-}\r
-\r
-/*******************************************************************/\r
-ShoppingTreeModel::removeRows(int row, int count, const QModelIndex &parent)\r
-{\r
- ShoppingTreeItem *parentItem;\r
-\r
- if (!parent.isValid())\r
- parentItem = rootItem;\r
- else\r
- parentItem = static_cast<ShoppingTreeItem*>(parent.internalPointer());\r
-\r
- for(int i = 0; i < count; i++)\r
- {\r
- parentItem->node().removeChild(parentItem->child(i)->node());\r
- }\r
- emit documentChanged();\r
-}\r
-\r
-/*******************************************************************/\r
-ShoppingTreeModel::addCategory(QModelIndex &index)\r
-{\r
- ShoppingTreeItem *selectedItem;\r
-\r
- if (!parent.isValid())\r
- selectedItem = rootItem;\r
- else\r
- selectedItem = static_cast<ShoppingTreeItem*>(parent.internalPointer());\r
-\r
- insertRows(selectedItem->row(), 1, parent(index), category);\r
-}\r
-\r
-/*******************************************************************/\r
-ShoppingTreeModel::addItem(QModelIndex &index)\r
-{\r
- ShoppingTreeItem *selectedItem;\r
-\r
- if (!parent.isValid())\r
- selectedItem = rootItem;\r
- else\r
- selectedItem = static_cast<ShoppingTreeItem*>(parent.internalPointer());\r
-\r
- insertRows(selectedItem->row(), 1, parent(index), item);\r
-}\r
-\r
-/*******************************************************************/\r
-ShoppingTreeModel::removeElement(QModelIndex &index)\r
-{\r
- ShoppingTreeItem *selectedItem;\r
-\r
- if (!parent.isValid())\r
- selectedItem = rootItem;\r
- else\r
- selectedItem = static_cast<ShoppingTreeItem*>(parent.internalPointer());\r
-\r
- removeRows(selectedItem->row(), 1, parent(index));\r
-}\r
-\r
-/*******************************************************************/\r
-ShoppingTreeModel::save()\r
-{\r
- QFile file(m_fileName);\r
- file.remove();\r
- if (!file.open(QIODevice::WriteOnly))\r
- return;\r
- if (!file.write(domDocument.toString(4).toUtf8())) {\r
- file.close();\r
- return;\r
- }\r
- file.close();\r
-}\r