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

---

IndexConcepts.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_IndexConcepts_h
#define Impala_Core_ImageSet_IndexConcepts_h

#include "Persistency/SimilarityTableSetRepository.h"
#include "Core/ImageSet/Reporter.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class IndexConcepts : public Listener
{
public:
    typedef Feature::FeatureDefinition FeatureDefinition;
    typedef Table::SimilarityTableSet SimilarityTableSet;
    typedef Persistency::SimilarityTableSetLocator SimilarityTableSetLocator;
    typedef Persistency::SimilarityTableSetRepository SimilarityTableSetRepository;

    IndexConcepts(Reporter* reporter, CmdOptions& options, ImageSet* is)
    {
        if (options.GetNrArg() < 5)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        mLoc = SimilarityTableSetLocator
            (is->GetLocator(), false, "", options.GetArg(2), options.GetArg(3),
             options.GetArg(4), "");
        mSimSet = 0;
    }

    virtual void
    HandleNewWalk(ImageSet* is, String walkType)
    {
        mLoc.SetWalkType(walkType);
    }

    virtual void
    HandleDoneDir(ImageSet* is, int dirId)
    {
        mLoc.SetContainer(is->GetContainer(dirId));
        SimilarityTableSet* simSet = SimilarityTableSetRepository().Get(mLoc);
        if (mSimSet == 0)
        {
            std::vector<String> names = simSet->GetNames();
            mSimSet = new SimilarityTableSet(names, 1000000);
        }
        mSimSet->Append(simSet);
        delete simSet;
    }

    virtual void
    HandleDoneWalk(ImageSet* is)
    {
        mSimSet->ComputeRanks(true);
        mLoc.SetInIndex(true);
        SimilarityTableSetRepository().Add(mLoc, mSimSet);
        delete mSimSet;
    }

private:

    SimilarityTableSetLocator mLoc;
    SimilarityTableSet*       mSimSet;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(IndexConcepts, Impala.Core.ImageSet);

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

#endif

---