usermodel.cpp

#include "usermodel.h"

#include <domain/user.h>
#include <domain/taskmanagerdata.h>
#include <domain/usertype.h>

using namespace control;

UserModel::UserModel(domain::TaskManagerData* data) : DomainModel(data)
{
    connect(data, SIGNAL(usersChanged()), this, SLOT(emitDataChange()));
}

int UserModel::columnCount(const QModelIndex& parent) const
{
    if (!parent.isValid())
        return LastColumn;

    return 0;
}

int UserModel::rowCount(const QModelIndex& parent) const
{
    if (!parent.isValid())
        return domainData->allUsers().size();

    return 0;
}

QVariant UserModel::headerData(int section, Qt::Orientation orientation,
                               int role) const
{
    if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
    {
        switch (section)
        {
            case Name:          return "Name";
            case Type:          return "Type";

            default:            return QVariant();
        }
    }

    return QVariant(); //keep compiler happy
}

QVariant UserModel::data(const QModelIndex& index, int role) const
{
    const domain::User* user = domainData->allUsers()[index.row()];

    if (role == Qt::DisplayRole)
    {
        switch (index.column())
        {
            case Name:          return user->name();
            case Type:          return user->userType()->name();

            default:            return QVariant();
        }
    }
    else if (role == Qt::ToolTipRole)
    {
        QString toolTip = "%1\nHas %2 tasks assigned";

        return toolTip.arg(user->name())
                      .arg(user->tasks().size());
    }
    else
        return QVariant();
}

const domain::User* UserModel::userForIndex(const QModelIndex& index) const
{
    return domainData->allUsers()[index.row()];
}

QModelIndex UserModel::indexForUser(const domain::User* constUser) const
{
    domain::User* user = domainData->user(constUser);

    if (!domainData->allUsers().contains(user))
        return QModelIndex();

    return index(domainData->allUsers().indexOf(user), 0);
}

---