|| || || || || || || ||

---

Walker.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_Walker_h
#define Impala_Core_ImageSet_Walker_h

#include <vector>
#include "Core/ImageSet/ImageSet.h"
#include "Core/ImageSet/Listener.h"
#include "Core/Geometry/FileReadIxRectangle.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class Walker
{
public:

    Walker(ImageSet* imageSet)
    {
        mImageSet = imageSet;
    }

    void
    AddListener(Listener* listener)
    {
        mListeners.push_back(listener);
    }

    int
    NrListeners() const
    {
        return mListeners.size();
    }

    void
    LoadBookmarks(String fileName)
    {
        mIrList.clear();
        Geometry::FileReadIxRectangle(std::back_inserter(mIrList), fileName);
    }

    // Walking part

    void
    DoWalk(int startDir, int numberDirs)
    {
        if (startDir >= mImageSet->NrDirs())
        {
            ILOG_ERROR("DoWalk: startDir out of range");
            return;
        }
        CmdOptions& options = CmdOptions::GetInstance();
        bool useIm = ! options.GetBool("virtualWalk");
        if (numberDirs == -1)
            numberDirs = mImageSet->NrDirs() - startDir;
        if (startDir + numberDirs >= mImageSet->NrDirs())
            numberDirs = mImageSet->NrDirs() - startDir;

        for (size_t l=0 ; l<mListeners.size() ; l++)
        {
            mListeners[l]->HandleNewWalk(mImageSet);
            mListeners[l]->HandleNewWalk(mImageSet, String("Image"));
        }
        for (int dirId=startDir ; dirId<startDir+numberDirs ; dirId++)
        {
            for (size_t l=0 ; l<mListeners.size() ; l++)
                mListeners[l]->HandleNewDir(mImageSet, dirId);

            int startFile = options.GetInt("startFile");
            int numberFiles = options.GetInt("numberFiles");
            int startF = mImageSet->GetFirstFileId(dirId);
            int numberF = mImageSet->GetNrFiles(dirId);
            if (startFile < numberF)
                startF += startFile;
            if (numberFiles == -1)
                numberFiles = numberF;
            if (startFile + numberFiles <= numberF)
                numberF = numberFiles;
            for (int fileId=startF ; fileId<startF+numberF ; fileId++)
            {
                Array::Array2dVec3UInt8* im = 0;
                if (useIm)
                    im = mImageSet->GetImage(fileId);
                for (size_t l=0 ; l<mListeners.size() ; l++)
                    mListeners[l]->HandleNewFile(mImageSet, fileId, im);
                if (im)
                    delete im;
            }
            for (size_t l=0 ; l<mListeners.size() ; l++)
                mListeners[l]->HandleDoneDir(mImageSet, dirId);
        }
        for (size_t l=0 ; l<mListeners.size() ; l++)
            mListeners[l]->HandleDoneWalk(mImageSet);
    }

    void
    DoWalkBookmarks()
    {
        CmdOptions& options = CmdOptions::GetInstance();
        bool useIm = ! options.GetBool("virtualWalk");
        for (size_t l=0 ; l<mListeners.size() ; l++)
        {
            mListeners[l]->HandleNewWalk(mImageSet, mIrList.size());
            mListeners[l]->HandleNewWalk(mImageSet, String("Bookmarks"));
        }
        for (size_t i=0 ; i<mIrList.size() ; i++)
        {
            Geometry::IxRectangle ir = mIrList[i];
            int fileId = mImageSet->GetFileId(ir.mSection, ir.mDir,
                                              ir.mImageName);
            Array::Array2dVec3UInt8* im = 0;
            if (useIm)
                im = mImageSet->GetImage(fileId);
            Geometry::Rectangle r = ir.mRect;
            String s = ir.mAnno;
            for (size_t l=0 ; l<mListeners.size() ; l++)
                mListeners[l]->HandleNewFile(mImageSet, fileId, im, r, s);
            if (im)
                delete im;
        }
        for (size_t l=0 ; l<mListeners.size() ; l++)
            mListeners[l]->HandleDoneWalk(mImageSet, mIrList.size());
    }

protected:

    ImageSet*                          mImageSet;
    std::vector<Geometry::IxRectangle> mIrList;
    std::vector<Listener*>             mListeners;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(Walker, Impala.Core.ImageSet);

} // namespace ImageSet
} // namespace Core
} // namespace Impala

#endif

---