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

---

IndexFeatures.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_IndexFeatures_h
#define Impala_Core_VideoSet_IndexFeatures_h

#include "Persistency/FeatureTableRepository.h"
#include "Core/Feature/FeatureTable.h"
#include "Core/VideoSet/Reporter.h"
#include "Core/VideoSet/Keyframes.h"
#include "Core/Table/Select.h"
#include "Core/Table/Write.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


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;
        mKeyframes = 0;
        mKeyframeMask = 0;
        mWalkType = "unknown";
    }

    virtual void
    HandleNewWalk(VideoSet* vs, String walkType)
    {
        mWalkType = walkType;
    }

    virtual void
    HandleNewWalk(VideoSet* vs, Keyframes* keyframes, bool* keyframeMask)
    {
        mKeyframes = keyframes;
        mKeyframeMask = keyframeMask;
    }

    virtual void
    HandleDoneFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        String fName = vs->GetFilePathFeatureData(mWalkType, mDef,
                                                  fileId, false, -1, false,
                                                  false);
        if (fName.empty())
        {
            return;
        }
        Feature::FeatureTable tmp(mDef);
        Table::Read(&tmp, fName, vs->GetDatabase());
        if (mTable == 0)
        {
            mTable = new Feature::FeatureTable(mDef, 0,
                                               tmp.GetFeatureVectorLength());
        }
#else // REPOSITORY_USED
        Persistency::FeatureLocator loc(vs->GetLocator(), false, false,
                                        mWalkType, mDef.AsString(),
                                        vs->GetContainerFile(fileId));
        FeatureTable* tmp = Persistency::FeatureTableRepository().Get(loc);
        if (mTable == 0)
        {
            mTable = new FeatureTable(mDef, 12000,
                                      tmp->GetFeatureVectorLength());
        }
#endif // REPOSITORY_USED
        bool* mask = mKeyframeMask;
        if (mask)
            mask = mKeyframeMask + mKeyframes->GetFirstKeyframeVideo(fileId);
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        Table::Select(mTable, &tmp, mask, false);
#else // REPOSITORY_USED
        Table::Select(mTable, tmp, mask, false);
        delete tmp;
#endif // REPOSITORY_USED
    }

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

private:

    Reporter*                  mReporter;
    Feature::FeatureDefinition mDef;
    Feature::FeatureTable*     mTable;
    Keyframes*                 mKeyframes;
    bool*                      mKeyframeMask;
    String                     mWalkType;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(IndexFeatures, Impala.Core.VideoSet);

} // namespace VideoSet
} // namespace Core
} // namespace Impala

#endif

---