reservationmodel.cpp

#include "reservationmodel.h"

#include <domain/taskmanagerdata.h>
#include <domain/reservation.h>
#include <domain/duration.h>
#include <domain/resource.h>
#include <domain/user.h>

#include <QDateTime>

using namespace control;

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

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

    return 0;
}

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

    return 0;
}

QVariant ReservationModel::headerData(int section, Qt::Orientation orientation,
                               int role) const
{
    if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
    {
        switch (section)
        {
            case Resource:      return "Resource";
            case Task:          return "Task";
            case Time:          return "Time";
            case Duration:      return "Duration";

            default:            return QVariant();
        }
    }

    return QVariant(); //keep compiler happy
}

QVariant ReservationModel::data(const QModelIndex& index, int role) const
{
    const domain::Reservation* reservation =
                                domainData->reservations()[index.row()];

    if (role == Qt::DisplayRole)
    {
        switch (index.column())
        {
            case Resource:      return reservation->resource()->description();
            case Task:          return (int)reservation->task();
            case Time:          return reservation->time();
            case Duration:      return reservation->duration().toString();

            default:            return QVariant();
        }
    }
    else if (role == Qt::ToolTipRole)
    {
        QString toolTip("Reservation of resource %1 for task %2\n"
                        "Starting %3 for a period of %4");

        return toolTip.arg(reservation->resource()->description())
                      .arg(reservation->task()->id())
                      .arg(reservation->time().toString())
                      .arg(reservation->duration().toString());
    }
    else
        return QVariant();
}

---