RoomModel.cpp

/***************************************************************************
 *   Copyright (C) 2006 by Graeme Foster                                   *
 *   email    foster.graeme@gmail.com                                      *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include "RoomModel.h"

RoomModel::RoomModel(dbListManager * dataLists, QObject *parent)
        : RootModel(parent)
{
        QList<QVariant> rootData;
        Institute * rootObj;
        rootObj = new Institute (0,tr("Institute Name"), tr("Abbreviation"));
        rootItem = new InstItem(rootObj);
        setupInstData(*dataLists,rootItem);
}


/******************************************************************************
 * Inherited from RootModel
 ******************************************************************************/
int RoomModel::columnCount(const QModelIndex &) const
{
        return 3;
}

QVariant RoomModel::headerData(int section, Qt::Orientation orientation,
                                                                                 int role) const
{
        if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
                switch (section)
                {
                        case 0:
                                return tr("Name");
                        case 1:
                                return tr("Code");
                        case 2:
                                return tr("Capacity");
                        default:
                                return "";
                }
        return QVariant();
}

/******************************************************************************
 * Private Methods
 ******************************************************************************/
void RoomModel::setupInstData(dbListManager &dataLists, RootItem * parent)
{
        Institute * instObj;
        InstItem   * iiObj;
        QList<Institute * > * instList = dataLists.getInstituteList();
        
        for (int iCnt = 0; iCnt < instList->size(); ++iCnt)
        {
                instObj = instList->at(iCnt);
                iiObj = new InstItem(instObj, parent);
                parent->insertChild(iiObj);
                setupRoomData(dataLists, iiObj);
        } // end loop for each institute
}

void RoomModel::setupRoomData(dbListManager &dataLists, InstItem * parent)
{
        Room * roomObj;
        RoomItem   * riObj;
        QList<Room * > * roomList;
        
        roomList = dataLists.getRoomList(parent->getItemData());
        for (int rCnt = 0; rCnt < roomList->size(); ++rCnt)
        {
                roomObj = roomList->at(rCnt);
                riObj = new RoomItem(roomObj, parent);
                parent->insertChild(riObj);
        } // end loop for each room
}

---