resourcereservationmodel.cpp

#include "resourcereservationmodel.h"

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

using namespace control;

ResourceReservationModel::ResourceReservationModel (const domain::Resource* resource)
{
    reservations = resource->reservations();
}

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

    return 0;
}

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

    return 0;
}

QVariant ResourceReservationModel::data (const QModelIndex& index, int role) const
{
    const domain::Reservation* reservation = reservationForIndex(index);

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

            default:            return QVariant();
        }
    }
    else
        return QVariant();
}

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

            default:            return QVariant();
        }
    }
    else
        return QVariant();
}

const domain::Reservation* ResourceReservationModel::reservationForIndex
                                                (const QModelIndex& index) const
{
    return reservations[index.row()];
}

---