taskmodel.cpp

#include "taskmodel.h"

#include <domain/task.h>
#include <domain/user.h>
#include <domain/project.h>
#include <domain/taskstates.h>
#include <domain/taskmanagerdata.h>
#include <domain/schedule.h>
#include <domain/tasktype.h>
#include <tools/functions.h>
#include <QStringList>

using namespace control;

TaskModel::TaskModel(domain::TaskManagerData* data,
               QList<const domain::Task*> shownTasks) : DomainModel(data),
                                                        taskList(shownTasks),
                                                        showAll(false)
{
    Q_FOREACH (const domain::Task* task, taskList)
    {
        connect(data->task(task), SIGNAL(dataChanged()),
                this, SLOT(emitDataChange()));
    }
}

TaskModel::TaskModel(domain::TaskManagerData* data) : DomainModel(data),
                                                        showAll(true)
{
    connect(data, SIGNAL(tasksChanged()), this, SLOT(emitDataChange()));
}

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

    return 0;
}

int TaskModel::rowCount(const QModelIndex& parent) const
{
    if (!parent.isValid())
        return tasks().size();

    return 0;
}

QVariant TaskModel::headerData(int section, Qt::Orientation orientation,
                               int role) const
{
    if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
    {
        switch (section)
        {
            case Id:            return "ID";
            case Type:          return "Type";
            case Status:        return "Status";
            case User:          return "User";
            case Project:       return "Project";
            case StartTime:     return "Start time";
            case Deadline:      return "Deadline";
            case Duration:      return "Duration";
            case Fields:        return "Fields";

            default:            return QVariant();
        }
    }

    return QVariant();
}

QVariant TaskModel::data(const QModelIndex& index, int role) const
{
    const domain::Task* task = taskForIndex(index);

    if (role == Qt::DisplayRole)
    {
        switch (index.column())
        {
            case Id:            return task->id();
            case Type:          return task->taskType()->name();
            case Status:        return task->stateString();
            case User:          return task->user()->name();
            case Project:       return projectString(task);
            case StartTime:     return task->schedule().startDate();
            case Deadline:      return task->schedule().dueDate();
            case Duration:      return task->schedule().duration().toString();
            case Fields:        return fieldsString(task).join(", ");

            default:            return QVariant();
        }
    }
    else if (role == Qt::ToolTipRole)
    {
        QString toolTip("Task %1\nHas %2 sub task(s)\nFields:\n - %3");
        return toolTip.arg(task->id()).arg(task->subTasks().size())
                      .arg(fieldsString(task).join("\n - "));
    }
    else
        return QVariant();
}

const domain::Task* TaskModel::taskForIndex(const QModelIndex& index) const
{
    return tasks()[index.row()];
}

QModelIndex TaskModel::indexForTask(const domain::Task* task) const
{
    if (!tasks().contains(task))
        return QModelIndex();

    return index(tasks().indexOf(task), 0);
}

QList<const domain::Task*> TaskModel::tasks() const
{
    return showAll ? tools::makeConstList(domainData->tasks()) : taskList;
}

QString TaskModel::projectString(const domain::Task* task) const
{
    return task->project() == 0 ? "/" : task->project()->description();
}

QStringList TaskModel::fieldsString(const domain::Task* task) const
{
    QStringList ret;

    Q_FOREACH (QString id, task->fields().keys())
    {
        QString name = task->taskType()->fieldById(id).name;
        QVariant value = task->field(id);
        ret << QString("%1: %2").arg(name).arg(value.toString());
    }

    if (ret.isEmpty())
        ret << "/";

    return ret;
}

---