From e710b1ff68c29f5a2331cc36b5fdf6d67f4b1c63 Mon Sep 17 00:00:00 2001 From: Arno Date: Sun, 6 Jun 2010 16:26:58 +0200 Subject: Finished SmTreeModel, started SmUberModel kinda finished the treemodel. Don't really know if it's complete yet. More programming will show :) --- shemov.pro | 6 ++- smtreeitem.cpp | 2 +- smtreeitem.h | 2 +- smtreemodel.cpp | 144 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ smtreemodel.h | 22 +++++++++ smubermodel.cpp | 14 ++++++ smubermodel.h | 30 ++++++++++++ 7 files changed, 216 insertions(+), 4 deletions(-) create mode 100644 smubermodel.cpp create mode 100644 smubermodel.h diff --git a/shemov.pro b/shemov.pro index 47d985f..937117b 100644 --- a/shemov.pro +++ b/shemov.pro @@ -47,7 +47,8 @@ SOURCES = main.cpp \ archiveitemcoveredit.cpp \ archiveitemeditdialog.cpp \ smtreeitem.cpp \ - smtreemodel.cpp + smtreemodel.cpp \ + smubermodel.cpp HEADERS = listitem.h \ listmodel.h \ movieitem.h \ @@ -90,6 +91,7 @@ HEADERS = listitem.h \ archiveitemcoveredit.h \ archiveitemeditdialog.h \ smtreeitem.h \ - smtreemodel.h + smtreemodel.h \ + smubermodel.h LIBS += -lmagic RESOURCES = shemov.qrc diff --git a/smtreeitem.cpp b/smtreeitem.cpp index 43998c3..950926e 100644 --- a/smtreeitem.cpp +++ b/smtreeitem.cpp @@ -50,7 +50,7 @@ QVariant SmTreeItem::data(int column) const{ return mData.at(column); } -void SmTreeItem::setData(int column, QVariant &data){ +void SmTreeItem::setData(int column, const QVariant &data){ mData[column] = data; } diff --git a/smtreeitem.h b/smtreeitem.h index 4696edf..807e3d7 100644 --- a/smtreeitem.h +++ b/smtreeitem.h @@ -23,7 +23,7 @@ class SmTreeItem { int row() const; SmTreeItem *parent(); QVariant data(int column) const; - void setData(int column, QVariant &data); + void setData(int column, const QVariant &data); bool insertChild(int where, SmTreeItem *child); bool removeChild(int where); diff --git a/smtreemodel.cpp b/smtreemodel.cpp index 60cc977..2c7a2d6 100644 --- a/smtreemodel.cpp +++ b/smtreemodel.cpp @@ -16,3 +16,147 @@ SmTreeModel::SmTreeModel(const QStringList &headers, QObject *parent) : QAbstrac SmTreeModel::~SmTreeModel(){ delete mRootItem; } + +int SmTreeModel::rowCount(const QModelIndex &parent) const{ + SmTreeItem *parentItem = itemAt(parent); + return parentItem->childCount(); +} + +int SmTreeModel::columnCount(const QModelIndex &parent) const{ + Q_UNUSED(parent); + return mRootItem->columnCount(); +} + +QModelIndex SmTreeModel::index(int row, int column, const QModelIndex &parent) const{ + if(parent.isValid() && (parent.column() != 0)){ + return QModelIndex(); + } + SmTreeItem *parentItem = itemAt(parent); + SmTreeItem *childItem = parentItem->child(row); + if(childItem){ + return createIndex(row, column, childItem); + } + return QModelIndex(); +} + +Qt::ItemFlags SmTreeModel::flags(const QModelIndex &index) const{ + if(!index.isValid()){ + return 0; + } + return Qt::ItemIsEditable | Qt::ItemIsEnabled | Qt::ItemIsSelectable; +} + +QVariant SmTreeModel::headerData(int section, Qt::Orientation orientation, int role) const{ + if((orientation == Qt::Horizontal) && (role == Qt::DisplayRole)){ + return mRootItem->data(section); + } + return QVariant(); +} + +bool SmTreeModel::setHeaderData(int section, Qt::Orientation orientation, const QVariant &value, int role){ + if((orientation == Qt::Horizontal) && (role == Qt::DisplayRole)){ + mRootItem->setData(section, value); + emit headerDataChanged(orientation, section, section); + return true; + } + return false; +} + +QVariant SmTreeModel::data(const QModelIndex &index, int role) const{ + if(!index.isValid()){ + return QVariant(); + } + SmTreeItem *item = itemAt(index); + if(role == Qt::DisplayRole){ + return item->data(index.column()); + } + if(role == Qt::EditRole){ + return item->data(index.column()); + } + return QVariant(); +} + +bool SmTreeModel::setData(const QModelIndex &index, const QVariant &value, int role){ + if(role != Qt::EditRole){ + return false; + } + SmTreeItem *item = itemAt(index); + item->setData(index.column(), value); + emit dataChanged(index, index); + return true; +} + +bool SmTreeModel::setRoot(SmTreeItem *rootItem){ + if(mRootItem){ + beginResetModel(); + delete mRootItem; + mRootItem = rootItem; + endResetModel(); + return true; + } + return false; +} + +bool SmTreeModel::insertRows(int row, int count, const QModelIndex &parent){ + SmTreeItem *parentItem = itemAt(parent); + bool retval; + + if(row > parentItem->childCount()){ + return false; + } + + beginInsertRows(parent, row, row + count - 1); + for(int i = row; i < row + count; ++i){ + SmTreeItem *newItem = new SmTreeItem(mRootItem->columnCount()); + retval = parentItem->insertChild(i, newItem); + } + endInsertRows(); + return retval; +} + +bool SmTreeModel::removeRows(int row, int count, const QModelIndex &parent){ + SmTreeItem *parentItem = itemAt(parent); + bool retval; + + if(row > parentItem->childCount()){ + return false; + } + + beginRemoveRows(parent, row, row + count - 1); + for(int i = row + count; i > row; --i){ + retval = parentItem->removeChild(i); + + } + endRemoveRows(); + return retval; +} + +bool SmTreeModel::addRow(const QList &data, const QModelIndex &parent){ + if(data.count() != mRootItem->columnCount()){ + return false; + } + + SmTreeItem *parentItem = itemAt(parent); + + if(insertRows(parentItem->childCount(), 1, parent)){ + SmTreeItem *child = parentItem->child(parentItem->childCount()); + delete child; + SmTreeItem *newChild = new SmTreeItem(data, parentItem); + child = newChild; + QModelIndex start = index(parentItem->childCount() - 1, 0, parent); + QModelIndex end = index(parentItem->childCount() - 1, parentItem->columnCount() - 1, parent); + emit dataChanged(start, end); + return true; + } + return false; +} + +SmTreeItem *SmTreeModel::itemAt(const QModelIndex &index) const{ + if(index.isValid()){ + SmTreeItem *item = static_cast(index.internalPointer()); + if(item){ + return item; + } + } + return mRootItem; +} diff --git a/smtreemodel.h b/smtreemodel.h index ede4b39..46f852f 100644 --- a/smtreemodel.h +++ b/smtreemodel.h @@ -19,7 +19,29 @@ class SmTreeModel : public QAbstractItemModel { explicit SmTreeModel(const QStringList &headers, QObject *parent = 0); ~SmTreeModel(); + // counts + int rowCount(const QModelIndex &parent) const; + int columnCount(const QModelIndex &parent) const; + + // index and flags + QModelIndex index(int row, int column, const QModelIndex &parent) const; + Qt::ItemFlags flags(const QModelIndex &index) const; + + // headers + data + QVariant headerData(int section, Qt::Orientation orientation, int role) const; + bool setHeaderData(int section, Qt::Orientation orientation, const QVariant &value, int role); + QVariant data(const QModelIndex &index, int role) const; + bool setData(const QModelIndex &index, const QVariant &value, int role); + bool setRoot(SmTreeItem *rootItem); + + // row manipulation + bool insertRows(int row, int count, const QModelIndex &parent); + bool removeRows(int row, int count, const QModelIndex &parent); + bool addRow(const QList &data, const QModelIndex &parent); + + private: + SmTreeItem *itemAt(const QModelIndex &index) const; QStringList mHeaders; SmTreeItem *mRootItem; }; diff --git a/smubermodel.cpp b/smubermodel.cpp new file mode 100644 index 0000000..1134e18 --- /dev/null +++ b/smubermodel.cpp @@ -0,0 +1,14 @@ +/* + This program is free software; you can redistribute it and/or + modify it under the terms of the GNU General Public License + as published by the Free Software Foundation; either version + 2 of the License, or (at your option) any later version. +*/ + +#include "smubermodel.h" +#include "smtreemodel.h" +#include "actormodel.h" + +SmUberModel::SmUberModel(QObject *parent) : QObject(parent), mSeriesModel(0), mFileModel(0), mActorModel(0){ + +} diff --git a/smubermodel.h b/smubermodel.h new file mode 100644 index 0000000..edfa008 --- /dev/null +++ b/smubermodel.h @@ -0,0 +1,30 @@ +/* + This program is free software; you can redistribute it and/or + modify it under the terms of the GNU General Public License + as published by the Free Software Foundation; either version + 2 of the License, or (at your option) any later version. +*/ + +#ifndef SMUBERMODEL_H +#define SMUBERMODEL_H + +#include + +class SmTreeModel; +class ActorModel; + +class SmUberModel : public QObject { + Q_OBJECT + public: + explicit SmUberModel(QObject *parent = 0); + ~SmUberModel(); + + private: + void populateSeriesmodel(); + SmTreeModel *mSeriesModel; + SmTreeModel *mFileModel; + ActorModel *mActorModel; + +}; + +#endif -- cgit v1.2.3-70-g09d2