ObjectIterator.cpp

/***************************************************************************
 *   This file is part of OpenCAL: Open Computer Animation Library         *
 *   I created OpenCAL as my master's thesis Computer Science (multimedia) *
 *   at the tUL university in Diepenbeek, Belgium                          *
 *                                                                         *
 *   Copyright (C) 2003-2004 by Jeroen Dierckx                             *
 *   jeroen.dierckx@student.luc.ac.be                                      *
 *                                                                         *
 ***************************************************************************/

// Includes
#include "ObjectIterator.h"
#include <OpenCAL/System.h>
#include <OpenCAL/Object.h>
using namespace OpenCAL;

using namespace std;


/******************************
* Constructors and destructor *
******************************/

ObjectIterator::ObjectIterator(System *system)
: m_system(system)
{
    reset();
}

ObjectIterator::~ObjectIterator()
{
}


/************************
* Get and set functions *
************************/


/*********************
* Operator functions *
*********************/

ObjectIterator &ObjectIterator::operator++()
{
    if(!m_currentObject) return *this;

    m_currentObject = 0;
    System *system = m_currentSystem;

    if(m_objectIndex < (system->numObjects() - 1))
    {
        m_currentObject = system->getObject(++m_objectIndex);
    }
    else if(system->numSystems() > 0)
    {
        for(unsigned int i = 0; i < system->numSystems(); ++i)
        {
            m_systemStack.push_back(i);
            if(gotoFirstObject(system->getSystem(i)))
                break;
            m_systemStack.pop_back();
        }
    }
    else if(!m_systemStack.empty())
    {
        while(!m_currentObject && !m_systemStack.empty())
        {
            unsigned int index = m_systemStack.back();
            m_systemStack.pop_back();
            m_currentSystem = m_currentSystem->getParent();
            system = m_currentSystem;
            for(unsigned int i = index + 1; i < system->numSystems(); ++i)
            {
                m_systemStack.push_back(i);
                if(gotoFirstObject(system->getSystem(i)))
                    break;
                m_systemStack.pop_back();
            }
        }
    }
    else
    {
        m_currentObject = 0;
    }

    return *this;
}

const ObjectIterator ObjectIterator::operator++(int)
{
    ObjectIterator current = *this;
    operator++();
    return current;
}


/******************
* Other functions *
******************/

void ObjectIterator::reset()
{
    m_systemStack.clear();
    gotoFirstObject(m_system);
}


/**********************
* Protected functions *
**********************/

Object *ObjectIterator::gotoFirstObject(System *system)
{
    // If we find an object, it will always be the first one of a system
    m_objectIndex = 0;

    m_currentObject = 0;
    m_currentSystem = system;

    // If the system has objects in it, return the first one
    if(system->numObjects() > 0)
    {
        m_currentObject = system->getObject(0);
        return m_currentObject;
    }

    // If the system hasn't any objects, look if the child systems have any
    for(unsigned int i = 0; i < system->numSystems(); ++i)
    {
        m_systemStack.push_back(i);
        if(gotoFirstObject(system->getSystem(i)))
            return m_currentObject;
        m_systemStack.pop_back();
    }

    return 0;
}

---