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

---

ProtoSimilarityEval.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_ProtoSimilarityEval_h
#define Impala_Core_ImageSet_ProtoSimilarityEval_h

#include "Persistency/FeatureTableRepository.h"
#include "Core/ImageSet/Reporter.h"
#include "Core/Feature/Computor.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class ProtoSimilarityEval : public Listener
{
public:

    ProtoSimilarityEval(Feature::Computor* computor, Reporter* reporter,
                        CmdOptions& options)
    {
        mReporter = reporter;
        mComputor = computor;
        if (options.GetNrArg() < 5)
            ILOG_ERROR("Need more arguments");
        String protoSet = options.GetArg(2);
        if (protoSet == "nil")
            protoSet = "";
        String maskSet = options.GetArg(3);
        if (maskSet == "nil")
            maskSet = "";
        int maxNr = atol(options.GetArg(4));
        mComputor->ReadPrototypes(protoSet, maskSet, maxNr);
    }

    virtual void
    HandleNewFile(ImageSet* is, int fileId, Array::Array2dVec3UInt8* im)
    {
        Quid quid = is->GetQuidImage(fileId);
        mComputor->ComputeFeatures(im, quid);
    }

    virtual void
    HandleDoneDir(ImageSet* is, int dirId)
    {
        bool binary = true;
        Feature::FeatureTableSet* tableSet = mComputor->GetFeatureTableSet();
        Persistency::FeatureLocator loc(is->GetLocator(), false, false, "",
                                        "empty", is->GetContainerDir(dirId));
        for (int i=0 ; i<tableSet->Size() ; i++)
        {
            Feature::FeatureDefinition def = tableSet->GetFeatureDefinition(i);
            def = mComputor->ExtendFeatureDefinition(def);
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
            String fName = is->GetFilePathFeatureData(def, dirId, false,
                                                      0, true, false);
            if (!fName.empty())
            {
                Table::Write(tableSet->GetTable(i), fName, is->GetDatabase(),
                             binary);
            }
#else // REPOSITORY_USED
            loc.SetFeatureDef(def);
            Persistency::FeatureTableRepository().Add(loc,
                                                      tableSet->GetTable(i));
#endif // REPOSITORY_USED
            tableSet->GetTable(i)->SetSize(0);
        }
        String mask = mComputor->GetMaskSetBase();
        if (!mask.empty())
        {
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
            String fName = is->GetFilePathFeatureData(mask + ".txt", dirId,
                                                      false, 0, true, false);
            if (!fName.empty())
                mComputor->WriteFeatureDefinitions(fName, is->GetDatabase());
#else // REPOSITORY_USED
            loc.SetFeatureString(mask + ".txt");
            mComputor->WriteFeatureDefinitions(loc);
#endif // REPOSITORY_USED
        }
    }

private:

    Reporter*          mReporter;
    Feature::Computor* mComputor;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(ProtoSimilarityEval, Impala.Core.ImageSet);

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

#endif

---