File indexing completed on 2024-05-12 05:09:51

0001 /***************************************************************************
0002     Copyright (C) 2008-2009 Robby Stephenson <robby@periapsis.org>
0003  ***************************************************************************/
0004 
0005 /***************************************************************************
0006  *                                                                         *
0007  *   This program is free software; you can redistribute it and/or         *
0008  *   modify it under the terms of the GNU General Public License as        *
0009  *   published by the Free Software Foundation; either version 2 of        *
0010  *   the License or (at your option) version 3 or any later version        *
0011  *   accepted by the membership of KDE e.V. (or its successor approved     *
0012  *   by the membership of KDE e.V.), which shall act as a proxy            *
0013  *   defined in Section 14 of version 3 of the license.                    *
0014  *                                                                         *
0015  *   This program is distributed in the hope that it will be useful,       *
0016  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
0017  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
0018  *   GNU General Public License for more details.                          *
0019  *                                                                         *
0020  *   You should have received a copy of the GNU General Public License     *
0021  *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
0022  *                                                                         *
0023  ***************************************************************************/
0024 
0025 #ifndef TELLICO_BORROWERMODEL_H
0026 #define TELLICO_BORROWERMODEL_H
0027 
0028 #include "../datavectors.h"
0029 #include "../borrower.h"
0030 
0031 #include <QAbstractItemModel>
0032 
0033 namespace Tellico {
0034 
0035 /**
0036  * @author Robby Stephenson
0037  */
0038 class BorrowerModel : public QAbstractItemModel {
0039 Q_OBJECT
0040 
0041 public:
0042   BorrowerModel(QObject* parent);
0043   virtual ~BorrowerModel();
0044 
0045   virtual int rowCount(const QModelIndex& parent = QModelIndex()) const Q_DECL_OVERRIDE;
0046   virtual int columnCount(const QModelIndex& parent = QModelIndex()) const Q_DECL_OVERRIDE;
0047   virtual QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const Q_DECL_OVERRIDE;
0048   virtual bool setHeaderData(int section, Qt::Orientation orientation, const QVariant& value, int role=Qt::EditRole) Q_DECL_OVERRIDE;
0049   virtual QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const Q_DECL_OVERRIDE;
0050   virtual QModelIndex index(int row, int column=0, const QModelIndex& parent = QModelIndex()) const Q_DECL_OVERRIDE;
0051   virtual QModelIndex parent(const QModelIndex& index) const Q_DECL_OVERRIDE;
0052 
0053   void clear();
0054   void addBorrowers(const Data::BorrowerList& borrowers);
0055   QModelIndex addBorrower(Data::BorrowerPtr borrower);
0056   QModelIndex modifyBorrower(Data::BorrowerPtr borrower);
0057   void removeBorrower(Data::BorrowerPtr borrower);
0058 
0059   Data::BorrowerPtr borrower(const QModelIndex& index) const;
0060   Data::EntryPtr entry(const QModelIndex& index) const;
0061   Data::LoanPtr loan(const QModelIndex& index) const;
0062 
0063 private:
0064   Data::BorrowerList m_borrowers;
0065   QString m_header;
0066   class Node;
0067   Node* m_rootNode;
0068 };
0069 
0070 } // end namespace
0071 #endif