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

---

IndexConcepts.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_IndexConcepts_h
#define Impala_Core_ImageSet_IndexConcepts_h

#include "Core/Table/SimilarityTableSet.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;

    IndexConcepts(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        if (options.GetNrArg() < 5)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        mConceptFileName = options.GetArg(2);
        mModel = options.GetArg(3);
        mFeatureDef = FeatureDefinition(options.GetArg(4));
        mSimSet = 0;
    }

    virtual void
    HandleNewWalk(ImageSet* is, String walkType)
    {
        mWalkType = walkType;
    }

    virtual void
    HandleDoneDir(ImageSet* is, int dirId)
    {
        SimilarityTableSet* simSet = SimilarityTableSet::MakeFromFile
            (is, mWalkType, mConceptFileName, mModel, mFeatureDef.AsString(), dirId);

        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);
        mSimSet->Save(is, mConceptFileName, mModel, mFeatureDef.AsString(),
                      true);
        delete mSimSet;
    }

private:

    Reporter*           mReporter;
    String              mConceptFileName;
    String              mModel;
    FeatureDefinition   mFeatureDef;
    SimilarityTableSet* mSimSet;
    String              mWalkType;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(IndexConcepts, Impala.Core.ImageSet);

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

#endif

---