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

---

ImportConcepts.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_ImportConcepts_h
#define Impala_Core_VideoSet_ImportConcepts_h

#include "Persistency/SimilarityTableSetRepository.h"
#include "Persistency/Mpeg7DocRepository.h"
#include "Core/VideoSet/Reporter.h"
#include "Core/VideoSet/Segmentation.h"
#include "Core/VideoSet/Keyframes.h"
#include "Core/VideoSet/Mpeg7Doc.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class ImportConcepts : public Listener
{
public:
    typedef Table::SimilarityTableSet SimilarityTableSet;
    typedef Table::SimilarityTableSet::SimTableType SimTableType;
    typedef Persistency::SimilarityTableSetLocator SimilarityTableSetLocator;
    typedef Persistency::SimilarityTableSetRepository SimilarityTableSetRepository;

    ImportConcepts(Reporter* reporter, CmdOptions& options)
    {
        if (options.GetNrArg() < 5)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        mConceptSet = options.GetArg(2);
        mModel = options.GetArg(3);
        mFeature = options.GetArg(4);
        mSimSet = 0;
    }

    virtual void
    HandleNewWalk(VideoSet* vs)
    {
        Persistency::KeywordListLocator loc(vs->GetLocator(), mConceptSet);
        Table::KeywordList names =
            *(Persistency::KeywordListRepository().Get(loc));
        mSimSet = new SimilarityTableSet(names, 100);
    }

    virtual void
    HandleNewFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        String vidName = vs->GetFile(fileId);
        int frameRateNum = (src) ? src->FrameRateNum() : 0;
        int frameRateDen = (src) ? src->FrameRateDen() : 0;
        for (int c=0 ; c<mSimSet->NrTables() ; c++)
        {
            String concept = mSimSet->GetName(c);
            SimTableType* table = mSimSet->GetSimTable(c);            
            String container = mConceptSet + "/" + vs->GetContainerFile(fileId);
            typedef Persistency::Mpeg7DocLocator MP7Loc;
            MP7Loc loc(vs->GetLocator(), MP7Loc::MPEG7_CONCEPT, container,
                       concept + ".xml");
            if (!Persistency::Mpeg7DocRepository().Exists(loc))
                continue;

            Mpeg7Doc* mp7 = Persistency::Mpeg7DocRepository().Get(loc);
            for (int s=0 ; s<mp7->NrShots() ; s++)
            {
                double sim = mp7->Confidence(s);
                table->Add(sim);
            }
            if (c == 0)
            {
                Table::QuidTable* qTable = mSimSet->GetQuidTable();
                for (int s=0 ; s<mp7->NrShots() ; s++)
                {
                    Quid quid = vs->GetQuidFrame(fileId, mp7->StartFrame(s));
                    qTable->Add(quid);
                }
            }
            delete mp7;
        }
    }

    virtual void
    HandleDoneWalk(VideoSet* vs)
    {
        mSimSet->ComputeRanks(true);
        SimilarityTableSetLocator loc(vs->GetLocator(), true, "", mConceptSet,
                                      mModel, mFeature, "");
        SimilarityTableSetRepository().Add(loc, mSimSet);
        delete mSimSet;
    }

private:

    String              mConceptSet;
    String              mModel;
    String              mFeature;
    SimilarityTableSet* mSimSet;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(ImportConcepts, Impala.Core.VideoSet);

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

#endif

---