projectmodel.cpp

#include "projectmodel.h"

#include <domain/taskmanagerdata.h>
#include <domain/project.h>

using namespace control;

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

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

    return 0;
}

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

    return 0;
}

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

            default:            return QVariant();
        }
    }

    return QVariant(); //keep compiler happy
}

QVariant ProjectModel::data(const QModelIndex& index, int role) const
{
    const domain::Project* project = projectForIndex(index);

    if (role == Qt::DisplayRole)
    {
        switch (index.column())
        {
            case Description:   return project->description();

            default:            return QVariant();
        }
    }
    else if (role == Qt::ToolTipRole)
    {
        QString toolTip = project->description() + "\n";
        toolTip += "Has " + QString::number(project->tasks().size())
                            + " tasks\n";
        return toolTip;
    }
    else
        return QVariant();
}

const domain::Project* ProjectModel::projectForIndex(const QModelIndex& index) const
{
    return domainData->projects()[index.row()];
}

QModelIndex ProjectModel::indexForProject(domain::Project* project) const
{
    if (!domainData->projects().contains(project))
        return QModelIndex();

    return index(domainData->projects().indexOf(project),0);
}

---