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

---

ImportConcepts.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_ImportConcepts_h
#define Impala_Core_VideoSet_ImportConcepts_h

#include "Util/DatabaseReadString.h"
#include "Util/DatabaseWriteString.h"
#include "Core/Table/SimilarityTableSet.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;

    ImportConcepts(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        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)
    {
        String fName = vs->GetFilePathAnnotation(mConceptSet, false, false);
        if (fName.empty())
        {
            ILOG_ERROR("Unable to read conceptSet");
            return;
        }
        std::vector<String> names;
        Util::DatabaseReadString(std::back_inserter(names), fName,
                                 vs->GetDatabase(), true);
        mSimSet = new SimilarityTableSet(names, 100);
    }

    virtual void
    HandleNewFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        String vidName = vs->GetFile(fileId);
        for (int c=0 ; c<mSimSet->NrTables() ; c++)
        {
            String concept = mSimSet->GetName(c);
            SimTableType* table = mSimSet->GetSimTable(c);
            String fName;
            fName = vs->GetFilePathMpeg7Concept(fileId, mConceptSet,
                                                concept + ".xml", false, true);
            if (fName.empty())
            {
                continue;
            }
            Mpeg7Doc mp7(fName, vs->GetDatabase());
            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);
                }
            }
        }
    }

    virtual void
    HandleDoneWalk(VideoSet* vs)
    {
        mSimSet->ComputeRanks(true);
        mSimSet->Save(vs, mConceptSet, mModel, mFeature, true);
        delete mSimSet;
    }

private:

    Reporter*           mReporter;
    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

---