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

---

CombineConcepts.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_CombineConcepts_h
#define Impala_Core_VideoSet_CombineConcepts_h

#include "Core/Column/AddAssign.h"
#include "Core/Column/MulAssign.h"
#include "Core/Column/DivAssign.h"
#include "Persistency/SimilarityTableSetRepository.h"
#include "Core/VideoSet/Reporter.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


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

    CombineConcepts(Reporter* reporter, CmdOptions& options)
    {
        if (options.GetNrArg() < 7)
        {
            ILOG_ERROR("Missing argument");
            return;
        }
        mConceptFileName = options.GetArg(2);
        mModel = options.GetArg(3);
        mResultName = options.GetArg(4);
        mCombi = options.GetArg(5);
        if ((mCombi != "mul") && (mCombi != "avg"))
        {
            ILOG_ERROR("Unknown combination method : " << mCombi);
            return;
        }
        for (int i=6 ; i<options.GetNrArg() ; i++)
            mFeatureDefs.push_back(options.GetArg(i));
    }

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

    virtual void
    HandleDoneFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        SimilarityTableSetLocator loc(vs->GetLocator(), false, mWalkType,
                                      mConceptFileName, mModel, mFeatureDefs[0],
                                      vs->GetContainer(fileId));
        SimilarityTableSet* combinedSet = SimilarityTableSetRepository().Get(loc);
        for (int i=1 ; i<mFeatureDefs.size() ; i++)
        {
            loc.SetFeature(mFeatureDefs[i]);
            SimilarityTableSet* simSet = SimilarityTableSetRepository().Get(loc);
            for (int t=0 ; t<simSet->NrTables() ; t++)
            {
                if (mCombi == "mul")
                    MulAssign(combinedSet->GetSimTable(t)->GetColumn1(),
                              simSet->GetSimTable(t)->GetColumn1(), 0, -1);
                else
                    AddAssign(combinedSet->GetSimTable(t)->GetColumn1(),
                              simSet->GetSimTable(t)->GetColumn1(), 0, -1);
            }
            delete simSet;
        }
        if (mCombi == "avg")
            for (int t=0 ; t<combinedSet->NrTables() ; t++)
                DivAssign(combinedSet->GetSimTable(t)->GetColumn1(),
                          mFeatureDefs.size(), 0, -1);
        combinedSet->ComputeRanks(true);

        loc.SetFeature(mResultName);
        SimilarityTableSetRepository().Add(loc, combinedSet);
        delete combinedSet;
    }

private:

    String              mConceptFileName;
    String              mModel;
    String              mResultName;
    String              mCombi;
    std::vector<String> mFeatureDefs;
    String              mWalkType;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(CombineConcepts, Impala.Core.VideoSet);

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

#endif

---