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

---

CheckSizes.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_CheckSizes_h
#define Impala_Core_ImageSet_CheckSizes_h

#include "Core/ImageSet/Reporter.h"
#include "Core/ImageSet/MakeImageSet.h"
#include "Persistency/ImageSetRepository.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class CheckSizes : public Listener
{
public:

    CheckSizes(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        mMinWidth = 100;
        mMaxWidth = 2000;
        mMinHeight = 100;
        mMaxHeight = 2000;
        mWriteCorrected = false;
        mNewSet = 0;
        if (options.GetNrArg() != 7)
        {
            ILOG_ERROR("Need 7 arguments");
        }
        else
        {
            mMinWidth = atol(options.GetArg(2));
            mMaxWidth = atol(options.GetArg(3));
            mMinHeight = atol(options.GetArg(4));
            mMaxHeight = atol(options.GetArg(5));
            mWriteCorrected = atol(options.GetArg(6)) != 0;
        }
        ILOG_DEBUG("range: width: " << mMinWidth << " - " << mMaxWidth
                   << ", height: " << mMinHeight << " - " << mMaxHeight);
    }

    virtual void
    HandleNewWalk(ImageSet* is)
    {
        if (mWriteCorrected)
        {
            String name = FileNameBase(is->GetSetName()) + ".ok.txt";
            mNewSet = new ImageSet(name);
        }
    }

    virtual void
    HandleNewFile(ImageSet* is, int fileId, Array::Array2dVec3UInt8* im)
    {
        String fileName = is->GetAsPath(fileId);
        if (im == 0)
        {
            ILOG_ERROR("No image " << fileName);
            return;
        }
        bool error = false;
        if ((im->CW() < mMinWidth) || (im->CW() > mMaxWidth))
        {
            ILOG_ERROR("width " << im->CW() << " out of range: " << fileName);
            error = true;
        }
        if ((im->CH() < mMinHeight) || (im->CH() > mMaxHeight))
        {
            ILOG_ERROR("height " << im->CH() << " out of range: " << fileName);
            error = true;
        }
        if (error)
        {
            return;
        }
        if (mNewSet)
        {
            int dirId = is->GetDirIdOfFile(fileId);
            mNewSet->AddFile(is->GetFile(fileId), is->GetSection(dirId),
                             is->GetDir(dirId));
        }
    }

    virtual void
    HandleDoneWalk(ImageSet* is)
    {
        if (mNewSet)
        {
            Persistency::ImageSetLocator loc(is->GetLocator(), "");
            Persistency::ImageSetRepository().Add(loc, mNewSet);
        }
    }

private:

    Reporter* mReporter;
    int       mMinWidth;
    int       mMaxWidth;
    int       mMinHeight;
    int       mMaxHeight;
    bool      mWriteCorrected;
    ImageSet* mNewSet;
    
    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(CheckSizes, Core.ImageSet);

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

#endif

---