include/vrfobjcore/recSRIter.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2001 MaK Technologies, Inc.
** All rights reserved.
*******************************************************************************/
/*******************************************************************************
** $RCSfile: recSRIter.h,v $ $Revision: 1.7 $ $State: Exp $
*******************************************************************************/



#ifndef DtRecursiveStateRepositoryIterator_H_
#define DtRecursiveStateRepositoryIterator_H_

#include "vrfutil/iteratorBase.h"
#include "vrfobjcore/attachMgr.h"
#include "vrfobjcore/simObjSR.h"

#include "vrfobjcore/vrfobjcoreDefines.h"
class DT_DLL_vrfobjcore DtRecursiveStateRepositoryIterator : public DtIteratorBase<DtSimObjectStateRepository>
{

private:
   DtRecursiveStateRepositoryIterator();

   DtRecursiveStateRepositoryIterator(
      const DtRecursiveStateRepositoryIterator& orig);

   const DtRecursiveStateRepositoryIterator& operator=(
      const DtRecursiveStateRepositoryIterator& orig);

public:
   DtRecursiveStateRepositoryIterator(const DtList& list) :
      DtIteratorBase<DtSimObjectStateRepository>(list), myChildIterator(NULL) {};

   virtual ~DtRecursiveStateRepositoryIterator() { delete myChildIterator; };

   DtSimObjectStateRepository* first(bool reset = true)
      {
         if (reset)
         {
            DtIteratorBase<DtSimObjectStateRepository>::reset();
         }

         return DtIteratorBase<DtSimObjectStateRepository>::first();
      }

   DtSimObjectStateRepository* current() const
      {
         if (myChildIterator)
         {
            DtSimObjectStateRepository* currRep = myChildIterator->current();

            if (currRep)
            {
               return currRep;
            }

            else 
            {
               delete myChildIterator;
               DtRecursiveStateRepositoryIterator* wThis =
                  (DtRecursiveStateRepositoryIterator*)this;
               wThis->myChildIterator = NULL;
            }
         }

         if (myCursor)
         {
            return (DtSimObjectStateRepository*) myCursor->data();
         }

         return NULL;
      }

   DtSimObjectStateRepository* next()
      {
         DtSimObjectStateRepository* currRep;

         if (myChildIterator)
         {
            currRep = myChildIterator->next();

            if (currRep)
            {
               return currRep;
            }

            else
            {
               delete myChildIterator;
               myChildIterator = NULL;
            }
         }

         else 
         {
            DtSimObjectStateRepository* currRep = current();

            if (!currRep) 
            {
               return NULL;
            }

            if (currRep->attachments() && currRep->attachments()->numAttachments())
            {
               myChildIterator = new DtRecursiveStateRepositoryIterator(currRep->attachments()->elements());
               if (myChildIterator)
               {
                  return myChildIterator->first();
               }

               else
               {
                  return NULL;
               }
            }
         }

         if (myCursor)
         {
            myCursor = myCursor->next();
         }

         return current();
      }

   void reset()
      {
         DtIteratorBase<DtSimObjectStateRepository>::reset();

         if (myChildIterator)
         {
            delete myChildIterator;
            myChildIterator = NULL;
         }
      }

protected:
   DtRecursiveStateRepositoryIterator* myChildIterator;
};

#endif