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

---

ExportConcepts.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_ExportConcepts_h
#define Impala_Core_VideoSet_ExportConcepts_h

#include "Persistency/SimilarityTableSetRepository.h"
#include "Core/VideoSet/Reporter.h"
#include "Core/VideoSet/Segmentation.h"
#include "Core/VideoSet/Mpeg7DocWrite.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


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

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

    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;
        }
        int frameRateNum = (src) ? src->FrameRateNum() : 0;
        int frameRateDen = (src) ? src->FrameRateDen() : 0;
        String subDir = (mFeatureDef.Empty()) ? "" : mFeatureDef.AsString();
        SimilarityTableSetLocator loc(vs->GetLocator(), false, mWalkType,
                                      mConceptFileName, mModel, subDir,
                                      vs->GetContainer(fileId));
        SimilarityTableSet* simSet = SimilarityTableSetRepository().Get(loc);

        for (int c=0 ; c<simSet->NrTables() ; c++)
        {
            String concept = simSet->GetName(c);
            ILOG_INFO("doing concept " << concept);
            ILOG_NDC_PUSH(concept);
            SimilarityTableSet::SimTableType* table = simSet->GetSimTable(c);
            String fileName = vs->GetFile(fileId);
            ILOG_DEBUG("doing file " << fileName);
            String id = fileName;
            int nrFrames = mSegmentation->GetNrFramesVideo(fileId);
            Mpeg7Doc mp7(id, fileName, nrFrames, mFractions, false,
                         mConceptFileName, concept, frameRateNum, frameRateDen);
            mp7.SetAnnoIsConcept();
            for (int i=0 ; i<table->Size() ; i++)
            {
                Quid quid = simSet->GetQuidTable()->Get1(i);
                if (QuidClass(quid) != QUID_CLASS_FRAME)
                    ILOG_ERROR("Quid is not a frame");
                int frameNr = QuidId(quid);
                String name = "frame_" + MakeString(frameNr);
                double sim = table->Get1(i);
                mp7.AddAnnotation(concept, name, frameNr, frameNr, 1, sim);
            }
            Core::VideoSet::Mpeg7DocWrite(&mp7, vs, fileId);
            ILOG_NDC_POP;
        }
        delete simSet;
    }

private:

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

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(ExportConcepts, Impala.Core.VideoSet);

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

#endif

---