git.maemo.org Git - movie-schedule/blob - src/ui/theatermodel.cpp

   1 // Copyright 2010 Jochen Becher
   2 //
   3 // This file is part of MovieSchedule.
   4 //
   5 // MovieSchedule is free software: you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation, either version 3 of the License, or
   8 // (at your option) any later version.
   9 //
  10 // MovieSchedule is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14 //
  15 // You should have received a copy of the GNU General Public License
  16 // along with MovieSchedule.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 #include "theatermodel.h"
  19
  20 #include "data/cinemaschedule.h"
  21 #include "data/cinema.h"
  22 #include "utils/assertedlocker.h"
  23
  24 #include <iostream>
  25
  26 TheaterModel::TheaterModel(const CinemaSchedule *cinema_schedule, QObject *parent)
  27     : QAbstractListModel(parent),
  28     _cinema_schedule(cinema_schedule)
  29 {
  30 }
  31
  32 int TheaterModel::rowCount(const QModelIndex &parent) const
  33 {
  34     Q_UNUSED(parent);
  35     AssertedReadLocker locker(_cinema_schedule->GetLock());
  36     return _cinema_schedule->GetCinemas().values().size();
  37 }
  38
  39 QVariant TheaterModel::data(const QModelIndex &index, int role) const
  40 {
  41     AssertedReadLocker locker(_cinema_schedule->GetLock());
  42     if (!(index.isValid() && index.row() >= 0 && index.row() < _cinema_schedule->GetCinemas().values().size())) {
  43         std::cout << "TheaterModel: invalid index" << std::endl;
  44         return QVariant();
  45     }
  46     Cinema *cinema = _cinema_schedule->GetCinemas().values().operator [](index.row());
  47     if (cinema == 0) {
  48         std::cout << "cinema is 0 (row = " << index.row() << ", size = " << _cinema_schedule->GetCinemas().values().size() << ")" << std::endl;
  49         return QVariant();
  50     }
  51     //std::cout << "row = " << index.row() << " role = " << role << std::endl;
  52     switch (role) {
  53     case Qt::DisplayRole:
  54     case TheaterItemRole:
  55         return qVariantFromValue(TheaterItem(*cinema));
  56     case TheaterKeyRole:
  57         return qVariantFromValue(cinema->GetKey());
  58     case SortRole:
  59         return QVariant(cinema->GetName());
  60     }
  61     return QVariant();
  62 }