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

---

ExportFeatures.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_ExportFeatures_h
#define Impala_Core_VideoSet_ExportFeatures_h

#include "Persistency/FeatureTableRepository.h"
#include "Core/Feature/FeatureTable.h"
#include "Core/VideoSet/Reporter.h"
#include "Core/VideoSet/Segmentation.h"
#include "Core/VideoSet/Mpeg7DocWrite.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class ExportFeatures : public Listener
{
public:
    typedef Feature::FeatureDefinition FeatureDefinition;

    ExportFeatures(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        if (options.GetNrArg() < 3)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        mFractions = atol(options.GetArg(2));
        if ((mFractions != 25) && (mFractions != 30000))
            ILOG_WARN("Never tried fractions = " << mFractions);
        mFeatureDef = FeatureDefinition(options.GetArg(3));
        mSegmentation = 0;
        mWalkType = "unknown";
    }

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

    virtual void
    HandleNewWalk(VideoSet* vs, Segmentation* segmentation)
    {
        mSegmentation = segmentation;
    }

    virtual void
    HandleDoneFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        if ((mSegmentation == 0))
        {
            ILOG_ERROR("Need segmentation");
            return;
        }
        Persistency::FeatureLocator loc(vs->GetLocator(), false, false,
                                        mWalkType, mFeatureDef.AsString(),
                                        vs->GetContainerFile(fileId));
        typedef Feature::FeatureTable FeatureTable;
        FeatureTable* table = Persistency::FeatureTableRepository().Get(loc);
        if (table == 0)
        {
            ILOG_ERROR("Unable to find " << mFeatureDef.AsString());
            return;
        }
        String fileName = vs->GetFile(fileId);
        String id = fileName;
        int nrFrames = mSegmentation->GetNrFramesVideo(fileId);
        int frameRateNum = (src) ? src->FrameRateNum() : 0;
        int frameRateDen = (src) ? src->FrameRateDen() : 0;            
        Mpeg7Doc mp7(id, fileName, nrFrames, mFractions, false, "", "",
                     frameRateNum, frameRateDen);
        for (int i=0 ; i<table->Size() ; i++)
        {
            Quid quid = table->Get1(i);
            if (QuidClass(quid) != QUID_CLASS_FRAME)
                ILOG_ERROR("Quid is not a frame");
            int frameNr = QuidId(quid);
            String name = "frame_" + MakeString(frameNr);
            Feature::FeatureTable::VectorReal64 v = table->Get2(i);
            std::vector<double> vec;
            for (int j=0 ; j<v.Size() ; j++)
                vec.push_back(v[j]);
            mp7.AddFeature(mFeatureDef.GetName(), name, frameNr, frameNr, vec);
        }
        delete table;
        Core::VideoSet::Mpeg7DocWrite(&mp7, vs, fileId);
    }

private:

    Reporter*           mReporter;
    int                 mFractions;
    FeatureDefinition   mFeatureDef;
    Segmentation*       mSegmentation;
    String              mWalkType;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(ExportFeatures, Impala.Core.VideoSet);

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

#endif

---