include/vrfutil/recRscrIter.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2004 MaK Technologies, Inc.
** All rights reserved.
*******************************************************************************/
/*******************************************************************************
** $RCSfile: recRscrIter.h,v $ $Revision: 1.3 $ $State: Exp $
Created: 4/5/04 MEM
*******************************************************************************/



#ifndef DtRecursiveResourceIterator_H_
#define DtRecursiveResourceIterator_H_

#include "vrfutil/simRsrc.h"
#include "vrfutil/rsrcList.h"
#include "vrfutil/iteratorBase.h"

#include "vrfutil/vrfutilDefines.h"
class DT_DLL_vrfutil DtRecursiveResourceIterator : public DtIteratorBase<DtSimResource>
{
private:
   DtRecursiveResourceIterator();

   DtRecursiveResourceIterator(
      const DtRecursiveResourceIterator& orig);

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

public:

   DtRecursiveResourceIterator(const DtList& list) :
      DtIteratorBase<DtSimResource>(list), myChildIterator(NULL) {};

   virtual ~DtRecursiveResourceIterator() { delete myChildIterator; };

   DtSimResource* first(bool move=true)
      {
         return DtIteratorBase<DtSimResource>::first(move);
      }

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

            if (currRep)
            {
               return currRep;
            }

            else 
            {
               delete myChildIterator;
               DtRecursiveResourceIterator* pRsrcIter =
                  (DtRecursiveResourceIterator*)this;
               pRsrcIter->myChildIterator = NULL;
            }
         }

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

         return NULL;
      }

   DtSimResource* next(bool move=true, bool recurse=false)
      {
         DtSimResource* currRep;

         if (myChildIterator)
         {
            currRep = myChildIterator->next(move,recurse);

            if (currRep)
            {
               return currRep;
            }

            else
            {
               delete myChildIterator;
               myChildIterator = NULL;
            }
         }

         else if ( recurse ) 
         {
            DtSimResource* currRep = current();

            if (!currRep) 
            {
               return NULL;
            }

            if (currRep->subresources() && currRep->subresources()->count())
            {
               myChildIterator = new DtRecursiveResourceIterator(*(currRep->subresources()));
               if (myChildIterator)
               {
                  return myChildIterator->first();
               }

               else
               {
                  return NULL;
               }
            }
         }

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

         return current();
      }

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

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

protected:
   DtRecursiveResourceIterator* myChildIterator;

};

#endif