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

---

IndexFeatures.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_IndexFeatures_h
#define Impala_Core_ImageSet_IndexFeatures_h

#include "Persistency/FeatureTableRepository.h"
#include "Core/Feature/FeatureTable.h"
#include "Core/ImageSet/Reporter.h"
#include "Core/Table/Append.h"
#include "Core/Table/Write.h"
#include "Core/Feature/DumpFeatureTableHistogram.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{


class IndexFeatures : public Listener
{
public:
    typedef Feature::FeatureTable FeatureTable;

    IndexFeatures(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        if (Link::Mpi::NrProcs() == 8) // quick hack for sift
        {
            String feat = options.GetArg(2);
            int nr = Link::Mpi::MyId();
            String sub = (nr == 0) ? "" : "_sub_" + MakeString(nr);
            mDef = Feature::FeatureDefinition(feat + sub);            
            ILOG_INFO_NODE("indexing feature " << mDef.AsString());
        }
        else
        {
            mDef = Feature::FeatureDefinition(options.GetArg(2));
        }
        mTable = 0;
    }

    virtual void
    HandleDoneDir(ImageSet* is, int dirId)
    {
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        String fName = is->GetFilePathFeatureData(mDef, dirId, false, -1, 
                                                  false, false);
        if (fName.empty())
        {
            return;
        }
        FeatureTable tmp(mDef);
        Table::Read(&tmp, fName, is->GetDatabase());
        if (mTable == 0)
        {
            mTable = new FeatureTable(mDef, 12000, tmp.GetFeatureVectorLength());
        }
        Table::Append(mTable, &tmp);
#else // REPOSITORY_USED
        Persistency::FeatureLocator loc(is->GetLocator(), false, false, "",
                                        mDef.AsString(),
                                        is->GetContainerDir(dirId));
        FeatureTable* tmp = Persistency::FeatureTableRepository().Get(loc);
        if (mTable == 0)
        {
            mTable = new FeatureTable(mDef, 12000,
                                      tmp->GetFeatureVectorLength());
        }
        Table::Append(mTable, tmp);
        delete tmp;
#endif // REPOSITORY_USED
    }

    virtual void
    HandleDoneWalk(ImageSet* is)
    {
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        String fName = is->GetFilePathFeatureIndex(mDef, "", true, false);
        if (!fName.empty())
            Table::Write(mTable, fName, is->GetDatabase(), true);
#else // REPOSITORY_USED
        Persistency::FeatureLocator loc(is->GetLocator(), false, true, "",
                                        mDef.AsString(), "");
        Persistency::FeatureTableRepository().Add(loc, mTable);
#endif // REPOSITORY_USED
    }

private:

    Reporter*                  mReporter;
    Feature::FeatureDefinition mDef;
    Feature::FeatureTable*     mTable;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(IndexFeatures, Impala.Core.ImageSet);

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

#endif

---