MLEnrolModel.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 "MLEnrolModel.h"

MLEnrolModel::MLEnrolModel(dbListManager * dataLists, Institute* inst, QObject *parent)
        : RootModel(parent)
{
        rootItem = new InstItem(inst);
        setupData(*dataLists, rootItem);
}

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

QVariant MLEnrolModel::headerData(int section, Qt::Orientation orientation,
                                                                                 int role) const
{
        if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
                switch (section)
                {
                        case 0:
                                return tr("Module");
                        case 1:
                                return tr("Lecturer");
                        case 2:
                                return tr("Hours");
                        case 3:
                                return tr("Size");
                        case 4:
                                return tr("Groups");
                        case 5:
                                return tr("Labs");
                        case 6:
                                return tr("Lab Size");
                        default:
                                return "";
                }
        return QVariant();
}

/******************************************************************************
 * Private Methods
 ******************************************************************************/
void MLEnrolModel::setupData(dbListManager &dataLists, RootItem* parent)
{
        ModLecAlloc * enrolObj;
        MLAllocItem * eiObj;
        ProgItem* parentObj;
        QList<ModLecAlloc *> * enrolList;
        Institute* inst = static_cast<InstItem*>(parent)->getInstitute();
        Programme* prog = NULL;
        int sem =0;
        
        enrolList = dataLists.getMLAllocList(static_cast<InstItem*>(parent)->getItemData());
        for (int Cnt = 0; Cnt < enrolList->size(); ++Cnt)
        {
                enrolObj = enrolList->at(Cnt);
                if (enrolObj->getProg() != prog || enrolObj->getSem() != sem)
                {
                        prog = enrolObj->getProg();
                        sem = enrolObj->getSem();
                        parentObj = new ProgItem(prog,parent,2,sem);
                        parentObj->setColumnOrder(2
                                                        ,ProgItem::COL_ABBREV
                                                        ,ProgItem::COL_SEMESTER
                                                        );
                        parent->appendChild(parentObj);
                }
                //eiObj = new MLEnrolItem(enrolObj, parentObj);
                eiObj = new MLAllocItem(enrolObj, parentObj);
                eiObj->setColumnOrder(7
                                        ,MLAllocItem::COL_MODULE
                                        ,MLAllocItem::COL_LECTURER
                                        ,MLAllocItem::COL_HOURS
                                        ,MLAllocItem::COL_CLASS_SIZE
                                        ,MLAllocItem::COL_GROUPS
                                        ,MLAllocItem::COL_LAB_HOURS
                                        ,MLAllocItem::COL_LAB_GROUPS
                                        );
                parentObj->appendChild(eiObj);
        } // end loop for each allocated object
}

QString MLEnrolModel::dump()
{
        QString msg;
        msg = "MLEnrolModel: \n";
        return msg;
}

Module * MLEnrolModel::getModuleAt(int i)
{
        RootItem * ritem = rootItem->child(i);
        MLAllocItem* MLEitem = dynamic_cast<MLAllocItem*>(ritem);
        return MLEitem->getModule();
}

---