--- /dev/null
+/*\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