StaffModel.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 <QtGui>

#include "Institute.h"
#include "InstItem.h"
#include "ProgItem.h"

#include "StaffModel.h"

StaffModel::StaffModel(dbListManager * dataLists
                      ,DataClass::ClassName cn
                      ,RootItem * root
                      ,QObject *parent
                      )
        : RootModel(parent)
{
        QList<QVariant> rootData;
        Institute * rootObj;
        if (root == 0)
        {
                rootObj = new Institute (0,tr("First Name"), tr("Last Name"));
                rootItem = new InstItem(rootObj);
        }
        else
                rootItem = root;
        setupData(*dataLists,cn,root);
}


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


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


/******************************************************************************
 * Private Methods
 ******************************************************************************/
void StaffModel::setupData(dbListManager &dataLists
                          ,DataClass::ClassName cn
                          ,RootItem * parent
                          )
{
        if (parent == 0)
                setupInstData(dataLists, rootItem);
        else if (parent->rtti() == DataClass::INSTITUTE_CLASS)
        {
                if (cn == DataClass::INSTITUTE_CLASS)
                        setupFacultyData(dataLists, static_cast<InstItem*>(parent));
                else if (cn == DataClass::DEPARTMENT_CLASS)
                        setupInstStaffData(dataLists, static_cast<InstItem*>(parent));
        }
        else if (parent->rtti() == DataClass::FACULTY_CLASS)
                setupDeptData(dataLists, static_cast<FacItem*>(parent));
        else if  (parent->rtti() == DataClass::DEPARTMENT_CLASS)
                setupStaffData(dataLists, static_cast<DeptItem*>(parent));
}

void StaffModel::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);
                setupFacultyData(dataLists, iiObj);
        } // end loop for each institute
}

void StaffModel::setupFacultyData(dbListManager &dataLists, InstItem * parent)
{
        Faculty * facultyObj;
        FacItem   * fiObj;
        QList<Faculty * > * facultyList = dataLists.getFacultyList(parent->getItemData());
        
        for (int fCnt = 0; fCnt < facultyList->size(); ++fCnt)
        {
                facultyObj = facultyList->at(fCnt);
                fiObj = new FacItem(facultyObj, parent);
                parent->insertChild(fiObj);
                setupDeptData(dataLists, fiObj);
        } // end loop for each faculty
}

void StaffModel::setupDeptData(dbListManager &dataLists, FacItem * parent)
{
        Department * deptObj;
        DeptItem   * diObj;
        QList<Department * > * deptList = dataLists.getDepartmentList(parent->getItemData());
        
        for (int dCnt = 0; dCnt < deptList->size(); ++dCnt)
        {
                deptObj = deptList->at(dCnt);
                diObj = new DeptItem(deptObj, parent);
                parent->insertChild(diObj);
                setupStaffData(dataLists, diObj);
        } // end loop for each department
}

void StaffModel::setupStaffData(dbListManager &dataLists, DeptItem * parent)
{
        Staff * staffObj;
        StaffItem   * siObj;
        QList<Staff * > * staffList = dataLists.getStaffList(parent->getItemData());
        
        for (int dCnt = 0; dCnt < staffList->size(); ++dCnt)
        {
                staffObj = staffList->at(dCnt);
                siObj = new StaffItem(staffObj, parent);
                parent->insertChild(siObj);
        } // end loop for each staff
}

void StaffModel::setupInstStaffData(dbListManager &dataLists, InstItem * parent)
{
        Staff * staffObj;
        StaffItem   * siObj;
        QList<Staff * > * staffList = dataLists.getStaffList(parent->getItemData());
        
        for (int dCnt = 0; dCnt < staffList->size(); ++dCnt)
        {
                staffObj = staffList->at(dCnt);
                siObj = new StaffItem(staffObj, parent);
                parent->insertChild(siObj);
        } // end loop for each staff
}

---